Important changes to repositories hosted on mbed.com
Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.
To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.
It is also possible to export all your personal repositories from the account settings page.
main.cpp@2:5ee1cbcf8e04, 2020-12-01 (annotated)
- Committer:
- markschwarzer
- Date:
- Tue Dec 01 15:06:41 2020 +0000
- Revision:
- 2:5ee1cbcf8e04
- Parent:
- 1:558ab0afbbc5
- Child:
- 3:a87f5671ea8b
Third commit
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
markschwarzer | 0:852fb667f2d6 | 1 | /* |
markschwarzer | 0:852fb667f2d6 | 2 | Author Mingxi Zhou |
markschwarzer | 0:852fb667f2d6 | 3 | OCE360 underwater float template program |
markschwarzer | 0:852fb667f2d6 | 4 | */ |
markschwarzer | 0:852fb667f2d6 | 5 | #include "mbed.h" |
markschwarzer | 0:852fb667f2d6 | 6 | #include "LSM9DS1.h" //IMU library |
markschwarzer | 0:852fb667f2d6 | 7 | #include "MS5837.h" //pressure sensor library |
markschwarzer | 0:852fb667f2d6 | 8 | #include "SCI_SENSOR.h" //science sensor |
markschwarzer | 0:852fb667f2d6 | 9 | #include "SDFileSystem.h" // SD card |
markschwarzer | 0:852fb667f2d6 | 10 | |
markschwarzer | 0:852fb667f2d6 | 11 | DigitalOut myled(LED1); |
markschwarzer | 0:852fb667f2d6 | 12 | Serial pc(USBTX, USBRX); //initial serial |
markschwarzer | 0:852fb667f2d6 | 13 | Serial BLE(p13,p14); //Bluetooth |
markschwarzer | 0:852fb667f2d6 | 14 | LSM9DS1 IMU(p28, p27, 0xD6, 0x3C); //initial IMU |
markschwarzer | 0:852fb667f2d6 | 15 | LM19 temp(p19); |
markschwarzer | 0:852fb667f2d6 | 16 | PhotoCell light(p20); |
markschwarzer | 0:852fb667f2d6 | 17 | MS5837 p_sensor(p9, p10, ms5837_addr_no_CS); //pressure sensor |
markschwarzer | 0:852fb667f2d6 | 18 | PwmOut thruster(p21); //set PWM pin //max 1.3ms min 1.1ms |
markschwarzer | 0:852fb667f2d6 | 19 | PwmOut thruster2(p22); //set PWM pin |
markschwarzer | 0:852fb667f2d6 | 20 | SDFileSystem sd(p5, p6, p7, p8, "sd"); // the pinout on the mbed Cool Components workshop board |
markschwarzer | 0:852fb667f2d6 | 21 | |
markschwarzer | 0:852fb667f2d6 | 22 | //global ticker |
markschwarzer | 0:852fb667f2d6 | 23 | Ticker log_ticker; |
markschwarzer | 0:852fb667f2d6 | 24 | Ticker imu_ticker; |
markschwarzer | 0:852fb667f2d6 | 25 | // global timer |
markschwarzer | 0:852fb667f2d6 | 26 | Timer t; |
markschwarzer | 0:852fb667f2d6 | 27 | //MS5837 p_sensor(p9, p10, ms5837_addr_no_CS); |
markschwarzer | 2:5ee1cbcf8e04 | 28 | ///File |
markschwarzer | 0:852fb667f2d6 | 29 | FILE *fp; |
markschwarzer | 0:852fb667f2d6 | 30 | char fname[100]; |
markschwarzer | 0:852fb667f2d6 | 31 | float PI = 3.14159265358979323846f; |
markschwarzer | 0:852fb667f2d6 | 32 | |
markschwarzer | 0:852fb667f2d6 | 33 | //float operation parameters |
markschwarzer | 0:852fb667f2d6 | 34 | float target_depth=0; //global target depth default 0 |
markschwarzer | 0:852fb667f2d6 | 35 | int yo_num=0; //global yo_num default 0 |
markschwarzer | 0:852fb667f2d6 | 36 | float thrust_on_time=0; //global thrust_on time default 0 |
markschwarzer | 0:852fb667f2d6 | 37 | float accel[3], mag[3], gyro[3], euler[3]; //global IMU data |
markschwarzer | 0:852fb667f2d6 | 38 | |
markschwarzer | 0:852fb667f2d6 | 39 | //functions |
markschwarzer | 0:852fb667f2d6 | 40 | void welcome(); |
markschwarzer | 0:852fb667f2d6 | 41 | void log_data(); |
markschwarzer | 0:852fb667f2d6 | 42 | //IMU related |
markschwarzer | 0:852fb667f2d6 | 43 | void IMU_update(); //update IMU related varibles. we use imu_ticker to call this function |
markschwarzer | 0:852fb667f2d6 | 44 | void mag_correction(float mx, float my, float mz, float mag_c[3]); //raw mag -> mag[3], mag_c[3] calibrated |
markschwarzer | 0:852fb667f2d6 | 45 | void pose_estimate(float euler[3], float accel[3], float gyro[3], float mag[3]); //pose estimation function |
markschwarzer | 0:852fb667f2d6 | 46 | //Control related functions |
markschwarzer | 0:852fb667f2d6 | 47 | void thrust_on(float pw, float on_time); //input is pulse width |
markschwarzer | 0:852fb667f2d6 | 48 | |
markschwarzer | 0:852fb667f2d6 | 49 | //-------------Main functions----------------------------------------------------------------------------------------- |
markschwarzer | 0:852fb667f2d6 | 50 | int main() |
markschwarzer | 0:852fb667f2d6 | 51 | { |
markschwarzer | 0:852fb667f2d6 | 52 | //-----Initialization realted code-------// |
markschwarzer | 0:852fb667f2d6 | 53 | //inital set the thruster esc to 1ms duty cycle |
markschwarzer | 0:852fb667f2d6 | 54 | thruster.period(0.002); // 2 ms period |
markschwarzer | 0:852fb667f2d6 | 55 | thruster.pulsewidth(1.0/1000.000); /////IMU initial and begin |
markschwarzer | 0:852fb667f2d6 | 56 | thruster2.period(0.002); // 2 ms period |
markschwarzer | 0:852fb667f2d6 | 57 | thruster2.pulsewidth(1.0/1000.000); /////IMU initial and begin |
markschwarzer | 0:852fb667f2d6 | 58 | IMU.begin(); |
markschwarzer | 0:852fb667f2d6 | 59 | IMU.calibrate(true); |
markschwarzer | 0:852fb667f2d6 | 60 | myled=1; |
markschwarzer | 0:852fb667f2d6 | 61 | //initialize pressure sensor |
markschwarzer | 0:852fb667f2d6 | 62 | pc.printf("setting the pressure sensor\r\n"); |
markschwarzer | 0:852fb667f2d6 | 63 | p_sensor.MS5837Reset(); |
markschwarzer | 0:852fb667f2d6 | 64 | p_sensor.MS5837Init(); |
markschwarzer | 0:852fb667f2d6 | 65 | pc.printf("settting the tickers\r\n"); |
markschwarzer | 0:852fb667f2d6 | 66 | t.start(); |
markschwarzer | 0:852fb667f2d6 | 67 | myled=0; |
markschwarzer | 0:852fb667f2d6 | 68 | welcome(); |
markschwarzer | 0:852fb667f2d6 | 69 | //-----setup ticker-------// |
markschwarzer | 0:852fb667f2d6 | 70 | //setup ticker to separate log and IMU data update. |
markschwarzer | 0:852fb667f2d6 | 71 | //so we could have all our control code in the while loop |
markschwarzer | 0:852fb667f2d6 | 72 | // //log at 2 Hz |
markschwarzer | 0:852fb667f2d6 | 73 | imu_ticker.attach(&IMU_update,0.1); //10Hz |
markschwarzer | 0:852fb667f2d6 | 74 | log_ticker.attach(&log_data,0.5); |
markschwarzer | 0:852fb667f2d6 | 75 | wait(1); |
markschwarzer | 0:852fb667f2d6 | 76 | while(1) |
markschwarzer | 1:558ab0afbbc5 | 77 | { |
markschwarzer | 1:558ab0afbbc5 | 78 | p_sensor.Barometer_MS5837(); |
markschwarzer | 1:558ab0afbbc5 | 79 | |
markschwarzer | 1:558ab0afbbc5 | 80 | printf("Pressure: "); |
markschwarzer | 1:558ab0afbbc5 | 81 | printf("%f",p_sensor.MS5837_Pressure()); |
markschwarzer | 1:558ab0afbbc5 | 82 | printf(" mbar"); |
markschwarzer | 1:558ab0afbbc5 | 83 | |
markschwarzer | 1:558ab0afbbc5 | 84 | printf("Temperature: "); |
markschwarzer | 1:558ab0afbbc5 | 85 | printf("%f",p_sensor.MS5837_Temperature()); |
markschwarzer | 1:558ab0afbbc5 | 86 | printf(" deg C "); |
markschwarzer | 1:558ab0afbbc5 | 87 | |
markschwarzer | 1:558ab0afbbc5 | 88 | printf("Depth: "); |
markschwarzer | 1:558ab0afbbc5 | 89 | printf("%f",p_sensor.depth()); |
markschwarzer | 1:558ab0afbbc5 | 90 | printf(" m "); |
markschwarzer | 1:558ab0afbbc5 | 91 | |
markschwarzer | 1:558ab0afbbc5 | 92 | printf("Altitude: "); |
markschwarzer | 1:558ab0afbbc5 | 93 | printf("%f",p_sensor.altitude()); |
markschwarzer | 1:558ab0afbbc5 | 94 | printf(" m above mean sea level\r\n"); |
markschwarzer | 1:558ab0afbbc5 | 95 | |
markschwarzer | 1:558ab0afbbc5 | 96 | myled=!myled; |
markschwarzer | 1:558ab0afbbc5 | 97 | wait(1); |
markschwarzer | 0:852fb667f2d6 | 98 | } |
markschwarzer | 0:852fb667f2d6 | 99 | |
markschwarzer | 1:558ab0afbbc5 | 100 | } |
markschwarzer | 0:852fb667f2d6 | 101 | |
markschwarzer | 0:852fb667f2d6 | 102 | //-------------Customized functions---------------------------------------------//---------------------------------------- |
markschwarzer | 0:852fb667f2d6 | 103 | ///-----------Welcome menu---------------------/// |
markschwarzer | 0:852fb667f2d6 | 104 | void welcome() |
markschwarzer | 0:852fb667f2d6 | 105 | { |
markschwarzer | 0:852fb667f2d6 | 106 | char buffer[100]={0}; |
markschwarzer | 0:852fb667f2d6 | 107 | int flag=1; |
markschwarzer | 0:852fb667f2d6 | 108 | //Flush the port |
markschwarzer | 0:852fb667f2d6 | 109 | while(BLE.readable()) |
markschwarzer | 0:852fb667f2d6 | 110 | { |
markschwarzer | 0:852fb667f2d6 | 111 | BLE.getc(); |
markschwarzer | 0:852fb667f2d6 | 112 | } |
markschwarzer | 0:852fb667f2d6 | 113 | while(flag) |
markschwarzer | 0:852fb667f2d6 | 114 | { |
markschwarzer | 0:852fb667f2d6 | 115 | BLE.printf("### I am alive\r\n"); |
markschwarzer | 0:852fb667f2d6 | 116 | BLE.printf("### Please enter the log file name you want\r\n"); |
markschwarzer | 0:852fb667f2d6 | 117 | if(BLE.readable()) |
markschwarzer | 0:852fb667f2d6 | 118 | { |
markschwarzer | 0:852fb667f2d6 | 119 | BLE.scanf("%s",buffer); |
markschwarzer | 0:852fb667f2d6 | 120 | sprintf(fname,"/sd/mydir/%s.txt",buffer); //make file name |
markschwarzer | 0:852fb667f2d6 | 121 | |
markschwarzer | 0:852fb667f2d6 | 122 | flag = 0; //set the flag to 0 to break the while |
markschwarzer | 0:852fb667f2d6 | 123 | } |
markschwarzer | 0:852fb667f2d6 | 124 | myled=!myled; |
markschwarzer | 0:852fb667f2d6 | 125 | wait(1); |
markschwarzer | 0:852fb667f2d6 | 126 | } |
markschwarzer | 0:852fb667f2d6 | 127 | //print name |
markschwarzer | 0:852fb667f2d6 | 128 | BLE.printf("### name received\r\n"); |
markschwarzer | 0:852fb667f2d6 | 129 | BLE.printf("### file name and directory is: \r\n %s\r\n",fname); //file name and location |
markschwarzer | 0:852fb667f2d6 | 130 | //open file test |
markschwarzer | 0:852fb667f2d6 | 131 | mkdir("/sd/mydir",0777); //keep 0777, this is magic # |
markschwarzer | 0:852fb667f2d6 | 132 | fp = fopen(fname, "a"); |
markschwarzer | 0:852fb667f2d6 | 133 | if(fp == NULL){ |
markschwarzer | 0:852fb667f2d6 | 134 | BLE.printf("Could not open file for write\n"); |
markschwarzer | 0:852fb667f2d6 | 135 | } |
markschwarzer | 0:852fb667f2d6 | 136 | else |
markschwarzer | 0:852fb667f2d6 | 137 | { |
markschwarzer | 0:852fb667f2d6 | 138 | BLE.printf("##file open good \n"); //open file and tell if open |
markschwarzer | 0:852fb667f2d6 | 139 | fprintf(fp, "Hello\r\n"); |
markschwarzer | 0:852fb667f2d6 | 140 | fclose(fp); |
markschwarzer | 0:852fb667f2d6 | 141 | } |
markschwarzer | 0:852fb667f2d6 | 142 | |
markschwarzer | 0:852fb667f2d6 | 143 | BLE.printf("### The main program will start in 10 seconds\r\n"); |
markschwarzer | 0:852fb667f2d6 | 144 | wait(5); |
markschwarzer | 0:852fb667f2d6 | 145 | } |
markschwarzer | 0:852fb667f2d6 | 146 | |
markschwarzer | 0:852fb667f2d6 | 147 | ///-----------log functions---------------------/// |
markschwarzer | 0:852fb667f2d6 | 148 | void log_data() |
markschwarzer | 0:852fb667f2d6 | 149 | { |
markschwarzer | 2:5ee1cbcf8e04 | 150 | fp= fopen(fname, "a"); |
markschwarzer | 2:5ee1cbcf8e04 | 151 | fprintf(fp, "$IMU,3,10, %f, %3.3f, %3.3f, %3.3f, %3.3f, %3.3f, %3.3f, %3.3f, %3.3f, %3.3f;\r\n",t.read(),accel[0],accel[1],accel[2],gyro[0],gyro[1],gyro[2],euler[0],euler[1],euler[2]); |
markschwarzer | 2:5ee1cbcf8e04 | 152 | fclose(fp); |
markschwarzer | 0:852fb667f2d6 | 153 | //log system time t.read() |
markschwarzer | 0:852fb667f2d6 | 154 | // log imu data, log sciene data |
markschwarzer | 0:852fb667f2d6 | 155 | // log pulse width |
markschwarzer | 0:852fb667f2d6 | 156 | // log pressure sensor data. |
markschwarzer | 0:852fb667f2d6 | 157 | //science sensor: temp.temp(), light.light() |
markschwarzer | 0:852fb667f2d6 | 158 | //IMU sensor |
markschwarzer | 0:852fb667f2d6 | 159 | |
markschwarzer | 0:852fb667f2d6 | 160 | } |
markschwarzer | 0:852fb667f2d6 | 161 | |
markschwarzer | 0:852fb667f2d6 | 162 | ///-----------IMU related functions---------------------/// |
markschwarzer | 0:852fb667f2d6 | 163 | void IMU_update() |
markschwarzer | 0:852fb667f2d6 | 164 | { |
markschwarzer | 0:852fb667f2d6 | 165 | IMU.readMag(); |
markschwarzer | 0:852fb667f2d6 | 166 | IMU.readGyro(); |
markschwarzer | 0:852fb667f2d6 | 167 | IMU.readAccel(); |
markschwarzer | 0:852fb667f2d6 | 168 | accel[0] = IMU.calcAccel(IMU.ax); |
markschwarzer | 0:852fb667f2d6 | 169 | accel[1] = IMU.calcAccel(IMU.ay); |
markschwarzer | 0:852fb667f2d6 | 170 | accel[2] = -IMU.calcAccel(IMU.az); |
markschwarzer | 0:852fb667f2d6 | 171 | gyro[0] = IMU.calcGyro(IMU.gx); |
markschwarzer | 0:852fb667f2d6 | 172 | gyro[1] = IMU.calcGyro(IMU.gy); |
markschwarzer | 0:852fb667f2d6 | 173 | gyro[2] = -IMU.calcGyro(IMU.gz); |
markschwarzer | 0:852fb667f2d6 | 174 | mag_correction(IMU.calcMag(IMU.mx), IMU.calcMag(IMU.my), IMU.calcMag(IMU.mz), mag); //mag correction |
markschwarzer | 0:852fb667f2d6 | 175 | mag[2] = - mag[2]; |
markschwarzer | 0:852fb667f2d6 | 176 | pose_estimate(euler, accel, gyro, mag); //pose update |
markschwarzer | 0:852fb667f2d6 | 177 | } |
markschwarzer | 0:852fb667f2d6 | 178 | |
markschwarzer | 0:852fb667f2d6 | 179 | void mag_correction(float mx, float my, float mz, float mag_c[3]) |
markschwarzer | 0:852fb667f2d6 | 180 | { |
markschwarzer | 0:852fb667f2d6 | 181 | float bias[3] = {0.0793,0.0357,0.2333}; |
markschwarzer | 0:852fb667f2d6 | 182 | float scale[3][3] = {{1.0070, 0.0705, 0.0368}, |
markschwarzer | 0:852fb667f2d6 | 183 | {0.0705, 1.0807, 0.0265}, |
markschwarzer | 0:852fb667f2d6 | 184 | {0.0368, 0.0265, 0.9250}}; |
markschwarzer | 0:852fb667f2d6 | 185 | //mag_c = (mag-bias)*scale |
markschwarzer | 0:852fb667f2d6 | 186 | |
markschwarzer | 0:852fb667f2d6 | 187 | mag_c[0] = (mx - bias[0]) *scale[0][0] + (my - bias[1]) *scale[1][0] + (mz - bias[2]) *scale[2][0]; |
markschwarzer | 0:852fb667f2d6 | 188 | mag_c[1] = (mx - bias[0]) *scale[0][1] + (my - bias[1]) *scale[1][1] + (mz - bias[2]) *scale[2][1]; |
markschwarzer | 0:852fb667f2d6 | 189 | mag_c[2] = (mx - bias[0]) *scale[0][2] + (my - bias[1]) *scale[1][2] + (mz - bias[2]) *scale[2][2]; |
markschwarzer | 0:852fb667f2d6 | 190 | } |
markschwarzer | 0:852fb667f2d6 | 191 | |
markschwarzer | 0:852fb667f2d6 | 192 | void pose_estimate(float euler[3], float accel[3], float gyro[3], float mag[3]) //pose estimation function |
markschwarzer | 0:852fb667f2d6 | 193 | { |
markschwarzer | 0:852fb667f2d6 | 194 | euler[0] = atan2 (accel[1] , accel[2]/abs(accel[2])*(sqrt ((accel[0] * accel[0]) + (accel[2] * accel[2])))); |
markschwarzer | 0:852fb667f2d6 | 195 | euler[1] = - atan2( -accel[0] ,( sqrt((accel[1] * accel[1]) + (accel[2] * accel[2])))); |
markschwarzer | 0:852fb667f2d6 | 196 | float Yh = (mag[1] * cos(euler[0])) - (mag[2] * sin(euler[0])); |
markschwarzer | 0:852fb667f2d6 | 197 | float Xh = (mag[0] * cos(euler[1]))+(mag[1] * sin(euler[0])*sin(euler[1])) |
markschwarzer | 0:852fb667f2d6 | 198 | + (mag[2] * cos(euler[0]) * sin(euler[1])); |
markschwarzer | 0:852fb667f2d6 | 199 | euler[2] = atan2(Yh, Xh); |
markschwarzer | 0:852fb667f2d6 | 200 | //convert into degrees |
markschwarzer | 0:852fb667f2d6 | 201 | euler[0] *= 180.0f / PI; |
markschwarzer | 0:852fb667f2d6 | 202 | euler[1] *= 180.0f / PI; |
markschwarzer | 0:852fb667f2d6 | 203 | euler[2] *= 180.0f / PI; |
markschwarzer | 0:852fb667f2d6 | 204 | //wrap the values to be within 0 to 360. |
markschwarzer | 0:852fb667f2d6 | 205 | for (int i=0;i<3;i++) |
markschwarzer | 0:852fb667f2d6 | 206 | { |
markschwarzer | 0:852fb667f2d6 | 207 | if(euler[i]<=0) |
markschwarzer | 0:852fb667f2d6 | 208 | { |
markschwarzer | 0:852fb667f2d6 | 209 | euler[i]=euler[i]+360; |
markschwarzer | 0:852fb667f2d6 | 210 | } |
markschwarzer | 0:852fb667f2d6 | 211 | if(euler[i]>360) |
markschwarzer | 0:852fb667f2d6 | 212 | { |
markschwarzer | 0:852fb667f2d6 | 213 | euler[i]=euler[i]-360; |
markschwarzer | 0:852fb667f2d6 | 214 | } |
markschwarzer | 0:852fb667f2d6 | 215 | } |
markschwarzer | 0:852fb667f2d6 | 216 | |
markschwarzer | 0:852fb667f2d6 | 217 | } |
markschwarzer | 0:852fb667f2d6 | 218 | |
markschwarzer | 0:852fb667f2d6 | 219 | ///-----------Control related functions---------------------/// |
markschwarzer | 0:852fb667f2d6 | 220 | ////Thruster on control, pw->pulse width in milli-second// |
markschwarzer | 0:852fb667f2d6 | 221 | //// pw range between 1 to 1.5// |
markschwarzer | 0:852fb667f2d6 | 222 | //// on_time-> thruster on time. |
markschwarzer | 0:852fb667f2d6 | 223 | void thrust_on(float pw, float on_time) //input is pulse width |
markschwarzer | 0:852fb667f2d6 | 224 | { |
markschwarzer | 0:852fb667f2d6 | 225 | float pw_max=2.0; |
markschwarzer | 0:852fb667f2d6 | 226 | if(pw>pw_max) |
markschwarzer | 0:852fb667f2d6 | 227 | { |
markschwarzer | 0:852fb667f2d6 | 228 | pw=pw_max; //hard limitation |
markschwarzer | 0:852fb667f2d6 | 229 | } |
markschwarzer | 0:852fb667f2d6 | 230 | Timer tt; |
markschwarzer | 0:852fb667f2d6 | 231 | tt.reset(); |
markschwarzer | 0:852fb667f2d6 | 232 | tt.start(); |
markschwarzer | 0:852fb667f2d6 | 233 | // lets set the pulse width |
markschwarzer | 0:852fb667f2d6 | 234 | //thruster.period(20.0/1000.00); // 20 ms period |
markschwarzer | 0:852fb667f2d6 | 235 | thruster.pulsewidth(pw/1000.00); |
markschwarzer | 0:852fb667f2d6 | 236 | thruster2.pulsewidth(pw/1000.00); |
markschwarzer | 0:852fb667f2d6 | 237 | //PWM will be kept until time out |
markschwarzer | 0:852fb667f2d6 | 238 | while(tt.read()<=on_time) |
markschwarzer | 0:852fb667f2d6 | 239 | { |
markschwarzer | 0:852fb667f2d6 | 240 | } |
markschwarzer | 0:852fb667f2d6 | 241 | //stop the timer |
markschwarzer | 0:852fb667f2d6 | 242 | tt.stop(); |
markschwarzer | 0:852fb667f2d6 | 243 | //turn off the thruster |
markschwarzer | 0:852fb667f2d6 | 244 | thruster.pulsewidth(1.0/1000.00); |
markschwarzer | 0:852fb667f2d6 | 245 | thruster2.pulsewidth(1.0/1000.00); |
markschwarzer | 0:852fb667f2d6 | 246 | |
markschwarzer | 0:852fb667f2d6 | 247 | } |