osd-contiki/tools/cooja/apps/powertracker/java/PowerTracker.java

525 lines
16 KiB
Java
Raw Normal View History

/*
* Copyright (c) 2012, 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.
*/
import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Component;
import java.awt.Toolkit;
import java.awt.datatransfer.Clipboard;
import java.awt.datatransfer.StringSelection;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseEvent;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Observable;
import java.util.Observer;
import javax.swing.AbstractAction;
import javax.swing.Action;
import javax.swing.Box;
import javax.swing.JButton;
import javax.swing.JScrollPane;
import javax.swing.JTable;
import javax.swing.Timer;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.DefaultTableCellRenderer;
import org.apache.log4j.Logger;
import org.jdom.Element;
import se.sics.cooja.ClassDescription;
import se.sics.cooja.GUI;
import se.sics.cooja.Mote;
import se.sics.cooja.PluginType;
import se.sics.cooja.SimEventCentral.MoteCountListener;
import se.sics.cooja.Simulation;
import se.sics.cooja.VisPlugin;
import se.sics.cooja.interfaces.Radio;
/**
* Tracks radio events to sum up transmission, reception, and radio on times.
* This plugin can be run without visualization, i.e. from a Contiki test.
*
* @author Fredrik Osterlind, Adam Dunkels
*/
@ClassDescription("PowerTracker")
@PluginType(PluginType.SIM_PLUGIN)
public class PowerTracker extends VisPlugin {
private static Logger logger = Logger.getLogger(PowerTracker.class);
private static final int POWERTRACKER_UPDATE_INTERVAL = 100; /* ms */
private static final int COLUMN_MOTE = 0;
private static final int COLUMN_RADIOON = 1;
private static final int COLUMN_RADIOTX = 2;
private static final int COLUMN_RADIORX = 3;
private Simulation simulation;
private MoteCountListener moteCountListener;
private ArrayList<MoteTracker> moteTrackers = new ArrayList<MoteTracker>();
private JTable table;
private int tableMaxRadioOnIndex = -1;
public PowerTracker(final Simulation simulation, final GUI gui) {
super("PowerTracker", gui, false);
this.simulation = simulation;
/* Automatically add/delete motes */
simulation.getEventCentral().addMoteCountListener(moteCountListener = new MoteCountListener() {
public void moteWasAdded(Mote mote) {
addMote(mote);
table.invalidate();
table.repaint();
}
public void moteWasRemoved(Mote mote) {
removeMote(mote);
table.invalidate();
table.repaint();
}
});
for (Mote m: simulation.getMotes()) {
addMote(m);
}
if (!GUI.isVisualized()) {
return;
}
AbstractTableModel model = new AbstractTableModel() {
public int getRowCount() {
return moteTrackers.size()+1;
}
public int getColumnCount() {
return 4;
}
public String getColumnName(int col) {
if (col == COLUMN_MOTE) {
return "Mote";
}
if (col == COLUMN_RADIOON) {
return "Radio on (%)";
}
if (col == COLUMN_RADIOTX) {
return "Radio TX (%)";
}
if (col == COLUMN_RADIORX) {
return "Radio RX (%)";
}
return null;
}
public Object getValueAt(int rowIndex, int col) {
if (rowIndex < 0 || rowIndex >= moteTrackers.size()+1) {
return null;
}
if (rowIndex == moteTrackers.size()) {
/* Average */
long radioOn = 0;
long radioTx = 0;
long radioRx = 0;
long duration = 0;
for (MoteTracker mt: moteTrackers) {
radioOn += mt.radioOn;
radioTx += mt.radioTx;
radioRx += mt.radioRx;
duration += mt.duration;
}
if (col == COLUMN_MOTE) {
return "AVERAGE";
}
if (col == COLUMN_RADIOON) {
return String.format("%2.2f%%", 100.0*radioOn/duration);
}
if (col == COLUMN_RADIOTX) {
return String.format("%2.2f%%", 100.0*radioTx/duration);
}
if (col == COLUMN_RADIORX) {
return String.format("%2.2f%%", 100.0*radioRx/duration);
}
return null;
}
MoteTracker rule = moteTrackers.get(rowIndex);
if (col == COLUMN_MOTE) {
return rule.mote.toString();
}
if (col == COLUMN_RADIOON) {
return String.format("%2.2f%%", 100.0*rule.getRadioOnRatio());
}
if (col == COLUMN_RADIOTX) {
return String.format("%2.2f%%", 100.0*rule.getRadioTxRatio());
}
if (col == COLUMN_RADIORX) {
return String.format("%2.2f%%", 100.0*rule.getRadioRxRatio());
}
return null;
}
};
table = new JTable(model) {
public String getToolTipText(MouseEvent e) {
java.awt.Point p = e.getPoint();
int rowIndex = table.rowAtPoint(p);
if (rowIndex < 0 || rowIndex >= moteTrackers.size()) {
return null;
}
MoteTracker mt = moteTrackers.get(rowIndex);
return "<html><pre>" + mt.toString() + "</html>";
}
};
table.setDefaultRenderer(Object.class, new DefaultTableCellRenderer() {
public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
Component c = super.getTableCellRendererComponent(table, value, isSelected, hasFocus, row, column);
if (row == tableMaxRadioOnIndex) {
setBackground(isSelected ? table.getSelectionBackground():Color.RED);
setForeground(isSelected ? table.getSelectionForeground():Color.WHITE);
} else {
setBackground(isSelected ? table.getSelectionBackground():table.getBackground());
setForeground(isSelected ? table.getSelectionForeground():table.getForeground());
}
return c;
}
});
Box control = Box.createHorizontalBox();
control.add(Box.createHorizontalGlue());
control.add(new JButton(printAction));
control.add(new JButton(resetAction));
this.getContentPane().add(BorderLayout.CENTER, new JScrollPane(table));
this.getContentPane().add(BorderLayout.SOUTH, control);
setSize(400, 400);
repaintTimer.start();
}
private Action resetAction = new AbstractAction("Reset") {
public void actionPerformed(ActionEvent e) {
Runnable r = new Runnable() {
public void run() {
reset();
}
};
if (simulation.isRunning()) {
simulation.invokeSimulationThread(r);
} else {
r.run();
}
}
};
private Action printAction = new AbstractAction("Print to console/Copy to clipboard") {
public void actionPerformed(ActionEvent e) {
String output = radioStatistics(true, true, false);
logger.info("PowerTracker output:\n\n" + output);
Clipboard clipboard = Toolkit.getDefaultToolkit().getSystemClipboard();
StringSelection stringSelection = new StringSelection(output);
clipboard.setContents(stringSelection, null);
}
};
public String radioStatistics() {
return radioStatistics(true, true, false);
}
public String radioStatistics(boolean radioHW, boolean radioRXTX, boolean onlyAverage) {
StringBuilder sb = new StringBuilder();
/* Average */
long radioOn = 0;
long radioTx = 0;
long radioRx = 0;
long radioInterfered = 0;
long duration = 0;
for (MoteTracker mt: moteTrackers) {
radioOn += mt.radioOn;
radioTx += mt.radioTx;
radioRx += mt.radioRx;
radioInterfered += mt.radioInterfered;
duration += mt.duration;
}
if (radioHW) {
sb.append(String.format("AVG" + " ON " + (radioOn + " us ") + "%2.2f %%", 100.0*radioOn/duration) + "\n");
}
if (radioRXTX) {
sb.append(String.format("AVG" + " TX " + (radioTx + " us ") + "%2.2f %%", 100.0*radioTx/duration) + "\n");
sb.append(String.format("AVG" + " RX " + (radioRx + " us ") + "%2.2f %%", 100.0*radioRx/duration) + "\n");
sb.append(String.format("AVG" + " INT " + (radioInterfered + " us ") + "%2.2f %%", 100.0*radioInterfered/duration) + "\n");
}
if (onlyAverage) {
return sb.toString();
}
for (MoteTracker mt: moteTrackers) {
sb.append(mt.toString(radioHW, radioRXTX));
}
return sb.toString();
}
private static class MoteTracker implements Observer {
/* last radio state */
private boolean radioWasOn;
private RadioState lastRadioState;
private long lastUpdateTime;
/* accumulating radio state durations */
long duration = 0;
long radioOn = 0;
long radioTx = 0;
long radioRx = 0;
long radioInterfered = 0;
private Simulation simulation;
private Mote mote;
private Radio radio;
public MoteTracker(Mote mote) {
this.simulation = mote.getSimulation();
this.mote = mote;
this.radio = mote.getInterfaces().getRadio();
radioWasOn = radio.isReceiverOn();
if (radio.isTransmitting()) {
lastRadioState = RadioState.TRANSMITTING;
} else if (radio.isReceiving()) {
lastRadioState = RadioState.RECEIVING;
} else if (radio.isInterfered()){
lastRadioState = RadioState.INTERFERED;
} else {
lastRadioState = RadioState.IDLE;
}
lastUpdateTime = simulation.getSimulationTime();
radio.addObserver(this);
}
public void update(Observable o, Object arg) {
update();
}
public void update() {
long now = simulation.getSimulationTime();
accumulateDuration(now - lastUpdateTime);
/* Radio on/off */
if (radioWasOn) {
accumulateRadioOn(now - lastUpdateTime);
}
/* Radio tx/rx */
if (lastRadioState == RadioState.TRANSMITTING) {
accumulateRadioTx(now - lastUpdateTime);
} else if (lastRadioState == RadioState.RECEIVING) {
accumulateRadioRx(now - lastUpdateTime);
} else if (lastRadioState == RadioState.INTERFERED) {
accumulateRadioIntefered(now - lastUpdateTime);
}
/* Await next radio event */
if (radio.isTransmitting()) {
lastRadioState = RadioState.TRANSMITTING;
} else if (!radio.isReceiverOn()) {
lastRadioState = RadioState.IDLE;
} else if (radio.isInterfered()) {
lastRadioState = RadioState.INTERFERED;
} else if (radio.isReceiving()) {
lastRadioState = RadioState.RECEIVING;
} else {
lastRadioState = RadioState.IDLE;
}
radioWasOn = radio.isReceiverOn();
lastUpdateTime = now;
}
protected void accumulateDuration(long t) {
duration += t;
}
protected void accumulateRadioOn(long t) {
radioOn += t;
}
protected void accumulateRadioTx(long t) {
radioTx += t;
}
protected void accumulateRadioRx(long t) {
radioRx += t;
}
protected void accumulateRadioIntefered(long t) {
radioInterfered += t;
}
protected double getRadioOnRatio() {
return 1.0*radioOn/duration;
}
protected double getRadioTxRatio() {
return 1.0*radioTx/duration;
}
protected double getRadioInterferedRatio() {
return 1.0*radioInterfered/duration;
}
protected double getRadioRxRatio() {
return 1.0*radioRx/duration;
}
public Mote getMote() {
return mote;
}
public void dispose() {
radio.deleteObserver(this);
radio = null;
mote = null;
}
public String toString() {
return toString(true, true);
}
public String toString(boolean radioHW, boolean radioRXTX) {
StringBuilder sb = new StringBuilder();
String moteString = mote.toString().replace(' ', '_');
sb.append(moteString + " MONITORED " + duration + " us\n");
if (radioHW) {
sb.append(String.format(moteString + " ON " + (radioOn + " us ") + "%2.2f %%", 100.0*getRadioOnRatio()) + "\n");
}
if (radioRXTX) {
sb.append(String.format(moteString + " TX " + (radioTx + " us ") + "%2.2f %%", 100.0*getRadioTxRatio()) + "\n");
sb.append(String.format(moteString + " RX " + (radioRx + " us ") + "%2.2f %%", 100.0*getRadioRxRatio()) + "\n");
sb.append(String.format(moteString + " INT " + (radioInterfered + " us ") + "%2.2f %%", 100.0*getRadioInterferedRatio()) + "\n");
}
return sb.toString();
}
}
private MoteTracker createMoteTracker(Mote mote) {
final Radio moteRadio = mote.getInterfaces().getRadio();
if (moteRadio == null) {
return null;
}
/* Radio observer */
MoteTracker tracker = new MoteTracker(mote);
tracker.update(null, null);
return tracker;
}
public void reset() {
while (moteTrackers.size() > 0) {
removeMote(moteTrackers.get(0).mote);
}
for (Mote m: simulation.getMotes()) {
addMote(m);
}
}
private void addMote(Mote mote) {
if (mote == null) {
return;
}
MoteTracker t = createMoteTracker(mote);
if (t != null) {
moteTrackers.add(t);
}
setTitle("PowerTracker: " + moteTrackers.size() + " motes");
}
private void removeMote(Mote mote) {
/* Remove mote tracker(s) */
MoteTracker[] trackers = moteTrackers.toArray(new MoteTracker[0]);
for (MoteTracker t: trackers) {
if (t.getMote() == mote) {
t.dispose();
moteTrackers.remove(t);
}
}
setTitle("PowerTracker: " + moteTrackers.size() + " motes");
}
public void closePlugin() {
/* Remove repaint timer */
repaintTimer.stop();
simulation.getEventCentral().removeMoteCountListener(moteCountListener);
/* Remove mote trackers */
for (Mote m: simulation.getMotes()) {
removeMote(m);
}
if (!moteTrackers.isEmpty()) {
logger.fatal("Mote observers not cleaned up correctly");
for (MoteTracker t: moteTrackers.toArray(new MoteTracker[0])) {
t.dispose();
}
}
}
public enum RadioState {
IDLE, RECEIVING, TRANSMITTING, INTERFERED
}
private Timer repaintTimer = new Timer(POWERTRACKER_UPDATE_INTERVAL, new ActionListener() {
public void actionPerformed(ActionEvent e) {
/* Identify max radio on */
double maxRadioOn = 0;
int maxRadioOnIndex = -1;
for (int i=0; i < moteTrackers.size(); i++) {
MoteTracker mt = moteTrackers.get(i);
if (mt.getRadioOnRatio() > maxRadioOn) {
maxRadioOn = mt.getRadioOnRatio();
maxRadioOnIndex = i;
}
}
if (maxRadioOnIndex >= 0) {
tableMaxRadioOnIndex = maxRadioOnIndex;
}
table.repaint();
}
});
public Collection<Element> getConfigXML() {
return null;
}
public boolean setConfigXML(Collection<Element> configXML, boolean visAvailable) {
return true;
}
}