FFMPEG视频转码

#include <jni.h>
#include "android/log.h"
#include <string>

#define TAG "LEVY"
#define LOGE(...) __android_log_print(ANDROID_LOG_ERROR,TAG,__VA_ARGS__)

extern "C" {

#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libswscale/swscale.h"

JNIEXPORT void JNICALL
Java_com_levylin_study_1ffmpeg_MainActivity_transform(JNIEnv *env, jclass type, jstring input_,
                                                      jstring output_) {
    const char *input = env->GetStringUTFChars(input_, 0);
    const char *output = env->GetStringUTFChars(output_, 0);
    LOGE("input=%s", input);
    LOGE("output=%s", output);

    //该函数在所有基于ffmpeg的应用程序中几乎都是第一个被调用的。只有调用了该函数，才能使用复用器，编码器等。
    av_register_all();

//创建并分配一个AVFormatContext,AVFormatContext：全局的基本结构体,主要用于处理封装格式（FLV/MKV/RMVB等）。
    AVFormatContext *formatContext = avformat_alloc_context();

    if (avformat_open_input(&formatContext, input, NULL, NULL) < 0) {
        LOGE("视频打开失败");
        return;
    }
    if (avformat_find_stream_info(formatContext, NULL) < 0) {
        LOGE("找寻流信息失败");
        return;
    }
    //寻找流信息
    int video_stream_index = -1;
    for (int i = 0; i < formatContext->nb_streams; ++i) {
        LOGE("循环 %d", i);
        //特别注意，这里的codec_type千万别打成coder_type
        if (formatContext->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
            video_stream_index = i;
        }
    }
    LOGE("video_stream_index=%d", video_stream_index);
    if (video_stream_index == -1) {
        LOGE("找寻视频流信息失败");
        return;
    }
    //获取上下文
    AVCodecContext *codecContext = formatContext->streams[video_stream_index]->codec;
    LOGE("获取上下文成功,宽 %d  高 %d", codecContext->width, codecContext->height);
    //获取解码器
    AVCodec *codec = avcodec_find_decoder(codecContext->codec_id);
    LOGE("获取解码器成功");
    if (avcodec_open2(codecContext, codec, NULL) < 0) {
        LOGE("解码失败");
        return;
    }
    //分配内存
    AVPacket *packet = (AVPacket *) av_malloc(sizeof(AVPacket));
    //初始化结构体
    av_init_packet(packet);
    LOGE("创建并初始化AVPacket成功");
    AVFrame *frame = av_frame_alloc();

    //声明一个yuvFrame
    AVFrame *yuvFrame = av_frame_alloc();
    //yuvFrame中缓存区的初始化
    uint8_t *out_buffer = (uint8_t *) av_malloc(
            (size_t) avpicture_get_size(AV_PIX_FMT_YUV420P, codecContext->width,
                                        codecContext->height));
    int re = avpicture_fill((AVPicture *) yuvFrame, out_buffer, AV_PIX_FMT_YUV420P,
                            codecContext->width,
                            codecContext->height);
    LOGE("创建和初始化yuvFrame成功");

    FILE *outFile = fopen(output, "wb");
    int got_frame = 0;

    LOGE("开始获取swsContext");
    SwsContext *swsContext = sws_getContext(codecContext->width, codecContext->height,
                                            codecContext->pix_fmt,
                                            codecContext->width, codecContext->height,
                                            AV_PIX_FMT_YUV420P,
                                            SWS_BILINEAR, NULL, NULL, NULL);
    LOGE("开始解码");
    while (av_read_frame(formatContext, packet) >= 0) {
        avcodec_decode_video2(codecContext, frame, &got_frame, packet);
        if (got_frame > 0) {
            sws_scale(swsContext, (const uint8_t *const *) frame->data, frame->linesize, 0,
                      frame->height, yuvFrame->data, yuvFrame->linesize);
            int y_size = codecContext->width * codecContext->height;
            //        y 亮度信息写完了
            fwrite(yuvFrame->data[0], 1, (size_t) y_size, outFile);
            fwrite(yuvFrame->data[1], 1, (size_t) (y_size / 4), outFile);
            fwrite(yuvFrame->data[2], 1, (size_t) (y_size / 4), outFile);
        }
        av_free_packet(packet);
    }
    LOGE("转码成功");
    fclose(outFile);
    av_frame_free(&frame);
    av_frame_free(&yuvFrame);
    avcodec_close(codecContext);
    avformat_free_context(formatContext);
    env->ReleaseStringUTFChars(input_, input);
    env->ReleaseStringUTFChars(output_, output);
}
}