Android中，使用C++类MediaCodec，调用MediaCodec::getOutputBuffer的解码例子

以下是一个简单的Android JNI代码示例，演示如何使用C++类MediaCodec和MediaFormat来解码视频文件并获取输出缓冲区：

#include <jni.h>
#include <android/log.h>
#include <android/native_window_jni.h>
#include <media/NdkMediaCodec.h>

#define LOG_TAG "MediaCodecExample"
#define LOGD(...) __android_log_print(ANDROID_LOG_DEBUG, LOG_TAG, __VA_ARGS__)

extern "C" {

JNIEXPORT void JNICALL
Java_com_example_MediaCodecUtils_decode(JNIEnv *env, jobject instance, jstring filePath_,
                                        jobject surface) {

    const char *filePath = env->GetStringUTFChars(filePath_, NULL);
    FILE *fp = fopen(filePath, "rb");
    if (!fp) {
        LOGD("Failed to open file: %s", filePath);
        return;
    }

    // 创建解码器
    AMediaCodec *codec = AMediaCodec_createDecoderByType("video/avc");
    AMediaFormat *format = AMediaFormat_new();
    AMediaFormat_setString(format, AMEDIAFORMAT_KEY_MIME, "video/avc");
    AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_WIDTH, 1920);
    AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_HEIGHT, 1080);
    AMediaFormat_setBuffer(format, "csd-0", (void *) csd_data, csd_size);

    AMediaCodec_configure(codec, format, ANativeWindow_fromSurface(env, surface),
                           NULL /* crypto */, 0 /* flags */);
    AMediaCodec_start(codec);

    // 循环解码
    bool inputDone = false;
    bool outputDone = false;
    while (!outputDone) {
        if (!inputDone) {
            // 获取可用输入缓冲区的索引
            ssize_t inputIndex = AMediaCodec_dequeueInputBuffer(codec, 2000);
            if (inputIndex >= 0) {
                AMediaCodecBufferInfo inputBufferInfo;
                auto inputBuffer = AMediaCodec_getInputBuffer(codec, inputIndex);
                size_t bytesRead = fread(inputBuffer->data, 1, inputBuffer->capacity, fp);
                if (bytesRead > 0) {
                    // 提交输入缓冲区
                    AMediaCodec_queueInputBuffer(codec, inputIndex, 0, bytesRead, 0,
                                                 inputDone ? AMEDIACODEC_BUFFER_FLAG_END_OF_STREAM : 0);
                } else {
                    inputDone = true;
                    AMediaCodec_queueInputBuffer(codec, inputIndex, 0, 0, 0, AMEDIACODEC_BUFFER_FLAG_END_OF_STREAM);
                }
            }
        }

        // 获取可用输出缓冲区的索引
        AMediaCodecBufferInfo outputBufferInfo;
        ssize_t outputIndex = AMediaCodec_dequeueOutputBuffer(codec, &outputBufferInfo, 0);
        if (outputIndex == AMEDIACODEC_INFO_OUTPUT_FORMAT_CHANGED) {
            // 更新输出格式
            format = AMediaCodec_getOutputFormat(codec);
            LOGD("Output format changed: %s", AMediaFormat_toString(format));
        } else if (outputIndex == AMEDIACODEC_INFO_OUTPUT_BUFFERS_CHANGED) {
            // 忽略此消息
            LOGD("Output buffers changed");
        } else if (outputIndex >= 0) {
            auto outputBuffer = AMediaCodec_getOutputBuffer(codec, outputIndex);
            // 处理输出缓冲区
            LOGD("Output buffer %zd, size %d, flags %d, pts %lld, dts %lld", outputIndex,
                 outputBufferInfo.size, outputBufferInfo.flags, outputBufferInfo.presentationTimeUs,
                 outputBufferInfo.decodingTimeUs);
            // 获取输出缓冲区的数据
            uint8_t *data = outputBuffer->data;
            int dataSize = outputBufferInfo.size;
            // 处理数据...
            // 处理完输出缓冲区后，释放缓冲区
            AMediaCodec_releaseOutputBuffer(codec, outputIndex, true /* render */);
            outputDone = (outputBufferInfo.flags & AMEDIACODEC_BUFFER_FLAG_END_OF_STREAM) != 0;
        }
    }

    // 释放资源
    AMediaCodec_stop(codec);
    AMediaCodec_delete(codec);
    fclose(fp);

    env->ReleaseStringUTFChars(filePath_, filePath);
}

}

这个例子中，我们使用AMediaCodec_createDecoderByType函数创建一个H.264视频解码器，然后使用AMediaFormat来指定视频的宽度、高度、MIME类型和CSD数据。在configure方法中，我们将解码器与一个Surface关联起来，以便输出图像可以在屏幕上渲染。在循环中，我们首先获取可用的输入缓冲区索引，并从输入文件中读取数据填充缓冲区。然后，我们提交输入缓冲区，并等待解码器输出缓冲区。当我们获取可用的输出缓冲区索引时，我们可以从输出缓冲区中获取数据并进行处理。最后，我们释放输出缓冲区并检查是否到达了输入文件的结尾。