osd-contiki/tools/collect-view/src/se/sics/contiki/collect/MoteProgrammer.java

280 lines
8.4 KiB
Java

/*
* 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.
*
*
* -----------------------------------------------------------------
*
* MoteProgrammer
*
* Authors : Joakim Eriksson, Niclas Finne
* Created : 10 jul 2008
* Updated : $Date: 2010/11/03 14:53:05 $
* $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.JProgressBar;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.SwingUtilities;
/**
*
*/
public class MoteProgrammer {
private MoteProgrammerProcess[] processes;
private String[] motes;
private String firmwareFile;
private Window parent;
private JProgressBar progressBar;
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 String[] getMotes() {
return motes;
}
public void setMotes(String[] 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");
progressBar = new JProgressBar(0, 100);
progressBar.setValue(0);
progressBar.setString("Programming...");
progressBar.setStringPainted(true);
progressBar.setIndeterminate(true);
dialog.getContentPane().add(progressBar, BorderLayout.NORTH);
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.getMoteID() + "> 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;
final String doneMessage = "Programming finished with " + errors + " errors.";
log(doneMessage, null);
if (closeButton != null) {
SwingUtilities.invokeLater(new Runnable() {
public void run() {
progressBar.setValue(100);
progressBar.setIndeterminate(false);
progressBar.setString(doneMessage);
closeButton.setText("Close");
}});
}
synchronized (this) {
notifyAll();
}
}
}
}
protected boolean handleLogLine(MoteProgrammerProcess moteProgrammerProcess,
String line, boolean stderr, final Throwable e) {
log("Mote@" + moteProgrammerProcess.getMoteID() + "> " + 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 String[] { args[1] });
} else {
mp.searchForMotes();
}
if (!mp.hasMotes()) {
System.err.println("No motes connected");
System.exit(1);
}
mp.programMotes();
}
}