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:
- EmilyJCZ
- Date:
- 2015-10-27
- Revision:
- 8:151e43b99156
- Parent:
- 7:d55569b92f30
- Child:
- 9:a5c2ddf2cb8e
File content as of revision 8:151e43b99156:
//======================================================================= Script: Ball-E ========================================================================== // Authors: 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 ‘arm’ that can turn in the vertical plane and move in the horizontal plane. */ //*********************************************** LIBRARY DECLARATION ******************************************** // Libraries are files which contain standard formulas for reading surtain information. Every library contains its own information. #include "mbed.h" // Standard library. This includes the reading of AnalogIn, DigitalOut, PwmOut and other standard formules. #include "QEI.h" // This library includes the reading of of the Encoders from motors. #include "MODSERIAL.h" // MODSERIAL inherits Serial and adds extensions for buffering. #include "HIDScope.h" // This sends the measured EMG signal to the HIDScope. #include "biquadFilter.h" // Because of this library we can make different filters. #include <cmath> // This library declares a set of functions to compute common mathematical operations and transformations. Only used fabs. #include <stdio.h> // This library defines three variable types, several macros, and various functions for performing input and output. //*********************************************** FUNCTION DECLARATION ******************************************* //**************************** INPUTS ****************************************************** AnalogIn EMG1(A0); // Input left biceps EMG. AnalogIn EMG2(A1); // Input right biceps EMG. QEI wheel1 (D10, D11, NC, 64,QEI::X4_ENCODING); // Function for Encoder1 on the motor1 to the Microcontroller. QEI wheel2 (D12, D13, NC, 64,QEI::X4_ENCODING); // Function for Encoder2 on the motor2 to the Microcontroller. //**************************** OUTPUTS ***************************************************** DigitalOut led_red(D1); // Output for red LED decleared. DigitalOut led_yellow(D3); // Output for yellow LED decleared. DigitalOut led_green(D9); // Output for green LED delceared. DigitalOut magnet(D2); // Output for magnet. DigitalOut motor1direction(D4); // Direction for motor 2 on motorshield. This motor moves the arm in fase 2. PwmOut motor1speed(D5); // Speed for motor 2 on motorshield. This motor moves the arm in fase 2. DigitalOut motor2direction(D7); // Direction for motor 1 on motorshield. This motor moves the board in fase 1. PwmOut motor2speed(D6); // Speed for motor 1 on motorshield. This motor moves the board in fase 1. //**************************** FUNCTIONS *************************************************** //HIDScope scope(2); // HIDScope declared with 2 scopes. MODSERIAL pc(USBTX, USBRX); // Function for Serial communication with the Microcontroller to the pc. //*********************************************** GLOBAL VARIABLES DECLARATION *********************************** const int led_on = 0; // This constant turns the led on. const int led_off = 1; // This constant turns the led off. int games_played1 = 0; // Counts number of games played. int games_played = -1; // Shows real number of games played. There is a -1 because the game is first reset before the first throw. float dt = 0.01; // Time staps //**************************** VARIABLES FOR MOTOR CONTROL ********************************* const float cw = 1; // Motor1 moves counter clock wise and Motor2 moves clock wise. const float ccw = 0; // Motor1 moves clock wise and Motor2 moves counter clock wise. bool flag_s = false; // Var flag_s sensor ticker const int relax = 0; // The motor speed is zero. //**************************** VARIABLES FOR CONTROL 1 ************************************* int Fs = 512; // Sampling frequency. int calibration_measurements = 0; // Integer counts the number of calibrations done. It starts at 0. //Filter coefficients. a1 is normalized to 1. const double low_b1 = /*0.0055427172102806843;*/1.480219865318266e-04; const double low_b2 = /*0.011085434420561369;*/2.960439730636533e-04; const double low_b3 = /*0.0055427172102806843; */1.480219865318266e-04; const double low_a2 = /*-1.778631777824585; */-1.965293372622690e+00; const double low_a3 = /*0.80080264666570777; */9.658854605688177e-01; const double high_b1 = /*0.63894552515902237;*/8.047897937631126e-01; const double high_b2 = /*-1.2778910503180447; */-1.609579587526225e+00; const double high_b3 = /*0.63894552515902237;*/8.047897937631126e-01; const double high_a2 = /*-1.1429805025399009;*/-1.571102440190402e+00; const double high_a3 = /*0.41280159809618855;*/6.480567348620491e-01; // Declaring the input and output variables. double u1; // Input from EMG 1. The left biceps. double y1; // Output from the filters from u1. double u2; // Input from EMG 2. The right biceps. double y2; // Output from the filters from u2. double cali_fact1 = 0.9; // Calibration factor to normalize filter output to a scale of 0 - 1. For signal y1. double cali_fact2 = 0.9; // Calibration factor to normalize filter output to a scale of 0 - 1. For signal y2. double cali_max1 = 0; // Declares max y1. double cali_max2 = 0; // Declares max y2. double cali_array1[2560] = {}; // Array to store values in for the 5 seconds calibartion for y1. double cali_array2[2560] = {}; // Array to store values in for the 5 seconds calibartion for y2. bool flag_calibration = true; // Flag to start the calibration. //**************************** VARIABLES FOR FASE SWITCH *********************************** int fase = 3; // The fase is used in a switch to switch between the fases. It starts at the reset fase. int j = 1; // Wait time. Used in problem1 and fase switch int N = 200; // Maximum value of j. bool fase_switch_wait = true; // Timer wait to switch between different fases. //**************************** VARIABLES FOR CONTROL 2 ************************************* const float contract = 0.5; // The minimum value for y1 and y2 for which the motor moves. const float fasecontract = 0.7; // The value y1 and y2 must be for the fase change. const float maxleft = -200; // With this angle the motor should stop moving. const float maxright = 200; // With this angle the motor should stop moving. const float speed_plate_1 = 0.1; // The speed of the plate in control 2. bool flag_left = true; // This flag determines if the signals from the left biceps should be measured. This is signal y1. bool flag_right = true; // This flag determines if the signals from the right biceps should be measured. This is signal y2. float pos_board = 0; // The begin position of the board. It begins at zero. Reads Encoder2 in degrees. Formula is applied in SensorTicker. int pos_board1 = 0; // float Encoderread2 = 0; // Reads travelled distance from Motor2. Reads pulses. //**************************** VARIABLES FOR CONTROL 3 ************************************* const float minimum_contract = 0.4; // The minimum value for y2 for whicht the motor moves. const float medium_contract = 0.5; // Value for different speeds of the motor. const float maximum_leftbiceps = 0.7; // Value for y1 for which the magnet turns off. const float on = 1.0; // This value turns the magnet on. const float off = 0.0; // This value turns the magnet off. const float minimum_throw_angle = 20; // The minimum angle the arm has to be in to be able to turn the magnet off. const float maximum_throw_angle = 110; // The maximum angle for the arm to turn the magnet off. float pos_arm = 0; // The begin position of the arm. It begins at zero. Reads Encoder1 in degrees. Formula is applied in SensorTicker. int pos_arm1 = 0; // This makes from the positon of the arm degrees within a cirkel as an integer. float pos_arm2 = 0; // This makes from the positon of the arm degrees within a cirkel as an float. float previous_pos_arm = 0; // Needed to calculate the v_arm. float v_arm = 0; // The speed of the arm. float v_arm_com = 0; // The compensated speed of the arm. float speed_control_arm = 0.000; // float Encoderread1 = 0; // Reads travelled distance from Motor1. int switch_rounds = 2; // Value of a switch to calculate the number of rounds made. int rounds = 0; // Number of rounds made by the arm. float pos_armrounds_max = 3; // Max rounds the arm can make. bool problem1 = false; // Stop for fase 2. It is a wait for problem2 begins. bool problem2 = false; // The reset of fase 2 when the arm reaches its maximum value. bool flag_v_arm = false; // float problem_velocity = 0; // //**************************** VARIABLES FOR CONTROL 4 ************************************* float reset_position = 0; // The reset position of the arm. int reset_arm = 0; // The reset position of the arm (?). int reset_board = 0; // The reset position of the board. float speedcompensation; // Speed of Motor2 during reset. float speedcompensation2; // Speed of Motor1 during reset. int t = 5; // Integer for count down to new game. int switch_reset = 1; // Value of a switch for the different parts of the reset. bool board = false; // // **************************************** Tickers **************************************** /* Tickers count constantly. The formulas attacht to the ticker count with them. */ Ticker Sensor_control; // This ticker counts for the position of the motors and the speed of the arm. Ticker EMG_Control; // This ticker counts for the filtering of the EMG signal //=============================================================================================== 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); 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() { Encoderread1 = wheel1.getPulses(); previous_pos_arm = pos_arm; pos_arm = (Encoderread1/((64.0*131.0)/360.0)); // Omrekenen naar graden van arm pos_arm1 = pos_arm; v_arm = (pos_arm - previous_pos_arm)/dt; Encoderread2 = wheel2.getPulses(); pos_board = (Encoderread2/((64.0*131.0)/360.0)); // Omrekenen naar graden van board pos_board1 = pos_board; flag_s = true; } void EMG_LOOP() // ticker function, set flag to true every sample interval { if(flag_calibration == false) { hoog_laag_filter(); } } //================================================================================================== HEAD LOOP ================================================================================================================ int main() { pc.baud(115200); Sensor_control.attach(&SENSOR_LOOP, 0.01); // TICKER FUNCTION EMG_Control.attach(&EMG_LOOP, (float)1/Fs); led_green.write(0); led_yellow.write(0); led_red.write(0); pc.printf("===============================================================\n"); pc.printf(" \t\t\t Ball-E\n"); pc.printf("In the module Biorobotics on the University of Twente this script is 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 (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. { calibration_measurements ++; pc.printf(" \n\n EMG Signal starting up Calibration measurement........... \n"); cali_max1 = 0; // declare max y1 cali_max2 = 0; //declare max y2 cali_fact1 = 0.9; // calibration factor to normalize filter output to a scale of 0 - 1 cali_fact2 = 0.9; wait(2); led_red.write(0); wait(0.2); led_red.write(1); //Toggles red calibration LED on wait(0.2); led_red.write(0); wait(0.2); led_red.write(1); wait(0.2); led_red.write(0); wait(0.2); led_red.write(1); wait(1); pc.printf("\t.....Calibrating Signal of EMG 1 and 2 .......\n"); led_red.write(0); led_yellow.write(1); 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 led_yellow.write(0); pc.printf(" \t....... Calibration has been completed!\n"); wait(0.2); led_green.write(led_off); wait(0.2); led_green.write(led_on); wait(0.2); led_green.write(led_off); wait(0.2); led_green.write(led_on); wait(4); pc.printf("Beginning with Ball-E board settings\n"); led_green.write(led_off); wait(2); y1 = 0; y2 = 0; j = 1; // Wachten van fase switch initialiseren fase_switch_wait = true; flag_calibration = false; } //************************* CONTROL MOTOR **************************************** if (flag_s) { flag_calibration = false; } //************************* FASE SWITCH ****************************************** //******************** Fase determination ************* if (fase_switch_wait) { 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); led_red.write(0); led_green.write(1); led_yellow.write(0); } if( j > N) { fase_switch_wait = false; switch(fase) { //******************* Fase 1 ************************** case(1): led_red.write(1); led_green.write(0); led_yellow.write(0); rounds = 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) { if (y1> contract) { motor2direction.write(ccw); motor2speed.write(speed_plate_1); } else { motor2speed.write(relax); } } if (flag_right) { 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; led_red.write(0); led_green.write(0); led_yellow.write(1); j = 0; } break; //******************* Fase 2 ************************** case(2): led_red.write(0); led_green.write(0); led_yellow.write(1); motor1direction.write(cw); pos_arm1 = (pos_arm - (rounds * 360)); pos_arm2 = pos_arm1; switch(switch_rounds) { case(1): rounds ++; switch_rounds = 2; break; case(2): if(pos_arm1>360 & 368<pos_arm1) { switch_rounds = 1; } break; } if (y2 > minimum_contract & y2 < medium_contract) { speed_control_arm = 1;//((v_arm/304) + 0.15*(1 -(v_arm/304))); motor1speed.write(speed_control_arm); } if (y2 > medium_contract) // Hoger dan drempelwaarde = Actief { speed_control_arm = 1;//((v_arm/304) + 0.15*(1 -(v_arm/304))); motor1speed.write(speed_control_arm); } if (y2 < minimum_contract) // Lager dan drempel = Rust { motor1speed.write(relax); } if(rounds == pos_armrounds_max) // max aantal draaing gemaakt!!!!!! { problem1 = true; problem2 = true; motor1speed.write(relax); while (problem1) { 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. if( j > N) { problem1 = false; } } wait(0.1); led_red.write(0); wait(0.1); led_red.write(1); wait(0.1); led_red.write(0); wait(0.1); led_red.write(1); wait(0.1); led_red.write(0); wait(0.1); led_red.write(1); wait(1.5); while(problem2) { motor1direction.write(ccw); if(pos_arm < 170) { if(v_arm > 200) { flag_v_arm = true; } } if(flag_v_arm) { v_arm_com = v_arm; } speed_control_arm = (0.4*((pos_arm2 - reset_arm)/500.0) + (0.05) - (v_arm_com/304)*0.3); motor1speed.write(speed_control_arm); if (pos_arm < 10) { flag_v_arm = false; problem2 = false; motor1speed.write(0); rounds = 0; wait(1); } } } if (pos_arm1 > minimum_throw_angle & pos_arm1 < maximum_throw_angle) { if (y1> maximum_leftbiceps) { magnet.write(off); wait(0.1); motor1speed.write(relax); fase = 3; pc.printf("Van fase 2 naar fase 3\n"); wait(2); j = 0; fase_switch_wait = true; } } pc.printf("Snelheid arm = \t %f \t EMG1 en EMG2 = %f \t %f \n", speed_control_arm, y1, y2); break; //********************************************* Fase 3 ********************************************** case(3): led_red.write(0); led_green.write(1); led_yellow.write(0); switch(switch_reset) { case(1): if(pos_arm < reset_position) //ene kant op draaien { motor1direction.write(cw); speedcompensation2 = 0.05; //((reset_arm - pos_arm1)/900.0 + (0.02)); motor1speed.write(speedcompensation2); } if(pos_arm > reset_position) //andere kant op { motor1direction.write(ccw); speedcompensation2 = 0.05;//((pos_arm1 - reset_arm)/500.0 + (0.02)); motor1speed.write(speedcompensation2); } if(pos_arm < reset_position+5 && pos_arm > reset_position-5) // Dit gaat niet goed komen, omdat het precies die waarde moet zijn { motor1speed.write(0); 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(0.2); magnet.write(on); wait(2); switch_reset = 3; break; case(3): if(pos_board < reset_board) { motor2direction.write(cw); speedcompensation = 0.05; //((reset_board - pos_board1)/500.0 + (0.1)); motor2speed.write(speedcompensation); } if(pos_board > reset_board) { motor2direction.write(ccw); speedcompensation = 0.05;//((pos_board1 - reset_board)/500.0 + (0.05)); motor2speed.write(speedcompensation); } if(pos_board < reset_board+5 && pos_board > reset_board-5) { motor2speed.write(0); board = true; } if(board) { pc.printf("fase switch naar 1\n\n"); board = false; wait(2); games_played ++; games_played1 = games_played - (5*calibration_measurements - 5); pc.printf("Games played total count = %i and loop %i\n",games_played,games_played1); if(games_played1 == 5) { flag_calibration = true; while(t) { pc.printf("\tCalibratie begint over %i \n",t); t--; led_yellow.write(1); wait(0.5); led_yellow.write(0); wait(0.5); } } while(t) { pc.printf("\tNieuw spel begint over %i \n",t); t--; led_yellow.write(1); wait(0.5); led_yellow.write(0); wait(0.5); } 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 ============================================================ } } } }