This repository has been archived on 2022-04-22. You can view files and clone it, but cannot push or open issues or pull requests.
GPXprocessor/huawei.py
2021-12-25 23:05:03 +08:00

79 lines
2.7 KiB
Python

import os
import gpxpy
import gpxpy.gpx
from datetime import datetime, timedelta
from geopy.distance import geodesic
start_time = {
'20190625骑行.gpx': '2019-06-25 16:48',
'20190627骑行.gpx': '2019-06-27 19:42',
'20190705骑行.gpx': '2019-07-05 18:26',
'20190706骑行.gpx': '2019-07-06 19:37',
'20190709骑行.gpx': '2019-07-09 18:25',
'20190912跑步.gpx': '2019-09-12 18:03',
'20190915跑步.gpx': '2019-09-15 21:02',
'20190916跑步.gpx': '2019-09-16 21:38',
'20190920跑步.gpx': '2019-09-20 21:36',
'20190928跑步.gpx': '2019-09-28 22:41',
'20190930跑步.gpx': '2019-09-30 21:27',
'20191001跑步.gpx': '2019-10-01 17:12',
'20191006跑步.gpx': '2019-10-06 18:10',
'20191010跑步.gpx': '2019-10-10 22:06',
'20191011跑步.gpx': '2019-10-11 21:52',
'20191020跑步.gpx': '2019-10-20 18:08',
'20191030跑步.gpx': '2019-10-30 10:47',
'20191103跑步.gpx': '2019-11-03 17:16',
'20191107跑步.gpx': '2019-11-07 21:33',
'20191108跑步.gpx': '2019-11-08 22:14',
'20200321跑步.gpx': '2020-03-21 19:27',
'20200828骑行.gpx': '2020-08-28 22:19'
}
def process(file):
gpx_file = open('./old/'+file, 'r', encoding='utf-8')
gpx = gpxpy.parse(gpx_file)
s = 0
p = None
for track in gpx.tracks:
time = float(track.extensions[0].text)
for segment in track.segments:
for point in segment.points:
if p:
s += geodesic((point.latitude, point.longitude), (p.latitude, p.longitude)).m
p = point
ngpx = gpxpy.gpx.GPX()
gpx_track = gpxpy.gpx.GPXTrack()
ngpx.tracks.append(gpx_track)
gpx_segment = gpxpy.gpx.GPXTrackSegment()
gpx_track.segments.append(gpx_segment)
p = None
time0 = datetime.strptime(start_time[file], '%Y-%m-%d %H:%M') - timedelta(hours=8)
for track in gpx.tracks:
for segment in track.segments:
for point in segment.points:
if p and point.latitude == p.latitude and point.longitude == p.longitude:
continue
if p:
dis = geodesic((point.latitude, point.longitude), (p.latitude, p.longitude)).m
dtime += time * dis / s
else:
dtime = 0
gpx_segment.points.append(gpxpy.gpx.GPXTrackPoint(point.latitude, point.longitude,
time=time0 + timedelta(seconds=dtime)))
# datetime.strftime(time0 + timedelta(seconds=dtime), '%Y-%m-%dT%H:%M:%S.%fZ')
p = point
with open('./new/'+file, 'w', encoding='utf-8') as f:
f.write(ngpx.to_xml())
files = os.listdir('./old')
for file in files:
process(file)