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

相关文章

中文分词jieba库的使用与实景应用(一)

知识星球:https://articles.zsxq.com/id_fxvgc803qmr2.html 目录 一.定义: 精确模式(默认模式): 全模式: 搜索引擎模式: paddle 模式(基于深度学习的分词模式): 二 自定义词典 三.文本解析   调整词出现的频率 四. 关键词提取 A. 基于TF-IDF算法的关键词提取 B. 基于TextRank算法的关键词提取

使用SecondaryNameNode恢复NameNode的数据

1)需求: NameNode进程挂了并且存储的数据也丢失了,如何恢复NameNode 此种方式恢复的数据可能存在小部分数据的丢失。 2)故障模拟 (1)kill -9 NameNode进程 [lytfly@hadoop102 current]$ kill -9 19886 (2)删除NameNode存储的数据(/opt/module/hadoop-3.1.4/data/tmp/dfs/na

Hadoop数据压缩使用介绍

一、压缩原则 (1)运算密集型的Job,少用压缩 (2)IO密集型的Job,多用压缩 二、压缩算法比较 三、压缩位置选择 四、压缩参数配置 1)为了支持多种压缩/解压缩算法,Hadoop引入了编码/解码器 2)要在Hadoop中启用压缩,可以配置如下参数

Makefile简明使用教程

文章目录 规则makefile文件的基本语法:加在命令前的特殊符号:.PHONY伪目标: Makefilev1 直观写法v2 加上中间过程v3 伪目标v4 变量 make 选项-f-n-C Make 是一种流行的构建工具,常用于将源代码转换成可执行文件或者其他形式的输出文件(如库文件、文档等)。Make 可以自动化地执行编译、链接等一系列操作。 规则 makefile文件

使用opencv优化图片(画面变清晰)

文章目录 需求影响照片清晰度的因素 实现降噪测试代码 锐化空间锐化Unsharp Masking频率域锐化对比测试 对比度增强常用算法对比测试 需求 对图像进行优化,使其看起来更清晰,同时保持尺寸不变,通常涉及到图像处理技术如锐化、降噪、对比度增强等 影响照片清晰度的因素 影响照片清晰度的因素有很多,主要可以从以下几个方面来分析 1. 拍摄设备 相机传感器:相机传

活用c4d官方开发文档查询代码

当你问AI助手比如豆包,如何用python禁止掉xpresso标签时候,它会提示到 这时候要用到两个东西。https://developers.maxon.net/论坛搜索和开发文档 比如这里我就在官方找到正确的id描述 然后我就把参数标签换过来

pdfmake生成pdf的使用

实际项目中有时会有根据填写的表单数据或者其他格式的数据,将数据自动填充到pdf文件中根据固定模板生成pdf文件的需求 文章目录 利用pdfmake生成pdf文件1.下载安装pdfmake第三方包2.封装生成pdf文件的共用配置3.生成pdf文件的文件模板内容4.调用方法生成pdf 利用pdfmake生成pdf文件 1.下载安装pdfmake第三方包 npm i pdfma

零基础学习Redis(10) -- zset类型命令使用

zset是有序集合,内部除了存储元素外,还会存储一个score,存储在zset中的元素会按照score的大小升序排列,不同元素的score可以重复,score相同的元素会按照元素的字典序排列。 1. zset常用命令 1.1 zadd  zadd key [NX | XX] [GT | LT]   [CH] [INCR] score member [score member ...]

poj 1258 Agri-Net(最小生成树模板代码)

感觉用这题来当模板更适合。 题意就是给你邻接矩阵求最小生成树啦。~ prim代码:效率很高。172k...0ms。 #include<stdio.h>#include<algorithm>using namespace std;const int MaxN = 101;const int INF = 0x3f3f3f3f;int g[MaxN][MaxN];int n

git使用的说明总结

Git使用说明 下载安装(下载地址) macOS: Git - Downloading macOS Windows: Git - Downloading Windows Linux/Unix: Git (git-scm.com) 创建新仓库 本地创建新仓库:创建新文件夹,进入文件夹目录,执行指令 git init ,用以创建新的git 克隆仓库 执行指令用以创建一个本地仓库的