java cv实现每秒对视频进行截图

主要使用java cv中的ffmpeg对视频进行每秒截图

用到的maven dependency如下：

        <dependency>
            <groupId>org.bytedeco</groupId>
            <artifactId>javacpp</artifactId>
            <version>1.4.3</version>
        </dependency>
        <dependency>
            <groupId>org.bytedeco</groupId>
            <artifactId>javacv</artifactId>
            <version>1.4.3</version>
        </dependency>
        <dependency>
            <groupId>org.bytedeco.javacpp-presets</groupId>
            <artifactId>ffmpeg-platform</artifactId>
            <version>4.0.2-1.4.3</version>
        </dependency>

代码如下：

package com.pingan.video.quality.controller;

import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.Java2DFrameConverter;

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

public class PictureGrabberController {
    /**
     * 获取视频图片
     * @Title: gainVedioImg
     * @param: filePath  文件夹路径
     * @return: void
     * @throws
     */
    public static void getVedioImg(String filePath, String fileName, int second) {
        getImage(filePath, fileName, second);
    }
    /**
     * 获取视频图片
     * @Title: gainVedioImg
     * @param:
     * @return: void
     * @throws
     */
    public static void getVedioImg(String filePath, String fileName) {
        getImage(filePath, fileName, 30);
    }

    private static void getImage(String filePath, String fileName, int second) {
        File folder = new File(filePath);
        if (!folder.exists())
        {
            folder.mkdirs();//如果文件夹不存在则创建
        }
        try{
            FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(filePath + File.separator + fileName);
            grabber.start();
            long ftp = grabber.getLengthInFrames();
            int flag=0;
            Frame frame = null;
            int count = (int)Math.round(grabber.getFrameRate());
            while (flag <= ftp) {
                if(count<ftp && count>flag){
                    flag++;
                    continue;
                } else if(count > ftp){
                    break;
                } else{
                    count += (int)Math.round(grabber.getFrameRate());
                }
                //获取帧
                frame = grabber.grabImage();
                //过滤前 second 帧，避免出现全黑图片
                /*if ((flag>second)&&(frame != null)) {
                    break;
                }*/
                String rotate = grabber.getVideoMetadata("rotate");
                //创建BufferedImage对象
                Java2DFrameConverter converter = new Java2DFrameConverter();
                BufferedImage bufferedImage = converter.getBufferedImage(frame);
                if (rotate != null) {
                    //旋转图片
                    bufferedImage = rotate(bufferedImage, Integer.parseInt(rotate));
                }
                String newFileName = filePath + File.separator + fileName.substring(0, fileName.lastIndexOf(".")) + flag +".jpeg";
                ImageIO.write(bufferedImage, "jpeg", new File(newFileName));
                flag += (int)Math.round(grabber.getFrameRate());
            }

            grabber.close();
            grabber.stop();
        }catch(Exception e){
            System.out.printf("获取视频图片失败！", e);
        }

    }

    private static BufferedImage rotate(BufferedImage src, int angel) {
        int src_width = src.getWidth(null);
        int src_height = src.getHeight(null);
        int type = src.getColorModel().getTransparency();
        Rectangle rect_des = calcRotatedSize(new Rectangle(new Dimension(src_width, src_height)), angel);
        BufferedImage bi = new BufferedImage(rect_des.width, rect_des.height, type);
        Graphics2D g2 = bi.createGraphics();
        g2.translate((rect_des.width - src_width) / 2, (rect_des.height - src_height) / 2);
        g2.rotate(Math.toRadians(angel), src_width / 2, src_height / 2);
        g2.drawImage(src, 0, 0, null);
        g2.dispose();
        return bi;
    }
    private static Rectangle calcRotatedSize(Rectangle src, int angel) {
        if (angel >= 90) {
            if(angel / 90 % 2 == 1) {
                int temp = src.height;
                src.height = src.width;
                src.width = temp;
            }
            angel = angel % 90;
        }
        double r = Math.sqrt(src.height * src.height + src.width * src.width) / 2;
        double len = 2 * Math.sin(Math.toRadians(angel) / 2) * r;
        double angel_alpha = (Math.PI - Math.toRadians(angel)) / 2;
        double angel_dalta_width = Math.atan((double) src.height / src.width);
        double angel_dalta_height = Math.atan((double) src.width / src.height);
        int len_dalta_width = (int) (len * Math.cos(Math.PI - angel_alpha - angel_dalta_width));
        int len_dalta_height = (int) (len * Math.cos(Math.PI - angel_alpha - angel_dalta_height));
        int des_width = src.width + len_dalta_width * 2;
        int des_height = src.height + len_dalta_height * 2;
        return new Rectangle(new Dimension(des_width, des_height));
    }

    public static void main(String[] args) {
        try {
            String filePath = "E:\\video\\";
            String fileName = "test.mp4";
            getVedioImg(filePath,fileName);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}