control for robotic arm that can play chess using a granular gripper

Dependencies:   Encoder mbed HIDScope Servo MODSERIAL

Fork of chessRobot by a steenbeek

Committer:
annesteenbeek
Date:
Wed Oct 28 11:51:23 2015 +0100
Revision:
117:b1667291748d
Parent:
116:8b812e268b85
Parent:
112:7b964afb97b4
Child:
118:49605b5bd802
merged

Who changed what in which revision?

UserRevisionLine numberNew contents of line
annesteenbeek 13:4837b36b9a68 1 #include "actuators.h"
annesteenbeek 13:4837b36b9a68 2 #include "PID.h"
annesteenbeek 13:4837b36b9a68 3 #include "mbed.h"
annesteenbeek 13:4837b36b9a68 4 #include "config.h"
annesteenbeek 13:4837b36b9a68 5 #include "encoder.h"
annesteenbeek 25:874675516927 6 #include "HIDScope.h"
annesteenbeek 107:de47331612d9 7 #include "buttons.h"
annesteenbeek 107:de47331612d9 8 #include "Servo.h"
annesteenbeek 98:25528494287d 9
annesteenbeek 109:026abd708dce 10 // Motor control constants
annesteenbeek 109:026abd708dce 11 #define pwm_frequency 50000 // still High, could be lowered
annesteenbeek 109:026abd708dce 12 #define PI 3.14159265
annesteenbeek 106:1773bf7b95c5 13
annesteenbeek 109:026abd708dce 14 // functions for controlling the motors
annesteenbeek 109:026abd708dce 15 bool motorsEnable = false;
annesteenbeek 109:026abd708dce 16 bool safetyOn = true;
annesteenbeek 76:0aa90e728e4a 17
annesteenbeek 6:b957d8809e7c 18
annesteenbeek 109:026abd708dce 19 double encoder1Counts = 0;
annesteenbeek 109:026abd708dce 20 double encoder2Counts = 0;
annesteenbeek 68:21cb054f1399 21
annesteenbeek 109:026abd708dce 22 bool direction1 = false; // CCW is false(positive rotation), CW is true (neg rotation)
annesteenbeek 109:026abd708dce 23 bool direction2 = false;
annesteenbeek 6:b957d8809e7c 24
annesteenbeek 109:026abd708dce 25 double motor1Pos = 0;
annesteenbeek 109:026abd708dce 26 double motor2Pos = 0;
annesteenbeek 6:b957d8809e7c 27
annesteenbeek 109:026abd708dce 28 double motor1Speed = 0;
annesteenbeek 109:026abd708dce 29 double motor2Speed = 0;
annesteenbeek 6:b957d8809e7c 30
annesteenbeek 109:026abd708dce 31 double motor1SetSpeed = 0;
annesteenbeek 109:026abd708dce 32 double motor2SetSpeed = 0;
annesteenbeek 6:b957d8809e7c 33
annesteenbeek 109:026abd708dce 34 double motor1PWM = 0;
annesteenbeek 109:026abd708dce 35 double motor2PWM = 0;
annesteenbeek 6:b957d8809e7c 36
annesteenbeek 109:026abd708dce 37 // Set PID values
annesteenbeek 109:026abd708dce 38 double Kp1 = 1;
annesteenbeek 109:026abd708dce 39 double Ki1 = 0;
annesteenbeek 109:026abd708dce 40 double Kd1 = 0;
annesteenbeek 0:525558a26464 41
annesteenbeek 109:026abd708dce 42 double Kp2 = 0.008;
annesteenbeek 109:026abd708dce 43 double Ki2 = 0.08;
annesteenbeek 109:026abd708dce 44 double Kd2 = 0;
annesteenbeek 20:3cba803cd771 45
annesteenbeek 109:026abd708dce 46 double motor1PrevCounts = 0;
annesteenbeek 109:026abd708dce 47 double motor2PrevCounts = 0;
annesteenbeek 109:026abd708dce 48 double prevTime = 0;
annesteenbeek 109:026abd708dce 49 double now = 0;
annesteenbeek 109:026abd708dce 50 double timechange;
annesteenbeek 109:026abd708dce 51
annesteenbeek 109:026abd708dce 52 // Set servo values
annesteenbeek 110:a6439e13be8b 53 double servoSpeed = 0;
annesteenbeek 109:026abd708dce 54 const int servoStartPos = 1300; // Servo ranges from 600ms(-90) to 2000ms(90), 1300 is 0 deg
annesteenbeek 26:0a9e4147a31a 55
annesteenbeek 109:026abd708dce 56 // Set calibration values
annesteenbeek 111:43c0881fe7e7 57 double motor1CalSpeed = -5; // deg/sec
annesteenbeek 111:43c0881fe7e7 58 double motor2CalSpeed = 5;
annesteenbeek 109:026abd708dce 59 bool springHit = false;
annesteenbeek 109:026abd708dce 60 float lastCall = 0;
annesteenbeek 109:026abd708dce 61 bool calibrating1 = true;
annesteenbeek 109:026abd708dce 62 bool calibrating2 = false;
annesteenbeek 111:43c0881fe7e7 63 double looseTime = 0;
annesteenbeek 98:25528494287d 64
annesteenbeek 112:7b964afb97b4 65 bool calReady = false;
annesteenbeek 112:7b964afb97b4 66
annesteenbeek 109:026abd708dce 67 // Create object instances
annesteenbeek 109:026abd708dce 68 // Safety Pin
annesteenbeek 109:026abd708dce 69 DigitalIn safetyIn(safetyPin);
annesteenbeek 32:2006977785f5 70
annesteenbeek 109:026abd708dce 71 // Initialze motors
annesteenbeek 109:026abd708dce 72 PwmOut motor1(motor1PWMPin);
annesteenbeek 109:026abd708dce 73 PwmOut motor2(motor2PWMPin);
annesteenbeek 109:026abd708dce 74
annesteenbeek 109:026abd708dce 75 // initialize Servo
annesteenbeek 109:026abd708dce 76 Servo servo(servoPin);
annesteenbeek 98:25528494287d 77
annesteenbeek 98:25528494287d 78
annesteenbeek 109:026abd708dce 79 // Initialize encoders
annesteenbeek 109:026abd708dce 80 Encoder encoder1(enc1A, enc1B);
annesteenbeek 109:026abd708dce 81 Encoder encoder2(enc2A, enc2B);
annesteenbeek 25:874675516927 82
annesteenbeek 109:026abd708dce 83 // Set direction pins
annesteenbeek 109:026abd708dce 84 DigitalOut motor1Dir(motor1DirPin);
annesteenbeek 109:026abd708dce 85 DigitalOut motor2Dir(motor2DirPin);
annesteenbeek 26:0a9e4147a31a 86
annesteenbeek 109:026abd708dce 87 // create PID instances
annesteenbeek 109:026abd708dce 88 PID motor1PID(&motor1Speed, &motor1PWM, &motor1SetSpeed, Kp1, Ki1, Kd1);
annesteenbeek 109:026abd708dce 89 PID motor2PID(&motor2Speed, &motor2PWM, &motor2SetSpeed, Kp2, Ki2, Kd2);
annesteenbeek 31:8fbee6c92753 90
annesteenbeek 109:026abd708dce 91 Timer t;
annesteenbeek 25:874675516927 92
annesteenbeek 25:874675516927 93 void motorInit(){
annesteenbeek 25:874675516927 94
annesteenbeek 3:47c76be6d402 95 motor1Dir.write(direction1);
annesteenbeek 3:47c76be6d402 96 motor2Dir.write(direction2);
annesteenbeek 3:47c76be6d402 97
annesteenbeek 3:47c76be6d402 98 // Set motor PWM period
annesteenbeek 3:47c76be6d402 99 motor1.period(1/pwm_frequency);
annesteenbeek 3:47c76be6d402 100 motor2.period(1/pwm_frequency);
annesteenbeek 55:ee5257fb73df 101
annesteenbeek 55:ee5257fb73df 102 motor1PID.SetSampleTime(motorCall);
annesteenbeek 55:ee5257fb73df 103 motor2PID.SetSampleTime(motorCall);
annesteenbeek 55:ee5257fb73df 104
annesteenbeek 69:37f75a7d36d8 105 motor1PID.SetOutputLimits(0,1);
annesteenbeek 69:37f75a7d36d8 106 motor2PID.SetOutputLimits(0,1);
annesteenbeek 69:37f75a7d36d8 107
annesteenbeek 30:a20f16bf8dda 108 // Turn PID on
annesteenbeek 55:ee5257fb73df 109 motor1PID.SetMode(AUTOMATIC);
annesteenbeek 55:ee5257fb73df 110 motor2PID.SetMode(AUTOMATIC);
annesteenbeek 98:25528494287d 111
annesteenbeek 98:25528494287d 112 // set servo period
annesteenbeek 107:de47331612d9 113 servo.Enable(servoStartPos, 20000);
annesteenbeek 98:25528494287d 114
annesteenbeek 30:a20f16bf8dda 115 // start the timer
annesteenbeek 29:e4f3455aaa0b 116 t.start();
annesteenbeek 4:80e2280058ed 117 }
annesteenbeek 0:525558a26464 118
annesteenbeek 26:0a9e4147a31a 119
annesteenbeek 0:525558a26464 120 void motorControl(){
annesteenbeek 107:de47331612d9 121 // EMG signals to motor speeds
annesteenbeek 112:7b964afb97b4 122 // const double scaleVel = 20;
annesteenbeek 112:7b964afb97b4 123 // motor1SetSpeed = x_velocity*scaleVel;
annesteenbeek 112:7b964afb97b4 124 // motor2SetSpeed = y_velocity*scaleVel;
annesteenbeek 112:7b964afb97b4 125 // servoSpeed = z_velocity*scaleVel;
annesteenbeek 30:a20f16bf8dda 126 // get encoder positions in degrees
annesteenbeek 30:a20f16bf8dda 127 // 131.25:1 gear ratio
annesteenbeek 30:a20f16bf8dda 128 // getPosition uses X2 configuration, so 32 counts per revolution
annesteenbeek 30:a20f16bf8dda 129 // encoder reads CCW negative, and CW positive, so multiply by -1 to make CCW positive
annesteenbeek 68:21cb054f1399 130
annesteenbeek 68:21cb054f1399 131 encoder1Counts = encoder1.getPosition();
annesteenbeek 68:21cb054f1399 132 encoder2Counts = encoder2.getPosition();
annesteenbeek 68:21cb054f1399 133
annesteenbeek 68:21cb054f1399 134
annesteenbeek 111:43c0881fe7e7 135 motor1Pos = ((encoder1Counts/32)/131.25)*360;
annesteenbeek 111:43c0881fe7e7 136 motor2Pos = ((encoder2Counts/32)/131.25)*360;
annesteenbeek 3:47c76be6d402 137
annesteenbeek 0:525558a26464 138 // check if motor's are within rotational boundarys
annesteenbeek 30:a20f16bf8dda 139 // get encoder speeds in deg/sec
annesteenbeek 66:d1ab5904f8e5 140 now = t.read();
annesteenbeek 50:b0cf07ca53cf 141 timechange = (now - prevTime);
annesteenbeek 111:43c0881fe7e7 142 motor1Speed = ((((encoder1Counts - motor1PrevCounts)/32)/131.25)*360)/timechange;
annesteenbeek 111:43c0881fe7e7 143 motor2Speed = ((((encoder2Counts - motor2PrevCounts)/32)/131.25)*360)/timechange;
annesteenbeek 50:b0cf07ca53cf 144 prevTime = now;
annesteenbeek 68:21cb054f1399 145 motor1PrevCounts = encoder1Counts;
annesteenbeek 68:21cb054f1399 146 motor2PrevCounts = encoder2Counts;
annesteenbeek 29:e4f3455aaa0b 147
annesteenbeek 31:8fbee6c92753 148 // calculate motor setpoint speed in deg/sec from setpoint x/y speed
annesteenbeek 30:a20f16bf8dda 149
annesteenbeek 112:7b964afb97b4 150 // exclude kinematics when still calibrating
annesteenbeek 112:7b964afb97b4 151 if (calReady){
annesteenbeek 112:7b964afb97b4 152 kinematics();
annesteenbeek 112:7b964afb97b4 153 }
annesteenbeek 112:7b964afb97b4 154
annesteenbeek 48:3cf1eaf34926 155 if(motorsEnable){ // only run motors if switch is enabled
annesteenbeek 32:2006977785f5 156 // compute new PID parameters using setpoint angle speeds and encoder speed
annesteenbeek 31:8fbee6c92753 157 writeMotors();
annesteenbeek 98:25528494287d 158 servoControl();
annesteenbeek 0:525558a26464 159 }else{
annesteenbeek 0:525558a26464 160 // write 0 to motors
annesteenbeek 2:95ba9f6f0128 161 motor1.write(0);
annesteenbeek 2:95ba9f6f0128 162 motor2.write(0);
annesteenbeek 0:525558a26464 163 }
annesteenbeek 0:525558a26464 164 }
annesteenbeek 0:525558a26464 165
annesteenbeek 31:8fbee6c92753 166 void writeMotors(){
annesteenbeek 54:c14c3bc48b8a 167 motor1PID.Compute(); // calculate PID outputs, output changes automatically
annesteenbeek 76:0aa90e728e4a 168 motor2PID.Compute();
annesteenbeek 31:8fbee6c92753 169 // write new values to motor's
annesteenbeek 82:4cc8f9ad3fec 170 if (motor1SetSpeed > 0 ){ // CCW rotation
annesteenbeek 111:43c0881fe7e7 171 direction1 = true;
annesteenbeek 84:df770ec4df61 172 motor1PID.SetOutputLimits(0,1); // change pid output direction
annesteenbeek 31:8fbee6c92753 173 }else{
annesteenbeek 111:43c0881fe7e7 174 direction1 = false; // CW rotation
annesteenbeek 83:8fa05f53fc73 175 motor1PID.SetOutputLimits(-1,0);
annesteenbeek 31:8fbee6c92753 176 }
annesteenbeek 82:4cc8f9ad3fec 177 if (motor2SetSpeed > 0 ){ // CCW rotation
annesteenbeek 111:43c0881fe7e7 178 direction2 = true;
annesteenbeek 83:8fa05f53fc73 179 motor2PID.SetOutputLimits(0,1);
annesteenbeek 68:21cb054f1399 180 }else{
annesteenbeek 111:43c0881fe7e7 181 direction2 = false; // CW rotation
annesteenbeek 83:8fa05f53fc73 182 motor2PID.SetOutputLimits(-1,0);
annesteenbeek 68:21cb054f1399 183 }
annesteenbeek 31:8fbee6c92753 184 motor1Dir.write(direction1);
annesteenbeek 31:8fbee6c92753 185 motor2Dir.write(direction2);
annesteenbeek 69:37f75a7d36d8 186
annesteenbeek 52:2ac9dee099ce 187 motor1.write(abs(motor1PWM));
annesteenbeek 76:0aa90e728e4a 188 motor2.write(abs(motor2PWM));
annesteenbeek 31:8fbee6c92753 189 }
annesteenbeek 31:8fbee6c92753 190
annesteenbeek 0:525558a26464 191 void servoControl(){
annesteenbeek 107:de47331612d9 192 servo.SetPosition(1300 + 700*servoSpeed);
annesteenbeek 25:874675516927 193 }
annesteenbeek 25:874675516927 194
annesteenbeek 112:7b964afb97b4 195 const double motor1StartPos = (-10*32*131.25)/360; // angle to encoder counts
annesteenbeek 112:7b964afb97b4 196 const double motor2StartPos = (114*32*131.25)/360;
annesteenbeek 112:7b964afb97b4 197
annesteenbeek 106:1773bf7b95c5 198 bool calibrateMotors(){
annesteenbeek 98:25528494287d 199 safetyOn = false; // safety springs off
annesteenbeek 98:25528494287d 200 motorsEnable = true; // motors on
annesteenbeek 98:25528494287d 201 while (calibrating1 || calibrating2){
annesteenbeek 98:25528494287d 202 if (calibrating1){
annesteenbeek 111:43c0881fe7e7 203 motor1SetSpeed = motor1CalSpeed;
annesteenbeek 98:25528494287d 204 redLed.write(1); greenLed.write(0); blueLed.write(1);
annesteenbeek 111:43c0881fe7e7 205 if(safetyIn.read() !=1 && !springHit){ // check if arm reached safety position
annesteenbeek 112:7b964afb97b4 206 encoder1.setPosition(motor1StartPos); // set motor 1 cal angle
annesteenbeek 112:7b964afb97b4 207 motor1PrevCounts = motor1StartPos; // set previous count to prevent speed spike
annesteenbeek 111:43c0881fe7e7 208 motor1CalSpeed = -motor1CalSpeed; // move away
annesteenbeek 98:25528494287d 209 springHit = true;
annesteenbeek 111:43c0881fe7e7 210 looseTime = t.read(); // timer to compensate spring movement
annesteenbeek 98:25528494287d 211 }else{
annesteenbeek 111:43c0881fe7e7 212 // if hit and after is no longer touching spring and 0.5seconds passed
annesteenbeek 111:43c0881fe7e7 213 if(springHit && ((t.read() - looseTime) > 2)){
annesteenbeek 98:25528494287d 214 motor1SetSpeed = 0;
annesteenbeek 98:25528494287d 215 springHit = false;
annesteenbeek 111:43c0881fe7e7 216 calibrating2 = true; // start calibrating 2
annesteenbeek 98:25528494287d 217 calibrating1 = false;
annesteenbeek 98:25528494287d 218 }
annesteenbeek 98:25528494287d 219 }
annesteenbeek 98:25528494287d 220 }
annesteenbeek 98:25528494287d 221 if (calibrating2){
annesteenbeek 111:43c0881fe7e7 222 motor2SetSpeed = motor2CalSpeed;
annesteenbeek 98:25528494287d 223 redLed.write(1); greenLed.write(1); blueLed.write(0);
annesteenbeek 111:43c0881fe7e7 224 if(safetyIn.read() !=1 && !springHit){ // check if arm reached safety position
annesteenbeek 112:7b964afb97b4 225 encoder2.setPosition(motor2StartPos); // set motor 2 cal angle
annesteenbeek 112:7b964afb97b4 226 motor2PrevCounts = motor2StartPos; // set previous cunt to prevent speed spike
annesteenbeek 111:43c0881fe7e7 227 motor2CalSpeed = -motor2CalSpeed; // move away
annesteenbeek 98:25528494287d 228 springHit = true;
annesteenbeek 111:43c0881fe7e7 229 looseTime = t.read();
annesteenbeek 98:25528494287d 230 }else{
annesteenbeek 111:43c0881fe7e7 231 if(springHit && ((t.read() - looseTime) > 2)){
annesteenbeek 98:25528494287d 232 motor2SetSpeed = 0;
annesteenbeek 98:25528494287d 233 springHit = false;
annesteenbeek 98:25528494287d 234 calibrating2 = false; // stop calibrating 2
annesteenbeek 98:25528494287d 235 }
annesteenbeek 98:25528494287d 236 }
annesteenbeek 98:25528494287d 237 }
annesteenbeek 111:43c0881fe7e7 238 // now = t.read(); // call motor using timer instead of wait
annesteenbeek 111:43c0881fe7e7 239 // if(now - lastCall > motorCall){
annesteenbeek 98:25528494287d 240 motorControl();
annesteenbeek 111:43c0881fe7e7 241 wait(motorCall); // keep calling PID's with motorCall frequency
annesteenbeek 111:43c0881fe7e7 242 // lastCall = now;
annesteenbeek 111:43c0881fe7e7 243 // }
annesteenbeek 98:25528494287d 244
annesteenbeek 98:25528494287d 245 }
annesteenbeek 111:43c0881fe7e7 246 redLed.write(0); greenLed.write(1); blueLed.write(1);
annesteenbeek 99:7030e9790b1d 247 motorsEnable = false; // turn motor's off again
annesteenbeek 76:0aa90e728e4a 248 safetyOn = true; // turn safety on after callibration
annesteenbeek 110:a6439e13be8b 249 return true; // return true when finished
annesteenbeek 74:75be98779124 250 }
annesteenbeek 74:75be98779124 251
annesteenbeek 74:75be98779124 252
annesteenbeek 74:75be98779124 253 void safety(){
annesteenbeek 74:75be98779124 254 if (safetyOn){
annesteenbeek 81:71e7e98deb2c 255 if (safetyIn.read() != 1){
annesteenbeek 79:cf500b63f349 256 motorsEnable = false;
annesteenbeek 79:cf500b63f349 257 }
annesteenbeek 74:75be98779124 258 }
annesteenbeek 79:cf500b63f349 259 }
annesteenbeek 106:1773bf7b95c5 260
annesteenbeek 112:7b964afb97b4 261 const double L1 = 0.436; // first arm in m
annesteenbeek 112:7b964afb97b4 262 const double L2 = 0.120; // first arm in m
annesteenbeek 112:7b964afb97b4 263 const double L3 = 0.255; // servo arm in m
annesteenbeek 112:7b964afb97b4 264 const double Xmax = 0.3;
annesteenbeek 112:7b964afb97b4 265 const double Xmin; = -0.3;
annesteenbeek 112:7b964afb97b4 266 const double Ymax = 0.645;
annesteenbeek 112:7b964afb97b4 267 const double Ymin = 0.33;
annesteenbeek 106:1773bf7b95c5 268
annesteenbeek 112:7b964afb97b4 269
annesteenbeek 106:1773bf7b95c5 270
annesteenbeek 112:7b964afb97b4 271 bool kinematics(){
annesteenbeek 112:7b964afb97b4 272 // calculate current x and Y
annesteenbeek 112:7b964afb97b4 273 double X = (L2+L3)*cos((motor1Pos + motor2Pos)*PI/180) + L1*cos(motor1Pos*PI/180);
annesteenbeek 112:7b964afb97b4 274 double Y = (L2+L3)*sin((motor1Pos + motor2Pos)*PI/180) + L1*sin(motor1Pos*PI/180);
annesteenbeek 112:7b964afb97b4 275 // check if x and y are within limits
annesteenbeek 112:7b964afb97b4 276 // else Store the constraint line
annesteenbeek 112:7b964afb97b4 277 // check if movement is in direction of constraint
annesteenbeek 112:7b964afb97b4 278 // else return false no movement (anglespeed = 0)
annesteenbeek 112:7b964afb97b4 279 // calculate required angle speeds
annesteenbeek 112:7b964afb97b4 280 if( (X>Xmax && setXSpeed > 0 )|| \
annesteenbeek 112:7b964afb97b4 281 (X<Xmin && setXSpeed < 0 )|| \
annesteenbeek 112:7b964afb97b4 282 (Y>Ymax && setYSpeed > 0 )|| \
annesteenbeek 112:7b964afb97b4 283 (Y<Ymin && setYSpeed < 0 ) \
annesteenbeek 112:7b964afb97b4 284 ){
annesteenbeek 112:7b964afb97b4 285 motor1SetSpeed = 0;
annesteenbeek 112:7b964afb97b4 286 motor2SetSpeed = 0;
annesteenbeek 112:7b964afb97b4 287 return false;
annesteenbeek 112:7b964afb97b4 288 break;
annesteenbeek 112:7b964afb97b4 289 }
annesteenbeek 112:7b964afb97b4 290 motor1SetSpeed = (setXSpeed*cos((motor1Pos + motor2Pos)*PI/180) + \
annesteenbeek 112:7b964afb97b4 291 setYSpeed*sin((motor1Pos + motor2Pos)*PI/180))/(L1*sin(motor2Pos*PI/180));
annesteenbeek 112:7b964afb97b4 292 motor2SetSpeed = -(setXSpeed*(L2+L3)*cos((motor1Pos + motor2Pos)*PI/180) + \
annesteenbeek 112:7b964afb97b4 293 setYSpeed*(L2+L3)*sin((motor1Pos + motor2Pos)*PI/180) + \
annesteenbeek 112:7b964afb97b4 294 setXSpeed*L1*cos(motor1Pos*PI/180) + \
annesteenbeek 112:7b964afb97b4 295 setYSpeed*L1*sin(motor1Pos*PI/180))/(L1*(L2+L3)*sin(motor2Pos*PI/180));
annesteenbeek 112:7b964afb97b4 296
annesteenbeek 112:7b964afb97b4 297 }