package com.ld.igds.util;
|
|
import java.io.IOException;
|
import java.io.InputStreamReader;
|
import java.io.LineNumberReader;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.concurrent.*;
|
|
/**
|
* 使用ffmpeg.exe的命令执行截图，不要添加任何JAR包。
|
* <p>
|
* 链接：https://pan.baidu.com/s/1MEvUXPMe6bDOsQYt5LMIxA 提取码：y177
|
*
|
* @author: andy.jia
|
* @description:
|
* @version:
|
* @data:2020年6月13日
|
*/
|
public class FfmpegUtil {
|
|
/**
|
* 定义ffmpeg工具的位置
|
*/
|
public static final String FFMPEG_PATH = "D:/IGDS/ffmpeg-win64-static/bin/ffmpeg.exe";
|
|
public static final String IMG_SIZE = "1280*720";
|
|
public static final Map<String, Process> contextProcess = new HashMap<>();
|
|
/**
|
* 关闭进程
|
*
|
* @param key
|
*/
|
public static void destroyProcess(String key) {
|
Process process = contextProcess.get(key);
|
contextProcess.remove(key);
|
if (null != process) {
|
process.destroy();
|
}
|
}
|
|
public static void addProcess(String key, Process process) {
|
contextProcess.put(key, process);
|
}
|
|
/**
|
* 根据视频流转换为图片
|
*
|
* @param mediaAddr
|
* @param key
|
* @return
|
*/
|
public static String cutImgByMediaAddr(String mediaAddr, String baseFile,String fileName, String key) {
|
// 图片命令
|
StringBuilder command = new StringBuilder();
|
command.append(FFMPEG_PATH);
|
command.append(" -i ");
|
command.append(mediaAddr);
|
command.append(" -vframes");
|
command.append(" 1");
|
command.append(" -y");
|
command.append(" -f");
|
command.append(" image2");
|
command.append(" -t");
|
command.append(" 1");
|
command.append(" -s ");
|
command.append(IMG_SIZE);
|
command.append(" ");
|
command.append(baseFile+fileName);
|
|
InputStreamReader isr = null;
|
LineNumberReader input = null;
|
Process process;
|
try {
|
process = Runtime.getRuntime().exec(command.toString());
|
addProcess(key, process);
|
isr = new InputStreamReader(process.getInputStream());
|
input = new LineNumberReader(isr);
|
while (input.readLine() != null) {
|
}
|
} catch (java.io.IOException e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
if (null != input)
|
input.close();
|
if (null != isr)
|
isr.close();
|
} catch (IOException e1) {
|
e1.printStackTrace();
|
}
|
|
destroyProcess(key);
|
}
|
return fileName;
|
}
|
|
public static String cutImgByMediaAddr(String mediaAddr, String diskFile, String key) {
|
// 图片命令
|
StringBuilder command = new StringBuilder();
|
command.append(FFMPEG_PATH);
|
command.append(" -i ");
|
command.append(mediaAddr);
|
command.append(" -vframes");
|
command.append(" 1");
|
command.append(" -y");
|
command.append(" -f");
|
command.append(" image2");
|
command.append(" -t");
|
command.append(" 1");
|
command.append(" -s ");
|
command.append(IMG_SIZE);
|
command.append(" ");
|
command.append(diskFile);
|
|
InputStreamReader isr = null;
|
LineNumberReader input = null;
|
Process process;
|
try {
|
process = Runtime.getRuntime().exec(command.toString());
|
addProcess(key, process);
|
isr = new InputStreamReader(process.getInputStream());
|
input = new LineNumberReader(isr);
|
while (input.readLine() != null) {
|
}
|
} catch (java.io.IOException e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
if (null != input)
|
input.close();
|
if (null != isr)
|
isr.close();
|
} catch (IOException e1) {
|
e1.printStackTrace();
|
}
|
|
destroyProcess(key);
|
}
|
return diskFile;
|
}
|
|
|
/**
|
* 调用抓拍，每个动作执行6S
|
*
|
* @param mediaAddr
|
* @param baseFile
|
* @return
|
*/
|
public static String snapImg(String mediaAddr, String baseFile, String fileName) {
|
final ExecutorService exec = Executors.newFixedThreadPool(1);
|
String key = ContextUtil.getUUID();
|
|
Callable<String> call = new Callable<String>() {
|
public String call() throws Exception {
|
return cutImgByMediaAddr(mediaAddr, baseFile,fileName, key);
|
}
|
};
|
try {
|
Future<String> future = exec.submit(call);
|
future.get(6 * 1000, TimeUnit.MILLISECONDS);
|
} catch (Exception ex) {
|
destroyProcess(key);
|
return null;
|
}
|
exec.shutdown();
|
|
return fileName;
|
}
|
|
public static String cutImgByMediaAddr1(String mediaAddr, String outFile) {
|
final ExecutorService exec = Executors.newFixedThreadPool(1);
|
String key = ContextUtil.getUUID();
|
Callable<String> call = new Callable<String>() {
|
public String call() throws Exception {
|
return cutImgByMediaAddr(mediaAddr, outFile, key);
|
}
|
};
|
try {
|
Future<String> future = exec.submit(call);
|
future.get(5 * 1000, TimeUnit.MILLISECONDS);
|
} catch (Exception ex) {
|
destroyProcess(key);
|
return "=============后台保存图片执行时间超时======";
|
}
|
exec.shutdown();
|
|
return outFile;
|
}
|
|
public static String cutImgByMediaAddr2(String mediaAddr, String outFile) {
|
final ExecutorService exec = Executors.newFixedThreadPool(1);
|
String key = ContextUtil.getUUID();
|
Callable<String> call = new Callable<String>() {
|
public String call() throws Exception {
|
return cutImgByMediaAddr(mediaAddr, outFile, key);
|
}
|
};
|
try {
|
Future<String> future = exec.submit(call);
|
future.get(5 * 1000, TimeUnit.MILLISECONDS);
|
} catch (Exception ex) {
|
destroyProcess(key);
|
return "=============后台保存图片执行时间超时======";
|
}
|
exec.shutdown();
|
|
return outFile;
|
}
|
|
public static String cutImgByMediaAddr3(String mediaAddr, String outFile) {
|
final ExecutorService exec = Executors.newFixedThreadPool(1);
|
String key = ContextUtil.getUUID();
|
Callable<String> call = new Callable<String>() {
|
public String call() throws Exception {
|
return cutImgByMediaAddr(mediaAddr, outFile, key);
|
}
|
};
|
try {
|
Future<String> future = exec.submit(call);
|
future.get(5 * 1000, TimeUnit.MILLISECONDS);
|
} catch (Exception ex) {
|
destroyProcess(key);
|
return "=============后台保存图片执行时间超时======";
|
}
|
exec.shutdown();
|
|
return outFile;
|
}
|
|
public static String getImgStr(String imgFile) {
|
// FileInputStream in = null;
|
// byte[] data = null;
|
// // 读取图片字节数组
|
// try {
|
// in = new FileInputStream(imgFile);
|
// data = new byte[in.available()];
|
// in.read(data);
|
// in.close();
|
// } catch (IOException e) {
|
// e.printStackTrace();
|
// return null;
|
// }
|
// String imgData = "data:image/jpg;base64,"
|
// + Base64Util.encode2String(data);
|
// return imgData;
|
|
return Base64Util.getImageStr(imgFile);
|
}
|
|
public static void main(String[] args) {
|
|
String rtsp = "rtsp://admin:admin@192.168.1.102:554/h264/ch1/sub/av_stream";
|
|
rtsp = "rtsp://admin:smd123456@192.168.2.61:554/cam/realmonitor?channel=1&subtype=1&proto=Dahua3";
|
|
FfmpegUtil
|
.cutImgByMediaAddr1(rtsp,
|
"D:\\IGDS\\FILE\\INOUT\\" + System.currentTimeMillis()
|
+ ".jpg");
|
|
System.out.println("===================");
|
}
|
|
}
|
