|
8 | 8 |
|
9 | 9 | #include "pico/stdlib.h" |
10 | 10 | #include "hardware/pwm.h" |
| 11 | +#include "pindefs.h" |
11 | 12 |
|
12 | | -#define PWMA_GPIO 16 |
13 | | -#define PWMB_GPIO 17 |
| 13 | +static void setupBtn(uint pin) { |
| 14 | + gpio_init(pin); |
| 15 | + gpio_set_dir(pin, GPIO_IN); |
| 16 | + gpio_pull_up(pin); |
| 17 | +} |
| 18 | + |
| 19 | +static void setupLed(uint pin) { |
| 20 | + gpio_init(pin); |
| 21 | + gpio_set_drive_strength(pin, GPIO_DRIVE_STRENGTH_4MA); |
| 22 | + gpio_set_dir(pin, GPIO_OUT); |
| 23 | +} |
14 | 24 |
|
15 | 25 | int main() { |
| 26 | + setupLed(LED_MODE_1_GPIO); |
| 27 | + setupLed(LED_MODE_2_GPIO); |
| 28 | + setupLed(LED_MODE_3_GPIO); |
| 29 | + setupLed(LED_MODE_4_GPIO); |
16 | 30 |
|
17 | 31 | // Tell GPIO 0 and 1 they are allocated to the PWM |
18 | | - gpio_set_function(PWMA_GPIO, GPIO_FUNC_PWM); |
19 | | - gpio_set_function(PWMB_GPIO, GPIO_FUNC_PWM); |
| 32 | + gpio_set_function(PWM_MODE_GPIO, GPIO_FUNC_PWM); |
| 33 | + gpio_set_function(PWM_STEER_GPIO, GPIO_FUNC_PWM); |
| 34 | + |
| 35 | + setupBtn(BTN_MODE1_GPIO); |
| 36 | + setupBtn(BTN_MODE2_GPIO); |
| 37 | + setupBtn(BTN_MODE3_GPIO); |
| 38 | + setupBtn(BTN_MODE4_GPIO); |
| 39 | + |
| 40 | + setupBtn(BTN_L_GPIO); |
| 41 | + setupBtn(BTN_L_FINE_GPIO); |
| 42 | + setupBtn(BTN_R_GPIO); |
| 43 | + setupBtn(BTN_R_FINE_GPIO); |
| 44 | + |
20 | 45 |
|
21 | 46 | // Find out which PWM slice is connected to GPIO 0 (it's slice 0) |
22 | | - uint slice_num = pwm_gpio_to_slice_num(PWMA_GPIO); |
| 47 | + uint slice_num = pwm_gpio_to_slice_num(PWM_MODE_GPIO); |
23 | 48 |
|
24 | | - // Set period of 4 cycles (0 to 3 inclusive) |
| 49 | + // Set PWM freq 100 Hz |
25 | 50 | pwm_set_clkdiv(slice_num, 125); |
26 | 51 | pwm_set_wrap(slice_num, 10000); |
27 | | - // Set channel A output high for one cycle before dropping |
28 | | - pwm_set_chan_level(slice_num, PWM_CHAN_A, 1500); |
29 | | - // Set initial B output high for three cycles before dropping |
30 | | - pwm_set_chan_level(slice_num, PWM_CHAN_B, 2000); |
| 52 | + |
31 | 53 | // Set the PWM running |
32 | 54 | pwm_set_enabled(slice_num, true); |
33 | | - |
34 | | - // Note we could also use pwm_set_gpio_level(gpio, x) which looks up the |
35 | | - // correct slice and channel for a given GPIO. |
36 | | - while(1) { |
37 | | - sleep_ms(700); |
38 | | - pwm_set_chan_level(slice_num, PWM_CHAN_A, 1200); |
39 | | - sleep_ms(700); |
40 | | - pwm_set_chan_level(slice_num, PWM_CHAN_A, 1500); |
| 55 | + uint mode = 0; |
| 56 | + uint steerPwm = 0; |
| 57 | + pwm_set_chan_level(slice_num, PWM_MODE_CHAN, 0); |
| 58 | + while (1) { |
| 59 | + uint newMode = 0; |
| 60 | + if (gpio_get(BTN_MODE4_GPIO) == 0) { |
| 61 | + newMode = 4; |
| 62 | + } |
| 63 | + if (gpio_get(BTN_MODE3_GPIO) == 0) { |
| 64 | + newMode = 3; |
| 65 | + } else if (gpio_get(BTN_MODE2_GPIO) == 0) { |
| 66 | + newMode = 2; |
| 67 | + } else if (gpio_get(BTN_MODE1_GPIO) == 0) { |
| 68 | + newMode = 1; |
| 69 | + } |
| 70 | + if (newMode == 0 && mode == 0) { |
| 71 | + newMode = 1; |
| 72 | + } |
| 73 | + if (newMode != 0 && newMode != mode) { |
| 74 | + mode = newMode; |
| 75 | + uint pwm = 0; |
| 76 | + switch (mode) { |
| 77 | + case 1: |
| 78 | + gpio_put(LED_MODE_1_GPIO, true); |
| 79 | + gpio_put(LED_MODE_2_GPIO, false); |
| 80 | + gpio_put(LED_MODE_3_GPIO, false); |
| 81 | + gpio_put(LED_MODE_4_GPIO, false); |
| 82 | + pwm_set_chan_level(slice_num, PWM_MODE_CHAN, 0); |
| 83 | + break; |
| 84 | + case 2: |
| 85 | + gpio_put(LED_MODE_1_GPIO, true); |
| 86 | + gpio_put(LED_MODE_2_GPIO, true); |
| 87 | + gpio_put(LED_MODE_3_GPIO, false); |
| 88 | + gpio_put(LED_MODE_4_GPIO, false); |
| 89 | + pwm_set_chan_level(slice_num, PWM_MODE_CHAN, 1300); |
| 90 | + break; |
| 91 | + case 3: |
| 92 | + gpio_put(LED_MODE_1_GPIO, true); |
| 93 | + gpio_put(LED_MODE_2_GPIO, true); |
| 94 | + gpio_put(LED_MODE_3_GPIO, true); |
| 95 | + gpio_put(LED_MODE_4_GPIO, false); |
| 96 | + pwm_set_chan_level(slice_num, PWM_MODE_CHAN, 1500); |
| 97 | + break; |
| 98 | + case 4: |
| 99 | + gpio_put(LED_MODE_1_GPIO, true); |
| 100 | + gpio_put(LED_MODE_2_GPIO, true); |
| 101 | + gpio_put(LED_MODE_3_GPIO, true); |
| 102 | + gpio_put(LED_MODE_4_GPIO, true); |
| 103 | + pwm_set_chan_level(slice_num, PWM_MODE_CHAN, 1700); |
| 104 | + break; |
| 105 | + default: |
| 106 | + gpio_put(LED_MODE_1_GPIO, false); |
| 107 | + gpio_put(LED_MODE_2_GPIO, false); |
| 108 | + gpio_put(LED_MODE_3_GPIO, false); |
| 109 | + gpio_put(LED_MODE_4_GPIO, false); |
| 110 | + pwm_set_chan_level(slice_num, PWM_MODE_CHAN, 0); |
| 111 | + break; |
| 112 | + } |
| 113 | + } |
| 114 | + uint newSteer = 0; |
| 115 | + if (gpio_get(BTN_L_GPIO) == 0) { |
| 116 | + newSteer = 1300; |
| 117 | + } else if (gpio_get(BTN_L_FINE_GPIO) == 0) { |
| 118 | + newSteer = 1450; |
| 119 | + } else if (gpio_get(BTN_R_FINE_GPIO) == 0) { |
| 120 | + newSteer = 1550; |
| 121 | + } else if (gpio_get(BTN_R_GPIO) == 0) { |
| 122 | + newSteer = 1700; |
| 123 | + } else { |
| 124 | + newSteer = 1500; |
| 125 | + } |
| 126 | + if (newSteer != steerPwm) { |
| 127 | + steerPwm = newSteer; |
| 128 | + pwm_set_chan_level(slice_num, PWM_STEER_CHAN, steerPwm); |
| 129 | + } |
41 | 130 | } |
| 131 | + |
42 | 132 | } |
0 commit comments