Important changes to repositories hosted on mbed.com
Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.
To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.
It is also possible to export all your personal repositories from the account settings page.
Dependencies: MatrixMath Matrix ExperimentServer QEI_pmw MotorShield
main.cpp@38:8ce2f6edba26, 2020-11-17 (annotated)
- Committer:
- uriel_magana
- Date:
- Tue Nov 17 09:11:44 2020 +0000
- Revision:
- 38:8ce2f6edba26
- Parent:
- 37:48a3017a958f
- Child:
- 39:c8da22fcf4f3
no
Who changed what in which revision?
| User | Revision | Line number | New contents of line |
|---|---|---|---|
| pwensing | 0:43448bf056e8 | 1 | #include "mbed.h" |
| pwensing | 0:43448bf056e8 | 2 | #include "rtos.h" |
| pwensing | 0:43448bf056e8 | 3 | #include "EthernetInterface.h" |
| pwensing | 0:43448bf056e8 | 4 | #include "ExperimentServer.h" |
| pwensing | 0:43448bf056e8 | 5 | #include "QEI.h" |
| saloutos | 16:f9ea2b2d410f | 6 | #include "BezierCurve.h" |
| elijahsj | 6:1faceb53dabe | 7 | #include "MotorShield.h" |
| elijahsj | 13:3a1f4e09789b | 8 | #include "HardwareSetup.h" |
| pwensing | 0:43448bf056e8 | 9 | |
| saloutos | 16:f9ea2b2d410f | 10 | #define BEZIER_ORDER_FOOT 7 |
| uriel_magana | 38:8ce2f6edba26 | 11 | #define BEZIER_ORDER_TORQUE 3 /// if x -> plots x+1 points, so should have x+1 Bezier points for T1 and T2 in MATLAB |
| dgdiaz | 35:88dbfefc1bbb | 12 | //#define NUM_INPUTS (14 + 2*(BEZIER_ORDER_FOOT+1)) |
| dgdiaz | 35:88dbfefc1bbb | 13 | #define NUM_INPUTS (14 + 2*(BEZIER_ORDER_TORQUE+1)) |
| saloutos | 16:f9ea2b2d410f | 14 | #define NUM_OUTPUTS 19 |
| pwensing | 0:43448bf056e8 | 15 | |
| saloutos | 16:f9ea2b2d410f | 16 | #define PULSE_TO_RAD (2.0f*3.14159f / 1200.0f) |
| saloutos | 16:f9ea2b2d410f | 17 | |
| saloutos | 16:f9ea2b2d410f | 18 | // Initializations |
| pwensing | 0:43448bf056e8 | 19 | Serial pc(USBTX, USBRX); // USB Serial Terminal |
| pwensing | 0:43448bf056e8 | 20 | ExperimentServer server; // Object that lets us communicate with MATLAB |
| elijahsj | 5:1ab9b2527794 | 21 | Timer t; // Timer to measure elapsed time of experiment |
| elijahsj | 5:1ab9b2527794 | 22 | |
| dgdiaz | 35:88dbfefc1bbb | 23 | QEI encoderA(PE_9, PE_11, NC, 1200, QEI::X4_ENCODING); // MOTOR A ENCODER (no index, 1200 counts/rev, Quadrature encoding) |
| elijahsj | 5:1ab9b2527794 | 24 | QEI encoderB(PA_5, PB_3, NC, 1200, QEI::X4_ENCODING); // MOTOR B ENCODER (no index, 1200 counts/rev, Quadrature encoding) |
| elijahsj | 5:1ab9b2527794 | 25 | QEI encoderC(PC_6, PC_7, NC, 1200, QEI::X4_ENCODING); // MOTOR C ENCODER (no index, 1200 counts/rev, Quadrature encoding) |
| elijahsj | 5:1ab9b2527794 | 26 | QEI encoderD(PD_12, PD_13, NC, 1200, QEI::X4_ENCODING);// MOTOR D ENCODER (no index, 1200 counts/rev, Quadrature encoding) |
| elijahsj | 5:1ab9b2527794 | 27 | |
| elijahsj | 12:84a6dcb60422 | 28 | MotorShield motorShield(12000); //initialize the motor shield with a period of 12000 ticks or ~20kHZ |
| saloutos | 16:f9ea2b2d410f | 29 | Ticker currentLoop; |
| saloutos | 16:f9ea2b2d410f | 30 | |
| saloutos | 16:f9ea2b2d410f | 31 | // Variables for q1 |
| saloutos | 16:f9ea2b2d410f | 32 | float current1; |
| saloutos | 16:f9ea2b2d410f | 33 | float current_des1 = 0; |
| saloutos | 16:f9ea2b2d410f | 34 | float prev_current_des1 = 0; |
| saloutos | 16:f9ea2b2d410f | 35 | float current_int1 = 0; |
| saloutos | 16:f9ea2b2d410f | 36 | float angle1; |
| dgdiaz | 34:9a24d0f718ac | 37 | float angle_des1; |
| saloutos | 16:f9ea2b2d410f | 38 | float velocity1; |
| dgdiaz | 34:9a24d0f718ac | 39 | float velocity_des1; |
| saloutos | 16:f9ea2b2d410f | 40 | float duty_cycle1; |
| saloutos | 16:f9ea2b2d410f | 41 | float angle1_init; |
| saloutos | 16:f9ea2b2d410f | 42 | |
| saloutos | 16:f9ea2b2d410f | 43 | // Variables for q2 |
| saloutos | 16:f9ea2b2d410f | 44 | float current2; |
| saloutos | 16:f9ea2b2d410f | 45 | float current_des2 = 0; |
| saloutos | 16:f9ea2b2d410f | 46 | float prev_current_des2 = 0; |
| saloutos | 16:f9ea2b2d410f | 47 | float current_int2 = 0; |
| saloutos | 16:f9ea2b2d410f | 48 | float angle2; |
| dgdiaz | 34:9a24d0f718ac | 49 | float angle_des2; |
| saloutos | 16:f9ea2b2d410f | 50 | float velocity2; |
| dgdiaz | 34:9a24d0f718ac | 51 | float velocity_des2; |
| saloutos | 16:f9ea2b2d410f | 52 | float duty_cycle2; |
| saloutos | 16:f9ea2b2d410f | 53 | float angle2_init; |
| saloutos | 16:f9ea2b2d410f | 54 | |
| dgdiaz | 34:9a24d0f718ac | 55 | // Fixed kinematic parameters |
| uriel_magana | 36:a3beb771d8f7 | 56 | //const float l_OA=.011; |
| uriel_magana | 36:a3beb771d8f7 | 57 | //const float l_OB=.042; |
| uriel_magana | 36:a3beb771d8f7 | 58 | //const float l_AC=.096; |
| uriel_magana | 36:a3beb771d8f7 | 59 | //const float l_DE=.091; |
| suribe | 32:c60a5d33cd79 | 60 | |
| dgdiaz | 34:9a24d0f718ac | 61 | float l1; //=0.084125; |
| dgdiaz | 34:9a24d0f718ac | 62 | float l2; //=0.084125; |
| saloutos | 16:f9ea2b2d410f | 63 | |
| saloutos | 16:f9ea2b2d410f | 64 | // Timing parameters |
| saloutos | 16:f9ea2b2d410f | 65 | float current_control_period_us = 200.0f; // 5kHz current control loop |
| saloutos | 16:f9ea2b2d410f | 66 | float impedance_control_period_us = 1000.0f; // 1kHz impedance control loop |
| saloutos | 16:f9ea2b2d410f | 67 | float start_period, traj_period, end_period; |
| saloutos | 16:f9ea2b2d410f | 68 | |
| saloutos | 16:f9ea2b2d410f | 69 | // Control parameters |
| saloutos | 19:562c08086d71 | 70 | float current_Kp = 4.0f; |
| saloutos | 19:562c08086d71 | 71 | float current_Ki = 0.4f; |
| saloutos | 19:562c08086d71 | 72 | float current_int_max = 3.0f; |
| saloutos | 16:f9ea2b2d410f | 73 | float duty_max; |
| saloutos | 16:f9ea2b2d410f | 74 | float K_xx; |
| saloutos | 16:f9ea2b2d410f | 75 | float K_yy; |
| saloutos | 16:f9ea2b2d410f | 76 | float K_xy; |
| saloutos | 16:f9ea2b2d410f | 77 | float D_xx; |
| saloutos | 16:f9ea2b2d410f | 78 | float D_xy; |
| saloutos | 16:f9ea2b2d410f | 79 | float D_yy; |
| saloutos | 16:f9ea2b2d410f | 80 | |
| saloutos | 16:f9ea2b2d410f | 81 | // Model parameters |
| saloutos | 17:1bb5aa45826e | 82 | float supply_voltage = 12; // motor supply voltage |
| saloutos | 18:21c8d94eddee | 83 | float R = 2.0f; // motor resistance |
| saloutos | 18:21c8d94eddee | 84 | float k_t = 0.18f; // motor torque constant |
| saloutos | 17:1bb5aa45826e | 85 | float nu = 0.0005; // motor viscous friction |
| saloutos | 16:f9ea2b2d410f | 86 | |
| saloutos | 16:f9ea2b2d410f | 87 | // Current control interrupt function |
| saloutos | 16:f9ea2b2d410f | 88 | void CurrentLoop() |
| saloutos | 16:f9ea2b2d410f | 89 | { |
| saloutos | 19:562c08086d71 | 90 | // This loop sets the motor voltage commands using PI current controllers with feedforward terms. |
| saloutos | 16:f9ea2b2d410f | 91 | |
| saloutos | 16:f9ea2b2d410f | 92 | //use the motor shield as follows: |
| saloutos | 16:f9ea2b2d410f | 93 | //motorShield.motorAWrite(DUTY CYCLE, DIRECTION), DIRECTION = 0 is forward, DIRECTION =1 is backwards. |
| saloutos | 16:f9ea2b2d410f | 94 | |
| uriel_magana | 38:8ce2f6edba26 | 95 | current1 = -(((float(motorShield.readCurrentA())/65536.0f)*30.0f)-15.0f); // measure current DO WE NEED TO ADJUST GEAR RATIO HERE? |
| saloutos | 18:21c8d94eddee | 96 | velocity1 = encoderA.getVelocity() * PULSE_TO_RAD; // measure velocity |
| saloutos | 18:21c8d94eddee | 97 | float err_c1 = current_des1 - current1; // current errror |
| saloutos | 18:21c8d94eddee | 98 | current_int1 += err_c1; // integrate error |
| saloutos | 18:21c8d94eddee | 99 | current_int1 = fmaxf( fminf(current_int1, current_int_max), -current_int_max); // anti-windup |
| saloutos | 18:21c8d94eddee | 100 | float ff1 = R*current_des1 + k_t*velocity1; // feedforward terms |
| saloutos | 18:21c8d94eddee | 101 | duty_cycle1 = (ff1 + current_Kp*err_c1 + current_Ki*current_int1)/supply_voltage; // PI current controller |
| saloutos | 16:f9ea2b2d410f | 102 | |
| saloutos | 16:f9ea2b2d410f | 103 | float absDuty1 = abs(duty_cycle1); |
| saloutos | 16:f9ea2b2d410f | 104 | if (absDuty1 > duty_max) { |
| saloutos | 16:f9ea2b2d410f | 105 | duty_cycle1 *= duty_max / absDuty1; |
| saloutos | 16:f9ea2b2d410f | 106 | absDuty1 = duty_max; |
| saloutos | 16:f9ea2b2d410f | 107 | } |
| saloutos | 16:f9ea2b2d410f | 108 | if (duty_cycle1 < 0) { // backwards |
| saloutos | 16:f9ea2b2d410f | 109 | motorShield.motorAWrite(absDuty1, 1); |
| saloutos | 16:f9ea2b2d410f | 110 | } else { // forwards |
| saloutos | 16:f9ea2b2d410f | 111 | motorShield.motorAWrite(absDuty1, 0); |
| saloutos | 16:f9ea2b2d410f | 112 | } |
| saloutos | 16:f9ea2b2d410f | 113 | prev_current_des1 = current_des1; |
| saloutos | 16:f9ea2b2d410f | 114 | |
| dgdiaz | 34:9a24d0f718ac | 115 | current2 = -(((float(motorShield.readCurrentB())/65536.0f)*30.0f)-15.0f); // measure current |
| saloutos | 18:21c8d94eddee | 116 | velocity2 = encoderB.getVelocity() * PULSE_TO_RAD; // measure velocity |
| saloutos | 18:21c8d94eddee | 117 | float err_c2 = current_des2 - current2; // current error |
| saloutos | 18:21c8d94eddee | 118 | current_int2 += err_c2; // integrate error |
| saloutos | 18:21c8d94eddee | 119 | current_int2 = fmaxf( fminf(current_int2, current_int_max), -current_int_max); // anti-windup |
| saloutos | 18:21c8d94eddee | 120 | float ff2 = R*current_des2 + k_t*velocity2; // feedforward terms |
| saloutos | 18:21c8d94eddee | 121 | duty_cycle2 = (ff2 + current_Kp*err_c2 + current_Ki*current_int2)/supply_voltage; // PI current controller |
| saloutos | 16:f9ea2b2d410f | 122 | |
| saloutos | 16:f9ea2b2d410f | 123 | float absDuty2 = abs(duty_cycle2); |
| saloutos | 16:f9ea2b2d410f | 124 | if (absDuty2 > duty_max) { |
| saloutos | 16:f9ea2b2d410f | 125 | duty_cycle2 *= duty_max / absDuty2; |
| saloutos | 16:f9ea2b2d410f | 126 | absDuty2 = duty_max; |
| saloutos | 16:f9ea2b2d410f | 127 | } |
| saloutos | 16:f9ea2b2d410f | 128 | if (duty_cycle2 < 0) { // backwards |
| saloutos | 16:f9ea2b2d410f | 129 | motorShield.motorBWrite(absDuty2, 1); |
| saloutos | 16:f9ea2b2d410f | 130 | } else { // forwards |
| saloutos | 16:f9ea2b2d410f | 131 | motorShield.motorBWrite(absDuty2, 0); |
| saloutos | 16:f9ea2b2d410f | 132 | } |
| saloutos | 16:f9ea2b2d410f | 133 | prev_current_des2 = current_des2; |
| dgdiaz | 34:9a24d0f718ac | 134 | |
| saloutos | 16:f9ea2b2d410f | 135 | } |
| elijahsj | 6:1faceb53dabe | 136 | |
| elijahsj | 4:7a1b35f081bb | 137 | int main (void) |
| elijahsj | 4:7a1b35f081bb | 138 | { |
| dgdiaz | 34:9a24d0f718ac | 139 | |
| saloutos | 17:1bb5aa45826e | 140 | // Object for 7th order Cartesian foot trajectory |
| saloutos | 17:1bb5aa45826e | 141 | BezierCurve rDesFoot_bez(2,BEZIER_ORDER_FOOT); |
| dgdiaz | 35:88dbfefc1bbb | 142 | BezierCurve torque_bez(2,BEZIER_ORDER_TORQUE); |
| saloutos | 17:1bb5aa45826e | 143 | |
| pwensing | 0:43448bf056e8 | 144 | // Link the terminal with our server and start it up |
| pwensing | 0:43448bf056e8 | 145 | server.attachTerminal(pc); |
| pwensing | 0:43448bf056e8 | 146 | server.init(); |
| elijahsj | 13:3a1f4e09789b | 147 | |
| pwensing | 0:43448bf056e8 | 148 | // Continually get input from MATLAB and run experiments |
| pwensing | 0:43448bf056e8 | 149 | float input_params[NUM_INPUTS]; |
| elijahsj | 5:1ab9b2527794 | 150 | pc.printf("%f",input_params[0]); |
| elijahsj | 5:1ab9b2527794 | 151 | |
| pwensing | 0:43448bf056e8 | 152 | while(1) { |
| dgdiaz | 34:9a24d0f718ac | 153 | |
| saloutos | 16:f9ea2b2d410f | 154 | // If there are new inputs, this code will run |
| pwensing | 0:43448bf056e8 | 155 | if (server.getParams(input_params,NUM_INPUTS)) { |
| dgdiaz | 34:9a24d0f718ac | 156 | |
| dgdiaz | 34:9a24d0f718ac | 157 | |
| suribe | 32:c60a5d33cd79 | 158 | // Get inputs from MATLAB |
| dgdiaz | 34:9a24d0f718ac | 159 | start_period = input_params[0]; // First buffer time, before trajectory |
| dgdiaz | 34:9a24d0f718ac | 160 | traj_period = input_params[1]; // Trajectory time/length |
| dgdiaz | 34:9a24d0f718ac | 161 | end_period = input_params[2]; // Second buffer time, after trajectory |
| saloutos | 16:f9ea2b2d410f | 162 | |
| dgdiaz | 34:9a24d0f718ac | 163 | angle1_init = input_params[3]; // Initial angle for q1 (rad) |
| dgdiaz | 34:9a24d0f718ac | 164 | angle2_init = input_params[4]; // Initial angle for q2 (rad) |
| elijahsj | 4:7a1b35f081bb | 165 | |
| dgdiaz | 34:9a24d0f718ac | 166 | K_xx = input_params[5]; // Foot stiffness N/m |
| dgdiaz | 34:9a24d0f718ac | 167 | K_yy = input_params[6]; // Foot stiffness N/m |
| dgdiaz | 34:9a24d0f718ac | 168 | K_xy = input_params[7]; // Foot stiffness N/m |
| dgdiaz | 34:9a24d0f718ac | 169 | D_xx = input_params[8]; // Foot damping N/(m/s) |
| dgdiaz | 34:9a24d0f718ac | 170 | D_yy = input_params[9]; // Foot damping N/(m/s) |
| dgdiaz | 34:9a24d0f718ac | 171 | D_xy = input_params[10]; // Foot damping N/(m/s) |
| dgdiaz | 34:9a24d0f718ac | 172 | duty_max = input_params[11]; // Maximum duty factor |
| uriel_magana | 36:a3beb771d8f7 | 173 | l1 = input_params[12]; // Length of 1st link |
| uriel_magana | 36:a3beb771d8f7 | 174 | l2 = input_params[13]; // Length of 2nd link |
| saloutos | 16:f9ea2b2d410f | 175 | |
| saloutos | 19:562c08086d71 | 176 | // Get foot trajectory points |
| saloutos | 16:f9ea2b2d410f | 177 | float foot_pts[2*(BEZIER_ORDER_FOOT+1)]; |
| saloutos | 16:f9ea2b2d410f | 178 | for(int i = 0; i<2*(BEZIER_ORDER_FOOT+1);i++) { |
| uriel_magana | 36:a3beb771d8f7 | 179 | foot_pts[i] = input_params[14+i]; // Bezier curve points |
| saloutos | 16:f9ea2b2d410f | 180 | } |
| saloutos | 16:f9ea2b2d410f | 181 | rDesFoot_bez.setPoints(foot_pts); |
| saloutos | 16:f9ea2b2d410f | 182 | |
| dgdiaz | 35:88dbfefc1bbb | 183 | // Get torque traj points |
| dgdiaz | 35:88dbfefc1bbb | 184 | float torque_pts[2*(BEZIER_ORDER_FOOT+1)]; |
| dgdiaz | 35:88dbfefc1bbb | 185 | for(int i = 0; i<2*(BEZIER_ORDER_FOOT+1);i++) { |
| dgdiaz | 35:88dbfefc1bbb | 186 | torque_pts[i] = input_params[14+i]; |
| dgdiaz | 35:88dbfefc1bbb | 187 | } |
| dgdiaz | 35:88dbfefc1bbb | 188 | torque_bez.setPoints(torque_pts); |
| dgdiaz | 37:48a3017a958f | 189 | float torque_des[2]; |
| dgdiaz | 35:88dbfefc1bbb | 190 | |
| saloutos | 16:f9ea2b2d410f | 191 | // Attach current loop interrupt |
| saloutos | 16:f9ea2b2d410f | 192 | currentLoop.attach_us(CurrentLoop,current_control_period_us); |
| saloutos | 16:f9ea2b2d410f | 193 | |
| pwensing | 0:43448bf056e8 | 194 | // Setup experiment |
| pwensing | 0:43448bf056e8 | 195 | t.reset(); |
| pwensing | 0:43448bf056e8 | 196 | t.start(); |
| elijahsj | 5:1ab9b2527794 | 197 | encoderA.reset(); |
| elijahsj | 5:1ab9b2527794 | 198 | encoderB.reset(); |
| elijahsj | 5:1ab9b2527794 | 199 | encoderC.reset(); |
| elijahsj | 5:1ab9b2527794 | 200 | encoderD.reset(); |
| elijahsj | 10:a40d180c305c | 201 | |
| elijahsj | 15:495333b2ccf1 | 202 | motorShield.motorAWrite(0, 0); //turn motor A off |
| saloutos | 16:f9ea2b2d410f | 203 | motorShield.motorBWrite(0, 0); //turn motor B off |
| saloutos | 16:f9ea2b2d410f | 204 | |
| pwensing | 0:43448bf056e8 | 205 | // Run experiment |
| saloutos | 16:f9ea2b2d410f | 206 | while( t.read() < start_period + traj_period + end_period) { |
| saloutos | 16:f9ea2b2d410f | 207 | |
| saloutos | 19:562c08086d71 | 208 | // Read encoders to get motor states |
| saloutos | 16:f9ea2b2d410f | 209 | angle1 = encoderA.getPulses() *PULSE_TO_RAD + angle1_init; |
| saloutos | 16:f9ea2b2d410f | 210 | velocity1 = encoderA.getVelocity() * PULSE_TO_RAD; |
| saloutos | 16:f9ea2b2d410f | 211 | |
| saloutos | 16:f9ea2b2d410f | 212 | angle2 = encoderB.getPulses() * PULSE_TO_RAD + angle2_init; |
| saloutos | 16:f9ea2b2d410f | 213 | velocity2 = encoderB.getVelocity() * PULSE_TO_RAD; |
| saloutos | 16:f9ea2b2d410f | 214 | |
| saloutos | 16:f9ea2b2d410f | 215 | const float th1 = angle1; |
| saloutos | 16:f9ea2b2d410f | 216 | const float th2 = angle2; |
| saloutos | 16:f9ea2b2d410f | 217 | const float dth1= velocity1; |
| saloutos | 16:f9ea2b2d410f | 218 | const float dth2= velocity2; |
| saloutos | 16:f9ea2b2d410f | 219 | |
| saloutos | 16:f9ea2b2d410f | 220 | // Calculate the Jacobian |
| dgdiaz | 34:9a24d0f718ac | 221 | float Jx_th1 = l1*cos(th1)+l2*cos(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 222 | float Jx_th2 = l2*cos(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 223 | float Jy_th1 = l1*sin(th1)+l2*sin(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 224 | float Jy_th2 = l2*sin(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 225 | |
| dgdiaz | 34:9a24d0f718ac | 226 | |
| saloutos | 17:1bb5aa45826e | 227 | // Calculate the forward kinematics (position and velocity) |
| dgdiaz | 34:9a24d0f718ac | 228 | // float xFoot = l_DE*sin(th1)+l_OB*sin(th1)+l_AC*sin(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 229 | // float yFoot = -l_DE*cos(th1)-l_OB*cos(th1)-l_AC*cos(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 230 | // float dxFoot = dth1*(l_AC*cos(th1+th2)+l_DE*cos(th1)+l_OB*cos(th1))+dth2*l_AC*cos(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 231 | // float dyFoot = dth1*(l_AC*sin(th1+th2)+l_DE*sin(th1)+l_OB*sin(th1))+dth2*l_AC*sin(th1+th2); |
| dgdiaz | 34:9a24d0f718ac | 232 | float xFoot = l2*sin(th1+th2) + l1*sin(th1); |
| dgdiaz | 34:9a24d0f718ac | 233 | float yFoot = -l2*cos(th1+th2) - l1*cos(th1); |
| dgdiaz | 34:9a24d0f718ac | 234 | float dxFoot = l1*cos(th1)*dth1 + l2*cos(th1+th2)*(dth1+dth2); |
| dgdiaz | 34:9a24d0f718ac | 235 | float dyFoot = l1*sin(th1)*dth1 + l2*sin(th1+th2)*(dth1+dth2); |
| dgdiaz | 34:9a24d0f718ac | 236 | |
| saloutos | 16:f9ea2b2d410f | 237 | // Set gains based on buffer and traj times, then calculate desired x,y from Bezier trajectory at current time if necessary |
| saloutos | 16:f9ea2b2d410f | 238 | float teff = 0; |
| saloutos | 16:f9ea2b2d410f | 239 | float vMult = 0; |
| uriel_magana | 36:a3beb771d8f7 | 240 | if( t < start_period) { // impedance operational space to start point |
| dgdiaz | 37:48a3017a958f | 241 | |
| dgdiaz | 37:48a3017a958f | 242 | teff = 0.f; |
| dgdiaz | 37:48a3017a958f | 243 | float th1_d= 0.f; |
| dgdiaz | 37:48a3017a958f | 244 | float th2_d= -3.1415f/2.f; |
| dgdiaz | 37:48a3017a958f | 245 | float e_th1= th1_d-th1; |
| dgdiaz | 37:48a3017a958f | 246 | float e_th2= th2_d-th2; |
| uriel_magana | 36:a3beb771d8f7 | 247 | |
| dgdiaz | 37:48a3017a958f | 248 | float e_dth1= -dth1; |
| dgdiaz | 37:48a3017a958f | 249 | float e_dth2= -dth2; |
| uriel_magana | 36:a3beb771d8f7 | 250 | |
| uriel_magana | 38:8ce2f6edba26 | 251 | torque_des[0] = K_xx*e_th1 + D_xx*e_dth1; |
| uriel_magana | 38:8ce2f6edba26 | 252 | torque_des[1] = K_yy*e_th2 + D_yy*e_dth2; |
| dgdiaz | 37:48a3017a958f | 253 | |
| uriel_magana | 38:8ce2f6edba26 | 254 | current_des1 = torque_des[0]/k_t; |
| uriel_magana | 38:8ce2f6edba26 | 255 | current_des2 = torque_des[1]/k_t; |
| saloutos | 16:f9ea2b2d410f | 256 | } |
| uriel_magana | 36:a3beb771d8f7 | 257 | else if (t < start_period + traj_period) // torque control to follow Bezier |
| saloutos | 16:f9ea2b2d410f | 258 | { |
| dgdiaz | 34:9a24d0f718ac | 259 | K_xx = input_params[5]; // Foot stiffness N/m |
| dgdiaz | 34:9a24d0f718ac | 260 | K_yy = input_params[6]; // Foot stiffness N/m |
| dgdiaz | 34:9a24d0f718ac | 261 | K_xy = input_params[7]; // Foot stiffness N/m |
| dgdiaz | 34:9a24d0f718ac | 262 | D_xx = input_params[8]; // Foot damping N/(m/s) |
| dgdiaz | 34:9a24d0f718ac | 263 | D_yy = input_params[9]; // Foot damping N/(m/s) |
| dgdiaz | 34:9a24d0f718ac | 264 | D_xy = input_params[10]; // Foot damping N/(m/s) |
| uriel_magana | 36:a3beb771d8f7 | 265 | |
| saloutos | 16:f9ea2b2d410f | 266 | teff = (t-start_period); |
| saloutos | 16:f9ea2b2d410f | 267 | vMult = 1; |
| uriel_magana | 36:a3beb771d8f7 | 268 | torque_bez.evaluate(teff/traj_period,torque_des); |
| uriel_magana | 36:a3beb771d8f7 | 269 | current_des1 = torque_des[0]/k_t; |
| uriel_magana | 36:a3beb771d8f7 | 270 | current_des2 = torque_des[1]/k_t; |
| saloutos | 16:f9ea2b2d410f | 271 | } |
| uriel_magana | 36:a3beb771d8f7 | 272 | else // impedance operational space to end point |
| uriel_magana | 36:a3beb771d8f7 | 273 | { |
| dgdiaz | 37:48a3017a958f | 274 | teff = 0.f; |
| dgdiaz | 37:48a3017a958f | 275 | float th1_d= 0.f; |
| dgdiaz | 37:48a3017a958f | 276 | float th2_d= -3.1415f/2.f; |
| dgdiaz | 37:48a3017a958f | 277 | float e_th1= th1_d-th1; |
| dgdiaz | 37:48a3017a958f | 278 | float e_th2= th2_d-th2; |
| dgdiaz | 34:9a24d0f718ac | 279 | |
| dgdiaz | 37:48a3017a958f | 280 | float e_dth1= -dth1; |
| dgdiaz | 37:48a3017a958f | 281 | float e_dth2= -dth2; |
| uriel_magana | 36:a3beb771d8f7 | 282 | |
| uriel_magana | 38:8ce2f6edba26 | 283 | torque_des[0] = K_xx*e_th1 + D_xx*e_dth1; |
| uriel_magana | 38:8ce2f6edba26 | 284 | torque_des[1] = K_yy*e_th2 + D_yy*e_dth2; |
| dgdiaz | 37:48a3017a958f | 285 | |
| uriel_magana | 38:8ce2f6edba26 | 286 | current_des1 = torque_des[0]/k_t; |
| uriel_magana | 38:8ce2f6edba26 | 287 | current_des2 = torque_des[1]/k_t; |
| dgdiaz | 37:48a3017a958f | 288 | |
| dgdiaz | 37:48a3017a958f | 289 | // teff = 0; |
| dgdiaz | 37:48a3017a958f | 290 | // float rDesFoot[2] , vDesFoot[2]; |
| dgdiaz | 37:48a3017a958f | 291 | // rDesFoot[0] = l2; |
| dgdiaz | 37:48a3017a958f | 292 | // rDesFoot[1] = -l1; |
| dgdiaz | 37:48a3017a958f | 293 | // vDesFoot[0] = 0; |
| dgdiaz | 37:48a3017a958f | 294 | // vDesFoot[1] =0; |
| dgdiaz | 37:48a3017a958f | 295 | // |
| dgdiaz | 37:48a3017a958f | 296 | // // Calculate the inverse kinematics (joint positions and velocities) for desired joint angles |
| dgdiaz | 37:48a3017a958f | 297 | // float xFootd = -rDesFoot[0]; |
| dgdiaz | 37:48a3017a958f | 298 | // float yFootd = rDesFoot[1]; |
| dgdiaz | 37:48a3017a958f | 299 | // |
| dgdiaz | 37:48a3017a958f | 300 | // // Calculate error variables |
| dgdiaz | 37:48a3017a958f | 301 | // float e_x = xFootd-xFoot; |
| dgdiaz | 37:48a3017a958f | 302 | // float e_y = yFootd-yFoot; |
| dgdiaz | 37:48a3017a958f | 303 | // float de_x = vDesFoot[0]-dxFoot; |
| dgdiaz | 37:48a3017a958f | 304 | // float de_y = vDesFoot[1]-dyFoot; |
| dgdiaz | 37:48a3017a958f | 305 | // |
| dgdiaz | 37:48a3017a958f | 306 | // // Calculate virtual force on foot |
| dgdiaz | 37:48a3017a958f | 307 | // float fx = K_xx*e_x + K_xy*e_y + D_xx*de_x + D_xy*de_y; |
| dgdiaz | 37:48a3017a958f | 308 | // float fy = K_xy*e_x + K_yy*e_y + D_xy*de_x + D_yy*de_y; |
| dgdiaz | 37:48a3017a958f | 309 | // |
| dgdiaz | 37:48a3017a958f | 310 | // float T1 = Jx_th1*fx+Jy_th1*fy; |
| dgdiaz | 37:48a3017a958f | 311 | // float T2 = Jx_th2*fx+Jy_th2*fy; |
| dgdiaz | 37:48a3017a958f | 312 | // |
| dgdiaz | 37:48a3017a958f | 313 | // // Cartesian impedance |
| dgdiaz | 37:48a3017a958f | 314 | // // Note: As with the joint space laws, be careful with signs! |
| dgdiaz | 37:48a3017a958f | 315 | // current_des1 = T1/k_t; |
| dgdiaz | 37:48a3017a958f | 316 | // current_des2 = T2/k_t; |
| uriel_magana | 36:a3beb771d8f7 | 317 | } |
| uriel_magana | 36:a3beb771d8f7 | 318 | |
| saloutos | 19:562c08086d71 | 319 | // Form output to send to MATLAB |
| saloutos | 16:f9ea2b2d410f | 320 | float output_data[NUM_OUTPUTS]; |
| saloutos | 16:f9ea2b2d410f | 321 | // current time |
| pwensing | 0:43448bf056e8 | 322 | output_data[0] = t.read(); |
| saloutos | 16:f9ea2b2d410f | 323 | // motor 1 state |
| saloutos | 16:f9ea2b2d410f | 324 | output_data[1] = angle1; |
| saloutos | 16:f9ea2b2d410f | 325 | output_data[2] = velocity1; |
| saloutos | 16:f9ea2b2d410f | 326 | output_data[3] = current1; |
| saloutos | 16:f9ea2b2d410f | 327 | output_data[4] = current_des1; |
| saloutos | 16:f9ea2b2d410f | 328 | output_data[5] = duty_cycle1; |
| saloutos | 16:f9ea2b2d410f | 329 | // motor 2 state |
| saloutos | 16:f9ea2b2d410f | 330 | output_data[6] = angle2; |
| saloutos | 16:f9ea2b2d410f | 331 | output_data[7] = velocity2; |
| saloutos | 16:f9ea2b2d410f | 332 | output_data[8] = current2; |
| saloutos | 16:f9ea2b2d410f | 333 | output_data[9] = current_des2; |
| saloutos | 16:f9ea2b2d410f | 334 | output_data[10]= duty_cycle2; |
| saloutos | 16:f9ea2b2d410f | 335 | // foot state |
| saloutos | 16:f9ea2b2d410f | 336 | output_data[11] = xFoot; |
| saloutos | 16:f9ea2b2d410f | 337 | output_data[12] = yFoot; |
| saloutos | 17:1bb5aa45826e | 338 | output_data[13] = dxFoot; |
| saloutos | 17:1bb5aa45826e | 339 | output_data[14] = dyFoot; |
| dgdiaz | 35:88dbfefc1bbb | 340 | output_data[15] = current1*k_t; |
| dgdiaz | 35:88dbfefc1bbb | 341 | output_data[16] = current2*k_t; |
| dgdiaz | 35:88dbfefc1bbb | 342 | output_data[17] = torque_des[0]; |
| dgdiaz | 35:88dbfefc1bbb | 343 | output_data[18] = torque_des[1]; |
| dgdiaz | 35:88dbfefc1bbb | 344 | // output_data[15] = rDesFoot[0]; |
| dgdiaz | 35:88dbfefc1bbb | 345 | // output_data[16] = rDesFoot[1]; |
| dgdiaz | 35:88dbfefc1bbb | 346 | // output_data[17] = 0;//vDesFoot[0]; |
| dgdiaz | 35:88dbfefc1bbb | 347 | // output_data[18] = 0;//vDesFoot[1]; |
| elijahsj | 13:3a1f4e09789b | 348 | |
| pwensing | 0:43448bf056e8 | 349 | // Send data to MATLAB |
| pwensing | 0:43448bf056e8 | 350 | server.sendData(output_data,NUM_OUTPUTS); |
| saloutos | 16:f9ea2b2d410f | 351 | |
| saloutos | 16:f9ea2b2d410f | 352 | wait_us(impedance_control_period_us); |
| elijahsj | 4:7a1b35f081bb | 353 | } |
| saloutos | 16:f9ea2b2d410f | 354 | |
| pwensing | 0:43448bf056e8 | 355 | // Cleanup after experiment |
| pwensing | 0:43448bf056e8 | 356 | server.setExperimentComplete(); |
| saloutos | 16:f9ea2b2d410f | 357 | currentLoop.detach(); |
| elijahsj | 12:84a6dcb60422 | 358 | motorShield.motorAWrite(0, 0); //turn motor A off |
| saloutos | 16:f9ea2b2d410f | 359 | motorShield.motorBWrite(0, 0); //turn motor B off |
| dgdiaz | 34:9a24d0f718ac | 360 | |
| pwensing | 0:43448bf056e8 | 361 | } // end if |
| dgdiaz | 34:9a24d0f718ac | 362 | |
| pwensing | 0:43448bf056e8 | 363 | } // end while |
| dgdiaz | 34:9a24d0f718ac | 364 | |
| suribe | 32:c60a5d33cd79 | 365 | } // end main |
