111 lines
4.0 KiB
Python
111 lines
4.0 KiB
Python
|
import re
|
|||
|
|
import pandas as pd
|
|||
|
|
from dataclasses import dataclass
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class DjiSrt:
|
|||
|
|
"""解析 DJI SRT 文件中的元数据"""
|
|||
|
|
FrameCnt: int
|
|||
|
|
Datetime: int # 时间戳（毫秒）
|
|||
|
|
rel_alt: float
|
|||
|
|
abs_alt: float
|
|||
|
|
gb_yaw: float
|
|||
|
|
gb_pitch: float
|
|||
|
|
gb_roll: float
|
|||
|
|
dehaze_level: int
|
|||
|
|
dehaze_mode: int
|
|||
|
|
iso: Optional[int] = None
|
|||
|
|
shutter: Optional[str] = None
|
|||
|
|
fnum: Optional[float] = None
|
|||
|
|
ev: Optional[int] = None
|
|||
|
|
ae_meter_md: Optional[int] = None
|
|||
|
|
focal_len: Optional[float] = None
|
|||
|
|
dzoom_ratio: Optional[float] = None
|
|||
|
|
latitude: Optional[float] = None
|
|||
|
|
longitude: Optional[float] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def parse_srt_file(srt_file: str) -> list[DjiSrt]:
|
|||
|
|
"""解析 SRT 文件，返回 DjiSrt 对象列表"""
|
|||
|
|
data = []
|
|||
|
|
frame_pattern = re.compile(r"FrameCnt: (\d+) (\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}\.\d{3})")
|
|||
|
|
rel_abs_alt_pattern = re.compile(r"\[rel_alt: ([\d\.\-]+) abs_alt: ([\d\.\-]+)\]")
|
|||
|
|
gb_pattern = re.compile(r"\[gb_yaw: ([\d\.\-]+) gb_pitch: ([\d\.\-]+) gb_roll: ([\d\.\-]+)\]")
|
|||
|
|
dehaze_pattern = re.compile(r"\[dehaze_level: (\d+)\] \[dehaze_mode: (\d+)\]")
|
|||
|
|
fields_pattern = re.compile(r"\[([a-zA-Z_]+): ([\d\.\-\/]+)\]") # 匹配其他字段
|
|||
|
|
|
|||
|
|
with open(srt_file, 'r', encoding='utf-8') as file:
|
|||
|
|
content = file.read()
|
|||
|
|
|
|||
|
|
entries = content.split('\n\n') # 按 SRT 块分割
|
|||
|
|
|
|||
|
|
for entry in entries:
|
|||
|
|
if not entry.strip():
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 初始化 DjiSrt 对象（先设为 None，后续填充）
|
|||
|
|
frame_info = DjiSrt(
|
|||
|
|
FrameCnt=0, Datetime=0, rel_alt=0.0, abs_alt=0.0,
|
|||
|
|
gb_yaw=0.0, gb_pitch=0.0, gb_roll=0.0,
|
|||
|
|
dehaze_level=0, dehaze_mode=0
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 匹配 FrameCnt 和 Datetime
|
|||
|
|
frame_match = frame_pattern.search(entry)
|
|||
|
|
if frame_match:
|
|||
|
|
frame_info.FrameCnt = int(frame_match.group(1))
|
|||
|
|
datetime_str = frame_match.group(2)
|
|||
|
|
frame_info.Datetime = int(pd.to_datetime(datetime_str).timestamp() * 1000)
|
|||
|
|
|
|||
|
|
# 匹配 rel_alt 和 abs_alt
|
|||
|
|
rel_abs_alt_match = rel_abs_alt_pattern.search(entry)
|
|||
|
|
if rel_abs_alt_match:
|
|||
|
|
frame_info.rel_alt = float(rel_abs_alt_match.group(1))
|
|||
|
|
frame_info.abs_alt = float(rel_abs_alt_match.group(2))
|
|||
|
|
|
|||
|
|
# 匹配 gb_yaw, gb_pitch, gb_roll
|
|||
|
|
gb_match = gb_pattern.search(entry)
|
|||
|
|
if gb_match:
|
|||
|
|
frame_info.gb_yaw = float(gb_match.group(1))
|
|||
|
|
frame_info.gb_pitch = float(gb_match.group(2))
|
|||
|
|
frame_info.gb_roll = float(gb_match.group(3))
|
|||
|
|
|
|||
|
|
# 匹配 dehaze_level 和 dehaze_mode
|
|||
|
|
dehaze_match = dehaze_pattern.search(entry)
|
|||
|
|
if dehaze_match:
|
|||
|
|
frame_info.dehaze_level = int(dehaze_match.group(1))
|
|||
|
|
frame_info.dehaze_mode = int(dehaze_match.group(2))
|
|||
|
|
|
|||
|
|
# 匹配其他字段（iso, shutter, fnum, ev, ae_meter_md, focal_len, dzoom_ratio, latitude, longitude）
|
|||
|
|
fields_matches = fields_pattern.findall(entry)
|
|||
|
|
for field, value in fields_matches:
|
|||
|
|
if field == "iso":
|
|||
|
|
frame_info.iso = int(value)
|
|||
|
|
elif field == "shutter":
|
|||
|
|
frame_info.shutter = value
|
|||
|
|
elif field == "fnum":
|
|||
|
|
frame_info.fnum = float(value)
|
|||
|
|
elif field == "ev":
|
|||
|
|
frame_info.ev = int(value)
|
|||
|
|
elif field == "ae_meter_md":
|
|||
|
|
frame_info.ae_meter_md = int(value)
|
|||
|
|
elif field == "focal_len":
|
|||
|
|
frame_info.focal_len = float(value)
|
|||
|
|
elif field == "dzoom_ratio":
|
|||
|
|
frame_info.dzoom_ratio = float(value)
|
|||
|
|
elif field == "latitude":
|
|||
|
|
frame_info.latitude = float(value)
|
|||
|
|
elif field == "longitude":
|
|||
|
|
frame_info.longitude = float(value)
|
|||
|
|
|
|||
|
|
data.append(frame_info)
|
|||
|
|
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
srt_data = parse_srt_file(r"D:\project\verification\DJI_SRT_Tool-main\example\hongxian.srt")
|
|||
|
|
for entry in srt_data[:2]: # 打印前两条数据
|
|||
|
|
print(entry)
|