altos: Clean up -Wextra warnings
[fw/altos] / src / telelco-v0.2 / ao_lco.c
1 /*
2  * Copyright © 2012 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 #include <ao.h>
19 #include <ao_lco.h>
20 #include <ao_event.h>
21 #include <ao_seven_segment.h>
22 #include <ao_quadrature.h>
23 #include <ao_lco_func.h>
24 #include <ao_radio_cmac.h>
25
26 #define DEBUG   1
27
28 #if DEBUG
29 static uint8_t  ao_lco_debug;
30 #define PRINTD(...) do { if (!ao_lco_debug) break; printf ("\r%5u %s: ", ao_tick_count, __func__); printf(__VA_ARGS__); flush(); } while(0)
31 #else
32 #define PRINTD(...) 
33 #endif
34
35 #define AO_LCO_PAD_DIGIT        0
36 #define AO_LCO_BOX_DIGIT_1      1
37 #define AO_LCO_BOX_DIGIT_10     2
38
39 static uint8_t  ao_lco_min_box, ao_lco_max_box;
40 static uint8_t  ao_lco_pad;
41 static uint8_t  ao_lco_box;
42 static uint8_t  ao_lco_armed;
43 static uint8_t  ao_lco_firing;
44 static uint8_t  ao_lco_valid;
45 static uint8_t  ao_lco_got_channels;
46 static uint16_t ao_lco_tick_offset;
47
48 static struct ao_pad_query      ao_pad_query;
49
50 static void
51 ao_lco_set_pad(uint8_t pad)
52 {
53         ao_seven_segment_set(AO_LCO_PAD_DIGIT, pad + 1);
54 }
55
56 static void
57 ao_lco_set_box(uint8_t box)
58 {
59         ao_seven_segment_set(AO_LCO_BOX_DIGIT_1, box % 10);
60         ao_seven_segment_set(AO_LCO_BOX_DIGIT_10, box / 10);
61 }
62
63 #define MASK_SIZE(n)    (((n) + 7) >> 3)
64 #define MASK_ID(n)      ((n) >> 3)
65 #define MASK_SHIFT(n)   ((n) & 7)
66
67 static uint8_t  ao_lco_box_mask[MASK_SIZE(AO_PAD_MAX_BOXES)];
68
69 static uint8_t
70 ao_lco_box_present(uint8_t box)
71 {
72         if (box >= AO_PAD_MAX_BOXES)
73                 return 0;
74         return (ao_lco_box_mask[MASK_ID(box)] >> MASK_SHIFT(box)) & 1;
75 }
76
77 static uint8_t
78 ao_lco_pad_present(uint8_t pad)
79 {
80         if (!ao_lco_got_channels || !ao_pad_query.channels)
81                 return pad == 0;
82         if (pad >= AO_PAD_MAX_CHANNELS)
83                 return 0;
84         return (ao_pad_query.channels >> pad) & 1;
85 }
86
87 static uint8_t
88 ao_lco_pad_first(void)
89 {
90         uint8_t pad;
91
92         for (pad = 0; pad < AO_PAD_MAX_CHANNELS; pad++)
93                 if (ao_lco_pad_present(pad))
94                         return pad;
95         return 0;
96 }
97
98 static void
99 ao_lco_input(void)
100 {
101         static struct ao_event  event;
102         int8_t  dir, new_box, new_pad;
103
104         ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(200));
105         for (;;) {
106                 ao_event_get(&event);
107                 PRINTD("event type %d unit %d value %d\n",
108                        event.type, event.unit, event.value);
109                 switch (event.type) {
110                 case AO_EVENT_QUADRATURE:
111                         switch (event.unit) {
112                         case AO_QUADRATURE_PAD:
113                                 if (!ao_lco_armed) {
114                                         dir = (int8_t) event.value;
115                                         new_pad = ao_lco_pad;
116                                         do {
117                                                 new_pad += dir;
118                                                 if (new_pad > AO_PAD_MAX_CHANNELS)
119                                                         new_pad = 0;
120                                                 else if (new_pad < 0)
121                                                         new_pad = AO_PAD_MAX_CHANNELS - 1;
122                                                 if (new_pad == ao_lco_pad)
123                                                         break;
124                                         } while (!ao_lco_pad_present(new_pad));
125                                         if (new_pad != ao_lco_pad) {
126                                                 ao_lco_pad = new_pad;
127                                                 ao_lco_set_pad(ao_lco_pad);
128                                         }
129                                 }
130                                 break;
131                         case AO_QUADRATURE_BOX:
132                                 if (!ao_lco_armed) {
133                                         dir = (int8_t) event.value;
134                                         new_box = ao_lco_box;
135                                         do {
136                                                 new_box += dir;
137                                                 if (new_box > ao_lco_max_box)
138                                                         new_box = ao_lco_min_box;
139                                                 else if (new_box < ao_lco_min_box)
140                                                         new_box = ao_lco_max_box;
141                                                 if (new_box == ao_lco_box)
142                                                         break;
143                                         } while (!ao_lco_box_present(new_box));
144                                         if (ao_lco_box != new_box) {
145                                                 ao_lco_box = new_box;
146                                                 ao_lco_got_channels = 0;
147                                                 ao_lco_set_box(ao_lco_box);
148                                         }
149                                 }
150                                 break;
151                         }
152                         break;
153                 case AO_EVENT_BUTTON:
154                         switch (event.unit) {
155                         case AO_BUTTON_ARM:
156                                 ao_lco_armed = event.value;
157                                 PRINTD("Armed %d\n", ao_lco_armed);
158                                 ao_wakeup(&ao_lco_armed);
159                                 break;
160                         case AO_BUTTON_FIRE:
161                                 if (ao_lco_armed) {
162                                         ao_lco_firing = event.value;
163                                         PRINTD("Firing %d\n", ao_lco_firing);
164                                         ao_wakeup(&ao_lco_armed);
165                                 }
166                                 break;
167                         }
168                         break;
169                 }
170         }
171 }
172
173 static AO_LED_TYPE      continuity_led[AO_LED_CONTINUITY_NUM] = {
174 #ifdef AO_LED_CONTINUITY_0
175         AO_LED_CONTINUITY_0,
176 #endif
177 #ifdef AO_LED_CONTINUITY_1
178         AO_LED_CONTINUITY_1,
179 #endif
180 #ifdef AO_LED_CONTINUITY_2
181         AO_LED_CONTINUITY_2,
182 #endif
183 #ifdef AO_LED_CONTINUITY_3
184         AO_LED_CONTINUITY_3,
185 #endif
186 #ifdef AO_LED_CONTINUITY_4
187         AO_LED_CONTINUITY_4,
188 #endif
189 #ifdef AO_LED_CONTINUITY_5
190         AO_LED_CONTINUITY_5,
191 #endif
192 #ifdef AO_LED_CONTINUITY_6
193         AO_LED_CONTINUITY_6,
194 #endif
195 #ifdef AO_LED_CONTINUITY_7
196         AO_LED_CONTINUITY_7,
197 #endif
198 };
199
200 static void
201 ao_lco_update(void)
202 {
203         int8_t                  r;
204         uint8_t                 c;
205
206         r = ao_lco_query(ao_lco_box, &ao_pad_query, &ao_lco_tick_offset);
207         if (r == AO_RADIO_CMAC_OK) {
208                 c = ao_lco_got_channels;
209                 ao_lco_got_channels = 1;
210                 ao_lco_valid = 1;
211                 if (!c) {
212                         ao_lco_pad = ao_lco_pad_first();
213                         ao_lco_set_pad(ao_lco_pad);
214                 }
215         } else
216                 ao_lco_valid = 0;
217
218 #if 0
219         PRINTD("lco_query success arm_status %d i0 %d i1 %d i2 %d i3 %d\n",
220                query.arm_status,
221                query.igniter_status[0],
222                query.igniter_status[1],
223                query.igniter_status[2],
224                query.igniter_status[3]);
225 #endif
226         ao_wakeup(&ao_pad_query);
227 }
228
229 static void
230 ao_lco_box_reset_present(void)
231 {
232         ao_lco_min_box = 0xff;
233         ao_lco_max_box = 0x00;
234         memset(ao_lco_box_mask, 0, sizeof (ao_lco_box_mask));
235 }
236
237 static void
238 ao_lco_box_set_present(uint8_t box)
239 {
240         if (box < ao_lco_min_box)
241                 ao_lco_min_box = box;
242         if (box > ao_lco_max_box)
243                 ao_lco_max_box = box;
244         if (box >= AO_PAD_MAX_BOXES)
245                 return;
246         ao_lco_box_mask[MASK_ID(box)] |= 1 << MASK_SHIFT(box);
247 }
248
249 static void
250 ao_lco_search(void)
251 {
252         uint16_t        tick_offset;
253         int8_t          r;
254         uint8_t         box;
255
256         ao_lco_box_reset_present();
257         for (box = 0; box < AO_PAD_MAX_BOXES; box++) {
258                 if ((box % 10) == 0)
259                         ao_lco_set_box(box);
260                 tick_offset = 0;
261                 r = ao_lco_query(box, &ao_pad_query, &tick_offset);
262                 PRINTD("box %d result %d\n", box, r);
263                 if (r == AO_RADIO_CMAC_OK) {
264                         ao_lco_box_set_present(box);
265                         ao_delay(AO_MS_TO_TICKS(30));
266                 }
267         }
268         if (ao_lco_min_box <= ao_lco_max_box)
269                 ao_lco_box = ao_lco_min_box;
270         else
271                 ao_lco_min_box = ao_lco_max_box = ao_lco_box = 0;
272         ao_lco_valid = 0;
273         ao_lco_got_channels = 0;
274         ao_lco_pad = 0;
275         ao_lco_set_pad(ao_lco_pad);
276         ao_lco_set_box(ao_lco_box);
277 }
278
279 static void
280 ao_lco_igniter_status(void)
281 {
282         uint8_t         c;
283
284         for (;;) {
285                 ao_sleep(&ao_pad_query);
286                 if (!ao_lco_valid) {
287                         ao_led_on(AO_LED_RED);
288                         ao_led_off(AO_LED_GREEN|AO_LED_AMBER);
289                         continue;
290                 }
291                 PRINTD("RSSI %d\n", ao_radio_cmac_rssi);
292                 if (ao_radio_cmac_rssi < -90) {
293                         ao_led_on(AO_LED_AMBER);
294                         ao_led_off(AO_LED_RED|AO_LED_GREEN);
295                 } else {
296                         ao_led_on(AO_LED_GREEN);
297                         ao_led_off(AO_LED_RED|AO_LED_AMBER);
298                 }
299                 if (ao_pad_query.arm_status)
300                         ao_led_on(AO_LED_REMOTE_ARM);
301                 else
302                         ao_led_off(AO_LED_REMOTE_ARM);
303                 for (c = 0; c < AO_LED_CONTINUITY_NUM; c++) {
304                         uint8_t status;
305
306                         if (ao_pad_query.channels & (1 << c))
307                                 status = ao_pad_query.igniter_status[c];
308                         else
309                                 status = AO_PAD_IGNITER_STATUS_NO_IGNITER_RELAY_OPEN;
310                         if (status == AO_PAD_IGNITER_STATUS_GOOD_IGNITER_RELAY_OPEN)
311                                 ao_led_on(continuity_led[c]);
312                         else
313                                 ao_led_off(continuity_led[c]);
314                 }
315         }
316 }
317
318 static void
319 ao_lco_arm_warn(void)
320 {
321         for (;;) {
322                 while (!ao_lco_armed)
323                         ao_sleep(&ao_lco_armed);
324                 ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(200));
325                 ao_delay(AO_MS_TO_TICKS(200));
326         }
327 }
328
329 static struct ao_task ao_lco_input_task;
330 static struct ao_task ao_lco_monitor_task;
331 static struct ao_task ao_lco_arm_warn_task;
332 static struct ao_task ao_lco_igniter_status_task;
333
334 static void
335 ao_lco_monitor(void)
336 {
337         uint16_t                delay;
338
339         ao_lco_search();
340         ao_add_task(&ao_lco_input_task, ao_lco_input, "lco input");
341         ao_add_task(&ao_lco_arm_warn_task, ao_lco_arm_warn, "lco arm warn");
342         ao_add_task(&ao_lco_igniter_status_task, ao_lco_igniter_status, "lco igniter status");
343         for (;;) {
344                 PRINTD("monitor armed %d firing %d offset %d\n",
345                        ao_lco_armed, ao_lco_firing, ao_lco_tick_offset);
346
347                 if (ao_lco_armed && ao_lco_firing) {
348                         PRINTD("Firing box %d pad %d: valid %d\n",
349                                ao_lco_box, ao_lco_pad, ao_lco_valid);
350                         if (!ao_lco_valid)
351                                 ao_lco_update();
352                         if (ao_lco_valid)
353                                 ao_lco_ignite(ao_lco_box, 1 << ao_lco_pad, ao_lco_tick_offset);
354                 } else if (ao_lco_armed) {
355                         PRINTD("Arming box %d pad %d\n",
356                                ao_lco_box, ao_lco_pad);
357                         if (!ao_lco_valid)
358                                 ao_lco_update();
359                         ao_lco_arm(ao_lco_box, 1 << ao_lco_pad, ao_lco_tick_offset);
360                         ao_lco_update();
361                 } else {
362                         ao_lco_update();
363                 }
364                 if (ao_lco_armed && ao_lco_firing)
365                         delay = AO_MS_TO_TICKS(100);
366                 else
367                         delay = AO_SEC_TO_TICKS(1);
368                 ao_alarm(delay);
369                 ao_sleep(&ao_lco_armed);
370                 ao_clear_alarm();
371         }
372 }
373
374 #if DEBUG
375 void
376 ao_lco_set_debug(void)
377 {
378         ao_cmd_decimal();
379         if (ao_cmd_status == ao_cmd_success)
380                 ao_lco_debug = ao_cmd_lex_i != 0;
381 }
382
383 __code struct ao_cmds ao_lco_cmds[] = {
384         { ao_lco_set_debug,     "D <0 off, 1 on>\0Debug" },
385         { ao_lco_search,        "s\0Search for pad boxes" },
386         { 0, NULL }
387 };
388 #endif
389
390 void
391 ao_lco_init(void)
392 {
393         ao_add_task(&ao_lco_monitor_task, ao_lco_monitor, "lco monitor");
394 #if DEBUG
395         ao_cmd_register(&ao_lco_cmds[0]);
396 #endif
397 }