/
tracks.py
139 lines (125 loc) · 5.64 KB
/
tracks.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
from math import sqrt
import pandas as pd
from haversine import haversine, Unit
import requests
class Track(object):
"""
Assumes WGS84 coordinate system
Altitude:
look at this for altitude: https://github.com/tkrajina/srtm.py
"""
def __init__(self, df):
self.df = df
self.columns = df.columns
self.min_elevation = None
self.max_elevation = None
self.avg_elevation = None
self.ascent = None
self.descent = None
self.start_time = None
self.end_time = None
self.elapsed_time = None
self.activity_time = None
self.moving_time = None
self.total_distance = None
self.place_info = None
self.place_name = None
def _calc_moving_time(self, method="simple", min_movement=0.05):
"""
simple: requires a minimum distance value and if the distance moved in 1 sec is less then this, it is not
counted as moving. The default os 0.05meters is .1 mph, this is assuming time between points is 1 second, which
might be wrong.
"""
if method == "simple":
self.df["moving_time_between"] = self.df["time_between"]
if 'distance_between' in self.df.columns:
self.df.loc[self.df["distance_between"] < min_movement, ["moving_time_between"]] = pd.Timedelta(0)
else:
self.distance()
self.df.loc[self.df["distance_between"] < min_movement, ["moving_time_between"]] = pd.Timedelta(0)
return self.df["moving_time_between"].sum()
def elevation(self):
"""
:return: {max:, min:, average:, ascent:, descent:}
"""
# TODO implement api to get Altitude data if it is not in the data.
if "Altitude" in self.df.columns:
self.min_elevation = self.df["Altitude"].min()
self.max_elevation = self.df["Altitude"].max()
self.avg_elevation = self.df["Altitude"].mean()
self.df["altitude_change"] = self.df["Altitude"].diff() # differance between rows
self.ascent = self.df[self.df["altitude_change"] > 0][ "altitude_change"].sum()
self.descent = self.df[self.df["altitude_change"] < 0]["altitude_change"].sum()
return {
"min_elevation": self.min_elevation,
"max_elevation": self.max_elevation,
"avg_elevation": self.avg_elevation,
"ascent": self.ascent,
"descent": self.descent,
}
else:
return None
def time(self):
"""
elapsed_duration: The time between the first and last record.
activity_time: Total time between points, probably the same asn elapsed_duration
moving time: there are different methods, now only a very simple method is used
:return:
"""
self.start_time = self.df.iloc[0]["Date_Time"]
self.end_time = self.df.iloc[-1]["Date_Time"]
self.df["time_between"] = self.df["Date_Time"].diff()
self.elapsed_time = self.end_time - self.start_time
self.activity_time = self.df["time_between"].sum()
self.moving_time = self._calc_moving_time(method="simple", min_movement=0.05)
return {
"start_time": self.start_time,
"end_time": self.end_time,
"elapsed_duration": self.elapsed_time,
"activity_time": self.activity_time,
"moving_time": self.moving_time,
}
def distance(self):
"""
:return:
"""
self.df["shift_Longitude"] = self.df.shift(1)["Longitude"]
self.df["shift_Latitude"] = self.df.shift(1)["Latitude"]
# This is the flat distance between points
self.df["distance_between"] = self.df.apply(
lambda x: sqrt(
(haversine((x["Latitude"], x["Longitude"]),
(x["shift_Latitude"], x["shift_Longitude"]), unit="m",)** 2
+ x["altitude_change"] ** 2)),axis=1)
self.df.drop(['shift_Longitude', 'shift_Latitude'], axis=1)
self.total_distance = self.df["distance_between"].sum()
self.df['distance'] = self.df['distance_between'].cumsum()
return {'total_distance': self.total_distance}
def place(self, private_token):
"""
using mapbox, get place name, "where" the ride was.
see
https://docs.mapbox.com/api/search/#reverse-geocoding
r = requests.get('https://api.mapbox.com/geocoding/v5/mapbox.places/-105.2386,39.4667.json', params=params)
TODO Add a retry
https://findwork.dev/blog/advanced-usage-python-requests-timeouts-retries-hooks/
TODO Log errors
"""
params = (('access_token', private_token), ('types', 'place'))
Longitude, Latitude = self.df.iloc[0][['Longitude', 'Latitude']].values
try:
r = requests.get(f"https://api.mapbox.com/geocoding/v5/mapbox.places/{Longitude},{Latitude}.json", params=params)
self.place_info = r.json()
self.place_name = self.place_info['features'][0]['place_name']
return {'place_info': self.place_info, 'place_name': self.place_name}
except Exception as e:
return {'place_info': self.place_info, 'place_name': self.place_name} # The Values should be none
def export_lat_lon_alt(self, file_type='JSON'):
"""
export the latitude and longitude
:return: file
"""
if file_type == 'JSON':
return [dict(longitude=r.Longitude, latitude=r.Latitude, altitude=r.Altitude) for r in self.df.itertuples()]
elif file_type == 'csv':
self.df[['Latitude', 'Longitude']].to_csv()