springboot 获取视频第一帧和时长

package com.example.demo.utils; import com.example.demo.entity.VideoInfo; import lombok.extern.slf4j.Slf4j; 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.ByteArrayOutputStream; import java.io.File; @Slf4j public class VideoTools { // 存放截取视频某一帧的图片 public static String videoFramesPath = System.getProperty("user.dir") + "/images/"; /** * 将视频文件帧处理并以“jpg”格式进行存储。 依赖FrameToBufferedImage方法：将frame转换为bufferedImage对象 * * @param path */ public static VideoInfo grabberVideoFramer(String path) { VideoInfo videoInfo = new VideoInfo(); // Frame对象 Frame frame = null; // 标识 int flag = 0; try { FFmpegFrameGrabber fFmpegFrameGrabber = new FFmpegFrameGrabber(path); fFmpegFrameGrabber.start(); //视频旋转角度，可能是null String rotate_old = fFmpegFrameGrabber.getVideoMetadata("rotate"); // 获取视频总帧数 int ftp = fFmpegFrameGrabber.getLengthInFrames(); System.out.println("时长 " + ftp / fFmpegFrameGrabber.getFrameRate() / 60); while (flag <= ftp) { frame = fFmpegFrameGrabber.grabImage(); /* 对视频的第五帧进行处理 */ if (frame != null && flag == 1) { BufferedImage bufferedImage = FrameToBufferedImage(frame); ByteArrayOutputStream out = new ByteArrayOutputStream(); ImageIO.write(bufferedImage, "jpg", out); byte[] b = out.toByteArray(); fFmpegFrameGrabber.stop(); VideoInfo info; //有需要旋转 if (rotate_old != null && !rotate_old.isEmpty()) { int rotate = Integer.parseInt(rotate_old); info = VideoTools.rotatePhonePhoto(bufferedImage, rotate); } else { //不需要旋转 info = VideoTools.rotatePhonePhoto(bufferedImage, 0); } videoInfo.setWidth(info.getWidth()); videoInfo.setHeight(info.getHeight()); //保存图片做封面 videoInfo.setCoverUrl(info.getCoverUrl()); fFmpegFrameGrabber.close(); return videoInfo; } flag++; } } catch (Exception E) { E.printStackTrace(); } return null; } public static BufferedImage FrameToBufferedImage(Frame frame) { // 创建BufferedImage对象 Java2DFrameConverter converter = new Java2DFrameConverter(); BufferedImage bufferedImage = converter.getBufferedImage(frame); return bufferedImage; } /** * 旋转照片 * * @return */ public static VideoInfo rotatePhonePhoto(BufferedImage bufferedImage, int angel) { // String path=""; BufferedImage src; VideoInfo info = new VideoInfo(); try { src = bufferedImage; int src_width = src.getWidth(null); int src_height = src.getHeight(null); int swidth = src_width; int sheight = src_height; if (angel == 90 || angel == 270) { swidth = src_height; sheight = src_width; } Rectangle rect_des = new Rectangle(new Dimension(swidth, sheight)); BufferedImage res = new BufferedImage(rect_des.width, rect_des.height, BufferedImage.TYPE_INT_RGB); Graphics2D g2 = res.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, null, null); String imgName = System.currentTimeMillis() + ".jpg"; info.setWidth(rect_des.width); info.setHeight(rect_des.height); info.setCoverUrl(imgName); ImageIO.write(res, "jpg", new File(videoFramesPath + imgName)); System.out.println("***********" + videoFramesPath + imgName); System.out.println(rect_des.width + "******************" + rect_des.height); // path=imgName; } catch (Exception e) { e.printStackTrace(); } return info; } /** * 测试： 1、在D盘中新建一个test文件夹，test中再分成video和img，在video下存入一个视频，并命名为test D:/test/video D:/test/img * * @param args */ public static void main(String[] args) { //竖屏 VideoInfo videoInfo = grabberVideoFramer(videoFramesPath + "1616983133.mp4"); //横屏 // byte[] bytes= grabberVideoFramer("http://192.168.10.104:8181/file/20200218114723HDu3hhxqIT.mp4"); System.out.println(videoInfo); } }