FFmpeg 调用 Android MediaCodec 进行硬解码(附源码)

FFmpeg 在 3.1 版本之后支持调用平台硬件进行解码,也就是说可以通过 FFmpeg 的 C 代码去调用 Android 上的 MediaCodec 了。

在官网上有对应说明,地址如下:

https://trac.ffmpeg.org/wiki/HWAccelIntro

image

从图中可以看到,不仅仅是 Android 上支持 MediaCodec,iOS 上也支持 VideoToolbox,连 Windows 上的 Direct3D 11 都有支持了。

注意:Android MediaCodec 目前仅支持解码,还不支持编码呢。

不过,为了验证是否可行,做个简单的演示,最后会有完整的的代码给出。

首先是 FFmpeg 的编译。它的编译有很多开关选项,要确保打开了 mediacodec 相关的选项,具体如下:

--enable-mediacodec
--enable-decoder=h264_mediacodec
--enable-decoder=hevc_mediacodec
--enable-decoder=mpeg4_mediacodec
--enable-hwaccel=h264_mediacodec

可以看出 mediacodec 支持的编码格式有 h264、hevc、mpeg4 三种可选,不在范围内的就还是考虑软解吧。

关于如何编译,就不详细阐述了,后面再专门写一篇来介绍

编译出对应的 so 之后,可以打印一下 AVCodec 支持的格式列表,看看有没有 mediacodec 。

具体代码如下:

char info[40000] = {0};

AVCodec *c_temp = av_codec_next(NULL);
while (c_temp != NULL) {
    if (c_temp->decode != NULL) {
        sprintf(info, "%s[Dec]", info);
    } else {
        sprintf(info, "%s[Enc]", info);
    }

    switch (c_temp->type) {
        case AVMEDIA_TYPE_VIDEO:
            sprintf(info, "%s[Video]", info);
            break;
        case AVMEDIA_TYPE_AUDIO:
            sprintf(info, "%s[Audio]", info);
            break;
        default:
            sprintf(info, "%s[Other]", info);
            break;
    }
    sprintf(info, "%s %10s\n", info, c_temp->name);
    c_temp = c_temp->next;
}

通过 AVCodec 的 next 指针进行遍历,然后打印出结果,看到下面的内容说明编译成功了。

image

支持的格式里面已经有了 h264_mediacodec 和 mpeg4_mediacodec 了。


接下来就进行解码了。关于 FFmpeg 解码的 API 调用,在公众号以前发布的文章中说过多次,就不详细讲解流程了,简单概况一下:

  1. 首先通过 avformat_open_input 方法打开文件,得到 AVFormatContext 。
  2. 然后通过 avformat_find_stream_info 查找文件的视频流信息。
  3. 得到文件相关信息和视频流信息,主要还是为了得到编码格式信息,然后好找到对应的解码器。也可以通过 avcodec_find_decoder_by_name 方法直接找具体的解码器。
  4. 有了解码器就可以创建解码上下文 AVCodecContext,并通过 avcodec_open2 方法打开解码器
  5. 然后通过 av_read_frame 读取文件的内容好进行下一步的解码。
  6. 接下来就是熟悉的 avcodec_send_packet 发送给解码器,avcodec_receive_frame 从解码器取回解码后的数据。

重点讲解一下调用硬件解码和普通解码的一些区别:

第一步是要在 so 加载的 JNI_OnLoad 方法中将 JavaVM 设置给 FFmpeg 。

jint JNI_OnLoad(JavaVM *vm, void *res) {
    av_jni_set_java_vm(vm, 0);
    return JNI_VERSION_1_4;
}

缺少这一步就不能反射调用 Java 方法了。


接下来还是判断硬件解码类型支不支持,上面是通过 AVCodec 来判断的,实际上 FFmpeg 都给出了硬件类型的定义,在 AVHWDeviceType 枚举变量中。

enum AVHWDeviceType {
    AV_HWDEVICE_TYPE_NONE,
    AV_HWDEVICE_TYPE_VDPAU,
    AV_HWDEVICE_TYPE_CUDA,
    AV_HWDEVICE_TYPE_VAAPI,
    AV_HWDEVICE_TYPE_DXVA2,
    AV_HWDEVICE_TYPE_QSV,
    AV_HWDEVICE_TYPE_VIDEOTOOLBOX,
    AV_HWDEVICE_TYPE_D3D11VA,
    AV_HWDEVICE_TYPE_DRM,
    AV_HWDEVICE_TYPE_OPENCL,
    AV_HWDEVICE_TYPE_MEDIACODEC,
    AV_HWDEVICE_TYPE_VULKAN,
};

通过 av_hwdevice_get_type_name 方法可以将这些枚举值转换成对应的字符串,比如 AV_HWDEVICE_TYPE_MEDIACODEC 对应的字符串就是 mediacodec ,其实在源码里面也是有的:

static const char *const hw_type_names[] = {
    [AV_HWDEVICE_TYPE_CUDA]   = "cuda",
    [AV_HWDEVICE_TYPE_DRM]    = "drm",
    [AV_HWDEVICE_TYPE_DXVA2]  = "dxva2",
    [AV_HWDEVICE_TYPE_D3D11VA] = "d3d11va",
    [AV_HWDEVICE_TYPE_OPENCL] = "opencl",
    [AV_HWDEVICE_TYPE_QSV]    = "qsv",
    [AV_HWDEVICE_TYPE_VAAPI]  = "vaapi",
    [AV_HWDEVICE_TYPE_VDPAU]  = "vdpau",
    [AV_HWDEVICE_TYPE_VIDEOTOOLBOX] = "videotoolbox",
    [AV_HWDEVICE_TYPE_MEDIACODEC] = "mediacodec",
    [AV_HWDEVICE_TYPE_VULKAN] = "vulkan",
};

和遍历 AVCodec 一样,也要遍历 FFmpeg 是否支持 mediacodec 。

type = av_hwdevice_find_type_by_name(mediacodec);
if (type == AV_HWDEVICE_TYPE_NONE) {
    LOGE("Device type %s is not supported.\n", mediacodec);
    LOGE("Available device types:");
    while((type = av_hwdevice_iterate_types(type)) != AV_HWDEVICE_TYPE_NONE)
        LOGE(" %s", av_hwdevice_get_type_name(type));
    LOGE("\n");
    return -1;
}

确定支持 mediacodec ,那么解码就可以用了。前面提到,获取文件信息主要是为了打开解码器的,但比如文件编码格式的 H.264 ,而支持 H.264 的解码器除了软解,还有 mediacodec 要怎么选择呢?

为了方便,直接 avcodec_find_decoder_by_name 找到 mediacodec 的解码器就行。

if (!(decoder = avcodec_find_decoder_by_name("h264_mediacodec"))) {
    LOGE("avcodec_find_decoder_by_name failed.\n");
    return -1;
}

找到解码器之后,还要得到该解码器的一些配置信息,比如解码出的格式是什么样子的?mediacodec 解码就是 NV21 这种。

for (i = 0;; i++) {
    // 解码器的配置
    const AVCodecHWConfig *config = avcodec_get_hw_config(decoder, i);
    if (!config) {
        LOGE("Decoder %s does not support device type %s.\n",
                decoder->name, av_hwdevice_get_type_name(type));
        return -1;
    }
    if (config->methods & AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX &&
        config->device_type == type) {
        // 硬解的格式
        hw_pix_fmt = config->pix_fmt;
        break;
    }
}

目前 mediacodec 解码还只有 buffer 模式,没有直接解纹理的那种。

接下来就是给解码上下文 AVCodecContext 添加一些硬件解码的上下文。

static int hw_decoder_init(AVCodecContext *ctx, const enum AVHWDeviceType type)
{
    int err = 0;
    if ((err = av_hwdevice_ctx_create(&hw_device_ctx, type,
                                      NULL, NULL, 0)) < 0) {
        LOGE("Failed to create specified HW device.\n");
        return err;
    }
    // 硬解解码的上下文
    ctx->hw_device_ctx = av_buffer_ref(hw_device_ctx);
    return err;
}

完成了这一系列操作之后,就是正常的解码了,拿到解码后的 AVFrame 内容。

如果 AVFrame 格式和硬件解码的配置格式一样,那么要用 av_hwframe_transfer_data 方法将它做一下转换,转成正常的 YUV 格式。

if (frame->format == hw_pix_fmt) {
    /* retrieve data from GPU to CPU */
    if ((ret = av_hwframe_transfer_data(sw_frame, frame, 0)) < 0) {
        LOGE("Error transferring the data to system memory\n");
        goto fail;
    }
    tmp_frame = sw_frame;
} else
    tmp_frame = frame;

等完成这一些操作之后,就已经解码成功了,实际运行也是 OK 的。

获取完整源码的话,可以关注微信公众号:音视频开发进阶,回复 1019 获取下载地址。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,905评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,140评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,791评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,483评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,476评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,516评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,905评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,560评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,778评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,557评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,635评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,338评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,925评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,898评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,142评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,818评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,347评论 2 342

推荐阅读更多精彩内容