def __init__(self, net=None, parent=None, fname=None):
QMainWindow.__init__(self)
self.ui = Ui_SimulationWindow()
self.ui.setupUi(self)
if fname:
self.set_title(fname)
# context menu
self.ui.nodeInspector.addAction(self.ui.actionCopyInspectorData)
self.ui.nodeInspector.addAction(self.ui.actionShowLocalizedSubclusters)
# callbacks
self.ui.actionCopyInspectorData.activated.connect(self.on_actionCopyInspectorData_triggered)
self.ui.actionShowLocalizedSubclusters.activated.connect(self.on_actionShowLocalizedSubclusters_triggered)
self.dpi = 72
# take size of networDisplayWidget
self.fig = Figure((700 / self.dpi, 731 / self.dpi), self.dpi, facecolor="0.9")
self.canvas = FigureCanvas(self.fig)
self.canvas.setParent(self.ui.networkDisplayWidget)
self.nav = NavigationToolbar(self.canvas, self.ui.networkDisplayWidget, coordinates=True)
self.nav.setGeometry(QRect(0, 0, 651, 36))
self.nav.setIconSize(QSize(24, 24))
self.axes = self.fig.add_subplot(111)
# matplotlib.org/api/figure_api.html#matplotlib.figure.SubplotParams
self.fig.subplots_adjust(left=0.03, right=0.99, top=0.92)
if net:
self.init_sim(net)
self.connect(self.ui.showNodes, SIGNAL("stateChanged(int)"), self.refresh_visibility)
self.connect(self.ui.showEdges, SIGNAL("stateChanged(int)"), self.refresh_visibility)
self.connect(self.ui.showMessages, SIGNAL("stateChanged(int)"), self.refresh_visibility)
self.connect(self.ui.showLabels, SIGNAL("stateChanged(int)"), self.refresh_visibility)
self.connect(self.ui.redrawNetworkButton, SIGNAL("clicked(bool)"), self.redraw)
self.connect(self.ui.treeGroupBox, SIGNAL("toggled(bool)"), self.refresh_visibility)
self.connect(self.ui.treeKey, SIGNAL("textEdited(QString)"), self.redraw)
self.connect(self.ui.propagationError, SIGNAL("toggled(bool)"), self.refresh_visibility)
self.connect(self.ui.locKey, SIGNAL("textEdited(QString)"), self.redraw)
# callbacks
self.ui.actionOpenNetwork.activated.connect(self.on_actionOpenNetwork_triggered)
self.ui.actionSaveNetwork.activated.connect(self.on_actionSaveNetwork_triggered)
self.ui.actionRun.activated.connect(self.on_actionRun_triggered)
self.ui.actionStep.activated.connect(self.on_actionStep_triggered)
self.ui.actionReset.activated.connect(self.on_actionReset_triggered)
self.canvas.mpl_connect("pick_event", self.on_pick)
class SimulationGui(QMainWindow):
def __init__(self, net=None, parent=None, fname=None):
QMainWindow.__init__(self)
self.ui = Ui_SimulationWindow()
self.ui.setupUi(self)
if fname:
self.set_title(fname)
# context menu
self.ui.nodeInspector.addAction(self.ui.actionCopyInspectorData)
self.ui.nodeInspector.addAction(self.ui.actionShowLocalizedSubclusters)
# callbacks
self.ui.actionCopyInspectorData.activated\
.connect(self.on_actionCopyInspectorData_triggered)
self.ui.actionShowLocalizedSubclusters.activated\
.connect(self.on_actionShowLocalizedSubclusters_triggered)
self.dpi = 72
# take size of networDisplayWidget
self.fig = Figure((700 / self.dpi, 731 / self.dpi), self.dpi,
facecolor='0.9')
self.canvas = FigureCanvas(self.fig)
self.canvas.setParent(self.ui.networkDisplayWidget)
self.nav = NavigationToolbar(self.canvas, self.ui.networkDisplayWidget,
coordinates=True)
self.nav.setGeometry(QRect(0, 0, 651, 36))
self.nav.setIconSize(QSize(24, 24))
self.axes = self.fig.add_subplot(111)
# matplotlib.org/api/figure_api.html#matplotlib.figure.SubplotParams
self.fig.subplots_adjust(left=0.03, right=0.99, top=0.92)
if net:
self.init_sim(net)
self.connect(self.ui.showNodes, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showEdges, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showMessages, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showLabels, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.redrawNetworkButton, SIGNAL('clicked(bool)'),
self.redraw)
self.connect(self.ui.treeGroupBox, SIGNAL('toggled(bool)'),
self.refresh_visibility)
self.connect(self.ui.treeKey, SIGNAL('textEdited(QString)'),
self.redraw)
self.connect(self.ui.propagationError, SIGNAL('toggled(bool)'),
self.refresh_visibility)
self.connect(self.ui.locKey, SIGNAL('textEdited(QString)'),
self.redraw)
# callbacks
self.ui.actionOpenNetwork.activated\
.connect(self.on_actionOpenNetwork_triggered)
self.ui.actionSaveNetwork.activated\
.connect(self.on_actionSaveNetwork_triggered)
self.ui.actionRun.activated.connect(self.on_actionRun_triggered)
self.ui.actionStep.activated.connect(self.on_actionStep_triggered)
self.ui.actionReset.activated.connect(self.on_actionReset_triggered)
self.canvas.mpl_connect('pick_event', self.on_pick)
def handleInspectorMenu(self, pos):
menu = QMenu()
menu.addAction('Add')
menu.addAction('Delete')
menu.exec_(QCursor.pos())
def init_sim(self, net):
self.net = net
self.sim = Simulation(net)
self.connect(self.sim, SIGNAL("redraw()"), self.redraw)
self.connect(self.sim, SIGNAL("updateLog(QString)"), self.update_log)
self.redraw()
def update_log(self, text):
""" Add item to list widget """
print "Add: " + text
self.ui.logListWidget.insertItem(0, text)
# self.ui.logListWidget.sortItems()
def redraw(self):
self.refresh_network_inspector()
self.draw_network()
self.reset_zoom()
self.refresh_visibility()
def draw_network(self, net=None, clear=True, subclusters=None,
drawMessages=True):
if not net:
net = self.net
currentAlgorithm = self.net.get_current_algorithm()
if clear:
self.axes.clear()
self.axes.imshow(net.environment.im, vmin=0, cmap='binary_r',
origin='lower')
self.draw_tree(str(self.ui.treeKey.text()), net)
self.draw_edges(net)
self.draw_propagation_errors(str(self.ui.locKey.text()), net)
if subclusters:
node_colors = self.get_node_colors(net, subclusters=subclusters)
else:
node_colors = self.get_node_colors(net, algorithm=currentAlgorithm)
self.node_collection = self.draw_nodes(net, node_colors)
if drawMessages:
self.draw_messages(net)
self.draw_labels(net)
self.drawnNet = net
step_text = ' (step %d)' % self.net.algorithmState['step'] \
if isinstance(currentAlgorithm, NodeAlgorithm) else ''
self.axes.set_title((currentAlgorithm.name
if currentAlgorithm else '') + step_text)
self.refresh_visibility()
# To save multiple figs of the simulation uncomment next two lines:
#self.fig.savefig('network-alg-%d-step-%d.png' %
# (self.net.algorithmState['index'], self.net.algorithmState['step']))
def draw_nodes(self, net=None, node_colors={}, node_radius={}):
if not net:
net = self.net
if type(node_colors) == str:
node_colors = {node: node_colors for node in net.nodes()}
nodeCircles = []
for n in net.nodes():
c = NodeCircle(tuple(net.pos[n]), node_radius.get(n, 8.0),
color=node_colors.get(n, 'r'),
ec='k', lw=1.0, ls='solid', picker=3)
nodeCircles.append(c)
node_collection = PatchCollection(nodeCircles, match_original=True)
node_collection.set_picker(3)
self.axes.add_collection(node_collection)
return node_collection
def get_node_colors(self, net, algorithm=None, subclusters=None,
drawLegend=True):
COLORS = 'rgbcmyw' * 100
node_colors = {}
if algorithm:
color_map = {}
if isinstance(algorithm, NodeAlgorithm):
for ind, status in enumerate(algorithm.STATUS.keys()):
if status == 'IDLE':
color_map.update({status: 'k'})
else:
color_map.update({status: COLORS[ind]})
if drawLegend:
proxy = []
labels = []
for status, color in color_map.items():
proxy.append(Circle((0, 0), radius=8.0,
color=color, ec='k',
lw=1.0, ls='solid'))
labels.append(status)
self.fig.legends = []
self.fig.legend(proxy, labels, loc=8,
prop={'size': '10.0'}, ncol=len(proxy),
title='Statuses for %s:'
% algorithm.name)
for n in net.nodes():
if n.status == '' or not n.status in color_map.keys():
node_colors[n] = 'r'
else:
node_colors[n] = color_map[n.status]
elif subclusters:
for i, sc in enumerate(subclusters):
for n in sc:
if n in node_colors:
node_colors[n] = 'k'
else:
node_colors[n] = COLORS[i]
return node_colors
def draw_edges(self, net=None):
if not net:
net = self.net
self.edge_collection = nx.draw_networkx_edges(net, net.pos, alpha=0.6,
edgelist=None,
ax=self.axes)
def draw_messages(self, net=None):
if not net:
net = self.net
self.messages = []
msgCircles = []
for node in net.nodes():
for msg in node.outbox:
# broadcast
if msg.destination is None:
for neighbor in net.adj[node].keys():
nbr_msg = msg.copy()
nbr_msg.destination = neighbor
c = MessageCircle(nbr_msg, net, 'out', 3.0, lw=0,
picker=3, zorder=3, color='b')
self.messages.append(nbr_msg)
msgCircles.append(c)
else:
c = MessageCircle(msg, net, 'out', 3.0, lw=0, picker=3,
zorder=3, color='b')
self.messages.append(msg)
msgCircles.append(c)
for msg in node.inbox:
c = MessageCircle(msg, net, 'in', 3.0, lw=0, picker=3,
zorder=3, color='g')
self.messages.append(msg)
msgCircles.append(c)
if self.messages:
self.message_collection = PatchCollection(msgCircles,
match_original=True)
self.message_collection.set_picker(3)
self.axes.add_collection(self.message_collection)
def draw_labels(self, net=None):
if not net:
net = self.net
label_pos = {}
for n in net.nodes():
label_pos[n] = net.pos[n].copy() + 10
self.label_collection = nx.draw_networkx_labels(net, label_pos,
labels=net.labels,
ax=self.axes)
def refresh_visibility(self):
try:
self.node_collection.set_visible(self.ui.showNodes.isChecked())
self.edge_collection.set_visible(self.ui.showEdges.isChecked())
for label in self.label_collection.values():
label.set_visible(self.ui.showLabels.isChecked())
self.tree_collection.set_visible(self.ui.treeGroupBox.isChecked())
self.ini_error_collection.set_visible(self.ui.propagationError\
.isChecked())
self.propagation_error_collection.set_visible(self.ui\
.propagationError\
.isChecked())
# sould be last, sometimes there are no messages
self.message_collection.set_visible(self.ui.showMessages\
.isChecked())
except AttributeError:
print 'Refresh visibility warning'
self.canvas.draw()
def draw_tree(self, treeKey, net=None):
"""
Show tree representation of network.
Attributes:
treeKey (str):
key in nodes memory (dictionary) where tree data is stored
storage format can be a list off tree neighbors or a dict:
{'parent': parent_node,
'children': [child_node1, child_node2 ...]}
"""
if not net:
net = self.net
treeNet = net.get_tree_net(treeKey)
if treeNet:
self.tree_collection = draw_networkx_edges(treeNet, treeNet.pos,
treeNet.edges(),
width=1.8, alpha=0.6,
ax=self.axes)
def draw_propagation_errors(self, locKey, net):
SCALE_FACTOR = 0.6
if not net:
net = self.net
if any([not locKey in node.memory for node in net.nodes()]):
self.propagation_error_collection = []
self.ini_error_collection = []
return
rms = {'iniRms': {}, 'stitchRms': {}}
for node in net.nodes():
rms['iniRms'][node] = get_rms(self.net.pos,
(node.memory['iniLocs']), True) * \
SCALE_FACTOR
rms['stitchRms'][node] = get_rms(self.net.pos, node.memory[locKey],
True) * SCALE_FACTOR
self.propagation_error_collection = \
self.draw_nodes(net=net, node_colors='g',
node_radius=rms['stitchRms'])
self.ini_error_collection = self.draw_nodes(net=net, node_colors='b',
node_radius=rms['iniRms'])
def reset_zoom(self):
self.axes.set_xlim((0, self.net.environment.im.shape[1]))
self.axes.set_ylim((0, self.net.environment.im.shape[0]))
def set_title(self, fname):
new = ' - '.join([str(self.windowTitle()).split(' - ')[0], str(fname)])
self.setWindowTitle(new)
def refresh_network_inspector(self):
niModel = DictionaryTreeModel(dic=self.net.get_dic())
self.ui.networkInspector.setModel(niModel)
self.ui.networkInspector.expandToDepth(0)
"""
Callbacks
"""
def on_actionRun_triggered(self):
self.ui.logListWidget.clear()
print 'running ...',
self.sim.stepping = True
self.sim.run_all()
def on_actionStep_triggered(self):
print 'next step ...',
self.sim.run(self.ui.stepSize.value())
def on_actionReset_triggered(self):
print 'reset ...',
self.sim.reset()
self.redraw()
def on_actionCopyInspectorData_triggered(self):
string = 'Node inspector data\n-------------------'
# raise()
for qModelIndex in self.ui.nodeInspector.selectedIndexes():
string += '\n' + qModelIndex.internalPointer().toString(' ')
clipboard = app.clipboard()
clipboard.setText(string)
event = QEvent(QEvent.Clipboard)
app.sendEvent(clipboard, event)
def on_actionShowLocalizedSubclusters_triggered(self):
if len(self.ui.nodeInspector.selectedIndexes()) == 1:
qModelIndex = self.ui.nodeInspector.selectedIndexes()[0]
treeItem = qModelIndex.internalPointer()
assert(isinstance(treeItem.itemDataValue, Positions))
estimated = deepcopy(treeItem.itemDataValue)
estimatedsub = estimated.subclusters[0]
# rotate, translate and optionally scale
# w.r.t. original positions (pos)
align_clusters(Positions.create(self.net.pos), estimated, True)
net = self.net.subnetwork(estimatedsub.keys(), pos=estimatedsub)
self.draw_network(net=net, drawMessages=False)
edge_pos = numpy.asarray([(self.net.pos[node], estimatedsub[node][:2])
for node in net])
error_collection = LineCollection(edge_pos, colors='r')
self.axes.add_collection(error_collection)
rms = get_rms(self.net.pos, estimated, scale=False)
self.update_log('rms = %.3f' % rms)
self.update_log('localized = %.2f%% (%d/%d)' %
(len(estimatedsub) * 1. / len(self.net.pos) * 100,
len(estimatedsub), len(self.net.pos)))
def on_actionSaveNetwork_triggered(self, *args):
default_filetype = 'gz'
start = datetime.now().strftime('%Y%m%d') + default_filetype
filters = ['Network pickle (*.gz)', 'All files (*)']
selectedFilter = 'Network pickle (gz)'
filters = ';;'.join(filters)
fname = QFileDialog.getSaveFileName(self, "Choose a filename",
start, filters, selectedFilter)[0]
if fname:
try:
write_pickle(self.net, fname)
except Exception, e:
QMessageBox.critical(
self, "Error saving file", str(e),
QMessageBox.Ok, QMessageBox.NoButton)
else:
self.set_title(fname)
def __init__(self, net=None, parent=None, fname=None):
QMainWindow.__init__(self)
self.ui = Ui_SimulationWindow()
self.ui.setupUi(self)
if fname:
self.set_title(fname)
# context menu
self.ui.nodeInspector.addAction(self.ui.actionCopyInspectorData)
self.ui.nodeInspector.addAction(self.ui.actionShowLocalizedSubclusters)
# callbacks
self.ui.actionCopyInspectorData.activated\
.connect(self.on_actionCopyInspectorData_triggered)
self.ui.actionShowLocalizedSubclusters.activated\
.connect(self.on_actionShowLocalizedSubclusters_triggered)
self.dpi = 72
# take size of networDisplayWidget
self.fig = Figure((700 / self.dpi, 731 / self.dpi),
self.dpi,
facecolor='0.9')
self.canvas = FigureCanvas(self.fig)
self.canvas.setParent(self.ui.networkDisplayWidget)
self.nav = NavigationToolbar(self.canvas,
self.ui.networkDisplayWidget,
coordinates=True)
self.nav.setGeometry(QRect(0, 0, 651, 36))
self.nav.setIconSize(QSize(24, 24))
self.axes = self.fig.add_subplot(111)
# matplotlib.org/api/figure_api.html#matplotlib.figure.SubplotParams
self.fig.subplots_adjust(left=0.03, right=0.99, top=0.92)
if net:
self.init_sim(net)
self.connect(self.ui.showNodes, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showEdges, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showMessages, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showLabels, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.redrawNetworkButton, SIGNAL('clicked(bool)'),
self.redraw)
self.connect(self.ui.treeGroupBox, SIGNAL('toggled(bool)'),
self.refresh_visibility)
self.connect(self.ui.treeKey, SIGNAL('textEdited(QString)'),
self.redraw)
self.connect(self.ui.propagationError, SIGNAL('toggled(bool)'),
self.refresh_visibility)
self.connect(self.ui.locKey, SIGNAL('textEdited(QString)'),
self.redraw)
# callbacks
self.ui.actionOpenNetwork.activated\
.connect(self.on_actionOpenNetwork_triggered)
self.ui.actionSaveNetwork.activated\
.connect(self.on_actionSaveNetwork_triggered)
self.ui.actionRun.activated.connect(self.on_actionRun_triggered)
self.ui.actionStep.activated.connect(self.on_actionStep_triggered)
self.ui.actionReset.activated.connect(self.on_actionReset_triggered)
self.canvas.mpl_connect('pick_event', self.on_pick)
class SimulationGui(QMainWindow):
def __init__(self, net=None, parent=None, fname=None):
QMainWindow.__init__(self)
self.ui = Ui_SimulationWindow()
self.ui.setupUi(self)
if fname:
self.set_title(fname)
# context menu
self.ui.nodeInspector.addAction(self.ui.actionCopyInspectorData)
self.ui.nodeInspector.addAction(self.ui.actionShowLocalizedSubclusters)
# callbacks
self.ui.actionCopyInspectorData.activated\
.connect(self.on_actionCopyInspectorData_triggered)
self.ui.actionShowLocalizedSubclusters.activated\
.connect(self.on_actionShowLocalizedSubclusters_triggered)
self.dpi = 72
# take size of networDisplayWidget
self.fig = Figure((700 / self.dpi, 731 / self.dpi),
self.dpi,
facecolor='0.9')
self.canvas = FigureCanvas(self.fig)
self.canvas.setParent(self.ui.networkDisplayWidget)
self.nav = NavigationToolbar(self.canvas,
self.ui.networkDisplayWidget,
coordinates=True)
self.nav.setGeometry(QRect(0, 0, 651, 36))
self.nav.setIconSize(QSize(24, 24))
self.axes = self.fig.add_subplot(111)
# matplotlib.org/api/figure_api.html#matplotlib.figure.SubplotParams
self.fig.subplots_adjust(left=0.03, right=0.99, top=0.92)
if net:
self.init_sim(net)
self.connect(self.ui.showNodes, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showEdges, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showMessages, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.showLabels, SIGNAL('stateChanged(int)'),
self.refresh_visibility)
self.connect(self.ui.redrawNetworkButton, SIGNAL('clicked(bool)'),
self.redraw)
self.connect(self.ui.treeGroupBox, SIGNAL('toggled(bool)'),
self.refresh_visibility)
self.connect(self.ui.treeKey, SIGNAL('textEdited(QString)'),
self.redraw)
self.connect(self.ui.propagationError, SIGNAL('toggled(bool)'),
self.refresh_visibility)
self.connect(self.ui.locKey, SIGNAL('textEdited(QString)'),
self.redraw)
# callbacks
self.ui.actionOpenNetwork.activated\
.connect(self.on_actionOpenNetwork_triggered)
self.ui.actionSaveNetwork.activated\
.connect(self.on_actionSaveNetwork_triggered)
self.ui.actionRun.activated.connect(self.on_actionRun_triggered)
self.ui.actionStep.activated.connect(self.on_actionStep_triggered)
self.ui.actionReset.activated.connect(self.on_actionReset_triggered)
self.canvas.mpl_connect('pick_event', self.on_pick)
def handleInspectorMenu(self, pos):
menu = QMenu()
menu.addAction('Add')
menu.addAction('Delete')
menu.exec_(QCursor.pos())
def init_sim(self, net):
self.net = net
self.sim = Simulation(net)
self.connect(self.sim, SIGNAL("redraw()"), self.redraw)
self.connect(self.sim, SIGNAL("updateLog(QString)"), self.update_log)
self.redraw()
def update_log(self, text):
""" Add item to list widget """
print "Add: " + text
self.ui.logListWidget.insertItem(0, text)
# self.ui.logListWidget.sortItems()
def redraw(self):
self.refresh_network_inspector()
self.draw_network()
self.reset_zoom()
self.refresh_visibility()
def draw_network(self,
net=None,
clear=True,
subclusters=None,
drawMessages=True):
if not net:
net = self.net
currentAlgorithm = self.net.get_current_algorithm()
if clear:
self.axes.clear()
self.axes.imshow(net.environment.im,
vmin=0,
cmap='binary_r',
origin='lower')
self.draw_tree(str(self.ui.treeKey.text()), net)
self.draw_edges(net)
self.draw_propagation_errors(str(self.ui.locKey.text()), net)
if subclusters:
node_colors = self.get_node_colors(net, subclusters=subclusters)
else:
node_colors = self.get_node_colors(net, algorithm=currentAlgorithm)
self.node_collection = self.draw_nodes(net, node_colors)
if drawMessages:
self.draw_messages(net)
self.draw_labels(net)
self.drawnNet = net
step_text = ' (step %d)' % self.net.algorithmState['step'] \
if isinstance(currentAlgorithm, NodeAlgorithm) else ''
self.axes.set_title(
(currentAlgorithm.name if currentAlgorithm else '') + step_text)
self.refresh_visibility()
# To save multiple figs of the simulation uncomment next two lines:
#self.fig.savefig('network-alg-%d-step-%d.png' %
# (self.net.algorithmState['index'], self.net.algorithmState['step']))
def draw_nodes(self, net=None, node_colors={}, node_radius={}):
if not net:
net = self.net
if type(node_colors) == str:
node_colors = {node: node_colors for node in net.nodes()}
nodeCircles = []
for n in net.nodes():
c = NodeCircle(tuple(net.pos[n]),
node_radius.get(n, 8.0),
color=node_colors.get(n, 'r'),
ec='k',
lw=1.0,
ls='solid',
picker=3)
nodeCircles.append(c)
node_collection = PatchCollection(nodeCircles, match_original=True)
node_collection.set_picker(3)
self.axes.add_collection(node_collection)
return node_collection
def get_node_colors(self,
net,
algorithm=None,
subclusters=None,
drawLegend=True):
COLORS = 'rgbcmyw' * 100
node_colors = {}
if algorithm:
color_map = {}
if isinstance(algorithm, NodeAlgorithm):
for ind, status in enumerate(algorithm.STATUS.keys()):
if status == 'IDLE':
color_map.update({status: 'k'})
else:
color_map.update({status: COLORS[ind]})
if drawLegend:
proxy = []
labels = []
for status, color in color_map.items():
proxy.append(
Circle((0, 0),
radius=8.0,
color=color,
ec='k',
lw=1.0,
ls='solid'))
labels.append(status)
self.fig.legends = []
self.fig.legend(proxy,
labels,
loc=8,
prop={'size': '10.0'},
ncol=len(proxy),
title='Statuses for %s:' % algorithm.name)
for n in net.nodes():
if n.status == '' or not n.status in color_map.keys():
node_colors[n] = 'r'
else:
node_colors[n] = color_map[n.status]
elif subclusters:
for i, sc in enumerate(subclusters):
for n in sc:
if n in node_colors:
node_colors[n] = 'k'
else:
node_colors[n] = COLORS[i]
return node_colors
def draw_edges(self, net=None):
if not net:
net = self.net
self.edge_collection = nx.draw_networkx_edges(net,
net.pos,
alpha=0.6,
edgelist=None,
ax=self.axes)
def draw_messages(self, net=None):
if not net:
net = self.net
self.messages = []
msgCircles = []
for node in net.nodes():
for msg in node.outbox:
# broadcast
if msg.destination is None:
for neighbor in net.adj[node].keys():
nbr_msg = msg.copy()
nbr_msg.destination = neighbor
c = MessageCircle(nbr_msg,
net,
'out',
3.0,
lw=0,
picker=3,
zorder=3,
color='b')
self.messages.append(nbr_msg)
msgCircles.append(c)
else:
c = MessageCircle(msg,
net,
'out',
3.0,
lw=0,
picker=3,
zorder=3,
color='b')
self.messages.append(msg)
msgCircles.append(c)
for msg in node.inbox:
c = MessageCircle(msg,
net,
'in',
3.0,
lw=0,
picker=3,
zorder=3,
color='g')
self.messages.append(msg)
msgCircles.append(c)
if self.messages:
self.message_collection = PatchCollection(msgCircles,
match_original=True)
self.message_collection.set_picker(3)
self.axes.add_collection(self.message_collection)
def draw_labels(self, net=None):
if not net:
net = self.net
label_pos = {}
for n in net.nodes():
label_pos[n] = net.pos[n].copy() + 10
self.label_collection = nx.draw_networkx_labels(net,
label_pos,
labels=net.labels,
ax=self.axes)
def refresh_visibility(self):
try:
self.node_collection.set_visible(self.ui.showNodes.isChecked())
self.edge_collection.set_visible(self.ui.showEdges.isChecked())
for label in self.label_collection.values():
label.set_visible(self.ui.showLabels.isChecked())
self.tree_collection.set_visible(self.ui.treeGroupBox.isChecked())
self.ini_error_collection.set_visible(self.ui.propagationError\
.isChecked())
self.propagation_error_collection.set_visible(self.ui\
.propagationError\
.isChecked())
# sould be last, sometimes there are no messages
self.message_collection.set_visible(self.ui.showMessages\
.isChecked())
except AttributeError:
print 'Refresh visibility warning'
self.canvas.draw()
def draw_tree(self, treeKey, net=None):
"""
Show tree representation of network.
Attributes:
treeKey (str):
key in nodes memory (dictionary) where tree data is stored
storage format can be a list off tree neighbors or a dict:
{'parent': parent_node,
'children': [child_node1, child_node2 ...]}
"""
if not net:
net = self.net
treeNet = net.get_tree_net(treeKey)
if treeNet:
self.tree_collection = draw_networkx_edges(treeNet,
treeNet.pos,
treeNet.edges(),
width=1.8,
alpha=0.6,
ax=self.axes)
def draw_propagation_errors(self, locKey, net):
SCALE_FACTOR = 0.6
if not net:
net = self.net
if any([not locKey in node.memory for node in net.nodes()]):
self.propagation_error_collection = []
self.ini_error_collection = []
return
rms = {'iniRms': {}, 'stitchRms': {}}
for node in net.nodes():
rms['iniRms'][node] = get_rms(self.net.pos,
(node.memory['iniLocs']), True) * \
SCALE_FACTOR
rms['stitchRms'][node] = get_rms(self.net.pos, node.memory[locKey],
True) * SCALE_FACTOR
self.propagation_error_collection = \
self.draw_nodes(net=net, node_colors='g',
node_radius=rms['stitchRms'])
self.ini_error_collection = self.draw_nodes(net=net,
node_colors='b',
node_radius=rms['iniRms'])
def reset_zoom(self):
self.axes.set_xlim((0, self.net.environment.im.shape[1]))
self.axes.set_ylim((0, self.net.environment.im.shape[0]))
def set_title(self, fname):
new = ' - '.join([str(self.windowTitle()).split(' - ')[0], str(fname)])
self.setWindowTitle(new)
def refresh_network_inspector(self):
niModel = DictionaryTreeModel(dic=self.net.get_dic())
self.ui.networkInspector.setModel(niModel)
self.ui.networkInspector.expandToDepth(0)
"""
Callbacks
"""
def on_actionRun_triggered(self):
self.ui.logListWidget.clear()
print 'running ...',
self.sim.stepping = True
self.sim.run_all()
def on_actionStep_triggered(self):
print 'next step ...',
self.sim.run(self.ui.stepSize.value())
def on_actionReset_triggered(self):
print 'reset ...',
self.sim.reset()
self.redraw()
def on_actionCopyInspectorData_triggered(self):
string = 'Node inspector data\n-------------------'
# raise()
for qModelIndex in self.ui.nodeInspector.selectedIndexes():
string += '\n' + qModelIndex.internalPointer().toString(' ')
clipboard = app.clipboard()
clipboard.setText(string)
event = QEvent(QEvent.Clipboard)
app.sendEvent(clipboard, event)
def on_actionShowLocalizedSubclusters_triggered(self):
if len(self.ui.nodeInspector.selectedIndexes()) == 1:
qModelIndex = self.ui.nodeInspector.selectedIndexes()[0]
treeItem = qModelIndex.internalPointer()
assert (isinstance(treeItem.itemDataValue, Positions))
estimated = deepcopy(treeItem.itemDataValue)
estimatedsub = estimated.subclusters[0]
# rotate, translate and optionally scale
# w.r.t. original positions (pos)
align_clusters(Positions.create(self.net.pos), estimated, True)
net = self.net.subnetwork(estimatedsub.keys(), pos=estimatedsub)
self.draw_network(net=net, drawMessages=False)
edge_pos = numpy.asarray([
(self.net.pos[node], estimatedsub[node][:2]) for node in net
])
error_collection = LineCollection(edge_pos, colors='r')
self.axes.add_collection(error_collection)
rms = get_rms(self.net.pos, estimated, scale=False)
self.update_log('rms = %.3f' % rms)
self.update_log('localized = %.2f%% (%d/%d)' %
(len(estimatedsub) * 1. / len(self.net.pos) * 100,
len(estimatedsub), len(self.net.pos)))
def on_actionSaveNetwork_triggered(self, *args):
default_filetype = 'gz'
start = datetime.now().strftime('%Y%m%d') + default_filetype
filters = ['Network pickle (*.gz)', 'All files (*)']
selectedFilter = 'Network pickle (gz)'
filters = ';;'.join(filters)
fname = QFileDialog.getSaveFileName(self, "Choose a filename", start,
filters, selectedFilter)[0]
if fname:
try:
write_pickle(self.net, fname)
except Exception, e:
QMessageBox.critical(self, "Error saving file", str(e),
QMessageBox.Ok, QMessageBox.NoButton)
else:
self.set_title(fname)
