telegps: Auto-connect to any base stations plugged in at startup
[fw/altos] / altosui / AltosFlashUI.java
index f91c542dfd5e7b24cc1b53b762a5f04ddd407b54..d8c70a0675136b23112bbd18b9edd7cc756ef5e8 100644 (file)
@@ -21,15 +21,13 @@ import java.awt.*;
 import java.awt.event.*;
 import javax.swing.*;
 import javax.swing.filechooser.FileNameExtensionFilter;
-import javax.swing.table.*;
 import java.io.*;
-import java.util.*;
-import java.text.*;
-import java.util.prefs.*;
 import java.util.concurrent.*;
+import org.altusmetrum.altoslib_4.*;
+import org.altusmetrum.altosuilib_2.*;
 
 public class AltosFlashUI
-       extends AltosDialog
+       extends AltosUIDialog
        implements ActionListener
 {
        Container       pane;
@@ -47,18 +45,51 @@ public class AltosFlashUI
        File            file;
 
        // Debug connection
-       AltosDevice     debug_dongle;
+       AltosDevice     device;
+
+       AltosLink       link;
 
        // Desired Rom configuration
        AltosRomconfig  rom_config;
 
        // Flash controller
-       AltosFlash      flash;
+       AltosProgrammer programmer;
+
+       private static String[] pair_programmed = {
+               "teleballoon",
+               "telebt",
+               "teledongle",
+               "telefire",
+               "telemetrum-v0",
+               "telemetrum-v1",
+               "telemini",
+               "telenano",
+               "teleshield",
+               "teleterra"
+       };
+
+       private boolean is_pair_programmed() {
+
+               if (file != null) {
+                       String  name = file.getName();
+                       for (int i = 0; i < pair_programmed.length; i++) {
+                               if (name.startsWith(pair_programmed[i]))
+                                       return true;
+                       }
+               }
+               if (device != null) {
+                       if (!device.matchProduct(AltosLib.product_altusmetrum) &&
+                           (device.matchProduct(AltosLib.product_teledongle) ||
+                            device.matchProduct(AltosLib.product_telebt)))
+                               return true;
+               }
+               return false;
+       }
 
        public void actionPerformed(ActionEvent e) {
                if (e.getSource() == cancel) {
-                       if (flash != null)
-                               flash.abort();
+                       if (programmer != null)
+                               programmer.abort();
                        setVisible(false);
                        dispose();
                } else {
@@ -158,6 +189,33 @@ public class AltosFlashUI
                serial_value.setText(String.format("%d", serial_number));
        }
 
+       static class AltosHexfileFilter extends javax.swing.filechooser.FileFilter {
+               int product;
+               String head;
+               String description;
+
+               public AltosHexfileFilter(int product, String head, String description) {
+                       this.product = product;
+                       this.head = head;
+                       this.description = description;
+               }
+
+               public boolean accept(File file) {
+                       return !file.isFile() || (file.getName().startsWith(head) && file.getName().endsWith(".ihx"));
+               }
+
+               public String getDescription() {
+                       return description;
+               }
+       }
+
+       static AltosHexfileFilter[] filters = {
+               new AltosHexfileFilter(AltosLib.product_telemetrum, "telemetrum", "TeleMetrum Image"),
+               new AltosHexfileFilter(AltosLib.product_teledongle, "teledongle", "TeleDongle Image"),
+               new AltosHexfileFilter(AltosLib.product_telemega, "telemega", "TeleMega Image"),
+               new AltosHexfileFilter(AltosLib.product_easymini, "easymini", "EasyMini Image"),
+       };
+
        boolean select_source_file() {
                JFileChooser    hexfile_chooser = new JFileChooser();
 
@@ -166,7 +224,21 @@ public class AltosFlashUI
                        hexfile_chooser.setCurrentDirectory(firmwaredir);
 
                hexfile_chooser.setDialogTitle("Select Flash Image");
-               hexfile_chooser.setFileFilter(new FileNameExtensionFilter("Flash Image", "ihx"));
+
+               for (int i = 0; i < filters.length; i++) {
+                       hexfile_chooser.addChoosableFileFilter(filters[i]);
+               }
+               javax.swing.filechooser.FileFilter ihx_filter = new FileNameExtensionFilter("Flash Image", "ihx");
+               hexfile_chooser.addChoosableFileFilter(ihx_filter);
+               hexfile_chooser.setFileFilter(ihx_filter);
+
+               if (!is_pair_programmed() && !device.matchProduct(AltosLib.product_altusmetrum)) {
+                       for (int i = 0; i < filters.length; i++) {
+                               if (device != null && device.matchProduct(filters[i].product))
+                                       hexfile_chooser.setFileFilter(filters[i]);
+                       }
+               }
+
                int returnVal = hexfile_chooser.showOpenDialog(frame);
 
                if (returnVal != JFileChooser.APPROVE_OPTION)
@@ -175,13 +247,16 @@ public class AltosFlashUI
                if (file == null)
                        return false;
                AltosUIPreferences.set_firmwaredir(file.getParentFile());
+
                return true;
        }
 
-       boolean select_debug_dongle() {
-               debug_dongle = AltosDeviceDialog.show(frame, Altos.product_any);
+       boolean select_device() {
+               int     product = Altos.product_any;
+
+               device = AltosDeviceUIDialog.show(frame, Altos.product_any);
 
-               if (debug_dongle == null)
+               if (device == null)
                        return false;
                return true;
        }
@@ -206,7 +281,7 @@ public class AltosFlashUI
                } else if (e instanceof AltosSerialInUseException) {
                        JOptionPane.showMessageDialog(frame,
                                                      String.format("Device \"%s\" already in use",
-                                                                   debug_dongle.toShortString()),
+                                                                   device.toShortString()),
                                                      "Device in use",
                                                      JOptionPane.ERROR_MESSAGE);
                } else if (e instanceof IOException) {
@@ -217,39 +292,52 @@ public class AltosFlashUI
                }
        }
 
-       class flash_task implements Runnable {
+       class flash_task implements Runnable, AltosFlashListener {
                AltosFlashUI    ui;
                Thread          t;
-               AltosFlash      flash;
+               AltosProgrammer programmer;
+
+               public void position(String in_s, int in_percent) {
+                       final String s = in_s;
+                       final int percent = in_percent;
+                       Runnable r = new Runnable() {
+                                       public void run() {
+                                               try {
+                                                       ui.actionPerformed(new ActionEvent(this,
+                                                                                          percent,
+                                                                                          s));
+                                               } catch (Exception ex) {
+                                               }
+                                       }
+                               };
+                       SwingUtilities.invokeLater(r);
+               }
 
                public void run () {
                        try {
-                               flash = new AltosFlash(ui.file, ui.debug_dongle);
-                               flash.addActionListener(ui);
+                               if (ui.is_pair_programmed())
+                                       programmer = new AltosFlash(ui.file, link, this);
+                               else
+                                       programmer = new AltosSelfFlash(ui.file, link, this);
 
-                               final AltosRomconfig    current_config = flash.romconfig();
+                               final AltosRomconfig    current_config = programmer.romconfig();
 
                                final Semaphore await_rom_config = new Semaphore(0);
                                SwingUtilities.invokeLater(new Runnable() {
                                                public void run() {
-                                                       ui.flash = flash;
+                                                       ui.programmer = programmer;
                                                        ui.update_rom_config_info(current_config);
-                                                       System.out.printf("Done updating rom config info\n");
                                                        await_rom_config.release();
                                                }
                                        });
-                               System.out.printf("Waiting for rom configuration updates\n");
                                await_rom_config.acquire();
-                               System.out.printf("Got rom config update\n");
 
                                if (ui.rom_config != null) {
-                                       System.out.printf("rom_config not null\n");
-                                       flash.set_romconfig(ui.rom_config);
-                                       flash.flash();
+                                       programmer.set_romconfig(ui.rom_config);
+                                       programmer.flash();
                                }
                        } catch (InterruptedException ee) {
                                final Exception e = ee;
-                               System.out.printf("exception %s\n", e.toString());
                                SwingUtilities.invokeLater(new Runnable() {
                                                public void run() {
                                                        ui.exception(e);
@@ -257,23 +345,14 @@ public class AltosFlashUI
                                        });
                        } catch (IOException ee) {
                                final Exception e = ee;
-                               System.out.printf("exception %s\n", e.toString());
-                               SwingUtilities.invokeLater(new Runnable() {
-                                               public void run() {
-                                                       ui.exception(e);
-                                               }
-                                       });
-                       } catch (AltosSerialInUseException ee) {
-                               final Exception e = ee;
-                               System.out.printf("exception %s\n", e.toString());
                                SwingUtilities.invokeLater(new Runnable() {
                                                public void run() {
                                                        ui.exception(e);
                                                }
                                        });
                        } finally {
-                               if (flash != null)
-                                       flash.close();
+                               if (programmer != null)
+                                       programmer.close();
                        }
                }
 
@@ -286,16 +365,63 @@ public class AltosFlashUI
 
        flash_task      flasher;
 
+       private boolean open_device() throws InterruptedException {
+               try {
+                       link = new AltosSerial(device);
+                       if (is_pair_programmed())
+                               return true;
+
+                       if (link == null)
+                               throw new IOException(String.format("%s: open failed", device.toShortString()));
+
+                       while (!link.is_loader()) {
+                               link.to_loader();
+
+                               java.util.List<AltosDevice> devices = null;
+
+                               for (int tries = 0; tries < 10; tries++) {
+                                       Thread.sleep(100);
+                                       devices = AltosUSBDevice.list(AltosLib.product_altusmetrum);
+                                       if (devices.size() != 0)
+                                               break;
+                               }
+
+                               if (devices.size() == 1)
+                                       device = devices.get(0);
+                               else {
+                                       device = AltosDeviceUIDialog.show(frame, AltosLib.product_altusmetrum);
+                                       if (device == null)
+                                               return false;
+                               }
+                               link = new AltosSerial(device);
+                       }
+                       return true;
+               } catch (AltosSerialInUseException ee) {
+                       exception(ee);
+               } catch (FileNotFoundException fe) {
+                       exception(fe);
+               } catch (IOException ie) {
+                       exception (ie);
+               }
+               return false;
+       }
+
        /*
         * Execute the steps for flashing
         * a device. Note that this returns immediately;
         * this dialog is not modal
         */
        void showDialog() {
-               if (!select_debug_dongle())
+               if (!select_device())
                        return;
                if (!select_source_file())
                        return;
+               try {
+                       if (!open_device())
+                               return;
+               } catch (InterruptedException ie) {
+                       return;
+               }
                build_dialog();
                flash_task      f = new flash_task(this);
        }
@@ -311,4 +437,4 @@ public class AltosFlashUI
 
                frame = in_frame;
        }
-}
\ No newline at end of file
+}