464e05ab363c1bf115a7321cf0859b5d976eccb1
[fw/altos] / src / drivers / 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_selected[AO_PAD_MAX_BOXES];
41 static uint8_t  ao_lco_armed;
42 static uint8_t  ao_lco_firing;
43 static uint8_t  ao_lco_valid[AO_PAD_MAX_BOXES];
44 static uint8_t  ao_lco_channels[AO_PAD_MAX_BOXES];
45 static uint16_t ao_lco_tick_offset[AO_PAD_MAX_BOXES];
46
47 static uint8_t  ao_lco_pad;
48 static uint8_t  ao_lco_box;
49 static struct ao_pad_query      ao_pad_query;
50
51 static void
52 ao_lco_set_pad(uint8_t pad)
53 {
54         ao_seven_segment_set(AO_LCO_PAD_DIGIT, pad);
55 }
56
57 static void
58 ao_lco_set_box(uint8_t box)
59 {
60         ao_seven_segment_set(AO_LCO_BOX_DIGIT_1, box % 10);
61         ao_seven_segment_set(AO_LCO_BOX_DIGIT_10, box / 10);
62 }
63
64 static void
65 ao_lco_set_voltage(uint16_t decivolts)
66 {
67         uint8_t tens, ones, tenths;
68
69         tenths = decivolts % 10;
70         ones = (decivolts / 10) % 10;
71         tens = (decivolts / 100) % 10;
72         ao_seven_segment_set(AO_LCO_PAD_DIGIT, tenths);
73         ao_seven_segment_set(AO_LCO_BOX_DIGIT_1, ones | 0x10);
74         ao_seven_segment_set(AO_LCO_BOX_DIGIT_10, tens);
75 }
76
77 static void
78 ao_lco_set_display(void)
79 {
80         if (ao_lco_pad == 0) {
81                 ao_lco_set_voltage(ao_pad_query.battery);
82         } else {
83                 ao_lco_set_pad(ao_lco_pad);
84                 ao_lco_set_box(ao_lco_box);
85         }
86 }
87
88 #define MASK_SIZE(n)    (((n) + 7) >> 3)
89 #define MASK_ID(n)      ((n) >> 3)
90 #define MASK_SHIFT(n)   ((n) & 7)
91
92 static uint8_t  ao_lco_box_mask[MASK_SIZE(AO_PAD_MAX_BOXES)];
93
94 static uint8_t
95 ao_lco_box_present(uint8_t box)
96 {
97         if (box >= AO_PAD_MAX_BOXES)
98                 return 0;
99         return (ao_lco_box_mask[MASK_ID(box)] >> MASK_SHIFT(box)) & 1;
100 }
101
102 static uint8_t
103 ao_lco_pad_present(uint8_t box, uint8_t pad)
104 {
105         /* voltage measurement is always valid */
106         if (pad == 0)
107                 return 1;
108         if (!ao_lco_channels[box])
109                 return 0;
110         if (pad > AO_PAD_MAX_CHANNELS)
111                 return 0;
112         return (ao_lco_channels[box] >> (pad - 1)) & 1;
113 }
114
115 static uint8_t
116 ao_lco_pad_first(uint8_t box)
117 {
118         uint8_t pad;
119
120         for (pad = 1; pad <= AO_PAD_MAX_CHANNELS; pad++)
121                 if (ao_lco_pad_present(box, pad))
122                         return pad;
123         return 0;
124 }
125
126 static void
127 ao_lco_input(void)
128 {
129         static struct ao_event  event;
130         int8_t  dir, new_box, new_pad;
131
132         ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(200));
133         for (;;) {
134                 ao_event_get(&event);
135                 PRINTD("event type %d unit %d value %d\n",
136                        event.type, event.unit, event.value);
137                 switch (event.type) {
138                 case AO_EVENT_QUADRATURE:
139                         switch (event.unit) {
140                         case AO_QUADRATURE_PAD:
141                                 if (!ao_lco_armed) {
142                                         dir = (int8_t) event.value;
143                                         new_pad = ao_lco_pad;
144                                         do {
145                                                 new_pad += dir;
146                                                 if (new_pad > AO_PAD_MAX_CHANNELS)
147                                                         new_pad = 0;
148                                                 if (new_pad < 0)
149                                                         new_pad = AO_PAD_MAX_CHANNELS;
150                                                 if (new_pad == ao_lco_pad)
151                                                         break;
152                                         } while (!ao_lco_pad_present(ao_lco_box, new_pad));
153                                         if (new_pad != ao_lco_pad) {
154                                                 ao_lco_pad = new_pad;
155                                                 ao_lco_set_display();
156                                         }
157                                 }
158                                 break;
159                         case AO_QUADRATURE_BOX:
160                                 if (!ao_lco_armed) {
161                                         dir = (int8_t) event.value;
162                                         new_box = ao_lco_box;
163                                         do {
164                                                 new_box += dir;
165                                                 if (new_box > ao_lco_max_box)
166                                                         new_box = ao_lco_min_box;
167                                                 else if (new_box < ao_lco_min_box)
168                                                         new_box = ao_lco_max_box;
169                                                 if (new_box == ao_lco_box)
170                                                         break;
171                                         } while (!ao_lco_box_present(new_box));
172                                         if (ao_lco_box != new_box) {
173                                                 ao_lco_box = new_box;
174                                                 ao_lco_pad = 1;
175                                                 ao_lco_channels[ao_lco_box] = 0;
176                                                 ao_lco_set_display();
177                                         }
178                                 }
179                                 break;
180                         }
181                         break;
182                 case AO_EVENT_BUTTON:
183                         switch (event.unit) {
184                         case AO_BUTTON_ARM:
185                                 ao_lco_armed = event.value;
186                                 PRINTD("Armed %d\n", ao_lco_armed);
187                                 if (ao_lco_pad != 0) {
188                                         memset(ao_lco_selected, 0, sizeof (ao_lco_selected));
189                                         ao_lco_selected[ao_lco_box] = (1 << (ao_lco_pad - 1));
190                                         ao_wakeup(&ao_lco_armed);
191                                 }
192                                 break;
193                         case AO_BUTTON_FIRE:
194                                 if (ao_lco_armed) {
195                                         ao_lco_firing = event.value;
196                                         PRINTD("Firing %d\n", ao_lco_firing);
197                                         ao_wakeup(&ao_lco_armed);
198                                 }
199                                 break;
200                         }
201                         break;
202                 }
203         }
204 }
205
206 static AO_LED_TYPE      continuity_led[AO_LED_CONTINUITY_NUM] = {
207 #ifdef AO_LED_CONTINUITY_0
208         AO_LED_CONTINUITY_0,
209 #endif
210 #ifdef AO_LED_CONTINUITY_1
211         AO_LED_CONTINUITY_1,
212 #endif
213 #ifdef AO_LED_CONTINUITY_2
214         AO_LED_CONTINUITY_2,
215 #endif
216 #ifdef AO_LED_CONTINUITY_3
217         AO_LED_CONTINUITY_3,
218 #endif
219 #ifdef AO_LED_CONTINUITY_4
220         AO_LED_CONTINUITY_4,
221 #endif
222 #ifdef AO_LED_CONTINUITY_5
223         AO_LED_CONTINUITY_5,
224 #endif
225 #ifdef AO_LED_CONTINUITY_6
226         AO_LED_CONTINUITY_6,
227 #endif
228 #ifdef AO_LED_CONTINUITY_7
229         AO_LED_CONTINUITY_7,
230 #endif
231 };
232
233 static uint8_t
234 ao_lco_get_channels(uint8_t box, struct ao_pad_query *query)
235 {
236         int8_t                  r;
237
238         r = ao_lco_query(box, query, &ao_lco_tick_offset[box]);
239         if (r == AO_RADIO_CMAC_OK) {
240                 ao_lco_channels[box] = query->channels;
241                 ao_lco_valid[box] = 1;
242         } else
243                 ao_lco_valid[box] = 0;
244         PRINTD("ao_lco_get_channels(%d) valid %d\n", box, ao_lco_valid[box]);
245         ao_wakeup(&ao_pad_query);
246         return ao_lco_valid[box];
247 }
248
249 static void
250 ao_lco_update(void)
251 {
252         uint8_t previous_valid = ao_lco_valid[ao_lco_box];
253
254         if (ao_lco_get_channels(ao_lco_box, &ao_pad_query)) {
255                 if (!previous_valid) {
256                         if (ao_lco_pad != 0)
257                                 ao_lco_pad = ao_lco_pad_first(ao_lco_box);
258                         ao_lco_set_display();
259                 }
260                 if (ao_lco_pad == 0)
261                         ao_lco_set_display();
262         }
263 }
264
265 static void
266 ao_lco_box_reset_present(void)
267 {
268         ao_lco_min_box = 0xff;
269         ao_lco_max_box = 0x00;
270         memset(ao_lco_box_mask, 0, sizeof (ao_lco_box_mask));
271 }
272
273 static void
274 ao_lco_box_set_present(uint8_t box)
275 {
276         if (box < ao_lco_min_box)
277                 ao_lco_min_box = box;
278         if (box > ao_lco_max_box)
279                 ao_lco_max_box = box;
280         if (box >= AO_PAD_MAX_BOXES)
281                 return;
282         ao_lco_box_mask[MASK_ID(box)] |= 1 << MASK_SHIFT(box);
283 }
284
285 static void
286 ao_lco_search(void)
287 {
288         uint16_t        tick_offset;
289         int8_t          r;
290         int8_t          try;
291         uint8_t         box;
292         uint8_t         boxes = 0;
293
294         ao_lco_box_reset_present();
295         ao_lco_set_pad(0);
296         for (box = 0; box < AO_PAD_MAX_BOXES; box++) {
297                 if ((box % 10) == 0)
298                         ao_lco_set_box(box);
299                 for (try = 0; try < 3; try++) {
300                         tick_offset = 0;
301                         r = ao_lco_query(box, &ao_pad_query, &tick_offset);
302                         PRINTD("box %d result %d\n", box, r);
303                         if (r == AO_RADIO_CMAC_OK) {
304                                 ++boxes;
305                                 ao_lco_box_set_present(box);
306                                 ao_lco_set_pad(boxes % 10);
307                                 ao_delay(AO_MS_TO_TICKS(30));
308                                 break;
309                         }
310                 }
311         }
312         if (ao_lco_min_box <= ao_lco_max_box)
313                 ao_lco_box = ao_lco_min_box;
314         else
315                 ao_lco_min_box = ao_lco_max_box = ao_lco_box = 0;
316         memset(ao_lco_valid, 0, sizeof (ao_lco_valid));
317         memset(ao_lco_channels, 0, sizeof (ao_lco_channels));
318         ao_lco_pad = 1;
319         ao_lco_set_display();
320 }
321
322 static void
323 ao_lco_igniter_status(void)
324 {
325         uint8_t         c;
326
327         for (;;) {
328                 ao_sleep(&ao_pad_query);
329                 PRINTD("RSSI %d VALID %d\n", ao_radio_cmac_rssi, ao_lco_valid[ao_lco_box]);
330                 if (!ao_lco_valid[ao_lco_box]) {
331                         ao_led_on(AO_LED_RED);
332                         ao_led_off(AO_LED_GREEN|AO_LED_AMBER);
333                         continue;
334                 }
335                 if (ao_radio_cmac_rssi < -90) {
336                         ao_led_on(AO_LED_AMBER);
337                         ao_led_off(AO_LED_RED|AO_LED_GREEN);
338                 } else {
339                         ao_led_on(AO_LED_GREEN);
340                         ao_led_off(AO_LED_RED|AO_LED_AMBER);
341                 }
342                 if (ao_pad_query.arm_status)
343                         ao_led_on(AO_LED_REMOTE_ARM);
344                 else
345                         ao_led_off(AO_LED_REMOTE_ARM);
346                 for (c = 0; c < AO_LED_CONTINUITY_NUM; c++) {
347                         uint8_t status;
348
349                         if (ao_pad_query.channels & (1 << c))
350                                 status = ao_pad_query.igniter_status[c];
351                         else
352                                 status = AO_PAD_IGNITER_STATUS_NO_IGNITER_RELAY_OPEN;
353                         if (status == AO_PAD_IGNITER_STATUS_GOOD_IGNITER_RELAY_OPEN)
354                                 ao_led_on(continuity_led[c]);
355                         else
356                                 ao_led_off(continuity_led[c]);
357                 }
358         }
359 }
360
361 static void
362 ao_lco_arm_warn(void)
363 {
364         for (;;) {
365                 while (!ao_lco_armed)
366                         ao_sleep(&ao_lco_armed);
367                 ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(200));
368                 ao_delay(AO_MS_TO_TICKS(200));
369         }
370 }
371
372 static struct ao_task ao_lco_input_task;
373 static struct ao_task ao_lco_monitor_task;
374 static struct ao_task ao_lco_arm_warn_task;
375 static struct ao_task ao_lco_igniter_status_task;
376
377 static void
378 ao_lco_monitor(void)
379 {
380         uint16_t                delay;
381         uint8_t                 box;
382         struct ao_pad_query     pad_query;
383
384         ao_lco_search();
385         ao_add_task(&ao_lco_input_task, ao_lco_input, "lco input");
386         ao_add_task(&ao_lco_arm_warn_task, ao_lco_arm_warn, "lco arm warn");
387         ao_add_task(&ao_lco_igniter_status_task, ao_lco_igniter_status, "lco igniter status");
388         for (;;) {
389                 PRINTD("monitor armed %d firing %d\n",
390                        ao_lco_armed, ao_lco_firing);
391
392                 if (ao_lco_armed && ao_lco_firing) {
393                         ao_lco_ignite();
394                 } else {
395                         if (ao_lco_armed) {
396                                 for (box = 0; box < AO_PAD_MAX_BOXES; box++) {
397                                         if (ao_lco_selected[box]) {
398                                                 PRINTD("Arming box %d pads %x\n",
399                                                        box, ao_lco_selected[box]);
400                                                 if (!ao_lco_valid[box])
401                                                         ao_lco_get_channels(box, &pad_query);
402                                                 if (ao_lco_valid[box])
403                                                         ao_lco_arm(box, ao_lco_selected[box], ao_lco_tick_offset[ao_lco_box]);
404                                         }
405                                 }
406                         }
407                         ao_lco_update();
408                 }
409                 if (ao_lco_armed && ao_lco_firing)
410                         delay = AO_MS_TO_TICKS(100);
411                 else
412                         delay = AO_SEC_TO_TICKS(1);
413                 ao_sleep_for(&ao_lco_armed, delay);
414         }
415 }
416
417 #if DEBUG
418 void
419 ao_lco_set_debug(void)
420 {
421         ao_cmd_decimal();
422         if (ao_cmd_status == ao_cmd_success)
423                 ao_lco_debug = ao_cmd_lex_i != 0;
424 }
425
426 __code struct ao_cmds ao_lco_cmds[] = {
427         { ao_lco_set_debug,     "D <0 off, 1 on>\0Debug" },
428         { ao_lco_search,        "s\0Search for pad boxes" },
429         { 0, NULL }
430 };
431 #endif
432
433 void
434 ao_lco_init(void)
435 {
436         ao_add_task(&ao_lco_monitor_task, ao_lco_monitor, "lco monitor");
437 #if DEBUG
438         ao_cmd_register(&ao_lco_cmds[0]);
439 #endif
440 }