package per.qy.dexter.video;
import java.awt.image.BufferedImage;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import javax.imageio.ImageIO;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.Java2DFrameConverter;
import org.junit.Test;
public class FFmpegTest {
@Test
public void splitVideo() {
FFmpegFrameGrabber frameGrabber = null;
try {
frameGrabber = FFmpegFrameGrabber.createDefault("d:/111.mp4");
frameGrabber.start();
long frameTime = frameGrabber.getLengthInTime() / 1000000;
int frameLength = frameGrabber.getLengthInFrames();
List<Integer> list = random(frameTime, frameLength);
for (int i = 0; i < frameLength; i++) {
Frame frame = frameGrabber.grabImage();
if (list.contains(i)) {
Java2DFrameConverter converter = new Java2DFrameConverter();
BufferedImage image = converter.getBufferedImage(frame);
FileOutputStream outFile = null;
try {
outFile = new FileOutputStream(
"d:/111/" + UUID.randomUUID().toString().replaceAll("-", "") + ".png");
ImageIO.write(image, "png", outFile);
} catch (IOException e) {
} finally {
try {
if (outFile != null) {
outFile.close();
}
} catch (IOException e) {
}
}
}
}
System.out.println(String.format("视频总时长=%ss,总帧数=%s,抽取帧数=%s", frameTime, frameLength, list.size()));
} catch (org.bytedeco.javacv.FrameGrabber.Exception e) {
} finally {
if (frameGrabber != null) {
try {
frameGrabber.stop();
} catch (org.bytedeco.javacv.FrameGrabber.Exception e) {
}
}
}
}
private List<Integer> random(long frameTime, int frameLength) {
long count = 200;
if (frameTime <= 100) {
count = 10;
} else if (frameTime <= 1800) {
count = frameTime / 10;
}
List<Integer> list = new ArrayList<Integer>();
if (frameLength < count) {
for (int i = 0; i < frameLength; i++) {
list.add(i);
}
return list;
}
long range = (frameLength - 5) / count;
for (long i = range + 5; i < frameLength; i = i + range) {
int rand = (int) ((i - range) + Math.random() * range);
list.add(rand);
}
return list;
}
}
网友评论