安卓 VideoProcessor 示例

以下是一个简单的安卓 VideoProcessor 示例：

public class VideoProcessor {
    private static final String TAG = "VideoProcessor";

    // 视频宽高
    private int videoWidth = 0;
    private int videoHeight = 0;

    // 帧率
    private int frameRate = 24;

    // 视频编码器
    private MediaCodec videoEncoder;

    // 视频编码器参数
    private MediaFormat videoFormat;

    // 视频输入缓冲区
    private ByteBuffer[] inputBuffers;

    // 视频输出缓冲区
    private ByteBuffer[] outputBuffers;

    // 视频编码完成标志
    private boolean isVideoEncoderDone = false;

    // 视频帧时间戳
    private long videoFrameTimeStamp = 0;

    // 视频输出文件路径
    private String outputFilePath;

    // 视频输出文件
    private FileOutputStream outputStream;

    // 视频输出线程
    private Thread videoOutputThread;

    // 视频帧队列
    private LinkedBlockingQueue<Frame> videoFrameQueue = new LinkedBlockingQueue<>();

    public VideoProcessor(int width, int height, int frameRate, String outputPath) {
        this.videoWidth = width;
        this.videoHeight = height;
        this.frameRate = frameRate;
        this.outputFilePath = outputPath;
    }

    /**
     * 初始化视频编码器
     */
    public void init() {
        try {
            // 创建视频编码器
            videoEncoder = MediaCodec.createEncoderByType(MediaFormat.MIMETYPE_VIDEO_AVC);

            // 创建视频编码器参数
            videoFormat = MediaFormat.createVideoFormat(MediaFormat.MIMETYPE_VIDEO_AVC, videoWidth, videoHeight);
            videoFormat.setInteger(MediaFormat.KEY_BIT_RATE, 6000000);
            videoFormat.setInteger(MediaFormat.KEY_FRAME_RATE, frameRate);
            videoFormat.setInteger(MediaFormat.KEY_COLOR_FORMAT, MediaCodecInfo.CodecCapabilities.COLOR_FormatSurface);
            videoFormat.setInteger(MediaFormat.KEY_I_FRAME_INTERVAL, 1);

            // 配置视频编码器
            videoEncoder.configure(videoFormat, null, null, MediaCodec.CONFIGURE_FLAG_ENCODE);

            // 启动视频编码器
            videoEncoder.start();

            // 获取视频编码器输入缓冲区
            inputBuffers = videoEncoder.getInputBuffers();

            // 获取视频编码器输出缓冲区
            outputBuffers = videoEncoder.getOutputBuffers();

            // 创建视频输出文件
            outputStream = new FileOutputStream(outputFilePath);

            // 创建视频输出线程
            videoOutputThread = new Thread(new VideoOutputRunnable());
            videoOutputThread.start();

        } catch (IOException e) {
            Log.e(TAG, "init: " + e.getMessage());
        }
    }

    /**
     * 处理视频帧
     */
    public void process(Frame frame) {
        try {
            // 添加视频帧到队列
            videoFrameQueue.put(frame);
        } catch (InterruptedException e) {
            Log.e(TAG, "process: " + e.getMessage());
        }
    }

    /**
     * 释放资源
     */
    public void release() {
        try {
            // 释放视频编码器
            videoEncoder.stop();
            videoEncoder.release();

            // 关闭视频输出文件
            outputStream.flush();
            outputStream.close();

            // 等待视频输出线程结束
            videoOutputThread.join();
        } catch (IOException | InterruptedException e) {
            Log.e(TAG, "release: " + e.getMessage());
        }
    }

    /**
     * 视频编码器输入线程
     */
    private class VideoEncoderInputRunnable implements Runnable {
        @Override
        public void run() {
            while (!isVideoEncoderDone) {
                try {
                    // 获取视频编码器输入缓冲区
                    int inputBufferIndex = videoEncoder.dequeueInputBuffer(-1);
                    if (inputBufferIndex >= 0) {
                        // 获取视频帧数据
                        Frame frame = videoFrameQueue.take();

                        // 将视频帧数据写入视频编码器输入缓冲区
                        ByteBuffer inputBuffer = inputBuffers[inputBufferIndex];
                        inputBuffer.clear();
                        inputBuffer.put(frame.getData());
                        videoEncoder.queueInputBuffer(inputBufferIndex, 0, frame.getData().length, videoFrameTimeStamp, 0);

                        // 更新视频帧时间戳
                        videoFrameTimeStamp += 1000000 / frameRate;
                    }
                } catch (InterruptedException e) {
                    Log.e(TAG, "VideoEncoderInputRunnable: " + e.getMessage());
                }
            }
        }
    }

    /**
     * 视频输出线程
     */
    private class VideoOutputRunnable implements Runnable {
        @Override
        public void run() {
            try {
                // 开始视频编码器输入线程
                Thread videoEncoderInputThread = new Thread(new VideoEncoderInputRunnable());
                videoEncoderInputThread.start();

                // 循环从视频编码器输出缓冲区中获取视频数据
                while (!isVideoEncoderDone) {
                    MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();
                    int outputBufferIndex = videoEncoder.dequeueOutputBuffer(bufferInfo, 0);
                    if (outputBufferIndex == MediaCodec.INFO_OUTPUT_FORMAT_CHANGED) {
                        // 获取视频编码器输出格式
                        MediaFormat outputFormat = videoEncoder.getOutputFormat();
                        Log.d(TAG, "Video output format changed: " + outputFormat);
                    } else if (outputBufferIndex >= 0) {
                        // 获取视频编码器输出缓冲区
                        ByteBuffer outputBuffer = outputBuffers[outputBufferIndex];

                        // 写入视频输出文件
                        byte[] data = new byte[bufferInfo.size];
                        outputBuffer.get(data);
                        outputStream.write(data);

                        // 释放视频编码器输出缓冲区
                        videoEncoder.releaseOutputBuffer(outputBufferIndex, false);
                    }
                }

                // 释放视频编码器输入线程
                videoEncoderInputThread.interrupt();
                videoEncoderInputThread.join();

            } catch (IOException | InterruptedException e) {
                Log.e(TAG, "VideoOutputRunnable: " + e.getMessage());
            }
        }
    }
}

使用 VideoProcessor 示例的代码如下：

// 创建 VideoProcessor 对象
VideoProcessor videoProcessor = new VideoProcessor(640, 480, 24, "/sdcard/output.mp4");

// 初始化 VideoProcessor
videoProcessor.init();

// 处理视频帧
for (int i = 0; i < 100; i++) {
    byte[] data = getVideoFrameData(i);
    Frame frame = new Frame(data);
    videoProcessor.process(frame);
}

// 释放 VideoProcessor 资源
videoProcessor.release();

其中，`Frame` 类表示视频帧数据，`getVideoFrameData` 方法用于获取视频帧数据。在实际使用中，需要根据具体需求进行修改。