Shih-Ho Hsieh / Mbed 2 deprecated pololu5mag_with_platform

5 sensors on a board with a moving platform.

Dependencies:   mbed

Fork of Motor_XYZ_UI_SPI_I2C_5mag by Shih-Ho Hsieh

ui.cpp@13:6850d2b41b2c, 2018-03-02 (annotated)

Committer:
hober
Date:
Fri Mar 02 08:53:17 2018 +0000
Branch:
envelope
Revision:
13:6850d2b41b2c
Parent:
12:2e1b1b1726fa
Child:
14:9672e91010a3
2018/03/02 update

Who changed what in which revision?

UserRevisionLine numberNew contents of line
hober 6:ce02d396c961 1 #include "motor.h"
hober 6:ce02d396c961 2 #include "xyz_sensor_platform.h"
hober 8:33d34a775873 3 //#include "ParseArray.h"
hober 8:33d34a775873 4 #include "envelopetracker.h"
hober 6:ce02d396c961 5
hober 7:ee0569d49c52 6 #define I2C_FREQUENCY 400000
hober 6:ce02d396c961 7
hober 8:33d34a775873 8 typedef unsigned char byte;
hober 10:a49cdab3810f 9 uint8_t* dataToSend;
hober 8:33d34a775873 10 int sendArrayIndex = 0;
hober 8:33d34a775873 11 int sendBufferMax = 10000;
hober 8:33d34a775873 12 const int BAUD = 921600;
hober 8:33d34a775873 13 Serial pc(SERIAL_TX, SERIAL_RX, BAUD );
hober 7:ee0569d49c52 14 DigitalOut led(LED2),led3(LED3);
hober 7:ee0569d49c52 15 DigitalOut mag_test(D11);
hober 12:2e1b1b1726fa 16 InterruptIn button(USER_BUTTON);
hober 6:ce02d396c961 17 XYZSensorPlatform platform;
hober 8:33d34a775873 18 Envelope *command;
hober 8:33d34a775873 19 EnvelopeTracker tracker;
hober 7:ee0569d49c52 20 byte commandToSend[10]= {'H','O','1','2','3','4','5','6','E',0};
hober 7:ee0569d49c52 21 bool isRecording = false;
hober 9:c4f7257dee47 22 Envelope* result;
hober 9:c4f7257dee47 23 float x, y, z;
hober 9:c4f7257dee47 24 float pos[3];
hober 9:c4f7257dee47 25 int n = 0;
hober 9:c4f7257dee47 26 int getMag = 0;
hober 9:c4f7257dee47 27 int leftCount = 0;
hober 9:c4f7257dee47 28 int rightCount = 0;
hober 9:c4f7257dee47 29 int upCount = 0;
hober 9:c4f7257dee47 30 int downCount = 0;
hober 9:c4f7257dee47 31 int forwardCount = 0;
hober 9:c4f7257dee47 32 int backwardCount = 0;
hober 9:c4f7257dee47 33 bool commandToDo = false;
hober 9:c4f7257dee47 34 int recordTime;
hober 9:c4f7257dee47 35 float waitTime;
hober 12:2e1b1b1726fa 36 bool isEcho = false;
hober 12:2e1b1b1726fa 37 bool isReset = true;
hober 12:2e1b1b1726fa 38 enum PatternRecordState{NONE,INIT,MAGNET,TO};
hober 12:2e1b1b1726fa 39 struct Pattern
hober 12:2e1b1b1726fa 40 {
hober 12:2e1b1b1726fa 41 float xStart, xEnd, xStep, yStart, yEnd, yStep, zStart, zEnd, zStep;
hober 12:2e1b1b1726fa 42 int num;
hober 12:2e1b1b1726fa 43 float x, y, z;
hober 12:2e1b1b1726fa 44 int count;
hober 12:2e1b1b1726fa 45 PatternRecordState state;
hober 12:2e1b1b1726fa 46 } pattern;
hober 12:2e1b1b1726fa 47
hober 8:33d34a775873 48
hober 9:c4f7257dee47 49 const int Fs = 1000; // sampling rate -- max: 1kHz
hober 8:33d34a775873 50
hober 8:33d34a775873 51
hober 6:ce02d396c961 52 void echo(char typ, float x, float y, float z);
hober 6:ce02d396c961 53 void echo(char typ, int16_t *p_data);
hober 9:c4f7257dee47 54 void Rx_interrupt();
hober 12:2e1b1b1726fa 55 void released()
hober 12:2e1b1b1726fa 56 {
hober 12:2e1b1b1726fa 57 led = !led;
hober 12:2e1b1b1726fa 58 pc.attach(&Rx_interrupt, Serial::RxIrq);
hober 12:2e1b1b1726fa 59 }
hober 8:33d34a775873 60
hober 6:ce02d396c961 61 int main()
hober 6:ce02d396c961 62 {
hober 6:ce02d396c961 63 led=1;
hober 8:33d34a775873 64 command = new Envelope;
hober 8:33d34a775873 65 command->enableHeader(std::string("H"));// 48 H
hober 8:33d34a775873 66 command->enableFooter(std::string("E"),8);// 45 E
hober 8:33d34a775873 67 command->enableCheckXOR(9);
hober 8:33d34a775873 68 tracker.setEnvelope(*command);
hober 8:33d34a775873 69 tracker.setBufferLength(100);
hober 12:2e1b1b1726fa 70 pattern.state = NONE;
hober 7:ee0569d49c52 71 pc.format(8,SerialBase::None,1);
hober 12:2e1b1b1726fa 72 button.rise(&released);
hober 6:ce02d396c961 73 platform.set_speed(2.5);
hober 12:2e1b1b1726fa 74 // platform.reset(); // need to be modified here
hober 7:ee0569d49c52 75 platform.setSensorI2cFrequency(I2C_FREQUENCY);
hober 9:c4f7257dee47 76 // Setup a serial interrupt function to receive data
hober 9:c4f7257dee47 77 pc.attach(&Rx_interrupt, Serial::RxIrq);
hober 12:2e1b1b1726fa 78 echo('B',0,0,0);
hober 6:ce02d396c961 79 while(1) {
hober 12:2e1b1b1726fa 80 if(isReset){
hober 12:2e1b1b1726fa 81 platform.set_speed(1);
hober 12:2e1b1b1726fa 82 platform.reset();
hober 12:2e1b1b1726fa 83 isReset = false;
hober 12:2e1b1b1726fa 84 platform.set_speed(2.5);
hober 12:2e1b1b1726fa 85 }
hober 12:2e1b1b1726fa 86 if(pattern.state != NONE)
hober 12:2e1b1b1726fa 87 {
hober 13:6850d2b41b2c 88 if(pattern.state == INIT)
hober 13:6850d2b41b2c 89 {
hober 13:6850d2b41b2c 90 pattern.x = pattern.xStart;
hober 13:6850d2b41b2c 91 pattern.y = pattern.yStart;
hober 13:6850d2b41b2c 92 pattern.z = pattern.zStart;
hober 13:6850d2b41b2c 93 pattern.count = pattern.num;
hober 13:6850d2b41b2c 94 pattern.state = MAGNET;
hober 13:6850d2b41b2c 95 isEcho = true;
hober 13:6850d2b41b2c 96 platform.to(pattern.x,pattern.y,pattern.z);
hober 13:6850d2b41b2c 97 }
hober 12:2e1b1b1726fa 98 if(pattern.count == 0) pattern.state = TO;
hober 12:2e1b1b1726fa 99 if(pattern.state == TO)
hober 12:2e1b1b1726fa 100 {
hober 13:6850d2b41b2c 101 if(abs(pattern.x-pattern.xEnd)>abs(pattern.xStep))
hober 13:6850d2b41b2c 102 {
hober 13:6850d2b41b2c 103 pattern.x += pattern.xStep;
hober 13:6850d2b41b2c 104 platform.set_speed(1.5);
hober 13:6850d2b41b2c 105 }
hober 12:2e1b1b1726fa 106 else if(abs(pattern.y-pattern.yEnd)>abs(pattern.yStep)){
hober 12:2e1b1b1726fa 107 pattern.x = pattern.xStart;
hober 12:2e1b1b1726fa 108 pattern.y += pattern.yStep;
hober 12:2e1b1b1726fa 109 }
hober 12:2e1b1b1726fa 110 else if(abs(pattern.z-pattern.zEnd)>abs(pattern.zStep)){
hober 12:2e1b1b1726fa 111 pattern.x = pattern.xStart;
hober 12:2e1b1b1726fa 112 pattern.y = pattern.yStart;
hober 12:2e1b1b1726fa 113 pattern.z += pattern.zStep;
hober 13:6850d2b41b2c 114 platform.to(0,0,0);
hober 13:6850d2b41b2c 115 platform.set_speed(1);
hober 13:6850d2b41b2c 116 platform.reset();
hober 13:6850d2b41b2c 117 platform.set_speed(2.5);
hober 12:2e1b1b1726fa 118 }
hober 12:2e1b1b1726fa 119 else{
hober 12:2e1b1b1726fa 120 pattern.state = NONE;
hober 13:6850d2b41b2c 121 echo('S',0,0,0);
hober 12:2e1b1b1726fa 122 continue;
hober 12:2e1b1b1726fa 123 }
hober 12:2e1b1b1726fa 124 platform.to(pattern.x,pattern.y,pattern.z);
hober 12:2e1b1b1726fa 125 isEcho = true;
hober 12:2e1b1b1726fa 126 pattern.count = pattern.num;
hober 12:2e1b1b1726fa 127 pattern.state = MAGNET;
hober 13:6850d2b41b2c 128 platform.set_speed(2.5);
hober 12:2e1b1b1726fa 129 }
hober 12:2e1b1b1726fa 130 if(pattern.state == MAGNET)
hober 12:2e1b1b1726fa 131 {
hober 12:2e1b1b1726fa 132 if(pattern.count-- <= 0) pattern.state = TO;
hober 12:2e1b1b1726fa 133 getMag++;
hober 13:6850d2b41b2c 134 // wait(0.01);
hober 12:2e1b1b1726fa 135 }
hober 12:2e1b1b1726fa 136 }
hober 12:2e1b1b1726fa 137 if(isEcho)
hober 12:2e1b1b1726fa 138 {
hober 12:2e1b1b1726fa 139 platform.position(pos);
hober 12:2e1b1b1726fa 140 echo('O',pos[0],pos[1],pos[2]);
hober 12:2e1b1b1726fa 141 led3 = !led3;
hober 12:2e1b1b1726fa 142 isEcho = false;
hober 12:2e1b1b1726fa 143 }
hober 8:33d34a775873 144 if(isRecording && n < recordTime) {
hober 7:ee0569d49c52 145 int16_t mag[3];
hober 7:ee0569d49c52 146 if(platform.get_mag_raw(mag)==0&&pc.writeable()) {
hober 7:ee0569d49c52 147 echo('M',mag);
hober 7:ee0569d49c52 148 mag_test=!mag_test;
hober 8:33d34a775873 149 wait(waitTime);
hober 7:ee0569d49c52 150 }
hober 7:ee0569d49c52 151 n++;
hober 8:33d34a775873 152 } else if(isRecording) {
hober 8:33d34a775873 153 isRecording = false;
hober 8:33d34a775873 154 echo('S',0,0,0);
hober 8:33d34a775873 155 }// end recording if
hober 9:c4f7257dee47 156 if(getMag>0) {
hober 9:c4f7257dee47 157 int16_t mag[3];
hober 9:c4f7257dee47 158 if(platform.get_mag_raw(mag)==0) echo('M',mag);
hober 9:c4f7257dee47 159 getMag--;
hober 9:c4f7257dee47 160 }
hober 9:c4f7257dee47 161 if(commandToDo) {
hober 9:c4f7257dee47 162 platform.to(x,y,z);
hober 9:c4f7257dee47 163 platform.position(pos);
hober 9:c4f7257dee47 164 echo('O',pos[0],pos[1],pos[2]);
hober 9:c4f7257dee47 165 commandToDo = false;
hober 9:c4f7257dee47 166 }
hober 9:c4f7257dee47 167 if(leftCount > 0||rightCount > 0||upCount > 0||downCount > 0||forwardCount > 0||backwardCount > 0) {
hober 9:c4f7257dee47 168 if(leftCount > 0) {
hober 9:c4f7257dee47 169 platform.go_left();
hober 9:c4f7257dee47 170 leftCount--;
hober 9:c4f7257dee47 171 }
hober 9:c4f7257dee47 172 if(rightCount > 0) {
hober 9:c4f7257dee47 173 platform.go_right();
hober 9:c4f7257dee47 174 rightCount--;
hober 9:c4f7257dee47 175 }
hober 9:c4f7257dee47 176 if(upCount > 0) {
hober 9:c4f7257dee47 177 platform.go_up();
hober 9:c4f7257dee47 178 upCount--;
hober 9:c4f7257dee47 179 }
hober 9:c4f7257dee47 180 if(downCount > 0) {
hober 9:c4f7257dee47 181 platform.go_down();
hober 9:c4f7257dee47 182 downCount--;
hober 9:c4f7257dee47 183 }
hober 9:c4f7257dee47 184 if(forwardCount > 0) {
hober 9:c4f7257dee47 185 platform.go_forward();
hober 9:c4f7257dee47 186 forwardCount--;
hober 9:c4f7257dee47 187 }
hober 9:c4f7257dee47 188 if(backwardCount > 0) {
hober 9:c4f7257dee47 189 platform.go_backward();
hober 9:c4f7257dee47 190 backwardCount--;
hober 9:c4f7257dee47 191 }
hober 9:c4f7257dee47 192 platform.position(pos);
hober 9:c4f7257dee47 193 echo('O',pos[0],pos[1],pos[2]);
hober 9:c4f7257dee47 194 }
hober 7:ee0569d49c52 195 } // end while
hober 6:ce02d396c961 196
hober 6:ce02d396c961 197 }
hober 6:ce02d396c961 198
hober 6:ce02d396c961 199 void echo(char typ,float x, float y, float z)
hober 6:ce02d396c961 200 {
hober 7:ee0569d49c52 201 int16_t p_data[3]= {(int16_t)(x*10), int16_t(y*10), int16_t(z*10)};
hober 7:ee0569d49c52 202 echo(typ,p_data);
hober 6:ce02d396c961 203 }
hober 6:ce02d396c961 204
hober 6:ce02d396c961 205 void echo(char typ, int16_t *p_data)
hober 6:ce02d396c961 206 {
hober 8:33d34a775873 207 char tmp[] = {typ, p_data[0]>>8, p_data[0], p_data[1]>>8, p_data[1], p_data[2]>>8, p_data[2]};
hober 8:33d34a775873 208 command->setEnvelopeData(tmp,7);
hober 10:a49cdab3810f 209 dataToSend = (uint8_t*)(command->getEnvelopeArray());
hober 8:33d34a775873 210 for(int i = 0; i < command->length(); i++) {
hober 10:a49cdab3810f 211 pc.putc(dataToSend[i]);
hober 8:33d34a775873 212 }
hober 6:ce02d396c961 213 }
hober 6:ce02d396c961 214
hober 9:c4f7257dee47 215 void Rx_interrupt()
hober 9:c4f7257dee47 216 {
hober 9:c4f7257dee47 217 char c;
hober 9:c4f7257dee47 218 while(pc.readable()) {
hober 9:c4f7257dee47 219 c = pc.getc();
hober 9:c4f7257dee47 220 tracker.parse(&c,1);
hober 6:ce02d396c961 221
hober 9:c4f7257dee47 222 result = tracker.getEnvelope();
hober 9:c4f7257dee47 223 if(result!=NULL) {
hober 9:c4f7257dee47 224 char *dataArray = result->getPayload();
hober 9:c4f7257dee47 225 switch(dataArray[0]) {
hober 12:2e1b1b1726fa 226 case 'I': // pattern record
hober 12:2e1b1b1726fa 227 if(dataArray[1] == 'X')
hober 12:2e1b1b1726fa 228 {
hober 12:2e1b1b1726fa 229 pattern.xStart = (float)((dataArray[2]<<8)+dataArray[3])/10.0f;
hober 12:2e1b1b1726fa 230 pattern.xEnd = (float)((dataArray[4]<<8)+dataArray[5])/10.0f;
hober 12:2e1b1b1726fa 231 pattern.xStep = (float) dataArray[6]/10.0f;
hober 12:2e1b1b1726fa 232 if((pattern.xStart-pattern.xEnd)*pattern.xStep>0) pattern.xStep = -pattern.xStep;
hober 12:2e1b1b1726fa 233 }
hober 12:2e1b1b1726fa 234 else if(dataArray[1] == 'Y')
hober 12:2e1b1b1726fa 235 {
hober 12:2e1b1b1726fa 236 pattern.yStart = (float)((dataArray[2]<<8)+dataArray[3])/10.0f;
hober 12:2e1b1b1726fa 237 pattern.yEnd = (float)((dataArray[4]<<8)+dataArray[5])/10.0f;
hober 12:2e1b1b1726fa 238 pattern.yStep = (float) dataArray[6]/10.0f;
hober 12:2e1b1b1726fa 239 if((pattern.yStart-pattern.yEnd)*pattern.yStep>0) pattern.yStep = -pattern.yStep;
hober 12:2e1b1b1726fa 240 }
hober 12:2e1b1b1726fa 241 else if(dataArray[1] == 'Z')
hober 12:2e1b1b1726fa 242 {
hober 12:2e1b1b1726fa 243 pattern.zStart = (float)((dataArray[2]<<8)+dataArray[3])/10.0f;
hober 12:2e1b1b1726fa 244 pattern.zEnd = (float)((dataArray[4]<<8)+dataArray[5])/10.0f;
hober 12:2e1b1b1726fa 245 pattern.zStep = (float) dataArray[6]/10.0f;
hober 12:2e1b1b1726fa 246 if((pattern.zStart-pattern.zEnd)*pattern.zStep>0) pattern.zStep = -pattern.zStep;
hober 12:2e1b1b1726fa 247 }
hober 12:2e1b1b1726fa 248 else if(dataArray[1] == 'N')
hober 12:2e1b1b1726fa 249 {
hober 12:2e1b1b1726fa 250 pattern.num = (int)dataArray[2];
hober 12:2e1b1b1726fa 251 }
hober 12:2e1b1b1726fa 252 else if(dataArray[1] == 'M')
hober 12:2e1b1b1726fa 253 {
hober 12:2e1b1b1726fa 254 if(dataArray[2] == 'B'){
hober 12:2e1b1b1726fa 255 pattern.x = (float)((dataArray[3]<<8)+dataArray[4])/10.0f;
hober 12:2e1b1b1726fa 256 pattern.y = (float)((dataArray[5]<<8)+dataArray[6])/10.0f;
hober 12:2e1b1b1726fa 257 }
hober 12:2e1b1b1726fa 258 if(dataArray[2] == 'U'){
hober 12:2e1b1b1726fa 259 pattern.z = (float)((dataArray[3]<<8)+dataArray[4])/10.0f;
hober 12:2e1b1b1726fa 260 pattern.state = TO;
hober 12:2e1b1b1726fa 261 printf("%.2f %.2f %.2f\n %.2f %.2f %.2f\n %.2f %.2f %.2f\n %.2f %.2f %.2f\n %d\n",
hober 12:2e1b1b1726fa 262 pattern.xStart, pattern.xEnd, pattern.xStep, pattern.yStart, pattern.yEnd, pattern.yStep, pattern.zStart, pattern.zEnd, pattern.zStep, pattern.x, pattern.y, pattern.z, pattern.num);
hober 12:2e1b1b1726fa 263 }
hober 12:2e1b1b1726fa 264 }
hober 12:2e1b1b1726fa 265
hober 12:2e1b1b1726fa 266 else if(dataArray[1] == 'B')
hober 12:2e1b1b1726fa 267 {
hober 12:2e1b1b1726fa 268 pattern.state = INIT;
hober 12:2e1b1b1726fa 269 }
hober 12:2e1b1b1726fa 270 else if(dataArray[1] == 'S') pattern.state = NONE;
hober 12:2e1b1b1726fa 271
hober 9:c4f7257dee47 272 break;
hober 9:c4f7257dee47 273 case 'O': // echo
hober 12:2e1b1b1726fa 274 isEcho = true;
hober 9:c4f7257dee47 275 break;
hober 9:c4f7257dee47 276 case 'C': // command
hober 12:2e1b1b1726fa 277 if(commandToDo) break;
hober 9:c4f7257dee47 278 x=(float)((dataArray[1]<<8)+dataArray[2])/10.0f;
hober 9:c4f7257dee47 279 y=(float)((dataArray[3]<<8)+dataArray[4])/10.0f;
hober 9:c4f7257dee47 280 z=(float)((dataArray[5]<<8)+dataArray[6])/10.0f;
hober 9:c4f7257dee47 281 commandToDo = true;
hober 9:c4f7257dee47 282 break;
hober 9:c4f7257dee47 283 case 'X':
hober 9:c4f7257dee47 284 if(dataArray[1]&0x80) rightCount++;
hober 9:c4f7257dee47 285 else leftCount++;
hober 9:c4f7257dee47 286 break;
hober 9:c4f7257dee47 287 case 'Y':
hober 9:c4f7257dee47 288 if(dataArray[1]&0x80) forwardCount++;
hober 10:a49cdab3810f 289 else backwardCount++;
hober 9:c4f7257dee47 290 break;
hober 9:c4f7257dee47 291 case 'Z':
hober 9:c4f7257dee47 292 if(dataArray[1]&0x80) upCount++;
hober 9:c4f7257dee47 293 else downCount++;
hober 9:c4f7257dee47 294 break;
hober 9:c4f7257dee47 295 case 'M': // magnet
hober 9:c4f7257dee47 296 getMag++;
hober 9:c4f7257dee47 297 mag_test=!mag_test;
hober 9:c4f7257dee47 298 break;
hober 9:c4f7257dee47 299 case 'R': // record
hober 9:c4f7257dee47 300 recordTime = dataArray[1];
hober 9:c4f7257dee47 301 recordTime *= Fs;
hober 11:bb86ffb378f4 302 waitTime = 1.0/Fs-0.00052-1/(BAUD/8/10);
hober 10:a49cdab3810f 303 n = 0;
hober 9:c4f7257dee47 304 if(waitTime < 0) waitTime = 0;
hober 9:c4f7257dee47 305 isRecording = true;
hober 9:c4f7257dee47 306 break;
hober 9:c4f7257dee47 307 case 'S': // stop
hober 9:c4f7257dee47 308 isRecording = false;
hober 9:c4f7257dee47 309 break;
hober 12:2e1b1b1726fa 310 case 'P': // set position
hober 12:2e1b1b1726fa 311 x=(float)((dataArray[1]<<8)+dataArray[2])/10.0f;
hober 12:2e1b1b1726fa 312 y=(float)((dataArray[3]<<8)+dataArray[4])/10.0f;
hober 12:2e1b1b1726fa 313 z=(float)((dataArray[5]<<8)+dataArray[6])/10.0f;
hober 12:2e1b1b1726fa 314 platform.setPosition(x,y,z);
hober 12:2e1b1b1726fa 315 break;
hober 12:2e1b1b1726fa 316 case 'N': // new set up
hober 12:2e1b1b1726fa 317 isReset = true;
hober 12:2e1b1b1726fa 318 break;
hober 9:c4f7257dee47 319 default:
hober 9:c4f7257dee47 320 break;
hober 9:c4f7257dee47 321 } // end switch
hober 9:c4f7257dee47 322 result = NULL;
hober 9:c4f7257dee47 323 dataArray = NULL;
hober 9:c4f7257dee47 324 } // end result if
hober 9:c4f7257dee47 325 } // end parsing if
hober 9:c4f7257dee47 326 }
hober 9:c4f7257dee47 327