ps:该工具类不仅适用于监控视频抓拍,也适用于各种视屏类链接抓拍
一、添加依赖
<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>
<!--这个依赖区分Linux系统和windows系统-->
<dependency>
<groupId>com.github.hoary.ffmpeg</groupId>
<artifactId>FFmpeg-windows-x86_64</artifactId>
<version>2.0.1.1</version>
</dependency>
<!-- <dependency>-->
<!-- <groupId>com.github.hoary.ffmpeg</groupId>-->
<!-- <artifactId>FFmpeg-linux-x86</artifactId>-->
<!-- <version>2.0.1.1</version>-->
<!-- </dependency>-->
二、工具类(开箱即用)
public class VideoUtil {
/**
* 抓拍图片
* @param streamURL 监控链接
* @param filePath 抓拍保存的文件路径
*/
public void grabImage(String streamURL, String filePath) {
// 获取视频源
FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(streamURL);
try {
grabber.start();
Frame frame = grabber.grabImage();
Java2DFrameConverter converter = new Java2DFrameConverter();
BufferedImage bi = converter.convert(frame);
// 对frameRate帧进行一次抓取
ImageIO.write(bi, "jpg", new File(filePath));
System.out.println("抓拍成功");
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
grabber.close();
} catch (FrameGrabber.Exception e) {
throw new RuntimeException(e);
}
}
}
/**
* 录屏
* @param filePath 录屏保存的文件路径
* @param timesSec 录屏时长,单位秒
* @param streamURL streamURL 监控链接
*/
public void recordVideo(String filePath, Long timesSec, String streamURL) {
CompletableFuture.runAsync(() -> {
// 获取视频源
FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(streamURL);
FFmpegFrameRecorder recorder = null;
try {
grabber.start();
Frame frame = grabber.grabFrame();
createFile(filePath);
// 流媒体输出地址,分辨率(长,高),是否录制音频(0:不录制/1:录制)
recorder = new FFmpegFrameRecorder(filePath, 1280, 720, 1);
recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);// 直播流格式
recorder.setFormat("mp4");// 录制的视频格式
recorder.setFrameRate(25);// 帧数
//百度翻译的比特率,默认400000,调成800000比较合适
recorder.setVideoBitrate(800000);
recorder.start();
System.out.println("推流开始");
// 计算结束时间
long endTime = System.currentTimeMillis() + timesSec * 1000;
// 如果没有到录制结束时间并且获取到了下一帧则继续录制
while ((System.currentTimeMillis() < endTime) && (frame != null)) {
recorder.record(frame);//录制
frame = grabber.grabFrame();//获取下一帧
}
recorder.record(frame);
} catch (Exception e) {
e.printStackTrace();
} finally {
//停止录制
try {
grabber.stop();
} catch (FrameGrabber.Exception e) {
e.printStackTrace();
}
if (recorder != null) {
try {
recorder.stop();
} catch (FrameRecorder.Exception e) {
e.printStackTrace();
}
}
System.out.println("录制完成,录制时长:" + timesSec + "秒");
}
});
}
private void createFile(String filePath) {
File outFile = new File(filePath);
if (filePath.isEmpty() || !outFile.exists() || outFile.isFile()) {
try {
outFile.createNewFile();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}