altosui: Add support for downloading TeleMini/TeleNano flight logs
[fw/altos] / altosui / AltosEepromDownload.java
index 9a748710a3fa5063c91f4316544ee08c786c95cf..f96a3dc90527559b685d2c28b94a090a53ba9467 100644 (file)
@@ -33,21 +33,24 @@ import libaltosJNI.*;
 public class AltosEepromDownload implements Runnable {
 
        JFrame                  frame;
-       AltosDevice             device;
        AltosSerial             serial_line;
        boolean                 remote;
        Thread                  eeprom_thread;
        AltosEepromMonitor      monitor;
-       int                     serial = 0;
-       int                     flight = 0;
-       int                     year = 0, month = 0, day = 0;
-       boolean                 want_file = false;
-       FileWriter              eeprom_file = null;
-       LinkedList<String>      eeprom_pending = new LinkedList<String>();
-       AltosConfigData         config_data;
+
+       int                     flight;
+       int                     year, month, day;
+       boolean                 want_file;
+       FileWriter              eeprom_file;
+       LinkedList<String>      eeprom_pending;
+
+       AltosEepromList         flights;
+       ActionListener          listener;
+       boolean                 success;
+       ParseException          parse_exception;
 
        private void FlushPending() throws IOException {
-               for (String s : config_data) {
+               for (String s : flights.config_data) {
                        eeprom_file.write(s);
                        eeprom_file.write('\n');
                }
@@ -62,9 +65,9 @@ public class AltosEepromDownload implements Runnable {
                if (force || (flight != 0 && want_file)) {
                        AltosFile               eeprom_name;
                        if (year != 0 && month != 0 && day != 0)
-                               eeprom_name = new AltosFile(year, month, day, serial, flight, "eeprom");
+                               eeprom_name = new AltosFile(year, month, day, flights.config_data.serial, flight, "eeprom");
                        else
-                               eeprom_name = new AltosFile(serial, flight, "eeprom");
+                               eeprom_name = new AltosFile(flights.config_data.serial, flight, "eeprom");
 
                        eeprom_file = new FileWriter(eeprom_name);
                        if (eeprom_file != null) {
@@ -75,59 +78,146 @@ public class AltosEepromDownload implements Runnable {
                }
        }
 
-       void CaptureLog(int start_block, int end_block) throws IOException, InterruptedException, TimeoutException {
-               int                     block, state_block = 0;
-               int                     state = 0;
-               boolean                 done = false;
-               int                     record;
+       void Log(AltosEepromRecord r) throws IOException {
+               if (r.cmd != Altos.AO_LOG_INVALID) {
+                       String log_line = String.format("%c %4x %4x %4x\n",
+                                                       r.cmd, r.tick, r.a, r.b);
+                       if (eeprom_file != null)
+                               eeprom_file.write(log_line);
+                       else
+                               eeprom_pending.add(log_line);
+               }
+       }
 
-               config_data = new AltosConfigData(serial_line);
-               serial = config_data.serial;
-               if (serial == 0)
-                       throw new IOException("no serial number found");
+       static final int        log_full = 1;
+       static final int        log_tiny = 2;
 
-               monitor.set_serial(serial);
-               /* Now scan the eeprom, reading blocks of data and converting to .eeprom file form */
+       boolean                 done;
+       int                     state;
 
-               state = 0; state_block = start_block;
-               for (block = start_block; !done && block < end_block; block++) {
-                       monitor.set_value(Altos.state_to_string[state], state, block - state_block);
+       void CaptureFull(AltosEepromChunk eechunk) throws IOException {
+               AltosEepromBlock        eeblock = new AltosEepromBlock(eechunk);
 
-                       AltosEepromBlock        eeblock = new AltosEepromBlock(serial_line, block);
-                       if (eeblock.has_flight) {
-                               flight = eeblock.flight;
-                               monitor.set_flight(flight);
-                       }
-                       if (eeblock.has_date) {
-                               year = eeblock.year;
-                               month = eeblock.month;
-                               day = eeblock.day;
+               if (eeblock.has_flight) {
+                       flight = eeblock.flight;
+                       monitor.set_flight(flight);
+               }
+               if (eeblock.has_date) {
+                       year = eeblock.year;
+                       month = eeblock.month;
+                       day = eeblock.day;
+                       want_file = true;
+               }
+
+               if (eeblock.size() == 0 ||
+                   eeblock.has_state && eeblock.state == Altos.ao_flight_landed)
+                       done = true;
+
+               /* Monitor state transitions to update display */
+               if (eeblock.has_state) {
+                       if (eeblock.state > Altos.ao_flight_pad)
                                want_file = true;
-                       }
+                       if (eeblock.state > state)
+                               state = eeblock.state;
+               }
+
+               if (parse_exception == null && eeblock.parse_exception != null)
+                       parse_exception = eeblock.parse_exception;
+
+               CheckFile(false);
+
+               for (int record = 0; record < eeblock.size(); record++)
+                       Log(eeblock.get(record));
+       }
+
+       boolean start;
+       int     tiny_tick;
 
-                       if (eeblock.size() == 0 ||
-                           eeblock.has_state && eeblock.state == Altos.ao_flight_landed)
-                                       done = true;
+       void CaptureTiny (AltosEepromChunk eechunk) throws IOException {
+               boolean some_reasonable_data = false;
 
-                       /* Monitor state transitions to update display */
-                       if (eeblock.has_state) {
-                               if (eeblock.state > Altos.ao_flight_pad)
-                                       want_file = true;
-                               if (eeblock.state > state)
-                                       state = eeblock.state;
+               for (int i = 0; i < eechunk.data.length; i += 2) {
+                       int     v = eechunk.data16(i);
+
+                       if (i == 0 && start) {
+                               tiny_tick = 0;
+                               start = false;
+                               flight = v;
+                               Log(new AltosEepromRecord(Altos.AO_LOG_FLIGHT, tiny_tick, 0, v));
+                               some_reasonable_data = true;
+                       } else {
+                               int     s = v ^ 0x8000;
+                               if (Altos.ao_flight_startup <= s && s <= Altos.ao_flight_invalid) {
+                                       Log(new AltosEepromRecord(Altos.AO_LOG_STATE, tiny_tick, s, 0));
+                                       if (s == Altos.ao_flight_landed) {
+                                               done = true;
+                                               break;
+                                       }
+                                       some_reasonable_data = true;
+                               } else {
+                                       if (v != 0xffff)
+                                               some_reasonable_data = true;
+                                       Log(new AltosEepromRecord(Altos.AO_LOG_HEIGHT, tiny_tick, v, 0));
+                                       if (state < Altos.ao_flight_drogue)
+                                               tiny_tick += 10;
+                                       else
+                                               tiny_tick += 100;
+                               }
                        }
+               }
+               CheckFile(false);
+               if (!some_reasonable_data)
+                       done = true;
+       }
+
+       void CaptureLog(AltosEepromLog log) throws IOException, InterruptedException, TimeoutException {
+               int                     block, state_block = 0;
+               int                     log_style = 0;
 
-                       CheckFile(true);
+               state = 0;
+               done = false;
+               start = true;
 
-                       for (record = 0; record < eeblock.size(); record++) {
-                               AltosEepromRecord r = eeblock.get(record);
+               if (flights.config_data.serial == 0)
+                       throw new IOException("no serial number found");
+
+               /* Reset per-capture variables */
+               flight = 0;
+               year = 0;
+               month = 0;
+               day = 0;
+               want_file = false;
+               eeprom_file = null;
+               eeprom_pending = new LinkedList<String>();
+
+               /* Set serial number in the monitor dialog window */
+               monitor.set_serial(flights.config_data.serial);
+               /* Now scan the eeprom, reading blocks of data and converting to .eeprom file form */
+
+               state = 0; state_block = log.start_block;
+               for (block = log.start_block; !done && block < log.end_block; block++) {
+                       monitor.set_value(Altos.state_to_string[state], state, block - state_block);
+
+                       AltosEepromChunk        eechunk = new AltosEepromChunk(serial_line, block);
 
-                               String log_line = String.format("%c %4x %4x %4x\n",
-                                                               r.cmd, r.tick, r.a, r.b);
-                               if (eeprom_file != null)
-                                       eeprom_file.write(log_line);
+                       /*
+                        * Figure out what kind of data is there
+                        */
+
+                       if (block == log.start_block) {
+                               if (eechunk.data(0) == Altos.AO_LOG_FLIGHT)
+                                       log_style = log_full;
                                else
-                                       eeprom_pending.add(log_line);
+                                       log_style = log_tiny;
+                       }
+
+                       switch (log_style) {
+                       case log_full:
+                               CaptureFull(eechunk);
+                               break;
+                       case log_tiny:
+                               CaptureTiny(eechunk);
+                               break;
                        }
                }
                CheckFile(true);
@@ -137,20 +227,21 @@ public class AltosEepromDownload implements Runnable {
                }
        }
 
-       private void show_error_internal(String message, String title) {
+       private void show_message_internal(String message, String title, int message_type) {
                JOptionPane.showMessageDialog(frame,
                                              message,
                                              title,
-                                             JOptionPane.ERROR_MESSAGE);
+                                             message_type);
        }
 
-       private void show_error(String in_message, String in_title) {
+       private void show_message(String in_message, String in_title, int in_message_type) {
                final String message = in_message;
                final String title = in_title;
+               final int message_type = in_message_type;
                Runnable r = new Runnable() {
                                public void run() {
                                        try {
-                                               show_error_internal(message, title);
+                                               show_message_internal(message, title, message_type);
                                        } catch (Exception ex) {
                                        }
                                }
@@ -158,75 +249,84 @@ public class AltosEepromDownload implements Runnable {
                SwingUtilities.invokeLater(r);
        }
 
-       int     start_block, end_block;
-
        public void run () {
-               if (remote) {
-                       serial_line.set_radio();
-                       serial_line.printf("p\nE 0\n");
-                       serial_line.flush_input();
-               }
+               if (remote)
+                       serial_line.start_remote();
 
                try {
-                       CaptureLog(start_block, end_block);
+                       boolean failed = false;
+                       for (AltosEepromLog log : flights) {
+                               parse_exception = null;
+                               if (log.download) {
+                                       monitor.reset();
+                                       CaptureLog(log);
+                               }
+                               if (parse_exception != null) {
+                                       failed = true;
+                                       show_message(String.format("Flight %d download error\n%s\nValid log data saved",
+                                                                  log.flight,
+                                                                  parse_exception.getMessage()),
+                                                    serial_line.device.toShortString(),
+                                                    JOptionPane.WARNING_MESSAGE);
+                               }
+                       }
+                       success = !failed;
                } catch (IOException ee) {
-                       show_error (device.toShortString(),
-                                   ee.getLocalizedMessage());
+                       show_message(ee.getLocalizedMessage(),
+                                    serial_line.device.toShortString(),
+                                    JOptionPane.ERROR_MESSAGE);
                } catch (InterruptedException ie) {
                } catch (TimeoutException te) {
-                       show_error (String.format("Connection to \"%s\" failed",
-                                                 device.toShortString()),
-                                   "Connection Failed");
+                       show_message(String.format("Connection to \"%s\" failed",
+                                                  serial_line.device.toShortString()),
+                                    "Connection Failed",
+                                    JOptionPane.ERROR_MESSAGE);
                }
                if (remote)
-                       serial_line.printf("~");
+                       serial_line.stop_remote();
                monitor.done();
                serial_line.flush_output();
-               serial_line.close();
+               if (listener != null) {
+                       Runnable r = new Runnable() {
+                                       public void run() {
+                                               try {
+                                                       listener.actionPerformed(new ActionEvent(this,
+                                                                                                success ? 1 : 0,
+                                                                                                "download"));
+                                               } catch (Exception ex) {
+                                               }
+                                       }
+                               };
+                       SwingUtilities.invokeLater(r);
+               }
        }
 
-       public AltosEepromDownload(JFrame given_frame) {
-               frame = given_frame;
-               device = AltosDeviceDialog.show(frame, AltosDevice.product_any);
+       public void start() {
+               eeprom_thread = new Thread(this);
+               eeprom_thread.start();
+       }
 
-               remote = false;
+       public void addActionListener(ActionListener l) {
+               listener = l;
+       }
 
-               if (device != null) {
-                       try {
-                               serial_line = new AltosSerial(device);
-                               if (!device.matchProduct(AltosDevice.product_telemetrum))
-                                       remote = true;
+       public AltosEepromDownload(JFrame given_frame,
+                                  AltosSerial given_serial_line,
+                                  boolean given_remote,
+                                  AltosEepromList given_flights) {
 
-                               monitor = new AltosEepromMonitor(frame, Altos.ao_flight_boost, Altos.ao_flight_landed);
-                               monitor.addActionListener(new ActionListener() {
-                                               public void actionPerformed(ActionEvent e) {
-                                                       if (eeprom_thread != null)
-                                                               eeprom_thread.interrupt();
-                                               }
-                                       });
-
-                               eeprom_thread = new Thread(this);
-                               start_block = 0;
-                               end_block = 0xfff;
-                               eeprom_thread.start();
-                       } catch (FileNotFoundException ee) {
-                               JOptionPane.showMessageDialog(frame,
-                                                             String.format("Cannot open device \"%s\"",
-                                                                           device.toShortString()),
-                                                             "Cannot open target device",
-                                                             JOptionPane.ERROR_MESSAGE);
-                       } catch (AltosSerialInUseException si) {
-                               JOptionPane.showMessageDialog(frame,
-                                                             String.format("Device \"%s\" already in use",
-                                                                           device.toShortString()),
-                                                             "Device in use",
-                                                             JOptionPane.ERROR_MESSAGE);
-                       } catch (IOException ee) {
-                               JOptionPane.showMessageDialog(frame,
-                                                             device.toShortString(),
-                                                             ee.getLocalizedMessage(),
-                                                             JOptionPane.ERROR_MESSAGE);
-                       }
-               }
+               frame = given_frame;
+               serial_line = given_serial_line;
+               remote = given_remote;
+               flights = given_flights;
+               success = false;
+
+               monitor = new AltosEepromMonitor(frame, Altos.ao_flight_boost, Altos.ao_flight_landed);
+               monitor.addActionListener(new ActionListener() {
+                               public void actionPerformed(ActionEvent e) {
+                                       if (eeprom_thread != null)
+                                               eeprom_thread.interrupt();
+                               }
+                       });
        }
 }