Disable heart rate tracking while power is connected (charging)

Author: Andersama

src/heartratetask/HeartRateTask.cpp

@@ -1,6 +1,7 @@
 #include "heartratetask/HeartRateTask.h"
 #include <drivers/Hrs3300.h>
 #include <components/heartrate/HeartRateController.h>
+#include <components/battery/BatteryController.h>
 #include <limits>
 
 using namespace Pinetime::Applications;
@@ -90,8 +91,9 @@ TickType_t HeartRateTask::CurrentTaskDelay() {
 
 HeartRateTask::HeartRateTask(Drivers::Hrs3300& heartRateSensor,
                              Controllers::HeartRateController& controller,
+                             Controllers::Battery& battery,
                              Controllers::Settings& settings)
-  : heartRateSensor {heartRateSensor}, controller {controller}, settings {settings} {
+  : heartRateSensor {heartRateSensor}, controller {controller}, battery {battery}, settings {settings} {
 }
 
 void HeartRateTask::Start() {
@@ -153,10 +155,15 @@ void HeartRateTask::Work() {
           break;
       }
     }
-    if (newState == States::Waiting && BackgroundMeasurementNeeded()) {
-      newState = States::BackgroundMeasuring;
-    } else if (newState == States::BackgroundMeasuring && !BackgroundMeasurementNeeded()) {
+
+    if (newState != States::Disabled && battery.IsPowerPresent()) {
       newState = States::Waiting;
+    } else {
+      if (newState == States::Waiting && BackgroundMeasurementNeeded()) {
+        newState = States::BackgroundMeasuring;
+      } else if (newState == States::BackgroundMeasuring && !BackgroundMeasurementNeeded()) {
+        newState = States::Waiting;
+      }
     }
 
     // Apply state transition (switch sensor on/off)

src/heartratetask/HeartRateTask.h

@@ -14,6 +14,7 @@ namespace Pinetime {
 
   namespace Controllers {
     class HeartRateController;
+    class Battery;
   }
 
   namespace Applications {
@@ -23,6 +24,7 @@ namespace Pinetime {
 
       explicit HeartRateTask(Drivers::Hrs3300& heartRateSensor,
                              Controllers::HeartRateController& controller,
+                             Controllers::Battery& battery,
                              Controllers::Settings& settings);
       void Start();
       void Work();
@@ -47,6 +49,7 @@ namespace Pinetime {
       uint16_t count;
       Drivers::Hrs3300& heartRateSensor;
       Controllers::HeartRateController& controller;
+      Controllers::Battery& battery;
       Controllers::Settings& settings;
       Controllers::Ppg ppg;
       TickType_t lastMeasurementTime;

src/main.cpp

@@ -99,7 +99,7 @@ Pinetime::Controllers::Settings settingsController {fs};
 Pinetime::Controllers::MotorController motorController {};
 
 Pinetime::Controllers::HeartRateController heartRateController;
-Pinetime::Applications::HeartRateTask heartRateApp(heartRateSensor, heartRateController, settingsController);
+Pinetime::Applications::HeartRateTask heartRateApp(heartRateSensor, heartRateController, batteryController, settingsController);
 
 Pinetime::Controllers::DateTime dateTimeController {settingsController};
 Pinetime::Drivers::Watchdog watchdog;