altos: Integrate telescience support
[fw/altos] / src / core / ao_log_telescience.c
diff --git a/src/core/ao_log_telescience.c b/src/core/ao_log_telescience.c
new file mode 100644 (file)
index 0000000..aac780f
--- /dev/null
@@ -0,0 +1,274 @@
+/*
+ * Copyright © 2011 Keith Packard <keithp@keithp.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; version 2 of the License.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
+ */
+
+#include "ao.h"
+#include "ao_product.h"
+
+static struct ao_task ao_log_task;
+//static struct ao_task ao_spi_task;
+
+uint8_t        ao_log_running;
+uint8_t                ao_log_mutex;
+uint32_t       ao_log_start_pos;
+uint32_t       ao_log_end_pos;
+uint32_t       ao_log_current_pos;
+
+#define AO_LOG_TELESCIENCE_START       ((uint8_t) 's')
+#define AO_LOG_TELESCIENCE_DATA                ((uint8_t) 'd')
+
+struct ao_log_telescience ao_log_store;
+struct ao_log_telescience ao_log_fetch;
+
+static uint8_t ao_log_adc_pos;
+
+__code uint8_t ao_log_format = AO_LOG_FORMAT_TELESCIENCE;
+
+static uint8_t
+ao_log_csum(__xdata uint8_t *b) __reentrant
+{
+       uint8_t sum = 0x5a;
+       uint8_t i;
+
+       for (i = 0; i < sizeof (struct ao_log_telescience); i++)
+               sum += *b++;
+       return -sum;
+}
+
+static uint8_t
+ao_log_telescience_write(void)
+{
+       uint8_t wrote = 0;
+
+       ao_log_store.csum = 0;
+       ao_log_store.csum = ao_log_csum((__xdata uint8_t *) &ao_log_store);
+       ao_mutex_get(&ao_log_mutex); {
+               if (ao_log_current_pos >= ao_log_end_pos && ao_log_running)
+                       ao_log_stop();
+               if (ao_log_running) {
+                       wrote = 1;
+                       ao_storage_write(ao_log_current_pos,
+                                        (__xdata uint8_t *) &ao_log_store,
+                                        sizeof (struct ao_log_telescience));
+                       ao_log_current_pos += sizeof (struct ao_log_telescience);
+               }
+       } ao_mutex_put(&ao_log_mutex);
+       return wrote;
+}
+
+static uint8_t
+ao_log_valid(struct ao_log_telescience *log)
+{
+       uint8_t *d;
+       uint8_t i;
+       d = (uint8_t *) log;
+       for (i = 0; i < sizeof (struct ao_log_telescience); i++)
+               if (d[i] != 0xff)
+                       return 1;
+       return 0;
+}
+
+static uint8_t
+ao_log_telescience_read(uint32_t pos)
+{
+       if (!ao_storage_read(pos, &ao_log_fetch, sizeof (struct ao_log_telescience)))
+               return 0;
+       return ao_log_valid(&ao_log_fetch);
+}
+
+void
+ao_log_start(void)
+{
+       if (!ao_log_running) {
+               ao_log_running = 1;
+               ao_wakeup(&ao_log_running);
+       }
+}
+
+void
+ao_log_stop(void)
+{
+       if (ao_log_running) {
+               ao_log_running = 0;
+       }
+}
+
+void
+ao_log_restart(void)
+{
+       /* Find end of data */
+       ao_log_end_pos = ao_storage_config;
+       for (ao_log_current_pos = 0;
+            ao_log_current_pos < ao_storage_config;
+            ao_log_current_pos += ao_storage_block)
+       {
+               if (!ao_log_telescience_read(ao_log_current_pos))
+                       break;
+       }
+       if (ao_log_current_pos > 0) {
+               ao_log_current_pos -= ao_storage_block;
+               for (; ao_log_current_pos < ao_storage_config;
+                    ao_log_current_pos += sizeof (struct ao_log_telescience))
+               {
+                       if (!ao_log_telescience_read(ao_log_current_pos))
+                               break;
+               }
+       }
+}
+
+void
+ao_log_telescience(void)
+{
+       ao_storage_setup();
+
+       /* This can take a while, so let the rest
+        * of the system finish booting before we start
+        */
+       ao_delay(AO_SEC_TO_TICKS(10));
+
+       ao_log_restart();
+       for (;;) {
+               while (!ao_log_running)
+                       ao_sleep(&ao_log_running);
+
+               ao_log_start_pos = ao_log_current_pos;
+               ao_log_store.type = AO_LOG_TELESCIENCE_START;
+               ao_log_store.tick = ao_time();
+               ao_log_store.adc[0] = ao_companion_command.serial;
+               ao_log_store.adc[1] = ao_companion_command.flight;
+               ao_log_telescience_write();
+               /* Write the whole contents of the ring to the log
+                * when starting up.
+                */
+               ao_log_adc_pos = ao_adc_ring_next(ao_adc_head);
+               ao_log_store.type = AO_LOG_TELESCIENCE_DATA;
+               while (ao_log_running) {
+                       /* Write samples to EEPROM */
+                       while (ao_log_adc_pos != ao_adc_head) {
+                               ao_log_store.tick = ao_adc_ring[ao_log_adc_pos].tick;
+                               memcpy(&ao_log_store.adc, (void *) ao_adc_ring[ao_log_adc_pos].adc,
+                                      AO_LOG_TELESCIENCE_NUM_ADC * sizeof (uint16_t));
+                               ao_log_telescience_write();
+                               ao_log_adc_pos = ao_adc_ring_next(ao_log_adc_pos);
+                       }
+                       /* Wait for more ADC data to arrive */
+                       ao_sleep((void *) &ao_adc_head);
+               }
+               memset(&ao_log_store.adc, '\0', sizeof (ao_log_store.adc));
+       }
+}
+
+void
+ao_log_set(void)
+{
+       printf("Logging currently %s\n", ao_log_running ? "on" : "off");
+       ao_cmd_hex();
+       if (ao_cmd_status == ao_cmd_success) {
+               if (ao_cmd_lex_i) {
+                       printf("Logging from %ld to %ld\n", ao_log_current_pos, ao_log_end_pos);
+                       ao_log_start();
+               } else {
+                       printf ("Log stopped at %ld\n", ao_log_current_pos);
+                       ao_log_stop();
+               }
+       }
+       ao_cmd_status = ao_cmd_success;
+}
+
+void
+ao_log_list(void)
+{
+       uint32_t        pos;
+       uint32_t        start = 0;
+       uint8_t         flight = 0;
+
+       for (pos = 0; ; pos += sizeof (struct ao_log_telescience)) {
+               if (pos >= ao_storage_config ||
+                   !ao_log_telescience_read(pos) ||
+                   ao_log_fetch.type == AO_LOG_TELESCIENCE_START)
+               {
+                       if (pos != start) {
+                               printf("flight %d start %x end %x\n",
+                                      flight,
+                                      (uint16_t) (start >> 8),
+                                      (uint16_t) ((pos + 0xff) >> 8)); flush();
+                       }
+                       if (ao_log_fetch.type != AO_LOG_TELESCIENCE_START)
+                               break;
+                       start = pos;
+                       flight++;
+               }
+       }
+       printf ("done\n");
+}
+
+void
+ao_log_delete(void)
+{
+       uint32_t        pos;
+
+       ao_cmd_hex();
+       if (ao_cmd_status != ao_cmd_success)
+               return;
+       if (ao_cmd_lex_i != 1) {
+               ao_cmd_status = ao_cmd_syntax_error;
+               printf("No such flight: %d\n", ao_cmd_lex_i);
+               return;
+       }
+       ao_log_stop();
+       for (pos = 0; pos < ao_storage_config; pos += ao_storage_block) {
+               if (!ao_log_telescience_read(pos))
+                       break;
+               ao_storage_erase(pos);
+       }
+       ao_log_current_pos = ao_log_start_pos = 0;
+       if (pos == 0)
+               printf("No such flight: %d\n", ao_cmd_lex_i);
+       else
+               printf ("Erased\n");
+}
+
+static void
+ao_log_query(void)
+{
+       printf("Logging enabled: %d\n", ao_log_running);
+       printf("Log start: %ld\n", ao_log_start_pos);
+       printf("Log cur: %ld\n", ao_log_current_pos);
+       printf("Log end: %ld\n", ao_log_end_pos);
+       printf("log data tick: %04x\n", ao_log_store.tick);
+       printf("TM data tick: %04x\n", ao_log_store.tm_tick);
+       printf("TM state: %d\n", ao_log_store.tm_state);
+       printf("TM serial: %d\n", ao_companion_command.serial);
+       printf("TM flight: %d\n", ao_companion_command.flight);
+}
+
+const struct ao_cmds ao_log_cmds[] = {
+       { ao_log_set,   "L <0 off, 1 on>\0Set logging mode" },
+       { ao_log_list,  "l\0List stored flight logs" },
+       { ao_log_delete, "d 1\0Delete all stored flights" },
+       { ao_log_query, "q\0Query log status" },
+       { 0,    NULL },
+};
+
+void
+ao_log_init(void)
+{
+       ao_log_running = 0;
+
+       ao_cmd_register(&ao_log_cmds[0]);
+
+       ao_add_task(&ao_log_task, ao_log_telescience, "log");
+}