Michael Sprich
/
PM2_Example_PES_board
first publish Workshop2
main.cpp@31:20bd07533f96, 2022-03-23 (annotated)
- Committer:
- raomen
- Date:
- Wed Mar 23 07:08:36 2022 +0000
- Revision:
- 31:20bd07533f96
- Parent:
- 30:1e8295770bc1
first commit;
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
raomen | 31:20bd07533f96 | 1 | // Workshope 2 |
pmic | 1:93d997d6b232 | 2 | #include "mbed.h" |
pmic | 17:c19b471f05cb | 3 | #include "PM2_Libary.h" |
pmic | 6:e1fa1a2d7483 | 4 | |
pmic | 24:86f1a63e35a0 | 5 | // logical variable main task |
pmic | 24:86f1a63e35a0 | 6 | bool do_execute_main_task = false; // this variable will be toggled via the user button (blue button) to or not to execute the main task |
pmic | 17:c19b471f05cb | 7 | |
pmic | 24:86f1a63e35a0 | 8 | // user button on nucleo board |
pmic | 24:86f1a63e35a0 | 9 | Timer user_button_timer; // create Timer object which we use to check if user button was pressed for a certain time (robust against signal bouncing) |
pmic | 24:86f1a63e35a0 | 10 | InterruptIn user_button(PC_13); // create InterruptIn interface object to evaluate user button falling and rising edge (no blocking code in ISR) |
pmic | 24:86f1a63e35a0 | 11 | void user_button_pressed_fcn(); // custom functions which gets executed when user button gets pressed and released, definition below |
pmic | 24:86f1a63e35a0 | 12 | void user_button_released_fcn(); |
pmic | 6:e1fa1a2d7483 | 13 | |
pmic | 24:86f1a63e35a0 | 14 | // while loop gets executed every main_task_period_ms milliseconds |
pmic | 24:86f1a63e35a0 | 15 | int main_task_period_ms = 50; // define main task period time in ms e.g. 50 ms -> main task runns 20 times per second |
pmic | 24:86f1a63e35a0 | 16 | Timer main_task_timer; // create Timer object which we use to run the main task every main task period time in ms |
pmic | 6:e1fa1a2d7483 | 17 | |
pmic | 24:86f1a63e35a0 | 18 | // led on nucleo board |
pmic | 24:86f1a63e35a0 | 19 | DigitalOut user_led(LED1); // create DigitalOut object to command user led |
pmic | 17:c19b471f05cb | 20 | |
pmic | 24:86f1a63e35a0 | 21 | // additional Led |
pmic | 24:86f1a63e35a0 | 22 | DigitalOut extra_led(PB_9); // create DigitalOut object to command extra led (do add an aditional resistor, e.g. 220...500 Ohm) |
pmic | 17:c19b471f05cb | 23 | |
pmic | 24:86f1a63e35a0 | 24 | // mechanical button |
pmic | 24:86f1a63e35a0 | 25 | DigitalIn mechanical_button(PC_5); // create DigitalIn object to evaluate extra mechanical button, you need to specify the mode for proper usage, see below |
pmic | 24:86f1a63e35a0 | 26 | |
pmic | 24:86f1a63e35a0 | 27 | // Sharp GP2Y0A41SK0F, 4-40 cm IR Sensor |
pmic | 24:86f1a63e35a0 | 28 | float ir_distance_mV = 0.0f; // define variable to store measurement |
pmic | 24:86f1a63e35a0 | 29 | AnalogIn ir_analog_in(PC_2); // create AnalogIn object to read in infrared distance sensor, 0...3.3V are mapped to 0...1 |
pmic | 6:e1fa1a2d7483 | 30 | |
pmic | 24:86f1a63e35a0 | 31 | // 78:1, 100:1, ... Metal Gearmotor 20Dx44L mm 12V CB |
pmic | 24:86f1a63e35a0 | 32 | DigitalOut enable_motors(PB_15); // create DigitalOut object to enable dc motors |
pmic | 17:c19b471f05cb | 33 | |
pmic | 24:86f1a63e35a0 | 34 | float pwm_period_s = 0.00005f; // define pwm period time in seconds and create FastPWM objects to command dc motors |
pmic | 24:86f1a63e35a0 | 35 | FastPWM pwm_M1(PB_13); // motor M1 is used open loop |
pmic | 30:1e8295770bc1 | 36 | FastPWM pwm_M2(PA_9); // motor M2 is closed-loop speed controlled (angle velocity) |
pmic | 30:1e8295770bc1 | 37 | FastPWM pwm_M3(PA_10); // motor M3 is closed-loop position controlled (angle controlled) |
pmic | 17:c19b471f05cb | 38 | |
pmic | 30:1e8295770bc1 | 39 | EncoderCounter encoder_M1(PA_6, PC_7); // create encoder objects to read in the encoder counter values |
pmic | 17:c19b471f05cb | 40 | EncoderCounter encoder_M2(PB_6, PB_7); |
pmic | 17:c19b471f05cb | 41 | EncoderCounter encoder_M3(PA_0, PA_1); |
pmic | 17:c19b471f05cb | 42 | |
pmic | 30:1e8295770bc1 | 43 | // create SpeedController and PositionController objects, default parametrization is for 78.125:1 gear box |
pmic | 24:86f1a63e35a0 | 44 | float max_voltage = 12.0f; // define maximum voltage of battery packs, adjust this to 6.0f V if you only use one batterypack |
pmic | 24:86f1a63e35a0 | 45 | float counts_per_turn = 20.0f * 78.125f; // define counts per turn at gearbox end: counts/turn * gearratio |
pmic | 25:ea1d6e27c895 | 46 | float kn = 180.0f / 12.0f; // define motor constant in rpm per V |
pmic | 30:1e8295770bc1 | 47 | float k_gear = 100.0f / 78.125f; // define additional ratio in case you are using a dc motor with a different gear box, e.g. 100:1 |
pmic | 30:1e8295770bc1 | 48 | float kp = 0.1f; // define custom kp, this is the default speed controller gain for gear box 78.125:1 |
pmic | 6:e1fa1a2d7483 | 49 | |
pmic | 30:1e8295770bc1 | 50 | // SpeedController speedController_M2(counts_per_turn, kn, max_voltage, pwm_M2, encoder_M2); // default 78.125:1 gear box with default contoller parameters |
pmic | 30:1e8295770bc1 | 51 | SpeedController speedController_M2(counts_per_turn * k_gear, kn / k_gear, max_voltage, pwm_M2, encoder_M2); // parameters adjusted to 100:1 gear |
pmic | 17:c19b471f05cb | 52 | |
pmic | 24:86f1a63e35a0 | 53 | float max_speed_rps = 0.5f; // define maximum speed that the position controller is changig the speed, has to be smaller or equal to kn * max_voltage |
pmic | 30:1e8295770bc1 | 54 | // PositionController positionController_M3(counts_per_turn, kn, max_voltage, pwm_M3, encoder_M3); // default 78.125:1 gear with default contoller parameters |
pmic | 30:1e8295770bc1 | 55 | PositionController positionController_M3(counts_per_turn * k_gear, kn / k_gear, kp * k_gear, max_voltage, pwm_M3, encoder_M3); // parameters adjusted to 100:1 gear, we need a different speed controller gain here |
pmic | 17:c19b471f05cb | 56 | |
pmic | 17:c19b471f05cb | 57 | // Futaba Servo S3001 20mm 3kg Analog |
pmic | 24:86f1a63e35a0 | 58 | Servo servo_S1(PB_2); // create servo objects |
pmic | 10:c5d85e35758c | 59 | Servo servo_S2(PC_8); |
pmic | 30:1e8295770bc1 | 60 | float servo_S1_angle = 0; // servo S1 normalized angle |
pmic | 30:1e8295770bc1 | 61 | float servo_S2_angle = 0; // servo S2 normalized angle |
pmic | 24:86f1a63e35a0 | 62 | int servo_period_mus = 20000; // define servo period time in mus |
pmic | 17:c19b471f05cb | 63 | |
pmic | 24:86f1a63e35a0 | 64 | int servo_counter = 0; // define servo counter, this is an additional variable to make the servos move |
pmic | 30:1e8295770bc1 | 65 | int loops_per_seconds = static_cast<int>(ceilf(1.0f/(0.001f*(float)main_task_period_ms))); // define loops per second |
pmic | 1:93d997d6b232 | 66 | |
pmic | 11:af0f165f8761 | 67 | // Groove Ultrasonic Ranger V2.0 |
pmic | 24:86f1a63e35a0 | 68 | float us_distance_cm = 0.0f; // define variable to store measurement |
pmic | 30:1e8295770bc1 | 69 | RangeFinder us_range_finder(PB_12, 5782.0f, 0.02f, 17500); // create range finder object (ultra sonic distance sensor), 20 Hz parametrization |
pmic | 30:1e8295770bc1 | 70 | // RangeFinder us_range_finder(PB_12, 5782.0f, 0.02f, 7000); // create range finder object (ultra sonic distance sensor), 50 Hz parametrization |
pmic | 17:c19b471f05cb | 71 | |
pmic | 24:86f1a63e35a0 | 72 | // LSM9DS1 IMU, carefull: not all PES boards have an imu (chip shortage) |
pmic | 25:ea1d6e27c895 | 73 | // LSM9DS1 imu(PC_9, PA_8); // create LSM9DS1 comunication object, if you want to be able to use the imu you need to #include "LSM9DS1_i2c.h" |
pmic | 20:7e7325edcf5c | 74 | |
pmic | 1:93d997d6b232 | 75 | int main() |
pmic | 23:26b3a25fc637 | 76 | { |
pmic | 24:86f1a63e35a0 | 77 | // attach button fall and rise functions to user button object |
pmic | 24:86f1a63e35a0 | 78 | user_button.fall(&user_button_pressed_fcn); |
pmic | 24:86f1a63e35a0 | 79 | user_button.rise(&user_button_released_fcn); |
pmic | 17:c19b471f05cb | 80 | |
pmic | 29:d6f1ccf42a31 | 81 | // start timer |
pmic | 24:86f1a63e35a0 | 82 | main_task_timer.start(); |
pmic | 6:e1fa1a2d7483 | 83 | |
pmic | 24:86f1a63e35a0 | 84 | // set pullup mode: add resistor between pin and 3.3 V, so that there is a defined potential |
pmic | 24:86f1a63e35a0 | 85 | mechanical_button.mode(PullUp); |
pmic | 24:86f1a63e35a0 | 86 | |
pmic | 24:86f1a63e35a0 | 87 | // enable hardwaredriver dc motors: 0 -> disabled, 1 -> enabled |
pmic | 10:c5d85e35758c | 88 | enable_motors = 1; |
pmic | 17:c19b471f05cb | 89 | |
pmic | 24:86f1a63e35a0 | 90 | // motor M1 is used open-loop, we need to initialize the pwm and set pwm output to zero at the beginning, range: 0...1 -> u_min...u_max: 0.5 -> 0 V |
pmic | 25:ea1d6e27c895 | 91 | pwm_M1.period(pwm_period_s); |
pmic | 30:1e8295770bc1 | 92 | pwm_M1.write(0.5f); |
pmic | 9:f10b974d01e0 | 93 | |
pmic | 24:86f1a63e35a0 | 94 | // enable servos, you can also disable them at any point in your program if you don't want your servos to become warm |
pmic | 30:1e8295770bc1 | 95 | servo_S1.Enable(servo_S1_angle, servo_period_mus); |
pmic | 30:1e8295770bc1 | 96 | servo_S2.Enable(servo_S2_angle, servo_period_mus); |
pmic | 6:e1fa1a2d7483 | 97 | |
pmic | 24:86f1a63e35a0 | 98 | while (true) { // this loop will run forever |
pmic | 6:e1fa1a2d7483 | 99 | |
pmic | 24:86f1a63e35a0 | 100 | main_task_timer.reset(); |
pmic | 6:e1fa1a2d7483 | 101 | |
pmic | 24:86f1a63e35a0 | 102 | if (do_execute_main_task) { |
pmic | 17:c19b471f05cb | 103 | |
pmic | 24:86f1a63e35a0 | 104 | // read analog input |
pmic | 24:86f1a63e35a0 | 105 | ir_distance_mV = 1.0e3f * ir_analog_in.read() * 3.3f; |
pmic | 6:e1fa1a2d7483 | 106 | |
pmic | 24:86f1a63e35a0 | 107 | // command dc motors if mechanical button is pressed |
pmic | 29:d6f1ccf42a31 | 108 | if (mechanical_button.read()) { |
pmic | 30:1e8295770bc1 | 109 | pwm_M1.write(0.75f); // write output voltage to motor M1 |
pmic | 24:86f1a63e35a0 | 110 | speedController_M2.setDesiredSpeedRPS(0.5f); // set a desired speed for speed controlled dc motors M2 |
pmic | 24:86f1a63e35a0 | 111 | positionController_M3.setDesiredRotation(1.5f, max_speed_rps); // set a desired rotation for position controlled dc motors M3 |
pmic | 24:86f1a63e35a0 | 112 | } else { |
pmic | 30:1e8295770bc1 | 113 | pwm_M1.write(0.5f); |
pmic | 24:86f1a63e35a0 | 114 | speedController_M2.setDesiredSpeedRPS(0.0f); |
pmic | 24:86f1a63e35a0 | 115 | positionController_M3.setDesiredRotation(0.0f, max_speed_rps); |
pmic | 24:86f1a63e35a0 | 116 | } |
pmic | 6:e1fa1a2d7483 | 117 | |
pmic | 30:1e8295770bc1 | 118 | // check if servos are enabled |
pmic | 30:1e8295770bc1 | 119 | if (!servo_S1.isEnabled()) servo_S1.Enable(servo_S1_angle, servo_period_mus); |
pmic | 30:1e8295770bc1 | 120 | if (!servo_S2.isEnabled()) servo_S2.Enable(servo_S2_angle, servo_period_mus); |
pmic | 24:86f1a63e35a0 | 121 | // command servo position, this needs to be calibrated |
pmic | 30:1e8295770bc1 | 122 | servo_S1.SetPosition(servo_S1_angle); |
pmic | 30:1e8295770bc1 | 123 | if (servo_S1_angle <= 1.0f & servo_counter%loops_per_seconds == 0 & servo_counter != 0) { |
pmic | 30:1e8295770bc1 | 124 | servo_S1_angle += 0.01f; |
pmic | 8:9bb806a7f585 | 125 | } |
pmic | 30:1e8295770bc1 | 126 | servo_S2.SetPosition(servo_S2_angle); |
pmic | 30:1e8295770bc1 | 127 | if (servo_S2_angle <= 1.0f & servo_counter%loops_per_seconds == 0 & servo_counter != 0) { |
pmic | 30:1e8295770bc1 | 128 | servo_S2_angle += 0.01f; |
pmic | 8:9bb806a7f585 | 129 | } |
pmic | 10:c5d85e35758c | 130 | servo_counter++; |
pmic | 6:e1fa1a2d7483 | 131 | |
pmic | 24:86f1a63e35a0 | 132 | // read ultra sonic distance sensor |
pmic | 24:86f1a63e35a0 | 133 | us_distance_cm = us_range_finder.read_cm(); |
pmic | 11:af0f165f8761 | 134 | |
pmic | 24:86f1a63e35a0 | 135 | // visual feedback that the main task is executed |
pmic | 24:86f1a63e35a0 | 136 | extra_led = 1; |
pmic | 9:f10b974d01e0 | 137 | |
pmic | 1:93d997d6b232 | 138 | } else { |
pmic | 6:e1fa1a2d7483 | 139 | |
pmic | 24:86f1a63e35a0 | 140 | ir_distance_mV = 0.0f; |
pmic | 1:93d997d6b232 | 141 | |
pmic | 30:1e8295770bc1 | 142 | pwm_M1.write(0.5f); |
pmic | 17:c19b471f05cb | 143 | speedController_M2.setDesiredSpeedRPS(0.0f); |
pmic | 17:c19b471f05cb | 144 | positionController_M3.setDesiredRotation(0.0f, max_speed_rps); |
pmic | 6:e1fa1a2d7483 | 145 | |
pmic | 30:1e8295770bc1 | 146 | servo_S1_angle = 0; |
pmic | 30:1e8295770bc1 | 147 | servo_S2_angle = 0; |
pmic | 30:1e8295770bc1 | 148 | // servo_S1.SetPosition(servo_S1_angle); |
pmic | 30:1e8295770bc1 | 149 | // servo_S2.SetPosition(servo_S2_angle); |
pmic | 30:1e8295770bc1 | 150 | if (servo_S1.isEnabled()) servo_S1.Disable(); |
pmic | 30:1e8295770bc1 | 151 | if (servo_S2.isEnabled()) servo_S2.Disable(); |
pmic | 17:c19b471f05cb | 152 | |
pmic | 24:86f1a63e35a0 | 153 | us_distance_cm = 0.0f; |
pmic | 6:e1fa1a2d7483 | 154 | |
pmic | 24:86f1a63e35a0 | 155 | extra_led = 0; |
pmic | 1:93d997d6b232 | 156 | } |
pmic | 6:e1fa1a2d7483 | 157 | |
pmic | 24:86f1a63e35a0 | 158 | user_led = !user_led; |
pmic | 24:86f1a63e35a0 | 159 | |
pmic | 24:86f1a63e35a0 | 160 | // do only output via serial what's really necessary (this makes your code slow) |
pmic | 30:1e8295770bc1 | 161 | printf("IR sensor (mV): %3.3f, Encoder M1: %3d, Speed M2 (rps) %3.3f, Position M3 (rot): %3.3f, Servo S1 angle (normalized): %3.3f, Servo S2 angle (normalized): %3.3f, US sensor (cm): %3.3f\r\n", |
pmic | 24:86f1a63e35a0 | 162 | ir_distance_mV, |
pmic | 17:c19b471f05cb | 163 | encoder_M1.read(), |
pmic | 17:c19b471f05cb | 164 | speedController_M2.getSpeedRPS(), |
pmic | 17:c19b471f05cb | 165 | positionController_M3.getRotation(), |
pmic | 30:1e8295770bc1 | 166 | servo_S1_angle, |
pmic | 30:1e8295770bc1 | 167 | servo_S2_angle, |
pmic | 24:86f1a63e35a0 | 168 | us_distance_cm); |
pmic | 6:e1fa1a2d7483 | 169 | |
pmic | 24:86f1a63e35a0 | 170 | // read out the imu, the actual frames of the sensor reading needs to be figured out |
pmic | 24:86f1a63e35a0 | 171 | // imu.updateGyro(); |
pmic | 24:86f1a63e35a0 | 172 | // imu.updateAcc(); |
pmic | 24:86f1a63e35a0 | 173 | // imu.updateMag(); |
pmic | 24:86f1a63e35a0 | 174 | // printf("%.6f, %.6f, %.6f, %.6f, %.6f, %.6f, %.6f, %.6f, %.6f\r\n", imu.readGyroX(), imu.readGyroY(), imu.readGyroZ(), |
pmic | 24:86f1a63e35a0 | 175 | // imu.readAccX(), imu.readAccY(), imu.readAccZ(), imu.readMagX(), imu.readMagY(), imu.readMagZ()); |
pmic | 17:c19b471f05cb | 176 | |
pmic | 24:86f1a63e35a0 | 177 | // read timer and make the main thread sleep for the remaining time span (non blocking) |
pmic | 24:86f1a63e35a0 | 178 | int main_task_elapsed_time_ms = std::chrono::duration_cast<std::chrono::milliseconds>(main_task_timer.elapsed_time()).count(); |
pmic | 24:86f1a63e35a0 | 179 | thread_sleep_for(main_task_period_ms - main_task_elapsed_time_ms); |
pmic | 1:93d997d6b232 | 180 | } |
pmic | 1:93d997d6b232 | 181 | } |
pmic | 6:e1fa1a2d7483 | 182 | |
pmic | 24:86f1a63e35a0 | 183 | void user_button_pressed_fcn() |
pmic | 25:ea1d6e27c895 | 184 | { |
pmic | 26:28693b369945 | 185 | user_button_timer.start(); |
pmic | 6:e1fa1a2d7483 | 186 | user_button_timer.reset(); |
pmic | 6:e1fa1a2d7483 | 187 | } |
pmic | 6:e1fa1a2d7483 | 188 | |
pmic | 24:86f1a63e35a0 | 189 | void user_button_released_fcn() |
pmic | 6:e1fa1a2d7483 | 190 | { |
pmic | 24:86f1a63e35a0 | 191 | // read timer and toggle do_execute_main_task if the button was pressed longer than the below specified time |
pmic | 24:86f1a63e35a0 | 192 | int user_button_elapsed_time_ms = std::chrono::duration_cast<std::chrono::milliseconds>(user_button_timer.elapsed_time()).count(); |
pmic | 6:e1fa1a2d7483 | 193 | user_button_timer.stop(); |
pmic | 24:86f1a63e35a0 | 194 | if (user_button_elapsed_time_ms > 200) { |
pmic | 24:86f1a63e35a0 | 195 | do_execute_main_task = !do_execute_main_task; |
pmic | 8:9bb806a7f585 | 196 | } |
pmic | 6:e1fa1a2d7483 | 197 | } |