Mirrors/srt-to-gpx
1
0
Fork 0
mirror of https://github.com/arru/srt-to-gpx.git synced 2026-03-05 17:56:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Output DJI metadata under extensions in 'dji' namespace

Browse Source
This commit is contained in:
Arvid Rudling
2018-06-13 23:03:17 +02:00
parent cd9f0085df
commit bea90d37a3
1 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
dji_srt.py
View File

@@ -17,6 +17,8 @@ import xml.etree.ElementTree as etree
CHUNK_PARTS_RE = re.compile( CHUNK_PARTS_RE = re.compile(
'^(\d+)\s+([\d:,]+) --> ([\d:,]+)\s+HOME\((.+)\) ([\d\.]+ [\d:]+)\s+GPS\((.+)\) (.+)$', flags=re.S) '^(\d+)\s+([\d:,]+) --> ([\d:,]+)\s+HOME\((.+)\) ([\d\.]+ [\d:]+)\s+GPS\((.+)\) (.+)$', flags=re.S)
EXT_PARTS_RE = re.compile('(\w+):\s?(\S+)')
TIME_PATTERN = "%H:%M:%S,000" TIME_PATTERN = "%H:%M:%S,000"
DATETIME_PATTERN = "%Y.%m.%d %H:%M:%S" DATETIME_PATTERN = "%Y.%m.%d %H:%M:%S"
VIDEO_TIME_FORMAT = "%M:%S" VIDEO_TIME_FORMAT = "%M:%S"
@@ -46,8 +48,15 @@ class DJITrackPoint(object):
if self.time: if self.time:
time = etree.SubElement(wpt, 'time') time = etree.SubElement(wpt, 'time')
time.text = self.time.astimezone(datetime.timezone.utc).strftime(GPX_UTC_DATETIME_FORMAT) time.text = self.time.astimezone(datetime.timezone.utc).strftime(GPX_UTC_DATETIME_FORMAT)
return wpt if self.extra:
extra_container = etree.SubElement(wpt, 'extensions')
for name, value in self.extra.items():
normalized_name = name.lower()
extra_item = etree.SubElement(extra_container, "dji:%s" % normalized_name)
extra_item.text = value
return wpt
class Track(object): class Track(object):
@@ -125,6 +134,7 @@ class GPXDocument(object):
arg_parser = argparse.ArgumentParser(description='Convert DJI format .srt files to .gpx') arg_parser = argparse.ArgumentParser(description='Convert DJI format .srt files to .gpx')
arg_parser.add_argument('-e', action='store_true', help='Output DJI metadata such as exposure and air pressure in GPX extension tag')
arg_parser.add_argument('input', metavar='SRT', help='Input .srt file path') arg_parser.add_argument('input', metavar='SRT', help='Input .srt file path')
args = arg_parser.parse_args() args = arg_parser.parse_args()
@@ -152,7 +162,17 @@ for c in chunks:
aircraft_lon, aircraft_lat, aircraft_ele = map(lambda l: float(l), chunk_parts[5].split(',')) aircraft_lon, aircraft_lat, aircraft_ele = map(lambda l: float(l), chunk_parts[5].split(','))
points.append(DJITrackPoint(aircraft_lon, aircraft_lat, aircraft_ele, point_time)) ext_items = {}
if args.e:
ext_items = dict(EXT_PARTS_RE.findall(chunk_parts[6]))
ext_items.update({
'video_begin': video_begin_time.strftime(VIDEO_TIME_FORMAT),
'video_end': video_end_time.strftime(VIDEO_TIME_FORMAT),
'home_lon': home_lon,
'home_lat': home_lat
})
points.append(DJITrackPoint(aircraft_lon, aircraft_lat, aircraft_ele, point_time, ext_items))
dest_dir, input_filename = os.path.split(args.input) dest_dir, input_filename = os.path.split(args.input)
dest_filename = os.path.splitext(input_filename)[0] dest_filename = os.path.splitext(input_filename)[0]