193 lines
6.4 KiB
C
193 lines
6.4 KiB
C
/*
|
||
* Copyright (c) 2022 Jinan Bosai Network Technology Co., LTD Ltd.
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
#include <stdio.h>
|
||
#include <string.h>
|
||
#include <unistd.h>
|
||
|
||
#include "max30102.h"
|
||
#include "algorithm.h"
|
||
#include "cmsis_os2.h"
|
||
#include "ohos_init.h"
|
||
|
||
#define TASK_STACK_SIZE 1024 * 8
|
||
#define TASK_PRIO 25
|
||
|
||
uint32_t aun_ir_buffer[500]; //IR LED sensor data
|
||
int32_t n_ir_buffer_length; //data length
|
||
uint32_t aun_red_buffer[500]; //Red LED sensor data
|
||
int32_t n_sp02; //SPO2 value
|
||
int8_t ch_spo2_valid; //indicator to show if the SP02 calculation is valid
|
||
int32_t n_heart_rate; //heart rate value
|
||
int8_t ch_hr_valid; //indicator to show if the heart rate calculation is valid
|
||
uint8_t uch_dummy;
|
||
|
||
#define MAX_BRIGHTNESS 255
|
||
|
||
static void ExampleTask(void)
|
||
{
|
||
//variables to calculate the on-board LED brightness that reflects the heartbeats
|
||
uint32_t un_min, un_max, un_prev_data;
|
||
int i;
|
||
int32_t n_brightness;
|
||
float f_temp;
|
||
uint8_t temp_num=0;
|
||
uint8_t temp[6]={0,0,0,0,0,0};
|
||
uint8_t str[100];
|
||
uint8_t dis_hr=0,dis_spo2=0;
|
||
|
||
BoardInit();
|
||
|
||
maxim_max30102_init();
|
||
|
||
printf("\r\n MAX30102 init \r\n");
|
||
|
||
un_min=0x3FFFF;
|
||
un_max=0;
|
||
|
||
n_ir_buffer_length=500; //buffer length of 100 stores 5 seconds of samples running at 100sps
|
||
//read the first 500 samples, and determine the signal range
|
||
for(i=0;i<n_ir_buffer_length;i++)
|
||
{
|
||
while(GetInit()==1); //wait until the interrupt pin asserts
|
||
|
||
max30102_FIFO_ReadBytes(REG_FIFO_DATA,temp);
|
||
aun_red_buffer[i] = (long)((long)((long)temp[0]&0x03)<<16) | (long)temp[1]<<8 | (long)temp[2]; // Combine values to get the actual number
|
||
aun_ir_buffer[i] = (long)((long)((long)temp[3] & 0x03)<<16) |(long)temp[4]<<8 | (long)temp[5]; // Combine values to get the actual number
|
||
|
||
if(un_min>aun_red_buffer[i])
|
||
un_min=aun_red_buffer[i]; //update signal min
|
||
if(un_max<aun_red_buffer[i])
|
||
un_max=aun_red_buffer[i]; //update signal max
|
||
// printf("red=");
|
||
// printf("%i", aun_red_buffer[i]);
|
||
// printf(", ir=");
|
||
// printf("%i\n\r", aun_ir_buffer[i]);
|
||
}
|
||
un_prev_data=aun_red_buffer[i];
|
||
//calculate heart rate and SpO2 after first 500 samples (first 5 seconds of samples)
|
||
maxim_heart_rate_and_oxygen_saturation(aun_ir_buffer, n_ir_buffer_length, aun_red_buffer, &n_sp02, &ch_spo2_valid, &n_heart_rate, &ch_hr_valid);
|
||
|
||
while(1)
|
||
{
|
||
i=0;
|
||
un_min=0x3FFFF;
|
||
un_max=0;
|
||
|
||
//dumping the first 100 sets of samples in the memory and shift the last 400 sets of samples to the top
|
||
for(i=100;i<500;i++)
|
||
{
|
||
aun_red_buffer[i-100]=aun_red_buffer[i];
|
||
aun_ir_buffer[i-100]=aun_ir_buffer[i];
|
||
|
||
//update the signal min and max
|
||
if(un_min>aun_red_buffer[i])
|
||
un_min=aun_red_buffer[i];
|
||
if(un_max<aun_red_buffer[i])
|
||
un_max=aun_red_buffer[i];
|
||
}
|
||
//take 100 sets of samples before calculating the heart rate.
|
||
for(i=400;i<500;i++)
|
||
{
|
||
un_prev_data=aun_red_buffer[i-1];
|
||
while(GetInit()==1);
|
||
max30102_FIFO_ReadBytes(REG_FIFO_DATA,temp);
|
||
aun_red_buffer[i] = (long)((long)((long)temp[0]&0x03)<<16) | (long)temp[1]<<8 | (long)temp[2]; // Combine values to get the actual number
|
||
aun_ir_buffer[i] = (long)((long)((long)temp[3] & 0x03)<<16) |(long)temp[4]<<8 | (long)temp[5]; // Combine values to get the actual number
|
||
|
||
if(aun_red_buffer[i]>un_prev_data)
|
||
{
|
||
f_temp=aun_red_buffer[i]-un_prev_data;
|
||
f_temp/=(un_max-un_min);
|
||
f_temp*=MAX_BRIGHTNESS;
|
||
n_brightness-=(int)f_temp;
|
||
if(n_brightness<0)
|
||
n_brightness=0;
|
||
}
|
||
else
|
||
{
|
||
f_temp=un_prev_data-aun_red_buffer[i];
|
||
f_temp/=(un_max-un_min);
|
||
f_temp*=MAX_BRIGHTNESS;
|
||
n_brightness+=(int)f_temp;
|
||
if(n_brightness>MAX_BRIGHTNESS)
|
||
n_brightness=MAX_BRIGHTNESS;
|
||
}
|
||
//send samples and calculation result to terminal program through UART
|
||
if(ch_hr_valid == 1 && n_heart_rate<120)//**/ ch_hr_valid == 1 && ch_spo2_valid ==1 && n_heart_rate<120 && n_sp02<101
|
||
{
|
||
dis_hr = n_heart_rate;
|
||
dis_spo2 = n_sp02;
|
||
}
|
||
else
|
||
{
|
||
dis_hr = 0;
|
||
dis_spo2 = 0;
|
||
}
|
||
|
||
}
|
||
// for(i=0;i<500;i++){
|
||
// printf("0x%02x,",aun_ir_buffer[i]);
|
||
// }
|
||
// printf("\r\n");
|
||
// printf("--------0-0-----\r\n");
|
||
// for(i=0;i<500;i++){
|
||
// printf("0x%02x,",aun_red_buffer[i]);
|
||
// }
|
||
// printf("\r\n");
|
||
// printf("--------0-0-----\r\n");
|
||
maxim_heart_rate_and_oxygen_saturation(aun_ir_buffer, n_ir_buffer_length, aun_red_buffer, &n_sp02, &ch_spo2_valid, &n_heart_rate, &ch_hr_valid);
|
||
printf("HR=%i, ", n_heart_rate);
|
||
printf("HRvalid=%i, ", ch_hr_valid);
|
||
// printf("SpO2=%i, ", n_sp02);
|
||
// printf("SPO2Valid=%i\r\n", ch_spo2_valid);
|
||
|
||
sleep(1);
|
||
// //显示刷新
|
||
// LED0=0;
|
||
// if(dis_hr == 0 && dis_spo2 == 0) //**dis_hr == 0 && dis_spo2 == 0
|
||
// {
|
||
// sprintf((char *)str,"HR:--- SpO2:--- ");//**HR:--- SpO2:---
|
||
// }
|
||
// else{
|
||
// sprintf((char *)str,"HR:%3d SpO2:%3d ",dis_hr,dis_spo2);//**HR:%3d SpO2:%3d
|
||
// }
|
||
// OLED_ShowString(0,0,str,16);
|
||
// OLED_Fill(0,23,127,63,0);
|
||
// //红光在上，红外在下
|
||
// dis_DrawCurve(aun_red_buffer,20);
|
||
// dis_DrawCurve(aun_ir_buffer,0);
|
||
// OLED_Refresh_Gram();//更新显示到OLED
|
||
}
|
||
}
|
||
|
||
static void ExampleEntry(void)
|
||
{
|
||
osThreadAttr_t attr;
|
||
|
||
attr.name = "ExampleTask";
|
||
attr.attr_bits = 0U;
|
||
attr.cb_mem = NULL;
|
||
attr.cb_size = 0U;
|
||
attr.stack_mem = NULL;
|
||
attr.stack_size = TASK_STACK_SIZE;
|
||
attr.priority = TASK_PRIO;
|
||
|
||
if (osThreadNew((osThreadFunc_t)ExampleTask, NULL, &attr) == NULL) {
|
||
printf("Failed to create ExampleTask!\n");
|
||
}
|
||
}
|
||
|
||
APP_FEATURE_INIT(ExampleEntry); |