yuv420转yuv422使用neno加速的示例代码

2024-05-29 18:36

本文主要是介绍yuv420转yuv422使用neno加速的示例代码,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

一、使用none实现yuv420快速转换为yuv422

#include <arm_neon.h>

static vs_int32_t yuv420_yuv422_conver(vs_uint8_t *p_yuv420_y, vs_uint8_t *p_yuv420_uv, vs_uint32_t yuv420_y_stride, vs_uint32_t yuv420_uv_stride, vs_int32_t width, vs_int32_t height, vs_uint8_t* p_yuv422_virt_addr)
{
    vs_int32_t w = 0, h = 0, remain = 0;
    vs_uint8_t* p_yuv422_out = NULL;
    uint8x8_t u8rslt;
    uint8x8x2_t u8yuvdata;
    uint8x8x2_t u8trans_idx;
    vs_int32_t p_yuv422_stride = (width << 1);
    const vs_uint8_t data_idx[2][8] = { { 0, 8, 1, 9, 2, 10, 3, 11 },
                                         { 4, 12, 5, 13, 6, 14, 7, 15 } };

    //width must be even
    if (width % 2 != 0) {
        VS_UVC_LOG_E("width[%u] must be even!\n", width);
        return VS_FAILED;
    }

    p_yuv422_out = p_yuv422_virt_addr;
    u8trans_idx.val[0] = vld1_u8(data_idx[0]);
    u8trans_idx.val[1] = vld1_u8(data_idx[1]);

    h = height;
    for (; h > 0; h--) {

        w = (width >> 3);
        remain = width - (w << 3);
        for (; w > 0; w--) {

            u8yuvdata.val[0] = vld1_u8(p_yuv420_y);
            u8yuvdata.val[1] = vld1_u8(p_yuv420_uv);
            u8rslt = vtbl2_u8(u8yuvdata, u8trans_idx.val[0]);
            vst1_u8(p_yuv422_out, u8rslt);

            u8rslt = vtbl2_u8(u8yuvdata, u8trans_idx.val[1]);

            vst1_u8(p_yuv422_out + 8, u8rslt);
            p_yuv420_y += 8;
            p_yuv420_uv += 8;
            p_yuv422_out += 16;
        }

        for (; remain > 0; remain -= 2) {

            p_yuv422_out[0] = p_yuv420_y[0];
            p_yuv422_out[1] = p_yuv420_uv[1];
            p_yuv422_out[2] = p_yuv420_y[1];
            p_yuv422_out[3] = p_yuv420_uv[0];
            p_yuv420_y += 2;
            p_yuv420_uv += 2;
            p_yuv422_out += 4;
        }

        if (h & 1) {
            p_yuv420_uv += (yuv420_uv_stride - width);
        } else {
            p_yuv420_uv -= width;
        }
        p_yuv420_y += (yuv420_y_stride - width);
        p_yuv422_out += (p_yuv422_stride - (width << 1));
    }

    return VS_SUCCESS;
}

static vs_int32_t frame_yuv420_yuv422_conver(vs_video_frame_s* p_yuv420_frame, vs_uint8_t* p_yuv422_virt_addr, vs_int32_t yuv422_w, vs_int32_t yuv422_h, FILE *p_yuv420_file)
{
    vs_int32_t ret = VS_SUCCESS;
    vs_uint32_t yuv420_frame_size = 0;
    vs_uint8_t *p_yuv420_virt_addr = NULL, *p_yuv420_y = NULL, *p_yuv420_uv = NULL;


    if(p_yuv420_frame->width != yuv422_w || p_yuv420_frame->height != yuv422_h) {
        VS_UVC_LOG_E("p_yuv420_frame->width != yuv422_w\n");
        return VS_FAILED;
    }

    yuv420_frame_size = p_yuv420_frame->stride[0] * p_yuv420_frame->height * 3 / 2;
    VS_UVC_LOG_I("===========p_yuv420_frame->stride[0] = %d =============\n", p_yuv420_frame->stride[0]);
    VS_UVC_LOG_I("===========p_yuv420_frame->stride[1] = %d =============\n", p_yuv420_frame->stride[1]);
    p_yuv420_virt_addr = (vs_uint8_t*)vs_mal_sys_mmap_cached(p_yuv420_frame->phys_addr[0], yuv420_frame_size);
    if (!p_yuv420_virt_addr) {
        VS_UVC_LOG_E("vs_mal_sys_mmap_cached failed\n");
        return VS_FAILED;
    }


    p_yuv420_y = p_yuv420_virt_addr;
    p_yuv420_uv = p_yuv420_y + p_yuv420_frame->stride[0] * p_yuv420_frame->height;

    ret = yuv420_yuv422_conver(p_yuv420_y, p_yuv420_uv, p_yuv420_frame->stride[0], p_yuv420_frame->stride[1], yuv422_w, yuv422_h, p_yuv422_virt_addr);
    if (ret != VS_SUCCESS) {
        VS_UVC_LOG_E("yuv420_2_yuv422_conver failed\n");
        vs_mal_sys_unmap(p_yuv420_virt_addr, yuv420_frame_size);
        return ret;
    }

    fwrite(p_yuv420_virt_addr, 1, yuv420_frame_size, p_yuv420_file);
    fflush(p_yuv420_file);

    vs_mal_sys_unmap(p_yuv420_virt_addr, yuv420_frame_size);

    return VS_SUCCESS;
}

static vs_int32_t frame_yuv420_witre_2_yuv422file(vs_video_frame_s* p_yuv420_frame)
{
    vs_int32_t ret = VS_SUCCESS;
    static vs_int32_t savefile_index = 0;
    vs_uint32_t yuv422_frame_size = 0;
    vs_uint8_t* p_yuv422_virt_addr = VS_NULL;
    FILE *p_yuv420_file = VS_NULL, *p_yuv422_file = VS_NULL;
    vs_char_t yuv420_file_name[64], yuv422_file_name[64];

    //yuv420_frame_size = p_yuv420_frame->stride[0] * p_yuv420_frame->height * 3 / 2;
    yuv422_frame_size = p_yuv420_frame->stride[0] * p_yuv420_frame->height * 2;
    p_yuv422_virt_addr = malloc(yuv422_frame_size);
    if (p_yuv422_virt_addr == VS_NULL) {
        VS_UVC_LOG_E("malloc size[%u] failed!\n", yuv422_frame_size);
        return VS_FAILED;
    }

    memset(yuv420_file_name, 0, sizeof(yuv420_file_name));
    memset(yuv422_file_name, 0, sizeof(yuv422_file_name));
    sprintf((char *)yuv420_file_name, "yuv420_idx%d.yuv", savefile_index);
    sprintf((char *)yuv422_file_name, "yuv422_idx%d.yuv", savefile_index);
    savefile_index++;
    p_yuv420_file = fopen((char *)yuv420_file_name, "wb");
    if (VS_NULL == p_yuv420_file) {
        VS_UVC_LOG_E("fopen file[%s] failed\n", yuv420_file_name);
        free(p_yuv422_virt_addr);
        return VS_FAILED;
    }
    p_yuv422_file = fopen((char *)yuv422_file_name, "wb");
    if (VS_NULL == p_yuv422_file) {
        VS_UVC_LOG_E("fopen file[%s] failed\n", yuv422_file_name);
        free(p_yuv422_virt_addr);
        fclose(p_yuv420_file);
        return VS_FAILED;
    }

    ret = frame_yuv420_yuv422_conver(p_yuv420_frame, p_yuv422_virt_addr, p_yuv420_frame->width, p_yuv420_frame->height, p_yuv420_file);
    if (ret != VS_SUCCESS) {
        VS_UVC_LOG_E("yuv420_2_yuv422_conver failed\n");
        free(p_yuv422_virt_addr);
        fclose(p_yuv420_file);
        fclose(p_yuv422_file);
        return ret;
    }

    fwrite(p_yuv422_virt_addr, 1, yuv422_frame_size, p_yuv422_file);
    fflush(p_yuv422_file);

    free(p_yuv422_virt_addr);
    fclose(p_yuv420_file);
    fclose(p_yuv422_file);

    return VS_SUCCESS;
}

二、使用none实现yuv420快速转换为nv12

static vs_int32_t yuv420_nv12_conver(vs_uint8_t* p_yuv420_virt_addr, vs_uint32_t yuv420_y_stride, vs_uint32_t yuv420_uv_stride, vs_int32_t width, vs_int32_t height, vs_uint8_t* p_nv12_virt_addr)
{
    vs_int32_t w, h;
    vs_uint8_t* p_yuv420_in = p_yuv420_virt_addr, *p_nv12_out = p_nv12_virt_addr;
    vs_int32_t remain = 0;
    uint8x16_t u8data;

    //width must be even
    if (width % 2 != 0) {
        VS_UVC_LOG_E("width[%u] must be even!\n", width);
        return VS_FAILED;
    }

    // for y
    h = height;
    for (; h > 0; h--) {
        w = (width >> 4);
        remain = width % 16;
        for (; w > 0; w--) {
            u8data = vld1q_u8(p_yuv420_in);
            vst1q_u8(p_nv12_out, u8data);
            p_nv12_out += 16;
            p_yuv420_in += 16;
        }
        for (; remain > 0; remain--) {
            *p_nv12_out = *p_yuv420_in;
            p_nv12_out++;
            p_yuv420_in++;
        }
        p_yuv420_in += (yuv420_y_stride - width);
    }

    // for yv
    h = (height >> 1);
    for (; h > 0; h--) {
        w = (width >> 4);
        remain = width % 16;
        for (; w > 0; w--) {
            u8data = vld1q_u8(p_yuv420_in);
            //u8data = vrev16q_u8(u8data);
            vst1q_u8(p_nv12_out, u8data);
            p_nv12_out += 16;
            p_yuv420_in += 16;
        }

        for (; remain > 0; remain -= 2) {
            p_nv12_out[0] = p_yuv420_in[0];
            p_nv12_out[1] = p_yuv420_in[1];
            p_nv12_out += 2;
            p_yuv420_in += 2;
        }
        p_yuv420_in += (yuv420_uv_stride - width);
    }

    return  VS_SUCCESS;
}

static vs_int32_t frame_yuv420_nv12_conver(vs_video_frame_s* p_yuv420_frame, vs_uint8_t* p_nv12_virt_addr,vs_int32_t nv12_w,vs_int32_t nv12_h, FILE *p_yuv420_file)
{
    vs_int32_t ret = VS_SUCCESS;
    vs_uint32_t yuv420_frame_size = 0;
    vs_uint8_t *p_yuv420_virt_addr = NULL;

    if (p_yuv420_frame->width != nv12_w || p_yuv420_frame->height != nv12_h) {
        VS_UVC_LOG_E("p_yuv420_frame->width != nv12_w\n");
        return 0;
    }

    yuv420_frame_size = p_yuv420_frame->stride[0] * p_yuv420_frame->height * 3 / 2;
    VS_UVC_LOG_I("===========p_yuv420_frame->stride[0] = %d =============\n", p_yuv420_frame->stride[0]);
    VS_UVC_LOG_I("===========p_yuv420_frame->stride[1] = %d =============\n", p_yuv420_frame->stride[1]);
    p_yuv420_virt_addr = (vs_uint8_t*)vs_mal_sys_mmap_cached(p_yuv420_frame->phys_addr[0], yuv420_frame_size);
    if (!p_yuv420_virt_addr) {
        VS_UVC_LOG_E(" vs_mal_sys_mmap_cached failed\n");
        return 0;
    }

    ret = yuv420_nv12_conver(p_yuv420_virt_addr, p_yuv420_frame->stride[0], p_yuv420_frame->stride[1], nv12_w, nv12_h, p_nv12_virt_addr);
    if (ret != VS_SUCCESS) {
        VS_UVC_LOG_E("yuv420_2_yuv422_conver failed\n");
        vs_mal_sys_unmap(p_yuv420_virt_addr, yuv420_frame_size);
        return ret;
    }

    fwrite(p_yuv420_virt_addr, 1, yuv420_frame_size, p_yuv420_file);
    fflush(p_yuv420_file);

    vs_mal_sys_unmap(p_yuv420_virt_addr, yuv420_frame_size);

    return VS_SUCCESS;
}

static vs_int32_t frame_yuv420_witre_2_nv12file(vs_video_frame_s* p_yuv420_frame)
{
    vs_int32_t ret = VS_SUCCESS;
    static vs_int32_t savefile_index = 0;
    vs_uint32_t nv12_frame_size = 0;
    vs_uint8_t* p_yuv422_virt_addr = VS_NULL;
    FILE *p_yuv420_file = VS_NULL, *p_nv12_file = VS_NULL;
    vs_char_t yuv420_file_name[64], nv12_file_name[64];

    //yuv420_frame_size = p_yuv420_frame->stride[0] * p_yuv420_frame->height * 3 / 2;
    nv12_frame_size = p_yuv420_frame->stride[0] * p_yuv420_frame->height * 3 / 2;
    p_yuv422_virt_addr = malloc(nv12_frame_size);
    if (p_yuv422_virt_addr == VS_NULL) {
        VS_UVC_LOG_E("malloc size[%u] failed!\n", nv12_frame_size);
        return VS_FAILED;
    }

    memset(yuv420_file_name, 0, sizeof(yuv420_file_name));
    memset(nv12_file_name, 0, sizeof(nv12_file_name));
    sprintf((char *)yuv420_file_name, "yuv420_idx%d.yuv", savefile_index);
    sprintf((char *)nv12_file_name, "nv12_idx%d.yuv", savefile_index);
    savefile_index++;
    p_yuv420_file = fopen((char *)yuv420_file_name, "wb");
    if (VS_NULL == p_yuv420_file) {
        VS_UVC_LOG_E("fopen file[%s] failed\n", yuv420_file_name);
        free(p_yuv422_virt_addr);
        return VS_FAILED;
    }
    p_nv12_file = fopen((char *)nv12_file_name, "wb");
    if (VS_NULL == p_nv12_file) {
        VS_UVC_LOG_E("fopen file[%s] failed\n", nv12_file_name);
        free(p_yuv422_virt_addr);
        fclose(p_yuv420_file);
        return VS_FAILED;
    }

    ret = frame_yuv420_nv12_conver(p_yuv420_frame, p_yuv422_virt_addr, p_yuv420_frame->width, p_yuv420_frame->height, p_yuv420_file);
    if (ret != VS_SUCCESS) {
        VS_UVC_LOG_E("yuv420_2_yuv422_conver failed\n");
        free(p_yuv422_virt_addr);
        fclose(p_yuv420_file);
        fclose(p_nv12_file);
        return ret;
    }

    fwrite(p_yuv422_virt_addr, 1, nv12_frame_size, p_nv12_file);
    fflush(p_nv12_file);

    free(p_yuv422_virt_addr);
    fclose(p_yuv420_file);
    fclose(p_nv12_file);

    return VS_SUCCESS;
}
 

这篇关于yuv420转yuv422使用neno加速的示例代码的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1014316

相关文章

Java Stream流与使用操作指南

《JavaStream流与使用操作指南》Stream不是数据结构,而是一种高级的数据处理工具,允许你以声明式的方式处理数据集合,类似于SQL语句操作数据库,本文给大家介绍JavaStream流与使用... 目录一、什么是stream流二、创建stream流1.单列集合创建stream流2.双列集合创建str

Python使用FastAPI实现大文件分片上传与断点续传功能

《Python使用FastAPI实现大文件分片上传与断点续传功能》大文件直传常遇到超时、网络抖动失败、失败后只能重传的问题,分片上传+断点续传可以把大文件拆成若干小块逐个上传,并在中断后从已完成分片继... 目录一、接口设计二、服务端实现(FastAPI)2.1 运行环境2.2 目录结构建议2.3 serv

C#实现千万数据秒级导入的代码

《C#实现千万数据秒级导入的代码》在实际开发中excel导入很常见,现代社会中很容易遇到大数据处理业务,所以本文我就给大家分享一下千万数据秒级导入怎么实现,文中有详细的代码示例供大家参考,需要的朋友可... 目录前言一、数据存储二、处理逻辑优化前代码处理逻辑优化后的代码总结前言在实际开发中excel导入很

Spring Security简介、使用与最佳实践

《SpringSecurity简介、使用与最佳实践》SpringSecurity是一个能够为基于Spring的企业应用系统提供声明式的安全访问控制解决方案的安全框架,本文给大家介绍SpringSec... 目录一、如何理解 Spring Security?—— 核心思想二、如何在 Java 项目中使用?——

SpringBoot+RustFS 实现文件切片极速上传的实例代码

《SpringBoot+RustFS实现文件切片极速上传的实例代码》本文介绍利用SpringBoot和RustFS构建高性能文件切片上传系统,实现大文件秒传、断点续传和分片上传等功能,具有一定的参考... 目录一、为什么选择 RustFS + SpringBoot?二、环境准备与部署2.1 安装 RustF

springboot中使用okhttp3的小结

《springboot中使用okhttp3的小结》OkHttp3是一个JavaHTTP客户端,可以处理各种请求类型,比如GET、POST、PUT等,并且支持高效的HTTP连接池、请求和响应缓存、以及异... 在 Spring Boot 项目中使用 OkHttp3 进行 HTTP 请求是一个高效且流行的方式。

Python实现Excel批量样式修改器(附完整代码)

《Python实现Excel批量样式修改器(附完整代码)》这篇文章主要为大家详细介绍了如何使用Python实现一个Excel批量样式修改器,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一... 目录前言功能特性核心功能界面特性系统要求安装说明使用指南基本操作流程高级功能技术实现核心技术栈关键函

Java使用Javassist动态生成HelloWorld类

《Java使用Javassist动态生成HelloWorld类》Javassist是一个非常强大的字节码操作和定义库,它允许开发者在运行时创建新的类或者修改现有的类,本文将简单介绍如何使用Javass... 目录1. Javassist简介2. 环境准备3. 动态生成HelloWorld类3.1 创建CtC

使用Python批量将.ncm格式的音频文件转换为.mp3格式的实战详解

《使用Python批量将.ncm格式的音频文件转换为.mp3格式的实战详解》本文详细介绍了如何使用Python通过ncmdump工具批量将.ncm音频转换为.mp3的步骤,包括安装、配置ffmpeg环... 目录1. 前言2. 安装 ncmdump3. 实现 .ncm 转 .mp34. 执行过程5. 执行结

Java使用jar命令配置服务器端口的完整指南

《Java使用jar命令配置服务器端口的完整指南》本文将详细介绍如何使用java-jar命令启动应用,并重点讲解如何配置服务器端口,同时提供一个实用的Web工具来简化这一过程,希望对大家有所帮助... 目录1. Java Jar文件简介1.1 什么是Jar文件1.2 创建可执行Jar文件2. 使用java