Code for controlling the Magna table
Dependencies: HIDScope MODSERIAL mbed-dsp mbed
main.cpp
- Committer:
- Technical_Muffin
- Date:
- 2016-01-08
- Revision:
- 2:083d74325bfb
- Parent:
- 1:c8ad338ba312
- Child:
- 3:3d5ad874add0
File content as of revision 2:083d74325bfb:
/*Code originally by Jesse Kaiser, s1355783 for control of the 2DOF Planar Table Some variables are also numbered at the end. The numbers stands for the muscle that controls it. Biceps = 1 = lower right arm(wrist flexors) Triceps = 2 = upper right arm(wrist extensors) Pectoralis Major = 3 = upper left arm(wrist flexors) Deltoid posterior = 4 = lower left arm(wrist extensors) The "x" and "y" at the end of variables stand for the X-movement or Y-movement respectivly. The code has been revised to work with the new board and also has a secondary way of controlling it using a joystick */ #include "mbed.h" #include "MODSERIAL.h" #include "arm_math.h" #include "HIDScope.h" #define P_Gain 0.99 #define K_Gain 150 //Gain of the filtered EMG signal #define Damp 5 //Deceleration of the motor #define Mass 1 // Mass value #define dt 0.01 //Sample frequency #define EMG_tresh1 0.02 #define EMG_tresh2 0.02 #define EMG_tresh3 0.02 #define EMG_tresh4 0.02 #define H_Gain 5 #define Pt_x 0.80 #define Pt_y 0.50 #define error_tresh 0.02 MODSERIAL pc(USBTX,USBRX); //joystick control AnalogIn X_control(A1); AnalogIn Y_control(A0); //Motor control 1 DigitalOut Diry(D12); PwmOut Stepy(PTA1); DigitalOut Enabley(PTC3); //motor control 2 DigitalOut Dirx(PTC17); PwmOut Stepx(PTD1); DigitalOut Enablex(D0); //Microstepping 1 DigitalOut MS11(D11); DigitalOut MS21(D10); DigitalOut MS31(D9); //Microstepping 2 DigitalOut MS12(PTC2); DigitalOut MS22(PTA2); DigitalOut MS32(PTB23); DigitalOut Ledr(LED1); DigitalOut Ledg(LED2); DigitalOut Ledb(LED3); //EMG inputs AnalogIn emg1(A2); //biceps bottom emg board AnalogIn emg2(A3); //triceps AnalogIn emg3(A4); //Pectoralis major AnalogIn emg4(A5); //Deltoid top emg board HIDScope scope(4); Ticker scopeTimer; Ticker emgtimer; Ticker looptimer1; Ticker looptimer2; //Variables for motor control float setpoint = 3200; //Frequentie setpoint float step_freq1 = 1; float step_freq2 = 1; //EMG filter arm_biquad_casd_df1_inst_f32 lowpass1_biceps; arm_biquad_casd_df1_inst_f32 lowpass1_triceps; arm_biquad_casd_df1_inst_f32 lowpass1_pect; arm_biquad_casd_df1_inst_f32 lowpass1_deltoid; arm_biquad_casd_df1_inst_f32 lowpass2_biceps; arm_biquad_casd_df1_inst_f32 lowpass2_triceps; arm_biquad_casd_df1_inst_f32 lowpass2_pect; arm_biquad_casd_df1_inst_f32 lowpass2_deltoid; //used as extra filter for wrist motion arm_biquad_casd_df1_inst_f32 bandstop_biceps; arm_biquad_casd_df1_inst_f32 bandstop_triceps; arm_biquad_casd_df1_inst_f32 bandstop_pect; arm_biquad_casd_df1_inst_f32 bandstop_deltoid; //lowpass 1 filter settings: Fc = 49 Hz, Fs = 100 Hz, Gain = -3 dB //lowpass 2 filter settings: Fc = 0.8 Hz, Fs = 100 Hz, Gain = -3 dB float lowpass1_const[] = {0.956543225556877,1.91308645111375,0.956543225556877,-1.91119706742607,-0.914975834801434}; float lowpass2_const[] = {0.000609854718717299,0.00121970943743460,0.000609854718717299,1.92894226325203,-0.931381682126903}; //highpass filter settings: Fc = 20 Hz, Fs = 100 Hz float highpass_const[] = {0.391335772501769,-0.782671545003538,0.391335772501769,0.369527377351241,-0.195815712655833}; //bandstop filter settings Fc=[29Hz 36Hz], Fs= 100Hz float bandstop_const[] = {0.732022476556623, 0.681064744276583,0.732022476556623,-0.535944148680739,-0.713976335521464,1,0.930387749130681,1,-1.04147956553087,-0.752398361226849}; /* //values are usable for triceps and biceps continuous motion, not for wrist motion. //lowpass 1 filter settings: Fc = 45 Hz, Fs = 100 Hz, Gain = -3 dB //lowpass 2 filter settings: Fc = 0.3 Hz, Fs = 100 Hz, Gain = -3 dB float lowpass1_const[] = {0.800592403464570,1.60118480692914,0.800592403464570,-1.56101807580072,-0.641351538057563}; float lowpass2_const[] = {8.76555487540147e-05, 0.000175311097508029, 8.76555487540147e-05 , 1.97334424978130, -0.973694871976315}; //highpass filter settings: Fc = 20 Hz, Fs = 100 Hz float highpass_const[] = {0.391335772501769 ,-0.782671545003538, 0.391335772501769,0.369527377351241, -0.195815712655833}; */ /* //suffer from high wire motion influence and dont react to continuous tension only rapid activation //lowpass 1 filter settings: Fc = 20 Hz, Fs = 100 Hz, Gain = -3 dB //lowpass 2 filter settings: Fc = 0.3 Hz, Fs = 100 Hz, Gain = -3 dB float lowpass1_const[] = {0.206572083826148,0.413144167652296,0.206572083826148,0.369527377351241,-0.195815712655833}; float lowpass2_const[] = {0.0000876555487540065, 0.000175311097508013, 0.0000876555487540065, 1.97334424978130, -0.973694871976315}; //highpass filter settings: Fc = 3 Hz, Fs = 100 Hz float highpass_const[] = {0.875183092438135,-1.75036618487627,0.875183092438135,1.73472576880928,-0.766006600943264}; */ /* Previous coefficients testing to see if new variants work lowpass 1 filter settings: Fc = 45 Hz, Fs = 100 Hz, Gain = -3 dB lowpass 2 filter settings: Fc = 2 Hz, Fs = 100 Hz, Gain = -3 dB float lowpass1_const[] = {0.800592403464570, 1.60118480692914, 0.800592403464570, -1.56101807580072, -0.641351538057563}; float lowpass2_const[] = {0.0000876555487540065, 0.000175311097508013, 0.0000876555487540065, 1.97334424978130, -0.973694871976315}; //highpass filter settings: Fc = 20 Hz, Fs = 100 Hz float highpass_const[] = {0.391335772501769, -0.782671545003537,0.391335772501769,0.369527377351241,-0.195815712655833 }; */ arm_biquad_casd_df1_inst_f32 highpass_biceps; arm_biquad_casd_df1_inst_f32 highpass_triceps; arm_biquad_casd_df1_inst_f32 highpass_pect; arm_biquad_casd_df1_inst_f32 highpass_deltoid; //state values float lowpass1_biceps_states[4]; float lowpass2_biceps_states[4]; float highpass_biceps_states[4]; float bandstop_biceps_states[8]; float lowpass1_triceps_states[4]; float lowpass2_triceps_states[4]; float highpass_triceps_states[4]; float bandstop_triceps_states[8]; float lowpass1_pect_states[4]; float lowpass2_pect_states[4]; float highpass_pect_states[4]; float bandstop_pect_states[8]; float lowpass1_deltoid_states[4]; float lowpass2_deltoid_states[4]; float highpass_deltoid_states[4]; float bandstop_deltoid_states[8]; //global variabels float filtered_biceps, filtered_triceps, filtered_pect, filtered_deltoid; float speed_old1, speed_old2, speed_old3, speed_old4; float acc1, acc2, acc3, acc4; float force1, force2, force3, force4; float speed1, speed2, speed3, speed4; float damping1, damping2, damping3, damping4; float emg_x, emg_y; float cx = 0; float cy = 0; float errorx = 0.2; float errory = 0.2; float Ps_x = 0; float Ps_y = 0; float hstep_freqx = 1; float hstep_freqy = 1; float emg_y_abs = 0; float emg_x_abs = 0; void looper_emg() { float emg_value1_f32, emg_value2_f32, emg_value3_f32, emg_value4_f32; emg_value1_f32 = emg1.read(); emg_value2_f32 = emg2.read(); emg_value3_f32 = emg3.read(); emg_value4_f32 = emg4.read(); //process emg biceps arm_biquad_cascade_df1_f32(&bandstop_biceps, &emg_value1_f32, &filtered_biceps, 1 );//used for wrist motion arm_biquad_cascade_df1_f32(&lowpass1_biceps, &filtered_biceps, &filtered_biceps, 1 ); arm_biquad_cascade_df1_f32(&highpass_biceps, &filtered_biceps, &filtered_biceps, 1 ); filtered_biceps = fabs(filtered_biceps); //Rectifier, The Gain is already implemented. arm_biquad_cascade_df1_f32(&lowpass2_biceps, &filtered_biceps, &filtered_biceps, 1 ); //low pass filter //process emg triceps arm_biquad_cascade_df1_f32(&bandstop_triceps, &emg_value2_f32, &filtered_triceps, 1 ); //used for wrist motion arm_biquad_cascade_df1_f32(&lowpass1_triceps, &filtered_triceps, &filtered_triceps, 1 ); arm_biquad_cascade_df1_f32(&highpass_triceps, &filtered_triceps, &filtered_triceps, 1 ); filtered_triceps = fabs(filtered_triceps); arm_biquad_cascade_df1_f32(&lowpass2_triceps, &filtered_triceps, &filtered_triceps, 1 ); //process emg pectoralis major arm_biquad_cascade_df1_f32(&bandstop_pect, &emg_value3_f32, &filtered_pect, 1 );//used for wrist motion arm_biquad_cascade_df1_f32(&lowpass1_pect, &filtered_pect, &filtered_pect, 1 ); arm_biquad_cascade_df1_f32(&highpass_pect, &filtered_pect, &filtered_pect, 1 ); filtered_pect = fabs(filtered_pect); arm_biquad_cascade_df1_f32(&lowpass2_pect, &filtered_pect, &filtered_pect, 1 ); //process emg deltoid arm_biquad_cascade_df1_f32(&bandstop_deltoid, &emg_value4_f32, &filtered_deltoid, 1 );//used for wrist motion arm_biquad_cascade_df1_f32(&lowpass1_deltoid, &filtered_deltoid, &filtered_deltoid, 1 ); arm_biquad_cascade_df1_f32(&highpass_deltoid, &filtered_deltoid, &filtered_deltoid, 1 ); filtered_deltoid = fabs(filtered_deltoid); arm_biquad_cascade_df1_f32(&lowpass2_deltoid, &filtered_deltoid, &filtered_deltoid, 1 ); // send value to PC. scope.set(0,filtered_pect); //Filtered EMG signal scope.set(1,filtered_deltoid); /*scope.set(2,filtered_pect); scope.set(3,filtered_deltoid);*/ scope.set(2,emg3.read());//testing emg signal for cause of variance scope.set(3,emg4.read()); scope.send(); } void looper_motory() { /*emg_y = (filtered_biceps - filtered_triceps); emg_y_abs = fabs(emg_y); force1 = emg_y_abs*K_Gain; force1 = force1 - damping1; acc1 = force1/Mass; speed1 = speed_old1 + (acc1 * dt); damping1 = speed1 * Damp; step_freq1 = setpoint * speed1;*/ // Stepy.period(/*1.0/step_freq1*/0.000625); //speed_old1 = speed1; //0.06 for biceps and triceps movement, 0.04 for wrist lower and upper motion right arm lower2 connections if (filtered_triceps > 0.04) {//upward cart movement Diry.write(1); Enabley.write(0); //Enable = 1 turns the motor off. Ledr.write(0); Ledb.write(1); } else if (filtered_biceps > 0.04) {// downward cart movement Diry.write(0); Enabley.write(0); //Enable = 1 turns the motor off. Ledr.write(1); Ledb.write(0); } else if (filtered_triceps < 0.015 && filtered_biceps < 0.015) {//0.04 for biceps and triceps stop, 0.015, 0.015 for wrist lower and upper stop Enabley.write(1); } else{} //Speed limit /*if (speed1 > 1) { speed1 = 1; step_freq1 = setpoint; }*/ } void looper_motorx() { /* emg_x = (filtered_pect - filtered_deltoid); emg_x_abs = fabs(emg_x); force2 = emg_x_abs*K_Gain; force2 = force2 - damping2; acc2 = force2/Mass; speed2 = speed_old2 + (acc2 * dt); damping2 = speed2 * Damp; step_freq2 = setpoint * speed2; Stepx.period(1.0/step_freq2); speed_old2 = speed2; if (emg_x > 0) { Dirx = 1; } if (emg_x < 0) { Dirx = 0; } //Speed limit if (speed2 > 1) { speed2 = 1; step_freq2 = setpoint; } //EMG treshold if (filtered_pect < EMG_tresh3 && filtered_deltoid < EMG_tresh4) { Enablex = 1; //Enable = 1 turns the motor off. } else { Enablex = 0; } */ //0.06 for biceps and triceps movement, 0.03 for wrist lower and upper motion left arm upper 2 connections if (filtered_pect > 0.03) {//right cart movement Dirx.write(1); Enablex.write(0); //Enable = 1 turns the motor off. Ledr.write(0); Ledb.write(1); } else if (filtered_deltoid > 0.03) {//left cart movement Dirx.write(0); Enablex.write(0); //Enable = 1 turns the motor off. Ledr.write(1); Ledb.write(0); } else if (filtered_pect < 0.01 && filtered_deltoid < 0.01) {//0.04 for biceps and triceps stop, 0.01 for wrist lower and upper stop Enablex.write(1); } else{} } int main() { //scopeTimer.attach_us(&scopeSend, 2e3); pc.baud(115200); Ledr=1; Ledg=1; Ledb=1; MS11=1;//higher microstepping in combination with a higher step frequency reduces the vibration significantly MS21=1;//it is now in 16th step mode, which seems to work really well and smooth without causing major vibrations MS31=1; MS12=1; MS22=1; MS32=1; Stepy.write(0.5f); Enabley.write(1); Stepx.write(0.5f); Enablex.write(1); Stepy.period(0.000625);//use period change for speed adjustments Stepx.period(0.000625);//frequency of 1600 Hz /*//code for controlling the mechanism with a joystick float Left_value = 0.6; float Right_value = 0.9; float Up_value = 0.6; float Down_value = 0.9; while(1){ if (X_control.read() < Left_value){ Dirx.write(0); Enablex.write(0); } else if (X_control.read() > Right_value){ Dirx.write(1); Enablex.write(0); } else{ Enablex.write(1); } if (Y_control.read() < Up_value){ Diry.write(0); Enabley.write(0); } else if (Y_control.read() > Down_value){ Diry.write(1); Enabley.write(0); } else{ Enabley.write(1); } pc.printf("X value is %f and Y value is %f\n", X_control.read(), Y_control.read()); }*/ //biceps arm_biquad_cascade_df1_init_f32(&lowpass1_biceps, 1 , lowpass1_const, lowpass1_biceps_states); arm_biquad_cascade_df1_init_f32(&lowpass2_biceps, 1 , lowpass2_const, lowpass2_biceps_states); arm_biquad_cascade_df1_init_f32(&highpass_biceps, 1 , highpass_const, highpass_biceps_states); arm_biquad_cascade_df1_init_f32(&bandstop_biceps, 2 , bandstop_const, bandstop_biceps_states); //triceps arm_biquad_cascade_df1_init_f32(&lowpass1_triceps, 1 , lowpass1_const, lowpass1_triceps_states); arm_biquad_cascade_df1_init_f32(&lowpass2_triceps, 1 , lowpass2_const, lowpass2_triceps_states); arm_biquad_cascade_df1_init_f32(&highpass_triceps, 1 , highpass_const, highpass_triceps_states); arm_biquad_cascade_df1_init_f32(&bandstop_triceps, 2 , bandstop_const, bandstop_triceps_states); //pectoralis major arm_biquad_cascade_df1_init_f32(&lowpass1_pect, 1 , lowpass1_const, lowpass1_pect_states); arm_biquad_cascade_df1_init_f32(&lowpass2_pect, 1 , lowpass2_const, lowpass2_pect_states); arm_biquad_cascade_df1_init_f32(&highpass_pect, 1 , highpass_const, highpass_pect_states); arm_biquad_cascade_df1_init_f32(&bandstop_pect, 2 , bandstop_const, bandstop_pect_states); //deltoid arm_biquad_cascade_df1_init_f32(&lowpass1_deltoid, 1 , lowpass1_const, lowpass1_deltoid_states); arm_biquad_cascade_df1_init_f32(&lowpass2_deltoid, 1 , lowpass2_const, lowpass2_deltoid_states); arm_biquad_cascade_df1_init_f32(&highpass_deltoid, 1 , highpass_const, highpass_deltoid_states); arm_biquad_cascade_df1_init_f32(&bandstop_deltoid, 2 , bandstop_const, bandstop_deltoid_states); emgtimer.attach(looper_emg, 0.01); looptimer1.attach(looper_motorx, 0.01); //X-Spindle motor looptimer2.attach(looper_motory, 0.01); //Y-Spindle motor while(1){}; }