diff options
author | The King <27705324+jlukanc1@users.noreply.github.com> | 2021-01-24 16:01:14 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-01-24 16:01:14 -0500 |
commit | 8c3df5f0211e0cc31de90039a73fe48b9a9aafe0 (patch) | |
tree | 88fc0671a629b27793a418df76f543c7913b644a /src/heartratetask | |
parent | 51c8cadcb78bdbe9013f5aace629c96ed3dfd06f (diff) | |
parent | 80838d1e42e83b50188d6237d16c81cfa27781a6 (diff) |
Merge branch 'develop' into upstream-dev
Diffstat (limited to 'src/heartratetask')
-rw-r--r-- | src/heartratetask/HeartRateTask.cpp | 97 | ||||
-rw-r--r-- | src/heartratetask/HeartRateTask.h | 43 |
2 files changed, 140 insertions, 0 deletions
diff --git a/src/heartratetask/HeartRateTask.cpp b/src/heartratetask/HeartRateTask.cpp new file mode 100644 index 00000000..5a6d2a5e --- /dev/null +++ b/src/heartratetask/HeartRateTask.cpp @@ -0,0 +1,97 @@ +#include "HeartRateTask.h" +#include <drivers/Hrs3300.h> +#include <components/heartrate/HeartRateController.h> +#include <nrf_log.h> + +using namespace Pinetime::Applications; + +HeartRateTask::HeartRateTask(Drivers::Hrs3300 &heartRateSensor, Controllers::HeartRateController& controller) : + heartRateSensor{heartRateSensor}, + controller{controller}, + ppg{static_cast<float>(heartRateSensor.ReadHrs())} { + messageQueue = xQueueCreate(10, 1); + controller.SetHeartRateTask(this); +} + +void HeartRateTask::Start() { + if (pdPASS != xTaskCreate(HeartRateTask::Process, "Heartrate", 500, this, 0, &taskHandle)) + APP_ERROR_HANDLER(NRF_ERROR_NO_MEM); +} + +void HeartRateTask::Process(void *instance) { + auto *app = static_cast<HeartRateTask *>(instance); + app->Work(); +} + +void HeartRateTask::Work() { + int lastBpm = 0; + while (true) { + Messages msg; + uint32_t delay; + if (state == States::Running) { + if (measurementStarted) delay = 40; + else delay = 100; + } else + delay = portMAX_DELAY; + + if (xQueueReceive(messageQueue, &msg, delay)) { + switch (msg) { + case Messages::GoToSleep: + StopMeasurement(); + state = States::Idle; + break; + case Messages::WakeUp: + state = States::Running; + if(measurementStarted) { + lastBpm = 0; + StartMeasurement(); + } + break; + case Messages::StartMeasurement: + if(measurementStarted) break; + lastBpm = 0; + StartMeasurement(); + measurementStarted = true; + break; + case Messages::StopMeasurement: + if(!measurementStarted) break; + StopMeasurement(); + measurementStarted = false; + break; + } + } + + if (measurementStarted) { + auto hrs = heartRateSensor.ReadHrs(); + ppg.Preprocess(hrs); + auto bpm = ppg.HeartRate(); + + if (lastBpm == 0 && bpm == 0) controller.Update(Controllers::HeartRateController::States::NotEnoughData, 0); + if(bpm != 0) { + lastBpm = bpm; + controller.Update(Controllers::HeartRateController::States::Running, lastBpm); + } + } + } +} + +void HeartRateTask::PushMessage(HeartRateTask::Messages msg) { + BaseType_t xHigherPriorityTaskWoken; + xHigherPriorityTaskWoken = pdFALSE; + xQueueSendFromISR(messageQueue, &msg, &xHigherPriorityTaskWoken); + if (xHigherPriorityTaskWoken) { + /* Actual macro used here is port specific. */ + // TODO : should I do something here? + } +} + +void HeartRateTask::StartMeasurement() { + heartRateSensor.Enable(); + vTaskDelay(100); + ppg.SetOffset(static_cast<float>(heartRateSensor.ReadHrs())); +} + +void HeartRateTask::StopMeasurement() { + heartRateSensor.Disable(); + vTaskDelay(100); +} diff --git a/src/heartratetask/HeartRateTask.h b/src/heartratetask/HeartRateTask.h new file mode 100644 index 00000000..f142af1a --- /dev/null +++ b/src/heartratetask/HeartRateTask.h @@ -0,0 +1,43 @@ +#pragma once +#include <FreeRTOS.h> +#include <task.h> +#include <queue.h> +#include <components/heartrate/Ppg.h> + + +namespace Pinetime { + namespace Drivers { + class Hrs3300; + } + namespace Controllers{ + class HeartRateController; + } + namespace Applications { + class HeartRateTask { + public: + enum class Messages : uint8_t {GoToSleep, WakeUp, StartMeasurement, StopMeasurement }; + enum class States {Idle, Running}; + + explicit HeartRateTask(Drivers::Hrs3300& heartRateSensor, Controllers::HeartRateController& controller); + void Start(); + void Work(); + void PushMessage(Messages msg); + + private: + static void Process(void* instance); + void StartMeasurement(); + void StopMeasurement(); + + TaskHandle_t taskHandle; + QueueHandle_t messageQueue; + States state = States::Running; + Drivers::Hrs3300 &heartRateSensor; + Controllers::HeartRateController& controller; + Controllers::Ppg ppg; + bool measurementStarted = false; + + }; + + } +} + |