Eurobot2012_Primary

Dependencies:   mbed Eurobot_2012_Primary

Committer:
narshu
Date:
Sun Apr 29 00:09:35 2012 +0000
Revision:
17:bafcef1c3579
Parent:
13:57ea4e520dbd
Child:
19:06610e1c0895
uptodate kalman lib; edit this one only!

Who changed what in which revision?

UserRevisionLine numberNew contents of line
narshu 0:f3bf6c7e2283 1 #include "mbed.h"
narshu 0:f3bf6c7e2283 2 #include "rtos.h"
narshu 0:f3bf6c7e2283 3 #include "TSH.h"
narshu 0:f3bf6c7e2283 4 #include "Kalman.h"
narshu 0:f3bf6c7e2283 5 #include "globals.h"
narshu 0:f3bf6c7e2283 6 #include "motors.h"
narshu 0:f3bf6c7e2283 7 #include "math.h"
narshu 0:f3bf6c7e2283 8 #include "system.h"
narshu 1:bbabbd997d21 9 #include "geometryfuncs.h"
narshu 10:294b9adbc9d3 10 #include "motion.h"
narshu 2:cffa347bb943 11 #include "ai.h"
narshu 2:cffa347bb943 12 #include "ui.h"
narshu 0:f3bf6c7e2283 13
narshu 0:f3bf6c7e2283 14 //#include <iostream>
narshu 0:f3bf6c7e2283 15
narshu 0:f3bf6c7e2283 16 //Interface declaration
narshu 0:f3bf6c7e2283 17 Serial pc(USBTX, USBRX); // tx, rx
narshu 0:f3bf6c7e2283 18
narshu 0:f3bf6c7e2283 19 Motors motors;
narshu 2:cffa347bb943 20 UI ui;
narshu 4:7b7334441da9 21 Kalman kalman(motors,ui,p23,p14,p14,p14,p15,p15,p15,p5,p6,p7,p8,p11);
narshu 2:cffa347bb943 22 AI ai;
narshu 10:294b9adbc9d3 23 Motion motion(motors, ai, kalman);
narshu 1:bbabbd997d21 24
narshu 0:f3bf6c7e2283 25 //TODO mutex on kalman state, and on motor commands (i.e. on the i2c bus)
narshu 0:f3bf6c7e2283 26 //NOTE! Recieving data with RF12B now DISABLED due to interferance with rtos!
narshu 0:f3bf6c7e2283 27
narshu 1:bbabbd997d21 28
narshu 0:f3bf6c7e2283 29 void vMotorThread(void const *argument);
narshu 0:f3bf6c7e2283 30 void vPrintState(void const *argument);
narshu 0:f3bf6c7e2283 31 void motion_thread(void const *argument);
narshu 1:bbabbd997d21 32
narshu 2:cffa347bb943 33 //bool flag_terminate = false;
narshu 0:f3bf6c7e2283 34
narshu 0:f3bf6c7e2283 35 float temp = 0;
narshu 0:f3bf6c7e2283 36
narshu 0:f3bf6c7e2283 37 //Main loop
narshu 0:f3bf6c7e2283 38 int main() {
narshu 0:f3bf6c7e2283 39 pc.baud(115200);
narshu 2:cffa347bb943 40 //Init kalman
narshu 2:cffa347bb943 41 kalman.KalmanInit();
narshu 1:bbabbd997d21 42
narshu 9:377560539b74 43 //Thread tMotorThread(vMotorThread,NULL,osPriorityNormal,256);
narshu 7:f9c59a3e4155 44 //Thread tUpdateState(vPrintState,NULL,osPriorityNormal,1024);
narshu 2:cffa347bb943 45
narshu 10:294b9adbc9d3 46 //Thread thr_motion(motion_thread,NULL,osPriorityNormal,1024);
narshu 1:bbabbd997d21 47 //Motion_Thread_Ptr = &thr_motion;
narshu 1:bbabbd997d21 48
narshu 1:bbabbd997d21 49 //measure cpu usage. output updated once per second to symbol cpupercent
narshu 1:bbabbd997d21 50 //Thread mCPUthread(measureCPUidle, NULL, osPriorityIdle, 1024); //check if stack overflow with such a small staack
narshu 2:cffa347bb943 51
narshu 0:f3bf6c7e2283 52 pc.printf("We got to main! ;D\r\n");
narshu 0:f3bf6c7e2283 53
narshu 0:f3bf6c7e2283 54 //REMEMBERT TO PUT PULL UP RESISTORS ON I2C!!!!!!!!!!!!!!
narshu 0:f3bf6c7e2283 55 while (1) {
narshu 9:377560539b74 56
narshu 9:377560539b74 57 osThreadSetPriority (osThreadGetId(), osPriorityIdle);
narshu 9:377560539b74 58
narshu 9:377560539b74 59 Timer timer;
narshu 9:377560539b74 60 ui.regid(10, 1);
narshu 9:377560539b74 61
narshu 9:377560539b74 62 while(1) {
narshu 9:377560539b74 63 timer.reset();
narshu 9:377560539b74 64 timer.start();
narshu 9:377560539b74 65 nopwait(1000);
narshu 9:377560539b74 66
narshu 9:377560539b74 67 ui.updateval(10, timer.read_us());
narshu 9:377560539b74 68 }
narshu 9:377560539b74 69
narshu 0:f3bf6c7e2283 70 // do nothing
narshu 10:294b9adbc9d3 71 //Thread::wait(osWaitForever);
narshu 0:f3bf6c7e2283 72 }
narshu 0:f3bf6c7e2283 73 }
narshu 0:f3bf6c7e2283 74
narshu 9:377560539b74 75 void AI::ai_thread () {
narshu 9:377560539b74 76 /*
narshu 9:377560539b74 77 //printf("aithreadstart\r\n");
narshu 9:377560539b74 78 Thread::signal_wait(0x01);
narshu 9:377560539b74 79 settarget(660, 400, PI/2, true);
narshu 9:377560539b74 80
narshu 9:377560539b74 81 Thread::signal_wait(0x01);
narshu 9:377560539b74 82 settarget(660, 570, PI, true);
narshu 9:377560539b74 83
narshu 9:377560539b74 84 Thread::signal_wait(0x01);
narshu 9:377560539b74 85 settarget(400, 870, PI, true);
narshu 9:377560539b74 86
narshu 9:377560539b74 87 Thread::signal_wait(0x01);
narshu 9:377560539b74 88 settarget(660, 870, PI, false);
narshu 9:377560539b74 89
narshu 9:377560539b74 90 flag_terminate = true;
narshu 9:377560539b74 91 */
narshu 13:57ea4e520dbd 92
narshu 9:377560539b74 93 while (1) {
narshu 17:bafcef1c3579 94
narshu 17:bafcef1c3579 95
narshu 17:bafcef1c3579 96 // goes to the mid
narshu 17:bafcef1c3579 97 settarget(500, 1000, 0, true);
narshu 17:bafcef1c3579 98 Thread::signal_wait(0x01);
narshu 17:bafcef1c3579 99 Thread::wait(2000);
narshu 17:bafcef1c3579 100
narshu 17:bafcef1c3579 101 settarget(2500, 1000, PI, true);
narshu 17:bafcef1c3579 102 Thread::signal_wait(0x01);
narshu 17:bafcef1c3579 103 Thread::wait(2000);
narshu 17:bafcef1c3579 104
narshu 17:bafcef1c3579 105 }
narshu 9:377560539b74 106
narshu 17:bafcef1c3579 107
narshu 17:bafcef1c3579 108
narshu 17:bafcef1c3579 109 // goes to the mid
narshu 17:bafcef1c3579 110 settarget(700, 1500, 0, false);
narshu 17:bafcef1c3579 111 Thread::signal_wait(0x01);
narshu 17:bafcef1c3579 112 Thread::wait(2000);
narshu 17:bafcef1c3579 113 //////////////////////////////////////////////////////
narshu 9:377560539b74 114 // goes to the mid
narshu 13:57ea4e520dbd 115 settarget(1500, 1000, PI/2, true);
narshu 9:377560539b74 116 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 117 Thread::wait(2000);
narshu 9:377560539b74 118
narshu 9:377560539b74 119 // left roll
narshu 13:57ea4e520dbd 120 settarget(500, 1500, PI/2, true);
narshu 9:377560539b74 121 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 122 Thread::wait(2000);
narshu 13:57ea4e520dbd 123
narshu 9:377560539b74 124 // mid
narshu 13:57ea4e520dbd 125 settarget(1500, 1000, PI/2, true);
narshu 9:377560539b74 126 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 127 Thread::wait(2000);
narshu 9:377560539b74 128
narshu 9:377560539b74 129 // map
narshu 13:57ea4e520dbd 130 settarget(1500, 1500, PI/2, true);
narshu 9:377560539b74 131 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 132 Thread::wait(2000);
narshu 13:57ea4e520dbd 133
narshu 9:377560539b74 134
narshu 9:377560539b74 135 // mid
narshu 13:57ea4e520dbd 136 settarget(1500, 1000, -PI/2, true);
narshu 9:377560539b74 137 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 138 Thread::wait(2000);
narshu 13:57ea4e520dbd 139
narshu 9:377560539b74 140
narshu 9:377560539b74 141 // home
narshu 13:57ea4e520dbd 142 settarget(500, 500, 0, true);
narshu 9:377560539b74 143 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 144 Thread::wait(2000);
narshu 13:57ea4e520dbd 145
narshu 13:57ea4e520dbd 146 // oponents base
narshu 13:57ea4e520dbd 147 settarget(2500, 500, 0, true);
narshu 13:57ea4e520dbd 148 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 149 Thread::wait(2000);
narshu 13:57ea4e520dbd 150
narshu 13:57ea4e520dbd 151 // oponents ship
narshu 13:57ea4e520dbd 152 settarget(2500, 1500, 0, true);
narshu 13:57ea4e520dbd 153 Thread::signal_wait(0x01);
narshu 13:57ea4e520dbd 154 Thread::wait(2000);
narshu 13:57ea4e520dbd 155
narshu 13:57ea4e520dbd 156
narshu 17:bafcef1c3579 157 //}
narshu 9:377560539b74 158
narshu 9:377560539b74 159 Thread::signal_wait(0x01);
narshu 9:377560539b74 160 flag_terminate = true;
narshu 9:377560539b74 161 //OLED3 = true;
narshu 9:377560539b74 162
narshu 9:377560539b74 163 while (true) {
narshu 9:377560539b74 164 Thread::wait(osWaitForever);
narshu 9:377560539b74 165 }
narshu 9:377560539b74 166 }
narshu 9:377560539b74 167
narshu 0:f3bf6c7e2283 168
narshu 0:f3bf6c7e2283 169 void vMotorThread(void const *argument) {
narshu 0:f3bf6c7e2283 170 motors.resetEncoders();
narshu 0:f3bf6c7e2283 171 while (1) {
narshu 1:bbabbd997d21 172 motors.setSpeed(20,20);
narshu 1:bbabbd997d21 173 Thread::wait(2000);
narshu 1:bbabbd997d21 174 motors.stop();
narshu 1:bbabbd997d21 175 Thread::wait(5000);
narshu 1:bbabbd997d21 176 motors.setSpeed(-20,-20);
narshu 0:f3bf6c7e2283 177 Thread::wait(2000);
narshu 0:f3bf6c7e2283 178 motors.stop();
narshu 0:f3bf6c7e2283 179 Thread::wait(5000);
narshu 0:f3bf6c7e2283 180 motors.setSpeed(-20,20);
narshu 0:f3bf6c7e2283 181 Thread::wait(2000);
narshu 0:f3bf6c7e2283 182 motors.stop();
narshu 0:f3bf6c7e2283 183 Thread::wait(5000);
narshu 1:bbabbd997d21 184 motors.setSpeed(20,-20);
narshu 1:bbabbd997d21 185 Thread::wait(2000);
narshu 1:bbabbd997d21 186 motors.stop();
narshu 1:bbabbd997d21 187 Thread::wait(5000);
narshu 0:f3bf6c7e2283 188 }
narshu 0:f3bf6c7e2283 189 }
narshu 0:f3bf6c7e2283 190
narshu 0:f3bf6c7e2283 191
narshu 0:f3bf6c7e2283 192 void vPrintState(void const *argument) {
narshu 0:f3bf6c7e2283 193 float state[3];
narshu 1:bbabbd997d21 194 float SonarMeasures[3];
narshu 1:bbabbd997d21 195 float IRMeasures[3];
narshu 1:bbabbd997d21 196
narshu 0:f3bf6c7e2283 197
narshu 0:f3bf6c7e2283 198 while (1) {
narshu 0:f3bf6c7e2283 199 kalman.statelock.lock();
narshu 0:f3bf6c7e2283 200 state[0] = kalman.X(0);
narshu 0:f3bf6c7e2283 201 state[1] = kalman.X(1);
narshu 0:f3bf6c7e2283 202 state[2] = kalman.X(2);
narshu 1:bbabbd997d21 203 SonarMeasures[0] = kalman.SonarMeasures[0];
narshu 1:bbabbd997d21 204 SonarMeasures[1] = kalman.SonarMeasures[1];
narshu 1:bbabbd997d21 205 SonarMeasures[2] = kalman.SonarMeasures[2];
narshu 1:bbabbd997d21 206 IRMeasures[0] = kalman.IRMeasures[0];
narshu 1:bbabbd997d21 207 IRMeasures[1] = kalman.IRMeasures[1];
narshu 1:bbabbd997d21 208 IRMeasures[2] = kalman.IRMeasures[2];
narshu 0:f3bf6c7e2283 209 kalman.statelock.unlock();
narshu 0:f3bf6c7e2283 210 pc.printf("\r\n");
narshu 0:f3bf6c7e2283 211 pc.printf("current: %0.4f %0.4f %0.4f \r\n", state[0], state[1],state[2]);
narshu 1:bbabbd997d21 212 pc.printf("Sonar: %0.4f %0.4f %0.4f \r\n",SonarMeasures[0],SonarMeasures[1],SonarMeasures[2]);
narshu 1:bbabbd997d21 213 pc.printf("IR : %0.4f %0.4f %0.4f \r\n",IRMeasures[0]*180/PI,IRMeasures[1]*180/PI,IRMeasures[2]*180/PI);
narshu 1:bbabbd997d21 214 Thread::wait(100);
narshu 0:f3bf6c7e2283 215 }
narshu 0:f3bf6c7e2283 216 }