Capstone project files
Dependencies: mbed-dsp mbed capstone_display_2
main.cpp@7:fc55813f823e, 2014-04-24 (annotated)
- Committer:
- ryanyuyu
- Date:
- Thu Apr 24 21:35:49 2014 +0000
- Revision:
- 7:fc55813f823e
- Parent:
- 6:8441a6864784
- Child:
- 8:dcc69fc6d88b
Works for two-gain stages.
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
ryanyuyu | 0:3aae5d23d0db | 1 | #include "mbed.h" |
ryanyuyu | 2:8ae58834937f | 2 | #include "FIR_f32.h" |
ryanyuyu | 0:3aae5d23d0db | 3 | #include "arm_math.h" |
ryanyuyu | 3:30dcfcf9412c | 4 | #include "display.h" |
ryanyuyu | 6:8441a6864784 | 5 | #include "st7735.h" |
ryanyuyu | 7:fc55813f823e | 6 | //#include <string> |
ryanyuyu | 7:fc55813f823e | 7 | //#include <sstream> |
ryanyuyu | 2:8ae58834937f | 8 | #define f_sampling 2000 //the sampling frequency |
ryanyuyu | 4:9ee3ae61db7f | 9 | #define NumTaps 27 //the number of filter coefficients |
ryanyuyu | 4:9ee3ae61db7f | 10 | #define BlockSize 512 //the size of the buffer |
ryanyuyu | 6:8441a6864784 | 11 | #define numCallibrationSteps 6 //the number of callibration steps or points |
ryanyuyu | 6:8441a6864784 | 12 | #define numGainStages 2 |
ryanyuyu | 4:9ee3ae61db7f | 13 | Serial pc(USBTX, USBRX); //USB serial for PC, to be removed later |
ryanyuyu | 4:9ee3ae61db7f | 14 | AnalogOut waveOut(p18); //for debugging |
ryanyuyu | 4:9ee3ae61db7f | 15 | |
ryanyuyu | 4:9ee3ae61db7f | 16 | //-------------------- SPI communication |
ryanyuyu | 6:8441a6864784 | 17 | SPI spi(p5, p6, p7); //MOSI, MISO, SCLK |
ryanyuyu | 3:30dcfcf9412c | 18 | DigitalOut cs(p8); |
ryanyuyu | 6:8441a6864784 | 19 | DigitalIn button(p21); |
ryanyuyu | 4:9ee3ae61db7f | 20 | |
ryanyuyu | 4:9ee3ae61db7f | 21 | //-------------------- LCD display |
ryanyuyu | 3:30dcfcf9412c | 22 | ST7735_LCD disp( p14, p13, p12, p10, p11); //for digital display |
ryanyuyu | 3:30dcfcf9412c | 23 | display lcd(&disp); |
ryanyuyu | 6:8441a6864784 | 24 | char* newString(int length); //prototype for newString |
ryanyuyu | 7:fc55813f823e | 25 | char* outputString = newString(64); |
ryanyuyu | 7:fc55813f823e | 26 | char* strength = newString(64); |
ryanyuyu | 7:fc55813f823e | 27 | char* dist = newString(64); |
ryanyuyu | 2:8ae58834937f | 28 | |
ryanyuyu | 4:9ee3ae61db7f | 29 | //-------------------- signal-related stuff |
ryanyuyu | 2:8ae58834937f | 30 | AnalogIn input(p15); //pin 15 for analog reading |
ryanyuyu | 2:8ae58834937f | 31 | float32_t waveform[BlockSize]; //array of input data |
ryanyuyu | 2:8ae58834937f | 32 | float32_t postFilterData[BlockSize]; //array of filtered data |
ryanyuyu | 2:8ae58834937f | 33 | bool fullRead; //whether the MBED has finish |
ryanyuyu | 2:8ae58834937f | 34 | bool waitForNext; |
ryanyuyu | 4:9ee3ae61db7f | 35 | int index_g; //tracks the index for the waveform array |
ryanyuyu | 2:8ae58834937f | 36 | |
ryanyuyu | 6:8441a6864784 | 37 | //-------------------for distance calculation and calibration |
ryanyuyu | 6:8441a6864784 | 38 | bool adjusting = true; //whether the user is still adjusting the beacon's distance |
ryanyuyu | 4:9ee3ae61db7f | 39 | float minThreshold; |
ryanyuyu | 4:9ee3ae61db7f | 40 | float maxThreshold; |
ryanyuyu | 6:8441a6864784 | 41 | int callibrationStep; |
ryanyuyu | 6:8441a6864784 | 42 | int state; |
ryanyuyu | 6:8441a6864784 | 43 | int gainStage; |
ryanyuyu | 6:8441a6864784 | 44 | float gainMultiplier; |
ryanyuyu | 6:8441a6864784 | 45 | float gainCutoffs[numGainStages] = {20.0, 100.0}; |
ryanyuyu | 6:8441a6864784 | 46 | //gainCutoffs = {20.0, 100.0, 1200.0, 10000.0}; |
ryanyuyu | 4:9ee3ae61db7f | 47 | float gain1; |
ryanyuyu | 6:8441a6864784 | 48 | float gain0; |
ryanyuyu | 6:8441a6864784 | 49 | //These constants are for linear interpolation for the varius gain stage. Two linear equations per stage (piecewise) |
ryanyuyu | 6:8441a6864784 | 50 | float linearSamples[numCallibrationSteps]; |
ryanyuyu | 6:8441a6864784 | 51 | int callibrationPoints[numCallibrationSteps] = {6, 10, 14, 14, 20, 24}; |
ryanyuyu | 6:8441a6864784 | 52 | //callibrationPoints = {6, 10, 14, 14, 20, 24, 26, 36, 50, 50, 62, 78}; |
ryanyuyu | 6:8441a6864784 | 53 | |
ryanyuyu | 6:8441a6864784 | 54 | float mLower[numGainStages]; //m (slope) lower portion |
ryanyuyu | 6:8441a6864784 | 55 | float bLower[numGainStages]; //b (y-offset) lower portion |
ryanyuyu | 6:8441a6864784 | 56 | float mid[numGainStages]; //the middle x-value for the piecewise |
ryanyuyu | 6:8441a6864784 | 57 | float mUpper[numGainStages]; //m (slope) upper portion |
ryanyuyu | 6:8441a6864784 | 58 | float bUpper[numGainStages]; //b (y-offset) upper portion |
ryanyuyu | 6:8441a6864784 | 59 | |
ryanyuyu | 6:8441a6864784 | 60 | /* |
ryanyuyu | 6:8441a6864784 | 61 | float m10; |
ryanyuyu | 6:8441a6864784 | 62 | float b10; |
ryanyuyu | 6:8441a6864784 | 63 | float mid1; |
ryanyuyu | 6:8441a6864784 | 64 | float m11; |
ryanyuyu | 6:8441a6864784 | 65 | float b11; |
ryanyuyu | 6:8441a6864784 | 66 | |
ryanyuyu | 6:8441a6864784 | 67 | float m20; |
ryanyuyu | 6:8441a6864784 | 68 | float b20; |
ryanyuyu | 6:8441a6864784 | 69 | float mid2; |
ryanyuyu | 6:8441a6864784 | 70 | float m21; |
ryanyuyu | 6:8441a6864784 | 71 | float b21; |
ryanyuyu | 6:8441a6864784 | 72 | |
ryanyuyu | 6:8441a6864784 | 73 | float m30; |
ryanyuyu | 6:8441a6864784 | 74 | float b30; |
ryanyuyu | 6:8441a6864784 | 75 | float mid3; |
ryanyuyu | 6:8441a6864784 | 76 | float m31; |
ryanyuyu | 6:8441a6864784 | 77 | float b31; |
ryanyuyu | 6:8441a6864784 | 78 | */ |
ryanyuyu | 3:30dcfcf9412c | 79 | |
ryanyuyu | 4:9ee3ae61db7f | 80 | //------------------------the filter coefficients for FIR filter |
ryanyuyu | 2:8ae58834937f | 81 | float32_t pCoeffs[NumTaps] = |
ryanyuyu | 2:8ae58834937f | 82 | { 0.012000000000000, 0.012462263166161, -0.019562318415964, -0.026175892863747, |
ryanyuyu | 2:8ae58834937f | 83 | 0.031654803781611, 0.050648026372209, -0.032547136829180, -0.070997780956819, |
ryanyuyu | 2:8ae58834937f | 84 | 0.032992306874347, 0.094643188024724, -0.020568171368385, -0.106071176200193, |
ryanyuyu | 2:8ae58834937f | 85 | 0.009515198320277, 0.114090808482376, 0.009515198320275, -0.106071176200193, |
ryanyuyu | 2:8ae58834937f | 86 | -0.020568171368382, 0.094643188024728, 0.032992306874351, -0.070997780956815, |
ryanyuyu | 2:8ae58834937f | 87 | -0.032547136829177, 0.050648026372211, 0.031654803781612, -0.026175892863746, |
ryanyuyu | 2:8ae58834937f | 88 | -0.019562318415964, 0.012462263166161, 0.012000000000000 }; |
ryanyuyu | 4:9ee3ae61db7f | 89 | float32_t pState[NumTaps + BlockSize - 1]; |
ryanyuyu | 3:30dcfcf9412c | 90 | // */ |
ryanyuyu | 4:9ee3ae61db7f | 91 | |
ryanyuyu | 4:9ee3ae61db7f | 92 | |
ryanyuyu | 4:9ee3ae61db7f | 93 | //-----------------------IIR stuff (if needed) |
ryanyuyu | 4:9ee3ae61db7f | 94 | /* |
ryanyuyu | 4:9ee3ae61db7f | 95 | float32_t pkCoeffs[NumTaps] = |
ryanyuyu | 4:9ee3ae61db7f | 96 | { |
ryanyuyu | 4:9ee3ae61db7f | 97 | 1,-2.496708288,3.17779085,-2.022333713,0.6561,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0 |
ryanyuyu | 4:9ee3ae61db7f | 98 | }; |
ryanyuyu | 4:9ee3ae61db7f | 99 | |
ryanyuyu | 4:9ee3ae61db7f | 100 | float32_t pvCoeffs[NumTaps] = |
ryanyuyu | 4:9ee3ae61db7f | 101 | { |
ryanyuyu | 4:9ee3ae61db7f | 102 | 0.0000556000,0.0002167120,0.0004326320,0.0005056930,0.0002111890,-0.0004911030,-0.0013071920,-0.0017060250,-0.0012444070,0.0000684000,0.0016603140,0.0026622100,0.0024306750,0.0009787140,-0.0009787140,-0.0024306750,-0.0026622100,-0.0016603140,-0.0000684000,0.0012444070,0.0017060250,0.0013071920,0.0004911030,-0.0002111890,-0.0005056930,-0.0004326320,-0.0002167120,-0.0000556000 |
ryanyuyu | 4:9ee3ae61db7f | 103 | }; |
ryanyuyu | 4:9ee3ae61db7f | 104 | float32_t pState[NumTaps + BlockSize]; |
ryanyuyu | 4:9ee3ae61db7f | 105 | //*/ |
ryanyuyu | 4:9ee3ae61db7f | 106 | |
ryanyuyu | 4:9ee3ae61db7f | 107 | |
ryanyuyu | 4:9ee3ae61db7f | 108 | //--------------------------------if needed, the 4kHz FIR filter |
ryanyuyu | 3:30dcfcf9412c | 109 | /* |
ryanyuyu | 3:30dcfcf9412c | 110 | float32_t pCoeffs[NumTaps] = |
ryanyuyu | 3:30dcfcf9412c | 111 | { |
ryanyuyu | 3:30dcfcf9412c | 112 | -0.00130297171184699, -0.00456436168827987, -0.00757978930408609, -0.00696944302000657, |
ryanyuyu | 3:30dcfcf9412c | 113 | -0.00100059082174453, 0.00812867271498616, 0.0148953048520266, 0.0137935053264369, |
ryanyuyu | 3:30dcfcf9412c | 114 | 0.00350484996910501, -0.0112195199182290, -0.0216305356563913, -0.0202538386423356, |
ryanyuyu | 3:30dcfcf9412c | 115 | -0.00609419278464673, 0.0137348990478646, 0.0275645559768492, 0.0261107576153156, |
ryanyuyu | 3:30dcfcf9412c | 116 | 0.00866220574766616, -0.0156131009924596, -0.0324957126350438, -0.0311514181527343, |
ryanyuyu | 3:30dcfcf9412c | 117 | -0.0110879396617141, 0.0168179120126559, 0.0362758644669149, 0.0352058948414930, |
ryanyuyu | 3:30dcfcf9412c | 118 | 0.0132978095684398, -0.0172706692984796, -0.0386711719606551, -0.0379507530937637, |
ryanyuyu | 3:30dcfcf9412c | 119 | -0.0149419841919419, 0.0172996706397712, 0.0400000000000000, 0.0397279151377323, |
ryanyuyu | 3:30dcfcf9412c | 120 | 0.0164353142069562, -0.0164055618588934, -0.0396949785867063, -0.0399629114640568, |
ryanyuyu | 3:30dcfcf9412c | 121 | -0.0172605211576678, 0.0149790280104299, 0.0379815311949588, 0.0386933807609119, |
ryanyuyu | 3:30dcfcf9412c | 122 | 0.0172844840085185, -0.0132904115318555, -0.0352024033389307, -0.0362742608690452, |
ryanyuyu | 3:30dcfcf9412c | 123 | -0.0168170401765007, 0.0110885383139611, 0.0311518509994083, 0.0324959946809230, |
ryanyuyu | 3:30dcfcf9412c | 124 | 0.0156132578212073, -0.00866213238945794, -0.0261107291487171, -0.0275645472357883, |
ryanyuyu | 3:30dcfcf9412c | 125 | -0.0137348973043660, 0.00609419268963993, 0.0202538383407381, 0.0216305354798053, |
ryanyuyu | 3:30dcfcf9412c | 126 | 0.0112195198475825, -0.00350484999121515, -0.0137935053321021, -0.0148953048532365, |
ryanyuyu | 3:30dcfcf9412c | 127 | -0.00812867271519995, 0.00100059082171422, 0.00696944302000319, 0.00757978930408577, |
ryanyuyu | 3:30dcfcf9412c | 128 | 0.00456436168827984, 0.00130297171184699 |
ryanyuyu | 3:30dcfcf9412c | 129 | }; |
ryanyuyu | 3:30dcfcf9412c | 130 | //*/ |
ryanyuyu | 4:9ee3ae61db7f | 131 | |
ryanyuyu | 2:8ae58834937f | 132 | |
ryanyuyu | 2:8ae58834937f | 133 | |
ryanyuyu | 0:3aae5d23d0db | 134 | |
ryanyuyu | 6:8441a6864784 | 135 | char* newString(int length) //creates an initialized string of given length |
ryanyuyu | 6:8441a6864784 | 136 | { |
ryanyuyu | 6:8441a6864784 | 137 | char* temp = new char[length+1]; |
ryanyuyu | 6:8441a6864784 | 138 | for (int i = 0; i <= length; i++) temp[i] = '\0'; |
ryanyuyu | 6:8441a6864784 | 139 | return temp; |
ryanyuyu | 6:8441a6864784 | 140 | } |
ryanyuyu | 6:8441a6864784 | 141 | |
ryanyuyu | 7:fc55813f823e | 142 | /* |
ryanyuyu | 7:fc55813f823e | 143 | std::string convert(float number) |
ryanyuyu | 7:fc55813f823e | 144 | { |
ryanyuyu | 7:fc55813f823e | 145 | std::ostringstream buffer; |
ryanyuyu | 7:fc55813f823e | 146 | buffer<<number; |
ryanyuyu | 7:fc55813f823e | 147 | return buffer.str(); |
ryanyuyu | 7:fc55813f823e | 148 | } |
ryanyuyu | 7:fc55813f823e | 149 | */ |
ryanyuyu | 7:fc55813f823e | 150 | |
ryanyuyu | 2:8ae58834937f | 151 | |
ryanyuyu | 4:9ee3ae61db7f | 152 | /* |
ryanyuyu | 4:9ee3ae61db7f | 153 | This is a helper function for precision timing of Tickers |
ryanyuyu | 4:9ee3ae61db7f | 154 | */ |
ryanyuyu | 2:8ae58834937f | 155 | void readPoint() |
ryanyuyu | 2:8ae58834937f | 156 | { |
ryanyuyu | 2:8ae58834937f | 157 | waitForNext = false; |
ryanyuyu | 2:8ae58834937f | 158 | } |
ryanyuyu | 2:8ae58834937f | 159 | |
ryanyuyu | 2:8ae58834937f | 160 | |
ryanyuyu | 2:8ae58834937f | 161 | /** |
ryanyuyu | 2:8ae58834937f | 162 | * This function reads one full set of analog data into the uC |
ryanyuyu | 2:8ae58834937f | 163 | */ |
ryanyuyu | 2:8ae58834937f | 164 | void readSamples() |
ryanyuyu | 2:8ae58834937f | 165 | { |
ryanyuyu | 2:8ae58834937f | 166 | Ticker sampler; //allows for precision data reading |
ryanyuyu | 2:8ae58834937f | 167 | waitForNext = true; |
ryanyuyu | 2:8ae58834937f | 168 | sampler.attach_us(&readPoint, (int) (1000000/f_sampling) ); //read in data according the sampling freq |
ryanyuyu | 2:8ae58834937f | 169 | for (int i = 0; i < BlockSize; i++) |
ryanyuyu | 2:8ae58834937f | 170 | { |
ryanyuyu | 2:8ae58834937f | 171 | while (waitForNext); //wait until the ticker calls for the next sample |
ryanyuyu | 2:8ae58834937f | 172 | waveform[i] = input.read(); |
ryanyuyu | 2:8ae58834937f | 173 | waitForNext = true; |
ryanyuyu | 2:8ae58834937f | 174 | } |
ryanyuyu | 2:8ae58834937f | 175 | sampler.detach(); |
ryanyuyu | 2:8ae58834937f | 176 | } |
ryanyuyu | 3:30dcfcf9412c | 177 | |
ryanyuyu | 4:9ee3ae61db7f | 178 | /** |
ryanyuyu | 4:9ee3ae61db7f | 179 | This function spits out the waveform on the analogOut pin (p18) |
ryanyuyu | 4:9ee3ae61db7f | 180 | This function will be unused in the final version, but is still usefull for debugging. |
ryanyuyu | 4:9ee3ae61db7f | 181 | @param array (float32_t *): (array of data) pointer to the data to output over the analogOut pin |
ryanyuyu | 4:9ee3ae61db7f | 182 | @return none |
ryanyuyu | 4:9ee3ae61db7f | 183 | */ |
ryanyuyu | 4:9ee3ae61db7f | 184 | void outputWaveform(float32_t* array) |
ryanyuyu | 3:30dcfcf9412c | 185 | { |
ryanyuyu | 3:30dcfcf9412c | 186 | Ticker outputter; |
ryanyuyu | 3:30dcfcf9412c | 187 | waitForNext = true; |
ryanyuyu | 3:30dcfcf9412c | 188 | outputter.attach_us(&readPoint, (int) (1000000/f_sampling) ); //output data according the sampling freq |
ryanyuyu | 3:30dcfcf9412c | 189 | for (int i = 0; i < BlockSize; i++) |
ryanyuyu | 3:30dcfcf9412c | 190 | { |
ryanyuyu | 3:30dcfcf9412c | 191 | while (waitForNext); //wait until the ticker calls for the next data point |
ryanyuyu | 4:9ee3ae61db7f | 192 | waveOut.write(array[i]); |
ryanyuyu | 3:30dcfcf9412c | 193 | waitForNext = true; |
ryanyuyu | 3:30dcfcf9412c | 194 | } |
ryanyuyu | 3:30dcfcf9412c | 195 | outputter.detach(); |
ryanyuyu | 3:30dcfcf9412c | 196 | } |
ryanyuyu | 3:30dcfcf9412c | 197 | |
ryanyuyu | 4:9ee3ae61db7f | 198 | /* |
ryanyuyu | 4:9ee3ae61db7f | 199 | This method writes to the digital potentiometer (MCP4251) |
ryanyuyu | 4:9ee3ae61db7f | 200 | @param wiperNo (int): this is the wiper number to write to (either 0 or 1) |
ryanyuyu | 4:9ee3ae61db7f | 201 | @param kOhms (float): this is the value to set the resistance (in kilo Ohms) between the wiper and terminal B |
ryanyuyu | 4:9ee3ae61db7f | 202 | note |
ryanyuyu | 6:8441a6864784 | 203 | @return: the integer command actually sent (for debugging) |
ryanyuyu | 4:9ee3ae61db7f | 204 | */ |
ryanyuyu | 4:9ee3ae61db7f | 205 | int setPot(int wiperNo, float kOhms) |
ryanyuyu | 3:30dcfcf9412c | 206 | { |
ryanyuyu | 3:30dcfcf9412c | 207 | //257 steps (8 bits + 1), see section 7.0 for SPI instructions |
ryanyuyu | 3:30dcfcf9412c | 208 | float Rmax = 100000; |
ryanyuyu | 3:30dcfcf9412c | 209 | spi.frequency(2000000); |
ryanyuyu | 3:30dcfcf9412c | 210 | spi.format(16, 0); //16 bits, mode b00 |
ryanyuyu | 3:30dcfcf9412c | 211 | float ratio = kOhms * 1000.0 / Rmax; |
ryanyuyu | 3:30dcfcf9412c | 212 | if (ratio > 1) ratio = 1; |
ryanyuyu | 3:30dcfcf9412c | 213 | if (ratio < 0) ratio = 0; |
ryanyuyu | 3:30dcfcf9412c | 214 | int dataBits = (int) (ratio * 0x100); |
ryanyuyu | 3:30dcfcf9412c | 215 | int command = wiperNo << 12; //setting the Address and Command bits |
ryanyuyu | 4:9ee3ae61db7f | 216 | command += dataBits; //add in the data bits (digital settings) |
ryanyuyu | 3:30dcfcf9412c | 217 | spi.write(command); |
ryanyuyu | 3:30dcfcf9412c | 218 | return command; |
ryanyuyu | 3:30dcfcf9412c | 219 | } |
ryanyuyu | 3:30dcfcf9412c | 220 | |
ryanyuyu | 6:8441a6864784 | 221 | /** |
ryanyuyu | 6:8441a6864784 | 222 | This function uses both sides of the digital pot to produce an overall gain for the circuit. It uses side1 (post filter) before side0 (prefilter) |
ryanyuyu | 6:8441a6864784 | 223 | @param gain (float): the overall gain wanted (bound by [1, 10000] inclusive) |
ryanyuyu | 6:8441a6864784 | 224 | */ |
ryanyuyu | 6:8441a6864784 | 225 | void setGain(float gain) |
ryanyuyu | 6:8441a6864784 | 226 | { |
ryanyuyu | 6:8441a6864784 | 227 | if (gain < 0) return; |
ryanyuyu | 6:8441a6864784 | 228 | if (gain <= 100.0) //only side1 is used |
ryanyuyu | 6:8441a6864784 | 229 | { |
ryanyuyu | 6:8441a6864784 | 230 | setPot(0, 1.0); |
ryanyuyu | 6:8441a6864784 | 231 | setPot(1, gain); |
ryanyuyu | 6:8441a6864784 | 232 | } |
ryanyuyu | 6:8441a6864784 | 233 | else if (gain <= 10000) |
ryanyuyu | 6:8441a6864784 | 234 | { |
ryanyuyu | 6:8441a6864784 | 235 | setPot(1, 100.0); |
ryanyuyu | 6:8441a6864784 | 236 | setPot(0, gain / 100.0); |
ryanyuyu | 6:8441a6864784 | 237 | } |
ryanyuyu | 6:8441a6864784 | 238 | else |
ryanyuyu | 6:8441a6864784 | 239 | { |
ryanyuyu | 6:8441a6864784 | 240 | setPot(1, 100.0); |
ryanyuyu | 6:8441a6864784 | 241 | setPot(0, gain / 100.0); |
ryanyuyu | 6:8441a6864784 | 242 | } |
ryanyuyu | 6:8441a6864784 | 243 | } |
ryanyuyu | 6:8441a6864784 | 244 | |
ryanyuyu | 4:9ee3ae61db7f | 245 | /* |
ryanyuyu | 4:9ee3ae61db7f | 246 | This function calculates the RMS (root mean squared) of an array of float data. |
ryanyuyu | 4:9ee3ae61db7f | 247 | @param array (float32_t *): the array to calculate RMS from |
ryanyuyu | 4:9ee3ae61db7f | 248 | @return float_32: the resulting RMS value of the given array |
ryanyuyu | 4:9ee3ae61db7f | 249 | */ |
ryanyuyu | 4:9ee3ae61db7f | 250 | float32_t rms(float32_t* array) |
ryanyuyu | 3:30dcfcf9412c | 251 | { |
ryanyuyu | 3:30dcfcf9412c | 252 | float32_t rms = 0; |
ryanyuyu | 3:30dcfcf9412c | 253 | for(int i = 0; i < BlockSize; i++) |
ryanyuyu | 3:30dcfcf9412c | 254 | { |
ryanyuyu | 4:9ee3ae61db7f | 255 | rms += array[i]*array[i]; |
ryanyuyu | 3:30dcfcf9412c | 256 | } |
ryanyuyu | 4:9ee3ae61db7f | 257 | //pc.printf("Sum of squares %f\n\r", rms); |
ryanyuyu | 3:30dcfcf9412c | 258 | return sqrt(rms/BlockSize); |
ryanyuyu | 3:30dcfcf9412c | 259 | } |
ryanyuyu | 4:9ee3ae61db7f | 260 | |
ryanyuyu | 6:8441a6864784 | 261 | |
ryanyuyu | 6:8441a6864784 | 262 | /** |
ryanyuyu | 6:8441a6864784 | 263 | This function will wait for a button press. It will work 250ms after being called (to reduce double reads) |
ryanyuyu | 6:8441a6864784 | 264 | */ |
ryanyuyu | 6:8441a6864784 | 265 | void waitForButton() |
ryanyuyu | 4:9ee3ae61db7f | 266 | { |
ryanyuyu | 6:8441a6864784 | 267 | wait_ms(250); //to ward off double reads or sticky buttons |
ryanyuyu | 6:8441a6864784 | 268 | while(button.read() == 0) wait_ms(10); //poll button press every 10ms |
ryanyuyu | 6:8441a6864784 | 269 | //char* outputString = newString(32); |
ryanyuyu | 6:8441a6864784 | 270 | //outputString = "Button pressed."; |
ryanyuyu | 6:8441a6864784 | 271 | lcd.print("Button pressed."); |
ryanyuyu | 6:8441a6864784 | 272 | } |
ryanyuyu | 6:8441a6864784 | 273 | |
ryanyuyu | 6:8441a6864784 | 274 | |
ryanyuyu | 6:8441a6864784 | 275 | /** |
ryanyuyu | 6:8441a6864784 | 276 | This function takes RMS voltage and estimates the distance using linear interpolations. |
ryanyuyu | 6:8441a6864784 | 277 | Each gain stage is split into a 2-piece-wise linear funtion for estimation |
ryanyuyu | 6:8441a6864784 | 278 | @param value (float): the post-filter RMS value |
ryanyuyu | 6:8441a6864784 | 279 | @return (float): the distance estimate in inches (6 to 84) assuming perfect alignment, or special: |
ryanyuyu | 6:8441a6864784 | 280 | Special cases: |
ryanyuyu | 6:8441a6864784 | 281 | -1: clipping likely, too close adjust to a lower gain stage |
ryanyuyu | 6:8441a6864784 | 282 | 999: cannot detect signal (too far), adjust to higher gain stage |
ryanyuyu | 6:8441a6864784 | 283 | */ |
ryanyuyu | 6:8441a6864784 | 284 | float estimateDistance(float value) |
ryanyuyu | 6:8441a6864784 | 285 | { |
ryanyuyu | 6:8441a6864784 | 286 | //if outside range, then alert to try to adjust the gain settings |
ryanyuyu | 6:8441a6864784 | 287 | if (value < minThreshold*1.10) return 999; |
ryanyuyu | 6:8441a6864784 | 288 | if (value > maxThreshold*.97) return -1; |
ryanyuyu | 6:8441a6864784 | 289 | |
ryanyuyu | 6:8441a6864784 | 290 | switch (gainStage) |
ryanyuyu | 6:8441a6864784 | 291 | { |
ryanyuyu | 6:8441a6864784 | 292 | case 0: |
ryanyuyu | 6:8441a6864784 | 293 | if (value > mid[0]) return mLower[0]*value + bLower[0]; |
ryanyuyu | 6:8441a6864784 | 294 | else return mUpper[0]*value + bUpper[0]; |
ryanyuyu | 6:8441a6864784 | 295 | case 1: |
ryanyuyu | 6:8441a6864784 | 296 | if (value > mid[1]) return mLower[1]*value + bLower[1]; |
ryanyuyu | 6:8441a6864784 | 297 | else return mUpper[1]*value + bUpper[1]; |
ryanyuyu | 6:8441a6864784 | 298 | /* |
ryanyuyu | 6:8441a6864784 | 299 | case 2: |
ryanyuyu | 6:8441a6864784 | 300 | if (value > mid[2]) return mLower[2]*value + bLower[2]; |
ryanyuyu | 6:8441a6864784 | 301 | else return mUpper[2]*value + bUpper[2]; |
ryanyuyu | 6:8441a6864784 | 302 | case 3: |
ryanyuyu | 6:8441a6864784 | 303 | if (value > mid[3]) return mLower[3]*value + bLower[3]; |
ryanyuyu | 6:8441a6864784 | 304 | else return mUpper[3]*value + bUpper[3]; |
ryanyuyu | 6:8441a6864784 | 305 | */ |
ryanyuyu | 6:8441a6864784 | 306 | default: |
ryanyuyu | 6:8441a6864784 | 307 | return 0; |
ryanyuyu | 6:8441a6864784 | 308 | } |
ryanyuyu | 4:9ee3ae61db7f | 309 | } |
ryanyuyu | 4:9ee3ae61db7f | 310 | |
ryanyuyu | 6:8441a6864784 | 311 | /** |
ryanyuyu | 6:8441a6864784 | 312 | This function takes in a distance estimate and tries to change the gain stage |
ryanyuyu | 6:8441a6864784 | 313 | @param distance (float): the RMS from estimateDistance |
ryanyuyu | 6:8441a6864784 | 314 | */ |
ryanyuyu | 6:8441a6864784 | 315 | void adjustGains(float distance) |
ryanyuyu | 4:9ee3ae61db7f | 316 | { |
ryanyuyu | 7:fc55813f823e | 317 | //pc.printf("GainStage = %d Distance=%f\n\r", gainStage, distance); |
ryanyuyu | 6:8441a6864784 | 318 | if (distance == -1) //the special case for clipping |
ryanyuyu | 4:9ee3ae61db7f | 319 | { |
ryanyuyu | 7:fc55813f823e | 320 | //pc.printf(" Too close\n\r"); |
ryanyuyu | 7:fc55813f823e | 321 | if (gainStage > 0) gainStage--; |
ryanyuyu | 7:fc55813f823e | 322 | else lcd.print("Clipping. Back up."); |
ryanyuyu | 4:9ee3ae61db7f | 323 | } |
ryanyuyu | 7:fc55813f823e | 324 | else if (distance == 999) //the special case for being too far |
ryanyuyu | 4:9ee3ae61db7f | 325 | { |
ryanyuyu | 7:fc55813f823e | 326 | //pc.printf(" Too far.\n\r"); |
ryanyuyu | 7:fc55813f823e | 327 | if (gainStage < numGainStages) gainStage++; |
ryanyuyu | 7:fc55813f823e | 328 | else lcd.print("No beacon found."); |
ryanyuyu | 4:9ee3ae61db7f | 329 | } |
ryanyuyu | 7:fc55813f823e | 330 | else lcd.print(""); |
ryanyuyu | 6:8441a6864784 | 331 | setGain( gainCutoffs[gainStage] ); |
ryanyuyu | 6:8441a6864784 | 332 | //return gainStage; |
ryanyuyu | 4:9ee3ae61db7f | 333 | } |
ryanyuyu | 4:9ee3ae61db7f | 334 | |
ryanyuyu | 6:8441a6864784 | 335 | void enforceGainStage() |
ryanyuyu | 6:8441a6864784 | 336 | { |
ryanyuyu | 6:8441a6864784 | 337 | setGain( gainCutoffs[gainStage] * gainMultiplier); |
ryanyuyu | 6:8441a6864784 | 338 | } |
ryanyuyu | 6:8441a6864784 | 339 | |
ryanyuyu | 6:8441a6864784 | 340 | /** |
ryanyuyu | 6:8441a6864784 | 341 | This function takes one point of callibration data. |
ryanyuyu | 6:8441a6864784 | 342 | */ |
ryanyuyu | 6:8441a6864784 | 343 | void callibratePoint(float value) |
ryanyuyu | 4:9ee3ae61db7f | 344 | { |
ryanyuyu | 6:8441a6864784 | 345 | if (adjusting) |
ryanyuyu | 6:8441a6864784 | 346 | { |
ryanyuyu | 6:8441a6864784 | 347 | gainStage = (callibrationStep) / 3; |
ryanyuyu | 7:fc55813f823e | 348 | snprintf(outputString, 32, "%i in.", callibrationPoints[ callibrationStep-1 ]); |
ryanyuyu | 6:8441a6864784 | 349 | lcd.calibrationdist(outputString); |
ryanyuyu | 6:8441a6864784 | 350 | waitForButton(); |
ryanyuyu | 6:8441a6864784 | 351 | adjusting = false; |
ryanyuyu | 6:8441a6864784 | 352 | state = 2; |
ryanyuyu | 6:8441a6864784 | 353 | } |
ryanyuyu | 6:8441a6864784 | 354 | else |
ryanyuyu | 6:8441a6864784 | 355 | { |
ryanyuyu | 6:8441a6864784 | 356 | enforceGainStage(); |
ryanyuyu | 6:8441a6864784 | 357 | linearSamples[ callibrationStep - 1] = value; |
ryanyuyu | 6:8441a6864784 | 358 | callibrationStep++; //move to next callibration step |
ryanyuyu | 6:8441a6864784 | 359 | //get ready for next callibration step |
ryanyuyu | 6:8441a6864784 | 360 | adjusting = true; |
ryanyuyu | 6:8441a6864784 | 361 | state = 1; |
ryanyuyu | 6:8441a6864784 | 362 | } |
ryanyuyu | 4:9ee3ae61db7f | 363 | } |
ryanyuyu | 0:3aae5d23d0db | 364 | |
ryanyuyu | 0:3aae5d23d0db | 365 | int main() { |
ryanyuyu | 4:9ee3ae61db7f | 366 | //arm_iir_lattice_instance_f32* filter1 = new arm_iir_lattice_instance_f32(); |
ryanyuyu | 4:9ee3ae61db7f | 367 | arm_fir_instance_f32* filter = new arm_fir_instance_f32(); |
ryanyuyu | 6:8441a6864784 | 368 | float history[10]; //history of RMS voltages. |
ryanyuyu | 6:8441a6864784 | 369 | |
ryanyuyu | 6:8441a6864784 | 370 | state = 0; //which state of the state machine to be in, change to enum if desired |
ryanyuyu | 2:8ae58834937f | 371 | |
ryanyuyu | 2:8ae58834937f | 372 | uint16_t numTaps = NumTaps; |
ryanyuyu | 2:8ae58834937f | 373 | uint32_t blockSize = BlockSize; |
ryanyuyu | 3:30dcfcf9412c | 374 | char buffer[32]; //for debugging scanf things |
ryanyuyu | 6:8441a6864784 | 375 | //char* outputString = newString(30); //string to be printed to the LCD display (or other output) |
ryanyuyu | 6:8441a6864784 | 376 | //char* strength = newString(32); |
ryanyuyu | 6:8441a6864784 | 377 | //char* dist = newString(32); |
ryanyuyu | 7:fc55813f823e | 378 | float average = 0; |
ryanyuyu | 3:30dcfcf9412c | 379 | float32_t estimate = 0; |
ryanyuyu | 6:8441a6864784 | 380 | float RMS = 0; |
ryanyuyu | 6:8441a6864784 | 381 | int index_h = 0; |
ryanyuyu | 2:8ae58834937f | 382 | while(1) |
ryanyuyu | 2:8ae58834937f | 383 | { |
ryanyuyu | 2:8ae58834937f | 384 | switch(state) |
ryanyuyu | 2:8ae58834937f | 385 | { |
ryanyuyu | 2:8ae58834937f | 386 | case 0: //initialization |
ryanyuyu | 5:bc45ed158abf | 387 | for (int i = 0; i < NumTaps; i++) |
ryanyuyu | 5:bc45ed158abf | 388 | { |
ryanyuyu | 6:8441a6864784 | 389 | pCoeffs[i] *= 1.70; |
ryanyuyu | 5:bc45ed158abf | 390 | } |
ryanyuyu | 5:bc45ed158abf | 391 | |
ryanyuyu | 2:8ae58834937f | 392 | arm_fir_init_f32(filter, numTaps, pCoeffs, pState, blockSize); |
ryanyuyu | 4:9ee3ae61db7f | 393 | //arm_iir_lattice_init_f32(filter1, numTaps, pkCoeffs, pvCoeffs, pState, blockSize); |
ryanyuyu | 2:8ae58834937f | 394 | //pc.printf("Pre-attachment"); |
ryanyuyu | 3:30dcfcf9412c | 395 | spi.frequency(1000000); |
ryanyuyu | 2:8ae58834937f | 396 | state = 1; |
ryanyuyu | 6:8441a6864784 | 397 | callibrationStep = 0; |
ryanyuyu | 6:8441a6864784 | 398 | gainStage = 0; |
ryanyuyu | 6:8441a6864784 | 399 | gainMultiplier = 1.0; |
ryanyuyu | 7:fc55813f823e | 400 | //pc.printf("Done with init.\n\r"); |
ryanyuyu | 2:8ae58834937f | 401 | break; |
ryanyuyu | 2:8ae58834937f | 402 | |
ryanyuyu | 6:8441a6864784 | 403 | case 1: //callibration |
ryanyuyu | 7:fc55813f823e | 404 | //pc.printf(" Callibration step: %i\n\r", callibrationStep); |
ryanyuyu | 6:8441a6864784 | 405 | if (callibrationStep == 0) //calculate the offset (beacon is off, or at infinity) |
ryanyuyu | 6:8441a6864784 | 406 | { |
ryanyuyu | 6:8441a6864784 | 407 | setGain( gainCutoffs[0]); |
ryanyuyu | 6:8441a6864784 | 408 | if (adjusting) |
ryanyuyu | 6:8441a6864784 | 409 | { |
ryanyuyu | 6:8441a6864784 | 410 | //outputString = "Turn off the beacon. Press the button when done."; |
ryanyuyu | 6:8441a6864784 | 411 | lcd.calibrationunl(); |
ryanyuyu | 6:8441a6864784 | 412 | waitForButton(); |
ryanyuyu | 6:8441a6864784 | 413 | adjusting = false; |
ryanyuyu | 6:8441a6864784 | 414 | state = 2; |
ryanyuyu | 6:8441a6864784 | 415 | } |
ryanyuyu | 6:8441a6864784 | 416 | else |
ryanyuyu | 6:8441a6864784 | 417 | { |
ryanyuyu | 6:8441a6864784 | 418 | |
ryanyuyu | 6:8441a6864784 | 419 | minThreshold = average; //the average RMS of background noise |
ryanyuyu | 6:8441a6864784 | 420 | maxThreshold = .400; |
ryanyuyu | 6:8441a6864784 | 421 | callibrationStep = 1; //move to next callibration step |
ryanyuyu | 6:8441a6864784 | 422 | //get ready for next callibration step |
ryanyuyu | 6:8441a6864784 | 423 | adjusting = true; |
ryanyuyu | 6:8441a6864784 | 424 | state = 1; |
ryanyuyu | 6:8441a6864784 | 425 | } |
ryanyuyu | 6:8441a6864784 | 426 | } |
ryanyuyu | 6:8441a6864784 | 427 | else if (callibrationStep == 1) //at 6in, adjust gain scaling and take one datapoint |
ryanyuyu | 6:8441a6864784 | 428 | { |
ryanyuyu | 6:8441a6864784 | 429 | /* |
ryanyuyu | 6:8441a6864784 | 430 | gain1 = 20.0; |
ryanyuyu | 6:8441a6864784 | 431 | gain0 = 1.0; |
ryanyuyu | 6:8441a6864784 | 432 | setPot(1, gain1); |
ryanyuyu | 6:8441a6864784 | 433 | setPot(0, gain0); |
ryanyuyu | 6:8441a6864784 | 434 | */ |
ryanyuyu | 6:8441a6864784 | 435 | setGain( 20 ); |
ryanyuyu | 6:8441a6864784 | 436 | callibratePoint(average); |
ryanyuyu | 6:8441a6864784 | 437 | } |
ryanyuyu | 6:8441a6864784 | 438 | else if (callibrationStep <= numCallibrationSteps) |
ryanyuyu | 6:8441a6864784 | 439 | { |
ryanyuyu | 6:8441a6864784 | 440 | callibratePoint(average); |
ryanyuyu | 6:8441a6864784 | 441 | } |
ryanyuyu | 6:8441a6864784 | 442 | else //now all the points are captured, so create the coeffs |
ryanyuyu | 6:8441a6864784 | 443 | { |
ryanyuyu | 7:fc55813f823e | 444 | //pc.printf("calculating coeffs\n\r"); |
ryanyuyu | 6:8441a6864784 | 445 | for (int i = 0; i < numGainStages; i++) |
ryanyuyu | 6:8441a6864784 | 446 | { |
ryanyuyu | 6:8441a6864784 | 447 | mid[i] = linearSamples[i*3+1]; |
ryanyuyu | 6:8441a6864784 | 448 | mLower[i] = (callibrationPoints[i*3+1] - callibrationPoints[i*3+0]) / (linearSamples[i*3+1] - linearSamples[i*3+0]) ; |
ryanyuyu | 6:8441a6864784 | 449 | mUpper[i] = (callibrationPoints[i*3+2] - callibrationPoints[i*3+1]) / (linearSamples[i*3+2] - linearSamples[i*3+1]) ; |
ryanyuyu | 6:8441a6864784 | 450 | bLower[i] = callibrationPoints[i*3+0] - mLower[i]*linearSamples[i*3+0]; |
ryanyuyu | 6:8441a6864784 | 451 | bUpper[i] = callibrationPoints[i*3+1] - mUpper[i]*linearSamples[i*3+1]; |
ryanyuyu | 7:fc55813f823e | 452 | //pc.printf("mL=%f mU=%f bL=%f, bU=%f, mid=%f\n\r", mLower[i], mUpper[i], bLower[i], bUpper[i], mid[i]); |
ryanyuyu | 6:8441a6864784 | 453 | } |
ryanyuyu | 6:8441a6864784 | 454 | callibrationStep = -1; |
ryanyuyu | 6:8441a6864784 | 455 | state = 2; |
ryanyuyu | 6:8441a6864784 | 456 | gainStage = 0; |
ryanyuyu | 6:8441a6864784 | 457 | |
ryanyuyu | 6:8441a6864784 | 458 | for (int i = 0; i < numCallibrationSteps; i++) |
ryanyuyu | 6:8441a6864784 | 459 | { |
ryanyuyu | 7:fc55813f823e | 460 | //pc.printf("linear(x)=%f callibration(y)=%d \n\r", linearSamples[i], callibrationPoints[i]); |
ryanyuyu | 6:8441a6864784 | 461 | } |
ryanyuyu | 7:fc55813f823e | 462 | //pc.printf("End of callibration.\n\r"); |
ryanyuyu | 6:8441a6864784 | 463 | } |
ryanyuyu | 6:8441a6864784 | 464 | |
ryanyuyu | 6:8441a6864784 | 465 | case 2: //read data, take samples |
ryanyuyu | 5:bc45ed158abf | 466 | //pc.printf("Reading data.\n\r"); |
ryanyuyu | 2:8ae58834937f | 467 | readSamples(); |
ryanyuyu | 3:30dcfcf9412c | 468 | state = 3; |
ryanyuyu | 2:8ae58834937f | 469 | break; |
ryanyuyu | 2:8ae58834937f | 470 | case 3: //filter? |
ryanyuyu | 6:8441a6864784 | 471 | //pc.printf("RMS of waveform = %f\n\r", rms(waveform)); |
ryanyuyu | 6:8441a6864784 | 472 | //pc.printf("Filtering?\n\r"); |
ryanyuyu | 2:8ae58834937f | 473 | arm_fir_f32(filter, waveform, postFilterData, blockSize); |
ryanyuyu | 4:9ee3ae61db7f | 474 | //arm_iir_lattice_f32(filter1, waveform, postFilterData, blockSize); |
ryanyuyu | 6:8441a6864784 | 475 | RMS = rms(postFilterData); |
ryanyuyu | 6:8441a6864784 | 476 | estimate = estimateDistance(RMS); |
ryanyuyu | 6:8441a6864784 | 477 | if (callibrationStep == -1) state = 6; //done with callibration |
ryanyuyu | 6:8441a6864784 | 478 | else state = 7; //still callibrating |
ryanyuyu | 2:8ae58834937f | 479 | break; |
ryanyuyu | 2:8ae58834937f | 480 | case 4: //FFT? |
ryanyuyu | 2:8ae58834937f | 481 | break; |
ryanyuyu | 2:8ae58834937f | 482 | case 5: //output, write to display and PWM tone |
ryanyuyu | 4:9ee3ae61db7f | 483 | /* |
ryanyuyu | 3:30dcfcf9412c | 484 | sprintf(outputString, "RMS = %f", estimate); |
ryanyuyu | 3:30dcfcf9412c | 485 | lcd.print(outputString); |
ryanyuyu | 3:30dcfcf9412c | 486 | state = 1; |
ryanyuyu | 4:9ee3ae61db7f | 487 | //*/ |
ryanyuyu | 2:8ae58834937f | 488 | break; |
ryanyuyu | 4:9ee3ae61db7f | 489 | case 6: //calculate the average voltage |
ryanyuyu | 6:8441a6864784 | 490 | //pc.printf("post filter RMS = %f\n\n\r", estimate); |
ryanyuyu | 4:9ee3ae61db7f | 491 | adjustGains(estimate); |
ryanyuyu | 6:8441a6864784 | 492 | state = 8; |
ryanyuyu | 6:8441a6864784 | 493 | break; |
ryanyuyu | 6:8441a6864784 | 494 | case 7: //callibration-related, take 10pt average and record it |
ryanyuyu | 6:8441a6864784 | 495 | history[index_h] = RMS; |
ryanyuyu | 6:8441a6864784 | 496 | index_h++; |
ryanyuyu | 6:8441a6864784 | 497 | state = 2; |
ryanyuyu | 6:8441a6864784 | 498 | if (index_h >= 10) //ten-pt average done |
ryanyuyu | 6:8441a6864784 | 499 | { |
ryanyuyu | 6:8441a6864784 | 500 | average = 0; |
ryanyuyu | 6:8441a6864784 | 501 | for (int i = 0; i < 10; i++) average+= history[i]; |
ryanyuyu | 6:8441a6864784 | 502 | average /= 10; |
ryanyuyu | 6:8441a6864784 | 503 | //pc.printf("10-pt average of RMS = %f\n\r", average); |
ryanyuyu | 6:8441a6864784 | 504 | float t = (float) average; |
ryanyuyu | 7:fc55813f823e | 505 | int n = snprintf(strength, 32," %f", t); |
ryanyuyu | 6:8441a6864784 | 506 | lcd.displayStr(strength); |
ryanyuyu | 6:8441a6864784 | 507 | index_h = 0; |
ryanyuyu | 6:8441a6864784 | 508 | state = 1; //go back to callibration |
ryanyuyu | 6:8441a6864784 | 509 | } |
ryanyuyu | 6:8441a6864784 | 510 | //state is 2, unless 10pts are collected, then state is 1 |
ryanyuyu | 6:8441a6864784 | 511 | //continue taking and filtering data until full of 10pts |
ryanyuyu | 6:8441a6864784 | 512 | break; |
ryanyuyu | 6:8441a6864784 | 513 | case 8: //output |
ryanyuyu | 6:8441a6864784 | 514 | //int n = sprintf(outputString, "RMS = %f, distance = %fin", RMS, estimate); |
ryanyuyu | 7:fc55813f823e | 515 | //pc.printf(" RMS=%f, Dist=%f GainStage=%d\n\r", RMS, estimate, gainStage); |
ryanyuyu | 7:fc55813f823e | 516 | //strcpy( strength, (convert(RMS) + "\0").c_str() ); |
ryanyuyu | 7:fc55813f823e | 517 | |
ryanyuyu | 7:fc55813f823e | 518 | |
ryanyuyu | 7:fc55813f823e | 519 | //snprintf(strength, 32, " %f\0", RMS); |
ryanyuyu | 5:bc45ed158abf | 520 | /* |
ryanyuyu | 7:fc55813f823e | 521 | if (estimate == -1) dist = " Unknown (clipping)\0"; |
ryanyuyu | 7:fc55813f823e | 522 | else if (estimate == 999) dist = " Unknown (no sig)\0"; |
ryanyuyu | 7:fc55813f823e | 523 | else */ |
ryanyuyu | 7:fc55813f823e | 524 | //strcpy( dist, (convert(estimate) + "in\0").c_str() ); |
ryanyuyu | 7:fc55813f823e | 525 | snprintf(strength, 32, " %f\0", RMS); |
ryanyuyu | 7:fc55813f823e | 526 | snprintf(dist, 32, " %.1f in.\0", estimate); |
ryanyuyu | 7:fc55813f823e | 527 | //*/ |
ryanyuyu | 7:fc55813f823e | 528 | //pc.printf( strength); |
ryanyuyu | 7:fc55813f823e | 529 | //pc.printf( dist); |
ryanyuyu | 6:8441a6864784 | 530 | lcd.displayStr(strength); |
ryanyuyu | 6:8441a6864784 | 531 | lcd.displayDist(dist); |
ryanyuyu | 7:fc55813f823e | 532 | |
ryanyuyu | 7:fc55813f823e | 533 | /* |
ryanyuyu | 6:8441a6864784 | 534 | if (button == 1) state = 9; |
ryanyuyu | 6:8441a6864784 | 535 | else state = 2; |
ryanyuyu | 7:fc55813f823e | 536 | //*/ |
ryanyuyu | 7:fc55813f823e | 537 | state = 2; |
ryanyuyu | 7:fc55813f823e | 538 | //pc.printf(" end of display\n\r"); |
ryanyuyu | 3:30dcfcf9412c | 539 | break; |
ryanyuyu | 6:8441a6864784 | 540 | case 9: //digital pot interfacing and calibration |
ryanyuyu | 6:8441a6864784 | 541 | pc.printf("Gain?\n\r"); |
ryanyuyu | 3:30dcfcf9412c | 542 | pc.scanf("%s", buffer); |
ryanyuyu | 3:30dcfcf9412c | 543 | float value = atof(buffer); |
ryanyuyu | 6:8441a6864784 | 544 | setGain(value); |
ryanyuyu | 6:8441a6864784 | 545 | //int side = (int) value; |
ryanyuyu | 6:8441a6864784 | 546 | //float k = (value - side) * 100; |
ryanyuyu | 6:8441a6864784 | 547 | //pc.printf("Command: %x Scanned:%d %f\n\r", setPot(side, k), side, k); |
ryanyuyu | 6:8441a6864784 | 548 | pc.printf("Scanned:%f\n\r", value); |
ryanyuyu | 6:8441a6864784 | 549 | //lcd.print("Press button to continue."); |
ryanyuyu | 6:8441a6864784 | 550 | //waitForButton(); |
ryanyuyu | 6:8441a6864784 | 551 | state = 2; |
ryanyuyu | 4:9ee3ae61db7f | 552 | break; |
ryanyuyu | 6:8441a6864784 | 553 | case 10: |
ryanyuyu | 4:9ee3ae61db7f | 554 | state = 10; |
ryanyuyu | 3:30dcfcf9412c | 555 | break; |
ryanyuyu | 2:8ae58834937f | 556 | default: |
ryanyuyu | 2:8ae58834937f | 557 | break; |
ryanyuyu | 2:8ae58834937f | 558 | } |
ryanyuyu | 2:8ae58834937f | 559 | } //end of (infinite) while loop |
ryanyuyu | 0:3aae5d23d0db | 560 | } |
ryanyuyu | 4:9ee3ae61db7f | 561 | |
ryanyuyu | 4:9ee3ae61db7f | 562 | |
ryanyuyu | 4:9ee3ae61db7f | 563 | //-----------------------------Unused code, but potentially useful |
ryanyuyu | 4:9ee3ae61db7f | 564 | |
ryanyuyu | 4:9ee3ae61db7f | 565 | /* |
ryanyuyu | 4:9ee3ae61db7f | 566 | double sum = 0; |
ryanyuyu | 4:9ee3ae61db7f | 567 | for (int i = 0; i < BlockSize; i++) sum += postFilterData[i]; |
ryanyuyu | 4:9ee3ae61db7f | 568 | double average = sum/BlockSize*3.3; //*3.3 V_ref (array stored as fractions of V_ref) |
ryanyuyu | 4:9ee3ae61db7f | 569 | pc.printf("Average = %f\n\r", average); |
ryanyuyu | 4:9ee3ae61db7f | 570 | wait_ms(500); |
ryanyuyu | 4:9ee3ae61db7f | 571 | state = 2; |
ryanyuyu | 4:9ee3ae61db7f | 572 | */ |
ryanyuyu | 4:9ee3ae61db7f | 573 | |
ryanyuyu | 4:9ee3ae61db7f | 574 | //pc.printf("into print\n\r"); |
ryanyuyu | 4:9ee3ae61db7f | 575 | /* |
ryanyuyu | 4:9ee3ae61db7f | 576 | for (int i = 0; i < BlockSize; i++) |
ryanyuyu | 4:9ee3ae61db7f | 577 | { |
ryanyuyu | 4:9ee3ae61db7f | 578 | pc.printf("Waveform contents:%f\n\r", waveform[i]); |
ryanyuyu | 4:9ee3ae61db7f | 579 | } |
ryanyuyu | 4:9ee3ae61db7f | 580 | */ |
ryanyuyu | 5:bc45ed158abf | 581 | |
ryanyuyu | 5:bc45ed158abf | 582 | |
ryanyuyu | 5:bc45ed158abf | 583 | |
ryanyuyu | 6:8441a6864784 | 584 | /*---------------peak detection |
ryanyuyu | 5:bc45ed158abf | 585 | pc.printf("Into estimation\n\r"); |
ryanyuyu | 5:bc45ed158abf | 586 | int peaks = 0; |
ryanyuyu | 5:bc45ed158abf | 587 | float sum = 0.0; |
ryanyuyu | 5:bc45ed158abf | 588 | float prev, current, next; |
ryanyuyu | 5:bc45ed158abf | 589 | for (int i = 0+1; i < BlockSize-1; i++) |
ryanyuyu | 5:bc45ed158abf | 590 | { |
ryanyuyu | 5:bc45ed158abf | 591 | prev = postFilterData[i-1]; |
ryanyuyu | 5:bc45ed158abf | 592 | current = postFilterData[i]; |
ryanyuyu | 5:bc45ed158abf | 593 | next = postFilterData[i+1]; |
ryanyuyu | 5:bc45ed158abf | 594 | if (prev < current && next < current) //local max |
ryanyuyu | 5:bc45ed158abf | 595 | { |
ryanyuyu | 5:bc45ed158abf | 596 | sum += current; |
ryanyuyu | 5:bc45ed158abf | 597 | peaks++; |
ryanyuyu | 5:bc45ed158abf | 598 | } |
ryanyuyu | 5:bc45ed158abf | 599 | } |
ryanyuyu | 5:bc45ed158abf | 600 | float average = sum/peaks; |
ryanyuyu | 5:bc45ed158abf | 601 | pc.printf("Average of peaks (scalar) = %f\n\r", average); |
ryanyuyu | 5:bc45ed158abf | 602 | state = 1; |
ryanyuyu | 5:bc45ed158abf | 603 | //*/ |
ryanyuyu | 6:8441a6864784 | 604 | |
ryanyuyu | 6:8441a6864784 | 605 | /*---------------------------//purely for testing that the digital potentiometer is working. |
ryanyuyu | 6:8441a6864784 | 606 | pc.printf("Start of digital pot loop.\n\r"); |
ryanyuyu | 6:8441a6864784 | 607 | setPot(1,0); |
ryanyuyu | 6:8441a6864784 | 608 | wait_ms(1000); |
ryanyuyu | 6:8441a6864784 | 609 | setPot(1,20); |
ryanyuyu | 6:8441a6864784 | 610 | wait_ms(1000); |
ryanyuyu | 6:8441a6864784 | 611 | setPot(1,40); |
ryanyuyu | 6:8441a6864784 | 612 | wait_ms(1000); |
ryanyuyu | 6:8441a6864784 | 613 | setPot(1,50); |
ryanyuyu | 6:8441a6864784 | 614 | wait_ms(1000); |
ryanyuyu | 6:8441a6864784 | 615 | setPot(1, 80); |
ryanyuyu | 6:8441a6864784 | 616 | wait_ms(1000); |
ryanyuyu | 6:8441a6864784 | 617 | setPot(1, 100); |
ryanyuyu | 6:8441a6864784 | 618 | wait_ms(1000); |
ryanyuyu | 6:8441a6864784 | 619 | */ |
ryanyuyu | 6:8441a6864784 | 620 | |
ryanyuyu | 6:8441a6864784 | 621 | /* |
ryanyuyu | 6:8441a6864784 | 622 | m00 = -15.221; |
ryanyuyu | 6:8441a6864784 | 623 | b00 = 10.836; |
ryanyuyu | 6:8441a6864784 | 624 | mid0 = .088; |
ryanyuyu | 6:8441a6864784 | 625 | m01 = -142.2; |
ryanyuyu | 6:8441a6864784 | 626 | b01 = 22.101; |
ryanyuyu | 6:8441a6864784 | 627 | |
ryanyuyu | 6:8441a6864784 | 628 | m10 = -48.639; |
ryanyuyu | 6:8441a6864784 | 629 | b10 = 22.128; |
ryanyuyu | 6:8441a6864784 | 630 | mid1 = .068; |
ryanyuyu | 6:8441a6864784 | 631 | m11 = -363.74; |
ryanyuyu | 6:8441a6864784 | 632 | b11 = 22.352; |
ryanyuyu | 6:8441a6864784 | 633 | |
ryanyuyu | 6:8441a6864784 | 634 | m20 = -45.513; |
ryanyuyu | 6:8441a6864784 | 635 | b20 = 39.895; |
ryanyuyu | 6:8441a6864784 | 636 | mid2 = .115; |
ryanyuyu | 6:8441a6864784 | 637 | m21 = -314.87; |
ryanyuyu | 6:8441a6864784 | 638 | b21 = 70.387; |
ryanyuyu | 6:8441a6864784 | 639 | |
ryanyuyu | 6:8441a6864784 | 640 | m30 = -81.809; |
ryanyuyu | 6:8441a6864784 | 641 | b30 = 76.868; |
ryanyuyu | 6:8441a6864784 | 642 | mid3 = .194; |
ryanyuyu | 6:8441a6864784 | 643 | m31 = -201.48; |
ryanyuyu | 6:8441a6864784 | 644 | b31 = 99.556; |
ryanyuyu | 7:fc55813f823e | 645 | |
ryanyuyu | 7:fc55813f823e | 646 | |
ryanyuyu | 7:fc55813f823e | 647 | /* |
ryanyuyu | 7:fc55813f823e | 648 | std::string convert(float number) |
ryanyuyu | 7:fc55813f823e | 649 | { |
ryanyuyu | 7:fc55813f823e | 650 | std::ostringstream buffer; |
ryanyuyu | 7:fc55813f823e | 651 | buffer<<number; |
ryanyuyu | 7:fc55813f823e | 652 | return buffer.str(); |
ryanyuyu | 7:fc55813f823e | 653 | } |
ryanyuyu | 7:fc55813f823e | 654 | |
ryanyuyu | 7:fc55813f823e | 655 | |
ryanyuyu | 7:fc55813f823e | 656 | |
ryanyuyu | 7:fc55813f823e | 657 | |
ryanyuyu | 7:fc55813f823e | 658 | void callibratePoint(float value) |
ryanyuyu | 7:fc55813f823e | 659 | { |
ryanyuyu | 7:fc55813f823e | 660 | if (callibrationStep%3 == 1) //this looks for the gain stage thresholds |
ryanyuyu | 7:fc55813f823e | 661 | { |
ryanyuyu | 7:fc55813f823e | 662 | if (adjusting) |
ryanyuyu | 7:fc55813f823e | 663 | { |
ryanyuyu | 7:fc55813f823e | 664 | //outputString = "Turn off the beacon. Press the button when done."; |
ryanyuyu | 7:fc55813f823e | 665 | snprintf(outputString, 32, "%i in.", callibrationPoints[ callibrationStep-1 ]); |
ryanyuyu | 7:fc55813f823e | 666 | lcd.calibrationdist(outputString); |
ryanyuyu | 7:fc55813f823e | 667 | waitForButton(); |
ryanyuyu | 7:fc55813f823e | 668 | adjusting = false; |
ryanyuyu | 7:fc55813f823e | 669 | state = 2; |
ryanyuyu | 7:fc55813f823e | 670 | gainMultiplier = .5; |
ryanyuyu | 7:fc55813f823e | 671 | enforceGainStage(); |
ryanyuyu | 7:fc55813f823e | 672 | } |
ryanyuyu | 7:fc55813f823e | 673 | else |
ryanyuyu | 7:fc55813f823e | 674 | { |
ryanyuyu | 7:fc55813f823e | 675 | //pc.printf("Av=%f\n\r", gainMultiplier); |
ryanyuyu | 7:fc55813f823e | 676 | if (pastAverage*1.1 < average || pastAverage < minThreshold*3) //not yet maxed, so bump up gainMultiplier |
ryanyuyu | 7:fc55813f823e | 677 | { |
ryanyuyu | 7:fc55813f823e | 678 | gainMultiplier *= 1.3; |
ryanyuyu | 7:fc55813f823e | 679 | enforceGainStage(); |
ryanyuyu | 7:fc55813f823e | 680 | pastAverage = average; |
ryanyuyu | 7:fc55813f823e | 681 | callibrationStep = 1; //stay here until fining gainMultiplier |
ryanyuyu | 7:fc55813f823e | 682 | } |
ryanyuyu | 7:fc55813f823e | 683 | else //move onto next callibration step |
ryanyuyu | 7:fc55813f823e | 684 | { |
ryanyuyu | 7:fc55813f823e | 685 | callibrationStep++; |
ryanyuyu | 7:fc55813f823e | 686 | adjusting = true; |
ryanyuyu | 7:fc55813f823e | 687 | linearSamples[ gainStage ] = average; //record the intensity |
ryanyuyu | 7:fc55813f823e | 688 | maxThresholds[ gainStage ] = average; |
ryanyuyu | 7:fc55813f823e | 689 | gainCutoffs[ gainStage ] *= gainMultiplier; |
ryanyuyu | 7:fc55813f823e | 690 | pastAverage = minThreshold; |
ryanyuyu | 7:fc55813f823e | 691 | } |
ryanyuyu | 7:fc55813f823e | 692 | state = 1; |
ryanyuyu | 7:fc55813f823e | 693 | } |
ryanyuyu | 7:fc55813f823e | 694 | } |
ryanyuyu | 7:fc55813f823e | 695 | else // |
ryanyuyu | 7:fc55813f823e | 696 | { |
ryanyuyu | 7:fc55813f823e | 697 | if (adjusting) |
ryanyuyu | 7:fc55813f823e | 698 | { |
ryanyuyu | 7:fc55813f823e | 699 | gainStage = (callibrationStep) / 3; |
ryanyuyu | 7:fc55813f823e | 700 | snprintf(outputString, 32, "%i", callibrationPoints[ callibrationStep-1 ]); |
ryanyuyu | 7:fc55813f823e | 701 | lcd.calibrationdist(outputString); |
ryanyuyu | 7:fc55813f823e | 702 | waitForButton(); |
ryanyuyu | 7:fc55813f823e | 703 | adjusting = false; |
ryanyuyu | 7:fc55813f823e | 704 | state = 2; |
ryanyuyu | 7:fc55813f823e | 705 | } |
ryanyuyu | 7:fc55813f823e | 706 | else |
ryanyuyu | 7:fc55813f823e | 707 | { |
ryanyuyu | 7:fc55813f823e | 708 | enforceGainStage(); |
ryanyuyu | 7:fc55813f823e | 709 | linearSamples[ callibrationStep - 1] = value; |
ryanyuyu | 7:fc55813f823e | 710 | callibrationStep++; //move to next callibration step |
ryanyuyu | 7:fc55813f823e | 711 | //get ready for next callibration step |
ryanyuyu | 7:fc55813f823e | 712 | adjusting = true; |
ryanyuyu | 7:fc55813f823e | 713 | state = 1; |
ryanyuyu | 7:fc55813f823e | 714 | } |
ryanyuyu | 7:fc55813f823e | 715 | } |
ryanyuyu | 7:fc55813f823e | 716 | } |
ryanyuyu | 7:fc55813f823e | 717 | */ |