MediaCodec 添加水印

要在 Android 中使用 MediaCodec 添加水印，可以考虑以下步骤：

1. 创建一个带有水印的 Bitmap。
2. 获取原始视频的帧数据并解码。
3. 将水印 Bitmap 绘制到视频帧上。
4. 在编码器中编码带有水印的帧数据。
5. 将编码后的数据写入输出文件或传输到网络。

以下是一些示例代码，用于在视频帧上添加水印：

// 创建带有水印的 Bitmap
Bitmap watermark = BitmapFactory.decodeResource(getResources(), R.drawable.watermark);
watermark = Bitmap.createScaledBitmap(watermark, 100, 100, true);

// 获取原始视频的帧数据并解码
MediaExtractor extractor = new MediaExtractor();
extractor.setDataSource(videoFilePath);
int trackIndex = selectTrack(extractor, "video/");
extractor.selectTrack(trackIndex);
MediaFormat format = extractor.getTrackFormat(trackIndex);
int width = format.getInteger(MediaFormat.KEY_WIDTH);
int height = format.getInteger(MediaFormat.KEY_HEIGHT);
MediaCodec decoder = MediaCodec.createDecoderByType(format.getString(MediaFormat.KEY_MIME));
decoder.configure(format, null, null, 0);
decoder.start();
ByteBuffer[] inputBuffers = decoder.getInputBuffers();
ByteBuffer[] outputBuffers = decoder.getOutputBuffers();
MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();
boolean isEOS = false;

// 逐帧处理视频
while (!isEOS) {
    int inputBufferIndex = decoder.dequeueInputBuffer(10000);
    if (inputBufferIndex >= 0) {
        ByteBuffer inputBuffer = inputBuffers[inputBufferIndex];
        int sampleSize = extractor.readSampleData(inputBuffer, 0);
        if (sampleSize < 0) {
            decoder.queueInputBuffer(inputBufferIndex, 0, 0, 0, MediaCodec.BUFFER_FLAG_END_OF_STREAM);
            isEOS = true;
        } else {
            decoder.queueInputBuffer(inputBufferIndex, 0, sampleSize, extractor.getSampleTime(), 0);
            extractor.advance();
        }
    }

    int outputBufferIndex = decoder.dequeueOutputBuffer(bufferInfo, 10000);
    switch (outputBufferIndex) {
        case MediaCodec.INFO_OUTPUT_BUFFERS_CHANGED:
            outputBuffers = decoder.getOutputBuffers();
            break;
        case MediaCodec.INFO_OUTPUT_FORMAT_CHANGED:
            format = decoder.getOutputFormat();
            break;
        case MediaCodec.INFO_TRY_AGAIN_LATER:
            break;
        default:
            ByteBuffer outputBuffer = outputBuffers[outputBufferIndex];
            // 绘制水印到视频帧上
            Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
            outputBuffer.rewind();
            bitmap.copyPixelsFromBuffer(outputBuffer);
            Canvas canvas = new Canvas(bitmap);
            canvas.drawBitmap(watermark, 0, 0, null);
            // 将带有水印的视频帧编码并写入输出文件或传输到网络
            byte[] data = convertBitmapToNV21(bitmap, width, height);
            ByteBuffer inputBuffer = encoder.getInputBuffer(inputBufferIndex);
            inputBuffer.clear();
            inputBuffer.put(data);
            encoder.queueInputBuffer(inputBufferIndex, 0, data.length, bufferInfo.presentationTimeUs, 0);
            decoder.releaseOutputBuffer(outputBufferIndex, false);
            break;
    }

    if ((bufferInfo.flags & MediaCodec.BUFFER_FLAG_END_OF_STREAM) != 0) {
        isEOS = true;
    }
}

// 释放资源
decoder.stop();
decoder.release();
extractor.release();

在上面的示例中，我们使用 `MediaExtractor` 获取原始视频的帧数据，并通过 `MediaCodec` 解码每一帧。在处理每一帧数据时，我们将水印 Bitmap 绘制到视频帧上，并将带有水印的帧数据编码并写入输出文件或传输到网络。最后，我们释放所有资源。

注意，上面的示例代码仅用于演示如何在 Android 中使用 MediaCodec 添加水印。实际应用中，你需要根据实际情况进行适当的修改和优化。