Skip to content

Commit fdfe24a

Browse files
authored
Merge pull request #200 from MihaMi27/main
Add speed average and speed max to TCXFile's extract_integral_metrics()
2 parents e2657c9 + 953e6ae commit fdfe24a

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

sport_activities_features/tcx_manipulation.py

+16-1
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,9 @@ def extract_integral_metrics(self, filename: str) -> dict:
144144
'cadence_avg': cadence_avg,
145145
'cadence_max': cadence_max,
146146
'watts_avg': watts_avg,
147-
'watts_max': watts_max
147+
'watts_max': watts_max,
148+
'speed_avg': speed_avg,
149+
'speed_max': speed_max
148150
}.
149151
"""
150152
tcx = TCXReader().read(filename)
@@ -235,6 +237,17 @@ def extract_integral_metrics(self, filename: str) -> dict:
235237
except BaseException:
236238
watts_max = None
237239

240+
try:
241+
speed_avg = tcx.avg_speed
242+
except BaseException:
243+
speed_avg = None
244+
245+
try:
246+
speed_max = tcx.max_speed
247+
except BaseException:
248+
speed_max = None
249+
250+
238251

239252
int_metrics = {
240253
'activity_type': activity_type,
@@ -254,6 +267,8 @@ def extract_integral_metrics(self, filename: str) -> dict:
254267
'cadence_max': cadence_max,
255268
'watts_avg': watts_avg,
256269
'watts_max': watts_max,
270+
'speed_avg': speed_avg,
271+
'speed_max': speed_max
257272
}
258273
return int_metrics
259274

0 commit comments

Comments
 (0)