YUV420转RGBA之使用opencv

前面在《YUV_420_888介绍及YUV420转RGBA》一文中介绍了YUV420的转换，并提供了自己写的转换代码。但是实际项目中一般不会自己手写代码逐个像素去转换，因为这样转换的速度比较慢。通常我们可以使用opencv或者libyuv来进行转换。本文就介绍使用opencv进行转换。

YUV420转RGBA系列共三篇：

本文是其中的第二篇。

1. 下载opencv

https://github.com/opencv/opencv/releases
可以下载opencv源码自行编译，也可以选择编译好的opencv-xxx-android-sdk。这里只是简单说明opencv的用法，就直接使用了opencv-3.4.10-android-sdk。

2. 导入opencv到Android Studio

导入opencv有两种方式：

  • 直接使用opencv提供的Java版本 API
  • 导入头文件和库文件，自己编写JNI调用opencv C/C++版本 API
    这里选择第二种，自己编写JNI调用opencv C/C++版本 API。

2.1 导入头文件

在项目的app/src/main/cpp/下新建include文件夹。解压opencv-3.4.10-android-sdk压缩包，将～/opencv-3.4.10-android-sdk/OpenCV-android-sdk/sdk/native/jni/include/下面的头文件拷贝到项目的app/src/main/cpp/include下：


1.png
2.png

2.2 导入库文件

将~/sw/opencv-3.4.10-android-sdk/OpenCV-android-sdk/sdk/native/libs下的库文件拷贝项目的app/src/main/jniLibs下：


4.png

3.png

2.3 修改CMakeLists.txt

include_directories(${CMAKE_SOURCE_DIR}/include)
find_library(log-lib log)

add_library(opencv3 SHARED IMPORTED)
set_target_properties(
        opencv3
        PROPERTIES
        IMPORTED_LOCATION ${CMAKE_SOURCE_DIR}/../jniLibs/${ANDROID_ABI}/libopencv_java3.so
)

add_library(
        OpencvUtils
        SHARED
        opencv.cpp
        opencv_utils.cpp
)
target_link_libraries(OpencvUtils opencv3 ${log-lib})

修改完成后同步一下项目，接下来我们只要在opencv.cpp和 opencv_utils.cpp中编码调用opencv 的API就可以了。

3. 使用opencv将YUV420转换成RGBA

opencv.cpp：

#include <jni.h>
#include <string>
#include "opencv_utils.h"

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_I420ToRGBA(JNIEnv *env, jclass clazz,
                                           jbyteArray src, jbyteArray dst,
                                           jint width, jint height) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    jbyte *_dst = env->GetByteArrayElements(dst, nullptr);
    opencvI420ToRGBA(reinterpret_cast<unsigned char *>(_src),
                     reinterpret_cast<unsigned char *>(_dst), width, height);
    env->ReleaseByteArrayElements(src, _src, JNI_ABORT);
    env->ReleaseByteArrayElements(dst, _dst, 0);
}

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_YV12ToRGBA(JNIEnv *env, jclass clazz,
                                           jbyteArray src, jbyteArray dst,
                                           jint width, jint height) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    jbyte *_dst = env->GetByteArrayElements(dst, nullptr);
    opencvYV12ToRGBA(reinterpret_cast<unsigned char *>(_src),
                     reinterpret_cast<unsigned char *>(_dst), width, height);
    env->ReleaseByteArrayElements(src, _src, JNI_ABORT);
    env->ReleaseByteArrayElements(dst, _dst, 0);
}

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_NV12ToRGBA(JNIEnv *env, jclass clazz,
                                           jbyteArray src, jbyteArray dst,
                                           jint width, jint height) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    jbyte *_dst = env->GetByteArrayElements(dst, nullptr);
    opencvNV12ToRGBA(reinterpret_cast<unsigned char *>(_src),
                     reinterpret_cast<unsigned char *>(_dst), width, height);
    env->ReleaseByteArrayElements(src, _src, JNI_ABORT);
    env->ReleaseByteArrayElements(dst, _dst, 0);
}

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_NV21ToRGBA(JNIEnv *env, jclass clazz,
                                           jbyteArray src, jbyteArray dst,
                                           jint width, jint height) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    jbyte *_dst = env->GetByteArrayElements(dst, nullptr);
    opencvNV21ToRGBA(reinterpret_cast<unsigned char *>(_src),
                     reinterpret_cast<unsigned char *>(_dst), width, height);
    env->ReleaseByteArrayElements(src, _src, JNI_ABORT);
    env->ReleaseByteArrayElements(dst, _dst, 0);
}

opencv_utils.h:

#ifndef OPENCV_UTILS_H
#define OPENCV_UTILS_H

#ifdef __cplusplus
extern "C" {
#endif

void opencvI420ToRGBA(unsigned char *src, unsigned char *dst, int width, int height);

void opencvYV12ToRGBA(unsigned char *src, unsigned char *dst, int width, int height);

void opencvNV12ToRGBA(unsigned char *src, unsigned char *dst, int width, int height);

void opencvNV21ToRGBA(unsigned char *src, unsigned char *dst, int width, int height);

#ifdef __cplusplus
}
#endif
#endif //OPENCV_UTILS_H

opencv_utils.cpp:

#include <stdint.h>
#include <opencv2/opencv.hpp>
#include <opencv2/core/core.hpp>

#include "logger.h"
#include "opencv_utils.h"

using namespace std;
using namespace cv;

void opencvI420ToRGBA(unsigned char *src, unsigned char *dst, int width, int height) {
    Mat srcImg(height * 3 / 2, width, CV_8UC1, src);
    Mat dstImg(height, width, CV_8UC4, dst);
    cvtColor(srcImg, dstImg, CV_YUV2RGBA_I420);
}

void opencvYV12ToRGBA(unsigned char *src, unsigned char *dst, int width, int height) {
    Mat srcImg(height * 3 / 2, width, CV_8UC1, src);
    Mat dstImg(height, width, CV_8UC4, dst);
    cvtColor(srcImg, dstImg, CV_YUV2RGBA_YV12);
}

void opencvNV12ToRGBA(unsigned char *src, unsigned char *dst, int width, int height) {
    Mat srcImg(height * 3 / 2, width, CV_8UC1, src);
    Mat dstImg(height, width, CV_8UC4, dst);
    cvtColor(srcImg, dstImg, CV_YUV2RGBA_NV12);
}

void opencvNV21ToRGBA(unsigned char *src, unsigned char *dst, int width, int height) {
    Mat srcImg(height * 3 / 2, width, CV_8UC1, src);
    Mat dstImg(height, width, CV_8UC4, dst);
    cvtColor(srcImg, dstImg, CV_YUV2RGBA_NV21);
}

4. 使用opencv旋转RGBA和YUV420P

opencv.cpp：

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_rotateRGB(JNIEnv *env, jclass clazz, jbyteArray src,
                                          jint width, jint height, jfloat degree) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    opencvRotateRGB(reinterpret_cast<unsigned char *>(_src), width, height, degree);
    env->ReleaseByteArrayElements(src, _src, 0);
}

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_rotateRGBA(JNIEnv *env, jclass clazz, jbyteArray src,
                                           jint width, jint height, jfloat degree) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    opencvRotateRGBA(reinterpret_cast<unsigned char *>(_src), width, height, degree);
    env->ReleaseByteArrayElements(src, _src, 0);
}

extern "C"
JNIEXPORT void JNICALL
Java_com_qxt_yuv420_OpencvUtils_rotateYUV420P(JNIEnv *env, jclass clazz, jbyteArray src,
                                              jint width, jint height, jfloat degree) {
    jbyte *_src = env->GetByteArrayElements(src, nullptr);
    opencvRotateYUV420P(reinterpret_cast<unsigned char *>(_src), width, height, degree);
    env->ReleaseByteArrayElements(src, _src, 0);
}

opencv_utils.h:

void opencvRotateRGB(unsigned char *src, int width, int height, float degree);

void opencvRotateRGBA(unsigned char *src, int width, int height, float degree);

void opencvRotateYUV420P(unsigned char *src, int width, int height, float degree);

opencv_utils.cpp:

static inline void rotate(Mat srcImg, float degree, int stride) {
    Mat tmpImg;
    Mat finalImg;
    if (degree == 90.0f /*ROTATE_90_CLOCKWISE*/) {
        transpose(srcImg, tmpImg);
        flip(tmpImg, finalImg, 1);
    } else if (degree == 180.0f /*ROTATE_180*/) {
        flip(srcImg, finalImg, -1);
    } else if (degree == 270.0f /*ROTATE_90_COUNTERCLOCKWISE*/) {
        transpose(srcImg, tmpImg);
        flip(tmpImg, finalImg, 0);
    } else {
        return;
    }
    memcpy(srcImg.data, finalImg.data,
           static_cast<size_t>(finalImg.rows * finalImg.cols * stride));
    if (!tmpImg.empty()) {
        tmpImg.release();
    }

    if (!finalImg.empty()) {
        finalImg.release();
    }
}

void opencvRotateRGB(unsigned char *src, int width, int height, float degree) {
    Mat srcImg(height, width, CV_8UC3, src);
    rotate(srcImg, degree, 3);
}

void opencvRotateRGBA(unsigned char *src, int width, int height, float degree) {
    Mat srcImg(height, width, CV_8UC4, src);
    rotate(srcImg, degree, 4);
}

void opencvRotateYUV420P(unsigned char *src, int width, int height, float degree) {
    int uvWidth = width >> 1;
    int uvHeight = height >> 1;
    unsigned char *pY = src;
    unsigned char *pU = src + width * height;
    unsigned char *pV = src + width * height * 5 / 4;
    Mat y(height, width, CV_8UC1, pY);
    rotate(y, degree, 1);
    Mat u(uvHeight, uvWidth, CV_8UC1, pU);
    rotate(u, degree, 1);
    Mat v(uvHeight, uvWidth, CV_8UC1, pV);
    rotate(v, degree, 1);
}

本文中的代码已经上传到github：https://github.com/qiuxintai/YUV420Converter

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 214,504评论 6赞 496
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 91,434评论 3赞 389
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 160,089评论 0赞 349
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 57,378评论 1赞 288
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 66,472评论 6赞 386
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 50,506评论 1赞 292
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 39,519评论 3赞 413
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 38,292评论 0赞 270
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 44,738评论 1赞 307
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,022评论 2赞 329
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,194评论 1赞 342
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 34,873评论 5赞 338
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 40,536评论 3赞 322
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,162评论 0赞 21
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,413评论 1赞 268
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 47,075评论 2赞 365
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 44,080评论 2赞 352