java视频截取帧数_获取视频中指定帧数截图

packagecom.thinkgem.jeesite.modules.sys.utils;importjava.awt.Dimension;importjava.awt.Graphics2D;importjava.awt.Rectangle;importjava.awt.image.BufferedImage;importjava.io.File;importjava.io.IOException;importjava.util.HashMap;importjava.util.Map;importjavax.imageio.ImageIO;importorg.apache.logging.log4j.Logger;importorg.apache.logging.log4j.LogManager;importorg.bytedeco.javacv.FFmpegFrameGrabber;importorg.bytedeco.javacv.Frame;importorg.bytedeco.javacv.Java2DFrameConverter;public classVideoUtils {//main函数中slf4j无法正常显示,这里暂时使用log4j2输出日志,迁移到web项目时,建议使用slf4j//private static final Logger LOG = LoggerFactory.getLogger(ToolVideo.class);

private static final Logger LOG =LogManager.getLogger();//获取要取得的帧数

private static final int GET_FRAMES_LENGTH = 5;public static voidmain(String[] args) {

Map screenshot = getScreenshot("G:\\project\\demoProject\\src\\main\\webapp\\static\\flash\\QaVideo.mp4");

System.out.println(screenshot.toString());

}/***

功能:获取一张视频截图并保存同名的jpg文件到视频同目录下

*

*@paramfilePath 视频文件全路径

*@return

*/

public static MapgetScreenshot(String filePath) {return getScreenshot(filePath, null);

}/***

功能:获取一张视频截图并保存同名的jpg文件到指定目录

*

*@paramfilePath 视频文件地址

*@paramsaveImagePath 截图的图片存放路径(绝对路径,不包含文件名称)

*@return

*/

public static MapgetScreenshot(String filePath, String saveImagePath) {

LOG.info("视频文件[{}]截图开始", filePath);

Map result = new HashMap();

FFmpegFrameGrabber grabber;try{

grabber=FFmpegFrameGrabber.createDefault(filePath);//第一帧图片存储位置(也是视频路径)

String targerFilePath = filePath.substring(0, filePath.lastIndexOf("\\"));//视频文件名

String fileName = filePath.substring(filePath.lastIndexOf("\\") + 1);//图片名称

String targetFileName = fileName.substring(0, fileName.lastIndexOf("."));

grabber.start();//视频总帧数

int videoLength =grabber.getLengthInFrames();

Frame frame= null;int i = 0;while (i

frame =grabber.grabFrame();if ((i > GET_FRAMES_LENGTH) && (frame.image != null)) {break;

}

i++;

}//视频旋转度

String rotate = grabber.getVideoMetadata("rotate");

Java2DFrameConverter converter= newJava2DFrameConverter();//绘制图片

BufferedImage bi =converter.getBufferedImage(frame);if (rotate != null) {//旋转图片

bi =rotate(bi, Integer.parseInt(rotate));

}//图片的类型

String imageMat = "jpg";//图片的完整路径

String imagePath = targerFilePath + File.separator + targetFileName + "." +imageMat;if (null != saveImagePath && !"".equals(saveImagePath)) {//指定路径

imagePath = saveImagePath + "." +imageMat;

}//创建文件

File output = newFile(imagePath);

ImageIO.write(bi, imageMat, output);//拼接Map信息

result.put("videoLength", videoLength); //视频总帧数

result.put("videoWide", bi.getWidth()); //视频的宽

result.put("videoHigh", bi.getHeight());//频的高

long duration = grabber.getLengthInTime() / (1000 * 1000); //此视频时长(s/秒)

result.put("rotate", (null == rotate || "".equals(rotate))? "0" : rotate); //视频的旋转度

result.put("format", grabber.getFormat()); //视频的格式

result.put("imgPath", output.getPath());

result.put("duration", duration);

grabber.stop();

LOG.info("视频文件[{}]截图结束,图片地址为[{}]", filePath, imagePath);

}catch(IOException e) {

LOG.error("视频信息帧数处理发生异常 [{}]", e.getMessage());

e.printStackTrace();

}returnresult;

}//==================== private method ====================

/***

功能:根据视频旋转度来调整图片

*

*@paramsrc 捕获的图像

*@paramangel 视频旋转度

*@returnBufferedImage*/

private static BufferedImage rotate(BufferedImage src, intangel) {int src_width = src.getWidth(null);int src_height = src.getHeight(null);int type =src.getColorModel().getTransparency();

Rectangle rect_des= calcRotatedSize(new Rectangle(newDimension(src_width, src_height)), angel);

BufferedImage bi= newBufferedImage(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();returnbi;

}/***

功能:计算图片旋转大小

*

*@paramsrc 屏幕坐标中捕获的矩形区域

*@paramangel 视频旋转度

*@return

*/

private static Rectangle calcRotatedSize(Rectangle src, intangel) {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 java.awt.Rectangle(newDimension(des_width, des_height));

}

}