Working code for pc app 12/01/2018 commit
Dependencies: mbed MS5607 mbed-dsp
Fork of Turrentine_Code by
main.cpp@4:d4804771134a, 2018-01-25 (annotated)
- Committer:
- intrinseca
- Date:
- Thu Jan 25 10:20:46 2018 +0000
- Revision:
- 4:d4804771134a
- Parent:
- 3:c80aa39db5bb
- Child:
- 5:44cfc3e81d1c
Implement peak finding algorithm and regression-based calibration
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
AlexStokoe | 0:2c6d81be69d8 | 1 | #include "mbed.h" |
intrinseca | 3:c80aa39db5bb | 2 | #include "MS5607SPI.h" |
intrinseca | 3:c80aa39db5bb | 3 | #include "arm_math.h" |
AlexStokoe | 0:2c6d81be69d8 | 4 | |
AlexStokoe | 0:2c6d81be69d8 | 5 | #define M_PI 3.14159265358979323846 |
AlexStokoe | 0:2c6d81be69d8 | 6 | |
intrinseca | 2:3d3e21c907e4 | 7 | #define N_SAMPLES 1024 |
intrinseca | 1:ab3dacbfcde6 | 8 | |
intrinseca | 4:d4804771134a | 9 | #define PEAK_SEARCH_START 1 //first bin to include in peak search |
intrinseca | 4:d4804771134a | 10 | #define PEAK_SEARCH_END 50 //last bin to include in peak search |
intrinseca | 4:d4804771134a | 11 | #define SUM_AROUND_PEAK 1 //+/- this many bins |
intrinseca | 4:d4804771134a | 12 | |
intrinseca | 4:d4804771134a | 13 | #define REGRESSION_SLOPE 53.77986f //from Excel fit to test data |
intrinseca | 4:d4804771134a | 14 | #define REGRESSION_INTERCEPT -80.07f |
intrinseca | 4:d4804771134a | 15 | |
intrinseca | 4:d4804771134a | 16 | float thresholds[] = {10.f, 30.f, 50.f, 70.f}; //estimated t90 thresholds to turn on LEDs |
intrinseca | 4:d4804771134a | 17 | |
AlexStokoe | 0:2c6d81be69d8 | 18 | //mbed class def |
AlexStokoe | 0:2c6d81be69d8 | 19 | Serial pc(USBTX, USBRX); // tx, rx |
intrinseca | 3:c80aa39db5bb | 20 | MS5607SPI pressure_sensor(p5, p6, p7, p8); // mosi, miso, sclk, cs |
AlexStokoe | 0:2c6d81be69d8 | 21 | |
intrinseca | 3:c80aa39db5bb | 22 | BusOut leds(LED1, LED2, LED3, LED4); |
AlexStokoe | 0:2c6d81be69d8 | 23 | |
intrinseca | 3:c80aa39db5bb | 24 | PwmOut motor(p26); |
AlexStokoe | 0:2c6d81be69d8 | 25 | |
intrinseca | 3:c80aa39db5bb | 26 | Timer sample_timer; |
intrinseca | 3:c80aa39db5bb | 27 | Timer iter_timer; |
AlexStokoe | 0:2c6d81be69d8 | 28 | |
intrinseca | 3:c80aa39db5bb | 29 | float p_data[N_SAMPLES]; |
intrinseca | 3:c80aa39db5bb | 30 | float p_data_raw[N_SAMPLES]; |
intrinseca | 3:c80aa39db5bb | 31 | float f_data[N_SAMPLES]; |
intrinseca | 3:c80aa39db5bb | 32 | float mag_data[N_SAMPLES / 2 + 1]; |
AlexStokoe | 0:2c6d81be69d8 | 33 | |
intrinseca | 1:ab3dacbfcde6 | 34 | unsigned int tData[N_SAMPLES]; |
AlexStokoe | 0:2c6d81be69d8 | 35 | |
intrinseca | 3:c80aa39db5bb | 36 | arm_rfft_fast_instance_f32 rfft; |
AlexStokoe | 0:2c6d81be69d8 | 37 | |
AlexStokoe | 0:2c6d81be69d8 | 38 | float duty = 1; |
AlexStokoe | 0:2c6d81be69d8 | 39 | |
AlexStokoe | 0:2c6d81be69d8 | 40 | int round(float number) |
AlexStokoe | 0:2c6d81be69d8 | 41 | { |
AlexStokoe | 0:2c6d81be69d8 | 42 | return (number >= 0) ? (int)(number + 0.5) : (int)(number - 0.5); |
AlexStokoe | 0:2c6d81be69d8 | 43 | } |
AlexStokoe | 0:2c6d81be69d8 | 44 | |
intrinseca | 1:ab3dacbfcde6 | 45 | int main() |
intrinseca | 1:ab3dacbfcde6 | 46 | { |
intrinseca | 3:c80aa39db5bb | 47 | printf("Turrentine\n"); |
intrinseca | 3:c80aa39db5bb | 48 | |
intrinseca | 3:c80aa39db5bb | 49 | //Configure PC serial link |
AlexStokoe | 0:2c6d81be69d8 | 50 | pc.baud(115200); |
intrinseca | 3:c80aa39db5bb | 51 | |
intrinseca | 3:c80aa39db5bb | 52 | //Configure motor PWM |
intrinseca | 3:c80aa39db5bb | 53 | motor = 0; |
intrinseca | 3:c80aa39db5bb | 54 | motor.period_ms(10); |
intrinseca | 2:3d3e21c907e4 | 55 | |
intrinseca | 2:3d3e21c907e4 | 56 | printf("Pump On\n"); |
intrinseca | 2:3d3e21c907e4 | 57 | //turn pump on |
intrinseca | 3:c80aa39db5bb | 58 | motor.write(duty); |
intrinseca | 2:3d3e21c907e4 | 59 | |
intrinseca | 1:ab3dacbfcde6 | 60 | printf("Start loop\n"); |
intrinseca | 3:c80aa39db5bb | 61 | int iter_counter = 0; //Iteration counter |
intrinseca | 3:c80aa39db5bb | 62 | iter_timer.start(); |
intrinseca | 1:ab3dacbfcde6 | 63 | |
intrinseca | 4:d4804771134a | 64 | printf("fsmpl titer mean_press fmax sum_peak t90\n"); |
intrinseca | 4:d4804771134a | 65 | |
intrinseca | 3:c80aa39db5bb | 66 | //program loop |
intrinseca | 3:c80aa39db5bb | 67 | while(1) { |
intrinseca | 3:c80aa39db5bb | 68 | sample_timer.reset(); |
intrinseca | 3:c80aa39db5bb | 69 | sample_timer.start(); |
intrinseca | 3:c80aa39db5bb | 70 | |
intrinseca | 3:c80aa39db5bb | 71 | float temperature_raw = pressure_sensor.getRawTemperature(); |
intrinseca | 3:c80aa39db5bb | 72 | |
intrinseca | 3:c80aa39db5bb | 73 | //Capture raw pressure samples |
intrinseca | 1:ab3dacbfcde6 | 74 | for(int i = 0; i < N_SAMPLES; i++) { |
intrinseca | 3:c80aa39db5bb | 75 | p_data_raw[i] = pressure_sensor.calculatePressure(pressure_sensor.getRawPressure(), temperature_raw); |
intrinseca | 3:c80aa39db5bb | 76 | p_data[i] = p_data_raw[i]; |
intrinseca | 1:ab3dacbfcde6 | 77 | } |
intrinseca | 3:c80aa39db5bb | 78 | |
intrinseca | 3:c80aa39db5bb | 79 | sample_timer.stop(); |
intrinseca | 3:c80aa39db5bb | 80 | |
intrinseca | 3:c80aa39db5bb | 81 | //http://www.keil.com/pack/doc/CMSIS/DSP/html/group__RealFFT.html |
intrinseca | 3:c80aa39db5bb | 82 | //Compute the RFFT |
intrinseca | 3:c80aa39db5bb | 83 | // |
intrinseca | 3:c80aa39db5bb | 84 | //p_data is trashed in the process |
intrinseca | 3:c80aa39db5bb | 85 | //Result is packaged as [real_0, real_N/2, real_1, imag_1, real_2, imag_2 ... real_N/2-1, imag_N/2-1] |
intrinseca | 3:c80aa39db5bb | 86 | arm_rfft_fast_init_f32(&rfft, N_SAMPLES); |
intrinseca | 3:c80aa39db5bb | 87 | arm_rfft_fast_f32(&rfft, p_data, f_data, 0); |
intrinseca | 3:c80aa39db5bb | 88 | |
intrinseca | 3:c80aa39db5bb | 89 | //http://www.keil.com/pack/doc/CMSIS/DSP/html/group__cmplx__mag.html |
intrinseca | 3:c80aa39db5bb | 90 | //Convert to magntiude, skip over the DC and fundamental terms |
intrinseca | 3:c80aa39db5bb | 91 | |
intrinseca | 3:c80aa39db5bb | 92 | arm_cmplx_mag_f32(&f_data[2], &mag_data[1], (N_SAMPLES / 2) - 1); |
intrinseca | 3:c80aa39db5bb | 93 | //Fill in the first and last terms from the first entry in f_data |
intrinseca | 3:c80aa39db5bb | 94 | mag_data[0] = f_data[0]; |
intrinseca | 3:c80aa39db5bb | 95 | mag_data[N_SAMPLES / 2] = f_data[1]; |
intrinseca | 3:c80aa39db5bb | 96 | |
intrinseca | 3:c80aa39db5bb | 97 | //Output raw data |
intrinseca | 3:c80aa39db5bb | 98 | /*for(int i = 0; i < N_SAMPLES; i++) { |
intrinseca | 3:c80aa39db5bb | 99 | printf("%f %f\n", p_data_raw[i], f_data[i]); |
intrinseca | 3:c80aa39db5bb | 100 | }*/ |
intrinseca | 3:c80aa39db5bb | 101 | |
intrinseca | 3:c80aa39db5bb | 102 | |
intrinseca | 4:d4804771134a | 103 | //Find peak in spectrum |
intrinseca | 4:d4804771134a | 104 | float max_mag = -1; |
intrinseca | 4:d4804771134a | 105 | int max_mag_i = -1; |
intrinseca | 4:d4804771134a | 106 | for(int i = PEAK_SEARCH_START + SUM_AROUND_PEAK; i < PEAK_SEARCH_END; i++){ |
intrinseca | 4:d4804771134a | 107 | //printf("%10f ", mag_data[i]); |
intrinseca | 4:d4804771134a | 108 | if(mag_data[i] > max_mag) { |
intrinseca | 4:d4804771134a | 109 | max_mag_i = i; |
intrinseca | 4:d4804771134a | 110 | max_mag = mag_data[i]; |
intrinseca | 4:d4804771134a | 111 | } |
intrinseca | 4:d4804771134a | 112 | } |
intrinseca | 4:d4804771134a | 113 | //printf("\n"); |
intrinseca | 4:d4804771134a | 114 | |
intrinseca | 4:d4804771134a | 115 | //Sum surrounding |
intrinseca | 4:d4804771134a | 116 | float sum = 0.f; |
intrinseca | 4:d4804771134a | 117 | for(int i = max_mag_i - SUM_AROUND_PEAK; i < (max_mag_i + SUM_AROUND_PEAK + 1); i++) { |
intrinseca | 4:d4804771134a | 118 | sum += mag_data[i]; |
intrinseca | 4:d4804771134a | 119 | //printf("%10f ", mag_data[i]); |
intrinseca | 1:ab3dacbfcde6 | 120 | } |
intrinseca | 3:c80aa39db5bb | 121 | |
intrinseca | 4:d4804771134a | 122 | //printf("\n"); |
intrinseca | 4:d4804771134a | 123 | |
intrinseca | 4:d4804771134a | 124 | //Scale by N_SAMPLES to recover pressure in Pa |
intrinseca | 4:d4804771134a | 125 | sum /= N_SAMPLES; |
intrinseca | 4:d4804771134a | 126 | |
intrinseca | 4:d4804771134a | 127 | //Apply inverse of logarithmic regression from test data |
intrinseca | 4:d4804771134a | 128 | float t90 = exp((sum + 80.07) / 53.77986); |
intrinseca | 4:d4804771134a | 129 | |
intrinseca | 4:d4804771134a | 130 | |
intrinseca | 4:d4804771134a | 131 | //Output iteration data |
intrinseca | 4:d4804771134a | 132 | iter_counter++; |
intrinseca | 4:d4804771134a | 133 | printf("%5d %5d %10.0f %4d %10.5e %3.0f\n", 1000000 / (sample_timer.read_us() / N_SAMPLES), iter_timer.read_ms() / iter_counter, f_data[0] / N_SAMPLES, max_mag_i, sum, t90); |
intrinseca | 3:c80aa39db5bb | 134 | |
intrinseca | 3:c80aa39db5bb | 135 | //Set LEDs based on thresholds |
intrinseca | 4:d4804771134a | 136 | for(int i = 0; i < 4; i++) { |
intrinseca | 4:d4804771134a | 137 | leds[i] = t90 >= thresholds[i]; |
intrinseca | 4:d4804771134a | 138 | } |
intrinseca | 3:c80aa39db5bb | 139 | |
intrinseca | 3:c80aa39db5bb | 140 | //break; |
intrinseca | 1:ab3dacbfcde6 | 141 | } |
AlexStokoe | 0:2c6d81be69d8 | 142 | } |