爆裂吧现实

不想死就早点睡

0%

给FFmpeg加上MediaCodec和线程支持

Posted on Edited on In ,

之前一篇文章，我在 Android 上用 FFmpeg 来压缩视频 (文章链接)，对于 exit_program 的处理，是直接改成 return 的。稍微处理不好一点（因为 c 没有异常），程序就会 creash。还有就是在 Android 上，其实可以使用 MediaCodec 来硬解码视频的。本文讲述怎么解决上面两个问题。

一、使用线程来处理 exit(1)

FFmpeg 源码内，只要出现了一点问题，比如：文件找不到啊、解码器找不到啊、参数不合法啊。。。等等，就会调用 exit_program(int)，这个函数，此函数内又会调用 exit(int) 方法来结束当前进程。之前的做饭是把 exit(1) 改成了 return 1，这样虽然能解决 exit(1) 退出程序的问题，但是 exit_program(int) 这个函数调用的地方太多了！不好处理。

解决办法是使用线程来跑 FFmpeg 的命令，然后在 exit_program(int) 函数内调用 pthread_exit(NULL) 来退出线程，完美！

在前一篇文章工程的基础上 新建 ffmpeg_thread.c 和 ffmpeg_thread.h

ffmpeg_thread.h

1
2
3
4
5
6
7
8
9
10
11
12
#ifndef ANDROID_FFMPEG_THREAD_H
#define ANDROID_FFMPEG_THREAD_H

#include "ffmpeg.h"

// 在子线程跑
int ffmpeg_thread_run_command(int argc, char **argv);

// 结束 ffmepeg 线程，并设置结果 ret
void ffmpeg_thread_exit(int ret);

#endif //ANDROID_FFMPEG_THREAD_H

ffmpeg_thread.c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include "ffmpeg_thread.h"
// run_ffmpeg_command 参数个数
int ffmpeg_argc;
// 缓存 run_ffmpeg_command 返回结果
int ffmpeg_exec_ret;
// run_ffmpegm_command 参数
char **ffmpeg_argv;

// 子线程函数
void *run_thread(void *arg){
    ffmpeg_exec_ret = run_ffmpeg_command(ffmpeg_argc, ffmpeg_argv);
}

// 在子线程跑指令
int ffmpeg_thread_run_command(int argc, char **argv){
    // 初始化全局变量
    ffmpeg_argc = argc;
    ffmpeg_argv = argv;
    ffmpeg_exec_ret = -1;

    pthread_t thread_id;
    // 创建线程
    int thread_ret = pthread_create(&thread_id, NULL, run_thread, NULL);
    if (thread_ret){
        av_log(NULL, AV_LOG_ERROR, "can not create thread");
        return 1;
    }

    // join 子线程，模拟同步
    thread_ret = pthread_join(thread_id, NULL);
    if (thread_ret){
        av_log(NULL, AV_LOG_ERROR, "thread join error");
        return 1;
    }

    // 返回结果
    return ffmpeg_exec_ret;
}

void ffmpeg_thread_exit(int ret){
    // 缓存执行结果
    ffmpeg_exec_ret = ret;
    pthread_exit(NULL);
}

没几行代码，很简单。写完后当然还要去替换原先 ffmpeg_lib.c 中调用 run_ffmpeg_command 的地方为 ffmpeg_thread_run_command。然后替换到 exit_program(int) 函数内的 exit(ret) 为 ffmpg_thread_exit(ret)

二、添加硬解码支持

查了文档后发现 FFmpeg 是支持硬解码的 网址链接（不支持硬编码 MediaCodec 那行的 encoder 不是 Y），网上资料太少，有很多误人子弟，害得我搞了很久。

首先修改 build_ffmpeg.sh 中 ./config 的编译配置，修改或添加下面几项：

1
2
3
4
5
6
7
8
9
10
11
12
./configure \
    # 启用 mediacodec 必须要启用它
    --enable-jni \
    --enable-mediacodec \
    --disable-decoders \
    --enable-decoder=aac \
    --enable-decoder=aac_latm \
    --enable-decoder=mjpeg \
    --enable-decoder=png \
    # 替换原先解码器为硬解码
    --enable-decoder=h264_mediacodec \
    --enable-decoder=mpeg4_mediacodec \

ps: #… 注释经提供说明用，实际运行命令的时候请去掉

这样就可以用 h264_mediacodec 来解码视频了？很遗憾，不行。。。你还需要告诉 FFmpeg 你的 Java 虚拟机，方法在 libcodec/jni.h 内，叫 av_jni_set_java_vm(vm, NULL);。这个 vm 就是你要告诉它的 Java 虚拟机。vm 的获取方式可以通过实现 JNI_OnLoad 来拿到。

示例代码

1
2
3
4
5
6
7
8
#include <jni.h>
#include "libavcodec/jni.h"

// load JNI
jint JNI_OnLoad(JavaVM* vm, void* reserved){
    av_jni_set_java_vm(vm, NULL);
    return JNI_VERSION_1_6;
}

然而我发现并没有快多少，CPU 使用下降到是明显

三、在 FFmpeg 中输出日志到 Logcat

之前我没有启用日志，导致都不知道哪里错了，FFmpeg 支持通过 av_log_set_callback 函数自定义日志，没仔细研究里面的日志系统，要去研究的可以参考雷神的博文，这里悼念下雷神，停下键盘默哀三分钟。。。

三分钟到了，继续贴代码，这么简单的代码就不解释了

1
2
3
4
5
6
7
8
9
public class FFmpegNativeBridge{
    ...
    /**
     * 设置是否处于调试状态
     * @param debug
     */
    public static native void setDebug(boolean debug);
    ...
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#include <android/log.h>

#define ALOGE(fmt, ...) __android_log_vprint(ANDROID_LOG_ERROR, TAG, fmt, ##__VA_ARGS__)
#define ALOGI(fmt, ...) __android_log_vprint(ANDROID_LOG_INFO, TAG, fmt, ##__VA_ARGS__)
#define ALOGD(fmt, ...) __android_log_vprint(ANDROID_LOG_DEBUG, TAG, fmt, ##__VA_ARGS__)
#define ALOGW(fmt, ...) __android_log_vprint(ANDROID_LOG_WARN, TAG, fmt, ##__VA_ARGS__)
#define ALOGV(fmt, ...) __android_log_vprint(ANDROID_LOG_VERBOSE, TAG, fmt, ##__VA_ARGS__

jboolean FFMPEG_ANDROID_DEBUG = 0;

static void ffmpeg_android_log_callback(void *ptr, int level, const char *fmt, va_list vl){
    if (FFMPEG_ANDROID_DEBUG){
        switch(level) {
            case AV_LOG_DEBUG:
                ALOGD(fmt, vl);
                break;
            case AV_LOG_VERBOSE:
                ALOGV(fmt, vl);
                break;
            case AV_LOG_INFO:
                ALOGI(fmt, vl);
                break;
            case AV_LOG_WARNING:
                ALOGW(fmt, vl);
                break;
            case AV_LOG_ERROR:
                ALOGE(fmt, vl);
                break;
        }
    }
}

JNIEXPORT void JNICALL
Java_org_voiddog_ffmpeg_FFmpegNativeBridge_setDebug(JNIEnv *env, jclass type, jboolean debug) {
    FFMPEG_ANDROID_DEBUG = debug;
    if (debug){
        av_log_set_callback(ffmpeg_android_log_callback);
    } else {
        av_log_set_callback(NULL);
    }
}