灌篮少年 发表于 2024-10-24 11:53:36

Java ffmpeg 实现视频加文字/图片水印功能

引入依赖

<dependency>
<groupId>org.bytedeco</groupId>
<artifactId>javacv-platform</artifactId>
<version>1.5.4</version>
</dependency>
<dependency>
<groupId>org.bytedeco</groupId>
<artifactId>ffmpeg-platform</artifactId>
<version>4.3.1-1.5.4</version>
</dependency>代码示例

文字水印实现

import org.bytedeco.javacv.*;
import org.bytedeco.javacv.Frame;

import java.awt.*;
import java.awt.image.BufferedImage;

/**
* 视频添加文字水印
*
* @author alin
*/
public class VideoWatermark {

    public static void main(String[] args) throws Exception {
      // 视频文件
      String inputFile = "D:\\test\\video.mp4";
      // 输出文件
      String outputFile = "D:\\test\\output.mp4";

      // 初始化FFmpegFrameGrabber以读取输入视频文件
      try (FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(inputFile)) {
            grabber.start();
            // 初始化FFmpegFrameRecorder以写入输出视频文件
            try (FFmpegFrameRecorder recorder = new FFmpegFrameRecorder(outputFile, grabber.getImageWidth(), grabber.getImageHeight(), grabber.getAudioChannels())) {
                recorder.setFormat("mp4");
                recorder.setSampleRate(grabber.getSampleRate());
                recorder.setFrameRate(grabber.getFrameRate());
                recorder.setTimestamp(grabber.getTimestamp());
                recorder.setVideoBitrate(grabber.getVideoBitrate());
                recorder.setVideoCodec(grabber.getVideoCodec());
                recorder.start();

                // 处理每一帧
                Frame frame;
                Java2DFrameConverter converter = new Java2DFrameConverter();
                while ((frame = grabber.grab()) != null) {
                  if (frame.image != null) {
                        // 将Frame转换为BufferedImage
                        BufferedImage bufferedImage = converter.getBufferedImage(frame);

                        // 在BufferedImage上绘制文字水印
                        Graphics2D g = bufferedImage.createGraphics();
                        // 启用抗锯齿
                        g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
                        // 启用文本抗锯齿
                        g.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
                        // 使用支持中文的字体
                        g.setFont(new Font("Microsoft YaHei", Font.BOLD, 30));
                        g.setColor(Color.gray);
                        String watermarkText = "测试水印文字";

                        // 计算水印文字的位置 右下角
                        //FontMetrics fontMetrics = g.getFontMetrics();
                        //int textWidth = fontMetrics.stringWidth(watermarkText);
                        //int textHeight = fontMetrics.getHeight();
                        //int x = bufferedImage.getWidth() - textWidth - 10; // 右边距10像素
                        //int y = bufferedImage.getHeight() - textHeight + fontMetrics.getAscent() - 10; // 下边距10像素

                        // 计算水印文字的位置 左上角
                        int x = 10; // 左边距10像素
                        int y = 30; // 上边距30像素

                        g.drawString(watermarkText, x, y);
                        g.dispose();

                        // 将BufferedImage转换回Frame
                        frame = converter.convert(bufferedImage);
                  }

                  // 记录帧
                  recorder.record(frame);
                }
                // 关闭grabber和recorder
                recorder.stop();
                recorder.release();
                grabber.stop();
            }
      }
    }
}图片水印实现

import org.bytedeco.javacv.*;
import org.bytedeco.javacv.Frame;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.net.URL;

/**
* 视频添加图片水印
*
* @author alin
*/
public class VideoWatermark {

    public static void main(String[] args) throws Exception {
      // 视频文件
      String inputFile = "D:\\test\\video.mp4";
      // 输出文件
      String outputFile = "D:\\test\\output.mp4";
      // 水印照片
      String watermarkImageUrl = "D:\\test\\test.png";

      try (FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(inputFile)) {
            grabber.start();
            try (FFmpegFrameRecorder recorder = new FFmpegFrameRecorder(outputFile, grabber.getImageWidth(), grabber.getImageHeight(), grabber.getAudioChannels())) {
                recorder.setFormat("mp4");
                recorder.setSampleRate(grabber.getSampleRate());
                recorder.setFrameRate(grabber.getFrameRate());
                recorder.setTimestamp(grabber.getTimestamp());
                recorder.setVideoBitrate(grabber.getVideoBitrate());
                recorder.setVideoCodec(grabber.getVideoCodec());
                recorder.start();

                BufferedImage watermarkImage = loadWatermarkImage(watermarkImageUrl);

                Frame frame;
                while ((frame = grabber.grab()) != null) {
                  if (frame.image != null) {
                        Java2DFrameConverter converter = new Java2DFrameConverter();
                        BufferedImage bufferedImage = converter.getBufferedImage(frame);

                        // 在BufferedImage上绘制图片水印
                        Graphics2D g = bufferedImage.createGraphics();
                        g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.5f));
                        int x = 10; // 左边距10像素
                        int y = 30; // 上边距30像素
                        g.drawImage(watermarkImage, x, y, null);
                        g.dispose();
                        frame = converter.convert(bufferedImage);
                  }
                  recorder.record(frame);
                }
                recorder.stop();
                recorder.release();
                grabber.stop();
            }
      }
    }

    /**
   * 加载水印图片
   *
   * @param imagePath 水印图片路径
   * @return 水印图片
   * @throws IOException
   */
    private static BufferedImage loadWatermarkImage(String imagePath) throws IOException {
      //return ImageIO.read(new URL(imagePath));
      return ImageIO.read(new File(imagePath));
    }

}
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: Java ffmpeg 实现视频加文字/图片水印功能