File content as of revision 20:682ac5b520a0:

// Project BioRobotics - Opening a Door - Group 13 2019/2020
// Dion ten Berge       - s1864734
// Bas Rutteman         - s1854305
// Nick in het Veld     - s1915584
// Marleen van der Weij - s1800078
// Mevlid Yildirim      - s2005735 

/* To-Do
1. Kd, Ki, Kp waardes bepalen
2. Filter cutoff frequentie bepalen, zie https://github.com/tomlankhorst/biquad
3. Grenswaarde EMG signaal na het filteren
*/

//*****************************************************************************
// 1. Libraries ******************************************************************
//*****************************************************************************
#include "mbed.h"
#include "HIDScope.h"
#include "QEI.h"
#include "MODSERIAL.h"
#include "BiQuad.h"
#include "FastPWM.h"

//*****************************************************************************
// 2. States ******************************************************************
//*****************************************************************************
enum States {StartWait, MotorCalibration, EMGCalibration, Homing, Demo, Operating, EmergencyMode, Idle}; //All robot states
States State;

//*****************************************************************************
// 3. (Global) Variables ***********************************************************
//*****************************************************************************
// 3.1 Tickers *****************************************************************
Ticker ticker_mainloop;      // The ticker which runs the mainloop
Ticker ticker_hidscope; // The ticker which sends data to the HIDScope server

// 3.2 General variables *******************************************************

MODSERIAL pc(USBTX, USBRX); // Serial communication with the board
QEI encoder_motor1(D12,D13,NC,64); // Defines encoder for motor 1
QEI encoder_motor2(D10,D11,NC,64); // Defines encoder for motor 1
double f=1/100;                     // Frequency, currently unused
const double Ts = 0.001;            // Sampletime
HIDScope scope(2);                  // Amount of HIDScope servers


// 3.3 EMG Variables **********************************************************
static BiQuad LowPassFilter_motor1( 4.12535e-02, 8.25071e-02, 4.12535e-02, -1.34897e+00, 5.13982e-01 ); 
static BiQuad LowPassFilter_motor2( 4.12535e-02, 8.25071e-02, 4.12535e-02, -1.34897e+00, 5.13982e-01 ); 

static BiQuad highfilter0(9.56543e-01, -1.91309e+00, 9.56543e-01, -1.91120e+00, 9.14976e-01);   
static BiQuad LowPassFilter0( 4.12535e-02, 8.25071e-02, 4.12535e-02, -1.34897e+00, 5.13982e-01 ); 
    
static BiQuad highfilter1(9.56543e-01, -1.91309e+00, 9.56543e-01, -1.91120e+00, 9.14976e-01);   
static BiQuad LowPassFilter1( 4.12535e-02, 8.25071e-02, 4.12535e-02, -1.34897e+00, 5.13982e-01 ); 
   
static BiQuad highfilter2(9.56543e-01, -1.91309e+00, 9.56543e-01, -1.91120e+00, 9.14976e-01);   
static BiQuad LowPassFilter2( 4.12535e-02, 8.25071e-02, 4.12535e-02, -1.34897e+00, 5.13982e-01 ); 
   
static BiQuad highfilter3(9.56543e-01, -1.91309e+00, 9.56543e-01, -1.91120e+00, 9.14976e-01);   
static BiQuad LowPassFilter3( 4.12535e-02, 8.25071e-02, 4.12535e-02, -1.34897e+00, 5.13982e-01 ); 

double emg0_raw_signal ; double emg1_raw_signal ; double emg2_raw_signal ; double emg3_raw_signal ;
double high_emg0_signal ; double high_emg1_signal ; double high_emg2_signal ; double high_emg3_signal ;
double rec_emg0_signal ; double rec_emg1_signal ;  double rec_emg2_signal ;  double rec_emg3_signal ;   
double low_rec_high_emg0_signal ; double low_rec_high_emg1_signal ; double low_rec_high_emg2_signal ; double low_rec_high_emg3_signal ;
double emg0_signal ; double emg1_signal ; double emg2_signal ; double emg3_signal ;



// 3.4 Hardware ***************************************************************
    //3.4a Leds
    DigitalOut led_red(LED_RED); // Defines the red led on the K64 board (0=on, 1 = off)   
    DigitalOut led_green(LED_GREEN); // Defines the green led on the K64 board (0=on, 1 = off)
    DigitalOut led_blue(LED_BLUE); // Defines the blue led on the K64 board (0=on, 1 = off)  
//    FastPWM led1(D8);         //CODE DOES NOT WORK WITH D8 PIN DEFINED      //Defines Led1 on the BioRobotics Shield
    FastPWM led2(D9);               //Defines Led2 on the BioRobotics Shield
    
    //3.4b Potmeters and buttons
    AnalogIn pot1_links(A5); //Defines potmeter1 on the BioRobotics Shield
    AnalogIn pot2_rechts(A4); //Defines potmeter2 on the BioRobotics Shield
    DigitalIn button1(D2); //Defines button1 on the BioRobotics Shield
    DigitalIn button2(D3); //Defines button2 on the BioRobotics Shield
    DigitalIn sw2(SW2); //Defines button SW2 on the K64 board
    DigitalIn sw3(SW3); //Defines button SW3 on the K64 board
    
    //3.4c Motors
    DigitalOut motor1DirectionPin(D7);   // motor 1 direction control (1=cw, 0=ccw)
    FastPWM motor1(D6);                  // motor 1 velocity control (between 0-1)
    FastPWM motor2(D5);                  // motor 2 velocity control (between 0-1)
    DigitalOut motor2DirectionPin(D4);   // motor 2 direction control (1=cw, 0=ccw)
    bool motor1_calibrated=false;
    bool motor2_calibrated=false;
 
    //3.4d EMGs
    AnalogIn    emg0(A0); //Rechterarm
    AnalogIn    emg1(A1); //Linkerarm
    AnalogIn    emg2(A2); //Rechterbeen
    AnalogIn    emg3(A3); //Linkerbeen
    
    double emg0_max=0.0;
    int emg_tijd=0; 
    int homing_tijd=0;
// 3.5 Motor 1 variables ***************************************************************   
    //3.5a PID-controller motor 1
    double counts_per_rad_motor1 = (131.25*32)/(2*3.14159265359); // (gear ratio * counts per revolution) / (2* pi) = ~668.45 counts per rad
    static double error_integral_motor1 = 0;
    double Yref_motor1;                                 
    double kp_motor1;
    double Ki_motor1;
    double Kd_motor1;
    double Up_motor1;
    double Ui_motor1;
    double Ud_motor1;
    
    double positie_motor1;                              //counts encoder
    double error1_motor1;
    double error1_prev_motor1;
    double error1_derivative_motor1;
    double error1_derivative_filtered_motor1;
    double P_motor1;
    
    double positie_verschil_motor1;
    double positie_prev_motor1;

// 3.5 Motor 2 variables ***************************************************************
    //3.5b PID-controller motor 2
    double counts_per_rad_motor2 = (131.25*32)/(2*3.14159265359); // (gear ratio * counts per revolution) / (2* pi) = ~668.45 counts per rad
    static double error_integral_motor2 = 0;
    double Yref_motor2;                                 
    double kp_motor2;
    double Ki_motor2;
    double Kd_motor2;
    double Up_motor2;
    double Ui_motor2;
    double Ud_motor2;

    double positie_motor2;                              //counts encoder
    double error1_motor2;
    double error1_prev_motor2;
    double error1_derivative_motor2;
    double error1_derivative_filtered_motor2;
    double P_motor2;
    
    double positie_verschil_motor2;
    double positie_prev_motor2;    

//******************************************************************************
// 4. Functions ****************************************************************
//******************************************************************************

    // 4.1 Hidscope ****************************************************************
    void HIDScope() //voor HIDscope
    {
        scope.set(0, 1);
   //     scope.set(1, error1_motor1);
   //     scope.set(2, emg_tijd); 
   //     scope.set(3, emg3_signal);
   //    scope.set(4, Ui_motor1);
   //    scope.set(5, Uk_motor1);
       
        scope.send();   
    }
    
    // 4.x Encoder motor1 ****************************************************************
        double fencoder_motor1()                                    // bepaalt de positie van de motor
        {
            positie_motor1 = encoder_motor1.getPulses();                // haalt encoder waardes op
            positie_verschil_motor1 =  (positie_motor1-positie_prev_motor1)/Ts;
            positie_prev_motor1 = positie_motor1;
            
            return positie_motor1;                                 //geeft positie van motor
        }
    // 4.x Encoder motor2 ****************************************************************        
        double fencoder_motor2()                                    // bepaalt de positie van de motor
        {
            positie_motor2 = encoder_motor2.getPulses();                // haalt encoder waardes op
            positie_verschil_motor2 =  (positie_motor2-positie_prev_motor2)/Ts;
            positie_prev_motor2 = positie_motor2;
            
            return positie_motor2;                                 //geeft positie van motor
        }
        
    // 4.xa Calibration motors
        
    void motor_calibration()
    {
        // Calibration motor 2
        motor1DirectionPin=0; //direction of the motor
        motor1=1.0;
        wait(0.1);
        while (abs(positie_verschil_motor1)>5)
        {
            motor1=0.2  ; 
    //        pc.printf("\r\n Motor1 kalibratie = %s", motor1_calibrated ? "true" : "false");
        }
        motor1=0.0;
        motor1_calibrated=true;  
     //   pc.printf("\r\n Motor1 kalibratie = %s", motor1_calibrated ? "true" : "false");
    
    
    
        // Calibration motor 2
        motor2DirectionPin=0; //direction of the motor
        motor2=1.0;
        wait(0.1);
        while (abs(positie_verschil_motor2)>5)
        {
            motor2=0.2  ; 
  //          pc.printf("\r\n Motor2 kalibratie = %s", motor2_calibrated ? "true" : "false");
            led2=1;
        }
        motor2=0.0;
        led2=0;
        motor2_calibrated=true;  
    //    pc.printf("\r\n Motor2 kalibratie = %s", motor2_calibrated ? "true" : "false");
        
            
    }
    
    // 4.2a PID-Controller motor 1**************************************************
    double PID_controller_motor1(double &error_integral_motor1, double &error1_prev_motor1)
    {
        //Proportional part
        kp_motor1 = 0.01 ;      // moet nog getweaked worden
        Up_motor1 = kp_motor1 * error1_motor1;
    
        //Integral part
        Ki_motor1 = 0.001;     // moet nog getweaked worden
        error_integral_motor1 = error_integral_motor1 + (Ts*error1_motor1);       // integrale fout + (de sample tijd * fout)
        Ui_motor1 = Ki_motor1 * error_integral_motor1;                     // (fout * integrale fout)
    
        //Derivative part 
        Kd_motor1 = 0.001 ;// moet nog getweaked worden  
        error1_derivative_motor1 = (error1_motor1-error1_prev_motor1)/Ts; // (Fout - de vorige fout) / tijdstap = afgeleide
        error1_derivative_filtered_motor1 = LowPassFilter_motor1.step(error1_derivative_motor1); //derivative wordt gefiltered
        Ud_motor1 = Kd_motor1 * error1_derivative_filtered_motor1;           // (afgeleide gain) * (afgeleide gefilterde fout) 
        error1_prev_motor1 = error1_motor1;

        P_motor1 = Up_motor1 + Ui_motor1 + Ud_motor1;                                           //sommatie van de u's
        
        return P_motor1; 
        

    }

    // 4.2b PID-Controller motor 2**************************************************
    double PID_controller_motor2(double &error_integral_motor2, double &error1_prev_motor2)
    {
        //Proportional part
        kp_motor2 = 0.01 ;      // moet nog getweaked worden
        Up_motor2 = kp_motor2 * error1_motor2;
    
        //Integral part
        Ki_motor2 = 0.001;     // moet nog getweaked worden
        error_integral_motor2 = error_integral_motor2 + (Ts*error1_motor2);       // integrale fout + (de sample tijd * fout)
        Ui_motor2 = Ki_motor2 * error_integral_motor2;                     //de fout keer de integrale fout
    
        //Derivative part 
        Kd_motor2 = 0.001 ;// moet nog getweaked worden  
        error1_derivative_motor2 = (error1_motor2  -  error1_prev_motor2)/Ts;
        error1_derivative_filtered_motor2 = LowPassFilter_motor2.step(error1_derivative_motor2); //derivative wordt gefiltered, dit later aanpassen
        Ud_motor2 = Kd_motor2 * error1_derivative_filtered_motor2;
        error1_prev_motor2 = error1_motor2;

        P_motor2 = Up_motor2 + Ui_motor2 + Ud_motor2;                                           //sommatie van de u's

        return P_motor2; 
    }
    
    double motor1_pwm()
    {
        
        if (P_motor1 >=0 ) // Als de stuursignaal groter is als 0, dan clockwise rotatie, anders counterclockwise rotatie
        {                
            motor1DirectionPin=1;   // Clockwise rotation
        } 
        else 
        {
             motor1DirectionPin=0;   // Counterclockwise rotation
        }
            
        if (fabs(P_motor1) > 0.99 )  // als de absolute waarde van de motorsnelheid groter is als 1, terug schalen naar 1, anders de absolute waarde van de snelheid. (Bij een waarde lager als 0 draait de motor niet)
        {         
            motor1 = 0.99 ;
        } 
        else 
        {
            motor1 = fabs(P_motor1);
        }
    }  
   double motor2_pwm()
    {
        
        if (P_motor2 >=0 ) // Als de stuursignaal groter is als 0, dan clockwise rotatie, anders counterclockwise rotatie
        {                
            motor2DirectionPin=2;   // Clockwise rotation
        } 
        else 
        {
             motor2DirectionPin=0;   // Counterclockwise rotation
        }
            
        if (fabs(P_motor2) > 0.99 )  // als de absolute waarde van de motorsnelheid groter is als 1, terug schalen naar 1, anders de absolute waarde van de snelheid. (Bij een waarde lager als 0 draait de motor niet)
        {         
            motor2 = 0.99 ;
        } 
        else 
        {
            motor2 = fabs(P_motor2);
        }            
    }
    
    void motor1_controller(void)
    {
        error1_motor1 = (Yref_motor1 - positie_motor1);
        motor1_pwm();
        /*if (motor1_calibrated==true&&motor2_calibrated==true)
            {
                motor1_pwm();
            } */
        
    }

    void motor2_controller(void)
    {
        error1_motor2 = (Yref_motor2 - positie_motor2);
        motor2_pwm();
        /* if (motor1_calibrated==true&&motor2_calibrated==true)
            {
                motor2_pwm();
            } */
    }   
    
void emg0_processing()
{
    emg0_raw_signal=emg0.read();

    high_emg0_signal  =     highfilter0.step(emg0_raw_signal);
    rec_emg0_signal   =     abs(high_emg0_signal);    
    low_rec_high_emg0_signal  =     LowPassFilter0.step(rec_emg0_signal);
    emg0_signal = low_rec_high_emg0_signal;  
    
}     

void emg1_processing()
{
    emg1_raw_signal=emg1.read();

    high_emg1_signal  =     highfilter1.step(emg1_raw_signal);
    rec_emg1_signal   =     abs(high_emg1_signal);    
    low_rec_high_emg1_signal  =     LowPassFilter1.step(rec_emg1_signal);
    emg1_signal = low_rec_high_emg1_signal;
    
}    

void emg2_processing()
{
    emg2_raw_signal=emg2.read();

    high_emg2_signal  =     highfilter2.step(emg2_raw_signal);
    rec_emg2_signal   =     abs(high_emg2_signal);    
    low_rec_high_emg2_signal  =     LowPassFilter2.step(rec_emg2_signal);
    emg2_signal = low_rec_high_emg2_signal;
    
}    

void emg3_processing()
{
    emg3_raw_signal=emg3.read();

    high_emg3_signal  =     highfilter3.step(emg3_raw_signal);
    rec_emg3_signal   =     abs(high_emg3_signal);    
    low_rec_high_emg3_signal  =     LowPassFilter3.step(rec_emg3_signal);
    emg3_signal = low_rec_high_emg3_signal;
    
}    
    
    // 4.3 State-Machine *******************************************************

    void state_machine()
{
    if (sw2==0) {State = EmergencyMode;}
    switch(State)
    {       
        case MotorCalibration: 
//            pc.printf("\r\n State: MotorCalibration");
            led_blue.write(1);
            led_red.write(1);
            led_green.write(0); //Green Led on when in this state
            
            if (motor1_calibrated==true&&motor2_calibrated==true)
            {
                pc.printf("\r\n Motor Calibration is done!");
                encoder_motor1.reset();
                encoder_motor2.reset();

                
                State=StartWait; 
            }
            else {;} //pc.printf("\r\n Motor Calibration is not done!");}
            
            break; 
                
        case StartWait:  
 //           pc.printf("\r\n State: StartWait Button 1 = operation, Button 2 = Demo"); 
            led_blue.write(0);
            led_red.write(1);
            led_green.write(1);
            
            emg_tijd =0;
            State=EMGCalibration;
            break;  
                       
        case EMGCalibration: 
   //         pc.printf("\r\n State: EMGCalibration");
            led_blue.write(1);
            led_red.write(1);
            led_green.write(1);
            
            emg_tijd++;
            
            if (0<=emg_tijd&&emg_tijd<=5000)
            { led_green.write(0); }
            else if (5000<emg_tijd&&emg_tijd<8000)
            {   led_green.write(1);
                if (emg0_signal > emg0_max)
                {
                emg0_max=emg0_signal;
                }             
            }
            else if (8000<emg_tijd)
            {led_green.write(0);
                if(button1==0) {State=StartWait;}
                if(button2==0) {State=Homing;}
            }
            
            
            
//            Yref_motor1=5000;
//            Yref_motor2=2000;
//            State=Homing; 
            break; 
        case Homing: 
       //     pc.printf("\r\n State: Homing");
            led_blue.write(0);
            led_red.write(1);
            led_green.write(1);
            led2=1;

            homing_tijd++;
            
            if (homing_tijd>1000) //Voorkomt dat het automatisch naar Demo springt, omdat je bij de vorige nog knop 2 hebt ingedrukt
            {
                if(button1==0) {State=Operating;}
                if(button2==0) {State=Demo;} 
            }
            
            break; 
  
        case Operating:
  /*          pc.printf("\r\n State: Operating");
            led_blue.write(1);
            led_red.write(1);
            led_green.write(0);
            wait(0.5);
            led_green.write(1);
            wait(0.5); */
            break; 
        
        case Demo:
           pc.printf("\r\n State: Demo");
            led_blue.write(1);
            led_red.write(1);
            led_green.write(0); 

            break;
            
        case EmergencyMode:
            pc.printf("\r\n State: EMERGENCY MODE! Press RESET to restart");
            
            motor1=0; 
            motor2=0;
            
            led_blue.write(1);
            led_green.write(1);
            //SOS start
            led_red.write(0); // S
            wait(0.5);
            led_red.write(1); //pause
            wait(0.25);
            led_red.write(0); // O
            wait(1.5);
            led_red.write(1); // pause
            wait(0.25);
            led_red.write(0); // S
            wait(0.5);
            //SOS end
            break; 
        case Idle:
   /*         pc.printf("\r\n Idling..."); */
            break;

        }
}
    
//******************************************************************************
// 5. Main Loop ****************************************************************
//******************************************************************************

void main_loop() { //Beginning of main_loop()
//    pc.printf("main_loop is running succesfully \r\n"); //confirmation that main_loop is running (als je dit erin zet krijg je elke duizendste dit bericht. Dit is niet gewenst)






//test stuff:
//Yref_motor1= 4200*pot2_rechts.read();
//Yref_motor2= 4200*pot1_links.read();

// 5.1 Measure Analog and Digital input signals ********************************
emg0_processing();
emg1_processing();
emg2_processing();
emg3_processing();

fencoder_motor1() ;
fencoder_motor2() ;
// 5.2 Run state-machine(s) ****************************************************
state_machine() ;
// 5.3 Run controller(s) *******************************************************
PID_controller_motor1(error_integral_motor1, error1_prev_motor1);
PID_controller_motor2(error_integral_motor2, error1_prev_motor2);
// 5.4 Send output signals to digital and PWM output pins **********************
motor1_controller();
motor2_controller();
                    

}                   //Ending of main_loop()

//******************************************************************************
// 6. Main function ************************************************************
//******************************************************************************
int main() 
{ //Beginning of Main() Function  //All the things we do only once, some relevant things are now missing here: set pwmperiod to 60 microsec. Set Serial comm. Etc. Etc.
// 6.1 Initialization **********************************************************
    pc.baud(115200);
    pc.printf("\r\nStarting Project BioRobotics - Opening a Door " //print Project information
                "- Group 13 2019/2020 \r\n"
                "Dion ten Berge       - s1864734 \r\n"
                "Bas Rutteman         - s1854305 \r\n"
                "Nick in het Veld     - s1915584 \r\n"
                "Marleen van der Weij - s1800078 \r\n"
                "Mevlid Yildirim      - s2005735 \r\n");
    led_green.write(1);
    led_red.write(1);
    led_blue.write(1);
    State = EMGCalibration ; // veranderen naar MotorCalibration;
    ticker_hidscope.attach(&HIDScope, 0.001); //Ticker for Hidscope, different frequency compared to motors
    ticker_mainloop.attach(&main_loop,0.001); // change back to 0.001f //Run the function main_loop 1000 times per second       
    
   // motor_calibration();
    
    
// 6.2 While loop in main function**********************************************   
    while (true) { pc.printf("\r\n emg_tijd = %d, homing_tijd= %d", emg_tijd, homing_tijd); } //Is not used but has to remain in the code!!
    
}                  //Ending of Main() Function