git.gag.com Git - fw/altos/blob - altosui/AltosEepromDownload.java

   1 /*
   2  * Copyright © 2010 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 package altosui;
  19
  20 import java.awt.*;
  21 import java.awt.event.*;
  22 import javax.swing.*;
  23 import javax.swing.filechooser.FileNameExtensionFilter;
  24 import javax.swing.table.*;
  25 import java.io.*;
  26 import java.util.*;
  27 import java.text.*;
  28 import java.util.prefs.*;
  29 import java.util.concurrent.*;
  30
  31 import libaltosJNI.*;
  32
  33 public class AltosEepromDownload implements Runnable {
  34
  35         JFrame                  frame;
  36         AltosSerial             serial_line;
  37         boolean                 remote;
  38         Thread                  eeprom_thread;
  39         AltosEepromMonitor      monitor;
  40
  41         int                     flight;
  42         int                     year, month, day;
  43         boolean                 want_file;
  44         FileWriter              eeprom_file;
  45         LinkedList<String>      eeprom_pending;
  46
  47         AltosEepromList         flights;
  48         ActionListener          listener;
  49         boolean                 success;
  50         ParseException          parse_exception;
  51
  52         private void FlushPending() throws IOException {
  53                 for (String s : flights.config_data) {
  54                         eeprom_file.write(s);
  55                         eeprom_file.write('\n');
  56                 }
  57
  58                 for (String s : eeprom_pending)
  59                         eeprom_file.write(s);
  60         }
  61
  62         private void CheckFile(boolean force) throws IOException {
  63                 if (eeprom_file != null)
  64                         return;
  65                 if (force || (flight != 0 && want_file)) {
  66                         AltosFile               eeprom_name;
  67                         if (year != 0 && month != 0 && day != 0)
  68                                 eeprom_name = new AltosFile(year, month, day, flights.config_data.serial, flight, "eeprom");
  69                         else
  70                                 eeprom_name = new AltosFile(flights.config_data.serial, flight, "eeprom");
  71
  72                         eeprom_file = new FileWriter(eeprom_name);
  73                         if (eeprom_file != null) {
  74                                 monitor.set_file(eeprom_name.getName());
  75                                 FlushPending();
  76                                 eeprom_pending = null;
  77                         }
  78                 }
  79         }
  80
  81         void Log(AltosEepromRecord r) throws IOException {
  82                 if (r.cmd != Altos.AO_LOG_INVALID) {
  83                         String log_line = String.format("%c %4x %4x %4x\n",
  84                                                         r.cmd, r.tick, r.a, r.b);
  85                         if (eeprom_file != null)
  86                                 eeprom_file.write(log_line);
  87                         else
  88                                 eeprom_pending.add(log_line);
  89                 }
  90         }
  91
  92         static final int        log_full = 1;
  93         static final int        log_tiny = 2;
  94
  95         boolean                 done;
  96         int                     state;
  97
  98         void CaptureFull(AltosEepromChunk eechunk) throws IOException {
  99                 boolean any_valid = false;
 100                 for (int i = 0; i < eechunk.chunk_size && !done; i += AltosEepromRecord.record_length) {
 101                         try {
 102                                 AltosEepromRecord r = new AltosEepromRecord(eechunk, i);
 103                                 if (r.cmd == Altos.AO_LOG_FLIGHT) {
 104                                         flight = r.b;
 105                                         monitor.set_flight(flight);
 106                                 }
 107
 108                                 /* Monitor state transitions to update display */
 109                                 if (r.cmd == Altos.AO_LOG_STATE && r.a <= Altos.ao_flight_landed) {
 110                                         state = r.a;
 111                                         if (state > Altos.ao_flight_pad)
 112                                                 want_file = true;
 113                                 }
 114
 115                                 if (r.cmd == Altos.AO_LOG_GPS_DATE) {
 116                                         year = 2000 + (r.a & 0xff);
 117                                         month = (r.a >> 8) & 0xff;
 118                                         day = (r.b & 0xff);
 119                                         want_file = true;
 120                                 }
 121                                 if (r.cmd == Altos.AO_LOG_STATE && r.a == Altos.ao_flight_landed)
 122                                         done = true;
 123                                 any_valid = true;
 124                                 Log(r);
 125                         } catch (ParseException pe) {
 126                                 if (parse_exception == null)
 127                                         parse_exception = pe;
 128                         }
 129                 }
 130
 131                 if (!any_valid)
 132                         done = true;
 133
 134                 CheckFile(false);
 135         }
 136
 137         boolean start;
 138         int     tiny_tick;
 139
 140         void CaptureTiny (AltosEepromChunk eechunk) throws IOException {
 141                 boolean any_valid = false;
 142
 143                 for (int i = 0; i < eechunk.data.length && !done; i += 2) {
 144                         int                     v = eechunk.data16(i);
 145                         AltosEepromRecord       r;
 146
 147                         if (i == 0 && start) {
 148                                 tiny_tick = 0;
 149                                 start = false;
 150                                 flight = v;
 151                                 monitor.set_flight(flight);
 152                                 r = new AltosEepromRecord(Altos.AO_LOG_FLIGHT, tiny_tick, 0, v);
 153                                 any_valid = true;
 154                         } else {
 155                                 int     s = v ^ 0x8000;
 156
 157                                 if (Altos.ao_flight_startup <= s && s <= Altos.ao_flight_invalid) {
 158                                         state = s;
 159                                         r = new AltosEepromRecord(Altos.AO_LOG_STATE, tiny_tick, state, 0);
 160                                         if (state == Altos.ao_flight_landed)
 161                                                 done = true;
 162                                         state = s;
 163                                         any_valid = true;
 164                                 } else {
 165                                         if (v != 0xffff)
 166                                                 any_valid = true;
 167
 168                                         r = new AltosEepromRecord(Altos.AO_LOG_PRESSURE, tiny_tick, 0, v);
 169
 170                                         /*
 171                                          * The flight software records ascent data every 100ms, and descent
 172                                          * data every 1s.
 173                                          */
 174                                         if (state < Altos.ao_flight_drogue)
 175                                                 tiny_tick += 10;
 176                                         else
 177                                                 tiny_tick += 100;
 178                                 }
 179                         }
 180                         Log(r);
 181                 }
 182                 CheckFile(false);
 183                 if (!any_valid)
 184                         done = true;
 185         }
 186
 187         void CaptureLog(AltosEepromLog log) throws IOException, InterruptedException, TimeoutException {
 188                 int                     block, state_block = 0;
 189                 int                     log_style = 0;
 190
 191                 state = 0;
 192                 done = false;
 193                 start = true;
 194
 195                 if (flights.config_data.serial == 0)
 196                         throw new IOException("no serial number found");
 197
 198                 /* Reset per-capture variables */
 199                 flight = 0;
 200                 year = 0;
 201                 month = 0;
 202                 day = 0;
 203                 want_file = false;
 204                 eeprom_file = null;
 205                 eeprom_pending = new LinkedList<String>();
 206
 207                 /* Set serial number in the monitor dialog window */
 208                 monitor.set_serial(flights.config_data.serial);
 209                 /* Now scan the eeprom, reading blocks of data and converting to .eeprom file form */
 210
 211                 state = 0; state_block = log.start_block;
 212                 for (block = log.start_block; !done && block < log.end_block; block++) {
 213                         monitor.set_value(Altos.state_to_string[state], state, block - state_block);
 214
 215                         AltosEepromChunk        eechunk = new AltosEepromChunk(serial_line, block);
 216
 217                         /*
 218                          * Figure out what kind of data is there
 219                          */
 220
 221                         if (block == log.start_block) {
 222                                 if (eechunk.data(0) == Altos.AO_LOG_FLIGHT)
 223                                         log_style = log_full;
 224                                 else
 225                                         log_style = log_tiny;
 226                         }
 227
 228                         switch (log_style) {
 229                         case log_full:
 230                                 CaptureFull(eechunk);
 231                                 break;
 232                         case log_tiny:
 233                                 CaptureTiny(eechunk);
 234                                 break;
 235                         }
 236                 }
 237                 CheckFile(true);
 238                 if (eeprom_file != null) {
 239                         eeprom_file.flush();
 240                         eeprom_file.close();
 241                 }
 242         }
 243
 244         private void show_message_internal(String message, String title, int message_type) {
 245                 JOptionPane.showMessageDialog(frame,
 246                                               message,
 247                                               title,
 248                                               message_type);
 249         }
 250
 251         private void show_message(String in_message, String in_title, int in_message_type) {
 252                 final String message = in_message;
 253                 final String title = in_title;
 254                 final int message_type = in_message_type;
 255                 Runnable r = new Runnable() {
 256                                 public void run() {
 257                                         try {
 258                                                 show_message_internal(message, title, message_type);
 259                                         } catch (Exception ex) {
 260                                         }
 261                                 }
 262                         };
 263                 SwingUtilities.invokeLater(r);
 264         }
 265
 266         public void run () {
 267                 if (remote)
 268                         serial_line.start_remote();
 269
 270                 try {
 271                         boolean failed = false;
 272                         for (AltosEepromLog log : flights) {
 273                                 parse_exception = null;
 274                                 if (log.download) {
 275                                         monitor.reset();
 276                                         CaptureLog(log);
 277                                 }
 278                                 if (parse_exception != null) {
 279                                         failed = true;
 280                                         show_message(String.format("Flight %d download error\n%s\nValid log data saved",
 281                                                                    log.flight,
 282                                                                    parse_exception.getMessage()),
 283                                                      serial_line.device.toShortString(),
 284                                                      JOptionPane.WARNING_MESSAGE);
 285                                 }
 286                         }
 287                         success = !failed;
 288                 } catch (IOException ee) {
 289                         show_message(ee.getLocalizedMessage(),
 290                                      serial_line.device.toShortString(),
 291                                      JOptionPane.ERROR_MESSAGE);
 292                 } catch (InterruptedException ie) {
 293                 } catch (TimeoutException te) {
 294                         show_message(String.format("Connection to \"%s\" failed",
 295                                                    serial_line.device.toShortString()),
 296                                      "Connection Failed",
 297                                      JOptionPane.ERROR_MESSAGE);
 298                 }
 299                 if (remote)
 300                         serial_line.stop_remote();
 301                 monitor.done();
 302                 serial_line.flush_output();
 303                 if (listener != null) {
 304                         Runnable r = new Runnable() {
 305                                         public void run() {
 306                                                 try {
 307                                                         listener.actionPerformed(new ActionEvent(this,
 308                                                                                                  success ? 1 : 0,
 309                                                                                                  "download"));
 310                                                 } catch (Exception ex) {
 311                                                 }
 312                                         }
 313                                 };
 314                         SwingUtilities.invokeLater(r);
 315                 }
 316         }
 317
 318         public void start() {
 319                 eeprom_thread = new Thread(this);
 320                 eeprom_thread.start();
 321         }
 322
 323         public void addActionListener(ActionListener l) {
 324                 listener = l;
 325         }
 326
 327         public AltosEepromDownload(JFrame given_frame,
 328                                    AltosSerial given_serial_line,
 329                                    boolean given_remote,
 330                                    AltosEepromList given_flights) {
 331
 332                 frame = given_frame;
 333                 serial_line = given_serial_line;
 334                 serial_line.set_frame(frame);
 335                 remote = given_remote;
 336                 flights = given_flights;
 337                 success = false;
 338
 339                 monitor = new AltosEepromMonitor(frame, Altos.ao_flight_boost, Altos.ao_flight_landed);
 340                 monitor.addActionListener(new ActionListener() {
 341                                 public void actionPerformed(ActionEvent e) {
 342                                         if (eeprom_thread != null)
 343                                                 eeprom_thread.interrupt();
 344                                 }
 345                         });
 346         }
 347 }