HR Sensor : do not go to sleep when the HR app is running.

1 files changed, 13 insertions, 2 deletions

diff --git a/src/components/heartrate/HeartRateController.cpp b/src/components/heartrate/HeartRateController.cpp
index f296027..53b34e6 100644
--- a/src/components/heartrate/HeartRateController.cpp
+++ b/src/components/heartrate/HeartRateController.cpp
@@ -1,23 +1,34 @@
 #include "HeartRateController.h"
 #include <heartratetask/HeartRateTask.h>
+#include <systemtask/SystemTask.h>
 
 using namespace Pinetime::Controllers;
 
+HeartRateController::HeartRateController(Pinetime::System::SystemTask &systemTask) : systemTask{systemTask} {
+
+}
+
+
 void HeartRateController::Update(HeartRateController::States newState, uint8_t heartRate) {
   this->state = newState;
   this->heartRate = heartRate;
 }
 
 void HeartRateController::Start() {
-  if(task != nullptr)
+  if(task != nullptr) {
     task->PushMessage(Pinetime::Applications::HeartRateTask::Messages::StartMeasurement);
+    systemTask.PushMessage(System::SystemTask::Messages::HeartRateRunning);
+  }
 }
 
 void HeartRateController::Stop() {
-  if(task != nullptr)
+  if(task != nullptr) {
     task->PushMessage(Pinetime::Applications::HeartRateTask::Messages::StopMeasurement);
+    systemTask.PushMessage(System::SystemTask::Messages::HeartRateStopped);
+  }
 }
 
 void HeartRateController::SetHeartRateTask(Pinetime::Applications::HeartRateTask *task) {
   this->task = task;
 }
+