Java interface using the shell to collect and visualize sensor values from Sky nodes
This commit is contained in:
nifi
parent
5da504dcc7
commit
0fa1ae0fce
15 changed files with 3135 additions and 0 deletions
|
|
@ -0,0 +1,875 @@
|
|||
/*
|
||||
* 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: CollectServer.java,v 1.1 2008/07/09 23:18:06 nifi Exp $
|
||||
*
|
||||
* -----------------------------------------------------------------
|
||||
*
|
||||
* CollectServer
|
||||
*
|
||||
* Authors : Joakim Eriksson, Niclas Finne
|
||||
* Created : 3 jul 2008
|
||||
* Updated : $Date: 2008/07/09 23:18:06 $
|
||||
* $Revision: 1.1 $
|
||||
*/
|
||||
|
||||
package se.sics.contiki.collect;
|
||||
import java.awt.BorderLayout;
|
||||
import java.awt.GraphicsEnvironment;
|
||||
import java.awt.Rectangle;
|
||||
import java.awt.event.ActionEvent;
|
||||
import java.awt.event.ActionListener;
|
||||
import java.awt.event.KeyEvent;
|
||||
import java.awt.event.WindowAdapter;
|
||||
import java.awt.event.WindowEvent;
|
||||
import java.io.BufferedInputStream;
|
||||
import java.io.BufferedReader;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.FileNotFoundException;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.FileReader;
|
||||
import java.io.FileWriter;
|
||||
import java.io.IOException;
|
||||
import java.io.PrintWriter;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Hashtable;
|
||||
import java.util.Properties;
|
||||
import javax.swing.BorderFactory;
|
||||
import javax.swing.DefaultListModel;
|
||||
import javax.swing.JCheckBoxMenuItem;
|
||||
import javax.swing.JFileChooser;
|
||||
import javax.swing.JFrame;
|
||||
import javax.swing.JList;
|
||||
import javax.swing.JMenu;
|
||||
import javax.swing.JMenuBar;
|
||||
import javax.swing.JMenuItem;
|
||||
import javax.swing.JOptionPane;
|
||||
import javax.swing.JScrollPane;
|
||||
import javax.swing.JTabbedPane;
|
||||
import javax.swing.SwingUtilities;
|
||||
import javax.swing.event.ListSelectionEvent;
|
||||
import javax.swing.event.ListSelectionListener;
|
||||
import org.jfree.chart.axis.NumberAxis;
|
||||
import org.jfree.chart.axis.ValueAxis;
|
||||
import se.sics.contiki.collect.gui.BarChartPanel;
|
||||
import se.sics.contiki.collect.gui.MapPanel;
|
||||
import se.sics.contiki.collect.gui.SerialConsole;
|
||||
import se.sics.contiki.collect.gui.TimeChartPanel;
|
||||
|
||||
/**
|
||||
*
|
||||
*/
|
||||
public class CollectServer {
|
||||
|
||||
public static final String WINDOW_TITLE = "Sensor Data Collect with Contiki";
|
||||
public static final double TICKS_PER_SECOND = 4096; /* TODO Convert from TimerB ticks to seconds */
|
||||
public static final double UPDATE_PERIOD = 1; /* TODO Set update period (1 second?) */
|
||||
|
||||
public static final String CONFIG_FILE = "collect.conf";
|
||||
public static final String SENSORDATA_FILE = "sensordata.log";
|
||||
public static final String CONFIG_DATA_FILE = "collect-data.conf";
|
||||
public static final String INIT_SCRIPT = "collect-init.script";
|
||||
|
||||
private Properties config = new Properties();
|
||||
|
||||
private String configFile;
|
||||
private Properties configTable = new Properties();
|
||||
|
||||
private ArrayList<SensorData> sensorDataList = new ArrayList<SensorData>();
|
||||
private PrintWriter sensorDataOutput;
|
||||
|
||||
private Hashtable<String,Node> nodeTable = new Hashtable<String,Node>();
|
||||
private Node[] nodeCache;
|
||||
|
||||
private JFrame window;
|
||||
private JTabbedPane mainPanel;
|
||||
|
||||
private Visualizer[] visualizers;
|
||||
private MapPanel mapPanel;
|
||||
private SerialConsole serialConsole;
|
||||
private JFileChooser fileChooser;
|
||||
|
||||
private JList nodeList;
|
||||
private DefaultListModel nodeModel;
|
||||
private Node[] selectedNodes;
|
||||
|
||||
private SerialConnection serialConnection;
|
||||
|
||||
@SuppressWarnings("serial")
|
||||
public CollectServer(String comPort) {
|
||||
loadConfig(config, CONFIG_FILE);
|
||||
|
||||
this.configFile = config.getProperty("config.datafile", CONFIG_DATA_FILE);
|
||||
if (this.configFile != null) {
|
||||
loadConfig(configTable, this.configFile);
|
||||
}
|
||||
if (comPort == null) {
|
||||
comPort = configTable.getProperty("collect.serialport");
|
||||
}
|
||||
|
||||
/* Make sure we have nice window decorations */
|
||||
// JFrame.setDefaultLookAndFeelDecorated(true);
|
||||
// JDialog.setDefaultLookAndFeelDecorated(true);
|
||||
Rectangle maxSize = GraphicsEnvironment.getLocalGraphicsEnvironment()
|
||||
.getMaximumWindowBounds();
|
||||
|
||||
/* Create and set up the window */
|
||||
window = new JFrame(WINDOW_TITLE + " (not connected)");
|
||||
window.setLocationByPlatform(true);
|
||||
if (maxSize != null) {
|
||||
window.setMaximizedBounds(maxSize);
|
||||
}
|
||||
window.setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE);
|
||||
|
||||
window.addWindowListener(new WindowAdapter() {
|
||||
|
||||
public void windowClosing(WindowEvent e) {
|
||||
exit();
|
||||
}
|
||||
});
|
||||
|
||||
nodeModel = new DefaultListModel();
|
||||
nodeList = new JList(nodeModel);
|
||||
nodeList.setPrototypeCellValue("Node 88888");
|
||||
// DefaultListCellRenderer l = new DefaultListCellRenderer();
|
||||
// l.setHorizontalAlignment(DefaultListCellRenderer.RIGHT);
|
||||
// nodeList.setCellRenderer(l);
|
||||
nodeList.addListSelectionListener(new ListSelectionListener() {
|
||||
|
||||
@Override
|
||||
public void valueChanged(ListSelectionEvent e) {
|
||||
if (!e.getValueIsAdjusting() && e.getSource() == nodeList) {
|
||||
Node[] selected;
|
||||
int iMin = nodeList.getMinSelectionIndex();
|
||||
int iMax = nodeList.getMaxSelectionIndex();
|
||||
if ((iMin < 0) || (iMax < 0)) {
|
||||
selected = null;
|
||||
} else {
|
||||
Node[] tmp = new Node[1 + (iMax - iMin)];
|
||||
int n = 0;
|
||||
for(int i = iMin; i <= iMax; i++) {
|
||||
if (nodeList.isSelectedIndex(i)) {
|
||||
tmp[n++] = (Node) nodeModel.getElementAt(i);
|
||||
}
|
||||
}
|
||||
if (n != tmp.length) {
|
||||
Node[] t = new Node[n];
|
||||
System.arraycopy(tmp, 0, t, 0, n);
|
||||
tmp = t;
|
||||
}
|
||||
selected = tmp;
|
||||
}
|
||||
selectNodes(selected, false);
|
||||
}
|
||||
|
||||
}});
|
||||
nodeList.setBorder(BorderFactory.createTitledBorder("Nodes"));
|
||||
window.getContentPane().add(new JScrollPane(nodeList), BorderLayout.WEST);
|
||||
|
||||
mainPanel = new JTabbedPane();
|
||||
mainPanel.setBackground(nodeList.getBackground());
|
||||
mainPanel.setTabLayoutPolicy(JTabbedPane.WRAP_TAB_LAYOUT);
|
||||
|
||||
serialConsole = new SerialConsole(this);
|
||||
mapPanel = new MapPanel(this);
|
||||
String image = getConfig("collect.mapimage");
|
||||
if (image != null) {
|
||||
mapPanel.setMapBackground(image);
|
||||
}
|
||||
visualizers = new Visualizer[] {
|
||||
mapPanel,
|
||||
new BarChartPanel(this, "Instantaneous Power", "Instantaneous Power Consumption", null, "Power (mW)",
|
||||
new String[] { "LPM", "CPU", "Radio listen", "Radio transmit" }) {
|
||||
protected void addSensorData(SensorData data) {
|
||||
Node node = data.getNode();
|
||||
String nodeName = node.getName();
|
||||
dataset.addValue(data.getLPMPower(), categories[0], nodeName);
|
||||
dataset.addValue(data.getCPUPower(), categories[1], nodeName);
|
||||
dataset.addValue(data.getListenPower(), categories[2], nodeName);
|
||||
dataset.addValue(data.getTransmitPower(), categories[3], nodeName);
|
||||
ValueAxis axis = chart.getCategoryPlot().getRangeAxis();
|
||||
axis.setLowerBound(0.0);
|
||||
axis.setUpperBound(75.0);
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Power History", "Historical Power Consumption", "Time", "mW") {
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getAveragePower();
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Temperature", "Temperature", "Time", "Celsius") {
|
||||
{
|
||||
chart.getXYPlot().getRangeAxis().setStandardTickUnits(NumberAxis.createIntegerTickUnits());
|
||||
setRangeTick(5);
|
||||
setRangeMinimumSize(10.0);
|
||||
setGlobalRange(true);
|
||||
}
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getTemperature();
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Relative Humidity", "Humidity", "Time", "%") {
|
||||
{
|
||||
chart.getXYPlot().getRangeAxis().setRange(0.0, 100.0);
|
||||
}
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getHumidity();
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Light 1", "Light 1", "Time", "-") {
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getLight1();
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Light 2", "Light 2", "Time", "-") {
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getLight2();
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Network Hops", "Network Hops", "Time", "Hops") {
|
||||
{
|
||||
ValueAxis axis = chart.getXYPlot().getRangeAxis();
|
||||
axis.setLowerBound(0.0);
|
||||
axis.setUpperBound(4.0);
|
||||
axis.setStandardTickUnits(NumberAxis.createIntegerTickUnits());
|
||||
}
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getValue(SensorData.HOPS);
|
||||
}
|
||||
},
|
||||
new BarChartPanel(this, "Network Hops", "Network Hops", null, "Hops",
|
||||
new String[] { "Hops" }) {
|
||||
{
|
||||
chart.getCategoryPlot().getRangeAxis().setStandardTickUnits(NumberAxis.createIntegerTickUnits());
|
||||
}
|
||||
protected void addSensorData(SensorData data) {
|
||||
dataset.addValue(data.getValue(SensorData.HOPS), categories[0], data.getNode().getName());
|
||||
}
|
||||
},
|
||||
new TimeChartPanel(this, "Latency", "Latency", "Time", "Milliseconds") {
|
||||
protected double getSensorDataValue(SensorData data) {
|
||||
return data.getLatency();
|
||||
}
|
||||
},
|
||||
serialConsole
|
||||
};
|
||||
for (int i = 0, n = visualizers.length; i < n; i++) {
|
||||
mainPanel.add(visualizers[i].getTitle(), visualizers[i].getPanel());
|
||||
}
|
||||
window.getContentPane().add(mainPanel, BorderLayout.CENTER);
|
||||
|
||||
// Setup menu
|
||||
JMenuBar menuBar = new JMenuBar();
|
||||
|
||||
JMenu fileMenu = new JMenu("File");
|
||||
fileMenu.setMnemonic(KeyEvent.VK_F);
|
||||
menuBar.add(fileMenu);
|
||||
|
||||
final JMenuItem clearMapItem = new JMenuItem("Remove Map Background");
|
||||
clearMapItem.addActionListener(new ActionListener() {
|
||||
|
||||
public void actionPerformed(ActionEvent e) {
|
||||
mapPanel.setMapBackground(null);
|
||||
clearMapItem.setEnabled(false);
|
||||
configTable.remove("collect.mapimage");
|
||||
}
|
||||
|
||||
});
|
||||
clearMapItem.setEnabled(mapPanel.getMapBackground() != null);
|
||||
|
||||
JMenuItem item = new JMenuItem("Select Map Background...");
|
||||
item.addActionListener(new ActionListener() {
|
||||
|
||||
public void actionPerformed(ActionEvent e) {
|
||||
if (fileChooser == null) {
|
||||
fileChooser = new JFileChooser();
|
||||
int reply = fileChooser.showOpenDialog(window);
|
||||
if (reply == JFileChooser.APPROVE_OPTION) {
|
||||
File file = fileChooser.getSelectedFile();
|
||||
String name = file.getAbsolutePath();
|
||||
configTable.put("collect.mapimage", name);
|
||||
if (!mapPanel.setMapBackground(file.getAbsolutePath())) {
|
||||
JOptionPane.showMessageDialog(window, "Failed to set background image", "Error", JOptionPane.ERROR_MESSAGE);
|
||||
}
|
||||
clearMapItem.setEnabled(mapPanel.getMapBackground() != null);
|
||||
saveConfig(configTable, configFile);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
});
|
||||
fileMenu.add(item);
|
||||
fileMenu.add(clearMapItem);
|
||||
|
||||
fileMenu.addSeparator();
|
||||
item = new JMenuItem("Clear Sensor Data...");
|
||||
item.addActionListener(new ActionListener() {
|
||||
|
||||
public void actionPerformed(ActionEvent e) {
|
||||
int reply = JOptionPane.showConfirmDialog(window, "Also clear the sensor data log file?");
|
||||
if (reply == JOptionPane.YES_OPTION) {
|
||||
// Clear data from both memory and sensor log file
|
||||
clearSensorDataLog();
|
||||
clearSensorData();
|
||||
} else if (reply == JOptionPane.NO_OPTION) {
|
||||
// Only clear data from memory
|
||||
clearSensorData();
|
||||
}
|
||||
}
|
||||
|
||||
});
|
||||
fileMenu.add(item);
|
||||
|
||||
fileMenu.addSeparator();
|
||||
item = new JMenuItem("Exit", KeyEvent.VK_X);
|
||||
item.addActionListener(new ActionListener() {
|
||||
|
||||
public void actionPerformed(ActionEvent e) {
|
||||
exit();
|
||||
}
|
||||
|
||||
});
|
||||
fileMenu.add(item);
|
||||
|
||||
JMenu toolsMenu = new JMenu("Tools");
|
||||
toolsMenu.setMnemonic(KeyEvent.VK_T);
|
||||
menuBar.add(toolsMenu);
|
||||
|
||||
final JCheckBoxMenuItem scrollItem = new JCheckBoxMenuItem("Scroll Layout");
|
||||
scrollItem.addActionListener(new ActionListener() {
|
||||
|
||||
public void actionPerformed(ActionEvent e) {
|
||||
if (scrollItem.getState()) {
|
||||
mainPanel.setTabLayoutPolicy(JTabbedPane.SCROLL_TAB_LAYOUT);
|
||||
} else {
|
||||
mainPanel.setTabLayoutPolicy(JTabbedPane.WRAP_TAB_LAYOUT);
|
||||
}
|
||||
}
|
||||
|
||||
});
|
||||
toolsMenu.add(scrollItem);
|
||||
|
||||
window.setJMenuBar(menuBar);
|
||||
window.pack();
|
||||
|
||||
String bounds = configTable.getProperty("collect.bounds");
|
||||
if (bounds != null) {
|
||||
String[] b = bounds.split(",");
|
||||
if (b.length == 4) {
|
||||
window.setBounds(Integer.parseInt(b[0]), Integer.parseInt(b[1]),
|
||||
Integer.parseInt(b[2]), Integer.parseInt(b[3]));
|
||||
}
|
||||
}
|
||||
|
||||
for(Object key: configTable.keySet()) {
|
||||
String property = key.toString();
|
||||
if (!property.startsWith("collect")) {
|
||||
getNode(property, true);
|
||||
}
|
||||
}
|
||||
initSensorData();
|
||||
|
||||
SwingUtilities.invokeLater(new Runnable() {
|
||||
public void run() {
|
||||
window.setVisible(true);
|
||||
}
|
||||
});
|
||||
|
||||
if (comPort == null) {
|
||||
comPort = MoteFinder.selectComPort(window);
|
||||
// if (comPort == null) {
|
||||
// exit();
|
||||
// }
|
||||
}
|
||||
serialConnection = new SerialConnection() {
|
||||
|
||||
private boolean hasOpened;
|
||||
private boolean hasSentInit;
|
||||
|
||||
@Override
|
||||
protected void serialOpened() {
|
||||
serialConsole.addSerialData("*** Serial console listening on port: " + getComPort() + " ***");
|
||||
hasOpened = true;
|
||||
// Remember the last selected serial port
|
||||
configTable.put("collect.serialport", getComPort());
|
||||
setSystemMessage("connected to " + getComPort());
|
||||
|
||||
// Send any initial commands
|
||||
if (!hasSentInit) {
|
||||
hasSentInit = true;
|
||||
|
||||
String initScript = config.getProperty("init.script", INIT_SCRIPT);
|
||||
if (initScript != null) {
|
||||
// Wait a short time before running the init script
|
||||
sleep(3000);
|
||||
runScript(initScript);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void serialClosed() {
|
||||
String comPort = getComPort();
|
||||
String prefix;
|
||||
if (hasOpened) {
|
||||
serialConsole.addSerialData("*** Serial connection terminated ***");
|
||||
prefix = "Serial connection terminated.\n";
|
||||
hasOpened = false;
|
||||
setSystemMessage("not connected");
|
||||
} else {
|
||||
prefix = "Failed to connect to " + getComPort() + '\n';
|
||||
}
|
||||
String options[] = {"Retry", "Search for connected nodes", "Cancel"};
|
||||
int value = JOptionPane.showOptionDialog(window,
|
||||
prefix + "Do you want to retry or search for connected nodes?",
|
||||
"Reconnect to serial port?",
|
||||
JOptionPane.YES_NO_OPTION, JOptionPane.WARNING_MESSAGE,
|
||||
null, options, options[0]);
|
||||
if (value == JOptionPane.CLOSED_OPTION || value == 2) {
|
||||
// exit();
|
||||
} else {
|
||||
if (value == 1) {
|
||||
// Select new serial port
|
||||
comPort = MoteFinder.selectComPort(window);
|
||||
if (comPort == null) {
|
||||
// exit();
|
||||
}
|
||||
}
|
||||
// Try to open com port again
|
||||
if (comPort != null) {
|
||||
open(comPort);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void serialData(String line) {
|
||||
parseIncomingLine(line);
|
||||
}
|
||||
|
||||
};
|
||||
serialConnection.open(comPort);
|
||||
}
|
||||
|
||||
private void exit() {
|
||||
/* TODO Clean up resources */
|
||||
if (configFile != null) {
|
||||
configTable.setProperty("collect.bounds", "" + window.getX() + ',' + window.getY() + ',' + window.getWidth() + ',' + window.getHeight());
|
||||
saveConfig(configTable, configFile);
|
||||
}
|
||||
if (serialConnection != null) {
|
||||
serialConnection.close();
|
||||
}
|
||||
System.exit(0);
|
||||
}
|
||||
|
||||
private void sleep(long delay) {
|
||||
try {
|
||||
Thread.sleep(delay);
|
||||
} catch (InterruptedException e1) {
|
||||
// Ignore
|
||||
}
|
||||
}
|
||||
|
||||
private boolean runScript(String script) {
|
||||
try {
|
||||
BufferedReader in = new BufferedReader(new FileReader(script));
|
||||
String line;
|
||||
while ((line = in.readLine()) != null) {
|
||||
if (line.length() == 0 || line.charAt(0) == '#') {
|
||||
// Ignore empty lines and comments
|
||||
} else if (line.startsWith("echo ")) {
|
||||
line = line.substring(5).trim();
|
||||
if (line.indexOf('%') >= 0) {
|
||||
line = line.replace("%TIME%", "" + (System.currentTimeMillis() / 1000));
|
||||
}
|
||||
sendToNode(line);
|
||||
} else if (line.startsWith("sleep ")) {
|
||||
long delay = Integer.parseInt(line.substring(6).trim());
|
||||
Thread.sleep(delay * 1000);
|
||||
} else {
|
||||
System.err.println("Unknown script comand: " + line);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
in.close();
|
||||
return true;
|
||||
} catch (FileNotFoundException e) {
|
||||
return false;
|
||||
} catch (Exception e) {
|
||||
System.err.println("Failed to run script: " + script);
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
public String getConfig(String property) {
|
||||
return getConfig(property, null);
|
||||
}
|
||||
|
||||
public String getConfig(String property, String defaultValue) {
|
||||
return configTable.getProperty(property, config.getProperty(property, defaultValue));
|
||||
}
|
||||
|
||||
protected void setSystemMessage(final String message) {
|
||||
SwingUtilities.invokeLater(new Runnable() {
|
||||
|
||||
public void run() {
|
||||
if (message == null) {
|
||||
window.setTitle(WINDOW_TITLE);
|
||||
} else {
|
||||
window.setTitle(WINDOW_TITLE + " (" + message + ')');
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// Node Handling
|
||||
// -------------------------------------------------------------------
|
||||
|
||||
public synchronized Node[] getNodes() {
|
||||
if (nodeCache == null) {
|
||||
Node[] tmp = nodeTable.values().toArray(new Node[nodeTable.size()]);
|
||||
Arrays.sort(tmp);
|
||||
nodeCache = tmp;
|
||||
}
|
||||
return nodeCache;
|
||||
}
|
||||
|
||||
public Node addNode(String nodeID) {
|
||||
return getNode(nodeID, true);
|
||||
}
|
||||
|
||||
private Node getNode(final String nodeID, boolean notify) {
|
||||
Node node = nodeTable.get(nodeID);
|
||||
if (node == null) {
|
||||
node = new Node(nodeID);
|
||||
nodeTable.put(nodeID, node);
|
||||
updateNodeLocation(node);
|
||||
|
||||
synchronized (this) {
|
||||
nodeCache = null;
|
||||
}
|
||||
|
||||
if (notify) {
|
||||
final Node newNode = node;
|
||||
SwingUtilities.invokeLater(new Runnable() {
|
||||
public void run() {
|
||||
// Insert the node sorted by name
|
||||
String nodeName = newNode.getName();
|
||||
boolean added = false;
|
||||
for (int i = 0, n = nodeModel.size(); i < n; i++) {
|
||||
int cmp = nodeName.compareTo(((Node) nodeModel.get(i)).getName());
|
||||
if (cmp < 0) {
|
||||
nodeModel.insertElementAt(newNode, i);
|
||||
added = true;
|
||||
break;
|
||||
} else if (cmp == 0) {
|
||||
// node already added
|
||||
added = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!added) {
|
||||
nodeModel.addElement(newNode);
|
||||
}
|
||||
if (visualizers != null) {
|
||||
for (int i = 0, n = visualizers.length; i < n; i++) {
|
||||
visualizers[i].nodeAdded(newNode);
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
return node;
|
||||
}
|
||||
|
||||
public void selectNodes(Node[] nodes) {
|
||||
selectNodes(nodes, true);
|
||||
}
|
||||
|
||||
private void selectNodes(Node[] nodes, boolean updateList) {
|
||||
if (nodes != selectedNodes) {
|
||||
selectedNodes = nodes;
|
||||
if (updateList) {
|
||||
nodeList.clearSelection();
|
||||
if (selectedNodes != null) {
|
||||
for (int i = 0, n = selectedNodes.length; i < n; i++) {
|
||||
int index = nodeModel.indexOf(selectedNodes[i]);
|
||||
if (index >= 0) {
|
||||
nodeList.addSelectionInterval(index, index);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if (visualizers != null) {
|
||||
for (int i = 0, n = visualizers.length; i < n; i++) {
|
||||
visualizers[i].nodesSelected(nodes);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// Node location handling
|
||||
// -------------------------------------------------------------------
|
||||
|
||||
public boolean updateNodeLocation(Node node) {
|
||||
String id = node.getID();
|
||||
if (node.hasLocation()) {
|
||||
String location = "" + node.getX() + ',' + node.getY();
|
||||
if (!location.equals(configTable.get(id))) {
|
||||
configTable.put(id, location);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
String location = configTable.getProperty(id);
|
||||
if (location != null) {
|
||||
try {
|
||||
String[] pos = location.split(",");
|
||||
node.setLocation(Integer.parseInt(pos[0].trim()),
|
||||
Integer.parseInt(pos[1].trim()));
|
||||
return true;
|
||||
} catch (Exception e) {
|
||||
System.err.println("could not parse node location: " + location);
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
private boolean loadConfig(Properties properties, String configFile) {
|
||||
try {
|
||||
BufferedInputStream input =
|
||||
new BufferedInputStream(new FileInputStream(configFile));
|
||||
try {
|
||||
properties.load(input);
|
||||
} finally {
|
||||
input.close();
|
||||
}
|
||||
return true;
|
||||
} catch (FileNotFoundException e) {
|
||||
// No configuration file exists.
|
||||
} catch (IOException e) {
|
||||
System.err.println("Failed to read configuration file: " + configFile);
|
||||
e.printStackTrace();
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
private void saveConfig(Properties properties, String configFile) {
|
||||
try {
|
||||
File fp = new File(configFile);
|
||||
if (fp.exists()) {
|
||||
File targetFp = new File(configFile + ".bak");
|
||||
if (targetFp.exists()) {
|
||||
targetFp.delete();
|
||||
}
|
||||
fp.renameTo(targetFp);
|
||||
}
|
||||
FileOutputStream output = new FileOutputStream(configFile);
|
||||
try {
|
||||
properties.store(output, "Configuration for Collect");
|
||||
} finally {
|
||||
output.close();
|
||||
}
|
||||
} catch (IOException e) {
|
||||
System.err.println("failed to save configuration to " + configFile);
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// Serial communication
|
||||
// -------------------------------------------------------------------
|
||||
|
||||
public boolean sendToNode(String data) {
|
||||
if (serialConnection != null && serialConnection.isOpen()) {
|
||||
serialConsole.addSerialData("SEND: " + data);
|
||||
serialConnection.writeSerialData(data);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
protected void parseIncomingLine(String line) {
|
||||
SensorData sensorData = SensorData.parseSensorData(this, line);
|
||||
if (sensorData != null) {
|
||||
// Sensor data received
|
||||
handleSensorData(sensorData);
|
||||
return;
|
||||
}
|
||||
System.out.println("SERIAL: " + line);
|
||||
serialConsole.addSerialData(line);
|
||||
}
|
||||
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// SensorData handling
|
||||
// -------------------------------------------------------------------
|
||||
|
||||
public int getSensorDataCount() {
|
||||
return sensorDataList.size();
|
||||
}
|
||||
|
||||
public SensorData getSensorData(int i) {
|
||||
return sensorDataList.get(i);
|
||||
}
|
||||
|
||||
private void handleSensorData(final SensorData sensorData) {
|
||||
System.out.println("SENSOR DATA: " + sensorData);
|
||||
if (sensorData.getNode().addSensorData(sensorData)) {
|
||||
sensorDataList.add(sensorData);
|
||||
saveSensorData(sensorData);
|
||||
handleLinks(sensorData);
|
||||
if (visualizers != null) {
|
||||
SwingUtilities.invokeLater(new Runnable() {
|
||||
public void run() {
|
||||
for (int i = 0, n = visualizers.length; i < n; i++) {
|
||||
visualizers[i].nodeDataReceived(sensorData);
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void handleLinks(SensorData sensorData) {
|
||||
String nodeID = sensorData.getBestNeighborID();
|
||||
if (nodeID != null) {
|
||||
Node neighbor = addNode(nodeID);
|
||||
Node source = sensorData.getNode();
|
||||
Link link = source.getLink(neighbor);
|
||||
link.setETX(sensorData.getBestNeighborETX());
|
||||
link.setLastActive(sensorData.getTime());
|
||||
}
|
||||
}
|
||||
|
||||
private void initSensorData() {
|
||||
loadSensorData(SENSORDATA_FILE);
|
||||
}
|
||||
|
||||
private boolean loadSensorData(String filename) {
|
||||
File fp = new File(filename);
|
||||
if (fp.exists() && fp.canRead()) {
|
||||
BufferedReader in = null;
|
||||
try {
|
||||
in = new BufferedReader(new FileReader(fp));
|
||||
String line;
|
||||
int no = 0;
|
||||
while ((line = in.readLine()) != null) {
|
||||
no++;
|
||||
if (line.length() == 0 || line.charAt(0) == '#') {
|
||||
// Ignore empty lines and comments
|
||||
} else {
|
||||
SensorData data = SensorData.parseSensorData(this, line);
|
||||
if (data != null) {
|
||||
if (data.getNode().addSensorData(data)) {
|
||||
sensorDataList.add(data);
|
||||
handleLinks(data);
|
||||
}
|
||||
} else {
|
||||
// TODO exit here?
|
||||
System.err.println("Failed to parse sensor data from log line " + no + ": " + line);
|
||||
}
|
||||
}
|
||||
}
|
||||
in.close();
|
||||
} catch (IOException e) {
|
||||
System.err.println("Failed to read sensor data log from " + fp.getAbsolutePath());
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
private void saveSensorData(SensorData data) {
|
||||
PrintWriter output = this.sensorDataOutput;
|
||||
if (output == null) {
|
||||
try {
|
||||
output = sensorDataOutput = new PrintWriter(new FileWriter(SENSORDATA_FILE, true));
|
||||
} catch (IOException e) {
|
||||
System.err.println("Failed to add sensor data to log '" + SENSORDATA_FILE + '\'');
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
if (output != null) {
|
||||
output.println(data.toString());
|
||||
}
|
||||
}
|
||||
|
||||
private void clearSensorData() {
|
||||
sensorDataList.clear();
|
||||
Node[] nodes = getNodes();
|
||||
if (nodes != null) {
|
||||
for(Node node : nodes) {
|
||||
node.removeAllSensorData();
|
||||
}
|
||||
}
|
||||
if (visualizers != null) {
|
||||
for(Visualizer v : visualizers) {
|
||||
v.clearNodeData();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void clearSensorDataLog() {
|
||||
PrintWriter output = this.sensorDataOutput;
|
||||
if (output != null) {
|
||||
output.close();
|
||||
}
|
||||
// Remove the sensor data log
|
||||
new File(SENSORDATA_FILE).delete();
|
||||
this.sensorDataOutput = null;
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// Main
|
||||
// -------------------------------------------------------------------
|
||||
|
||||
public static void main(String[] args) {
|
||||
String comPort = null;
|
||||
if (args.length > 0) {
|
||||
if (args.length > 1 || args[0].startsWith("-h")) {
|
||||
System.err.println("Usage: java CollectServer COMPORT");
|
||||
System.exit(1);
|
||||
}
|
||||
comPort = args[0];
|
||||
}
|
||||
new CollectServer(comPort);
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue