altos: Add ao_lco_two.c; alternate LCO interface code
[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; 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_lco_func.h>
22 #include <ao_radio_cmac.h>
23
24 #define DEBUG   1
25
26 #if DEBUG
27 static uint8_t  ao_lco_debug;
28 #define DEBUG_EVENT     1
29 #define DEBUG_STATUS    2
30 #define PRINTD(l, ...) do { if (!(ao_lco_debug & l)) break; printf ("\r%5u %s: ", ao_tick_count, __func__); printf(__VA_ARGS__); flush(); } while(0)
31 #else
32 #define PRINTD(l,...) 
33 #endif
34
35 #define AO_LCO_VALID_LAST       1
36 #define AO_LCO_VALID_EVER       2
37
38 static uint8_t  ao_lco_selected;
39 static uint8_t  ao_lco_valid;
40 static uint8_t  ao_lco_channels;
41 static uint16_t ao_lco_tick_offset;
42
43 /* UI values */
44 static uint8_t  ao_lco_armed;
45 static uint8_t  ao_lco_firing;
46 static uint8_t  ao_lco_fire_down;
47
48 #define ao_lco_box      (ao_config.pad_box)
49
50 static struct ao_pad_query      ao_pad_query;
51
52 #define MASK_SIZE(n)    (((n) + 7) >> 3)
53 #define MASK_ID(n)      ((n) >> 3)
54 #define MASK_SHIFT(n)   ((n) & 7)
55
56 static void
57 ao_lco_set_armed(int pad, int armed)
58 {
59         uint8_t bit = (1 << pad);
60
61         if (armed) {
62                 ao_lco_selected |= bit;
63                 ao_lco_armed |= bit;
64         } else {
65                 ao_lco_selected &= ~bit;
66                 ao_lco_armed &= ~bit;
67         }
68         PRINTD(DEBUG_EVENT, "pad %d bit 0x%x armed %d ao_lco_selected 0x%x ao_lco_armed 0x%x\n",
69                pad, bit, armed, ao_lco_selected, ao_lco_armed);
70         ao_wakeup(&ao_lco_armed);
71 }
72
73 static void
74 ao_lco_input(void)
75 {
76         static struct ao_event  event;
77
78         for (;;) {
79                 ao_event_get(&event);
80                 PRINTD(DEBUG_EVENT, "event type %d unit %d value %d\n",
81                        event.type, event.unit, event.value);
82                 switch (event.type) {
83                 case AO_EVENT_BUTTON:
84                         switch (event.unit) {
85                         case AO_BUTTON_ARM_0:
86                                 ao_lco_set_armed(0, event.value);
87                                 break;
88 #if AO_BUTTON_ARM_NUM > 1
89                         case AO_BUTTON_ARM_1:
90                                 ao_lco_set_armed(1, event.value);
91                                 break;
92 #endif
93                         case AO_BUTTON_FIRE:
94                                 if (ao_lco_armed) {
95                                         ao_lco_fire_down = 0;
96                                         ao_lco_firing = event.value;
97                                         PRINTD(DEBUG_EVENT, "Firing %d\n", ao_lco_firing);
98                                         ao_wakeup(&ao_lco_armed);
99                                 }
100                                 break;
101                         }
102                         break;
103                 }
104         }
105 }
106
107 static AO_LED_TYPE      continuity_led[AO_LED_CONTINUITY_NUM] = {
108 #ifdef AO_LED_CONTINUITY_0
109         AO_LED_CONTINUITY_0,
110 #endif
111 #ifdef AO_LED_CONTINUITY_1
112         AO_LED_CONTINUITY_1,
113 #endif
114 #ifdef AO_LED_CONTINUITY_2
115         AO_LED_CONTINUITY_2,
116 #endif
117 #ifdef AO_LED_CONTINUITY_3
118         AO_LED_CONTINUITY_3,
119 #endif
120 #ifdef AO_LED_CONTINUITY_4
121         AO_LED_CONTINUITY_4,
122 #endif
123 #ifdef AO_LED_CONTINUITY_5
124         AO_LED_CONTINUITY_5,
125 #endif
126 #ifdef AO_LED_CONTINUITY_6
127         AO_LED_CONTINUITY_6,
128 #endif
129 #ifdef AO_LED_CONTINUITY_7
130         AO_LED_CONTINUITY_7,
131 #endif
132 };
133
134 static uint8_t
135 ao_lco_get_channels(void)
136 {
137         int8_t                  r;
138
139         r = ao_lco_query(ao_lco_box, &ao_pad_query, &ao_lco_tick_offset);
140         if (r == AO_RADIO_CMAC_OK) {
141                 ao_lco_channels = ao_pad_query.channels;
142                 ao_lco_valid = AO_LCO_VALID_LAST | AO_LCO_VALID_EVER;
143         } else
144                 ao_lco_valid &= ~AO_LCO_VALID_LAST;
145         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);
146         ao_wakeup(&ao_pad_query);
147         return ao_lco_valid;
148 }
149
150 static void
151 ao_lco_igniter_status(void)
152 {
153         uint8_t         c;
154         uint8_t         t = 0;
155
156         for (;;) {
157                 ao_sleep(&ao_pad_query);
158                 PRINTD(DEBUG_STATUS, "RSSI %d VALID %d\n", ao_radio_cmac_rssi, ao_lco_valid);
159                 if (!(ao_lco_valid & AO_LCO_VALID_LAST)) {
160                         ao_led_on(AO_LED_RED);
161                         ao_led_off(AO_LED_GREEN|AO_LED_AMBER);
162                         continue;
163                 }
164                 if (ao_radio_cmac_rssi < -90) {
165                         ao_led_on(AO_LED_AMBER);
166                         ao_led_off(AO_LED_RED|AO_LED_GREEN);
167                 } else {
168                         ao_led_on(AO_LED_GREEN);
169                         ao_led_off(AO_LED_RED|AO_LED_AMBER);
170                 }
171                 if (ao_pad_query.arm_status)
172                         ao_led_on(AO_LED_REMOTE_ARM);
173                 else
174                         ao_led_off(AO_LED_REMOTE_ARM);
175
176                 for (c = 0; c < AO_LED_CONTINUITY_NUM; c++) {
177                         uint8_t status;
178
179                         if (ao_pad_query.channels & (1 << c))
180                                 status = ao_pad_query.igniter_status[c];
181                         else
182                                 status = AO_PAD_IGNITER_STATUS_NO_IGNITER_RELAY_OPEN;
183                         PRINTD(DEBUG_STATUS, "\tchannel %d status %d\n", c, status);
184                         if (status == AO_PAD_IGNITER_STATUS_GOOD_IGNITER_RELAY_OPEN)
185                                 ao_led_on(continuity_led[c]);
186                         else
187                                 ao_led_off(continuity_led[c]);
188                 }
189                 t = 1-t;
190         }
191 }
192
193 static void
194 ao_lco_arm_warn(void)
195 {
196         int     i;
197         for (;;) {
198                 while (!ao_lco_armed)
199                         ao_sleep(&ao_lco_armed);
200                 for (i = 0; i < ao_lco_armed; i++) {
201                         ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(100));
202                         ao_delay(AO_MS_TO_TICKS(100));
203                 }
204                 ao_delay(AO_MS_TO_TICKS(300));
205         }
206 }
207
208 static struct ao_task ao_lco_input_task;
209 static struct ao_task ao_lco_monitor_task;
210 static struct ao_task ao_lco_arm_warn_task;
211 static struct ao_task ao_lco_igniter_status_task;
212
213 static void
214 ao_lco_monitor(void)
215 {
216         uint16_t                delay;
217
218         ao_add_task(&ao_lco_input_task, ao_lco_input, "lco input");
219         ao_add_task(&ao_lco_arm_warn_task, ao_lco_arm_warn, "lco arm warn");
220         ao_add_task(&ao_lco_igniter_status_task, ao_lco_igniter_status, "lco igniter status");
221         ao_beep_for(AO_BEEP_MID, AO_MS_TO_TICKS(200));
222         for (;;) {
223                 PRINTD(DEBUG_STATUS, "monitor armed %d firing %d\n",
224                        ao_lco_armed, ao_lco_firing);
225
226                 if (ao_lco_armed && ao_lco_firing) {
227                         ao_lco_ignite();
228                 } else {
229                         ao_lco_get_channels();
230                         if (ao_lco_armed) {
231                                 if (ao_lco_selected) {
232                                         PRINTD(DEBUG_STATUS, "Arming pads %x\n",
233                                                ao_lco_selected);
234                                         if (ao_lco_valid & AO_LCO_VALID_EVER) {
235                                                 ao_lco_arm(ao_lco_box, ao_lco_selected, ao_lco_tick_offset);
236                                                 ao_delay(AO_MS_TO_TICKS(10));
237                                         }
238                                 }
239                         }
240                 }
241                 if (ao_lco_armed && ao_lco_firing)
242                         delay = AO_MS_TO_TICKS(100);
243                 else
244                         delay = AO_SEC_TO_TICKS(1);
245                 ao_sleep_for(&ao_lco_armed, delay);
246         }
247 }
248
249 #if DEBUG
250 void
251 ao_lco_set_debug(void)
252 {
253         ao_cmd_decimal();
254         if (ao_cmd_status == ao_cmd_success)
255                 ao_lco_debug = ao_cmd_lex_i;
256 }
257
258 __code struct ao_cmds ao_lco_cmds[] = {
259         { ao_lco_set_debug,     "D <0 off, 1 on>\0Debug" },
260         { 0, NULL }
261 };
262 #endif
263
264 void
265 ao_lco_init(void)
266 {
267         ao_add_task(&ao_lco_monitor_task, ao_lco_monitor, "lco monitor");
268 #if DEBUG
269         ao_cmd_register(&ao_lco_cmds[0]);
270 #endif
271 }