With this script a Ball-E robot can be made and be operative for the use.
Dependencies: HIDScope MODSERIAL QEI biquadFilter mbed
Fork of Samenvoegen_7_5 by
main.cpp
- Committer:
- Wallie117
- Date:
- 2015-10-19
- Revision:
- 0:6616d722fed3
- Child:
- 1:0e1d91965b8e
File content as of revision 0:6616d722fed3:
//================================================================ Script: Ball-E ================================================================== // Author: Ewoud Velu, Lisa Verhoeven, Robert van der Wal, Thijs van der Wal, Emily Zoetbrood /* This is the script of a EMG measurment moving robot. The purpose of the robot is to amuse people with the disease of Ducenne. The robot is designed to throw a ball in to a certain chosen pocket. In order to achieve this movement we use a ‘shoulder’ that can turn in the vertical plane */ //******************************************** Library DECLARATION ************************************** #include "mbed.h" #include "QEI.h" #include "MODSERIAL.h" #include "HIDScope.h" #include "biquadFilter.h" #include <cmath> #include <stdio.h> //******************************************* FUNCTION DECLA ******************************************* //**************** INPUTS ***************** AnalogIn EMG1(A0); // Input EMG AnalogIn EMG2(A1); // Input EMG //AnalogIn pot(A2); //AnalogIn pot1(A3); QEI wheel2 (D10, D11, NC, 64,QEI::X4_ENCODING); // Function for Encoder1 on the motor1 to the Microcontroller QEI wheel1 (D12, D13, NC, 64,QEI::X4_ENCODING); // Function for Encoder2 on the motor2 to the Microcontroller //**************** OUTPUTS **************** DigitalOut red(LED_RED); // LED declared for calibration DigitalOut green(LED_GREEN); // LED declared for sampling DigitalOut magnet(D2); DigitalOut motor2direction(D4); // D4 en D5 zijn motor 2 (op het motorshield) PwmOut motor2speed(D5); DigitalOut motor1direction(D7); // D6 en D7 zijn motor 1 (op het motorshield) PwmOut motor1speed(D6); //**************** FUNCTIONS ************** /* HIDScope scope(4); */ // HIDScope declared with 4 scopes MODSERIAL pc(USBTX, USBRX); // Function for Serial communication with the Microcontroller to the pc. //******************************* GLOBAL VARIABLES DECLARATION ************************************ const int led_on = 0; // Needed for the LEDs to turn on and off const int led_off = 1; const int relax = 0; const int graden = 360; int games_played = -1; // -1 omdat het spel daar eerst loopt voor het spelen om alles na te checken int games_played1 = 0; bool flag_calibration = true; //********* VARIABLES FOR MOTOR CONTROL ******** const float cw = 1; // The number if: motor1 moves clock wise motor2 moves counterclockwise const float ccw = 0; // The number if: motor1 moves counterclock wise motor2 moves clockwise bool flag_s = false; // Var flag_s sensor ticker bool flag_m = false; // Var flag_m motor ticker float ain = 0; //********* VARIABLES FOR CONTROL 1 ************ volatile bool sample_go; // Ticker EMG function int Fs = 512; // Sampling frequency const double low_b1 = 1.480219865318266e-04; //Filter coefficients const double low_b2 = 2.960439730636533e-04; const double low_b3 = 1.480219865318266e-04; const double low_a2 = -1.965293372622690e+00; // a1 is normalized to 1 const double low_a3 = 9.658854605688177e-01; const double high_b1 = 8.047897937631126e-01; const double high_b2 = -1.609579587526225e+00; const double high_b3 = 8.047897937631126e-01; const double high_a2 = -1.571102440190402e+00; // a1 is normalized to 1 const double high_a3 = 6.480567348620491e-01; double u1; double y1; // Declaring the input variables double u2; double y2; int calibratie_metingen = 0; //********* VARIABLES FOR FASE SWITCH ********** bool begin = true; int fase = 3; // To switch between different phases and begins with the reset phase const float fasecontract = 0.7; // The value the EMG signal must be for fase change float rightbiceps = y2; float leftbiceps = ain; int reset = 0; int button_pressed = 0; int j = 1; int N = 200; bool fase_switch_wait = true; //********* VARIABLES FOR CONTROL 2 ************ const float contract = 0.2; // The minimum value for which the motor moves const float maxleft = -1000; // With this angle the motor should stop moving const float maxright = 1000; // With this angle the motor should stop moving const float speed_plate_1 = 0.4; // The speed of the plate in control 2 bool flag_left = true; bool flag_right = true; float pos_board = 0; // The position of the board begins at zero float Encoderread2 = 0; //********* VARIABLES FOR CONTROL 3 ************ const float minimum_contract = 0.2; // Certain levels for muscle activity to activate motor const float medium_contract = 0.5; // "Medium contract muscle" const float maximum_leftbiceps = 0.8; // "Maximum contract muscle" const float on = 1.0; const float off = 0.0; const float minimum_throw_angle = -5; const float maximum_throw_angle = 110; float pos_arm = 0; int pos_arm1 = 0; float Encoderread1 = 0; int switch_rondjes = 2; int rondjes = 0; //********* VARIABLES FOR CONTROL 4 ************ float reset_positie = 0; // Belangrijk float reset_board = 0; float speedcompensation; float speedcompensation2; int t = 5; // aftellen naar nieuw spel int switch_reset = 1; bool arm = false; bool board1 = false; // **************************************** Tickers ***************************************** Ticker Sensor_control; // Adds ticker function for the variable function : Sensors Ticker Motor_control; Ticker EMG_Control; //=============================================================================================== SUB LOOPS ================================================================================================================== //**************************** CONTROL 1-EMG CALIBRATION *********************************** biquadFilter highpass1(high_a2, high_a3, high_b1, high_b2, high_b3); // Different objects for different inputs, otherwise the v1 and v2 variables get fucked up biquadFilter highpass2(high_a2, high_a3, high_b1, high_b2, high_b3); biquadFilter lowpass1(low_a2, low_a3, low_b1, low_b2, low_b3); biquadFilter lowpass2(low_a2, low_a3, low_b1, low_b2, low_b3); double cali_fact1 = 0.9; // calibration factor to normalize filter output to a scale of 0 - 1 double cali_fact2 = 0.9; double cali_array1[2560] = {}; // array to store values in double cali_array2[2560] = {}; void hoog_laag_filter() { u1 = EMG1; u2 = EMG2; y1 = highpass1.step(u1); y2 = highpass2.step(u2); // filter order is: high-pass --> rectify --> low-pass y1 = fabs(y1); y2 = fabs(y2); y1 = lowpass1.step(y1)*cali_fact1; y2 = lowpass2.step(y2)*cali_fact2; // roughly normalize to a scale of 0 - 1, where 0 is minimum and 1 is roughly the maximum output. } //======================================= TICKER LOOPS ========================================== void SENSOR_LOOP() { flag_s = true; } void MOTOR_LOOP() { flag_m = true; } void samplego() { // ticker function, set flag to true every sample interval sample_go = 1; } //================================================================================================== HEAD LOOP ================================================================================================================ int main() { pc.baud(115200); Sensor_control.attach(&SENSOR_LOOP, 0.01); // TICKER FUNCTION Motor_control.attach(&MOTOR_LOOP, 0.01); EMG_Control.attach(&samplego, (float)1/Fs); pc.printf("===============================================================\n"); pc.printf(" \t\t\t Ball-E\n"); pc.printf("In the module Biorobotics on the University of Twente is this script created\n"); pc.printf("Autors:\tE. Velu, L. Verhoeven, R. v/d Wal, T. v/d Wal, E. Zoetbrood\n\n"); wait(3); pc.printf("The script will begin with a short calibration\n\n"); wait(2.5); pc.printf("===============================================================\n"); //************************* CONTROL 1-EMG CALIBRATION **************************** while(1) { if(sample_go) { red.write(led_off); // Toggles red calibration LED off green.write(led_on); // Toggles green sampling LED on hoog_laag_filter(); sample_go = 0; } if (flag_calibration) { // function to calibrate the emg signals from the user. It takes 5 seconds of measurements of maximum output, then takes the max and normalizes to that. calibratie_metingen ++; green.write(led_off); // Toggles green sampling LED off red.write(led_on); //Toggles red calibration LED on cali_fact1 = 0.9; // calibration factor to normalize filter output to a scale of 0 - 1 cali_fact2 = 0.9; double cali_max1 = 0; // declare max y1 double cali_max2 = 0; //declare max y2 pc.printf(" \n\n EMG Signal starting up Calibration measurement........... \n"); wait(2); pc.printf("\t.....Calibrating Signal of EMG 1 and 2 .......\n"); wait(0.5); pc.printf("\t......contract muscles..... \n"); for(int cali_index1 = 0; cali_index1 < 2560; cali_index1++) // Records 5 seconds of y1 { hoog_laag_filter(); cali_array1[cali_index1] = y1; wait((float)1/Fs); } for(int cali_index2 = 0; cali_index2 < 2560; cali_index2++) // Records 5 seconds of y2 { hoog_laag_filter(); cali_array2[cali_index2] = y2; wait((float)1/Fs); } for(int cali_index3 = 0; cali_index3 < 2560; cali_index3++) // Scales y1 { if(cali_array1[cali_index3] > cali_max1) { cali_max1 = cali_array1[cali_index3]; } } for(int cali_index4 = 0; cali_index4 < 2560; cali_index4++) // Scales y2 { if(cali_array2[cali_index4] > cali_max2) { cali_max2 = cali_array2[cali_index4]; } } cali_fact1 = (double)1/cali_max1; // Calibration factor for y1 cali_fact2 = (double)1/cali_max2; // Calibration factor for y2 // Toggles green sampling LED off red.write(led_off); green.write(led_on); pc.printf(" \t....... Calibration has been completed!\n"); wait(0.2); green.write(led_off); wait(0.2); green.write(led_on); wait(0.2); green.write(led_off); wait(0.2); green.write(led_on); wait(4); pc.printf("Beginning with Ball-E board settings\n"); green.write(led_off); wait(2); y1 = 0; y2 = 0; j = 1; // Wachten van fase switch initialiseren fase_switch_wait = true; } //************************* CONTROL MOTOR **************************************** if (flag_s) { Encoderread1 = wheel1.getPulses(); pos_arm = (Encoderread1/((64.0*131.0)/360.0)); // Omrekenen naar graden van arm Encoderread2 = wheel2.getPulses(); pos_board = (Encoderread2/((64.0*131.0)/360.0)); // Omrekenen naar graden van board flag_calibration = false; } //************************* FASE SWITCH ****************************************** //******************** Fase determination ************* if (fase_switch_wait == true) { j++; wait(0.01); // Problemen met EMG metingen die te hoog zijn op het begin van script na calibratie. vandaar ff wachten en de sample loop een paar keer doorlopen. pc.printf("waarde j = %i \n",j); } if( j > N){ fase_switch_wait = false; switch(fase) { //******************* Fase 1 ************************** case(1): rondjes = 0; if (y1> contract) { flag_right = false; flag_left = true; } if (y2 > contract) { flag_right = true; flag_left = false; } if (pos_board <= maxleft) { flag_left = false; motor2speed.write(relax); } if (pos_board >= maxright) { flag_right = false; motor2speed.write(relax); } if (flag_left == true) { if (y1> contract) { motor2direction.write(ccw); motor2speed.write(speed_plate_1); } else { motor2speed.write(relax); } } if (flag_right == true) { if (y2 > contract) { motor2direction.write(cw); motor2speed.write(speed_plate_1); } else { motor2speed.write(relax); } } pc.printf("Boardposition \t %f EMG1 en EMG2 signaal = %f \t %f\n", pos_board, y1, y2); if (y1> fasecontract && y2 > fasecontract) { motor2speed.write(relax); fase = 2; fase_switch_wait = true; j = 0; } break; //******************* Fase 2 ************************** case(2): motor1direction.write(cw); pos_arm1 = (pos_arm - (rondjes * 360)); switch(switch_rondjes){ case(1): rondjes ++; switch_rondjes = 2; break; case(2): if(pos_arm1>360 & 368<pos_arm1) { switch_rondjes = 1; } break; } if (y2 > minimum_contract & y2 < medium_contract) { motor1speed.write((y2 - minimum_contract)/(medium_contract - minimum_contract)); } if (y2 > medium_contract) // Hoger dan drempelwaarde = Actief { motor1speed.write((y2 - minimum_contract)/(medium_contract - minimum_contract)); } if (y2 < minimum_contract) // Lager dan drempel = Rust { motor1speed.write(relax); } if (pos_arm1 > minimum_throw_angle & pos_arm1 < maximum_throw_angle) { if (y1> maximum_leftbiceps) { magnet.write(off); motor1speed.write(relax); fase = 3; pc.printf("Van fase 2 naar fase 3\n"); wait(2); j = 0; fase_switch_wait = true; } } pc.printf("Armposition \t %i \t EMG1 en EMG2 = %f \t %f \n", pos_arm1, y1, y2); break; //********************************************* Fase 3 ********************************************** case(3): switch(switch_reset) { case(1): if(pos_arm < reset_positie) //ene kant op draaien { motor1direction.write(cw); speedcompensation2 = ((reset_positie - pos_arm)/900.0 + (0.1)); motor1speed.write(speedcompensation2); } if(pos_arm > reset_positie) //andere kant op { motor1direction.write(ccw); speedcompensation2 = ((pos_arm - reset_positie)/500.0 + (0.1)); motor1speed.write(speedcompensation2); } if(pos_arm < reset_positie+5 && pos_arm > reset_positie-5) // Dit gaat niet goed komen, omdat het precies die waarde moet zijn { motor1speed.write(0); arm = true; switch_reset = 2; } pc.printf("Positie_arm = %f \t \t snelheid = %f \n",pos_arm, speedcompensation); wait(0.0001); break; case(2): pc.printf("\t switch magneet automatisch \n"); wait(1); magnet.write(on); wait(0.5); switch_reset = 3; break; case(3): if(pos_board < reset_board) { motor2direction.write(cw); speedcompensation = ((reset_board - pos_board)/500.0 + (0.1)); motor2speed.write(speedcompensation); } if(pos_board > reset_board) { motor2direction.write(ccw); speedcompensation = ((pos_board - reset_board)/500.0 + (0.1)); motor2speed.write(speedcompensation); } if(pos_board < reset_board+5 && pos_board > reset_board-5) { motor2speed.write(0); board1 = true; } if(board1 == true) { red=0; pc.printf("fase switch naar 1\n\n"); board1 = false; arm = false; // flag_calibration = true; !!!!! Moet naar gekeken worden of we elke spel willen calibreren wait(2); games_played ++; games_played1 = games_played - (3*calibratie_metingen - 3); pc.printf("Games played total count = %i and loop %i\n",games_played,games_played1); if(games_played1 == 3){ flag_calibration = true; while(t){ pc.printf("\tCalibratie begint over %i \n",t); t--; wait(1); } } while(t){ pc.printf("\tNieuw spel begint over %i \n",t); t--; wait(1); } fase = 1; // Terug naar fase 1 switch_reset = 1; // De switch op orginele locatie zetten t = 5; } pc.printf("Positie_board = %f \t \t snelheid = %f \n",pos_board, speedcompensation); wait(0.0001); break; } break; //=================================================== STOP SCRIPT ============================================================ } } } }