yezhong yezhong / Motor_DRV8323RH_for_2019

11

Dependencies:   mbed-dev-f303 FastPWM3

main.cpp@53:89565c1d9115, 2021-12-27 (annotated)

Committer:
yezhong
Date:
Mon Dec 27 02:47:12 2021 +0000
Revision:
53:89565c1d9115
Parent:
52:8fff6f1a3f50
Child:
54:4ce8f97be6ae
1

Who changed what in which revision?

UserRevisionLine numberNew contents of line
benkatz 22:60276ba87ac6 1 /// high-bandwidth 3-phase motor control, for robots
benkatz 22:60276ba87ac6 2 /// Written by benkatz, with much inspiration from bayleyw, nkirkby, scolton, David Otten, and others
benkatz 22:60276ba87ac6 3 /// Hardware documentation can be found at build-its.blogspot.com
benkatz 22:60276ba87ac6 4 /// Written for the STM32F446, but can be implemented on other STM32 MCU's with some further register-diddling
benkatz 22:60276ba87ac6 5
benkatz 23:2adf23ee0305 6 #define REST_MODE 0
yezhong 53:89565c1d9115 7 #define CALIBRATION_MODE 1
benkatz 26:2b865c00d7e9 8 #define MOTOR_MODE 2
benkatz 23:2adf23ee0305 9 #define SETUP_MODE 4
benkatz 23:2adf23ee0305 10 #define ENCODER_MODE 5
yezhong 52:8fff6f1a3f50 11 #define SPEED_MODE 6
yezhong 52:8fff6f1a3f50 12 #define Position_MODE 7
benkatz 22:60276ba87ac6 13
benkatz 44:efcde0af8390 14 #define VERSION_NUM "1.6"
benkatz 26:2b865c00d7e9 15
benkatz 18:f1d56f4acb39 16
benkatz 26:2b865c00d7e9 17 float __float_reg[64]; // Floats stored in flash
benkatz 26:2b865c00d7e9 18 int __int_reg[256]; // Ints stored in flash. Includes position sensor calibration lookup table
benkatz 17:3c5df2982199 19
benkatz 0:4e1c4df6aabd 20 #include "mbed.h"
benkatz 0:4e1c4df6aabd 21 #include "PositionSensor.h"
benkatz 20:bf9ea5125d52 22 #include "structs.h"
benkatz 20:bf9ea5125d52 23 #include "foc.h"
benkatz 22:60276ba87ac6 24 #include "calibration.h"
benkatz 20:bf9ea5125d52 25 #include "hw_setup.h"
benkatz 23:2adf23ee0305 26 #include "math_ops.h"
benkatz 20:bf9ea5125d52 27 #include "current_controller_config.h"
benkatz 20:bf9ea5125d52 28 #include "hw_config.h"
benkatz 20:bf9ea5125d52 29 #include "motor_config.h"
benkatz 23:2adf23ee0305 30 #include "stm32f4xx_flash.h"
benkatz 23:2adf23ee0305 31 #include "FlashWriter.h"
benkatz 23:2adf23ee0305 32 #include "user_config.h"
benkatz 23:2adf23ee0305 33 #include "PreferenceWriter.h"
benkatz 42:738fa01b0346 34 #include "CAN_com.h"
benkatz 37:c0f352d6e8e3 35
benkatz 26:2b865c00d7e9 36
benkatz 23:2adf23ee0305 37 PreferenceWriter prefs(6);
benkatz 9:d7eb815cb057 38
benkatz 20:bf9ea5125d52 39 GPIOStruct gpio;
benkatz 20:bf9ea5125d52 40 ControllerStruct controller;
benkatz 20:bf9ea5125d52 41 COMStruct com;
benkatz 37:c0f352d6e8e3 42 ObserverStruct observer;
benkatz 43:dfb72608639c 43 Serial pc(PA_2, PA_3);
benkatz 9:d7eb815cb057 44
benkatz 17:3c5df2982199 45
yezhong 52:8fff6f1a3f50 46 CAN can(PB_8, PB_9, 800000); // CAN Rx pin name, CAN Tx pin name, 1000kbps
benkatz 26:2b865c00d7e9 47 CANMessage rxMsg;
benkatz 26:2b865c00d7e9 48 CANMessage txMsg;
benkatz 23:2adf23ee0305 49
benkatz 20:bf9ea5125d52 50
benkatz 8:10ae7bc88d6e 51
Rushu 47:55bdc4d5096b 52 PositionSensorAM5147 spi(16384, 0.0, NPP); //14 bits encoder, 21 NPP
benkatz 20:bf9ea5125d52 53
benkatz 23:2adf23ee0305 54 volatile int count = 0;
benkatz 23:2adf23ee0305 55 volatile int state = REST_MODE;
benkatz 23:2adf23ee0305 56 volatile int state_change;
benkatz 20:bf9ea5125d52 57
benkatz 26:2b865c00d7e9 58 void onMsgReceived() {
benkatz 26:2b865c00d7e9 59 //msgAvailable = true;
benkatz 45:aadebe074af6 60
benkatz 26:2b865c00d7e9 61 can.read(rxMsg);
benkatz 28:8c7e29f719c5 62 if((rxMsg.id == CAN_ID)){
benkatz 28:8c7e29f719c5 63 controller.timeout = 0;
benkatz 28:8c7e29f719c5 64 if(((rxMsg.data[0]==0xFF) & (rxMsg.data[1]==0xFF) & (rxMsg.data[2]==0xFF) & (rxMsg.data[3]==0xFF) & (rxMsg.data[4]==0xFF) & (rxMsg.data[5]==0xFF) & (rxMsg.data[6]==0xFF) & (rxMsg.data[7]==0xFC))){
benkatz 28:8c7e29f719c5 65 state = MOTOR_MODE;
benkatz 28:8c7e29f719c5 66 state_change = 1;
benkatz 28:8c7e29f719c5 67 }
benkatz 28:8c7e29f719c5 68 else if(((rxMsg.data[0]==0xFF) & (rxMsg.data[1]==0xFF) & (rxMsg.data[2]==0xFF) & (rxMsg.data[3]==0xFF) * (rxMsg.data[4]==0xFF) & (rxMsg.data[5]==0xFF) & (rxMsg.data[6]==0xFF) & (rxMsg.data[7]==0xFD))){
benkatz 28:8c7e29f719c5 69 state = REST_MODE;
benkatz 28:8c7e29f719c5 70 state_change = 1;
yezhong 49:8dcdb8a89d0e 71 gpio.led->write(0);
benkatz 28:8c7e29f719c5 72 }
yezhong 52:8fff6f1a3f50 73 //************WYC ADD **********2021.11.04**************//
yezhong 52:8fff6f1a3f50 74 else if(((rxMsg.data[0]==0xFF) & (rxMsg.data[1]==0xFF) & (rxMsg.data[2]==0xFF) & (rxMsg.data[3]==0xFF) * (rxMsg.data[4]==0xFF) & (rxMsg.data[5]==0xFF) & (rxMsg.data[6]==0xFF) & (rxMsg.data[7]==0xFA))){
yezhong 52:8fff6f1a3f50 75 state = SPEED_MODE;
yezhong 52:8fff6f1a3f50 76 state_change = 1;
yezhong 52:8fff6f1a3f50 77 gpio.led->write(1);
yezhong 52:8fff6f1a3f50 78 }
yezhong 52:8fff6f1a3f50 79 //************WYC ADD **********2021.11.04**************//
yezhong 52:8fff6f1a3f50 80 //************YZ ADD **********2021.11.05**************//
yezhong 52:8fff6f1a3f50 81 else if(((rxMsg.data[0]==0xFF) & (rxMsg.data[1]==0xFF) & (rxMsg.data[2]==0xFF) & (rxMsg.data[3]==0xFF) * (rxMsg.data[4]==0xFF) & (rxMsg.data[5]==0xFF) & (rxMsg.data[6]==0xFF) & (rxMsg.data[7]==0xFB))){
yezhong 52:8fff6f1a3f50 82 state = Position_MODE;
yezhong 52:8fff6f1a3f50 83 state_change = 1;
yezhong 52:8fff6f1a3f50 84 gpio.led->write(1);
yezhong 52:8fff6f1a3f50 85 }
yezhong 52:8fff6f1a3f50 86 //************YZ ADD **********2021.11.05**************//
benkatz 28:8c7e29f719c5 87 else if(((rxMsg.data[0]==0xFF) & (rxMsg.data[1]==0xFF) & (rxMsg.data[2]==0xFF) & (rxMsg.data[3]==0xFF) * (rxMsg.data[4]==0xFF) & (rxMsg.data[5]==0xFF) & (rxMsg.data[6]==0xFF) & (rxMsg.data[7]==0xFE))){
yezhong 53:89565c1d9115 88 /*
yezhong 53:89565c1d9115 89 quanshu=spi.aa;
yezhong 53:89565c1d9115 90 if (!prefs.ready()) prefs.open();
yezhong 53:89565c1d9115 91 prefs.flush(); // Write new prefs to flash
yezhong 53:89565c1d9115 92 prefs.close();
yezhong 53:89565c1d9115 93 prefs.load();
yezhong 53:89565c1d9115 94 */
benkatz 28:8c7e29f719c5 95 spi.ZeroPosition();
yezhong 53:89565c1d9115 96
benkatz 28:8c7e29f719c5 97 }
yezhong 52:8fff6f1a3f50 98 /*
benkatz 28:8c7e29f719c5 99 else if(state == MOTOR_MODE){
benkatz 28:8c7e29f719c5 100 unpack_cmd(rxMsg, &controller);
benkatz 28:8c7e29f719c5 101 }
yezhong 52:8fff6f1a3f50 102 */
yezhong 52:8fff6f1a3f50 103 else if(state == MOTOR_MODE ||state == SPEED_MODE||state == Position_MODE)
yezhong 52:8fff6f1a3f50 104 { //WYC ADD 2021.11.04
yezhong 52:8fff6f1a3f50 105 unpack_cmd(rxMsg, &controller);
yezhong 52:8fff6f1a3f50 106 }
benkatz 37:c0f352d6e8e3 107 pack_reply(&txMsg, controller.theta_mech, controller.dtheta_mech, controller.i_q_filt*KT_OUT);
benkatz 37:c0f352d6e8e3 108 can.write(txMsg);
benkatz 28:8c7e29f719c5 109 }
benkatz 26:2b865c00d7e9 110
benkatz 26:2b865c00d7e9 111 }
benkatz 26:2b865c00d7e9 112
benkatz 23:2adf23ee0305 113 void enter_menu_state(void){
benkatz 23:2adf23ee0305 114 printf("\n\r\n\r\n\r");
benkatz 23:2adf23ee0305 115 printf(" Commands:\n\r");
benkatz 44:efcde0af8390 116 wait_us(10);
benkatz 26:2b865c00d7e9 117 printf(" m - Motor Mode\n\r");
benkatz 44:efcde0af8390 118 wait_us(10);
benkatz 23:2adf23ee0305 119 printf(" c - Calibrate Encoder\n\r");
benkatz 44:efcde0af8390 120 wait_us(10);
benkatz 23:2adf23ee0305 121 printf(" s - Setup\n\r");
benkatz 44:efcde0af8390 122 wait_us(10);
benkatz 23:2adf23ee0305 123 printf(" e - Display Encoder\n\r");
benkatz 44:efcde0af8390 124 wait_us(10);
benkatz 37:c0f352d6e8e3 125 printf(" z - Set Zero Position\n\r");
benkatz 44:efcde0af8390 126 wait_us(10);
benkatz 23:2adf23ee0305 127 printf(" esc - Exit to Menu\n\r");
benkatz 44:efcde0af8390 128 wait_us(10);
benkatz 23:2adf23ee0305 129 state_change = 0;
benkatz 25:f5741040c4bb 130 gpio.enable->write(0);
benkatz 37:c0f352d6e8e3 131 gpio.led->write(0);
benkatz 23:2adf23ee0305 132 }
benkatz 24:58c2d7571207 133
benkatz 24:58c2d7571207 134 void enter_setup_state(void){
benkatz 24:58c2d7571207 135 printf("\n\r\n\r Configuration Options \n\r\n\n");
benkatz 44:efcde0af8390 136 wait_us(10);
benkatz 28:8c7e29f719c5 137 printf(" %-4s %-31s %-5s %-6s %-5s\n\r\n\r", "prefix", "parameter", "min", "max", "current value");
benkatz 44:efcde0af8390 138 wait_us(10);
benkatz 28:8c7e29f719c5 139 printf(" %-4s %-31s %-5s %-6s %.1f\n\r", "b", "Current Bandwidth (Hz)", "100", "2000", I_BW);
benkatz 44:efcde0af8390 140 wait_us(10);
benkatz 28:8c7e29f719c5 141 printf(" %-4s %-31s %-5s %-6s %-5i\n\r", "i", "CAN ID", "0", "127", CAN_ID);
benkatz 44:efcde0af8390 142 wait_us(10);
benkatz 28:8c7e29f719c5 143 printf(" %-4s %-31s %-5s %-6s %-5i\n\r", "m", "CAN Master ID", "0", "127", CAN_MASTER);
benkatz 44:efcde0af8390 144 wait_us(10);
benkatz 28:8c7e29f719c5 145 printf(" %-4s %-31s %-5s %-6s %.1f\n\r", "l", "Torque Limit (N-m)", "0.0", "18.0", TORQUE_LIMIT);
benkatz 44:efcde0af8390 146 wait_us(10);
benkatz 28:8c7e29f719c5 147 printf(" %-4s %-31s %-5s %-6s %d\n\r", "t", "CAN Timeout (cycles)(0 = none)", "0", "100000", CAN_TIMEOUT);
benkatz 44:efcde0af8390 148 wait_us(10);
benkatz 24:58c2d7571207 149 printf("\n\r To change a value, type 'prefix''value''ENTER'\n\r i.e. 'b1000''ENTER'\n\r\n\r");
benkatz 44:efcde0af8390 150 wait_us(10);
benkatz 24:58c2d7571207 151 state_change = 0;
benkatz 24:58c2d7571207 152 }
benkatz 22:60276ba87ac6 153
benkatz 23:2adf23ee0305 154 void enter_torque_mode(void){
benkatz 37:c0f352d6e8e3 155 controller.ovp_flag = 0;
benkatz 28:8c7e29f719c5 156 gpio.enable->write(1); // Enable gate drive
benkatz 28:8c7e29f719c5 157 reset_foc(&controller); // Tesets integrators, and other control loop parameters
benkatz 28:8c7e29f719c5 158 wait(.001);
benkatz 23:2adf23ee0305 159 controller.i_d_ref = 0;
benkatz 28:8c7e29f719c5 160 controller.i_q_ref = 0; // Current Setpoints
benkatz 37:c0f352d6e8e3 161 gpio.led->write(1); // Turn on status LED
benkatz 25:f5741040c4bb 162 state_change = 0;
benkatz 28:8c7e29f719c5 163 printf("\n\r Entering Motor Mode \n\r");
benkatz 23:2adf23ee0305 164 }
yezhong 52:8fff6f1a3f50 165
yezhong 52:8fff6f1a3f50 166 //****************WYC ADD ************2021.11.04**********//
yezhong 52:8fff6f1a3f50 167 void enter_speed_mode(void){
yezhong 52:8fff6f1a3f50 168 controller.ovp_flag = 0;
yezhong 52:8fff6f1a3f50 169 gpio.enable->write(1); // Enable gate drive
yezhong 52:8fff6f1a3f50 170 reset_foc(&controller); // Tesets integrators, and other control loop parameters
yezhong 52:8fff6f1a3f50 171 wait(.001);
yezhong 52:8fff6f1a3f50 172 controller.i_d_ref = 0;
yezhong 52:8fff6f1a3f50 173 controller.i_q_ref = 0; // Current Setpoints
yezhong 52:8fff6f1a3f50 174 gpio.led->write(1); // Turn on status LED
yezhong 52:8fff6f1a3f50 175 state_change = 0;
yezhong 52:8fff6f1a3f50 176 printf("\n\r Entering SPEED Mode \n\r");
yezhong 52:8fff6f1a3f50 177 }
yezhong 52:8fff6f1a3f50 178 //****************WYC ADD ************2021.11.04**********//
yezhong 52:8fff6f1a3f50 179
yezhong 52:8fff6f1a3f50 180 //****************YZ ADD ************2021.11.05**********//
yezhong 52:8fff6f1a3f50 181 void enter_Position_mode(void){
yezhong 52:8fff6f1a3f50 182 controller.ovp_flag = 0;
yezhong 52:8fff6f1a3f50 183 gpio.enable->write(1); // Enable gate drive
yezhong 52:8fff6f1a3f50 184 reset_foc(&controller); // Tesets integrators, and other control loop parameters
yezhong 52:8fff6f1a3f50 185 wait(.001);
yezhong 52:8fff6f1a3f50 186 controller.i_d_ref = 0;
yezhong 52:8fff6f1a3f50 187 controller.i_q_ref = 0; // Current Setpoints
yezhong 52:8fff6f1a3f50 188 gpio.led->write(1); // Turn on status LED
yezhong 52:8fff6f1a3f50 189 state_change = 0;
yezhong 52:8fff6f1a3f50 190 printf("\n\r Entering Position Mode \n\r");
yezhong 52:8fff6f1a3f50 191 }
yezhong 52:8fff6f1a3f50 192 //****************YZ ADD ************2021.11.05**********//
benkatz 22:60276ba87ac6 193
benkatz 23:2adf23ee0305 194 void calibrate(void){
benkatz 25:f5741040c4bb 195 gpio.enable->write(1); // Enable gate drive
benkatz 37:c0f352d6e8e3 196 gpio.led->write(1); // Turn on status LED
benkatz 25:f5741040c4bb 197 order_phases(&spi, &gpio, &controller, &prefs); // Check phase ordering
benkatz 25:f5741040c4bb 198 calibrate(&spi, &gpio, &controller, &prefs); // Perform calibration procedure
benkatz 37:c0f352d6e8e3 199 gpio.led->write(0);; // Turn off status LED
benkatz 23:2adf23ee0305 200 wait(.2);
benkatz 25:f5741040c4bb 201 gpio.enable->write(0); // Turn off gate drive
benkatz 23:2adf23ee0305 202 printf("\n\r Calibration complete. Press 'esc' to return to menu\n\r");
benkatz 23:2adf23ee0305 203 state_change = 0;
benkatz 23:2adf23ee0305 204 }
benkatz 23:2adf23ee0305 205
benkatz 23:2adf23ee0305 206 void print_encoder(void){
benkatz 23:2adf23ee0305 207 printf(" Mechanical Angle: %f Electrical Angle: %f Raw: %d\n\r", spi.GetMechPosition(), spi.GetElecPosition(), spi.GetRawPosition());
benkatz 23:2adf23ee0305 208 wait(.05);
benkatz 22:60276ba87ac6 209 }
benkatz 20:bf9ea5125d52 210
benkatz 23:2adf23ee0305 211 /// Current Sampling Interrupt ///
benkatz 23:2adf23ee0305 212 /// This runs at 40 kHz, regardless of of the mode the controller is in ///
benkatz 2:8724412ad628 213 extern "C" void TIM1_UP_TIM10_IRQHandler(void) {
benkatz 2:8724412ad628 214 if (TIM1->SR & TIM_SR_UIF ) {
benkatz 23:2adf23ee0305 215
benkatz 23:2adf23ee0305 216 ///Sample current always ///
benkatz 25:f5741040c4bb 217 ADC1->CR2 |= 0x40000000; // Begin sample and conversion
benkatz 22:60276ba87ac6 218 //volatile int delay;
benkatz 20:bf9ea5125d52 219 //for (delay = 0; delay < 55; delay++);
benkatz 37:c0f352d6e8e3 220 controller.adc2_raw = ADC2->DR; // Read ADC Data Registers
benkatz 23:2adf23ee0305 221 controller.adc1_raw = ADC1->DR;
benkatz 37:c0f352d6e8e3 222 controller.adc3_raw = ADC3->DR;
benkatz 45:aadebe074af6 223 spi.Sample(DT); // sample position sensor
benkatz 37:c0f352d6e8e3 224 controller.theta_elec = spi.GetElecPosition();
benkatz 37:c0f352d6e8e3 225 controller.theta_mech = (1.0f/GR)*spi.GetMechPosition();
benkatz 37:c0f352d6e8e3 226 controller.dtheta_mech = (1.0f/GR)*spi.GetMechVelocity();
benkatz 37:c0f352d6e8e3 227 controller.dtheta_elec = spi.GetElecVelocity();
benkatz 37:c0f352d6e8e3 228 controller.v_bus = 0.95f*controller.v_bus + 0.05f*((float)controller.adc3_raw)*V_SCALE;
benkatz 23:2adf23ee0305 229 ///
benkatz 20:bf9ea5125d52 230
benkatz 23:2adf23ee0305 231 /// Check state machine state, and run the appropriate function ///
benkatz 23:2adf23ee0305 232 switch(state){
benkatz 37:c0f352d6e8e3 233 case REST_MODE: // Do nothing
benkatz 23:2adf23ee0305 234 if(state_change){
benkatz 23:2adf23ee0305 235 enter_menu_state();
benkatz 23:2adf23ee0305 236 }
benkatz 23:2adf23ee0305 237 break;
benkatz 22:60276ba87ac6 238
yezhong 53:89565c1d9115 239 case CALIBRATION_MODE: // Run encoder calibration procedure
benkatz 23:2adf23ee0305 240 if(state_change){
benkatz 23:2adf23ee0305 241 calibrate();
benkatz 23:2adf23ee0305 242 }
benkatz 23:2adf23ee0305 243 break;
benkatz 23:2adf23ee0305 244
benkatz 26:2b865c00d7e9 245 case MOTOR_MODE: // Run torque control
benkatz 25:f5741040c4bb 246 if(state_change){
benkatz 25:f5741040c4bb 247 enter_torque_mode();
benkatz 28:8c7e29f719c5 248 count = 0;
benkatz 25:f5741040c4bb 249 }
benkatz 28:8c7e29f719c5 250 else{
benkatz 37:c0f352d6e8e3 251 /*
benkatz 37:c0f352d6e8e3 252 if(controller.v_bus>28.0f){ //Turn of gate drive if bus voltage is too high, to prevent FETsplosion if the bus is cut during regen
benkatz 37:c0f352d6e8e3 253 gpio.enable->write(0);
benkatz 37:c0f352d6e8e3 254 controller.ovp_flag = 1;
benkatz 37:c0f352d6e8e3 255 state = REST_MODE;
benkatz 37:c0f352d6e8e3 256 state_change = 1;
benkatz 37:c0f352d6e8e3 257 printf("OVP Triggered!\n\r");
benkatz 37:c0f352d6e8e3 258 }
benkatz 37:c0f352d6e8e3 259 */
benkatz 37:c0f352d6e8e3 260
benkatz 40:cd7e837b2b93 261 torque_control(&controller);
benkatz 28:8c7e29f719c5 262 if((controller.timeout > CAN_TIMEOUT) && (CAN_TIMEOUT > 0)){
benkatz 28:8c7e29f719c5 263 controller.i_d_ref = 0;
benkatz 28:8c7e29f719c5 264 controller.i_q_ref = 0;
benkatz 37:c0f352d6e8e3 265 controller.kp = 0;
benkatz 37:c0f352d6e8e3 266 controller.kd = 0;
benkatz 37:c0f352d6e8e3 267 controller.t_ff = 0;
benkatz 28:8c7e29f719c5 268 }
benkatz 37:c0f352d6e8e3 269 commutate(&controller, &observer, &gpio, controller.theta_elec); // Run current loop
benkatz 28:8c7e29f719c5 270 controller.timeout += 1;
benkatz 38:67e4e1453a4b 271
benkatz 39:3580a907ef93 272 /*
benkatz 37:c0f352d6e8e3 273 count++;
benkatz 40:cd7e837b2b93 274 if(count == 4000){
benkatz 40:cd7e837b2b93 275 printf("%.4f\n\r", controller.dtheta_mech);
benkatz 32:ccac5da77844 276 count = 0;
benkatz 23:2adf23ee0305 277 }
benkatz 39:3580a907ef93 278 */
benkatz 38:67e4e1453a4b 279
benkatz 37:c0f352d6e8e3 280
benkatz 37:c0f352d6e8e3 281 }
benkatz 23:2adf23ee0305 282 break;
yezhong 52:8fff6f1a3f50 283 /// wyc////
yezhong 52:8fff6f1a3f50 284 case SPEED_MODE: // Run SPEED control WYC 2021.11.04
yezhong 52:8fff6f1a3f50 285 if(state_change){
yezhong 52:8fff6f1a3f50 286 enter_speed_mode();
yezhong 52:8fff6f1a3f50 287 }
yezhong 52:8fff6f1a3f50 288 else{
yezhong 52:8fff6f1a3f50 289 velocity_control(&controller);
yezhong 52:8fff6f1a3f50 290 if((controller.timeout > CAN_TIMEOUT) && (CAN_TIMEOUT > 0)){
yezhong 52:8fff6f1a3f50 291 controller.i_d_ref = 0;
yezhong 52:8fff6f1a3f50 292 controller.i_q_ref = 0;
yezhong 52:8fff6f1a3f50 293 controller.kp = 0;
yezhong 52:8fff6f1a3f50 294 controller.kd = 0;
yezhong 52:8fff6f1a3f50 295 controller.t_ff = 0;
yezhong 52:8fff6f1a3f50 296 }
yezhong 52:8fff6f1a3f50 297 commutate(&controller, &observer, &gpio, controller.theta_elec); // Run current loop
yezhong 52:8fff6f1a3f50 298 controller.timeout += 1;
yezhong 52:8fff6f1a3f50 299 }
yezhong 52:8fff6f1a3f50 300 break;
yezhong 52:8fff6f1a3f50 301 ////wyc////////
yezhong 52:8fff6f1a3f50 302
yezhong 52:8fff6f1a3f50 303 /// YZ////
yezhong 52:8fff6f1a3f50 304 case Position_MODE: // Run Position control WYC 2021.11.05
yezhong 52:8fff6f1a3f50 305 if(state_change){
yezhong 52:8fff6f1a3f50 306 enter_Position_mode();
yezhong 52:8fff6f1a3f50 307 }
yezhong 52:8fff6f1a3f50 308 else{
yezhong 52:8fff6f1a3f50 309 Position_control(&controller);
yezhong 52:8fff6f1a3f50 310 if((controller.timeout > CAN_TIMEOUT) && (CAN_TIMEOUT > 0)){
yezhong 52:8fff6f1a3f50 311 controller.i_d_ref = 0;
yezhong 52:8fff6f1a3f50 312 controller.i_q_ref = 0;
yezhong 52:8fff6f1a3f50 313 controller.kp = 0;
yezhong 52:8fff6f1a3f50 314 controller.kd = 0;
yezhong 52:8fff6f1a3f50 315 controller.t_ff = 0;
yezhong 52:8fff6f1a3f50 316 }
yezhong 52:8fff6f1a3f50 317 commutate(&controller, &observer, &gpio, controller.theta_elec); // Run current loop
yezhong 52:8fff6f1a3f50 318 controller.timeout += 1;
yezhong 52:8fff6f1a3f50 319 }
yezhong 52:8fff6f1a3f50 320 break;
yezhong 52:8fff6f1a3f50 321 ////YZ////////
benkatz 23:2adf23ee0305 322 case SETUP_MODE:
benkatz 23:2adf23ee0305 323 if(state_change){
benkatz 24:58c2d7571207 324 enter_setup_state();
benkatz 23:2adf23ee0305 325 }
benkatz 23:2adf23ee0305 326 break;
benkatz 23:2adf23ee0305 327 case ENCODER_MODE:
benkatz 23:2adf23ee0305 328 print_encoder();
benkatz 23:2adf23ee0305 329 break;
benkatz 37:c0f352d6e8e3 330 }
benkatz 2:8724412ad628 331 }
benkatz 23:2adf23ee0305 332 TIM1->SR = 0x0; // reset the status register
benkatz 2:8724412ad628 333 }
benkatz 0:4e1c4df6aabd 334
benkatz 25:f5741040c4bb 335
benkatz 24:58c2d7571207 336 char cmd_val[8] = {0};
benkatz 24:58c2d7571207 337 char cmd_id = 0;
benkatz 25:f5741040c4bb 338 char char_count = 0;
benkatz 24:58c2d7571207 339
benkatz 25:f5741040c4bb 340 /// Manage state machine with commands from serial terminal or configurator gui ///
benkatz 25:f5741040c4bb 341 /// Called when data received over serial ///
benkatz 23:2adf23ee0305 342 void serial_interrupt(void){
benkatz 23:2adf23ee0305 343 while(pc.readable()){
benkatz 23:2adf23ee0305 344 char c = pc.getc();
benkatz 25:f5741040c4bb 345 if(c == 27){
benkatz 25:f5741040c4bb 346 state = REST_MODE;
benkatz 25:f5741040c4bb 347 state_change = 1;
benkatz 25:f5741040c4bb 348 char_count = 0;
benkatz 25:f5741040c4bb 349 cmd_id = 0;
benkatz 37:c0f352d6e8e3 350 gpio.led->write(0);;
benkatz 25:f5741040c4bb 351 for(int i = 0; i<8; i++){cmd_val[i] = 0;}
benkatz 25:f5741040c4bb 352 }
benkatz 24:58c2d7571207 353 if(state == REST_MODE){
benkatz 23:2adf23ee0305 354 switch (c){
benkatz 23:2adf23ee0305 355 case 'c':
benkatz 23:2adf23ee0305 356 state = CALIBRATION_MODE;
benkatz 23:2adf23ee0305 357 state_change = 1;
benkatz 23:2adf23ee0305 358 break;
benkatz 26:2b865c00d7e9 359 case 'm':
benkatz 26:2b865c00d7e9 360 state = MOTOR_MODE;
benkatz 23:2adf23ee0305 361 state_change = 1;
benkatz 23:2adf23ee0305 362 break;
benkatz 23:2adf23ee0305 363 case 'e':
benkatz 23:2adf23ee0305 364 state = ENCODER_MODE;
benkatz 23:2adf23ee0305 365 state_change = 1;
benkatz 23:2adf23ee0305 366 break;
benkatz 23:2adf23ee0305 367 case 's':
benkatz 23:2adf23ee0305 368 state = SETUP_MODE;
benkatz 23:2adf23ee0305 369 state_change = 1;
benkatz 23:2adf23ee0305 370 break;
benkatz 37:c0f352d6e8e3 371 case 'z':
benkatz 37:c0f352d6e8e3 372 spi.SetMechOffset(0);
benkatz 45:aadebe074af6 373 spi.Sample(DT);
benkatz 37:c0f352d6e8e3 374 wait_us(20);
benkatz 37:c0f352d6e8e3 375 M_OFFSET = spi.GetMechPosition();
benkatz 37:c0f352d6e8e3 376 if (!prefs.ready()) prefs.open();
benkatz 37:c0f352d6e8e3 377 prefs.flush(); // Write new prefs to flash
benkatz 37:c0f352d6e8e3 378 prefs.close();
benkatz 37:c0f352d6e8e3 379 prefs.load();
benkatz 37:c0f352d6e8e3 380 spi.SetMechOffset(M_OFFSET);
benkatz 37:c0f352d6e8e3 381 printf("\n\r Saved new zero position: %.4f\n\r\n\r", M_OFFSET);
benkatz 37:c0f352d6e8e3 382
benkatz 37:c0f352d6e8e3 383 break;
benkatz 37:c0f352d6e8e3 384 }
benkatz 37:c0f352d6e8e3 385
benkatz 24:58c2d7571207 386 }
benkatz 24:58c2d7571207 387 else if(state == SETUP_MODE){
benkatz 25:f5741040c4bb 388 if(c == 13){
benkatz 24:58c2d7571207 389 switch (cmd_id){
benkatz 24:58c2d7571207 390 case 'b':
benkatz 24:58c2d7571207 391 I_BW = fmaxf(fminf(atof(cmd_val), 2000.0f), 100.0f);
benkatz 24:58c2d7571207 392 break;
benkatz 24:58c2d7571207 393 case 'i':
benkatz 24:58c2d7571207 394 CAN_ID = atoi(cmd_val);
benkatz 24:58c2d7571207 395 break;
benkatz 26:2b865c00d7e9 396 case 'm':
benkatz 26:2b865c00d7e9 397 CAN_MASTER = atoi(cmd_val);
benkatz 26:2b865c00d7e9 398 break;
benkatz 24:58c2d7571207 399 case 'l':
benkatz 24:58c2d7571207 400 TORQUE_LIMIT = fmaxf(fminf(atof(cmd_val), 18.0f), 0.0f);
benkatz 24:58c2d7571207 401 break;
benkatz 28:8c7e29f719c5 402 case 't':
benkatz 28:8c7e29f719c5 403 CAN_TIMEOUT = atoi(cmd_val);
benkatz 28:8c7e29f719c5 404 break;
benkatz 24:58c2d7571207 405 default:
benkatz 24:58c2d7571207 406 printf("\n\r '%c' Not a valid command prefix\n\r\n\r", cmd_id);
benkatz 24:58c2d7571207 407 break;
benkatz 24:58c2d7571207 408 }
benkatz 24:58c2d7571207 409
benkatz 24:58c2d7571207 410 if (!prefs.ready()) prefs.open();
benkatz 24:58c2d7571207 411 prefs.flush(); // Write new prefs to flash
benkatz 24:58c2d7571207 412 prefs.close();
benkatz 24:58c2d7571207 413 prefs.load();
benkatz 24:58c2d7571207 414 state_change = 1;
benkatz 24:58c2d7571207 415 char_count = 0;
benkatz 24:58c2d7571207 416 cmd_id = 0;
benkatz 24:58c2d7571207 417 for(int i = 0; i<8; i++){cmd_val[i] = 0;}
benkatz 24:58c2d7571207 418 }
benkatz 24:58c2d7571207 419 else{
benkatz 24:58c2d7571207 420 if(char_count == 0){cmd_id = c;}
benkatz 24:58c2d7571207 421 else{
benkatz 24:58c2d7571207 422 cmd_val[char_count-1] = c;
benkatz 24:58c2d7571207 423
benkatz 24:58c2d7571207 424 }
benkatz 24:58c2d7571207 425 pc.putc(c);
benkatz 24:58c2d7571207 426 char_count++;
benkatz 23:2adf23ee0305 427 }
benkatz 23:2adf23ee0305 428 }
benkatz 24:58c2d7571207 429 else if (state == ENCODER_MODE){
benkatz 24:58c2d7571207 430 switch (c){
benkatz 24:58c2d7571207 431 case 27:
benkatz 24:58c2d7571207 432 state = REST_MODE;
benkatz 24:58c2d7571207 433 state_change = 1;
benkatz 24:58c2d7571207 434 break;
benkatz 24:58c2d7571207 435 }
benkatz 24:58c2d7571207 436 }
benkatz 24:58c2d7571207 437
benkatz 24:58c2d7571207 438 }
benkatz 22:60276ba87ac6 439 }
benkatz 0:4e1c4df6aabd 440
benkatz 0:4e1c4df6aabd 441 int main() {
benkatz 45:aadebe074af6 442
benkatz 20:bf9ea5125d52 443 controller.v_bus = V_BUS;
benkatz 22:60276ba87ac6 444 controller.mode = 0;
benkatz 23:2adf23ee0305 445 Init_All_HW(&gpio); // Setup PWM, ADC, GPIO
benkatz 20:bf9ea5125d52 446
benkatz 9:d7eb815cb057 447 wait(.1);
benkatz 26:2b865c00d7e9 448 gpio.enable->write(1);
benkatz 45:aadebe074af6 449 TIM1->CCR3 = PWM_ARR*(1.0f); // Write duty cycles
benkatz 45:aadebe074af6 450 TIM1->CCR2 = PWM_ARR*(1.0f);
benkatz 45:aadebe074af6 451 TIM1->CCR1 = PWM_ARR*(1.0f);
benkatz 23:2adf23ee0305 452 zero_current(&controller.adc1_offset, &controller.adc2_offset); // Measure current sensor zero-offset
benkatz 26:2b865c00d7e9 453 gpio.enable->write(0);
benkatz 23:2adf23ee0305 454 reset_foc(&controller); // Reset current controller
benkatz 26:2b865c00d7e9 455 TIM1->CR1 ^= TIM_CR1_UDIS;
benkatz 26:2b865c00d7e9 456 //TIM1->CR1 |= TIM_CR1_UDIS; //enable interrupt
benkatz 20:bf9ea5125d52 457
benkatz 20:bf9ea5125d52 458 wait(.1);
benkatz 37:c0f352d6e8e3 459 NVIC_SetPriority(TIM1_UP_TIM10_IRQn, 2); // commutation > communication
benkatz 43:dfb72608639c 460
benkatz 37:c0f352d6e8e3 461 NVIC_SetPriority(CAN1_RX0_IRQn, 3);
benkatz 26:2b865c00d7e9 462 can.filter(CAN_ID<<21, 0xFFE00004, CANStandard, 0);
benkatz 43:dfb72608639c 463
benkatz 28:8c7e29f719c5 464 txMsg.id = CAN_MASTER;
benkatz 28:8c7e29f719c5 465 txMsg.len = 6;
benkatz 26:2b865c00d7e9 466 rxMsg.len = 8;
benkatz 43:dfb72608639c 467 can.attach(&onMsgReceived); // attach 'CAN receive-complete' interrupt handler
benkatz 23:2adf23ee0305 468
benkatz 25:f5741040c4bb 469 prefs.load(); // Read flash
benkatz 37:c0f352d6e8e3 470 if(isnan(E_OFFSET)){E_OFFSET = 0.0f;}
benkatz 37:c0f352d6e8e3 471 if(isnan(M_OFFSET)){M_OFFSET = 0.0f;}
benkatz 25:f5741040c4bb 472 spi.SetElecOffset(E_OFFSET); // Set position sensor offset
benkatz 37:c0f352d6e8e3 473 spi.SetMechOffset(M_OFFSET);
benkatz 23:2adf23ee0305 474 int lut[128] = {0};
benkatz 23:2adf23ee0305 475 memcpy(&lut, &ENCODER_LUT, sizeof(lut));
benkatz 25:f5741040c4bb 476 spi.WriteLUT(lut); // Set potision sensor nonlinearity lookup table
benkatz 23:2adf23ee0305 477
Rushu 47:55bdc4d5096b 478 pc.baud(115200);//pc.baud(921600); // set serial baud rate
benkatz 20:bf9ea5125d52 479 wait(.01);
benkatz 23:2adf23ee0305 480 pc.printf("\n\r\n\r HobbyKing Cheetah\n\r\n\r");
benkatz 20:bf9ea5125d52 481 wait(.01);
benkatz 23:2adf23ee0305 482 printf("\n\r Debug Info:\n\r");
benkatz 32:ccac5da77844 483 printf(" Firmware Version: %s\n\r", VERSION_NUM);
benkatz 23:2adf23ee0305 484 printf(" ADC1 Offset: %d ADC2 Offset: %d\n\r", controller.adc1_offset, controller.adc2_offset);
benkatz 23:2adf23ee0305 485 printf(" Position Sensor Electrical Offset: %.4f\n\r", E_OFFSET);
benkatz 37:c0f352d6e8e3 486 printf(" Output Zero Position: %.4f\n\r", M_OFFSET);
benkatz 24:58c2d7571207 487 printf(" CAN ID: %d\n\r", CAN_ID);
benkatz 23:2adf23ee0305 488
benkatz 23:2adf23ee0305 489 pc.attach(&serial_interrupt); // attach serial interrupt
benkatz 22:60276ba87ac6 490
benkatz 23:2adf23ee0305 491 state_change = 1;
benkatz 20:bf9ea5125d52 492
benkatz 22:60276ba87ac6 493
benkatz 0:4e1c4df6aabd 494 while(1) {
yezhong 49:8dcdb8a89d0e 495 if(state == MOTOR_MODE)
yezhong 49:8dcdb8a89d0e 496 {
yezhong 49:8dcdb8a89d0e 497 // if(count >= 400){
yezhong 49:8dcdb8a89d0e 498 // printf("J: %.3f Mec: %.3f Jerr: %.3f JVerr: %.3f Kp: %.3f Kd: %.3f \n\r",controller.theta_joint*57.2957795, controller.theta_mech*57.2957795, (controller.p_des - controller.theta_mech)*57.2957795,(controller.v_des - controller.dtheta_mech)*57.2957795, controller.kp,controller.kd);
yezhong 49:8dcdb8a89d0e 499 //printf("Jraw:%.3f J: %.3f Mec: %.3f N: %.3d Nmod: %.3f Mmod: %.3f \n\r",controller.theta_joint_raw*57.2957795,controller.theta_joint*57.2957795, controller.theta_mech*57.2957795, controller.Ncycle,controller.Ncycle_mod*57.2957795,controller.Mech_mod*57.2957795);
benkatz 11:c83b18d41e54 500
yezhong 49:8dcdb8a89d0e 501 //printf("Pdes: %.3f Vdes: %.3f Kp: %.3f Kd: %.3f Tff: %.3f\n\r",controller.p_des*57.2957795, controller.v_des*57.2957795, controller.kp,controller.kd,controller.t_ff);
yezhong 49:8dcdb8a89d0e 502 pc.printf("Vdes: %.3f\n\rPrel: %.3f Vrel: %.3f T: %.3f \n\r",controller.v_des,controller.theta_mech, controller.dtheta_mech, controller.i_q_filt*KT_OUT);
yezhong 49:8dcdb8a89d0e 503 //printf("tor: %.3f\n\r",controller.i_q_filt*KT_OUT)
yezhong 49:8dcdb8a89d0e 504 // count = 0;
yezhong 49:8dcdb8a89d0e 505
yezhong 49:8dcdb8a89d0e 506 //}
yezhong 49:8dcdb8a89d0e 507
yezhong 49:8dcdb8a89d0e 508 }
benkatz 0:4e1c4df6aabd 509 }
benkatz 0:4e1c4df6aabd 510 }