package org.sxkj.common.utils;
|
|
import org.sxkj.common.model.SubtitleFrame;
|
|
import java.io.IOException;
|
import java.nio.file.Files;
|
import java.nio.file.Paths;
|
import java.time.LocalDateTime;
|
import java.time.LocalTime;
|
import java.time.format.DateTimeFormatter;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
/**
|
* srt 文件解析工具类
|
*/
|
public class SrtParserUtils {
|
|
private static final DateTimeFormatter TIME_FORMATTER = DateTimeFormatter.ofPattern("HH:mm:ss,SSS");
|
private static final DateTimeFormatter TIMESTAMP_FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss.SSS");
|
|
/**
|
* srt 文件解析
|
* @param filePath
|
* @return
|
* @throws IOException
|
*/
|
public static List<SubtitleFrame> parseSrtFile(String filePath) throws IOException {
|
List<SubtitleFrame> frames = new ArrayList<>();
|
String content = new String(Files.readAllBytes(Paths.get(filePath)));
|
String[] blocks = content.split("\\n\\n+");
|
|
for (String block : blocks) {
|
if (block.trim().isEmpty()) {
|
continue;
|
}
|
|
String[] lines = block.split("\\n");
|
if (lines.length < 3) {
|
continue;
|
}
|
|
try {
|
int index = Integer.parseInt(lines[0].trim());
|
String[] timeRange = lines[1].trim().split(" --> ");
|
String startTime = timeRange[0].trim();
|
String endTime = timeRange[1].trim();
|
|
StringBuilder contentBuilder = new StringBuilder();
|
for (int i = 2; i < lines.length; i++) {
|
contentBuilder.append(lines[i].trim()).append(" ");
|
}
|
String contentText = contentBuilder.toString().trim();
|
|
// 解析帧计数和时间戳
|
Pattern framePattern = Pattern.compile("FrameCnt:\\s*(\\d+)\\s+(\\d{4}-\\d{2}-\\d{2}\\s+\\d{2}:\\d{2}:\\d{2}\\.\\d{3})");
|
Matcher frameMatcher = framePattern.matcher(contentText);
|
if (!frameMatcher.find()) {
|
continue;
|
}
|
|
int frameCount = Integer.parseInt(frameMatcher.group(1));
|
LocalDateTime timestamp = LocalDateTime.parse(frameMatcher.group(2), TIMESTAMP_FORMATTER);
|
|
// 解析元数据
|
Map<String, String> metadata = parseMetadata(contentText);
|
|
// 构建相机参数
|
Map<String, String> cameraParams = new HashMap<>();
|
cameraParams.put("iso", metadata.get("iso"));
|
cameraParams.put("shutter", metadata.get("shutter"));
|
cameraParams.put("fnum", metadata.get("fnum"));
|
cameraParams.put("ev", metadata.get("ev"));
|
cameraParams.put("color_md", metadata.get("color_md"));
|
cameraParams.put("ae_meter_md", metadata.get("ae_meter_md"));
|
cameraParams.put("focal_len", metadata.get("focal_len"));
|
cameraParams.put("dzoom_ratio", metadata.get("dzoom_ratio"));
|
cameraParams.put("dehaze_level", metadata.get("dehaze_level"));
|
cameraParams.put("dehaze_mode", metadata.get("dehaze_mode"));
|
|
// 构建GPS数据
|
Map<String, Double> gpsData = new HashMap<>();
|
gpsData.put("latitude", Double.parseDouble(metadata.getOrDefault("latitude", "0")));
|
gpsData.put("longitude", Double.parseDouble(metadata.getOrDefault("longitude", "0")));
|
gpsData.put("rel_alt", Double.parseDouble(metadata.getOrDefault("rel_alt", "0")));
|
gpsData.put("abs_alt", Double.parseDouble(metadata.getOrDefault("abs_alt", "0")));
|
|
// 构建方向数据
|
Map<String, Double> orientation = new HashMap<>();
|
orientation.put("yaw", Double.parseDouble(metadata.getOrDefault("gb_yaw", "0")));
|
orientation.put("pitch", Double.parseDouble(metadata.getOrDefault("gb_pitch", "0")));
|
orientation.put("roll", Double.parseDouble(metadata.getOrDefault("gb_roll", "0")));
|
|
// 其他数据
|
Map<String, String> otherData = new HashMap<>();
|
for (Map.Entry<String, String> entry : metadata.entrySet()) {
|
if (!cameraParams.containsKey(entry.getKey()) &&
|
!gpsData.containsKey(entry.getKey()) &&
|
!orientation.containsKey(entry.getKey())) {
|
otherData.put(entry.getKey(), entry.getValue());
|
}
|
}
|
|
SubtitleFrame frame = new SubtitleFrame(
|
index, startTime, endTime, frameCount, timestamp,
|
cameraParams, gpsData, orientation, otherData
|
);
|
|
frames.add(frame);
|
} catch (Exception e) {
|
System.err.println("Error parsing block: " + e.getMessage());
|
continue;
|
}
|
}
|
|
return frames;
|
}
|
|
private static LocalTime parseSrtTime(String timeStr) {
|
// 处理SRT时间格式，如 00:00:00,000
|
timeStr = timeStr.replace(',', '.');
|
return LocalTime.parse(timeStr, TIME_FORMATTER);
|
}
|
|
private static Map<String, String> parseMetadata(String text) {
|
// 先处理特殊字段
|
text = text.replaceAll("\\[rel_alt:\\s*([-\\d.]+)\\s*abs_alt:\\s*([-\\d.]+)\\]",
|
"[rel_alt: $1] [abs_alt: $2]");
|
|
text = text.replaceAll("\\[gb_yaw:\\s*([-\\d.]+)\\s*gb_pitch:\\s*([-\\d.]+)\\s*gb_roll:\\s*([-\\d.]+)\\]",
|
"[gb_yaw: $1] [gb_pitch: $2] [gb_roll: $3]");
|
|
// 解析所有键值对
|
Map<String, String> metadata = new HashMap<>();
|
Pattern pattern = Pattern.compile("\\[([^:]+):\\s*([^\\]]+)\\]");
|
Matcher matcher = pattern.matcher(text);
|
|
while (matcher.find()) {
|
metadata.put(matcher.group(1).trim(), matcher.group(2).trim());
|
}
|
|
return metadata;
|
}
|
|
// public static void main(String[] args) {
|
// try {
|
// String srtPath = "C:\\Users\\Administrator\\Desktop\\temp\\视频\\倾斜视频\\DJI_20250527181054_0002_V\\out.srt";
|
// List<SubtitleFrame> frames = SrtParserUtils.parseSrtFile(srtPath);
|
//
|
// for (SubtitleFrame frame : frames) {
|
// System.out.println("Frame #" + frame.getIndex());
|
// System.out.println("Time: " + frame.getStartTime() + " - " + frame.getEndTime());
|
// System.out.println("Frame count: " + frame.getFrameCount());
|
// System.out.println("Timestamp: " + frame.getTimestamp());
|
// System.out.println("GPS: " + frame.getGpsData());
|
// System.out.println("Orientation: " + frame.getOrientation());
|
// System.out.println("----------------------");
|
// }
|
// } catch (IOException e) {
|
// e.printStackTrace();
|
// }
|
// }
|
}
|
