1cb0546c0fc0f8e5dc207977ae82d56b6da54ba1
[fw/altos] / src / drivers / ao_lco_two.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; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License along
15  * with this program; if not, write to the Free Software Foundation, Inc.,
16  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
17  */
18
19 #include <ao.h>
20 #include <ao_lco.h>
21 #include <ao_event.h>
22 #include <ao_button.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 DEBUG_EVENT     1
31 #define DEBUG_STATUS    2
32 #define PRINTD(l, ...) do { if (!(ao_lco_debug & l)) break; printf ("\r%5u %s: ", ao_tick_count, __func__); printf(__VA_ARGS__); flush(); } while(0)
33 #else
34 #define PRINTD(l,...)
35 #endif
36
37 #define AO_LCO_VALID_LAST       1
38 #define AO_LCO_VALID_EVER       2
39
40 static uint8_t  ao_lco_suspended;
41 static uint8_t  ao_lco_valid;
42 static uint8_t  ao_lco_channels;
43 static uint16_t ao_lco_tick_offset;
44
45 /* UI values */
46 static uint8_t  ao_lco_armed;
47 static uint8_t  ao_lco_firing;
48 static uint8_t  ao_lco_box;
49
50 static struct ao_pad_query      ao_pad_query;
51
52 /* TeleFireTwo boxes have a single pad */
53 #define ao_lco_pad      0
54
55 static void
56 ao_lco_set_box(int box)
57 {
58         ao_lco_box = ao_config.pad_box + box;
59         ao_lco_valid = 0;
60         ao_lco_armed = 0;
61         ao_wakeup(&ao_lco_armed);
62 }
63
64 static void
65 ao_lco_set_armed(int armed)
66 {
67         uint8_t bit = (1 << ao_lco_pad);
68
69         if (armed) {
70                 ao_lco_armed = bit;
71         } else {
72                 ao_lco_armed = 0;
73         }
74         PRINTD(DEBUG_EVENT, "pad %d bit 0x%x armed %d ao_lco_armed 0x%x\n",
75                ao_lco_pad, bit, armed, ao_lco_armed);
76         ao_wakeup(&ao_lco_armed);
77 }
78
79 static void
80 ao_lco_suspend(void)
81 {
82         if (!ao_lco_suspended) {
83                 PRINTD(DEBUG_EVENT, "suspend\n");
84                 ao_lco_suspended = 1;
85                 ao_lco_armed = 0;
86                 ao_wakeup(&ao_pad_query);
87         }
88 }
89
90 static void
91 ao_lco_wakeup(void)
92 {
93         if (ao_lco_suspended) {
94                 ao_lco_suspended = 0;
95                 ao_wakeup(&ao_lco_suspended);
96         }
97 }
98
99 static void
100 ao_lco_input(void)
101 {
102         static struct ao_event  event;
103         uint8_t timeout;
104
105         for (;;) {
106                 if (ao_config.pad_idle && !ao_lco_suspended) {
107                         timeout = ao_event_get_for(&event, AO_SEC_TO_TICKS(ao_config.pad_idle));
108                         if (timeout) {
109                                 ao_lco_suspend();
110                                 continue;
111                         }
112                 } else {
113                         ao_event_get(&event);
114                 }
115                 ao_lco_wakeup();
116                 PRINTD(DEBUG_EVENT, "event type %d unit %d value %d\n",
117                        event.type, event.unit, event.value);
118                 switch (event.type) {
119                 case AO_EVENT_BUTTON:
120                         switch (event.unit) {
121                         case AO_BUTTON_BOX:
122                                 ao_lco_set_box(event.value);
123                                 break;
124                         case AO_BUTTON_ARM:
125                                 ao_lco_set_armed(event.value);
126                                 break;
127                         case AO_BUTTON_FIRE:
128                                 if (ao_lco_armed) {
129                                         ao_lco_firing = event.value;
130                                         PRINTD(DEBUG_EVENT, "Firing %d\n", ao_lco_firing);
131                                         ao_wakeup(&ao_lco_armed);
132                                 }
133                                 break;
134                         }
135                         break;
136                 }
137         }
138 }
139
140 static AO_LED_TYPE      continuity_led[AO_LED_CONTINUITY_NUM] = {
141 #ifdef AO_LED_CONTINUITY_0
142         AO_LED_CONTINUITY_0,
143 #endif
144 #ifdef AO_LED_CONTINUITY_1
145         AO_LED_CONTINUITY_1,
146 #endif
147 #ifdef AO_LED_CONTINUITY_2
148         AO_LED_CONTINUITY_2,
149 #endif
150 #ifdef AO_LED_CONTINUITY_3
151         AO_LED_CONTINUITY_3,
152 #endif
153 #ifdef AO_LED_CONTINUITY_4
154         AO_LED_CONTINUITY_4,
155 #endif
156 #ifdef AO_LED_CONTINUITY_5
157         AO_LED_CONTINUITY_5,
158 #endif
159 #ifdef AO_LED_CONTINUITY_6
160         AO_LED_CONTINUITY_6,
161 #endif
162 #ifdef AO_LED_CONTINUITY_7
163         AO_LED_CONTINUITY_7,
164 #endif
165 };
166
167 static uint8_t
168 ao_lco_get_channels(void)
169 {
170         int8_t                  r;
171
172         r = ao_lco_query(ao_lco_box, &ao_pad_query, &ao_lco_tick_offset);
173         if (r == AO_RADIO_CMAC_OK) {
174                 ao_lco_channels = ao_pad_query.channels;
175                 ao_lco_valid = AO_LCO_VALID_LAST | AO_LCO_VALID_EVER;
176         } else
177                 ao_lco_valid &= ~AO_LCO_VALID_LAST;
178         PRINTD(DEBUG_STATUS, "ao_lco_get_channels() rssi %d valid %d ret %d offset %d\n", ao_radio_cmac_rssi, ao_lco_valid, r, ao_lco_tick_offset);
179         ao_wakeup(&ao_pad_query);
180         return ao_lco_valid;
181 }
182
183 static void
184 ao_lco_igniter_status(void)
185 {
186         uint8_t         c;
187         uint8_t         t = 0;
188
189         for (;;) {
190                 uint8_t all_status;
191                 ao_sleep(&ao_pad_query);
192                 while (ao_lco_suspended) {
193                         ao_led_off(AO_LED_GREEN|AO_LED_AMBER|AO_LED_RED|AO_LED_REMOTE_ARM);
194                         for (c = 0; c < AO_LED_CONTINUITY_NUM; c++)
195                                 ao_led_off(continuity_led[c]);
196                         ao_sleep(&ao_lco_suspended);
197                 }
198                 PRINTD(DEBUG_STATUS, "RSSI %d VALID %d channels %d arm_status %d\n",
199                        ao_radio_cmac_rssi, ao_lco_valid,
200                        ao_lco_channels, ao_pad_query.arm_status);
201                 if (!(ao_lco_valid & AO_LCO_VALID_LAST)) {
202                         ao_led_on(AO_LED_RED);
203                         ao_led_off(AO_LED_GREEN|AO_LED_AMBER);
204                         memset(&ao_pad_query, '\0', sizeof (ao_pad_query));
205                 } else if (ao_radio_cmac_rssi < -90) {
206                         ao_led_on(AO_LED_AMBER);
207                         ao_led_off(AO_LED_RED|AO_LED_GREEN);
208                 } else {
209                         ao_led_on(AO_LED_GREEN);
210                         ao_led_off(AO_LED_RED|AO_LED_AMBER);
211                 }
212                 if (ao_pad_query.arm_status)
213                         ao_led_on(AO_LED_REMOTE_ARM);
214                 else
215                         ao_led_off(AO_LED_REMOTE_ARM);
216
217                 all_status = AO_PAD_IGNITER_STATUS_NO_IGNITER_RELAY_OPEN;
218                 for (c = 0; c < 8; c++) {
219                         if (ao_pad_query.channels & (1 << c)) {
220                                 uint8_t status = ao_pad_query.igniter_status[c];
221                                 if (status > all_status)
222                                         all_status = status;
223                                 PRINTD(DEBUG_STATUS, "\tchannel %d status %d\n", c, status);
224                         }
225                 }
226                 for (c = 0; c < AO_LED_CONTINUITY_NUM; c++) {
227                         uint8_t on = 0;
228                         if (c == (ao_lco_box - ao_config.pad_box) % AO_LED_CONTINUITY_NUM) {
229                                 switch (all_status) {
230                                 case AO_PAD_IGNITER_STATUS_GOOD_IGNITER_RELAY_OPEN:
231                                         on = 1;
232                                         break;
233                                 case AO_PAD_IGNITER_STATUS_NO_IGNITER_RELAY_CLOSED:
234                                 case AO_PAD_IGNITER_STATUS_UNKNOWN:
235                                         on = t & 1;
236                                 }
237                         }
238                         if (on)
239                                 ao_led_on(continuity_led[c]);
240                         else
241                                 ao_led_off(continuity_led[c]);
242                 }
243                 t = 1-t;
244         }
245 }
246
247 static void
248 ao_lco_arm_warn(void)
249 {
250         int     i;
251         for (;;) {
252                 while (ao_lco_suspended)
253                         ao_sleep(&ao_lco_suspended);
254                 while (!ao_lco_armed)
255                         ao_sleep(&ao_lco_armed);
256                 for (i = 0; i < ao_lco_armed; i++) {
257                         ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(100));
258                         ao_delay(AO_MS_TO_TICKS(100));
259                 }
260                 ao_delay(AO_MS_TO_TICKS(300));
261         }
262 }
263
264 static struct ao_task ao_lco_input_task;
265 static struct ao_task ao_lco_monitor_task;
266 static struct ao_task ao_lco_arm_warn_task;
267 static struct ao_task ao_lco_igniter_status_task;
268
269 static void
270 ao_lco_monitor(void)
271 {
272         uint16_t                delay;
273
274         ao_config_get();
275         ao_lco_set_box(ao_button_get(AO_BUTTON_BOX));
276         ao_add_task(&ao_lco_input_task, ao_lco_input, "lco input");
277         ao_add_task(&ao_lco_arm_warn_task, ao_lco_arm_warn, "lco arm warn");
278         ao_add_task(&ao_lco_igniter_status_task, ao_lco_igniter_status, "lco igniter status");
279         ao_led_on(~0);
280         ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(200));
281         ao_led_off(~0);
282         for (;;) {
283                 while (ao_lco_suspended)
284                         ao_sleep(&ao_lco_suspended);
285
286                 PRINTD(DEBUG_STATUS, "monitor armed %d firing %d\n",
287                        ao_lco_armed, ao_lco_firing);
288
289                 if (ao_lco_armed && ao_lco_firing) {
290                         ao_lco_ignite();
291                 } else {
292                         ao_lco_get_channels();
293                         if (ao_lco_armed) {
294                                 PRINTD(DEBUG_STATUS, "Arming pads %x\n",
295                                        ao_lco_armed);
296                                 if (ao_lco_valid & AO_LCO_VALID_EVER) {
297                                         ao_lco_arm(ao_lco_box, ao_lco_armed, ao_lco_tick_offset);
298                                         ao_delay(AO_MS_TO_TICKS(10));
299                                 }
300                         }
301                 }
302                 if (ao_lco_armed && ao_lco_firing)
303                         delay = AO_MS_TO_TICKS(100);
304                 else {
305                         delay = AO_SEC_TO_TICKS(1);
306                 }
307                 ao_sleep_for(&ao_lco_armed, delay);
308         }
309 }
310
311 #if DEBUG
312 void
313 ao_lco_set_debug(void)
314 {
315         ao_cmd_decimal();
316         if (ao_cmd_status == ao_cmd_success)
317                 ao_lco_debug = ao_cmd_lex_i;
318 }
319
320 __code struct ao_cmds ao_lco_cmds[] = {
321         { ao_lco_set_debug,     "D <0 off, 1 on>\0Debug" },
322         { 0, NULL }
323 };
324 #endif
325
326 void
327 ao_lco_init(void)
328 {
329         ao_add_task(&ao_lco_monitor_task, ao_lco_monitor, "lco monitor");
330 #if DEBUG
331         ao_cmd_register(&ao_lco_cmds[0]);
332 #endif
333 }