git.gag.com Git - fw/altos/blob - teststand/TestStand.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; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
  17  */
  18
  19 package teststand;
  20
  21 import java.awt.*;
  22 import java.awt.event.*;
  23 import javax.swing.*;
  24 import java.io.*;
  25 import java.util.concurrent.*;
  26 import org.altusmetrum.altoslib_12.*;
  27 import org.altusmetrum.altosuilib_12.*;
  28
  29 public class TestStand extends AltosUIFrame implements AltosEepromGrapher {
  30         public AltosVoice voice = new AltosVoice();
  31
  32         public static boolean load_library(Frame frame) {
  33                 if (!Altos.load_library()) {
  34                         JOptionPane.showMessageDialog(frame,
  35                                                       String.format("No AltOS library in \"%s\"",
  36                                                                     System.getProperty("java.library.path","<undefined>")),
  37                                                       "Cannot load device access library",
  38                                                       JOptionPane.ERROR_MESSAGE);
  39                         return false;
  40                 }
  41                 return true;
  42         }
  43
  44         void telemetry_window(AltosDevice device) {
  45                 try {
  46                         AltosFlightReader reader = new AltosTelemetryReader(new AltosSerial(device));
  47                         if (reader != null)
  48                                 new AltosFlightUI(voice, reader, device.getSerial());
  49                 } catch (FileNotFoundException ee) {
  50                         JOptionPane.showMessageDialog(TestStand.this,
  51                                                       ee.getMessage(),
  52                                                       String.format ("Cannot open %s", device.toShortString()),
  53                                                       JOptionPane.ERROR_MESSAGE);
  54                 } catch (AltosSerialInUseException si) {
  55                         JOptionPane.showMessageDialog(TestStand.this,
  56                                                       String.format("Device \"%s\" already in use",
  57                                                                     device.toShortString()),
  58                                                       "Device in use",
  59                                                       JOptionPane.ERROR_MESSAGE);
  60                 } catch (IOException ee) {
  61                         JOptionPane.showMessageDialog(TestStand.this,
  62                                                       String.format ("Unknown I/O error on %s", device.toShortString()),
  63                                                       "Unknown I/O error",
  64                                                       JOptionPane.ERROR_MESSAGE);
  65                 } catch (TimeoutException te) {
  66                         JOptionPane.showMessageDialog(this,
  67                                                       String.format ("Timeout on %s", device.toShortString()),
  68                                                       "Timeout error",
  69                                                       JOptionPane.ERROR_MESSAGE);
  70                 } catch (InterruptedException ie) {
  71                         JOptionPane.showMessageDialog(this,
  72                                                       String.format("Interrupted %s", device.toShortString()),
  73                                                       "Interrupted exception",
  74                                                       JOptionPane.ERROR_MESSAGE);
  75                 }
  76         }
  77
  78         public void scan_device_selected(AltosDevice device) {
  79                 telemetry_window(device);
  80         }
  81
  82         Container       pane;
  83         GridBagLayout   gridbag;
  84
  85         JButton addButton(int x, int y, String label) {
  86                 GridBagConstraints      c;
  87                 JButton                 b;
  88
  89                 c = new GridBagConstraints();
  90                 c.gridx = x; c.gridy = y;
  91                 c.fill = GridBagConstraints.BOTH;
  92                 c.weightx = 1;
  93                 c.weighty = 1;
  94                 b = new JButton(label);
  95
  96                 //Dimension ps = b.getPreferredSize();
  97
  98                 gridbag.setConstraints(b, c);
  99                 add(b, c);
 100                 return b;
 101         }
 102
 103         /* OSXAdapter interfaces */
 104         public void macosx_file_handler(String path) {
 105                 process_graph(new File(path));
 106         }
 107
 108         public void macosx_quit_handler() {
 109                 System.exit(0);
 110         }
 111
 112         public void macosx_preferences_handler() {
 113                 ConfigureTestStand();
 114         }
 115
 116         public TestStand() {
 117
 118                 load_library(null);
 119
 120                 register_for_macosx_events();
 121
 122                 AltosUIPreferences.set_component(this);
 123
 124                 pane = getContentPane();
 125                 gridbag = new GridBagLayout();
 126                 pane.setLayout(gridbag);
 127
 128                 JButton b;
 129
 130                 b = addButton(0, 0, "Monitor Flight");
 131                 b.addActionListener(new ActionListener() {
 132                                         public void actionPerformed(ActionEvent e) {
 133                                                 ConnectToDevice();
 134                                         }
 135                                 });
 136                 b.setToolTipText("Connect to TeleDongle and monitor telemetry");
 137                 b = addButton(1, 0, "Save Flight Data");
 138                 b.addActionListener(new ActionListener() {
 139                                         public void actionPerformed(ActionEvent e) {
 140                                                 SaveFlightData();
 141                                         }
 142                                 });
 143                 b.setToolTipText("Download and/or delete flight data from an altimeter");
 144                 b = addButton(3, 0, "Graph Data");
 145                 b.addActionListener(new ActionListener() {
 146                                         public void actionPerformed(ActionEvent e) {
 147                                                 GraphData();
 148                                         }
 149                                 });
 150                 b.setToolTipText("Present flight data in a graph and table of statistics");
 151                 b = addButton(4, 0, "Export Data");
 152                 b.addActionListener(new ActionListener() {
 153                                         public void actionPerformed(ActionEvent e) {
 154                                                 ExportData();
 155                                         }
 156                                 });
 157                 b.setToolTipText("Convert flight data for a spreadsheet or GoogleEarth");
 158                 b = addButton(0, 1, "Configure TeleFire");
 159                 b.addActionListener(new ActionListener() {
 160                                         public void actionPerformed(ActionEvent e) {
 161                                                 ConfigureTeleMetrum();
 162                                         }
 163                                 });
 164                 b.setToolTipText("Set flight, storage and communication parameters");
 165                 b = addButton(1, 1, "Configure TestStand");
 166                 b.addActionListener(new ActionListener() {
 167                                 public void actionPerformed(ActionEvent e) {
 168                                         ConfigureTestStand();
 169                                 }
 170                         });
 171                 b.setToolTipText("Global TestStand settings");
 172
 173                 b = addButton(2, 1, "Configure Ground Station");
 174                 b.addActionListener(new ActionListener() {
 175                                 public void actionPerformed(ActionEvent e) {
 176                                         ConfigureTeleDongle();
 177                                 }
 178                         });
 179
 180                 b = addButton(3, 1, "Flash Image");
 181                 b.addActionListener(new ActionListener() {
 182                                 public void actionPerformed(ActionEvent e) {
 183                                         FlashImage();
 184                                 }
 185                         });
 186                 b.setToolTipText("Replace the firmware in any AltusMetrum product");
 187
 188                 b = addButton(4, 1, "Fire Igniter");
 189                 b.addActionListener(new ActionListener() {
 190                                 public void actionPerformed(ActionEvent e) {
 191                                         FireIgniter();
 192                                 }
 193                         });
 194                 b.setToolTipText("Remote control of igniters for deployment testing");
 195                 b = addButton(0, 2, "Scan Channels");
 196                 b.addActionListener(new ActionListener() {
 197                                 public void actionPerformed(ActionEvent e) {
 198                                         ScanChannels();
 199                                 }
 200                         });
 201                 b.setToolTipText("Find what channel an altimeter is sending telemetry on");
 202
 203                 b = addButton(4, 2, "Quit");
 204                 b.addActionListener(new ActionListener() {
 205                                 public void actionPerformed(ActionEvent e) {
 206                                         System.exit(0);
 207                                 }
 208                         });
 209                 b.setToolTipText("Close all active windows and terminate TestStand");
 210
 211                 setTitle("AltOS");
 212
 213                 pane.doLayout();
 214                 pane.validate();
 215
 216                 doLayout();
 217                 validate();
 218
 219                 setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE);
 220                 addWindowListener(new WindowAdapter() {
 221                         @Override
 222                         public void windowClosing(WindowEvent e) {
 223                                 System.exit(0);
 224                         }
 225                 });
 226
 227                 setLocationByPlatform(false);
 228
 229                 /* Insets aren't set before the window is visible */
 230                 setVisible(true);
 231         }
 232
 233         private void ConnectToDevice() {
 234                 AltosDevice     device = AltosDeviceUIDialog.show(TestStand.this,
 235                                                                 Altos.product_basestation);
 236
 237                 if (device != null)
 238                         telemetry_window(device);
 239         }
 240
 241         void ConfigureCallsign() {
 242                 String  result;
 243                 result = JOptionPane.showInputDialog(TestStand.this,
 244                                                      "Configure Callsign",
 245                                                      AltosUIPreferences.callsign());
 246                 if (result != null)
 247                         AltosUIPreferences.set_callsign(result);
 248         }
 249
 250         void ConfigureTeleMetrum() {
 251                 new AltosConfigFC(TestStand.this);
 252         }
 253
 254         void ConfigureTeleDongle() {
 255                 new AltosConfigTD(TestStand.this);
 256         }
 257
 258         void FlashImage() {
 259                 AltosFlashUI.show(TestStand.this);
 260         }
 261
 262         void FireIgniter() {
 263                 new AltosIgniteUI(TestStand.this);
 264         }
 265
 266         void ScanChannels() {
 267                 new AltosScanUI(TestStand.this, true);
 268         }
 269
 270         void LaunchController() {
 271                 new AltosLaunchUI(TestStand.this);
 272         }
 273
 274         /* Connect to TeleMetrum, either directly or through
 275          * a TeleDongle over the packet link
 276          */
 277
 278         public void graph_flights(AltosEepromList flights) {
 279                 for (AltosEepromLog flight : flights) {
 280                         if (flight.graph_selected && flight.file != null) {
 281                                 process_graph(flight.file);
 282                         }
 283                 }
 284         }
 285
 286         private void SaveFlightData() {
 287                 new AltosEepromManage(this, this, AltosLib.product_any);
 288         }
 289
 290         private static AltosFlightSeries make_series(AltosRecordSet set) {
 291                 AltosFlightSeries series = new AltosFlightSeries(set.cal_data());
 292                 set.capture_series(series);
 293                 series.finish();
 294                 return series;
 295         }
 296
 297         /* Load a flight log file and write out a CSV file containing
 298          * all of the data in standard units
 299          */
 300
 301         private void ExportData() {
 302                 AltosDataChooser chooser;
 303                 chooser = new AltosDataChooser(this);
 304                 AltosRecordSet set = chooser.runDialog();
 305                 if (set == null)
 306                         return;
 307                 AltosFlightSeries series = make_series(set);
 308                 new AltosCSVUI(TestStand.this, series, chooser.file());
 309         }
 310
 311         /* Load a flight log CSV file and display a pretty graph.
 312          */
 313
 314         private void GraphData() {
 315                 AltosDataChooser chooser;
 316                 chooser = new AltosDataChooser(this);
 317                 AltosRecordSet set = chooser.runDialog();
 318                 if (set == null)
 319                         return;
 320                 try {
 321                         new AltosGraphUI(set, chooser.file());
 322                 } catch (InterruptedException ie) {
 323                 } catch (IOException ie) {
 324                 }
 325         }
 326
 327         private void ConfigureTestStand() {
 328                 new AltosConfigureUI(TestStand.this, voice);
 329         }
 330
 331         static AltosWriter open_csv(File file) {
 332                 try {
 333                         return new AltosCSV(file);
 334                 } catch (FileNotFoundException fe) {
 335                         System.out.printf("%s\n", fe.getMessage());
 336                         return null;
 337                 }
 338         }
 339
 340         static AltosWriter open_kml(File file) {
 341                 try {
 342                         return new AltosKML(file);
 343                 } catch (FileNotFoundException fe) {
 344                         System.out.printf("%s\n", fe.getMessage());
 345                         return null;
 346                 }
 347         }
 348
 349         static AltosRecordSet record_set(File input) {
 350                 try {
 351                         return AltosLib.record_set(input);
 352                 } catch (IOException ie) {
 353                         String message = ie.getMessage();
 354                         if (message == null)
 355                                 message = String.format("%s (I/O error)", input.toString());
 356                         System.err.printf("%s: %s\n", input.toString(), message);
 357                 }
 358                 return null;
 359         }
 360
 361         static final int process_none = 0;
 362         static final int process_csv = 1;
 363         static final int process_kml = 2;
 364         static final int process_graph = 3;
 365         static final int process_summary = 4;
 366         static final int process_oneline = 5;
 367
 368         static boolean process_csv(File input) {
 369                 AltosRecordSet set = record_set(input);
 370                 if (set == null)
 371                         return false;
 372
 373                 File output = Altos.replace_extension(input,".csv");
 374                 System.out.printf("Processing \"%s\" to \"%s\"\n", input, output);
 375                 if (input.equals(output)) {
 376                         System.out.printf("Not processing '%s'\n", input);
 377                         return false;
 378                 } else {
 379                         AltosWriter writer = open_csv(output);
 380                         if (writer == null)
 381                                 return false;
 382                         AltosFlightSeries series = make_series(set);
 383                         writer.write(series);
 384                         writer.close();
 385                 }
 386                 return true;
 387         }
 388
 389         static boolean process_kml(File input) {
 390                 AltosRecordSet set = record_set(input);
 391                 if (set == null)
 392                         return false;
 393
 394                 File output = Altos.replace_extension(input,".kml");
 395                 System.out.printf("Processing \"%s\" to \"%s\"\n", input, output);
 396                 if (input.equals(output)) {
 397                         System.out.printf("Not processing '%s'\n", input);
 398                         return false;
 399                 } else {
 400                         AltosWriter writer = open_kml(output);
 401                         if (writer == null)
 402                                 return false;
 403                         AltosFlightSeries series = make_series(set);
 404                         series.finish();
 405                         writer.write(series);
 406                         writer.close();
 407                         return true;
 408                 }
 409         }
 410
 411         static boolean process_graph(File file) {
 412                 AltosRecordSet set = record_set(file);
 413                 if (set == null)
 414                         return false;
 415                 try {
 416                         new AltosGraphUI(set, file);
 417                         return true;
 418                 } catch (InterruptedException ie) {
 419                 } catch (IOException ie) {
 420                 }
 421                 return false;
 422         }
 423
 424         static boolean process_summary(File file) {
 425                 AltosRecordSet set = record_set(file);
 426                 if (set == null)
 427                         return false;
 428                 System.out.printf("%s:\n", file.toString());
 429                 AltosFlightSeries series = make_series(set);
 430                 AltosFlightStats stats = new AltosFlightStats(series);
 431                 if (stats.serial != AltosLib.MISSING)
 432                         System.out.printf("Serial:       %5d\n", stats.serial);
 433                 if (stats.flight != AltosLib.MISSING)
 434                         System.out.printf("Flight:       %5d\n", stats.flight);
 435                 if (stats.year != AltosLib.MISSING)
 436                         System.out.printf("Date:    %04d-%02d-%02d\n",
 437                                           stats.year, stats.month, stats.day);
 438                 if (stats.hour != AltosLib.MISSING)
 439                         System.out.printf("Time:      %02d:%02d:%02d UTC\n",
 440                                           stats.hour, stats.minute, stats.second);
 441                 if (stats.max_height != AltosLib.MISSING)
 442                         System.out.printf("Max height:  %6.0f m    %6.0f ft\n",
 443                                           stats.max_height,
 444                                           AltosConvert.meters_to_feet(stats.max_height));
 445                 if (stats.max_speed != AltosLib.MISSING)
 446                         System.out.printf("Max speed:   %6.0f m/s  %6.0f ft/s  %6.4f Mach\n",
 447                                           stats.max_speed,
 448                                           AltosConvert.meters_to_feet(stats.max_speed),
 449                                           AltosConvert.meters_to_mach(stats.max_speed));
 450                 if (stats.max_acceleration != AltosLib.MISSING) {
 451                         System.out.printf("Max accel:   %6.0f m/s² %6.0f ft/s² %6.2f g\n",
 452                                           stats.max_acceleration,
 453                                           AltosConvert.meters_to_feet(stats.max_acceleration),
 454                                           AltosConvert.meters_to_g(stats.max_acceleration));
 455                 }
 456                 if (stats.state_speed[Altos.ao_flight_drogue] != AltosLib.MISSING)
 457                         System.out.printf("Drogue rate: %6.0f m/s  %6.0f ft/s\n",
 458                                           stats.state_speed[Altos.ao_flight_drogue],
 459                                           AltosConvert.meters_to_feet(stats.state_speed[Altos.ao_flight_drogue]));
 460                 if (stats.state_speed[Altos.ao_flight_main] != AltosLib.MISSING)
 461                         System.out.printf("Main rate:   %6.0f m/s  %6.0f ft/s\n",
 462                                           stats.state_speed[Altos.ao_flight_main],
 463                                           AltosConvert.meters_to_feet(stats.state_speed[Altos.ao_flight_main]));
 464                 if (stats.landed_time != AltosLib.MISSING &&
 465                     stats.boost_time != AltosLib.MISSING &&
 466                     stats.landed_time > stats.boost_time)
 467                         System.out.printf("Flight time: %6.0f s\n",
 468                                           stats.landed_time -
 469                                           stats.boost_time);
 470                 System.out.printf("\n");
 471                 return true;
 472         }
 473
 474         static boolean process_oneline(File file) {
 475                 AltosRecordSet set = record_set(file);
 476                 if (set == null)
 477                         return false;
 478                 System.out.printf("%s", file.toString());
 479                 AltosFlightSeries series = make_series(set);
 480                 AltosFlightStats stats = new AltosFlightStats(series);
 481                 if (stats.max_height != AltosLib.MISSING)
 482                         System.out.printf(" height  %6.0f m", stats.max_height);
 483                 if (stats.max_speed != AltosLib.MISSING)
 484                         System.out.printf(" speed   %6.0f m/s", stats.max_speed);
 485                 if (stats.state_enter_speed[AltosLib.ao_flight_drogue] != AltosLib.MISSING)
 486                         System.out.printf(" drogue-deploy   %6.0f m/s", stats.state_enter_speed[AltosLib.ao_flight_drogue]);
 487                 if (stats.max_acceleration != AltosLib.MISSING)
 488                         System.out.printf(" accel   %6.0f m/s²", stats.max_acceleration);
 489                 System.out.printf("\n");
 490                 return true;
 491         }
 492
 493         public static void help(int code) {
 494                 System.out.printf("Usage: teststand [OPTION]... [FILE]...\n");
 495                 System.out.printf("  Options:\n");
 496                 System.out.printf("    --graph <filename>\t\tgraph a flight\n");
 497                 System.out.printf("    --summary <filename>\t\tText summary of a flight\n");
 498                 System.out.printf("    --oneline <filename>\t\tOne line summary of a flight\n");
 499                 System.out.printf("    --csv\tgenerate comma separated output for spreadsheets, etc\n");
 500                 System.out.printf("    --kml\tgenerate KML output for use with Google Earth\n");
 501                 System.exit(code);
 502         }
 503
 504         public static void main(final String[] args) {
 505                 int     errors = 0;
 506                 load_library(null);
 507                 try {
 508                         UIManager.setLookAndFeel(AltosUIPreferences.look_and_feel());
 509                 } catch (Exception e) {
 510                 }
 511                 TestStand teststand = null;
 512
 513                 /* Handle batch-mode */
 514                 if (args.length == 0) {
 515                         teststand = new TestStand();
 516                         java.util.List<AltosDevice> devices = AltosUSBDevice.list(Altos.product_basestation);
 517                         if (devices != null)
 518                                 for (AltosDevice device : devices)
 519                                         teststand.telemetry_window(device);
 520                 } else {
 521                         int process = process_none;
 522                         for (int i = 0; i < args.length; i++) {
 523                                 if (args[i].equals("--help"))
 524                                         help(0);
 525                                 else if (args[i].equals("--kml"))
 526                                         process = process_kml;
 527                                 else if (args[i].equals("--csv"))
 528                                         process = process_csv;
 529                                 else if (args[i].equals("--graph"))
 530                                         process = process_graph;
 531                                 else if (args[i].equals("--summary"))
 532                                         process = process_summary;
 533                                 else if (args[i].equals("--oneline"))
 534                                         process = process_oneline;
 535                                 else if (args[i].startsWith("--"))
 536                                         help(1);
 537                                 else {
 538                                         File file = new File(args[i]);
 539                                         switch (process) {
 540                                         case process_none:
 541                                                 if (teststand == null)
 542                                                         teststand = new TestStand();
 543                                         case process_graph:
 544                                                 if (!process_graph(file))
 545                                                         ++errors;
 546                                                 break;
 547                                         case process_kml:
 548                                                 if (!process_kml(file))
 549                                                         ++errors;
 550                                                 break;
 551                                         case process_csv:
 552                                                 if (!process_csv(file))
 553                                                         ++errors;
 554                                                 break;
 555                                         case process_summary:
 556                                                 if (!process_summary(file))
 557                                                         ++errors;
 558                                                 break;
 559                                         case process_oneline:
 560                                                 if (!process_oneline(file))
 561                                                         ++errors;
 562                                                 break;
 563                                         }
 564                                 }
 565                         }
 566                 }
 567                 if (errors != 0)
 568                         System.exit(errors);
 569         }
 570 }