@TOC
目录
简介
通过 javaCV 视频工具—截取视频缩略图、获取视频属性
依赖引入
<!--javaCV 视频工具 -->
<dependency>
<groupId>org.bytedeco</groupId>
<artifactId>javacv-platform</artifactId>
<version>1.5</version>
</dependency>实现
@Slf4j
public class VideoUtils {
private static final String IMAGEMAT = "png";
private static final String ROTATE = "rotate";
/**
* 默认截取视频的两头帧为封面
*/
public static final int MOD = 2;
/**
* 视频缩略图后缀
*/
private static final String VIDEO_THUMBNAIL_SUF = "th.png";
/**
* 视频缩略图前缀
*/
private static final String VIDEO_THUMBNAIL_PRE = "video/thumbnail/";
private static final String SYMBOL = ".";
/**
* 获取视频缩略图
* @param filePath：视频门路
* @param mod：视频长度 /mod 获取第几帧
* @throws Exception
*/
public static String randomGrabberFFmpegImage(String filePath, int mod) {
String targetFilePath = "";
try{FFmpegFrameGrabber ff = FFmpegFrameGrabber.createDefault(filePath);
ff.start();
// 图片地位是否正确
String rotate = ff.getVideoMetadata(ROTATE);
// 获取帧数
int ffLength = ff.getLengthInFrames();
Frame f;
int i = 0;
// 设置截取帧数
int index = ffLength / mod;
while (i < ffLength) {f = ff.grabImage();
if(i == index){if (null != rotate && rotate.length() > 1) {OpenCVFrameConverter.ToIplImage converter = new OpenCVFrameConverter.ToIplImage();
IplImage src = converter.convert(f);
f = converter.convert(rotate(src, Integer.parseInt(rotate)));
}
targetFilePath = getImagePath(filePath, i);
doExecuteFrame(f, targetFilePath);
break;
}
i++;
}
ff.stop();}catch (Exception e){log.error("获取视频缩略图异样：" + e.getMessage());
}
return targetFilePath;
}
/**
* 随机生成生成缩略图寄存门路
* @param filePath：视频门路
* @param index：第几帧
* @return：缩略图的寄存门路
*/
private static String getImagePath(String filePath, int index){String fileName = FileUtils.getName(filePath);
// 去后缀
fileName = fileName.substring(0, fileName.indexOf(SYMBOL));
return TencentCosConfig.baseUrl + VIDEO_THUMBNAIL_PRE + DateUtils.datePath() + "/" + fileName + "_" + index + VIDEO_THUMBNAIL_SUF;}
/**
* 旋转图片
* @param src
* @param angle
* @return
*/
public static IplImage rotate(IplImage src, int angle) {IplImage img = IplImage.create(src.height(), src.width(), src.depth(), src.nChannels());
opencv_core.cvTranspose(src, img);
opencv_core.cvFlip(img, img, angle);
return img;
}
/**
* 截取缩略图
* @param f
* @param targerFilePath: 封面图片
*/
public static void doExecuteFrame(Frame f, String targerFilePath) {COSClient cosClient = TencentCosUtils.initCosClient();
if (null == f || null == f.image) {return;}
Java2DFrameConverter converter = new Java2DFrameConverter();
BufferedImage bi = converter.getBufferedImage(f);
ByteArrayOutputStream out = new ByteArrayOutputStream();
try {ImageIO.write(bi, IMAGEMAT, out);
// 获取文件流
InputStream bufferedImage = new ByteArrayInputStream(out.toByteArray());
int length = out.size();
ObjectMetadata objectMetadata = new ObjectMetadata();
// 从输出流上传必须制订 content length, 否则 http 客户端可能会缓存所有数据，存在内存 OOM 的状况
objectMetadata.setContentLength(length);
// 默认下载时依据 cos 门路 key 的后缀返回响应的 contenttype, 上传时设置 contenttype 会笼罩默认值
PutObjectRequest putObjectRequest = new PutObjectRequest(TencentCosConfig.bucket, targerFilePath, bufferedImage, objectMetadata);
PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
log.info("腾讯 COS 上传视频缩略图胜利：{}", putObjectResult.getETag());
// 敞开输入输出流
bufferedImage.close();
out.close();} catch (IOException e) {e.printStackTrace();
} finally {cosClient.shutdown();
}
}
/**
* 依据视频长度随机生成随机数汇合
* @param baseNum: 根底数字，此处为视频长度
* @param length：随机数汇合长度
* @return: 随机数汇合
*/
public static List<Integer> random(int baseNum, int length) {List<Integer> list = new ArrayList<Integer>(length);
while (list.size() < length) {Integer next = (int) (Math.random() * baseNum);
if (list.contains(next)) {continue;}
list.add(next);
}
Collections.sort(list);
return list;
}
/**
* 获取视频时长 单位 / 秒
* @param video
* @return
*/
public static long getVideoDuration(File video) {
long duration = 0L;
FFmpegFrameGrabber ff = new FFmpegFrameGrabber(video);
try {ff.start();
duration = ff.getLengthInTime() / (1000 * 1000);
ff.stop();} catch (FrameGrabber.Exception e) {e.printStackTrace();
}
return duration;
}
/**
* 获取视频时长 单位 / 秒
* @param inputStream 输出流
* @return
*/
public static long getVideoDuration(InputStream inputStream) {
long duration = 0L;
FFmpegFrameGrabber ff = new FFmpegFrameGrabber(inputStream);
try {ff.start();
duration = ff.getLengthInTime() / (1000 * 1000);
ff.stop();} catch (FrameGrabber.Exception e) {e.printStackTrace();
}
return duration;
}
/**
* 转换视频文件为 mp4
* @param file
* @return
*/
public static String convertToMp4(File file) {FFmpegFrameGrabber frameGrabber = new FFmpegFrameGrabber(file);
String fileName = null;
Frame captured_frame = null;
FFmpegFrameRecorder recorder = null;
try {frameGrabber.start();
fileName = file.getAbsolutePath() + "__.mp4";
recorder = new FFmpegFrameRecorder(fileName, frameGrabber.getImageWidth(), frameGrabber.getImageHeight(), frameGrabber.getAudioChannels());
recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264); //avcodec.AV_CODEC_ID_H264 //AV_CODEC_ID_MPEG4
recorder.setFormat("mp4");
recorder.setFrameRate(frameGrabber.getFrameRate());
//recorder.setSampleFormat(frameGrabber.getSampleFormat()); //
recorder.setSampleRate(frameGrabber.getSampleRate());
recorder.setAudioChannels(frameGrabber.getAudioChannels());
recorder.setFrameRate(frameGrabber.getFrameRate());
recorder.start();
while ((captured_frame = frameGrabber.grabFrame()) != null) {
try {recorder.setTimestamp(frameGrabber.getTimestamp());
recorder.record(captured_frame);
} catch (FrameRecorder.Exception e) {e.printStackTrace();
}
}
recorder.stop();
recorder.release();
frameGrabber.stop();} catch (Exception | FrameRecorder.Exception e) {e.printStackTrace();
}
return fileName;
}
}