James Lam / Mbed 2 deprecated zumy_mbed

Dependencies:   MPU6050IMU QEI RPCInterface mbed

Fork of zumy_mbed by David McPherson

Committer:
jlamyi
Date:
Thu Apr 07 21:17:10 2016 +0000
Revision:
14:a153ce7cc4c8
Parent:
13:af3e8a8189ed
Child:
15:3cee564f9d47
Added test LED

Who changed what in which revision?

UserRevisionLine numberNew contents of line
MichaelW 0:78952cd3935b 1 #include "mbed.h"
abuchan 3:8b5700499eb8 2 #include "SerialRPCInterface.h"
abuchan 2:2c0cd1aaae83 3 #include "MPU6050.h"
DavidMcP555 4:b8eeb59b62d4 4 #include "QEI.h"
abuchan 2:2c0cd1aaae83 5
jlamyi 13:af3e8a8189ed 6 //bool debug = false;
jlamyi 9:6bff048c3b49 7
jlamyi 8:1e47a6b49537 8 extern "C" void mbed_reset();
jlamyi 6:18183b8b08f1 9
abuchan 3:8b5700499eb8 10 SerialRPCInterface SerialRPC(USBTX, USBRX, 115200);
jlamyi 9:6bff048c3b49 11
jlamyi 7:50b807559e1a 12 void GetSensorData(char* input, char* output);
jlamyi 7:50b807559e1a 13 RPCFunction gsd(&GetSensorData, "gsd");
jlamyi 8:1e47a6b49537 14 void Reset(char* input, char* output);
jlamyi 8:1e47a6b49537 15 RPCFunction rst(&Reset, "rst");
jlamyi 13:af3e8a8189ed 16 //void WD_init(char* input, char* output);
jlamyi 13:af3e8a8189ed 17 //RPCFunction wdinit(&WD_init, "wdinit");
abuchan 3:8b5700499eb8 18
jlamyi 8:1e47a6b49537 19 float accel_x, accel_y, accel_z, gyro_x, gyro_y, gyro_z;
jlamyi 8:1e47a6b49537 20 int r_enc, l_enc;
jlamyi 6:18183b8b08f1 21
jlamyi 11:090042660f15 22
DavidMcP555 4:b8eeb59b62d4 23 QEI l_wheel (p29, p30, NC, 624);
DavidMcP555 4:b8eeb59b62d4 24 QEI r_wheel (p11, p12, NC, 624);
abuchan 2:2c0cd1aaae83 25
abuchan 2:2c0cd1aaae83 26 MPU6050 mpu6050;
abuchan 2:2c0cd1aaae83 27
abuchan 2:2c0cd1aaae83 28 DigitalOut init_done(LED1);
abuchan 2:2c0cd1aaae83 29 DigitalOut imu_good(LED2);
abuchan 2:2c0cd1aaae83 30 DigitalOut main_loop(LED3);
jlamyi 14:a153ce7cc4c8 31 DigitalOut main_loop2(LED4);
abuchan 2:2c0cd1aaae83 32
jlamyi 11:090042660f15 33 // Simon's Watchdog code from
jlamyi 11:090042660f15 34 // http://mbed.org/forum/mbed/topic/508/
jlamyi 13:af3e8a8189ed 35 /*
jlamyi 11:090042660f15 36 class Watchdog {
jlamyi 11:090042660f15 37 public:
jlamyi 11:090042660f15 38 // Load timeout value in watchdog timer and enable
jlamyi 11:090042660f15 39 void kick(float s) {
jlamyi 11:090042660f15 40 LPC_WDT->WDCLKSEL = 0x1; // Set CLK src to PCLK
jlamyi 11:090042660f15 41 uint32_t clk = SystemCoreClock / 16; // WD has a fixed /4 prescaler, PCLK default is /4
jlamyi 11:090042660f15 42 LPC_WDT->WDTC = s * (float)clk;
jlamyi 11:090042660f15 43 LPC_WDT->WDMOD = 0x3; // Enabled and Reset
jlamyi 11:090042660f15 44 kick();
jlamyi 11:090042660f15 45 }
jlamyi 11:090042660f15 46 // "kick" or "feed" the dog - reset the watchdog timer
jlamyi 11:090042660f15 47 // by writing this required bit pattern
jlamyi 11:090042660f15 48 void kick() {
jlamyi 11:090042660f15 49 LPC_WDT->WDFEED = 0xAA;
jlamyi 11:090042660f15 50 LPC_WDT->WDFEED = 0x55;
jlamyi 11:090042660f15 51 }
jlamyi 11:090042660f15 52 };
jlamyi 13:af3e8a8189ed 53 */
jlamyi 11:090042660f15 54 // Setup the watchdog timer
jlamyi 13:af3e8a8189ed 55 // Watchdog wdt;
jlamyi 11:090042660f15 56
MichaelW 0:78952cd3935b 57 int main() {
abuchan 2:2c0cd1aaae83 58 init_done = 0;
abuchan 2:2c0cd1aaae83 59 imu_good = 0;
abuchan 2:2c0cd1aaae83 60 main_loop = 0;
jlamyi 11:090042660f15 61
abuchan 2:2c0cd1aaae83 62 //Set up I2C
abuchan 2:2c0cd1aaae83 63 i2c.frequency(400000); // use fast (400 kHz) I2C
abuchan 2:2c0cd1aaae83 64
abuchan 3:8b5700499eb8 65 volatile bool imu_ready = false;
abuchan 3:8b5700499eb8 66
abuchan 3:8b5700499eb8 67 wait_ms(100);
abuchan 2:2c0cd1aaae83 68
abuchan 2:2c0cd1aaae83 69 uint8_t whoami = mpu6050.readByte(MPU6050_ADDRESS, WHO_AM_I_MPU6050);
abuchan 2:2c0cd1aaae83 70
abuchan 2:2c0cd1aaae83 71 if (whoami == 0x68) // WHO_AM_I should always be 0x68
abuchan 2:2c0cd1aaae83 72 {
abuchan 2:2c0cd1aaae83 73 mpu6050.MPU6050SelfTest(SelfTest);
abuchan 2:2c0cd1aaae83 74 if(SelfTest[0] < 1.0f && SelfTest[1] < 1.0f && SelfTest[2] < 1.0f && SelfTest[3] < 1.0f && SelfTest[4] < 1.0f && SelfTest[5] < 1.0f) {
abuchan 2:2c0cd1aaae83 75 mpu6050.resetMPU6050(); // Reset registers to default in preparation for device calibration
abuchan 2:2c0cd1aaae83 76 mpu6050.calibrateMPU6050(gyroBias, accelBias); // Calibrate gyro and accelerometers, load biases in bias registers
abuchan 2:2c0cd1aaae83 77 mpu6050.initMPU6050();
abuchan 2:2c0cd1aaae83 78 mpu6050.getAres();
abuchan 2:2c0cd1aaae83 79 mpu6050.getGres();
abuchan 2:2c0cd1aaae83 80 imu_ready = true;
abuchan 2:2c0cd1aaae83 81 imu_good = 1;
abuchan 2:2c0cd1aaae83 82 }
abuchan 2:2c0cd1aaae83 83 }
abuchan 2:2c0cd1aaae83 84
abuchan 2:2c0cd1aaae83 85 init_done = 1;
abuchan 3:8b5700499eb8 86 uint8_t loop_count = 10;
MichaelW 0:78952cd3935b 87 while(1) {
abuchan 3:8b5700499eb8 88 wait_ms(10);
abuchan 2:2c0cd1aaae83 89
DavidMcP555 4:b8eeb59b62d4 90 // Handle the encoders
jlamyi 8:1e47a6b49537 91 r_enc = r_wheel.getPulses();
jlamyi 8:1e47a6b49537 92 l_enc = l_wheel.getPulses();
jlamyi 11:090042660f15 93 //if(debug) SerialRPC.pc.printf("Pulses are: %i, %i\r\n", l_enc,r_enc);
DavidMcP555 4:b8eeb59b62d4 94
jlamyi 14:a153ce7cc4c8 95
abuchan 3:8b5700499eb8 96 if (!(--loop_count)) {
abuchan 3:8b5700499eb8 97 loop_count = 10;
abuchan 3:8b5700499eb8 98 main_loop = !main_loop;
abuchan 3:8b5700499eb8 99 }
abuchan 2:2c0cd1aaae83 100
abuchan 3:8b5700499eb8 101 if (imu_ready) {
abuchan 2:2c0cd1aaae83 102
abuchan 2:2c0cd1aaae83 103 if(mpu6050.readByte(MPU6050_ADDRESS, INT_STATUS) & 0x01) { // check if data ready interrupt
abuchan 2:2c0cd1aaae83 104 mpu6050.readAccelData(accelCount); // Read the x/y/z adc values
abuchan 2:2c0cd1aaae83 105 mpu6050.readGyroData(gyroCount); // Read the x/y/z adc values
abuchan 2:2c0cd1aaae83 106
abuchan 2:2c0cd1aaae83 107 // Now we'll calculate the accleration value into actual g's
jlamyi 8:1e47a6b49537 108 accel_x = (float)accelCount[0]*aRes - accelBias[0]; // get actual g value, this depends on scale being set
jlamyi 8:1e47a6b49537 109 accel_y = (float)accelCount[1]*aRes - accelBias[1];
jlamyi 8:1e47a6b49537 110 accel_z = (float)accelCount[2]*aRes - accelBias[2];
jlamyi 6:18183b8b08f1 111
abuchan 2:2c0cd1aaae83 112 // Calculate the gyro value into actual degrees per second
jlamyi 8:1e47a6b49537 113 gyro_x = (float)gyroCount[0]*gRes - gyroBias[0]; // get actual gyro value, this depends on scale being set
jlamyi 8:1e47a6b49537 114 gyro_y = (float)gyroCount[1]*gRes - gyroBias[1];
jlamyi 8:1e47a6b49537 115 gyro_z = (float)gyroCount[2]*gRes - gyroBias[2];
abuchan 2:2c0cd1aaae83 116 }
abuchan 2:2c0cd1aaae83 117 }
jlamyi 14:a153ce7cc4c8 118
jlamyi 14:a153ce7cc4c8 119 // for testing where is the bug
jlamyi 14:a153ce7cc4c8 120 main_loop2 = main_loop;
jlamyi 14:a153ce7cc4c8 121
jlamyi 14:a153ce7cc4c8 122 //SerialRPC.pc.printf("Data: %d,%d,%f,%f,%f,%f,%f,%f\r\n", l_enc, r_enc, accel_x, accel_y, accel_z, gyro_x, gyro_y, gyro_z);
MichaelW 0:78952cd3935b 123 }
MichaelW 0:78952cd3935b 124 }
jlamyi 6:18183b8b08f1 125
jlamyi 11:090042660f15 126
jlamyi 11:090042660f15 127 // RPCFunction
jlamyi 6:18183b8b08f1 128 void GetSensorData(char* input, char* output){
jlamyi 14:a153ce7cc4c8 129 sprintf(output, "%d,%d,%f,%f,%f,%f,%f,%f", l_enc, r_enc, accel_x, accel_y, accel_z, gyro_x, gyro_y, gyro_z);
jlamyi 6:18183b8b08f1 130 }
jlamyi 8:1e47a6b49537 131
jlamyi 8:1e47a6b49537 132 void Reset(char* input, char* output){
jlamyi 14:a153ce7cc4c8 133 mbed_reset();
jlamyi 8:1e47a6b49537 134 }
jlamyi 11:090042660f15 135
jlamyi 13:af3e8a8189ed 136 /*
jlamyi 11:090042660f15 137 void WD_init(char* input, char* output){
jlamyi 13:af3e8a8189ed 138 //wdt.kick(0.5);
jlamyi 11:090042660f15 139 }
jlamyi 13:af3e8a8189ed 140 */