git.gag.com Git - fw/altos/blob - ao-tools/altosui/AltosGraphDataChooser.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
  30 import altosui.AltosPreferences;
  31 import altosui.AltosCsvReader;
  32 import altosui.AltosDataPointReader;
  33 import altosui.AltosEepromIterable;
  34 import altosui.AltosTelemetryIterable;
  35
  36 public class AltosGraphDataChooser extends JFileChooser {
  37         JFrame  frame;
  38         String  filename;
  39         File    file;
  40
  41         public String filename() {
  42                 return filename;
  43         }
  44
  45         public File file() {
  46                 return file;
  47         }
  48
  49         public Iterable<AltosDataPoint> runDialog() {
  50                 int     ret;
  51
  52                 ret = showOpenDialog(frame);
  53                 if (ret == APPROVE_OPTION) {
  54                         file = getSelectedFile();
  55                         if (file == null)
  56                                 return null;
  57                         filename = file.getName();
  58                         try {
  59                 if (filename.endsWith("eeprom")) {
  60                     FileInputStream in = new FileInputStream(file);
  61                     return new AltosDataPointReader(new AltosEepromIterable(in));
  62                 } else if (filename.endsWith("telem")) {
  63                     FileInputStream in = new FileInputStream(file);
  64                     return new AltosDataPointReader(new AltosTelemetryIterable(in));
  65                 } else if (filename.endsWith("csv")) {
  66                                     return new AltosCsvReader(new FileReader(file));
  67                 } else {
  68                     throw new FileNotFoundException();
  69                 }
  70                         } catch (FileNotFoundException fe) {
  71                                 JOptionPane.showMessageDialog(frame,
  72                                                               filename,
  73                                                               "Cannot open file",
  74                                                               JOptionPane.ERROR_MESSAGE);
  75                         }
  76                 }
  77                 return null;
  78         }
  79
  80         public AltosGraphDataChooser(JFrame in_frame) {
  81                 frame = in_frame;
  82                 setDialogTitle("Select Flight Record File");
  83                 setFileFilter(new FileNameExtensionFilter("Flight data file",
  84                                                           "csv", "telem", "eeprom"));
  85                 setCurrentDirectory(AltosPreferences.logdir());
  86         }
  87 }