altos/test: Use ao_convert.c instead of hand-coded pres → alt func
[fw/altos] / src / ao_flight_test.c
1 /*
2  * Copyright © 2009 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 #define _GNU_SOURCE
19
20 #include <stdint.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24
25 #define AO_HERTZ        100
26
27 #define AO_ADC_RING     64
28 #define ao_adc_ring_next(n)     (((n) + 1) & (AO_ADC_RING - 1))
29 #define ao_adc_ring_prev(n)     (((n) - 1) & (AO_ADC_RING - 1))
30
31 /*
32  * One set of samples read from the A/D converter
33  */
34 struct ao_adc {
35         uint16_t        tick;           /* tick when the sample was read */
36         int16_t         accel;          /* accelerometer */
37         int16_t         pres;           /* pressure sensor */
38         int16_t         temp;           /* temperature sensor */
39         int16_t         v_batt;         /* battery voltage */
40         int16_t         sense_d;        /* drogue continuity sense */
41         int16_t         sense_m;        /* main continuity sense */
42 };
43
44 #define __pdata
45 #define __data
46 #define __xdata
47 #define __code
48 #define __reentrant
49
50 enum ao_flight_state {
51         ao_flight_startup = 0,
52         ao_flight_idle = 1,
53         ao_flight_pad = 2,
54         ao_flight_boost = 3,
55         ao_flight_fast = 4,
56         ao_flight_coast = 5,
57         ao_flight_drogue = 6,
58         ao_flight_main = 7,
59         ao_flight_landed = 8,
60         ao_flight_invalid = 9
61 };
62
63 struct ao_adc ao_adc_ring[AO_ADC_RING];
64 uint8_t ao_adc_head;
65
66 #define ao_led_on(l)
67 #define ao_led_off(l)
68 #define ao_timer_set_adc_interval(i)
69 #define ao_wakeup(wchan) ao_dump_state()
70 #define ao_cmd_register(c)
71 #define ao_usb_disable()
72 #define ao_telemetry_set_interval(x)
73 #define ao_rdf_set(rdf)
74 #define ao_packet_slave_start()
75
76 enum ao_igniter {
77         ao_igniter_drogue = 0,
78         ao_igniter_main = 1
79 };
80
81 void
82 ao_ignite(enum ao_igniter igniter)
83 {
84         printf ("ignite %s\n", igniter == ao_igniter_drogue ? "drogue" : "main");
85 }
86
87 struct ao_task {
88         int dummy;
89 };
90
91 #define ao_add_task(t,f,n)
92
93 #define ao_log_start()
94 #define ao_log_stop()
95
96 #define AO_MS_TO_TICKS(ms)      ((ms) / 10)
97 #define AO_SEC_TO_TICKS(s)      ((s) * 100)
98
99 #define AO_FLIGHT_TEST
100
101 struct ao_adc ao_adc_static;
102
103 FILE *emulator_in;
104
105 void
106 ao_dump_state(void);
107
108 void
109 ao_sleep(void *wchan);
110
111 const char const * const ao_state_names[] = {
112         "startup", "idle", "pad", "boost", "fast",
113         "coast", "drogue", "main", "landed", "invalid"
114 };
115
116 struct ao_cmds {
117         void            (*func)(void);
118         const char      *help;
119 };
120
121 #include "ao_convert.c"
122
123 struct ao_config {
124         uint16_t        main_deploy;
125         int16_t         accel_plus_g;
126         int16_t         accel_minus_g;
127 };
128
129 #define ao_config_get()
130
131 struct ao_config ao_config;
132
133 #define DATA_TO_XDATA(x) (x)
134
135 #define HAS_FLIGHT 1
136 #define HAS_ADC 1
137 #define HAS_USB 1
138 #define HAS_GPS 1
139 #ifndef HAS_ACCEL
140 #define HAS_ACCEL 1
141 #endif
142 #define HAS_ACCEL_REF 0
143
144 #include "ao_flight.c"
145
146 void
147 ao_insert(void)
148 {
149         ao_adc_ring[ao_adc_head] = ao_adc_static;
150         ao_adc_head = ao_adc_ring_next(ao_adc_head);
151         if (ao_flight_state != ao_flight_startup) {
152                 printf("time %g accel %d pres %d\n",
153                        (double) ao_adc_static.tick / 100,
154                        ao_adc_static.accel,
155                        ao_adc_static.pres);
156         }
157 }
158
159 static int      ao_records_read = 0;
160 static int      ao_eof_read = 0;
161 static int      ao_flight_ground_accel;
162 static int      ao_flight_started = 0;
163
164 void
165 ao_sleep(void *wchan)
166 {
167         ao_dump_state();
168         if (wchan == &ao_adc_head) {
169                 char            type;
170                 uint16_t        tick;
171                 uint16_t        a, b;
172                 int             ret;
173                 char            line[1024];
174                 char            *saveptr;
175                 char            *l;
176                 char            *words[64];
177                 int             nword;
178
179                 for (;;) {
180                         if (ao_records_read > 2 && ao_flight_state == ao_flight_startup)
181                         {
182                                 ao_adc_static.accel = ao_flight_ground_accel;
183                                 ao_insert();
184                                 return;
185                         }
186
187                         if (!fgets(line, sizeof (line), emulator_in)) {
188                                 if (++ao_eof_read >= 1000) {
189                                         printf ("no more data, exiting simulation\n");
190                                         exit(0);
191                                 }
192                                 ao_adc_static.tick += 10;
193                                 ao_insert();
194                                 return;
195                         }
196                         l = line;
197                         for (nword = 0; nword < 64; nword++) {
198                                 words[nword] = strtok_r(l, " \t\n", &saveptr);
199                                 l = NULL;
200                                 if (words[nword] == NULL)
201                                         break;
202                         }
203                         if (nword == 4) {
204                                 type = words[0][0];
205                                 tick = strtoul(words[1], NULL, 16);
206                                 a = strtoul(words[2], NULL, 16);
207                                 b = strtoul(words[3], NULL, 16);
208                         } else if (nword >= 6 && strcmp(words[0], "Accel")) {
209                                 ao_config.accel_plus_g = atoi(words[3]);
210                                 ao_config.accel_minus_g = atoi(words[5]);
211                         } else if (nword >= 4 && strcmp(words[0], "Main")) {
212                                 ao_config.main_deploy = atoi(words[2]);
213                         } else if (nword >= 36 && strcmp(words[0], "CALL") == 0) {
214                                 tick = atoi(words[10]);
215                                 if (!ao_flight_started) {
216                                         type = 'F';
217                                         a = atoi(words[26]);
218                                         ao_flight_started = 1;
219                                 } else {
220                                         type = 'A';
221                                         a = atoi(words[12]);
222                                         b = atoi(words[14]);
223                                 }
224                         }
225                         if (type != 'F' && !ao_flight_started)
226                                 continue;
227
228                         switch (type) {
229                         case 'F':
230                                 ao_flight_ground_accel = a;
231                                 if (ao_config.accel_plus_g == 0) {
232                                         ao_config.accel_plus_g = a;
233                                         ao_config.accel_minus_g = a + 530;
234                                 }
235                                 if (ao_config.main_deploy == 0)
236                                         ao_config.main_deploy = 250;
237                                 ao_flight_started = 1;
238                                 break;
239                         case 'S':
240                                 break;
241                         case 'A':
242                                 ao_adc_static.tick = tick;
243                                 ao_adc_static.accel = a;
244                                 ao_adc_static.pres = b;
245                                 ao_records_read++;
246                                 ao_insert();
247                                 return;
248                         case 'T':
249                                 ao_adc_static.tick = tick;
250                                 ao_adc_static.temp = a;
251                                 ao_adc_static.v_batt = b;
252                                 break;
253                         case 'D':
254                         case 'G':
255                         case 'N':
256                         case 'W':
257                         case 'H':
258                                 break;
259                         }
260                 }
261
262         }
263 }
264 #define COUNTS_PER_G 264.8
265
266 void
267 ao_dump_state(void)
268 {
269         if (ao_flight_state == ao_flight_startup)
270                 return;
271         printf ("\t\t\t\t\t%s accel %g vel %g alt %d main %d\n",
272                 ao_state_names[ao_flight_state],
273                 (ao_ground_accel - ao_flight_accel) / COUNTS_PER_G * GRAVITY,
274                 (double) ao_flight_vel / 100 / COUNTS_PER_G * GRAVITY,
275                 ao_pres_to_altitude(ao_flight_pres) - ao_pres_to_altitude(ao_ground_pres),
276                 ao_pres_to_altitude(ao_main_pres) - ao_pres_to_altitude(ao_ground_pres));
277         if (ao_flight_state == ao_flight_landed)
278                 exit(0);
279 }
280
281 int
282 main (int argc, char **argv)
283 {
284         emulator_in = fopen (argv[1], "r");
285         if (!emulator_in) {
286                 perror(argv[1]);
287                 exit(1);
288         }
289         ao_flight_init();
290         ao_flight();
291 }