]>
Commit | Line | Data |
---|---|---|
1 | #include "pico/cyw43_arch.h" | |
2 | #include "pico/stdlib.h" | |
3 | #include "pico/util/pheap.h" | |
4 | #include "pico/util/queue.h" | |
5 | ||
6 | #include "blink.h" | |
7 | #include "button.h" | |
8 | #include "config.h" | |
9 | #include "net.h" | |
10 | #include "press.h" | |
11 | ||
12 | enum event_type { NEW_BUTTON_PRESS, RESEND_TIME }; | |
13 | typedef struct { | |
14 | enum event_type type; | |
15 | union { | |
16 | struct { | |
17 | uint32_t timestamp; | |
18 | } buttonpress; | |
19 | }; | |
20 | } event_t; | |
21 | ||
22 | queue_t queue; | |
23 | ||
24 | uint32_t time_s() { return time_us_64() / 1000000ul; } | |
25 | ||
26 | /* Often we don't bother checking for failure (full queue) because | |
27 | * 1. The best thing to do in this unfortunate situation is to blithely | |
28 | * continue, dropping some events; continuing is better than stopping. | |
29 | * 2. Neither interrupt context nor queue-processing context can block | |
30 | * until space is available, or even sit around & blink the LED to | |
31 | * signal a problem. | |
32 | * (We also get a bit of type safety by taking event_t* rather than void*.) */ | |
33 | static void queue_try_add_ignoring_errors(queue_t *q, event_t *e) { | |
34 | queue_try_add(q, e); | |
35 | } | |
36 | ||
37 | static void button_pressed() { | |
38 | /* This runs in interrupt context; don't linger. */ | |
39 | static uint64_t last_button_press_time = 0; | |
40 | uint32_t now = time_s(); | |
41 | uint32_t time_since_last_press = now - last_button_press_time; | |
42 | if (time_since_last_press >= config_minimum_seconds_between_button_presses) { | |
43 | last_button_press_time = now; | |
44 | event_t e; | |
45 | e.type = NEW_BUTTON_PRESS; | |
46 | e.buttonpress.timestamp = now; | |
47 | queue_try_add_ignoring_errors(&queue, &e); | |
48 | } | |
49 | } | |
50 | ||
51 | static void time_to_send(uint _) { | |
52 | /* This runs in interrupt context; don't linger. */ | |
53 | event_t e; | |
54 | e.type = RESEND_TIME; | |
55 | queue_try_add_ignoring_errors(&queue, &e); | |
56 | } | |
57 | ||
58 | void set_resend_alarm(int alarm, uint32_t now, uint32_t next_act_time) { | |
59 | uint32_t sleep_duration = next_act_time - now; | |
60 | if (hardware_alarm_set_target(alarm, | |
61 | make_timeout_time_ms(sleep_duration * 1000))) | |
62 | signal_error_by_blinking(); | |
63 | } | |
64 | ||
65 | void service_sleeps(int alarm, press_pile_t *pp) { | |
66 | hardware_alarm_cancel(alarm); | |
67 | ||
68 | while (1) { | |
69 | uint32_t now = time_s(); | |
70 | int32_t act_time = next_scheduled_send(pp); | |
71 | if (act_time == -1) | |
72 | return; | |
73 | if (act_time > now) { | |
74 | set_resend_alarm(alarm, now, act_time); | |
75 | return; | |
76 | } | |
77 | press_t press; | |
78 | if (!get_press_due_for_resend(pp, now, &press)) | |
79 | signal_error_by_blinking(); | |
80 | uint32_t ago = now - press.timestamp; | |
81 | send_report_packet(press.seq, ago); | |
82 | press.send_count++; | |
83 | if (press.send_count < config_resend_count) { | |
84 | add_press(pp, &press); | |
85 | } | |
86 | } | |
87 | } | |
88 | ||
89 | void service_queue() { | |
90 | int alarm = hardware_alarm_claim_unused(true); | |
91 | if (alarm == -1) | |
92 | signal_error_by_blinking(); | |
93 | hardware_alarm_set_callback(alarm, time_to_send); | |
94 | ||
95 | press_pile_t *pp = create_press_pile(); | |
96 | u16_t seq = 0; | |
97 | while (1) { | |
98 | service_sleeps(alarm, pp); | |
99 | ||
100 | event_t e; | |
101 | queue_remove_blocking(&queue, &e); | |
102 | switch (e.type) { | |
103 | case NEW_BUTTON_PRESS: { | |
104 | add_press(pp, &(press_t){e.buttonpress.timestamp, seq++, 0}); | |
105 | } break; | |
106 | case RESEND_TIME: { | |
107 | /* OK, we're awake. Cool. Thanks! (We actually do the sends in the | |
108 | * service_sleeps() call at the top of the loop.) */ | |
109 | } break; | |
110 | default: | |
111 | signal_error_by_blinking(); | |
112 | } | |
113 | } | |
114 | } | |
115 | ||
116 | int main() { | |
117 | stdio_init_all(); | |
118 | if (cyw43_arch_init_with_country(CYW43_COUNTRY_USA)) | |
119 | signal_error_by_blinking(); | |
120 | cyw43_arch_enable_sta_mode(); | |
121 | signal(3, 100); | |
122 | if (cyw43_arch_wifi_connect_timeout_ms(config_wifi_ssid, config_wifi_pass, | |
123 | CYW43_AUTH_WPA2_AES_PSK, 90000)) | |
124 | signal_error_by_blinking(); | |
125 | signal(2, 300); | |
126 | ||
127 | queue_init(&queue, sizeof(event_t), 99); | |
128 | ||
129 | begin_listening_for_button_press(button_pressed); | |
130 | ||
131 | service_queue(); | |
132 | } |