2 * Copyright © 2012 Keith Packard <keithp@keithp.com>
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.
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.
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.
19 #ifndef AO_FLIGHT_TEST
21 #include <ao_sample.h>
22 #include <ao_flight.h>
27 #include <ao_companion.h>
28 #define ao_accel ao_companion_command.accel
29 #define ao_speed ao_companion_command.speed
30 #define ao_height ao_companion_command.height
31 #define ao_flight_state ao_companion_command.flight_state
32 #define ao_motor_number ao_companion_command.motor_number
35 #define ao_lowbit(x) ((x) & (-x))
37 #ifndef AO_FLIGHT_TEST
38 enum ao_igniter_status
39 ao_pyro_status(uint8_t p)
41 struct ao_data packet;
48 value = (AO_IGNITER_CLOSED>>1);
49 value = AO_SENSE_PYRO(&packet, p);
50 if (value < AO_IGNITER_OPEN)
51 return ao_igniter_open;
52 else if (value > AO_IGNITER_CLOSED)
53 return ao_igniter_ready;
55 return ao_igniter_unknown;
59 ao_pyro_print_status(void)
63 for(p = 0; p < AO_PYRO_NUM; p++) {
64 enum ao_igniter_status status = ao_pyro_status(p);
65 printf("Igniter: %d Status: %s\n",
66 p, ao_igniter_status_names[status]);
71 uint16_t ao_pyro_fired;
79 #define DBG(...) do { if (pyro_dbg) { printf("\t%d: ", (int) (pyro - ao_config.pyro)); printf(__VA_ARGS__); } } while (0)
85 ao_sample_max_orient(void)
88 angle_t max = ao_sample_orients[0];
90 for (i = 1; i < AO_NUM_ORIENT; i++) {
91 angle_t a = ao_sample_orients[i];
98 * Given a pyro structure, figure out
99 * if the current flight state satisfies all
100 * of the requirements
103 ao_pyro_ready(struct ao_pyro *pyro)
105 enum ao_pyro_flag flag, flags;
111 while (flags != ao_pyro_none) {
112 flag = ao_lowbit(flags);
116 case ao_pyro_accel_less:
117 if (ao_accel <= pyro->accel_less)
119 DBG("accel %d > %d\n", ao_accel, pyro->accel_less);
121 case ao_pyro_accel_greater:
122 if (ao_accel >= pyro->accel_greater)
124 DBG("accel %d < %d\n", ao_accel, pyro->accel_greater);
126 case ao_pyro_speed_less:
127 if (ao_speed <= pyro->speed_less)
129 DBG("speed %d > %d\n", ao_speed, pyro->speed_less);
131 case ao_pyro_speed_greater:
132 if (ao_speed >= pyro->speed_greater)
134 DBG("speed %d < %d\n", ao_speed, pyro->speed_greater);
136 case ao_pyro_height_less:
137 if (ao_height <= pyro->height_less)
139 DBG("height %d > %d\n", ao_height, pyro->height_less);
141 case ao_pyro_height_greater:
142 if (ao_height >= pyro->height_greater)
144 DBG("height %d < %d\n", ao_height, pyro->height_greater);
148 case ao_pyro_orient_less:
149 max_orient = ao_sample_max_orient();
150 if (max_orient <= pyro->orient_less)
152 DBG("orient %d > %d\n", max_orient, pyro->orient_less);
154 case ao_pyro_orient_greater:
155 max_orient = ao_sample_max_orient();
156 if (max_orient >= pyro->orient_greater)
158 DBG("orient %d < %d\n", max_orient, pyro->orient_greater);
162 case ao_pyro_time_less:
163 if ((int16_t) (ao_time() - ao_launch_tick) <= pyro->time_less)
165 DBG("time %d > %d\n", (int16_t)(ao_time() - ao_launch_tick), pyro->time_less);
167 case ao_pyro_time_greater:
168 if ((int16_t) (ao_time() - ao_launch_tick) >= pyro->time_greater)
170 DBG("time %d < %d\n", (int16_t)(ao_time() - ao_launch_tick), pyro->time_greater);
173 case ao_pyro_ascending:
176 DBG("not ascending speed %d\n", ao_speed);
178 case ao_pyro_descending:
181 DBG("not descending speed %d\n", ao_speed);
184 case ao_pyro_after_motor:
185 if (ao_motor_number >= pyro->motor)
187 DBG("motor %d != %d\n", ao_motor_number, pyro->motor);
191 /* handled separately */
194 case ao_pyro_state_less:
195 if (ao_flight_state < pyro->state_less)
197 DBG("state %d >= %d\n", ao_flight_state, pyro->state_less);
199 case ao_pyro_state_greater_or_equal:
200 if (ao_flight_state >= pyro->state_greater_or_equal)
202 DBG("state %d < %d\n", ao_flight_state, pyro->state_greater_or_equal);
213 #ifndef AO_FLIGHT_TEST
215 ao_pyro_pin_set(uint8_t p, uint8_t v)
219 case 0: ao_gpio_set(AO_PYRO_PORT_0, AO_PYRO_PIN_0, v); break;
222 case 1: ao_gpio_set(AO_PYRO_PORT_1, AO_PYRO_PIN_1, v); break;
225 case 2: ao_gpio_set(AO_PYRO_PORT_2, AO_PYRO_PIN_2, v); break;
228 case 3: ao_gpio_set(AO_PYRO_PORT_3, AO_PYRO_PIN_3, v); break;
231 case 4: ao_gpio_set(AO_PYRO_PORT_4, AO_PYRO_PIN_4, v); break;
234 case 5: ao_gpio_set(AO_PYRO_PORT_5, AO_PYRO_PIN_5, v); break;
237 case 6: ao_gpio_set(AO_PYRO_PORT_6, AO_PYRO_PIN_6, v); break;
240 case 7: ao_gpio_set(AO_PYRO_PORT_7, AO_PYRO_PIN_7, v); break;
247 uint8_t ao_pyro_wakeup;
250 ao_pyro_pins_fire(uint16_t fire)
254 for (p = 0; p < AO_PYRO_NUM; p++) {
256 ao_pyro_pin_set(p, 1);
258 ao_delay(ao_config.pyro_time);
259 for (p = 0; p < AO_PYRO_NUM; p++) {
261 ao_pyro_pin_set(p, 0);
263 ao_delay(AO_MS_TO_TICKS(50));
269 struct ao_pyro *pyro;
270 uint8_t p, any_waiting;
274 for (p = 0; p < AO_PYRO_NUM; p++) {
275 pyro = &ao_config.pyro[p];
277 /* Ignore igniters which have already fired
279 if (ao_pyro_fired & (1 << p))
282 /* Ignore disabled igniters
288 /* Check pyro state to see if it should fire
290 if (!pyro->delay_done) {
291 if (!ao_pyro_ready(pyro))
294 /* If there's a delay set, then remember when
297 if (pyro->flags & ao_pyro_delay) {
298 pyro->delay_done = ao_time() + pyro->delay;
299 if (!pyro->delay_done)
300 pyro->delay_done = 1;
304 /* Check to see if we're just waiting for
305 * the delay to expire
307 if (pyro->delay_done) {
309 /* Check to make sure the required conditions
310 * remain valid. If not, inhibit the channel
311 * by setting the fired bit
313 if (!ao_pyro_ready(pyro)) {
314 ao_pyro_fired |= (1 << p);
318 if ((int16_t) (ao_time() - pyro->delay_done) < 0)
326 ao_pyro_fired |= fire;
327 ao_pyro_pins_fire(fire);
333 #define NO_VALUE 0xff
335 #define AO_PYRO_NAME_LEN 4
338 #define ENABLE_HELP 1
348 char name[AO_PYRO_NAME_LEN];
349 enum ao_pyro_flag flag;
354 } ao_pyro_values[] = {
355 { "a<", ao_pyro_accel_less, offsetof(struct ao_pyro, accel_less), HELP("accel less (m/ss * 16)") },
356 { "a>", ao_pyro_accel_greater, offsetof(struct ao_pyro, accel_greater), HELP("accel greater (m/ss * 16)") },
358 { "s<", ao_pyro_speed_less, offsetof(struct ao_pyro, speed_less), HELP("speed less (m/s * 16)") },
359 { "s>", ao_pyro_speed_greater, offsetof(struct ao_pyro, speed_greater), HELP("speed greater (m/s * 16)") },
361 { "h<", ao_pyro_height_less, offsetof(struct ao_pyro, height_less), HELP("height less (m)") },
362 { "h>", ao_pyro_height_greater, offsetof(struct ao_pyro, height_greater), HELP("height greater (m)") },
365 { "o<", ao_pyro_orient_less, offsetof(struct ao_pyro, orient_less), HELP("orient less (deg)") },
366 { "o>", ao_pyro_orient_greater, offsetof(struct ao_pyro, orient_greater), HELP("orient greater (deg)") },
369 { "t<", ao_pyro_time_less, offsetof(struct ao_pyro, time_less), HELP("time less (s * 100)") },
370 { "t>", ao_pyro_time_greater, offsetof(struct ao_pyro, time_greater), HELP("time greater (s * 100)") },
372 { "f<", ao_pyro_state_less, offsetof(struct ao_pyro, state_less), HELP("state less") },
373 { "f>=",ao_pyro_state_greater_or_equal, offsetof(struct ao_pyro, state_greater_or_equal), HELP("state greater or equal") },
375 { "A", ao_pyro_ascending, NO_VALUE, HELP("ascending") },
376 { "D", ao_pyro_descending, NO_VALUE, HELP("descending") },
378 { "m", ao_pyro_after_motor, offsetof(struct ao_pyro, motor), HELP("after motor") },
380 { "d", ao_pyro_delay, offsetof(struct ao_pyro, delay), HELP("delay before firing (s * 100)") },
381 { "", ao_pyro_none, NO_VALUE, HELP(NULL) },
384 #define NUM_PYRO_VALUES (sizeof ao_pyro_values / sizeof ao_pyro_values[0])
386 #ifndef AO_FLIGHT_TEST
393 while (ao_flight_state < ao_flight_boost)
394 ao_sleep(&ao_flight_state);
397 ao_sleep_for(&ao_pyro_wakeup, AO_MS_TO_TICKS(100));
398 if (ao_flight_state >= ao_flight_landed)
400 any_waiting = ao_pyro_check();
407 struct ao_task ao_pyro_task;
411 ao_pyro_print_name(uint8_t v)
413 const char *s = ao_pyro_values[v].name;
414 printf ("%s%s", s, " " + strlen(s));
422 for (v = 0; ao_pyro_values[v].flag != ao_pyro_none; v++) {
423 ao_pyro_print_name(v);
424 if (ao_pyro_values[v].offset != NO_VALUE)
428 printf ("%s\n", ao_pyro_values[v].help);
438 struct ao_pyro *pyro;
440 printf ("Pyro-count: %d\n", AO_PYRO_NUM);
441 for (p = 0; p < AO_PYRO_NUM; p++) {
442 printf ("Pyro %2d: ", p);
443 pyro = &ao_config.pyro[p];
445 printf ("<disabled>\n");
448 for (v = 0; ao_pyro_values[v].flag != ao_pyro_none; v++) {
449 if (!(pyro->flags & ao_pyro_values[v].flag))
451 ao_pyro_print_name(v);
452 if (ao_pyro_values[v].offset != NO_VALUE) {
455 if (ao_pyro_values[v].flag & AO_PYRO_8_BIT_VALUE)
456 value = *((uint8_t *) ((char *) pyro + ao_pyro_values[v].offset));
458 value = *((int16_t *) (void *) ((char *) pyro + ao_pyro_values[v].offset));
459 printf ("%6d ", value);
472 struct ao_pyro pyro_tmp;
473 char name[AO_PYRO_NAME_LEN];
480 switch (ao_cmd_lex_c) {
487 p = ao_cmd_decimal();
488 if (ao_cmd_status != ao_cmd_success)
490 if (AO_PYRO_NUM <= p) {
491 printf ("invalid pyro channel %d\n", p);
494 memset(&pyro_tmp, '\0', sizeof (pyro_tmp));
497 if (ao_cmd_lex_c == '\n')
500 for (c = 0; c < AO_PYRO_NAME_LEN - 1; c++) {
501 if (ao_cmd_is_white() || ao_cmd_lex_c == '\n')
503 name[c] = ao_cmd_lex_c;
507 for (v = 0; ao_pyro_values[v].flag != ao_pyro_none; v++) {
508 if (!strcmp (ao_pyro_values[v].name, name))
511 if (ao_pyro_values[v].flag == ao_pyro_none) {
512 printf ("invalid pyro field %s\n", name);
513 ao_cmd_status = ao_cmd_syntax_error;
516 pyro_tmp.flags |= ao_pyro_values[v].flag;
517 if (ao_pyro_values[v].offset != NO_VALUE) {
520 if (ao_cmd_lex_c == '-') {
524 r *= ao_cmd_decimal();
525 if (ao_cmd_status != ao_cmd_success)
527 if (ao_pyro_values[v].flag & AO_PYRO_8_BIT_VALUE) {
529 ao_cmd_status = ao_cmd_syntax_error;
532 *((uint8_t *) ((char *) &pyro_tmp + ao_pyro_values[v].offset)) = r;
534 *((int16_t *) (void *) ((char *) &pyro_tmp + ao_pyro_values[v].offset)) = r;
538 _ao_config_edit_start();
539 ao_config.pyro[p] = pyro_tmp;
540 _ao_config_edit_finish();
544 ao_pyro_manual(uint8_t p)
546 printf ("ao_pyro_manual %d\n", p);
547 if (p >= AO_PYRO_NUM) {
548 ao_cmd_status = ao_cmd_syntax_error;
551 ao_pyro_pins_fire(1 << p);
558 ao_enable_output(AO_PYRO_PORT_0, AO_PYRO_PIN_0, 0);
561 ao_enable_output(AO_PYRO_PORT_1, AO_PYRO_PIN_1, 0);
564 ao_enable_output(AO_PYRO_PORT_2, AO_PYRO_PIN_2, 0);
567 ao_enable_output(AO_PYRO_PORT_3, AO_PYRO_PIN_3, 0);
570 ao_enable_output(AO_PYRO_PORT_4, AO_PYRO_PIN_4, 0);
573 ao_enable_output(AO_PYRO_PORT_5, AO_PYRO_PIN_5, 0);
576 ao_enable_output(AO_PYRO_PORT_6, AO_PYRO_PIN_6, 0);
579 ao_enable_output(AO_PYRO_PORT_7, AO_PYRO_PIN_7, 0);
581 ao_add_task(&ao_pyro_task, ao_pyro, "pyro");