Lucas Rosina
/
PM2_Example_PES_board_ros2
Workshop 2
main.cpp
- Committer:
- pmic
- Date:
- 2021-04-06
- Revision:
- 6:e1fa1a2d7483
- Parent:
- 4:67506e285ad0
- Child:
- 7:c0f5bb355f41
File content as of revision 6:e1fa1a2d7483:
#include "mbed.h" #include "platform/mbed_thread.h" #include "string" /* PM2_Libary */ #include "EncoderCounter.h" #include "Servo.h" #include "SpeedController.h" // #include "FATFileSystem.h" // #include "SDBlockDevice.h" using namespace std::chrono; InterruptIn user_button(USER_BUTTON); DigitalOut led(LED1); BufferedSerial bufferedSerial(USBTX, USBRX); char buffer[100]; bool executeMainTask = false; Timer user_button_timer, loop_timer; int Ts_ms = 2; /* declaration of custom button functions */ void button_fall(); void button_rise(); /* create analog input object */ AnalogIn analogIn(PC_2); float dist = 0.0f; /* create pwm objects */ PwmOut pwmOut_m0(PB_13); PwmOut pwmOut_m1(PA_9); PwmOut pwmOut_m2(PA_10); float Ts_pwm = 0.00005f; /* create enable dc motor digital out object */ DigitalOut enable_motors(PB_15); /* create encoder read objects */ EncoderCounter encoderCounter_m0(PA_6, PC_7); EncoderCounter encoderCounter_m1(PB_6, PB_7); EncoderCounter encoderCounter_m2(PA_0, PA_1); /* create speed controller objects, only m0 and m1, m2 is used open-loop */ SpeedController speedController_m0(1562.5f, 15.0f, 0.1f, 12.0f, pwmOut_m0, encoderCounter_m0); SpeedController speedController_m1(1562.5f, 15.0f, 0.1f, 12.0f, pwmOut_m1, encoderCounter_m1); /* create servo objects */ Servo servo_0(PB_2); Servo servo_1(PC_8); Servo servo_2(PC_6); // not used in this example int Ts_servo_mus = 20000; int servo_desval_0 = 0; int servo_desval_1 = 0; /* create sd object */ // SDBlockDevice sd(PC_12, PC_11, PC_10, PD_2); // FATFileSystem fs("fs", &sd); int main() { bufferedSerial.set_baud(115200); user_button.fall(&button_fall); user_button.rise(&button_rise); loop_timer.start(); /* initialize pwm */ pwmOut_m0.period(Ts_pwm); pwmOut_m1.period(Ts_pwm); pwmOut_m2.period(Ts_pwm); /* set pwm output zero at the beginning, range: 0...1 -> u_min...u_max */ pwmOut_m1.write(0.5f); pwmOut_m0.write(0.5f); pwmOut_m2.write(0.5f); /* enable driver DC motors */ enable_motors = 1; /* initialize servo */ servo_0.Enable(servo_desval_0, Ts_servo_mus); // 1 ms / 20 ms servo_1.Enable(servo_desval_0, Ts_servo_mus); /* // example code for sd card, not tested from pmic, 02.04.2021 printf("Test writing... "); FILE* fp = fopen("/fs/data.csv", "w"); fprintf(fp, "test %.5f\r\n",1.23); fclose(fp); printf("done\r\n"); printf("Test reading... "); // read from SD card fp = fopen("/fs/data.csv", "r"); if (fp != NULL) { char c = fgetc(fp); if (c == 't') printf("done\r\n"); else printf("incorrect char (%c)!\n", c); fclose(fp); } else { printf("Reading failed!\n"); } */ while (true) { loop_timer.reset(); /* ------------- start hacking ------------- -------------*/ if (executeMainTask) { /* read analog input */ dist = analogIn.read() * 3.3f; speedController_m0.setDesiredSpeedRPS( 1.0f); speedController_m1.setDesiredSpeedRPS(-2.0f); pwmOut_m2.write(0.75f); servo_0.SetPosition(servo_desval_0); servo_1.SetPosition(servo_desval_1); if (servo_desval_0 < 10000) servo_desval_0 += 100; if (servo_desval_1 < 10000) servo_desval_1 += 100; /* visual feedback that the main task is executed */ led = !led; /* write output via serial buffer */ int act_buffer_length = snprintf (buffer, 100, "%3.6e, %3.6e; \r\n", speedController_m0.getSpeedRPM(), speedController_m1.getSpeedRPM()); bufferedSerial.write(buffer, act_buffer_length); } else { dist = 0.0f; speedController_m0.setDesiredSpeedRPS(0.0f); speedController_m1.setDesiredSpeedRPS(0.0f); pwmOut_m2.write(0.5f); servo_desval_0 = 0; servo_desval_1 = 0; servo_0.SetPosition(servo_desval_0); servo_1.SetPosition(servo_desval_1); dist = analogIn.read() * 3.3f; led = 0; } /* ------------- stop hacking ------------- -------------*/ int T_loop_ms = duration_cast<milliseconds>(loop_timer.elapsed_time()).count(); int dT_loop_ms = Ts_ms - T_loop_ms; thread_sleep_for(dT_loop_ms); } } void button_fall() { user_button_timer.reset(); user_button_timer.start(); } void button_rise() { int t_button_ms = duration_cast<milliseconds>(user_button_timer.elapsed_time()).count(); user_button_timer.stop(); if (t_button_ms > 1) executeMainTask = !executeMainTask; }