Android 录屏并添加水印

实现 Android 录屏并添加水印的一种方法是使用 MediaProjection API 和 Canvas 绘图。以下是实现的步骤：

1. 获取 MediaProjection 对象，在 Android 5.0 及以上版本中，需要用户授权获取 MediaProjection 对象，可以使用 startActivityForResult() 方法获取。

2. 创建 VirtualDisplay 对象，将 MediaProjection 对象传入，设置录制的屏幕宽高和 dpi。

3. 创建 Surface 对象，将 VirtualDisplay 对象的 surface 传入。

4. 在 Surface 上绘制屏幕内容，可以使用 Canvas 绘图实现。

5. 绘制水印，可以使用 Bitmap、Paint、Canvas 等方法实现。

6. 将绘制好的帧数据传入 MediaCodec 编码成视频数据。

7. 最后将视频数据保存到文件或者上传到服务器。

以下是示例代码：

// 获取 MediaProjection 对象
private void startScreenCapture() {
    MediaProjectionManager mediaProjectionManager = (MediaProjectionManager) getSystemService(MEDIA_PROJECTION_SERVICE);
    startActivityForResult(mediaProjectionManager.createScreenCaptureIntent(), REQUEST_MEDIA_PROJECTION);
}

// 获取 MediaProjection 授权结果
@Override
protected void onActivityResult(int requestCode, int resultCode, @Nullable Intent data) {
    if (requestCode == REQUEST_MEDIA_PROJECTION) {
        if (resultCode == RESULT_OK) {
            // 创建 VirtualDisplay 对象
            mediaProjection = ((MediaProjectionManager) getSystemService(MEDIA_PROJECTION_SERVICE)).getMediaProjection(resultCode, data);
            virtualDisplay = mediaProjection.createVirtualDisplay("ScreenCapture",
                    screenWidth, screenHeight, screenDensity,
                    DisplayManager.VIRTUAL_DISPLAY_FLAG_AUTO_MIRROR,
                    surface, null, null);
        }
    }
}

// 在 Surface 上绘制屏幕内容和水印
private void drawOnSurface(Canvas canvas) {
    // 绘制屏幕内容
    canvas.drawBitmap(screenBitmap, 0, 0, null);

    // 绘制水印
    Paint paint = new Paint();
    paint.setColor(Color.WHITE);
    paint.setTextSize(24);
    paint.setAntiAlias(true);
    canvas.drawText("Watermark", 10, 30, paint);
}

// 将帧数据传入 MediaCodec 编码成视频数据
private void encodeFrame() {
    try {
        mediaCodec = MediaCodec.createEncoderByType("video/avc");
        MediaFormat mediaFormat = MediaFormat.createVideoFormat("video/avc", screenWidth, screenHeight);
        mediaFormat.setInteger(MediaFormat.KEY_BIT_RATE, bitrate);
        mediaFormat.setInteger(MediaFormat.KEY_FRAME_RATE, framerate);
        mediaFormat.setInteger(MediaFormat.KEY_COLOR_FORMAT, MediaCodecInfo.CodecCapabilities.COLOR_FormatSurface);
        mediaFormat.setInteger(MediaFormat.KEY_I_FRAME_INTERVAL, 1);
        mediaCodec.configure(mediaFormat, null, null, MediaCodec.CONFIGURE_FLAG_ENCODE);
        surface = mediaCodec.createInputSurface();
        mediaCodec.start();
        ByteBuffer[] inputBuffers = mediaCodec.getInputBuffers();
        ByteBuffer[] outputBuffers = mediaCodec.getOutputBuffers();
        MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();
        boolean isEOS = false;
        while (!isEOS) {
            int inputBufferIndex = mediaCodec.dequeueInputBuffer(-1);
            if (inputBufferIndex >= 0) {
                long presentationTimeUs = System.nanoTime() / 1000;
                Bitmap bitmap = getBitmapFromSurface();
                if (bitmap != null) {
                    Canvas canvas = surface.lockCanvas(null);
                    drawOnSurface(canvas);
                    surface.unlockCanvasAndPost(canvas);
                }
                ByteBuffer inputBuffer = inputBuffers[inputBufferIndex];
                inputBuffer.clear();
                inputBuffer.put(screenBuffer);
                if (isEOS) {
                    mediaCodec.queueInputBuffer(inputBufferIndex, 0, 0, presentationTimeUs, MediaCodec.BUFFER_FLAG_END_OF_STREAM);
                } else {
                    mediaCodec.queueInputBuffer(inputBufferIndex, 0, screenBuffer.length, presentationTimeUs, 0);
                }
                int outputBufferIndex = mediaCodec.dequeueOutputBuffer(bufferInfo, 0);
                while (outputBufferIndex >= 0) {
                    ByteBuffer outputBuffer = outputBuffers[outputBufferIndex];
                    byte[] outData = new byte[bufferInfo.size];
                    outputBuffer.get(outData);
                    if (bufferInfo.flags == MediaCodec.BUFFER_FLAG_CODEC_CONFIG) {
                        configByte = new byte[bufferInfo.size];
                        configByte = outData;
                    } else if (bufferInfo.flags == MediaCodec.BUFFER_FLAG_KEY_FRAME) {
                        byte[] keyframe = new byte[bufferInfo.size + configByte.length];
                        System.arraycopy(configByte, 0, keyframe, 0, configByte.length);
                        System.arraycopy(outData, 0, keyframe, configByte.length, outData.length);
                        outData = keyframe;
                    }
                    if (outData != null) {
                        // 保存视频数据
                    }
                    mediaCodec.releaseOutputBuffer(outputBufferIndex, false);
                    outputBufferIndex = mediaCodec.dequeueOutputBuffer(bufferInfo, 0);
                }
            }
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}

// 从 Surface 中获取 Bitmap
private Bitmap getBitmapFromSurface() {
    Bitmap bitmap = null;
    try {
        bitmap = Bitmap.createBitmap(screenWidth, screenHeight, Bitmap.Config.ARGB_8888);
        surface.lockCanvas(null).drawBitmap(bitmap, 0, 0, null);
        surface.unlockCanvasAndPost(canvas);
    } catch (Exception e) {
        e.printStackTrace();
    }
    return bitmap;
}

需要注意的是，由于 Android 中录屏功能和实现方式在不同的版本中有所不同，因此上述代码仅供参考，实际应用中可能需要根据具体情况进行调整。