Android MediaCodec 编解码

import android.media.MediaCodec;
import android.media.MediaFormat;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;

public class H264Encoder {
   private static final String MIME_TYPE = "video/avc"; // H.264 MIME 类型
   private MediaCodec mEncoder;
   private FileOutputStream mOutputStream; // 用于写入 H.264 文件

   public void configure(int width, int height, int bitrate, int frameRate, int iframeInterval) {
       try {
           // 1. 创建编码器
           mEncoder = MediaCodec.createEncoderByType(MIME_TYPE);

           // 2. 创建 MediaFormat 并设置参数
           MediaFormat format = MediaFormat.createVideoFormat(MIME_TYPE, width, height);

           // 关键参数设置
           format.setInteger(MediaFormat.KEY_COLOR_FORMAT,
                   MediaCodecInfo.CodecCapabilities.COLOR_FormatYUV420Flexible); // 推荐使用 Flexible
           format.setInteger(MediaFormat.KEY_BIT_RATE, bitrate); // 码率，单位 bps
           format.setInteger(MediaFormat.KEY_FRAME_RATE, frameRate); // 帧率
           format.setInteger(MediaFormat.KEY_I_FRAME_INTERVAL, iframeInterval); // I帧间隔，单位秒
           format.setInteger(MediaFormat.KEY_BITRATE_MODE, MediaCodecInfo.EncoderCapabilities.BITRATE_MODE_CBR); // 码率模式，可选 CBR/VBR/CQ

           // 3. 配置编码器（指定为编码器，传入 null 表示不关联 Surface）
           mEncoder.configure(format, null, null, MediaCodec.CONFIGURE_FLAG_ENCODE);

           // 4. 启动编码器
           mEncoder.start();

           // 5. 打开输出文件
           mOutputStream = new FileOutputStream("output.h264");

       } catch (IOException e) {
           e.printStackTrace();
       }
   }
}

public void encodeFrame(byte[] yuvData, long presentationTimeUs) {
    try {
        // --- 处理输入 ---
        // 1. 获取输入缓冲区的索引
        int inputBufferIndex = mEncoder.dequeueInputBuffer(10000); // 超时 10ms
        if (inputBufferIndex >= 0) {
            ByteBuffer inputBuffer = mEncoder.getInputBuffer(inputBufferIndex);
            inputBuffer.clear();

            // 2. 将 YUV 数据复制到输入缓冲区
            // 注意：这里假设 yuvData 的格式与 MediaFormat 中设置的 COLOR_FORMAT 一致！
            // 如果格式不匹配，需要先进行转换（例如 NV21 -> I420 或 NV12）
            inputBuffer.put(yuvData);

            // 3. 提交输入缓冲区给编码器
            // presentationTimeUs 是此帧的时间戳，单位微秒 (us)
            // 如果是最后一帧，需要加上 BUFFER_FLAG_END_OF_STREAM 标志
            mEncoder.queueInputBuffer(inputBufferIndex, 0, yuvData.length, presentationTimeUs, 0);
        }

        // --- 处理输出 ---
        MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();
        int outputBufferIndex;
        while ((outputBufferIndex = mEncoder.dequeueOutputBuffer(bufferInfo, 10000)) >= 0) {
            if ((bufferInfo.flags & MediaCodec.BUFFER_FLAG_CODEC_CONFIG) != 0) {
                // 这是编解码器配置信息 (SPS/PPS)，通常需要保存并在文件开头或每个 I 帧前写入
                // 对于 H.264 文件，通常将 SPS/PPS 写在文件最开头
                ByteBuffer outputBuffer = mEncoder.getOutputBuffer(outputBufferIndex);
                byte[] spsPps = new byte[bufferInfo.size];
                outputBuffer.get(spsPps);
                // 将 spsPps 写入文件
                mOutputStream.write(spsPps);
                // 释放输出缓冲区
                mEncoder.releaseOutputBuffer(outputBufferIndex, false);
                continue;
            }

            if ((bufferInfo.flags & MediaCodec.BUFFER_FLAG_END_OF_STREAM) != 0) {
                // 编码结束
                break;
            }

            // 获取包含编码后 H.264 数据的输出缓冲区
            ByteBuffer outputBuffer = mEncoder.getOutputBuffer(outputBufferIndex);
            byte[] encodedData = new byte[bufferInfo.size];
            outputBuffer.get(encodedData);

            // 将编码后的 H.264 NAL 单元写入文件
            // 通常需要在每个 NAL 单元前添加起始码 0x00000001
            mOutputStream.write(new byte[]{0, 0, 0, 1});
            mOutputStream.write(encodedData);

            // 释放输出缓冲区
            mEncoder.releaseOutputBuffer(outputBufferIndex, false);
        }

    } catch (IOException e) {
        e.printStackTrace();
    }
}

public void stopAndRelease() {
    try {
        // 1. 发送结束信号
        int inputBufferIndex = mEncoder.dequeueInputBuffer(-1); // 阻塞等待
        if (inputBufferIndex >= 0) {
            mEncoder.queueInputBuffer(inputBufferIndex, 0, 0, 0, MediaCodec.BUFFER_FLAG_END_OF_STREAM);
        }

        // 2. 处理所有剩余的输出数据
        MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();
        int outputBufferIndex;
        do {
            outputBufferIndex = mEncoder.dequeueOutputBuffer(bufferInfo, 10000);
            if (outputBufferIndex >= 0) {
                if ((bufferInfo.flags & MediaCodec.BUFFER_FLAG_END_OF_STREAM) != 0) {
                    break; // 确认结束
                }
                ByteBuffer outputBuffer = mEncoder.getOutputBuffer(outputBufferIndex);
                byte[] encodedData = new byte[bufferInfo.size];
                outputBuffer.get(encodedData);

                mOutputStream.write(new byte[]{0, 0, 0, 1});
                mOutputStream.write(encodedData);

                mEncoder.releaseOutputBuffer(outputBufferIndex, false);
            }
        } while (outputBufferIndex >= 0 || outputBufferIndex == MediaCodec.INFO_TRY_AGAIN_LATER);

        // 3. 关闭文件流
        if (mOutputStream != null) {
            mOutputStream.close();
        }

        // 4. 停止并释放编码器资源
        mEncoder.stop();
        mEncoder.release();
        mEncoder = null;

    } catch (IOException e) {
        e.printStackTrace();
    }
}