Merge pull request #200 from MihaMi27/main

Add speed average and speed max to TCXFile's extract_integral_metrics()
firefly-cpp · Dec 3, 2024 · fdfe24a · fdfe24a
2 parents e2657c9 + 953e6ae
commit fdfe24a
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/sport_activities_features/tcx_manipulation.py b/sport_activities_features/tcx_manipulation.py
@@ -144,7 +144,9 @@ def extract_integral_metrics(self, filename: str) -> dict:
                 'cadence_avg': cadence_avg,
                 'cadence_max': cadence_max,
                 'watts_avg': watts_avg,
-                'watts_max': watts_max
+                'watts_max': watts_max,
+                'speed_avg': speed_avg,
+                'speed_max': speed_max
             }.
         """
         tcx = TCXReader().read(filename)
@@ -235,6 +237,17 @@ def extract_integral_metrics(self, filename: str) -> dict:
         except BaseException:
             watts_max = None     
 
+        try:
+            speed_avg = tcx.avg_speed
+        except BaseException:
+            speed_avg = None     
+
+        try:
+            speed_max = tcx.max_speed
+        except BaseException:
+            speed_max = None     
+
+
 
         int_metrics = {
             'activity_type': activity_type,
@@ -254,6 +267,8 @@ def extract_integral_metrics(self, filename: str) -> dict:
             'cadence_max': cadence_max,
             'watts_avg': watts_avg,
             'watts_max': watts_max,
+            'speed_avg': speed_avg,
+            'speed_max': speed_max
         }
         return int_metrics