src-avr: Control log with pin on companion connector
[fw/altos] / src-avr / ao_log_telescience.c
1 /*
2  * Copyright © 2011 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
20 static struct ao_task ao_log_task;
21
22 uint8_t         ao_log_running;
23 uint8_t         ao_log_mutex;
24 uint32_t        ao_log_start_pos;
25 uint32_t        ao_log_end_pos;
26 uint32_t        ao_log_current_pos;
27
28 #define AO_LOG_TELESCIENCE_START        ((uint8_t) 's')
29 #define AO_LOG_TELESCIENCE_DATA         ((uint8_t) 'd')
30
31 struct ao_log_telescience {
32         uint8_t         type;
33         uint8_t         csum;
34         uint16_t        tick;
35         union {
36                 uint8_t         bytes[28];
37                 uint16_t        adc[NUM_ADC];
38         } u;
39 };
40
41 static struct ao_log_telescience log;
42 static uint8_t  ao_log_adc_pos;
43
44 static uint8_t
45 ao_log_csum(__xdata uint8_t *b) __reentrant
46 {
47         uint8_t sum = 0x5a;
48         uint8_t i;
49
50         for (i = 0; i < sizeof (struct ao_log_telescience); i++)
51                 sum += *b++;
52         return -sum;
53 }
54
55 uint8_t
56 ao_log_telescience_data(struct ao_log_telescience *log)
57 {
58         uint8_t wrote = 0;
59
60         log->csum = 0;
61         log->csum = ao_log_csum((__xdata uint8_t *) log);
62         ao_mutex_get(&ao_log_mutex); {
63                 if (ao_log_current_pos >= ao_log_end_pos && ao_log_running)
64                         ao_log_stop();
65                 if (ao_log_running) {
66                         wrote = 1;
67                         ao_storage_write(ao_log_current_pos,
68                                          log,
69                                          sizeof (struct ao_log_telescience));
70                         ao_log_current_pos += sizeof (struct ao_log_telescience);
71                 }
72         } ao_mutex_put(&ao_log_mutex);
73         return wrote;
74 }
75
76 static uint8_t
77 ao_log_valid(struct ao_log_telescience *log)
78 {
79         uint8_t *d;
80         uint8_t i;
81         d = (uint8_t *) log;
82         for (i = 0; i < sizeof (struct ao_log_telescience); i++)
83                 if (d[i] != 0xff)
84                         return 1;
85         return 0;
86 }
87
88 void
89 ao_log_start(void)
90 {
91         printf("Log goes from %ld to %ld\n", ao_log_current_pos, ao_log_end_pos);
92         ao_log_running = 1;
93         ao_wakeup(&ao_log_running);
94 }
95
96 void
97 ao_log_stop(void)
98 {
99         printf ("Log stopped at %ld\n", ao_log_current_pos);
100         ao_log_running = 0;
101         ao_wakeup((void *) &ao_adc_head);
102 }
103
104 void
105 ao_log_check_pin(void)
106 {
107         if (PINB & (1 << PINB0))
108                 ao_log_stop();
109         else
110                 ao_log_start();
111 }
112
113 void
114 ao_log_telescience(void)
115 {
116         ao_log_check_pin();
117
118         ao_storage_setup();
119
120         /* Find end of data */
121         while (ao_log_start_pos < ao_log_end_pos) {
122                 if (!(ao_storage_read(ao_log_start_pos, &log, sizeof (struct ao_log_telescience))))
123                         break;
124                 if (!ao_log_valid(&log))
125                         break;
126         }
127         ao_log_current_pos = ao_log_start_pos;
128         ao_log_end_pos = ao_storage_config;
129         for (;;) {
130                 while (!ao_log_running)
131                         ao_sleep(&ao_log_running);
132
133                 memset(&log, '\0', sizeof (struct ao_log_telescience));
134                 log.type = AO_LOG_TELESCIENCE_START;
135                 log.tick = ao_time();
136                 ao_log_telescience_data(&log);
137                 /* Write the whole contents of the ring to the log
138                  * when starting up.
139                  */
140                 ao_log_adc_pos = ao_adc_ring_next(ao_adc_head);
141                 log.type = AO_LOG_TELESCIENCE_DATA;
142                 while (ao_log_running) {
143                         /* Write samples to EEPROM */
144                         while (ao_log_adc_pos != ao_adc_head) {
145                                 log.tick = ao_adc_ring[ao_log_adc_pos].tick;
146                                 memcpy(&log.u.adc, (void *) ao_adc_ring[ao_log_adc_pos].adc,
147                                        NUM_ADC * sizeof (uint16_t));
148                                 ao_log_telescience_data(&log);
149                                 ao_log_adc_pos = ao_adc_ring_next(ao_log_adc_pos);
150                         }
151                         /* Wait for more ADC data to arrive */
152                         ao_sleep((void *) &ao_adc_head);
153                 }
154         }
155 }
156
157 void
158 ao_log_set(void)
159 {
160         ao_cmd_hex();
161         if (ao_cmd_status == ao_cmd_success) {
162                 if (ao_cmd_lex_i)
163                         ao_log_start();
164                 else
165                         ao_log_stop();
166         }
167 }
168
169 void
170 ao_log_list(void)
171 {
172         uint32_t        pos;
173         uint32_t        start = 0;
174         uint8_t         flight = 0;
175
176         for (pos = 0; pos < ao_storage_config; pos += sizeof (struct ao_log_telescience)) {
177                 if (!ao_storage_read(pos, &log, sizeof (struct ao_log_telescience)))
178                         break;
179                 if (!ao_log_valid(&log) || log.type == AO_LOG_TELESCIENCE_START) {
180                         if (pos != start) {
181                                 printf("flight %d start %x end %x\n",
182                                        flight,
183                                        (uint16_t) (start >> 8),
184                                        (uint16_t) ((pos + 0xff) >> 8));
185                         }
186                         if (!ao_log_valid(&log))
187                                 break;
188                         start = pos;
189                         flight++;
190                 }
191         }
192         printf ("done\n");
193 }
194
195 void
196 ao_log_delete(void)
197 {
198         uint32_t        pos;
199
200         ao_cmd_hex();
201         if (ao_cmd_status != ao_cmd_success)
202                 return;
203         if (ao_cmd_lex_i != 1) {
204                 ao_cmd_status = ao_cmd_syntax_error;
205                 printf("No such flight: %d\n", ao_cmd_lex_i);
206                 return;
207         }
208         for (pos = 0; pos < ao_storage_config; pos += ao_storage_block) {
209                 if (!ao_storage_read(pos, &log, sizeof (struct ao_log_telescience)))
210                         break;
211                 if (!ao_log_valid(&log))
212                         break;
213                 ao_storage_erase(pos);
214         }
215         if (pos == 0)
216                 printf("No such flight: %d\n", ao_cmd_lex_i);
217         else
218                 printf ("Erased\n");
219 }
220
221 const struct ao_cmds ao_log_cmds[] = {
222         { ao_log_set,   "L <0 off, 1 on>\0Set logging mode" },
223         { ao_log_list,  "l\0List stored flight logs" },
224         { ao_log_delete, "d 1\0Delete all stored flights" },
225         { 0,    NULL },
226 };
227
228 ISR(PCINT0_vect)
229 {
230         ao_log_check_pin();
231 }
232
233 void
234 ao_log_init(void)
235 {
236         ao_log_running = 0;
237
238         PCMSK0 |= (1 << PCINT0);        /* Enable PCINT0 pin change */
239
240         PCICR |= (1 << PCIE0);          /* Enable pin change interrupt */
241
242         PORTB &= ~(1 << PORTB0);        /* Pull input down; always log if NC */
243
244         ao_cmd_register(&ao_log_cmds[0]);
245
246         ao_add_task(&ao_log_task, ao_log_telescience, "log");
247 }