added option to program Sky nodes from the Java interface

examples/sky-shell/src/se/sics/contiki/collect/MoteProgrammer.java

@@ -0,0 +1,269 @@
+/*
+ * Copyright (c) 2008, Swedish Institute of Computer Science.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. Neither the name of the Institute nor the names of its contributors
+ *    may be used to endorse or promote products derived from this software
+ *    without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * $Id: MoteProgrammer.java,v 1.1 2008/07/10 14:52:59 nifi Exp $
+ *
+ * -----------------------------------------------------------------
+ *
+ * MoteProgrammer
+ *
+ * Authors : Joakim Eriksson, Niclas Finne
+ * Created : 10 jul 2008
+ * Updated : $Date: 2008/07/10 14:52:59 $
+ *           $Revision: 1.1 $
+ */
+
+package se.sics.contiki.collect;
+import java.awt.BorderLayout;
+import java.awt.Window;
+import java.awt.event.ActionEvent;
+import java.awt.event.ActionListener;
+import java.io.File;
+import java.io.IOException;
+import javax.swing.JButton;
+import javax.swing.JDialog;
+import javax.swing.JPanel;
+import javax.swing.JScrollPane;
+import javax.swing.JTextArea;
+import javax.swing.SwingUtilities;
+
+/**
+ *
+ */
+public class MoteProgrammer {
+
+  private MoteProgrammerProcess[] processes;
+  private int[] motes;
+  private String firmwareFile;
+
+  private Window parent;
+  protected JTextArea logTextArea;
+  protected JDialog dialog;
+  protected JButton closeButton;
+  private boolean isDone;
+
+  public MoteProgrammer() {
+  }
+
+  public Window getParentComponent() {
+    return parent;
+  }
+
+  public void setParentComponent(Window parent) {
+    this.parent = parent;
+  }
+
+  public boolean hasMotes() {
+    return motes != null && motes.length > 0;
+  }
+
+  public int[] getMotes() {
+    return motes;
+  }
+
+  public void setMotes(int[] motes) {
+    this.motes = motes;
+  }
+
+  public void searchForMotes() throws IOException {
+    MoteFinder finder = new MoteFinder();
+    motes = finder.getMotes();
+    finder.close();
+  }
+
+  public String getFirmwareFile() {
+    return firmwareFile;
+  }
+
+  public void setFirmwareFile(String firmwareFile) {
+    this.firmwareFile = firmwareFile;
+  }
+
+  public void programMotes() throws IOException {
+    if (firmwareFile == null) {
+      throw new IllegalStateException("no firmware");
+    }
+    if (!hasMotes()) {
+      throw new IllegalStateException("no motes");
+    }
+    File fp = new File(firmwareFile);
+    if (!fp.canRead()) {
+      throw new IllegalStateException("can not read firmware file '" + fp.getAbsolutePath() + '\'');
+    }
+    if (parent != null) {
+      // Use GUI
+      dialog = new JDialog(parent, "Mote Programmer");
+      logTextArea = new JTextArea(28, 80);
+      logTextArea.setEditable(false);
+      logTextArea.setLineWrap(true);
+      dialog.getContentPane().add(new JScrollPane(logTextArea), BorderLayout.CENTER);
+      JPanel panel = new JPanel();
+      closeButton = new JButton("Cancel");
+      closeButton.addActionListener(new ActionListener() {
+
+        public void actionPerformed(ActionEvent e) {
+          MoteProgrammer.this.close();
+        }
+
+      });
+      panel.add(closeButton);
+      dialog.getContentPane().add(panel, BorderLayout.SOUTH);
+      dialog.pack();
+      dialog.setLocationRelativeTo(parent);
+      dialog.setVisible(true);
+    }
+    processes = new MoteProgrammerProcess[motes.length];
+    isDone = false;
+    try {
+      log("Programming " + motes.length + " motes with '" + firmwareFile + '\'', null);
+      for (int i = 0, n = processes.length; i < n; i++) {
+        processes[i] = new MoteProgrammerProcess(motes[i], firmwareFile) {
+          protected void logLine(String line, boolean stderr, Throwable e) {
+            if (!handleLogLine(this, line, stderr, e)) {
+              super.logLine(line, stderr, e);
+            }
+          }
+          protected void processEnded() {
+            handleProcessEnded(this);
+          }
+        };
+        processes[i].start();
+      }
+    } catch (Exception e) {
+      throw (IOException) new IOException("Failed to program motes").initCause(e);
+    }
+  }
+
+  public synchronized void waitForProcess() throws InterruptedException {
+    while (!isDone) {
+      wait();
+    }
+  }
+
+  public void close() {
+    MoteProgrammerProcess[] processes = this.processes;
+    if (processes != null) {
+      this.processes = null;
+      for (int i = 0, n = processes.length; i < n; i++) {
+        if (processes[i] != null) {
+          processes[i].stop();
+        }
+      }
+    }
+    if (dialog != null) {
+      SwingUtilities.invokeLater(new Runnable() {
+        public void run() {
+          dialog.setVisible(false);
+        }
+      });
+    }
+    isDone = true;
+    synchronized (this) {
+      notifyAll();
+    }
+  }
+
+  protected void handleProcessEnded(MoteProgrammerProcess process) {
+    // Another process has finished
+    log("Mote@" + process.getMote() + "> finished" + (process.hasError() ? " with errors": ""), null);
+    MoteProgrammerProcess[] processes = this.processes;
+    if (processes != null) {
+      int running = 0;
+      int errors = 0;
+      for(MoteProgrammerProcess p: processes) {
+        if (p.isRunning()) {
+          running++;
+        } else if (p.hasError()) {
+          errors++;
+        }
+      }
+      if (running == 0) {
+        // All processes has finished
+        isDone = true;
+
+        log("Programming finished with " + errors + " errors.", null);
+        if (closeButton != null) {
+          SwingUtilities.invokeLater(new Runnable() {
+            public void run() {
+              closeButton.setText("Close");
+            }});
+        }
+        synchronized (this) {
+          notifyAll();
+        }
+      }
+    }
+  }
+
+  protected boolean handleLogLine(MoteProgrammerProcess moteProgrammerProcess,
+      String line, boolean stderr, final Throwable e) {
+    log("Mote@" + moteProgrammerProcess.getMote() + "> " + line, e);
+    return true;
+  }
+
+  private void log(String line, final Throwable e) {
+    System.err.println(line);
+    if (e != null) {
+      e.printStackTrace();
+      line += "\n  " + e;
+    }
+    final String text = line;
+    SwingUtilities.invokeLater(new Runnable() {
+      public void run() {
+        int len = logTextArea.getDocument().getLength();
+        if (len == 0) {
+          logTextArea.append(text);
+        } else {
+          logTextArea.append('\n' + text);
+          len++;
+        }
+        logTextArea.setCaretPosition(len + text.length());
+      }
+    });
+  }
+
+  public static void main(String[] args) throws IOException {
+    MoteProgrammer mp = new MoteProgrammer();
+    if (args.length < 1 || args.length > 2) {
+      System.err.println("Usage: MoteProgrammer <firmware> [mote]");
+      System.exit(1);
+    }
+    mp.setFirmwareFile(args[0]);
+    if (args.length == 2) {
+      mp.setMotes(new int[] { Integer.parseInt(args[1]) });
+    } else {
+      mp.searchForMotes();
+    }
+    if (!mp.hasMotes()) {
+      System.err.println("No motes connected");
+      System.exit(1);
+    }
+    mp.programMotes();
+  }
+
+}