Jeannie Hur / Working_HRV

HRV -> Mood

Dependencies:   MAX30101 Hexi_KW40Z Hexi_OLED_SSD1351

8cee5929f4d8/main.cpp@12:95d141421e18, 2019-03-16 (annotated)

Committer:
jeannie9809
Date:
Sat Mar 16 08:08:25 2019 +0000
Revision:
12:95d141421e18
Parent:
11:58a443cf3f7c
Child:
13:45c2837ab056
done with valence, arousal

Who changed what in which revision?

UserRevisionLine numberNew contents of line
jeannie9809 0:338c50c9c8cd 1 #include "mbed.h"
jeannie9809 0:338c50c9c8cd 2 #include "mbed_events.h"
jeannie9809 0:338c50c9c8cd 3 #include "MAX30101.h"
jeannie9809 0:338c50c9c8cd 4 #include "string.h"
jeannie9809 0:338c50c9c8cd 5 #include "Hexi_OLED_SSD1351.h"
jeannie9809 0:338c50c9c8cd 6 #include "Hexi_KW40Z.h"
jeannie9809 0:338c50c9c8cd 7 #include <math.h>
jeannie9809 6:ff1c3560db84 8 #include <vector>
jeannie9809 6:ff1c3560db84 9 using namespace std;
jeannie9809 11:58a443cf3f7c 10
jeannie9809 0:338c50c9c8cd 11 #define FIFO_DATA_MAX 288
jeannie9809 11:58a443cf3f7c 12 //#define M_PI 3.14159265358979323846
jeannie9809 11:58a443cf3f7c 13 #define M_PI 3.14159
jeannie9809 11:58a443cf3f7c 14
jeannie9809 0:338c50c9c8cd 15 void UpdateSensorData(void);
jeannie9809 0:338c50c9c8cd 16 void StartHaptic(void);
jeannie9809 0:338c50c9c8cd 17 void StopHaptic(void const *n);
jeannie9809 0:338c50c9c8cd 18 void txTask(void);
jeannie9809 0:338c50c9c8cd 19
jeannie9809 0:338c50c9c8cd 20 DigitalOut pwr1v8(PTA29);
jeannie9809 0:338c50c9c8cd 21 DigitalOut pwr3v3b(PTC13);
jeannie9809 0:338c50c9c8cd 22 DigitalOut pwr15v(PTB12);
jeannie9809 0:338c50c9c8cd 23 I2C i2c0(PTB1, PTB0);
jeannie9809 0:338c50c9c8cd 24 InterruptIn maximInterrupt(PTB18);
jeannie9809 0:338c50c9c8cd 25 Serial pc(USBTX, USBRX);
jeannie9809 0:338c50c9c8cd 26 SSD1351 oled(PTB22,PTB21,PTC13,PTB20,PTE6, PTD15);
jeannie9809 0:338c50c9c8cd 27 KW40Z kw40z_device(PTE24, PTE25);
jeannie9809 0:338c50c9c8cd 28 DigitalOut haptic(PTB9);
jeannie9809 0:338c50c9c8cd 29 EventQueue evqueue(32 * EVENTS_EVENT_SIZE);
jeannie9809 0:338c50c9c8cd 30 Thread t;
jeannie9809 0:338c50c9c8cd 31 /* Define timer for haptic feedback */
jeannie9809 0:338c50c9c8cd 32 RtosTimer hapticTimer(StopHaptic, osTimerOnce);
jeannie9809 0:338c50c9c8cd 33
jeannie9809 3:0da9235c9069 34 /*Create a Thread to handle sending BLE Sensor Data */
jeannie9809 0:338c50c9c8cd 35 Thread txThread;
jeannie9809 0:338c50c9c8cd 36 MAX30101 hr(i2c0);
jeannie9809 1:eabf219849ab 37 int ppg_single_sample;
jeannie9809 0:338c50c9c8cd 38 int mask_ppg = 0;
jeannie9809 0:338c50c9c8cd 39 uint32_t count = 0;
jeannie9809 0:338c50c9c8cd 40 uint32_t num;
jeannie9809 0:338c50c9c8cd 41 uint8_t testsignal = 60;
jeannie9809 0:338c50c9c8cd 42
jeannie9809 2:3389bdfd9afa 43 // I added this
jeannie9809 11:58a443cf3f7c 44 const int num_samples = 500; // for 30 sec
jeannie9809 0:338c50c9c8cd 45 int ppg[num_samples];
jeannie9809 12:95d141421e18 46 int SDNN_n, SDNN, valence_arousal;
jeannie9809 12:95d141421e18 47 double HF_LF = 0.0;
jeannie9809 12:95d141421e18 48 double HF_LF_n = 0.0;
jeannie9809 9:83dcb7382516 49 bool ready = false;
jeannie9809 0:338c50c9c8cd 50
jeannie9809 3:0da9235c9069 51 void StartHaptic(void)
jeannie9809 3:0da9235c9069 52 {
jeannie9809 0:338c50c9c8cd 53 hapticTimer.start(50);
jeannie9809 0:338c50c9c8cd 54 haptic = 1;
jeannie9809 0:338c50c9c8cd 55 }
jeannie9809 0:338c50c9c8cd 56 void ButtonRight(void)
jeannie9809 0:338c50c9c8cd 57 {
jeannie9809 0:338c50c9c8cd 58 StartHaptic();
jeannie9809 0:338c50c9c8cd 59 kw40z_device.ToggleAdvertisementMode();
jeannie9809 0:338c50c9c8cd 60 }
jeannie9809 0:338c50c9c8cd 61
jeannie9809 0:338c50c9c8cd 62 void ButtonLeft(void)
jeannie9809 0:338c50c9c8cd 63 {
jeannie9809 0:338c50c9c8cd 64 StartHaptic();
jeannie9809 0:338c50c9c8cd 65 kw40z_device.ToggleAdvertisementMode();
jeannie9809 0:338c50c9c8cd 66 }
jeannie9809 0:338c50c9c8cd 67
jeannie9809 3:0da9235c9069 68 void StopHaptic(void const *n)
jeannie9809 3:0da9235c9069 69 {
jeannie9809 0:338c50c9c8cd 70 haptic = 0;
jeannie9809 0:338c50c9c8cd 71 hapticTimer.stop();
jeannie9809 0:338c50c9c8cd 72 }
jeannie9809 3:0da9235c9069 73 void txTask(void)
jeannie9809 3:0da9235c9069 74 {
jeannie9809 3:0da9235c9069 75
jeannie9809 3:0da9235c9069 76 while (true) {
jeannie9809 0:338c50c9c8cd 77 UpdateSensorData();
jeannie9809 3:0da9235c9069 78
jeannie9809 0:338c50c9c8cd 79 /*Notify Hexiwear App that it is running Sensor Tag mode*/
jeannie9809 0:338c50c9c8cd 80 kw40z_device.SendSetApplicationMode(GUI_CURRENT_APP_SENSOR_TAG);
jeannie9809 0:338c50c9c8cd 81 //send heartrate
jeannie9809 3:0da9235c9069 82 kw40z_device.SendHeartRate(testsignal);
jeannie9809 0:338c50c9c8cd 83 /*The following is sending dummy data over BLE. Replace with real data*/
jeannie9809 3:0da9235c9069 84
jeannie9809 3:0da9235c9069 85 /*Send Battery Level for 20%
jeannie9809 0:338c50c9c8cd 86 kw40z_device.SendBatteryLevel(battery);
jeannie9809 3:0da9235c9069 87
jeannie9809 3:0da9235c9069 88 Send Ambient Light Level at 50%
jeannie9809 0:338c50c9c8cd 89 kw40z_device.SendAmbientLight(light);*/
jeannie9809 3:0da9235c9069 90
jeannie9809 0:338c50c9c8cd 91 /*Send Humidity at 90% */
jeannie9809 0:338c50c9c8cd 92 //kw40z_device.SendHumidity(humidity);
jeannie9809 3:0da9235c9069 93
jeannie9809 3:0da9235c9069 94 /*Send Temperature at 25 degrees Celsius
jeannie9809 0:338c50c9c8cd 95 kw40z_device.SendTemperature(temperature);
jeannie9809 0:338c50c9c8cd 96
jeannie9809 11:58a443cf3f7c 97 //Send Pressure at 100kPA
jeannie9809 11:58a443cf3f7c 98 //kw40z_device.SendPressure(pressure); */
jeannie9809 3:0da9235c9069 99
jeannie9809 3:0da9235c9069 100 /*Send Mag,Accel,Gyro Data.
jeannie9809 0:338c50c9c8cd 101 kw40z_device.SendGyro(x,y,z);
jeannie9809 0:338c50c9c8cd 102 kw40z_device.SendAccel(z,x,y);
jeannie9809 0:338c50c9c8cd 103 kw40z_device.SendMag(y,z,x);*/
jeannie9809 0:338c50c9c8cd 104
jeannie9809 3:0da9235c9069 105 Thread::wait(1000);
jeannie9809 0:338c50c9c8cd 106 }
jeannie9809 0:338c50c9c8cd 107 }
jeannie9809 0:338c50c9c8cd 108 void UpdateSensorData(void)
jeannie9809 3:0da9235c9069 109 {
jeannie9809 3:0da9235c9069 110 testsignal+=1;
jeannie9809 0:338c50c9c8cd 111 /*battery -= 5;
jeannie9809 0:338c50c9c8cd 112 if(battery < 5) battery = 100;
jeannie9809 3:0da9235c9069 113
jeannie9809 0:338c50c9c8cd 114 light += 20;
jeannie9809 0:338c50c9c8cd 115 if(light > 100) light = 0;
jeannie9809 3:0da9235c9069 116
jeannie9809 0:338c50c9c8cd 117 humidity += 500;
jeannie9809 0:338c50c9c8cd 118 if(humidity > 8000) humidity = 2000;
jeannie9809 3:0da9235c9069 119
jeannie9809 0:338c50c9c8cd 120 temperature -= 200;
jeannie9809 0:338c50c9c8cd 121 if(temperature < 200) temperature = 4200;
jeannie9809 3:0da9235c9069 122
jeannie9809 0:338c50c9c8cd 123 pressure += 300;
jeannie9809 0:338c50c9c8cd 124 if(pressure > 10300) pressure = 7500;
jeannie9809 3:0da9235c9069 125
jeannie9809 0:338c50c9c8cd 126 x += 1400;
jeannie9809 0:338c50c9c8cd 127 y -= 2300;
jeannie9809 0:338c50c9c8cd 128 z += 1700;*/
jeannie9809 0:338c50c9c8cd 129 }
jeannie9809 0:338c50c9c8cd 130
jeannie9809 3:0da9235c9069 131 void interruptHandlerQueued()
jeannie9809 3:0da9235c9069 132 {
jeannie9809 4:94190624967a 133 // int temp_ppg[num_samples];
jeannie9809 3:0da9235c9069 134 for(int iter = 0; iter < num_samples; iter+=0) {
jeannie9809 0:338c50c9c8cd 135
jeannie9809 3:0da9235c9069 136 MAX30101::InterruptBitField_u interruptStatus;
jeannie9809 3:0da9235c9069 137 hr.getInterruptStatus(interruptStatus);
jeannie9809 3:0da9235c9069 138 // printf("Interrupt Status: 0x%02x\r\n", interruptStatus.all);
jeannie9809 3:0da9235c9069 139
jeannie9809 3:0da9235c9069 140 if (interruptStatus.bits.pwr_rdy == 0x1) {
jeannie9809 3:0da9235c9069 141 // printf("Powered on\r\n");
jeannie9809 3:0da9235c9069 142
jeannie9809 3:0da9235c9069 143 // Soft reset
jeannie9809 3:0da9235c9069 144 MAX30101::ModeConfiguration_u modeConf;
jeannie9809 3:0da9235c9069 145 modeConf.all = 0;
jeannie9809 3:0da9235c9069 146 modeConf.bits.reset = 1;
jeannie9809 3:0da9235c9069 147 hr.setModeConfiguration(modeConf);
jeannie9809 3:0da9235c9069 148 wait(0.01);
jeannie9809 3:0da9235c9069 149
jeannie9809 3:0da9235c9069 150 // Configure FIFO
jeannie9809 3:0da9235c9069 151 MAX30101::FIFO_Configuration_u fifoConf;
jeannie9809 3:0da9235c9069 152 hr.getFIFOConfiguration(fifoConf);
jeannie9809 0:338c50c9c8cd 153 // pc.printf("FIFO Configuration: 0x%02x\r\n", fifoConf.all);
jeannie9809 3:0da9235c9069 154
jeannie9809 3:0da9235c9069 155 // Set LED power
jeannie9809 3:0da9235c9069 156 hr.setLEDPulseAmplitude(MAX30101::LED1_PA, 0x0C);
jeannie9809 3:0da9235c9069 157 hr.setLEDPulseAmplitude(MAX30101::ProxModeLED_PA, 0x19);
jeannie9809 0:338c50c9c8cd 158 // pc.printf("LED set\r\n");
jeannie9809 3:0da9235c9069 159
jeannie9809 3:0da9235c9069 160 MAX30101::SpO2Configuration_u spo2Conf;
jeannie9809 3:0da9235c9069 161 hr.getSpO2Configuration(spo2Conf);
jeannie9809 3:0da9235c9069 162 spo2Conf.bits.led_pw = MAX30101::PW_1;
jeannie9809 3:0da9235c9069 163 spo2Conf.bits.spo2_sr = MAX30101::SR_100_Hz;
jeannie9809 3:0da9235c9069 164 hr.setSpO2Configuration(spo2Conf);
jeannie9809 3:0da9235c9069 165 hr.getSpO2Configuration(spo2Conf);
jeannie9809 0:338c50c9c8cd 166 // pc.printf("SpO2 Configuration: 0x%02x\r\n", spo2Conf.all);
jeannie9809 3:0da9235c9069 167
jeannie9809 3:0da9235c9069 168 // Proximity settings
jeannie9809 3:0da9235c9069 169 hr.setProxIntThreshold(0x14);
jeannie9809 3:0da9235c9069 170
jeannie9809 3:0da9235c9069 171 // Enable HR mode
jeannie9809 3:0da9235c9069 172 modeConf.all = 0;
jeannie9809 3:0da9235c9069 173 modeConf.bits.mode = MAX30101::HeartRateMode;
jeannie9809 3:0da9235c9069 174 hr.setModeConfiguration(modeConf);
jeannie9809 0:338c50c9c8cd 175 // printf("Mode set\r\n");
jeannie9809 3:0da9235c9069 176 }
jeannie9809 3:0da9235c9069 177
jeannie9809 3:0da9235c9069 178 if (interruptStatus.bits.prox_int == 0x1) {
jeannie9809 0:338c50c9c8cd 179 // printf("Proximity Triggered, entered HR Mode.");
jeannie9809 3:0da9235c9069 180 }
jeannie9809 3:0da9235c9069 181
jeannie9809 3:0da9235c9069 182 if (interruptStatus.bits.ppg_rdy == 0x1) {
jeannie9809 0:338c50c9c8cd 183 // printf("PPG Ready.\r\n");
jeannie9809 3:0da9235c9069 184 mask_ppg = 1;
jeannie9809 3:0da9235c9069 185 }
jeannie9809 3:0da9235c9069 186
jeannie9809 3:0da9235c9069 187 if (interruptStatus.bits.a_full == 0x1) {
jeannie9809 0:338c50c9c8cd 188 // printf("FIFO Almost Full.\r\n");
jeannie9809 3:0da9235c9069 189 uint8_t data[FIFO_DATA_MAX];
jeannie9809 3:0da9235c9069 190 uint16_t readBytes = 0;
jeannie9809 3:0da9235c9069 191
jeannie9809 3:0da9235c9069 192 hr.readFIFO(MAX30101::OneLedChannel, data, readBytes);
jeannie9809 9:83dcb7382516 193 // printf("data length: %u \r\n",readBytes);
jeannie9809 3:0da9235c9069 194 //printf("data length: %u \r\n",data);
jeannie9809 3:0da9235c9069 195 for (uint16_t i = 0; i < readBytes; i += 3) {
jeannie9809 3:0da9235c9069 196 uint8_t sample[4] = {0};
jeannie9809 3:0da9235c9069 197 sample[0] = data[i + 2];
jeannie9809 3:0da9235c9069 198 sample[1] = data[i + 1];
jeannie9809 3:0da9235c9069 199 sample[2] = data[i];
jeannie9809 3:0da9235c9069 200
jeannie9809 3:0da9235c9069 201 num = *(uint32_t *) sample;
jeannie9809 9:83dcb7382516 202
jeannie9809 11:58a443cf3f7c 203 if (num < 0) {
jeannie9809 3:0da9235c9069 204 ppg_single_sample = 0;
jeannie9809 9:83dcb7382516 205 // printf("keep closer to your hand \r\n");
jeannie9809 3:0da9235c9069 206 } else {
jeannie9809 3:0da9235c9069 207
jeannie9809 3:0da9235c9069 208 //ppg_single_sample = 65;
jeannie9809 6:ff1c3560db84 209 ppg_single_sample = num;
jeannie9809 9:83dcb7382516 210 if(iter < num_samples && !ready) {
jeannie9809 9:83dcb7382516 211 ppg[iter] = num;
jeannie9809 9:83dcb7382516 212 // printf("%d \n", ppg[iter]);
jeannie9809 9:83dcb7382516 213 if(iter == num_samples-1)
jeannie9809 9:83dcb7382516 214 ready = true;
jeannie9809 9:83dcb7382516 215 }
jeannie9809 9:83dcb7382516 216
jeannie9809 9:83dcb7382516 217 // printf("%d ", ppg_single_sample); // I commented this out
jeannie9809 3:0da9235c9069 218 iter++;
jeannie9809 3:0da9235c9069 219 }
jeannie9809 9:83dcb7382516 220 // printf("%d; %d\r\n", iter, ppg[iter-1]);
jeannie9809 3:0da9235c9069 221
jeannie9809 3:0da9235c9069 222
jeannie9809 0:338c50c9c8cd 223 }
jeannie9809 0:338c50c9c8cd 224 }
jeannie9809 3:0da9235c9069 225
jeannie9809 3:0da9235c9069 226 interruptStatus.all = 0xFF;
jeannie9809 11:58a443cf3f7c 227
jeannie9809 3:0da9235c9069 228 if (mask_ppg == 1) {
jeannie9809 3:0da9235c9069 229 interruptStatus.bits.ppg_rdy = 0;
jeannie9809 3:0da9235c9069 230 }
jeannie9809 3:0da9235c9069 231 hr.enableInterrupts(interruptStatus);
jeannie9809 0:338c50c9c8cd 232 }
jeannie9809 4:94190624967a 233 /*
jeannie9809 4:94190624967a 234 for(int i = 0; i < num_samples; i++) {
jeannie9809 4:94190624967a 235 ppg[i] = temp_ppg[i];
jeannie9809 4:94190624967a 236 }
jeannie9809 4:94190624967a 237 */
jeannie9809 4:94190624967a 238 //ppg = temp_ppg;
jeannie9809 11:58a443cf3f7c 239
jeannie9809 0:338c50c9c8cd 240 }
jeannie9809 0:338c50c9c8cd 241
jeannie9809 3:0da9235c9069 242 void interruptHandler()
jeannie9809 3:0da9235c9069 243 {
jeannie9809 0:338c50c9c8cd 244 evqueue.call(interruptHandlerQueued);
jeannie9809 9:83dcb7382516 245 // interruptHandlerQueued();
jeannie9809 11:58a443cf3f7c 246 // printf("\nLezzgo\n");
jeannie9809 9:83dcb7382516 247 if(ready) {
jeannie9809 11:58a443cf3f7c 248 printf("\nStarting... ");
jeannie9809 11:58a443cf3f7c 249
jeannie9809 11:58a443cf3f7c 250 int i = 0;
jeannie9809 11:58a443cf3f7c 251 int j = 0;
jeannie9809 11:58a443cf3f7c 252
jeannie9809 11:58a443cf3f7c 253 double TIME[num_samples];
jeannie9809 11:58a443cf3f7c 254 for(int i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 255 TIME[i] = (double)i*(5.0/(double)num_samples); // change to 30.0 later
jeannie9809 11:58a443cf3f7c 256 }
jeannie9809 11:58a443cf3f7c 257
jeannie9809 11:58a443cf3f7c 258 // moving average
jeannie9809 11:58a443cf3f7c 259 double movave[num_samples];
jeannie9809 11:58a443cf3f7c 260 int avecap = 25;
jeannie9809 11:58a443cf3f7c 261 for(i = 0; i < (avecap-1)/2; i++) {
jeannie9809 11:58a443cf3f7c 262 movave[i] = (double)ppg[0];
jeannie9809 11:58a443cf3f7c 263 for(j = 1; j < 1+(avecap-1)/2; j++) {
jeannie9809 11:58a443cf3f7c 264 movave[i] = movave[i] + (double)ppg[j];
jeannie9809 11:58a443cf3f7c 265 }
jeannie9809 11:58a443cf3f7c 266 movave[i] = (double)(movave[i]/(i+(avecap-1)/2-1));
jeannie9809 11:58a443cf3f7c 267 }
jeannie9809 11:58a443cf3f7c 268 for(i = (num_samples-(avecap-1)/2); i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 269 movave[i] = (double)ppg[i-(avecap-1)/2-1];
jeannie9809 11:58a443cf3f7c 270 for(j = (i-(avecap-1)/2); j < num_samples; j++) {
jeannie9809 11:58a443cf3f7c 271 movave[i] = movave[i] + (double)ppg[j];
jeannie9809 11:58a443cf3f7c 272 }
jeannie9809 11:58a443cf3f7c 273 movave[i] = movave[i]/(double)(num_samples-i+(avecap-1)/2);
jeannie9809 11:58a443cf3f7c 274 }
jeannie9809 11:58a443cf3f7c 275 for(i = (avecap-1)/2; i < (num_samples-(avecap-1)/2); i++) {
jeannie9809 11:58a443cf3f7c 276 movave[i] = (double)ppg[i-(avecap-1)/2-1];
jeannie9809 11:58a443cf3f7c 277 for(j = i-(avecap-1)/2; j < i+(avecap-1)/2; j++) {
jeannie9809 11:58a443cf3f7c 278 movave[i] = movave[i] + (double)ppg[j];
jeannie9809 11:58a443cf3f7c 279 }
jeannie9809 11:58a443cf3f7c 280 movave[i] = movave[i]/(double)avecap;
jeannie9809 4:94190624967a 281 }
jeannie9809 11:58a443cf3f7c 282
jeannie9809 11:58a443cf3f7c 283 // normalize ppg
jeannie9809 11:58a443cf3f7c 284 for(i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 285 ppg[i] = ppg[i] - (int)movave[i];
jeannie9809 11:58a443cf3f7c 286 if(ppg[i] > 1000 || ppg[i] < -1000)
jeannie9809 11:58a443cf3f7c 287 ppg[i] = 0;
jeannie9809 11:58a443cf3f7c 288 }
jeannie9809 11:58a443cf3f7c 289
jeannie9809 11:58a443cf3f7c 290 // smoothing curve
jeannie9809 11:58a443cf3f7c 291 for(i = 1; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 292 ppg[i] = ppg[i] + ppg[i-1];
jeannie9809 11:58a443cf3f7c 293 //printf("%d ", ppg[i]);
jeannie9809 4:94190624967a 294 }
jeannie9809 11:58a443cf3f7c 295
jeannie9809 11:58a443cf3f7c 296 // AMPD Algorithm
jeannie9809 11:58a443cf3f7c 297 const int kcap = 25;
jeannie9809 11:58a443cf3f7c 298 int m[kcap][num_samples];
jeannie9809 11:58a443cf3f7c 299 for(int k = 1; k <= kcap; k++) {
jeannie9809 11:58a443cf3f7c 300 for(i = 1; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 301 if(i-1 < 0 || i-k-1 < 0 || i+k-1 >= num_samples)
jeannie9809 11:58a443cf3f7c 302 m[k-1][i] = 0;
jeannie9809 11:58a443cf3f7c 303 else if(ppg[i-1] > ppg[i-k-1] && ppg[i-1] > ppg[i+k-1])
jeannie9809 11:58a443cf3f7c 304 m[k-1][i] = 1;
jeannie9809 11:58a443cf3f7c 305 else
jeannie9809 11:58a443cf3f7c 306 m[k-1][i] = 0;
jeannie9809 11:58a443cf3f7c 307 }
jeannie9809 11:58a443cf3f7c 308 }
jeannie9809 11:58a443cf3f7c 309
jeannie9809 11:58a443cf3f7c 310 int max_mult[num_samples];
jeannie9809 11:58a443cf3f7c 311 for(i = 0; i < num_samples; i++) { // max_mult = m(1,:)';
jeannie9809 11:58a443cf3f7c 312 max_mult[i] = m[0][i];
jeannie9809 11:58a443cf3f7c 313 }
jeannie9809 11:58a443cf3f7c 314
jeannie9809 11:58a443cf3f7c 315 for(int k = 1; k < kcap; k++) {
jeannie9809 11:58a443cf3f7c 316 for(i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 317 max_mult[i] = max_mult[i]*m[k][i];
jeannie9809 11:58a443cf3f7c 318 }
jeannie9809 11:58a443cf3f7c 319 }
jeannie9809 11:58a443cf3f7c 320
jeannie9809 11:58a443cf3f7c 321 int num_max = 0;
jeannie9809 11:58a443cf3f7c 322 // extract times that are max
jeannie9809 11:58a443cf3f7c 323 for(i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 324 num_max = num_max + max_mult[i];
jeannie9809 4:94190624967a 325 }
jeannie9809 11:58a443cf3f7c 326
jeannie9809 11:58a443cf3f7c 327 // printf("num_max = %d ", num_max);
jeannie9809 11:58a443cf3f7c 328
jeannie9809 11:58a443cf3f7c 329 vector<double> time_of_max;
jeannie9809 11:58a443cf3f7c 330 vector<int> index_of_max;
jeannie9809 11:58a443cf3f7c 331 vector<int> max_points;
jeannie9809 11:58a443cf3f7c 332 for(i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 333 if(max_mult[i] == 1) {
jeannie9809 11:58a443cf3f7c 334 time_of_max.push_back(TIME[i-1]);
jeannie9809 11:58a443cf3f7c 335 index_of_max.push_back(i-1);
jeannie9809 11:58a443cf3f7c 336 max_points.push_back(ppg[i-1]);
jeannie9809 11:58a443cf3f7c 337 }
jeannie9809 11:58a443cf3f7c 338 }
jeannie9809 11:58a443cf3f7c 339
jeannie9809 11:58a443cf3f7c 340 // calculating HRV
jeannie9809 11:58a443cf3f7c 341 vector<double> r;
jeannie9809 11:58a443cf3f7c 342 vector<int> index_r;
jeannie9809 11:58a443cf3f7c 343 double mean_inter_time = 0;
jeannie9809 11:58a443cf3f7c 344 for(i = 0; i < num_max-1; i++) {
jeannie9809 11:58a443cf3f7c 345 r.push_back(time_of_max.at(i+1)-time_of_max.at(i));
jeannie9809 11:58a443cf3f7c 346 index_r.push_back(index_of_max.at(i+1) - index_of_max.at(i));
jeannie9809 11:58a443cf3f7c 347 mean_inter_time = mean_inter_time + r.at(i);
jeannie9809 11:58a443cf3f7c 348 }
jeannie9809 11:58a443cf3f7c 349
jeannie9809 11:58a443cf3f7c 350 //printf("r: %.2f %.2f %.2f ", r.at(0), r.at(1), r.at(2));
jeannie9809 11:58a443cf3f7c 351 mean_inter_time = mean_inter_time/(double)(num_max-1);
jeannie9809 11:58a443cf3f7c 352
jeannie9809 11:58a443cf3f7c 353 // getting rid of outlier points in r
jeannie9809 11:58a443cf3f7c 354 for(i = 0; i < num_max-1; i++) {
jeannie9809 11:58a443cf3f7c 355 if(r.at(i) > mean_inter_time + 0.11)
jeannie9809 11:58a443cf3f7c 356 r.at(i) = mean_inter_time + 0.11;
jeannie9809 11:58a443cf3f7c 357 else if(r.at(i) < mean_inter_time - 0.11)
jeannie9809 11:58a443cf3f7c 358 r.at(i) = mean_inter_time - 0.11;
jeannie9809 11:58a443cf3f7c 359 }
jeannie9809 11:58a443cf3f7c 360
jeannie9809 11:58a443cf3f7c 361 // SDNN -- std of normal to normal R-R intervals
jeannie9809 11:58a443cf3f7c 362 mean_inter_time = 0;
jeannie9809 11:58a443cf3f7c 363 for(i = 0; i < num_max-1; i++) {
jeannie9809 11:58a443cf3f7c 364 mean_inter_time = mean_inter_time + r.at(i);
jeannie9809 5:a6b6d0e5a69f 365 }
jeannie9809 11:58a443cf3f7c 366 mean_inter_time = double(mean_inter_time/(num_max-1));
jeannie9809 11:58a443cf3f7c 367 double SDNN_doub = 0.0;
jeannie9809 11:58a443cf3f7c 368 for(i = 0; i < num_max-1; i++) {
jeannie9809 11:58a443cf3f7c 369 SDNN_doub = SDNN_doub + (r.at(i)-mean_inter_time)*(r.at(i)-mean_inter_time);
jeannie9809 11:58a443cf3f7c 370 }
jeannie9809 11:58a443cf3f7c 371 SDNN = (int)(sqrt(SDNN_doub/(num_max-1))*1000);
jeannie9809 11:58a443cf3f7c 372 printf("SDNN = %d ", SDNN);
jeannie9809 11:58a443cf3f7c 373
jeannie9809 11:58a443cf3f7c 374 // TIME TO CALCULATE HF/LF
jeannie9809 11:58a443cf3f7c 375 // FFT: use movave as fftppg
jeannie9809 5:a6b6d0e5a69f 376 for(i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 377 double real = 0;
jeannie9809 11:58a443cf3f7c 378 double im = 0;
jeannie9809 11:58a443cf3f7c 379 double dum_ppg;
jeannie9809 11:58a443cf3f7c 380 for(j = 0; j < num_samples; j++) {
jeannie9809 11:58a443cf3f7c 381 dum_ppg = (double) ppg[j];
jeannie9809 11:58a443cf3f7c 382 real = real + dum_ppg * cos(2.0*M_PI*(double)(j*i)/(double)num_samples);
jeannie9809 11:58a443cf3f7c 383 im = im + dum_ppg * sin(2.0*M_PI*(double)(j*i)/(double)num_samples);
jeannie9809 11:58a443cf3f7c 384 }
jeannie9809 11:58a443cf3f7c 385 movave[i] = sqrt(real*real + im*im)/(double)num_samples;
jeannie9809 11:58a443cf3f7c 386 }
jeannie9809 11:58a443cf3f7c 387
jeannie9809 11:58a443cf3f7c 388
jeannie9809 11:58a443cf3f7c 389 // make frequency array: use TIME
jeannie9809 11:58a443cf3f7c 390 double sampling_freq = (double)(1/TIME[1]);
jeannie9809 11:58a443cf3f7c 391 for(i = 0; i < num_samples; i++) {
jeannie9809 11:58a443cf3f7c 392 TIME[i] = sampling_freq*(double)(i)/(double)(num_samples);
jeannie9809 5:a6b6d0e5a69f 393 }
jeannie9809 11:58a443cf3f7c 394
jeannie9809 11:58a443cf3f7c 395 double LF = 0.0;
jeannie9809 11:58a443cf3f7c 396 i = 0;
jeannie9809 11:58a443cf3f7c 397 while(TIME[i] < 0.15) {
jeannie9809 11:58a443cf3f7c 398 LF = LF + movave[i];
jeannie9809 11:58a443cf3f7c 399 i++;
jeannie9809 11:58a443cf3f7c 400 }
jeannie9809 11:58a443cf3f7c 401
jeannie9809 11:58a443cf3f7c 402 double HF = 0.0;
jeannie9809 11:58a443cf3f7c 403 while(TIME[i] < 0.4) {
jeannie9809 11:58a443cf3f7c 404 HF = HF + movave[i];
jeannie9809 11:58a443cf3f7c 405 i++;
jeannie9809 11:58a443cf3f7c 406 }
jeannie9809 11:58a443cf3f7c 407 HF_LF = HF/LF;
jeannie9809 11:58a443cf3f7c 408 printf("HF/LF = %.2f \n", HF_LF);
jeannie9809 11:58a443cf3f7c 409
jeannie9809 12:95d141421e18 410 if(SDNN_n < 1.0) {
jeannie9809 12:95d141421e18 411 SDNN_n = SDNN;
jeannie9809 12:95d141421e18 412 HF_LF_n = HF_LF;
jeannie9809 12:95d141421e18 413 valence_arousal = 0;
jeannie9809 12:95d141421e18 414 }
jeannie9809 12:95d141421e18 415 else {
jeannie9809 12:95d141421e18 416 if(SDNN > SDNN_n) {
jeannie9809 12:95d141421e18 417 if(HF_LF > HF_LF_n)
jeannie9809 12:95d141421e18 418 valence_arousal = 0;
jeannie9809 12:95d141421e18 419 else
jeannie9809 12:95d141421e18 420 valence_arousal = 3;
jeannie9809 12:95d141421e18 421 }
jeannie9809 12:95d141421e18 422 else {
jeannie9809 12:95d141421e18 423 if(HF_LF > HF_LF_n)
jeannie9809 12:95d141421e18 424 valence_arousal = 1;
jeannie9809 12:95d141421e18 425 else
jeannie9809 12:95d141421e18 426 valence_arousal = 2;
jeannie9809 12:95d141421e18 427 }
jeannie9809 12:95d141421e18 428 }
jeannie9809 12:95d141421e18 429
jeannie9809 11:58a443cf3f7c 430
jeannie9809 11:58a443cf3f7c 431 ready = false; // last line
jeannie9809 11:58a443cf3f7c 432
jeannie9809 5:a6b6d0e5a69f 433 }
jeannie9809 11:58a443cf3f7c 434
jeannie9809 0:338c50c9c8cd 435 }
jeannie9809 0:338c50c9c8cd 436
jeannie9809 0:338c50c9c8cd 437 // main() runs in its own thread in the OS
jeannie9809 3:0da9235c9069 438 int main()
jeannie9809 3:0da9235c9069 439 {
jeannie9809 0:338c50c9c8cd 440 // printf("Hello world.\r\n");
jeannie9809 4:94190624967a 441
jeannie9809 0:338c50c9c8cd 442 t.start(callback(&evqueue, &EventQueue::dispatch_forever));
jeannie9809 0:338c50c9c8cd 443 kw40z_device.attach_buttonLeft(&ButtonLeft);
jeannie9809 0:338c50c9c8cd 444 kw40z_device.attach_buttonRight(&ButtonRight);
jeannie9809 3:0da9235c9069 445
jeannie9809 0:338c50c9c8cd 446 pwr1v8 = 1;
jeannie9809 0:338c50c9c8cd 447 pwr3v3b = 1;
jeannie9809 0:338c50c9c8cd 448 pwr15v = 0;
jeannie9809 3:0da9235c9069 449
jeannie9809 0:338c50c9c8cd 450 maximInterrupt.fall(interruptHandler);
jeannie9809 0:338c50c9c8cd 451 maximInterrupt.enable_irq();
jeannie9809 3:0da9235c9069 452
jeannie9809 0:338c50c9c8cd 453 MAX30101::InterruptBitField_u interruptStatus;
jeannie9809 0:338c50c9c8cd 454 interruptStatus.all = 0xFF;
jeannie9809 0:338c50c9c8cd 455 hr.enableInterrupts(interruptStatus);
jeannie9809 3:0da9235c9069 456
jeannie9809 3:0da9235c9069 457 char text[20]; /* Text Buffer */
jeannie9809 0:338c50c9c8cd 458 oled_text_properties_t textProperties = {0};
jeannie9809 3:0da9235c9069 459 oled.GetTextProperties(&textProperties);
jeannie9809 0:338c50c9c8cd 460 /* Turn on the backlight of the OLED Display */
jeannie9809 0:338c50c9c8cd 461 oled.DimScreenON();
jeannie9809 3:0da9235c9069 462
jeannie9809 3:0da9235c9069 463 /* Fills the screen with solid black */
jeannie9809 0:338c50c9c8cd 464 oled.FillScreen(COLOR_BLACK);
jeannie9809 1:eabf219849ab 465 strcpy((char *) text, "Raw PPG:");
jeannie9809 0:338c50c9c8cd 466 oled.Label((uint8_t *)text,7,0);
jeannie9809 3:0da9235c9069 467
jeannie9809 11:58a443cf3f7c 468 strcpy((char *) text, " LOFI");
jeannie9809 0:338c50c9c8cd 469 oled.Label((uint8_t *)text,7,40);
jeannie9809 0:338c50c9c8cd 470 //dynamic text setup
jeannie9809 0:338c50c9c8cd 471 textProperties.fontColor = COLOR_WHITE;
jeannie9809 0:338c50c9c8cd 472 textProperties.alignParam = OLED_TEXT_ALIGN_RIGHT;
jeannie9809 3:0da9235c9069 473 oled.SetTextProperties(&textProperties);
jeannie9809 3:0da9235c9069 474
jeannie9809 0:338c50c9c8cd 475 txThread.start(txTask);
jeannie9809 11:58a443cf3f7c 476
jeannie9809 12:95d141421e18 477
jeannie9809 0:338c50c9c8cd 478 while (true) {
jeannie9809 3:0da9235c9069 479
jeannie9809 11:58a443cf3f7c 480 // Format the time reading
jeannie9809 1:eabf219849ab 481 sprintf(text,"%d",ppg_single_sample);
jeannie9809 0:338c50c9c8cd 482 oled.TextBox((uint8_t *)text,55,15,35,15); //Increase textbox for more digits
jeannie9809 11:58a443cf3f7c 483
jeannie9809 11:58a443cf3f7c 484 /*
jeannie9809 11:58a443cf3f7c 485 // Display time reading in 35px by 15px textbox at(x=55, y=40)
jeannie9809 3:0da9235c9069 486 if (ppg_single_sample > 45) {
jeannie9809 1:eabf219849ab 487 sprintf(text,"%d; %.2f",SDNN, HF_LF);
jeannie9809 3:0da9235c9069 488
jeannie9809 11:58a443cf3f7c 489 // Display time reading in 35px by 15px textbox at(x=55, y=40)
jeannie9809 0:338c50c9c8cd 490 oled.TextBox((uint8_t *)text,55,55,35,15); //Increase textbox for more digits
jeannie9809 3:0da9235c9069 491 } else {
jeannie9809 0:338c50c9c8cd 492 sprintf(text,"wait HR");
jeannie9809 3:0da9235c9069 493
jeannie9809 11:58a443cf3f7c 494 // Display time reading in 35px by 15px textbox at(x=55, y=40)
jeannie9809 0:338c50c9c8cd 495 oled.TextBox((uint8_t *)text,55,55,35,15);
jeannie9809 11:58a443cf3f7c 496 }*/
jeannie9809 11:58a443cf3f7c 497
jeannie9809 12:95d141421e18 498
jeannie9809 12:95d141421e18 499
jeannie9809 0:338c50c9c8cd 500 Thread::wait(1000);
jeannie9809 3:0da9235c9069 501 }
jeannie9809 0:338c50c9c8cd 502 return 0;
jeannie9809 0:338c50c9c8cd 503 }
jeannie9809 0:338c50c9c8cd 504