Ian Hua / Quadcopter-mbedRTOS
Committer:
pHysiX
Date:
Mon May 12 13:55:34 2014 +0000
Revision:
34:228d87c45151
Parent:
32:7a9be7761c46
Child:
35:6aada2e9ffb8
Code tidied

Who changed what in which revision?

UserRevisionLine numberNew contents of line
pHysiX 22:ef8aa9728013 1 /* File: Setup.h
pHysiX 22:ef8aa9728013 2 * Author: Trung Tin Ian HUA
pHysiX 22:ef8aa9728013 3 * Date: May 2014
pHysiX 22:ef8aa9728013 4 * Purpose: Setup code to initialise all device
pHysiX 22:ef8aa9728013 5 */
pHysiX 30:d9b988f8d84f 6 float KP_PITCH_STABLE = 0.5;//0.8; // 0.5;
pHysiX 30:d9b988f8d84f 7 float KP_ROLL_STABLE = 0.5;//0.8; // 0.5;
pHysiX 32:7a9be7761c46 8
pHysiX 25:a7cfe421cb4a 9 float KP_YAW_RATE = 8.8; // 8.5 for RATE MODE
pHysiX 32:7a9be7761c46 10 float KP_PITCH_RATE = 7.0;//7.6; // 8.5 for RATE MODE
pHysiX 32:7a9be7761c46 11 float KP_ROLL_RATE = 7.0;//7.5;//7.6; // 8.5 for RATE MODE
pHysiX 21:b642c18eccd1 12
pHysiX 32:7a9be7761c46 13 float TI_YAW_RATE = 8.0;//2.0;
pHysiX 32:7a9be7761c46 14 float TI_PITCH_RATE = 2.0;//2.0;
pHysiX 32:7a9be7761c46 15 float TI_ROLL_RATE = 2.0;//2.7;
pHysiX 32:7a9be7761c46 16
pHysiX 32:7a9be7761c46 17 float PID_TI_STABLE = 1.0;
pHysiX 21:b642c18eccd1 18
pHysiX 0:8c28fac22d27 19 #include "setup.h"
pHysiX 0:8c28fac22d27 20 #include "tasks.h"
pHysiX 0:8c28fac22d27 21
pHysiX 4:01921a136f58 22 Serial BT(p28, p27);
pHysiX 4:01921a136f58 23 DigitalOut BT_CMD(p29);
pHysiX 22:ef8aa9728013 24 MPU6050 imu(p9, p10); // MPU6050-Tilty SDA, SCL (18, 19)
pHysiX 22:ef8aa9728013 25 MPL3115A2 altimeter(p9, p10);
pHysiX 12:953d25061417 26 #ifdef ENABLE_COMPASS
pHysiX 12:953d25061417 27 HMC5883L compass(p9, p10);
pHysiX 12:953d25061417 28 #endif
pHysiX 12:953d25061417 29
pHysiX 27:18b6580eb0b1 30 PID pitchPIDstable(KP_PITCH_STABLE, PID_TI_STABLE, 0.0, TASK2_MASTER_PERIOD/1000.0);
pHysiX 27:18b6580eb0b1 31 PID rollPIDstable(KP_ROLL_STABLE, PID_TI_STABLE, 0.0, TASK2_MASTER_PERIOD/1000.0);
pHysiX 12:953d25061417 32
pHysiX 32:7a9be7761c46 33 PID yawPIDrate(KP_YAW_RATE, 0, 0.0, TASK2_SLAVE_PERIOD/1000.0);
pHysiX 32:7a9be7761c46 34 PID pitchPIDrate(KP_PITCH_RATE, TI_PITCH_RATE, 0.0, TASK2_SLAVE_PERIOD/1000.0);
pHysiX 32:7a9be7761c46 35 PID rollPIDrate(KP_ROLL_RATE, TI_ROLL_RATE, 0.0, TASK2_SLAVE_PERIOD/1000.0);
pHysiX 20:b193a50a2ba3 36
pHysiX 3:605fbcb54e75 37 PwmOut ESC[4] = {p21, p22, p23, p24};
pHysiX 3:605fbcb54e75 38
pHysiX 3:605fbcb54e75 39
pHysiX 2:ab967d7b4346 40
pHysiX 34:228d87c45151 41 // ==========================
pHysiX 3:605fbcb54e75 42 // === MAIN SETUP ROUTINE ===
pHysiX 34:228d87c45151 43 // ==========================
pHysiX 3:605fbcb54e75 44 bool setupALLdevices(void)
pHysiX 3:605fbcb54e75 45 {
pHysiX 3:605fbcb54e75 46 bool error = false;
pHysiX 22:ef8aa9728013 47 box_demo = false;
pHysiX 27:18b6580eb0b1 48 mode = ATTITUDE;
pHysiX 3:605fbcb54e75 49
pHysiX 3:605fbcb54e75 50 if (!setup_ESC()) {
pHysiX 3:605fbcb54e75 51 imu.debugSerial.printf("ESC FAILED!!!\n");
pHysiX 3:605fbcb54e75 52 error = true;
pHysiX 3:605fbcb54e75 53 }
pHysiX 0:8c28fac22d27 54
pHysiX 3:605fbcb54e75 55 if (setup_bt())
pHysiX 21:b642c18eccd1 56 imu.debugSerial.printf("BT established!\n");
pHysiX 3:605fbcb54e75 57 else error = true;
pHysiX 3:605fbcb54e75 58
pHysiX 3:605fbcb54e75 59 if (setup_PID())
pHysiX 21:b642c18eccd1 60 imu.debugSerial.printf("PID established!\n");
pHysiX 3:605fbcb54e75 61 else error = true;
pHysiX 0:8c28fac22d27 62
pHysiX 3:605fbcb54e75 63 if (setup_mpu6050())
pHysiX 21:b642c18eccd1 64 imu.debugSerial.printf("MPU6050 established!\n");
pHysiX 3:605fbcb54e75 65 else error = true;
pHysiX 3:605fbcb54e75 66
pHysiX 22:ef8aa9728013 67 if (setup_altimeter())
pHysiX 22:ef8aa9728013 68 imu.debugSerial.printf("Altimeter established!\n");
pHysiX 22:ef8aa9728013 69 else {
pHysiX 22:ef8aa9728013 70 error = true;
pHysiX 22:ef8aa9728013 71 imu.debugSerial.printf("ALTIMETER FAILED\n");
pHysiX 22:ef8aa9728013 72 }
pHysiX 22:ef8aa9728013 73
pHysiX 12:953d25061417 74 #ifdef ENABLE_COMPASS
pHysiX 12:953d25061417 75 if (setup_compass())
pHysiX 21:b642c18eccd1 76 imu.debugSerial.printf("Compass established!\n");
pHysiX 12:953d25061417 77 else error = true;
pHysiX 12:953d25061417 78 #endif
pHysiX 12:953d25061417 79
pHysiX 3:605fbcb54e75 80 return error;
pHysiX 3:605fbcb54e75 81 }
pHysiX 0:8c28fac22d27 82
pHysiX 3:605fbcb54e75 83
pHysiX 3:605fbcb54e75 84
pHysiX 3:605fbcb54e75 85
pHysiX 34:228d87c45151 86 // *************************
pHysiX 34:228d87c45151 87 // *** ESC SETUP ROUTINE ***
pHysiX 34:228d87c45151 88 // *************************
pHysiX 2:ab967d7b4346 89 bool setup_ESC(void)
pHysiX 2:ab967d7b4346 90 {
pHysiX 3:605fbcb54e75 91 for (int i = 0; i < 4; i++)
pHysiX 3:605fbcb54e75 92 ESC[i].period_us(ESC_PERIOD_US);
pHysiX 2:ab967d7b4346 93
pHysiX 3:605fbcb54e75 94 for (int i = 0; i < 4; i++)
pHysiX 3:605fbcb54e75 95 ESC[i].pulsewidth_us(0);
pHysiX 3:605fbcb54e75 96
pHysiX 14:267368c83b6a 97 armed = false;
pHysiX 14:267368c83b6a 98
pHysiX 2:ab967d7b4346 99 return true;
pHysiX 2:ab967d7b4346 100 }
pHysiX 2:ab967d7b4346 101
pHysiX 3:605fbcb54e75 102 // ****************************************************************
pHysiX 3:605fbcb54e75 103 // === BLUETOOTH SETUP ROUTINE ===
pHysiX 3:605fbcb54e75 104 // ****************************************************************
pHysiX 0:8c28fac22d27 105 bool setup_bt(void)
pHysiX 0:8c28fac22d27 106 {
pHysiX 0:8c28fac22d27 107 BT.baud(115200);
pHysiX 22:ef8aa9728013 108 BT_CMD = 0; // Place bluetooth into normal mode
pHysiX 0:8c28fac22d27 109 BT.printf("Bluetooth online!\n");
pHysiX 0:8c28fac22d27 110 return true;
pHysiX 0:8c28fac22d27 111 }
pHysiX 0:8c28fac22d27 112
pHysiX 3:605fbcb54e75 113 // ****************************************************************
pHysiX 3:605fbcb54e75 114 // === PID SETUP ROUTINE ===
pHysiX 3:605fbcb54e75 115 // ****************************************************************
pHysiX 3:605fbcb54e75 116 bool setup_PID(void)
pHysiX 3:605fbcb54e75 117 {
pHysiX 21:b642c18eccd1 118 pitchPIDstable.setInputLimits(-90.0, 90.0);
pHysiX 24:54a8cdf17378 119 pitchPIDstable.setOutputLimits(-250, 250.0);
pHysiX 21:b642c18eccd1 120 pitchPIDstable.setBias(0.0);
pHysiX 21:b642c18eccd1 121 pitchPIDstable.setMode(AUTO_MODE);
pHysiX 22:ef8aa9728013 122
pHysiX 21:b642c18eccd1 123 rollPIDstable.setInputLimits(-90.0, 90.0);
pHysiX 24:54a8cdf17378 124 rollPIDstable.setOutputLimits(-250, 250.0);
pHysiX 21:b642c18eccd1 125 rollPIDstable.setBias(0.0);
pHysiX 21:b642c18eccd1 126 rollPIDstable.setMode(AUTO_MODE);
pHysiX 22:ef8aa9728013 127
pHysiX 21:b642c18eccd1 128 yawPIDrate.setInputLimits(-500.0, 500.0);
pHysiX 21:b642c18eccd1 129 yawPIDrate.setOutputLimits(-200.0, 200.0);
pHysiX 21:b642c18eccd1 130 yawPIDrate.setBias(0.0);
pHysiX 3:605fbcb54e75 131 yawPIDrate.setMode(AUTO_MODE);
pHysiX 12:953d25061417 132
pHysiX 21:b642c18eccd1 133 pitchPIDrate.setInputLimits(-500.0, 500.0);
pHysiX 21:b642c18eccd1 134 pitchPIDrate.setOutputLimits(-200.0, 200.0);
pHysiX 21:b642c18eccd1 135 pitchPIDrate.setBias(0.0);
pHysiX 3:605fbcb54e75 136 pitchPIDrate.setMode(AUTO_MODE);
pHysiX 22:ef8aa9728013 137
pHysiX 21:b642c18eccd1 138 rollPIDrate.setInputLimits(-500.0, 500.0);
pHysiX 21:b642c18eccd1 139 rollPIDrate.setOutputLimits(-200.0, 200.0);
pHysiX 21:b642c18eccd1 140 rollPIDrate.setBias(0.0);
pHysiX 3:605fbcb54e75 141 rollPIDrate.setMode(AUTO_MODE);
pHysiX 24:54a8cdf17378 142
pHysiX 24:54a8cdf17378 143 pitchPIDstable.reset();
pHysiX 24:54a8cdf17378 144 rollPIDstable.reset();
pHysiX 24:54a8cdf17378 145 yawPIDrate.reset();
pHysiX 24:54a8cdf17378 146 pitchPIDrate.reset();
pHysiX 24:54a8cdf17378 147 rollPIDrate.reset();
pHysiX 12:953d25061417 148
pHysiX 3:605fbcb54e75 149 return true;
pHysiX 3:605fbcb54e75 150 }
pHysiX 3:605fbcb54e75 151
pHysiX 3:605fbcb54e75 152 // ****************************************************************
pHysiX 3:605fbcb54e75 153 // === MPU6050 SETUP ROUTINE ===
pHysiX 3:605fbcb54e75 154 // ****************************************************************
pHysiX 3:605fbcb54e75 155 bool imu_available = false;
pHysiX 3:605fbcb54e75 156
pHysiX 3:605fbcb54e75 157 /*/ MPU control/status variables: */
pHysiX 3:605fbcb54e75 158 bool dmpReady = false; // set true if DMP init was successful
pHysiX 3:605fbcb54e75 159 uint8_t devStatus; // return status after each device operation (0 = success, !0 = error)
pHysiX 3:605fbcb54e75 160 uint16_t packetSize; // expected DMP packet size (default is 42 bytes)
pHysiX 3:605fbcb54e75 161
pHysiX 0:8c28fac22d27 162 bool setup_mpu6050(void)
pHysiX 0:8c28fac22d27 163 {
pHysiX 0:8c28fac22d27 164 imu.reset();
pHysiX 0:8c28fac22d27 165 imu.debugSerial.baud(115200);
pHysiX 0:8c28fac22d27 166 wait_ms(5);
pHysiX 0:8c28fac22d27 167
pHysiX 0:8c28fac22d27 168 imu.initialize();
pHysiX 0:8c28fac22d27 169 imu_available = imu.testConnection();
pHysiX 0:8c28fac22d27 170
pHysiX 0:8c28fac22d27 171 imu.debugSerial.printf("IMU status...\t\t\t");
pHysiX 0:8c28fac22d27 172 imu_available ? imu.debugSerial.printf("OK!\n") : imu.debugSerial.printf("NOT OK!\n");
pHysiX 0:8c28fac22d27 173
pHysiX 0:8c28fac22d27 174 if (imu_available) {
pHysiX 0:8c28fac22d27 175 devStatus = imu.dmpInitialize();
pHysiX 0:8c28fac22d27 176
pHysiX 3:605fbcb54e75 177 // supply your own gyro offsets here, scaled for min sensitivity
pHysiX 4:01921a136f58 178 //imu.setXGyroOffset(55);
pHysiX 4:01921a136f58 179 //imu.setYGyroOffset(3);
pHysiX 4:01921a136f58 180 //imu.setZGyroOffset(-2);
pHysiX 3:605fbcb54e75 181 //mpu.setZAccelOffset(16282); // 1688 factory default for my test chip
pHysiX 0:8c28fac22d27 182
pHysiX 0:8c28fac22d27 183 if(!devStatus) {
pHysiX 0:8c28fac22d27 184 imu.setDMPEnabled(true);
pHysiX 0:8c28fac22d27 185 while (!imu.getIntDataReadyStatus());
pHysiX 0:8c28fac22d27 186 packetSize = imu.dmpGetFIFOPacketSize();
pHysiX 16:9072cd6fa8d1 187 imu.debugSerial.printf("Packet Size: %d\n", packetSize);
pHysiX 0:8c28fac22d27 188 dmpReady = true;
pHysiX 16:9072cd6fa8d1 189
pHysiX 16:9072cd6fa8d1 190 int8_t xgOffsetTC = imu.getXGyroOffset();
pHysiX 16:9072cd6fa8d1 191 int8_t ygOffsetTC = imu.getYGyroOffset();
pHysiX 16:9072cd6fa8d1 192 int8_t zgOffsetTC = imu.getZGyroOffset();
pHysiX 16:9072cd6fa8d1 193
pHysiX 16:9072cd6fa8d1 194 imu.debugSerial.printf("X Offset: %4d Y Offset: %4d Z Offset: %4d\n", xgOffsetTC, ygOffsetTC, zgOffsetTC);
pHysiX 16:9072cd6fa8d1 195
pHysiX 0:8c28fac22d27 196 } else {
pHysiX 0:8c28fac22d27 197 imu.debugSerial.printf("\tDMP setup failed!\n");
pHysiX 0:8c28fac22d27 198 return false;
pHysiX 0:8c28fac22d27 199 }
pHysiX 0:8c28fac22d27 200
pHysiX 0:8c28fac22d27 201 imu.resetFIFO();
pHysiX 0:8c28fac22d27 202 } else {
pHysiX 0:8c28fac22d27 203 return false;
pHysiX 0:8c28fac22d27 204 }
pHysiX 0:8c28fac22d27 205
pHysiX 0:8c28fac22d27 206 imu.debugSerial.printf("IMU setup routine done!");
pHysiX 0:8c28fac22d27 207
pHysiX 0:8c28fac22d27 208 return dmpReady;
pHysiX 0:8c28fac22d27 209 }
pHysiX 0:8c28fac22d27 210
pHysiX 22:ef8aa9728013 211 // ****************************************************************
pHysiX 22:ef8aa9728013 212 // === MPL3115A2 SETUP ROUTINE ===
pHysiX 22:ef8aa9728013 213 // ****************************************************************
pHysiX 22:ef8aa9728013 214 bool setup_altimeter(void)
pHysiX 22:ef8aa9728013 215 {
pHysiX 22:ef8aa9728013 216 if (altimeter.init())
pHysiX 22:ef8aa9728013 217 return true;
pHysiX 22:ef8aa9728013 218 else
pHysiX 22:ef8aa9728013 219 return false;
pHysiX 22:ef8aa9728013 220 }
pHysiX 22:ef8aa9728013 221
pHysiX 12:953d25061417 222 #ifdef ENABLE_COMPASS
pHysiX 12:953d25061417 223 // ****************************************************************
pHysiX 12:953d25061417 224 // === HMC5883L SETUP ROUTINE ===
pHysiX 12:953d25061417 225 // ****************************************************************
pHysiX 12:953d25061417 226 bool setup_compass(void)
pHysiX 12:953d25061417 227 {
pHysiX 12:953d25061417 228 compass.setConfigurationA(AVG8_SAMPLES | OUTPUT_RATE_75);
pHysiX 12:953d25061417 229 return true;
pHysiX 12:953d25061417 230 }
pHysiX 12:953d25061417 231 #endif