derp

Dependencies:   FastPWM3 mbed

main.cpp

Committer:
bwang
Date:
2016-04-19
Revision:
21:3bfff547c5e2
Parent:
16:46703e957b30

File content as of revision 21:3bfff547c5e2:

#include "mbed.h"
#include "math.h"
#include "PositionSensor.h"
#include "FastPWM.h"

#define PWMA PA_8
#define PWMB PA_9
#define PWMC PA_10
#define EN PB_15

#define IA PA_4
#define IB PB_0

#define PI 3.141593f
#define CPR 4096
#define POS_OFFSET 4.5f

#define I_SCALE_RAW 25.0f //mv/A
#define R_UP 12000.0f //ohms
#define R_DOWN 3600.0f //ohms
#define R_BIAS 3600.0f //ohms
#define AVDD 3300.0f //mV

#define I_OFFSET (AVDD * R_DOWN * R_UP / (R_DOWN * R_UP + R_BIAS * (R_DOWN + R_UP)))
#define I_SCALE (R_BIAS * R_DOWN * I_SCALE_RAW / (R_DOWN * R_UP + R_BIAS * (R_DOWN + R_UP)))

#define K_LOOP 0.02
#define KI_BASE 0.008
#define BUS_VOLTAGE 200.0

#define KP (K_LOOP / BUS_VOLTAGE)
#define KI (KI_BASE * K_LOOP / BUS_VOLTAGE)

#define INTEGRAL_MAX 1.0f

#define Q_REF_MAX (-100.0)

FastPWM *a;
FastPWM *b;
FastPWM *c;
DigitalOut en(EN);

DigitalIn throttle_in(PB_8);
PositionSensorEncoder pos(CPR, 0);
Serial pc(USBTX, USBRX);

int adval1, adval2;
float ia, ib, ic, alpha, beta, q;
double vq = 0.0, q_integral = 0.0, last_q = 0.0;

float throttle_scaler = 0.0f;

int state = 0;
 
extern "C" void TIM1_UP_TIM10_IRQHandler(void) {
    if (TIM1->SR & TIM_SR_UIF) {
        float p = pos.GetElecPosition() - POS_OFFSET;
        if (p < 0) p += 2 * PI;
        
        //float pos_dac = 0.85f * p / (2 * PI) + 0.05f;
        //DAC->DHR12R2 = (unsigned int) (pos_dac * 4096);
        
        float sin_p = sinf(p);
        float cos_p = cosf(p);
        
        ADC1->CR2  |= 0x40000000; 
        volatile int delay;
        for (delay = 0; delay < 35; delay++);
        adval1 = ADC1->DR;
        adval2 = ADC2->DR;
        
        ia = ((float) adval1 / 4096.0f * AVDD - I_OFFSET) / I_SCALE;
        ib = ((float) adval2 / 4096.0f * AVDD - I_OFFSET) / I_SCALE;
        ic = -ia - ib;
        
        float u = ib;
        float v = ic;
    
        alpha = u;
        beta = 1 / sqrtf(3.0f) * u + 2 / sqrtf(3.0f) * v;
        
        q = -alpha * sin_p - beta * cos_p;
                
        if (throttle_in.read() == 0) {
            state = 0;
            throttle_scaler = 0.0f;
            en = 0;
        } else {
            state = 1;
            throttle_scaler = 1.0f;
            en = 1;
        }
        
        double q_err = Q_REF_MAX * (double) throttle_scaler - q;
        
        //DAC->DHR12R2 = (unsigned int) (q_err * 20 + 2048);
        
        q_integral += q_err * KI;
        if (q_integral > INTEGRAL_MAX) q_integral = INTEGRAL_MAX;
        if (q_integral < -INTEGRAL_MAX) q_integral = -INTEGRAL_MAX;
        
        vq = KP * q_err + q_integral;
        if (vq < -1.0f) vq = -1.0f;
        if (vq > 1.0f) vq = 1.0f;
        
        //DAC->DHR12R2 = (unsigned int) (vq * 2000 + 2048);
                    
        *a = 0.5f + 0.5f * vq * sinf(p);
        *b = 0.5f + 0.5f * vq * sinf(p + 2 * PI / 3);
        *c = 0.5f + 0.5f * vq * sinf(p - 2 * PI / 3);
    }
    TIM1->SR = 0x00;
}

int main() {
    //Enable clocks for GPIOs
    RCC->AHB1ENR |= RCC_AHB1ENR_GPIOAEN;
    RCC->AHB1ENR |= RCC_AHB1ENR_GPIOBEN;
    RCC->AHB1ENR |= RCC_AHB1ENR_GPIOCEN;
    
    RCC->APB2ENR |= RCC_APB2ENR_TIM1EN; //enable TIM1 clock
    
    a = new FastPWM(PWMA);
    b = new FastPWM(PWMB);
    c = new FastPWM(PWMC);
    
    NVIC_EnableIRQ(TIM1_UP_TIM10_IRQn); //Enable TIM1 IRQ

    TIM1->DIER |= TIM_DIER_UIE; //enable update interrupt
    //TIM1->CR1 = 0x40; //CMS = 10, interrupt only when counting up
    TIM1->CR1 |= TIM_CR1_ARPE; //autoreload on, 
    TIM1->RCR |= 0x01; //update event once per up/down count of tim1 
    TIM1->EGR |= TIM_EGR_UG;
    
    TIM1->PSC = 0x00; //no prescaler, timer counts up in sync with the peripheral clock
    TIM1->ARR = 0x2EE0;
    //TIM1->ARR = 0x1770; //15 Khz
    TIM1->CCER |= ~(TIM_CCER_CC1NP); //Interupt when low side is on.
    TIM1->CR1 |= TIM_CR1_CEN;
    
    TIM5->CR1 |= TIM_CR1_ARPE;
    TIM5->EGR |= TIM_EGR_UG;
    TIM5->PSC = 0x00;
    TIM5->ARR = 0xFFFFFFFF;
    TIM5->CR1 |= TIM_CR1_CEN;
    
    //ADC Setup
    RCC->APB2ENR |= RCC_APB2ENR_ADC1EN; // clock for ADC1
    RCC->APB2ENR |= RCC_APB2ENR_ADC2EN; // clock for ADC2
    
    ADC->CCR = 0x00000006; //Regular simultaneous mode, 3 channels
    
    ADC1->CR2 |= ADC_CR2_ADON; //ADC1 on
    ADC1->SQR3 = 0x0000004; //PA_4 as ADC1, sequence 0
    
    ADC2->CR2 |= ADC_CR2_ADON; //ADC2 ON
    ADC2->SQR3 = 0x00000008; //PB_0 as ADC2, sequence 1
    
    GPIOA->MODER |= (1 << 8);
    GPIOA->MODER |= (1 << 9);
    
    GPIOA->MODER |= (1 << 2);
    GPIOA->MODER |= (1 << 3);
    
    GPIOA->MODER |= (1 << 0);
    GPIOA->MODER |= (1 << 1);
    
    GPIOB->MODER |= (1 << 0);
    GPIOB->MODER |= (1 << 1);
    
    GPIOC->MODER |= (1 << 2);
    GPIOC->MODER |= (1 << 3);
    
    //DAC setup
    RCC->APB1ENR |= 0x20000000;
    DAC->CR |= DAC_CR_EN2;
    
    GPIOA->MODER |= (1 << 10);
    GPIOA->MODER |= (1 << 11);
    
    *a = 0.0f;
    *b = 0.0f;
    *c = 0.0f;
    
    en = 1;
    
    throttle_in.mode(PullUp);
    
    pc.baud(115200);
    
    pc.printf("%s\n\r", "THE DENTIST controller Rev. A");
    
    for (;;) {
    //    pc.printf("%d %f\n\r", state, throttle_scaler);
    //    wait_ms(100);
    }
}