ES2017 coursework 2

Dependencies:   PID

Fork of ES_CW2_Starter by Edward Stott

main.cpp

Committer:
david_s95
Date:
2017-03-10
Revision:
24:4032857546f4
Parent:
23:d48d51e5db97
Child:
25:9e6e870821d8

File content as of revision 24:4032857546f4:

#include "mbed.h"
#include "rtos.h"
#include <string>
#include "PID.h"

//PID controller configuration
float PIDrate = 0.1;
float Kc = 4.5;
float Ti = 0.1;
float Td = 0.5;
float speedControl = 0;
PID controller(Kc, Ti, Td, PIDrate);
Thread VPIDthread;

//Photointerrupter input pins
#define I1pin D2
#define I2pin D11
#define I3pin D12

//Incremental encoder input pins
#define CHA   D7
#define CHB   D8

//Motor Drive output pins   //Mask in output byte
#define L1Lpin D4           //0x01
#define L1Hpin D5           //0x02
#define L2Lpin D3           //0x04
#define L2Hpin D6           //0x08
#define L3Lpin D9           //0x10
#define L3Hpin D10          //0x20

//Define sized for command arrays
#define ARRAYSIZE 8

//Mapping from sequential drive states to motor phase outputs
/*
State   L1  L2  L3
0       H   -   L
1       -   H   L
2       L   H   -
3       L   -   H
4       -   L   H
5       H   L   -
6       -   -   -
7       -   -   -
*/
//Drive state to output table
//const int8_t driveTable[6] = {0x38, 0x2C, 0x0E, 0x0B, 0x23, 0x32};

//const int8_t driveTable[] = {0x12,0x18,0x09,0x21,0x24,0x06,0x00,0x00};


//Mapping from interrupter inputs to sequential rotor states. 0x00 and 0x07 are not valid
//const int8_t stateMap[] = {0x07,0x05,0x03,0x04,0x01,0x00,0x02,0x07};

const int8_t cwState[7] = {0x00, 0x23, 0x38, 0x32, 0x0E, 0x0B, 0x2C};
const int8_t AcwState[7] = {0x00, 0x0E, 0x23, 0x0B, 0x38, 0x2C, 0x32};


const int8_t FastStateCW[7] = {0x00, 0x32, 0x2C, 0x38, 0x0B, 0x23, 0x0E};
const int8_t FastStateACW[7] = {0x00, 0x2C, 0x0B, 0x0E, 0x32, 0x38, 0x23};

//Status LED
DigitalOut led1(LED1);

//Photointerrupter inputs
DigitalIn I1(I1pin);
//InterruptIn I1(I1pin);
InterruptIn I2(I2pin);
DigitalIn I3(I3pin);

InterruptIn qA(CHA);
InterruptIn qB(CHB);

//Motor Drive outputs
DigitalOut clk(LED1);
DigitalOut Direction(LED2);
DigitalOut testpin(D13);

//NOTE, BusOut declares things in reverse (ie, 0, 1, 2, 3) compared to binary represenation
BusOut motor(L1Hpin, L2Lpin, L2Hpin, L3Lpin, L3Hpin);//L1Lpin,
PwmOut singpin(L1Lpin);

//BusOut digitalpins(L1Hpin, L2Lpin, L2Hpin, L3Lpin, L3Hpin);
//PwmOut motor[6] = {L1Lpin, L1Hpin, L2Lpin, L2Hpin, L3Lpin, L3Hpin};
//PwmOut singpin(L1Lpin);

//motor[] = (singpin, digitalpins);

//PwmOut L1L(L1Lpin);
//DigitalOut L1H(L1Hpin);
//DigitalOut L2L(L2Lpin);
//DigitalOut L2H(L2Hpin);
//DigitalOut L3L(L3Lpin);
//DigitalOut L3H(L3Hpin);


//Timeout function for rotating at set speed
Timeout spinTimer;
float spinWait = 10;
float revsec = 0;

//Variables for spinning N revolutions
int8_t targetRevs = 0;
int8_t currRevs = 0;

//Timer used for calculating speed
Timer speedTimer;
float measuredRevs = 0, revtimer = 0;

Serial pc(SERIAL_TX, SERIAL_RX);

int8_t orState = 0;    //Rotor offset at motor state 0
int8_t intState = 0;
int8_t intStateOld = 0;
int position = 0;

int i=0;
int quadraturePosition=0;
bool spinCW=0;

//Set a given drive state
void motorOut(int8_t driveState)
{

    motor = 0x2A;

    if(revtimer<33) {
        clk=1;
        if(!spinCW) {
            motor = (FastStateACW[driveState]>>1);
            singpin = float(FastStateACW[driveState]&&0x01)/2;
        } else {
            motor = (FastStateCW[driveState]>>1);
            singpin = float(FastStateCW[driveState]&&0x01)/2;
        }
    } else {
        clk=0;
        if(!spinCW) {
            motor = (AcwState[driveState]>>1);
            singpin = float(AcwState[driveState]&&0x01)/2;
        } else {
            motor = (cwState[driveState]>>1);
            singpin = float(cwState[driveState]&&0x01)/2;
        }
    }
}

inline void motorStop()
{
    //revsec set to zero prevents recurring interrupt for constant speed
    revsec = 0;
    wait(spinWait);
    //0x2A turns all motor transistors off to prevent any power usage
    motor = 0x2A;
}

//Convert photointerrupter inputs to a rotor state
inline int8_t readRotorState()
{
    return (I1 + 2*I2 + 4*I3);
}

//Basic synchronisation routine
int8_t motorHome()
{
    //Put the motor in drive state 0 and wait for it to stabilise
    motor=cwState[1];
//    motorOut(1);
    wait(1.0);

    position = 0;

    //Get the rotor state
    return readRotorState();
}

void fixedSpeed()
{
    //If spinning is required, attach the necessary wait to the
    //timeout interrupt to call this function again and
    //keep the motor spinning at the right speed
    if(revsec) spinTimer.attach(&fixedSpeed, spinWait);

    intState = readRotorState();
    //Increment state machine to next state
    motorOut(intState);

}

//void spinToPos()
//{
//    currRevs = 0;
//    int remRevs = 0;
//    revsec = 50.0;
//    fixedSpeed();
//    while (targetRevs - currRevs > 100) {
//        printf("%d\r\n", currRevs);
//    }
//
//    remRevs = targetRevs - currRevs;
//    while (remRevs > 3) {
//        revsec = remRevs/3;
//        remRevs = targetRevs - currRevs;
//    }
//
//    motorStop();
//}

void rps()
{

//    clk=!clk;
    speedTimer.stop();
    revtimer = speedTimer.read_ms();
    speedTimer.reset();
    speedTimer.start();

    measuredRevs = 1000/(revtimer);
    quadraturePosition=0;
//    currRevs = currRevs + 1;

}

void VPID()
{
    controller.setMode(1);
    while(1) {
        controller.setSetPoint(revsec);
        controller.setProcessValue(measuredRevs);
        speedControl = controller.compute();
        if(speedControl<0) speedControl = -speedControl;
        else if (speedControl==0) speedControl = 1;
        spinWait = (1/speedControl)/6;
        Thread::wait(PIDrate);
    }
}

//void sing(float singFreq)
//{
//    motor = driveTable[1];
//    wait(5.0);
//    float singDelayus = 1000000/singFreq;
//    for(int count=0; count<singFreq; count++) {
//        motor = driveTable[2];
//        wait_us(singDelayus);
//        motor = driveTable[1];
//    }
//    singFreq = 15000;
//    singDelayus = 1000000/singFreq;
//    for(int count=0; count<singFreq; count++) {
//        motor = driveTable[2];
//        wait_us(singDelayus);
//        motor = driveTable[1];
//    }
//}


//Main function
int main()
{
    pc.printf("spin\n\r");
    spinWait = 0.01;
    motorStop();

    //Run the motor synchronisation
    orState = motorHome();
    //orState is subtracted from future rotor state inputs to align rotor and motor states

    pc.printf("Rotor origin: %x\n\r",orState);

    char command[ARRAYSIZE];
    int index=0;
    int units = 0, tens = 0, decimals = 0;
    char ch;
    testpin=0;
    int vartens = 0, varunits = 0, vardecs = 0;
    int hdrds = 0;
    float bias = 0;

    speedTimer.reset();
    speedTimer.start();
    I2.mode(PullNone);
    I2.fall(&rps);

    singpin.period_us(100);


    VPIDthread.start(VPID);

    while(1) {
//        clk = I2;
        //Toggle LED so we know something's happening
//        clk = !clk;

        //If there's a character to read from the serial port
        if (pc.readable()) {

            //Clear index counter and control variables
            index = 0;
//            revsec = spinWait = 0;

            //Read each value from the serial port until Enter key is pressed
            do {
                //Read character
                ch = pc.getc();
                //Print character to serial for visual feedback
                pc.putc(ch);
                //Add character to input array
                command[index++]=ch;  // put it into the value array and increment the index
                //d10 and d13 used for detecting Enter key on Windows/Unix/Mac
            } while(ch != 10 && ch != 13);

            //Start new line on terminal for printing data
            pc.putc('\n');
            pc.putc('\r');

            //Analyse the input string
            switch (command[0]) {
                    //If a V was typed...
                case 'V':
                    units = 0, tens = 0, decimals = 0;
                    //For each character received, subtract ASCII 0 from ASCII
                    //representation to obtain the integer value of the number
                    if(command[1]=='-') {
                        spinCW = 0;
                        //If decimal point is in the second character (eg, V-.1)
                        if(command[2]=='.') {
                            //Extract decimal rev/s
                            decimals = command[3] - '0';

                            //If decimal point is in the third character (eg, V-0.1)
                        } else if(command[3]=='.') {
                            units = command[2] - '0';
                            decimals = command[4] - '0';

                            //If decimal point is in the fourth character (eg, V-10.1)
                        } else if(command[4]=='.') {
                            tens = command[2] - '0';
                            units = command[3] - '0';
                            decimals = command[5] - '0';
                        }
                    } else {
                        spinCW = 1;
                        //If decimal point is in the second character (eg, V.1)
                        if(command[1]=='.') {
                            //Extract decimal rev/s
                            decimals = command[2] - '0';

                            //If decimal point is in the third character (eg, V0.1)
                        } else if(command[2]=='.') {
                            units = command[1] - '0';
                            decimals = command[3] - '0';

                            //If decimal point is in the fourth character (eg, V10.1)
                        } else if(command[3]=='.') {
                            tens = command[1] - '0';
                            units = command[2] - '0';
                            decimals = command[4] - '0';
                        }
                    }

                    //Calculate the number of revolutions per second required
                    revsec = float(tens)*10 + float(units) + float(decimals)/10;
                    //Calculate the required wait period

                    spinWait = (1/revsec)/6;
                    //Print values for verification
                    pc.printf("Rev/S: %2.4f\n\r", revsec);

                    //Run the function to start rotating at a fixed speed
                    fixedSpeed();
                    break;
                    //If anything unexpected was received

                case 'R':
                    hdrds = 0;
                    units = 0, tens = 0, decimals = 0;
                    //For each character received, subtract ASCII 0 from ASCII
                    //representation to obtain the integer value of the number
                    if(command[1]=='-') {
                        spinCW = 0;
                        //If decimal point is in the second character (eg, V-.1)
                        if(command[2]=='.') {
                            //Extract decimal rev/s
                            decimals = command[3] - '0';

                            //If decimal point is in the third character (eg, V-0.1)
                        } else if(command[3]=='.') {
                            units = command[2] - '0';
                            decimals = command[4] - '0';

                            //If decimal point is in the fourth character (eg, V-10.1)
                        } else if(command[4]=='.') {
                            tens = command[2] - '0';
                            units = command[3] - '0';
                            decimals = command[5] - '0';
                        } else if(command[5]=='.') {
                            hdrds = command[2] - '0';
                            tens = command[3] - '0';
                            units = command[4] - '0';
                            decimals = command[6] - '0';
                        }
                    } else {
                        spinCW = 1;
                        //If decimal point is in the second character (eg, V.1)
                        if(command[1]=='.') {
                            //Extract decimal rev/s
                            decimals = command[2] - '0';

                            //If decimal point is in the third character (eg, V0.1)
                        } else if(command[2]=='.') {
                            units = command[1] - '0';
                            decimals = command[3] - '0';

                            //If decimal point is in the fourth character (eg, V10.1)
                        } else if(command[3]=='.') {
                            tens = command[1] - '0';
                            units = command[2] - '0';
                            decimals = command[4] - '0';
                        } else if(command[4]=='.') {
                            hdrds = command[1] - '0';
                            tens = command[2] - '0';
                            units = command[3] - '0';
                            decimals = command[5] - '0';
                        }
                    }
                    //Calculate the number of revolutions required
                    targetRevs = float(hdrds)*100 + float(tens)*10 + float(units) + float(decimals)/10;
                    singpin.period_us(float(hdrds)*100 + float(tens)*10 + float(units) + float(decimals)/10);
                    //Print values for verification
                    pc.printf("Target revs: %2.4f\n\r", targetRevs);

                    //Run the function to start rotating at a fixed speed
//                    spinToPos();
                    break;


                case 's':
//                    pc.printf("Revs / sec: %2.2f\r", revs);
//                    printSpeed.attach(&speedo, 1.0);
                    printf("Measured: %2.3f, revsec: %2.3f\r\n", measuredRevs, revsec);
                    printf("PID: %2.3f\r\n", speedControl);
                    break;
                case 't':
//                    pc.printf("%d\n\r", pos);
                    break;

                case 'K':
                    vartens = command[1] - '0';
                    varunits = command[2] - '0';
                    vardecs = command[4] - '0';
                    Kc = float(vartens)*10 + float(varunits) + float(vardecs)/10;
                    printf("Kc: %2.1f\r\n", Kc);
                    controller.setTunings(Kc, Ti, Td);
//                    controller.setMode(1);
                    break;
                case 'i':
                    vartens = command[1] - '0';
                    varunits = command[2] - '0';
                    vardecs = command[4] - '0';
                    Ti = float(vartens)*10 + float(varunits) + float(vardecs)/10;
                    printf("Ti: %2.1f\r\n", Ti);
                    controller.setTunings(Kc, Ti, Td);
//                    controller.setMode(1);
                    break;
                case 'd':
                    vartens = command[1] - '0';
                    varunits = command[2] - '0';
                    vardecs = command[4] - '0';
                    Td = float(vartens)*10 + float(varunits) + float(vardecs)/10;
                    printf("Td: %2.1f\r\n", Td);
                    controller.setTunings(Kc, Ti, Td);
//                    controller.setMode(1);
                    break;
                case 'b':
                    if(command[1]=='.') {
                        //Extract decimal rev/s
                        vardecs = command[2] - '0';

                        //If decimal point is in the third character (eg, V-0.1)
                    } else if(command[2]=='.') {
                        varunits = command[1] - '0';
                        vardecs = command[3] - '0';

                        //If decimal point is in the fourth character (eg, V-10.1)
                    } else if(command[3]=='.') {
                        vartens = command[1] - '0';
                        varunits = command[2] - '0';
                        vardecs = command[4] - '0';
                    }
                    bias = float(vartens)*10 + float(varunits) + float(vardecs)/10;
                    printf("Bias: %2.1f\r\n", bias);
                    controller.setBias(bias);
                    break;
                case 'l':
//                    sing(261.63);
                    break;
                default:
                    //Set speed variables to zero to stop motor spinning
                    //Print error message
                    motorStop();
                    pc.printf("Error in received data 0\n\r");
                    motorStop();
                    break;
            }
        }
    }

}