Ironcup Mar 2020
Dependencies: mbed mbed-rtos MotionSensor EthernetInterface
main.cpp@7:27516a2b504b, 2016-04-17 (annotated)
- Committer:
- drelliak
- Date:
- Sun Apr 17 01:07:25 2016 +0000
- Revision:
- 7:27516a2b504b
- Parent:
- 6:c930555fd872
- Child:
- 8:a1067fcde341
added the main loop
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
drelliak | 0:88faaa1afb83 | 1 | #include "FXAS21002.h" |
drelliak | 0:88faaa1afb83 | 2 | #include "FXOS8700Q.h" |
drelliak | 0:88faaa1afb83 | 3 | #include "mbed.h" |
drelliak | 0:88faaa1afb83 | 4 | #include "CarPWM.h" |
drelliak | 0:88faaa1afb83 | 5 | #include "receiver.h" |
drelliak | 0:88faaa1afb83 | 6 | |
drelliak | 0:88faaa1afb83 | 7 | |
drelliak | 0:88faaa1afb83 | 8 | #define PI 3.141592653589793238462 |
drelliak | 0:88faaa1afb83 | 9 | #define Ts 0.02 // Seconds |
drelliak | 6:c930555fd872 | 10 | #define PWM_PERIOD 13.5 // ms |
drelliak | 0:88faaa1afb83 | 11 | #define INITIAL_P 0.452531214933414 |
drelliak | 0:88faaa1afb83 | 12 | #define INITIAL_I 5.45748932024049 |
drelliak | 0:88faaa1afb83 | 13 | #define INITIAL_D 0.000233453623255507 |
drelliak | 0:88faaa1afb83 | 14 | #define INITIAL_N 51.0605584484153 |
drelliak | 6:c930555fd872 | 15 | #define BRAKE_CONSTANT 40 |
drelliak | 6:c930555fd872 | 16 | #define BRAKE_WAIT 0.3 |
drelliak | 0:88faaa1afb83 | 17 | #define GYRO_OFFSET 0.0152 |
drelliak | 0:88faaa1afb83 | 18 | #define END_THRESH 4 |
drelliak | 0:88faaa1afb83 | 19 | #define START_THRESH 10 |
drelliak | 0:88faaa1afb83 | 20 | #define MINIMUM_VELOCITY 15 |
drelliak | 6:c930555fd872 | 21 | |
drelliak | 0:88faaa1afb83 | 22 | Serial ser(USBTX, USBRX); // Initialize Serial port |
drelliak | 0:88faaa1afb83 | 23 | PwmOut motor(PTD1); // Motor connected to pin PTD1 |
drelliak | 0:88faaa1afb83 | 24 | PwmOut servo(PTD3); // Servo connected to pin PTD3 |
drelliak | 0:88faaa1afb83 | 25 | |
drelliak | 0:88faaa1afb83 | 26 | FXOS8700Q_mag mag(PTE25,PTE24,FXOS8700CQ_SLAVE_ADDR1); |
drelliak | 0:88faaa1afb83 | 27 | FXAS21002 gyro(PTE25,PTE24); |
drelliak | 0:88faaa1afb83 | 28 | |
drelliak | 0:88faaa1afb83 | 29 | |
drelliak | 0:88faaa1afb83 | 30 | |
drelliak | 0:88faaa1afb83 | 31 | // PID controller parameters and functions |
drelliak | 0:88faaa1afb83 | 32 | float e[2], u, up[1],ui[2], ud[2]; // The vector coeficient means a time delay, for exemple e[a] = e(k-a) -> z^(-a)e(k) |
drelliak | 0:88faaa1afb83 | 33 | float P, I, D, N, reference = 0; |
drelliak | 0:88faaa1afb83 | 34 | void controlAnglePID(float P, float I, float D, float N); |
drelliak | 0:88faaa1afb83 | 35 | void initializeController(); |
drelliak | 0:88faaa1afb83 | 36 | |
drelliak | 0:88faaa1afb83 | 37 | // Gyroscope variables and functions |
drelliak | 0:88faaa1afb83 | 38 | float gyro_data[3], gyro_angle; |
drelliak | 0:88faaa1afb83 | 39 | Timer t; |
drelliak | 0:88faaa1afb83 | 40 | void processGyroAngle(); |
drelliak | 0:88faaa1afb83 | 41 | void startGyro(); |
drelliak | 0:88faaa1afb83 | 42 | void stopGyro(); |
drelliak | 0:88faaa1afb83 | 43 | |
drelliak | 0:88faaa1afb83 | 44 | // Magnetometer variables and functions |
drelliak | 0:88faaa1afb83 | 45 | float max_x, max_y, min_x, min_y,x,y; |
drelliak | 0:88faaa1afb83 | 46 | MotionSensorDataUnits mag_data; |
drelliak | 0:88faaa1afb83 | 47 | float processMagAngle(); |
drelliak | 0:88faaa1afb83 | 48 | void magCal(); |
drelliak | 0:88faaa1afb83 | 49 | |
drelliak | 0:88faaa1afb83 | 50 | // State variables |
drelliak | 6:c930555fd872 | 51 | float feedback, velocity = 0; |
drelliak | 0:88faaa1afb83 | 52 | void readProtocol(); |
drelliak | 0:88faaa1afb83 | 53 | void brakeMotor(); |
drelliak | 6:c930555fd872 | 54 | void reverseMotor(int speed); |
drelliak | 6:c930555fd872 | 55 | void setVelocity(int new_velocity); |
drelliak | 6:c930555fd872 | 56 | |
drelliak | 0:88faaa1afb83 | 57 | // Test functions |
drelliak | 0:88faaa1afb83 | 58 | void debug(); |
drelliak | 0:88faaa1afb83 | 59 | |
drelliak | 0:88faaa1afb83 | 60 | int main(){ |
drelliak | 7:27516a2b504b | 61 | printf("Initializing controller....\r\n\r\n"); |
drelliak | 7:27516a2b504b | 62 | initializeController(); |
drelliak | 7:27516a2b504b | 63 | printf("Controller Initialized. \r\n"); |
drelliak | 7:27516a2b504b | 64 | magCal(); |
drelliak | 7:27516a2b504b | 65 | gyro_angle = processMagAngle(); |
drelliak | 7:27516a2b504b | 66 | velocity = MINIMUM_VELOCITY; |
drelliak | 7:27516a2b504b | 67 | setMotorPWM(velocity,motor); |
drelliak | 7:27516a2b504b | 68 | startGyro(); |
drelliak | 7:27516a2b504b | 69 | while (true){ |
drelliak | 7:27516a2b504b | 70 | processGyroAngle(); |
drelliak | 7:27516a2b504b | 71 | controlAnglePID(P,I,D,N); |
drelliak | 7:27516a2b504b | 72 | debug(); |
drelliak | 7:27516a2b504b | 73 | if(t.read_us() < Ts*1000000) |
drelliak | 7:27516a2b504b | 74 | wait_us(Ts*1000000 - t.read_us()); |
drelliak | 7:27516a2b504b | 75 | if(pc.readable()) |
drelliak | 7:27516a2b504b | 76 | readProtocol(); |
drelliak | 7:27516a2b504b | 77 | } |
drelliak | 0:88faaa1afb83 | 78 | } |
drelliak | 0:88faaa1afb83 | 79 | void readProtocol(){ |
drelliak | 0:88faaa1afb83 | 80 | char msg = ser.getc(); |
drelliak | 0:88faaa1afb83 | 81 | switch(msg) |
drelliak | 0:88faaa1afb83 | 82 | { |
drelliak | 0:88faaa1afb83 | 83 | case NONE: |
drelliak | 0:88faaa1afb83 | 84 | //ser.printf("sending red signal to led\r\n"); |
drelliak | 0:88faaa1afb83 | 85 | return; |
drelliak | 0:88faaa1afb83 | 86 | break; |
drelliak | 0:88faaa1afb83 | 87 | case BRAKE: |
drelliak | 0:88faaa1afb83 | 88 | //ser.printf("sending green signal to led\r\n"); |
drelliak | 0:88faaa1afb83 | 89 | brakeMotor(); |
drelliak | 0:88faaa1afb83 | 90 | break; |
drelliak | 0:88faaa1afb83 | 91 | case ANG_RST: |
drelliak | 0:88faaa1afb83 | 92 | //ser.printf("sending blue signal to led\r\n"); |
drelliak | 0:88faaa1afb83 | 93 | stopGyro(); |
drelliak | 0:88faaa1afb83 | 94 | gyro_angle = 0; |
drelliak | 0:88faaa1afb83 | 95 | startGyro(); |
drelliak | 0:88faaa1afb83 | 96 | return; |
drelliak | 0:88faaa1afb83 | 97 | break; |
drelliak | 0:88faaa1afb83 | 98 | case ANG_REF: |
drelliak | 0:88faaa1afb83 | 99 | reference = get_ang_ref(ser); |
drelliak | 0:88faaa1afb83 | 100 | break; |
drelliak | 0:88faaa1afb83 | 101 | case GND_SPEED: |
drelliak | 0:88faaa1afb83 | 102 | velocity = get_gnd_speed(ser); |
drelliak | 0:88faaa1afb83 | 103 | setMotorPWM(velocity,motor); |
drelliak | 0:88faaa1afb83 | 104 | break; |
drelliak | 0:88faaa1afb83 | 105 | case PID_PARAMS: |
drelliak | 0:88faaa1afb83 | 106 | ser.putc('p'); |
drelliak | 0:88faaa1afb83 | 107 | get_pid_params(ser, &P, &I, &D, &N); |
drelliak | 0:88faaa1afb83 | 108 | break; |
drelliak | 0:88faaa1afb83 | 109 | default: |
drelliak | 0:88faaa1afb83 | 110 | // ser.flush(); |
drelliak | 0:88faaa1afb83 | 111 | |
drelliak | 0:88faaa1afb83 | 112 | } |
drelliak | 0:88faaa1afb83 | 113 | } |
drelliak | 0:88faaa1afb83 | 114 | /* Initialize the controller parameter P, I, D and N with the initial values and set the error and input to 0. */ |
drelliak | 0:88faaa1afb83 | 115 | void initializeController(){ |
drelliak | 0:88faaa1afb83 | 116 | for(int i =0; i<2; i++){ |
drelliak | 0:88faaa1afb83 | 117 | e[i] = 0; |
drelliak | 0:88faaa1afb83 | 118 | ui[i] = 0; |
drelliak | 0:88faaa1afb83 | 119 | ud[i] = 0; |
drelliak | 0:88faaa1afb83 | 120 | } |
drelliak | 0:88faaa1afb83 | 121 | P= INITIAL_P; |
drelliak | 0:88faaa1afb83 | 122 | I= INITIAL_I; |
drelliak | 0:88faaa1afb83 | 123 | D= INITIAL_D; |
drelliak | 0:88faaa1afb83 | 124 | N= INITIAL_N; |
drelliak | 0:88faaa1afb83 | 125 | } |
drelliak | 0:88faaa1afb83 | 126 | |
drelliak | 0:88faaa1afb83 | 127 | /* Start the Gyroscope timer and set the initial configuration */ |
drelliak | 0:88faaa1afb83 | 128 | void startGyro(){ |
drelliak | 0:88faaa1afb83 | 129 | gyro.gyro_config(); |
drelliak | 0:88faaa1afb83 | 130 | t.start(); |
drelliak | 0:88faaa1afb83 | 131 | } |
drelliak | 0:88faaa1afb83 | 132 | |
drelliak | 0:88faaa1afb83 | 133 | /* Stop and reset the Gyroscope */ |
drelliak | 0:88faaa1afb83 | 134 | void stopGyro(){ |
drelliak | 0:88faaa1afb83 | 135 | t.stop(); |
drelliak | 0:88faaa1afb83 | 136 | t.reset(); |
drelliak | 0:88faaa1afb83 | 137 | gyro_angle = 0; |
drelliak | 0:88faaa1afb83 | 138 | } |
drelliak | 0:88faaa1afb83 | 139 | |
drelliak | 0:88faaa1afb83 | 140 | /* Integrate the Gyroscope to get the angular position (Deegre/seconds) */ |
drelliak | 0:88faaa1afb83 | 141 | void processGyroAngle(){ |
drelliak | 0:88faaa1afb83 | 142 | gyro.acquire_gyro_data_dps(gyro_data); |
drelliak | 0:88faaa1afb83 | 143 | t.stop(); |
drelliak | 0:88faaa1afb83 | 144 | gyro_angle = gyro_angle + (gyro_data[2] + GYRO_OFFSET)*(double)(t.read_us())/1000000; |
drelliak | 0:88faaa1afb83 | 145 | t.reset(); |
drelliak | 0:88faaa1afb83 | 146 | t.start(); |
drelliak | 4:8ead8ada8a8b | 147 | feedback = gyro_angle; |
drelliak | 4:8ead8ada8a8b | 148 | if(feedback > 180) |
drelliak | 4:8ead8ada8a8b | 149 | feedback = feedback - 360; |
drelliak | 4:8ead8ada8a8b | 150 | if(feedback < -180) |
drelliak | 4:8ead8ada8a8b | 151 | feedback = feedback + 360; |
drelliak | 0:88faaa1afb83 | 152 | } |
drelliak | 0:88faaa1afb83 | 153 | |
drelliak | 0:88faaa1afb83 | 154 | /* PID controller for angular position */ |
drelliak | 0:88faaa1afb83 | 155 | void controlAnglePID(float P, float I, float D, float N){ |
drelliak | 0:88faaa1afb83 | 156 | /* Getting error */ |
drelliak | 0:88faaa1afb83 | 157 | e[1] = e[0]; |
drelliak | 4:8ead8ada8a8b | 158 | e[0] = reference - (feedback*PI/180); |
drelliak | 0:88faaa1afb83 | 159 | if(e[0] > PI) |
drelliak | 0:88faaa1afb83 | 160 | e[0]= e[0] - 2*PI; |
drelliak | 0:88faaa1afb83 | 161 | if(e[0] < -PI) |
drelliak | 0:88faaa1afb83 | 162 | e[0] = e[0] + 2*PI; |
drelliak | 0:88faaa1afb83 | 163 | /* Proportinal Part */ |
drelliak | 0:88faaa1afb83 | 164 | up[0] = e[0]*P; |
drelliak | 0:88faaa1afb83 | 165 | /* Integral Part */ |
drelliak | 0:88faaa1afb83 | 166 | ui[1] = ui[0]; |
drelliak | 0:88faaa1afb83 | 167 | if(abs(u) < PI/8){ |
drelliak | 0:88faaa1afb83 | 168 | ui[0] = (P*I*Ts)*e[1] + ui[1]; |
drelliak | 0:88faaa1afb83 | 169 | } |
drelliak | 0:88faaa1afb83 | 170 | else if(u > 0) |
drelliak | 0:88faaa1afb83 | 171 | ui[0] = PI/8 - up[0]; |
drelliak | 0:88faaa1afb83 | 172 | else if(u < 0) |
drelliak | 0:88faaa1afb83 | 173 | ui[0] = -PI/8 - up[0]; |
drelliak | 0:88faaa1afb83 | 174 | /* Derivative Part */ |
drelliak | 0:88faaa1afb83 | 175 | ud[1] = ud[0]; |
drelliak | 0:88faaa1afb83 | 176 | ud[0] = P*D*N*(e[0] - e[1]) - ud[1]*(N*Ts -1); |
drelliak | 0:88faaa1afb83 | 177 | /** Controller **/ |
drelliak | 0:88faaa1afb83 | 178 | u = up[0] + ud[0] + ui[0]; |
drelliak | 0:88faaa1afb83 | 179 | setServoPWM(u*100/(PI/8), servo); |
drelliak | 0:88faaa1afb83 | 180 | } |
drelliak | 0:88faaa1afb83 | 181 | /* Brake function, braking while the gyroscope is still integrating will cause considerably error in the measurement. */ |
drelliak | 0:88faaa1afb83 | 182 | void brakeMotor(){ |
drelliak | 6:c930555fd872 | 183 | if(velocity >= 0){ |
drelliak | 3:e213c44a9f6c | 184 | setMotorPWM(-BRAKE_CONSTANT, motor); |
drelliak | 6:c930555fd872 | 185 | wait(BRAKE_WAIT); |
drelliak | 6:c930555fd872 | 186 | velocity = 0; |
drelliak | 6:c930555fd872 | 187 | setMotorPWM(velocity,motor); |
drelliak | 6:c930555fd872 | 188 | } |
drelliak | 6:c930555fd872 | 189 | else { |
drelliak | 6:c930555fd872 | 190 | setVelocity(0); |
drelliak | 6:c930555fd872 | 191 | } |
drelliak | 6:c930555fd872 | 192 | } |
drelliak | 6:c930555fd872 | 193 | void reverseMotor(int speed){ |
drelliak | 6:c930555fd872 | 194 | for(int i=0 ; i >= -speed; i--){ |
drelliak | 6:c930555fd872 | 195 | setMotorPWM((float)i,motor); |
drelliak | 6:c930555fd872 | 196 | wait_ms(13.5); |
drelliak | 6:c930555fd872 | 197 | } |
drelliak | 6:c930555fd872 | 198 | for(int i=-speed ; i <= 0; i++){ |
drelliak | 6:c930555fd872 | 199 | setMotorPWM((float)i,motor); |
drelliak | 6:c930555fd872 | 200 | wait_ms(13.5); |
drelliak | 6:c930555fd872 | 201 | } |
drelliak | 6:c930555fd872 | 202 | for(int i=0 ; i >= -speed; i--){ |
drelliak | 6:c930555fd872 | 203 | setMotorPWM((float)i,motor); |
drelliak | 6:c930555fd872 | 204 | wait_ms(13.5); |
drelliak | 6:c930555fd872 | 205 | } |
drelliak | 6:c930555fd872 | 206 | } |
drelliak | 6:c930555fd872 | 207 | void setVelocity(int new_velocity){ |
drelliak | 6:c930555fd872 | 208 | if( velocity > new_velocity){ |
drelliak | 6:c930555fd872 | 209 | for(; velocity >= new_velocity; velocity--){ |
drelliak | 6:c930555fd872 | 210 | setMotorPWM(velocity,motor); |
drelliak | 6:c930555fd872 | 211 | wait_ms(PWM_PERIOD); |
drelliak | 6:c930555fd872 | 212 | } |
drelliak | 6:c930555fd872 | 213 | velocity++; |
drelliak | 6:c930555fd872 | 214 | } |
drelliak | 6:c930555fd872 | 215 | else if(velocity < new_velocity){ |
drelliak | 6:c930555fd872 | 216 | for(; velocity <= new_velocity; velocity++){ |
drelliak | 6:c930555fd872 | 217 | setMotorPWM(velocity,motor); |
drelliak | 6:c930555fd872 | 218 | wait_ms(PWM_PERIOD); |
drelliak | 6:c930555fd872 | 219 | } |
drelliak | 6:c930555fd872 | 220 | velocity--; |
drelliak | 6:c930555fd872 | 221 | } |
drelliak | 0:88faaa1afb83 | 222 | } |
drelliak | 0:88faaa1afb83 | 223 | /* Debug functions that prints the sensor and control inputs values. Since it's time consuming it should not be used */ |
drelliak | 0:88faaa1afb83 | 224 | /* in the main loop or the controller performance may be affected. */ |
drelliak | 0:88faaa1afb83 | 225 | void debug(){ |
drelliak | 0:88faaa1afb83 | 226 | //printf("ERROR: %f Up: %f Ui: %f Ud: %f U: %f \r\n", e[0]*180/3.1415, up[0]*100/(3.1415/8), ui[0]*100/(3.1415/8), ud[0]*100/(3.1415/8),u*100/(3.1415/8)); |
drelliak | 0:88faaa1afb83 | 227 | //printf("Erro: %f Sensor: %f Magnetometer: %f \r\n",e[0]*180/PI,sensor,processMagAngle(0)*180/PI); |
drelliak | 4:8ead8ada8a8b | 228 | printf(" %f \r\n",feedback); |
drelliak | 0:88faaa1afb83 | 229 | } |
drelliak | 0:88faaa1afb83 | 230 | |
drelliak | 0:88faaa1afb83 | 231 | /* Function to normalize the magnetometer reading */ |
drelliak | 0:88faaa1afb83 | 232 | void magCal(){ |
drelliak | 0:88faaa1afb83 | 233 | printf("Starting Calibration"); |
drelliak | 0:88faaa1afb83 | 234 | mag.enable(); |
drelliak | 0:88faaa1afb83 | 235 | wait(0.01); |
drelliak | 0:88faaa1afb83 | 236 | mag.getAxis(mag_data); |
drelliak | 0:88faaa1afb83 | 237 | float x0 = max_x = min_y = mag_data.x; |
drelliak | 0:88faaa1afb83 | 238 | float y0 = max_y = min_y = mag_data.y; |
drelliak | 0:88faaa1afb83 | 239 | bool began = false; |
drelliak | 0:88faaa1afb83 | 240 | while(!(began && abs(mag_data.x - x0) < END_THRESH && abs(mag_data.y - y0) < END_THRESH)){ |
drelliak | 0:88faaa1afb83 | 241 | mag.getAxis(mag_data); |
drelliak | 0:88faaa1afb83 | 242 | if(mag_data.x > max_x) |
drelliak | 0:88faaa1afb83 | 243 | max_x = mag_data.x; |
drelliak | 0:88faaa1afb83 | 244 | if(mag_data.y > max_y) |
drelliak | 0:88faaa1afb83 | 245 | max_y = mag_data.y; |
drelliak | 0:88faaa1afb83 | 246 | if(mag_data.y < min_y) |
drelliak | 0:88faaa1afb83 | 247 | min_y = mag_data.y; |
drelliak | 0:88faaa1afb83 | 248 | if(mag_data.x < min_x) |
drelliak | 0:88faaa1afb83 | 249 | min_x = mag_data.x; |
drelliak | 0:88faaa1afb83 | 250 | if(abs(mag_data.x-x0)>START_THRESH && abs(mag_data.y-y0) > START_THRESH) |
drelliak | 0:88faaa1afb83 | 251 | began = true; |
drelliak | 0:88faaa1afb83 | 252 | printf("began: %d X-X0: %f , Y-Y0: %f \n\r", began, abs(mag_data.x-x0),abs(mag_data.y-y0)); |
drelliak | 0:88faaa1afb83 | 253 | } |
drelliak | 0:88faaa1afb83 | 254 | printf("Calibration Completed: X_MAX = %f , Y_MAX = %f , X_MIN = %f and Y_MIN = %f \n\r",max_x,max_y,min_x,min_y); |
drelliak | 0:88faaa1afb83 | 255 | } |
drelliak | 0:88faaa1afb83 | 256 | |
drelliak | 0:88faaa1afb83 | 257 | /* Function to transform the magnetometer reading in angle(rad/s).*/ |
drelliak | 0:88faaa1afb83 | 258 | float processMagAngle(){ |
drelliak | 0:88faaa1afb83 | 259 | mag.getAxis(mag_data); |
drelliak | 0:88faaa1afb83 | 260 | x = 2*(mag_data.x-min_x)/float(max_x-min_x) - 1; |
drelliak | 0:88faaa1afb83 | 261 | y = 2*(mag_data.y-min_y)/float(max_y-min_y) - 1; |
drelliak | 0:88faaa1afb83 | 262 | return atan2(y,x); |
drelliak | 0:88faaa1afb83 | 263 | } |