class PM_Clipboard(PM_GroupBox):
"""
The PM_Clipboard class provides a groupbox containing a list of clipboard
items that can be pasted in the 3D Workspace. The selected item in this
list is shown by its elementViewer (an instance of L{PM_PreviewGroupBox})
The object being previewed can then be deposited into the 3D workspace.
"""
def __init__(self,
parentWidget,
title = 'Clipboard',
win = None,
elementViewer = None
):
"""
Appends a PM_Clipboard groupbox widget to I{parentWidget},a L{PM_Dialog}
@param parentWidget: The parent dialog (Property manager) containing
this widget.
@type parentWidget: L{PM_Dialog}
@param title: The title (button) text.
@type title: str
@param win: MainWindow object
@type win: L{MWsemantics} or None
@param elementViewer: The associated preview pane groupbox. If provided,
The selected item in L{self.clipboardListWidget}
is shown (previewed) by L{elementViewer}.
The object being previewed can then be deposited
into the 3D workspace.
@type elementViewer: L{PM_PreviewGroupBox} or None
"""
self.w = win
self.elementViewer = elementViewer
self.elementViewer.setDisplay(diTUBES)
self.pastableItems = None
PM_GroupBox.__init__(self, parentWidget, title)
self._loadClipboardGroupbox()
def _loadClipboardGroupbox(self):
"""
Load the L{self.clipboardListWidget} widget used to display a list of
clipboard items inside this clipboard groupbox.
"""
self.clipboardListWidget = QListWidget(self)
self.gridLayout.addWidget(self.clipboardListWidget)
#Append to the widget list. This is important for expand -collapse
#functions (of the groupbox) to work properly.
self._widgetList.append(self.clipboardListWidget)
def _updateElementViewer(self, newModel = None):
"""
Update the view of L{self.elementViewer}
@param newModel: The model correseponding to the item selected
in L{self.clipboardListWidget}.
@type newModel: L{molecule} or L{Group}
"""
if not self.elementViewer:
return
assert isinstance(self.elementViewer, MMKitView)
self.elementViewer.resetView()
if newModel:
self.elementViewer.updateModel(newModel)
def update(self):
"""
Updates the clipboard items in the L{PM_Clipboard} groupbox. Also
updates its element viewer.
"""
PM_GroupBox.update(self)
self.pastableItems = self.w.assy.shelf.getPastables()
i = self.clipboardListWidget.currentRow()
self.clipboardListWidget.clear()
newModel = None
if len(self.pastableItems):
for item in self.pastableItems:
self.clipboardListWidget.addItem(item.name)
if i >= self.clipboardListWidget.count():
i = self.clipboardListWidget.count() - 1
if i < 0:
i = 0
self.clipboardListWidget.setCurrentItem(
self.clipboardListWidget.item(i))
newModel = self.pastableItems[i]
self._updateElementViewer(newModel)
def clipboardListItemChanged(self, currentItem = None, previousItem = None):
"""
Slot method. Called when user clicks on a different pastable item
displayed in this groupbox
@param currentItem: Current item in the L{self.clipboardListWidget}
that is selected
@type currentItem: U{B{QListWidgetItem}
<http://doc.trolltech.com/4.2/qlistwidgetitem.html>}
@param previousItem: Previously selected item in the
L{self.clipboardListWidget}
@type previousItem: U{B{QListWidgetItem}
<http://doc.trolltech.com/4.2/qlistwidgetitem.html>}
"""
if not (currentItem or previousItem):
return
itemId = self.clipboardListWidget.row(currentItem)
if itemId != -1:
newChunk = self.pastableItems[itemId]
self.clipboardListWidget.setCurrentRow(itemId)
self._updateElementViewer(newChunk)
def connect_or_disconnect_signals(self, isConnect):
"""
Connect or disconnect widget signals sent to their slot methods.
@param isConnect: If True the widget will send the signals to the slot
method.
@type isConnect: boolean
"""
if isConnect:
change_connect = self.w.connect
else:
change_connect = self.w.disconnect
change_connect(
self.clipboardListWidget,
SIGNAL("currentItemChanged(QListWidgetItem*,QListWidgetItem*)"),
self.clipboardListItemChanged )
def currentRow(self):
"""
Return the current row of the selected item in this groupbox's
listwidget ( L{self.clipboardListWidget} )
@return: Current Row of the selected pastable item in the
clipboard groupbox.
@rtype: int
"""
currentRow = self.clipboardListWidget.currentRow()
return currentRow
class PM_Clipboard(PM_GroupBox):
"""
The PM_Clipboard class provides a groupbox containing a list of clipboard
items that can be pasted in the 3D Workspace. The selected item in this
list is shown by its elementViewer (an instance of L{PM_PreviewGroupBox})
The object being previewed can then be deposited into the 3D workspace.
"""
def __init__(self,
parentWidget,
title='Clipboard',
win=None,
elementViewer=None):
"""
Appends a PM_Clipboard groupbox widget to I{parentWidget},a L{PM_Dialog}
@param parentWidget: The parent dialog (Property manager) containing
this widget.
@type parentWidget: L{PM_Dialog}
@param title: The title (button) text.
@type title: str
@param win: MainWindow object
@type win: L{MWsemantics} or None
@param elementViewer: The associated preview pane groupbox. If provided,
The selected item in L{self.clipboardListWidget}
is shown (previewed) by L{elementViewer}.
The object being previewed can then be deposited
into the 3D workspace.
@type elementViewer: L{PM_PreviewGroupBox} or None
"""
self.w = win
self.elementViewer = elementViewer
self.elementViewer.setDisplay(diTUBES)
self.pastableItems = None
PM_GroupBox.__init__(self, parentWidget, title)
self._loadClipboardGroupbox()
def _loadClipboardGroupbox(self):
"""
Load the L{self.clipboardListWidget} widget used to display a list of
clipboard items inside this clipboard groupbox.
"""
self.clipboardListWidget = QListWidget(self)
self.gridLayout.addWidget(self.clipboardListWidget)
#Append to the widget list. This is important for expand -collapse
#functions (of the groupbox) to work properly.
self._widgetList.append(self.clipboardListWidget)
def _updateElementViewer(self, newModel=None):
"""
Update the view of L{self.elementViewer}
@param newModel: The model correseponding to the item selected
in L{self.clipboardListWidget}.
@type newModel: L{molecule} or L{Group}
"""
if not self.elementViewer:
return
assert isinstance(self.elementViewer, MMKitView)
self.elementViewer.resetView()
if newModel:
self.elementViewer.updateModel(newModel)
def update(self):
"""
Updates the clipboard items in the L{PM_Clipboard} groupbox. Also
updates its element viewer.
"""
PM_GroupBox.update(self)
self.pastableItems = self.w.assy.shelf.getPastables()
i = self.clipboardListWidget.currentRow()
self.clipboardListWidget.clear()
newModel = None
if len(self.pastableItems):
for item in self.pastableItems:
self.clipboardListWidget.addItem(item.name)
if i >= self.clipboardListWidget.count():
i = self.clipboardListWidget.count() - 1
if i < 0:
i = 0
self.clipboardListWidget.setCurrentItem(
self.clipboardListWidget.item(i))
newModel = self.pastableItems[i]
self._updateElementViewer(newModel)
def clipboardListItemChanged(self, currentItem=None, previousItem=None):
"""
Slot method. Called when user clicks on a different pastable item
displayed in this groupbox
@param currentItem: Current item in the L{self.clipboardListWidget}
that is selected
@type currentItem: U{B{QListWidgetItem}
<http://doc.trolltech.com/4.2/qlistwidgetitem.html>}
@param previousItem: Previously selected item in the
L{self.clipboardListWidget}
@type previousItem: U{B{QListWidgetItem}
<http://doc.trolltech.com/4.2/qlistwidgetitem.html>}
"""
if not (currentItem or previousItem):
return
itemId = self.clipboardListWidget.row(currentItem)
if itemId != -1:
newChunk = self.pastableItems[itemId]
self.clipboardListWidget.setCurrentRow(itemId)
self._updateElementViewer(newChunk)
def connect_or_disconnect_signals(self, isConnect):
"""
Connect or disconnect widget signals sent to their slot methods.
@param isConnect: If True the widget will send the signals to the slot
method.
@type isConnect: boolean
"""
if isConnect:
change_connect = self.w.connect
else:
change_connect = self.w.disconnect
change_connect(
self.clipboardListWidget,
SIGNAL("currentItemChanged(QListWidgetItem*,QListWidgetItem*)"),
self.clipboardListItemChanged)
def currentRow(self):
"""
Return the current row of the selected item in this groupbox's
listwidget ( L{self.clipboardListWidget} )
@return: Current Row of the selected pastable item in the
clipboard groupbox.
@rtype: int
"""
currentRow = self.clipboardListWidget.currentRow()
return currentRow
class IpetPbHistoryWindow(IpetMainWindow):
NO_SELECTION_TEXT = "no selection"
default_cmap = 'spectral'
imagepath = osp.sep.join((osp.dirname(__file__), osp.pardir, "images"))
DEBUG = True
def __init__(self, testrunfiles=[], evaluationfile=None):
QtGui.QMainWindow.__init__(self)
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
self.setWindowTitle("application main window")
self.file_menu = QtGui.QMenu('&File', self)
self.file_menu.addAction('&Quit', self.fileQuit,
QtCore.Qt.CTRL + QtCore.Qt.Key_Q)
self.menuBar().addMenu(self.file_menu)
loadaction = self.createAction(
"&Load",
self.loadTestruns,
QKeySequence.Open,
icon="Load-icon",
tip="Load testrun from trn file (current test run gets discarded)")
resetaction = self.createAction(
"&Reset selected names",
self.resetSelectedTestrunNames,
QKeySequence.Undo,
icon="",
tip="Reset names of selected test runs")
self.file_menu.addAction(loadaction)
self.help_menu = QtGui.QMenu('&Help', self)
self.edit_menu = QtGui.QMenu('&Edit', self)
self.edit_menu.addAction(resetaction)
self.menuBar().addSeparator()
self.menuBar().addMenu(self.help_menu)
self.menuBar().addMenu(self.edit_menu)
self.help_menu.addAction('&About', self.about)
self.main_widget = QtGui.QWidget(self)
lwframe = QFrame(self.main_widget)
l = QtGui.QVBoxLayout(self.main_widget)
self.sc = MyStaticMplCanvas(self.main_widget,
width=5,
height=4,
dpi=100)
toolbar = IpetNavigationToolBar(self.sc, self.main_widget)
l.addWidget(toolbar)
l.addWidget(self.sc)
h = QtGui.QHBoxLayout(lwframe)
l.addWidget(lwframe)
self.trListWidget = QListWidget()
# for item in list("ABC"):
# self.trListWidget.addItem((item))
self.trListWidget.setSelectionMode(QAbstractItemView.ExtendedSelection)
h.addWidget(self.trListWidget)
self.connect(self.trListWidget, SIGNAL("itemSelectionChanged()"),
self.selectionChanged)
self.trListWidget.itemChanged.connect(self.testrunItemChanged)
v = QtGui.QVBoxLayout(lwframe)
self.probListWidget = QListWidget()
# for item in list("12345"):
# self.probListWidget.addItem((item))
self.probListWidget.setSelectionMode(
QAbstractItemView.ExtendedSelection)
v.addWidget(self.probListWidget)
self.connect(self.probListWidget, SIGNAL("itemSelectionChanged()"),
self.selectionChanged)
self.testSetWidget = QListWidget()
v.addWidget(self.testSetWidget)
self.testSetWidget.addItem(self.NO_SELECTION_TEXT)
for t in TestSets.getTestSets():
self.testSetWidget.addItem(str(t))
h.addLayout(v)
self.connect(self.testSetWidget, SIGNAL("itemSelectionChanged()"),
self.selectionChanged)
self.main_widget.setFocus()
self.setCentralWidget(self.main_widget)
self.testruns = []
self.testrunnames = {}
for tr in testrunfiles:
tr = TestRun.loadFromFile(str(tr))
try:
self.addTestrun(tr)
except Exception as e:
print(e)
self.evaluation = None
if evaluationfile is not None:
self.evaluation = IPETEvaluation.fromXMLFile(evaluationfile)
# self.primallines = {}
# self.duallines = {}
# self.primalpatches = {}
self.statusBar().showMessage("Ready to load some test run data!", 5000)
def createAction(self,
text,
slot=None,
shortcut=None,
icon=None,
tip=None,
checkable=False,
signal="triggered()"):
action = QAction(text, self)
if icon is not None:
action.setIcon(
QIcon(osp.sep.join((self.imagepath, "%s.png" % icon))))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
self.connect(action, SIGNAL(signal), slot)
if checkable:
action.setCheckable(True)
return action
def getSelectedProblist(self):
selectedprobs = [
str(item.text()) for item in self.probListWidget.selectedItems()
]
selitem = self.testSetWidget.selectedItems()[0]
if selitem.text() != self.NO_SELECTION_TEXT:
testsetprobs = set(TestSets.getTestSetByName(selitem.text()))
selectedprobs = [p for p in selectedprobs if p in testsetprobs]
return selectedprobs
def getSelectedTestrunList(self):
return [
tr for idx, tr in enumerate(self.testruns)
if self.trListWidget.isItemSelected(self.trListWidget.item(idx))
]
def selectionChanged(self):
if len(self.testruns) == 0:
return
problist = self.getSelectedProblist()
if len(problist) == 0:
return
testruns = self.getSelectedTestrunList()
if len(testruns) == 0:
return
self.update_Axis(problist, testruns)
def testrunItemChanged(self):
curritem = self.trListWidget.currentItem()
if not curritem:
return
rowindex = self.trListWidget.currentRow()
if rowindex < 0 or rowindex > len(self.testruns):
return
newtext = str(curritem.text())
testrun = self.testruns[rowindex]
self.setTestrunName(testrun, newtext)
def resetSelectedTestrunNames(self):
for tr in self.getSelectedTestrunList():
self.resetTestrunName(tr)
def getTestrunName(self, testrun):
"""
returns the test run name as specified by the user
"""
return self.testrunnames.get(testrun.getName(), testrun.getName())
def setTestrunName(self, testrun, newname):
self.debugMessage("Changing testrun name from %s to %s" %
(self.getTestrunName(testrun), newname))
self.testrunnames[testrun.getName()] = newname
def resetTestrunName(self, testrun):
try:
del self.testrunnames[testrun.getName()]
item = self.trListWidget.item(self.testruns.index(testrun))
item.setText((self.getTestrunName(testrun)))
except KeyError:
pass
def updateStatus(self, message):
self.statusBar().showMessage(message, 5000)
def fileQuit(self):
self.close()
def addTestrun(self, tr):
self.testruns.append(tr)
self.probListWidget.clear()
self.trListWidget.clear()
for testrun in self.testruns:
item = QListWidgetItem((self.getTestrunName(testrun)))
self.trListWidget.addItem(item)
item.setFlags(item.flags() | QtCore.Qt.ItemIsEditable)
problems = []
if len(self.testruns) > 0:
problems = self.testruns[0].getProblemNames()
if len(self.testruns) > 1:
for testrun in self.testruns[1:]:
problems = [
p for p in problems if p in set(testrun.getProblemNames())
]
for prob in sorted(problems):
self.probListWidget.addItem(str(prob))
self.trListWidget.selectAll()
def closeEvent(self, ce):
self.fileQuit()
def debugMessage(self, message):
if self.DEBUG:
print(message)
else:
pass
def loadTestruns(self):
thedir = str(".")
filenames = QFileDialog.getOpenFileNames(
self,
caption=("%s - Load testruns" % QApplication.applicationName()),
directory=thedir,
filter=str("Testrun files (*.trn)"))
if filenames:
loadedtrs = 0
notloadedtrs = 0
for filename in filenames:
try:
print(filename)
tr = TestRun.loadFromFile(str(filename))
try:
self.addTestrun(tr)
except Exception as e:
print(e)
loadedtrs += 1
except Exception:
notloadedtrs += 1
message = "Loaded %d/%d test runs" % (loadedtrs,
loadedtrs + notloadedtrs)
self.updateStatus(message)
pass
def update_Axis(self, probnames, testruns):
"""
update method called every time a new instance was selected
"""
#self.resetAxis()
# make up data for plotting method
x = {}
y = {}
z = {}
zx = {}
kws = {}
duallinekws = {}
dualbarkws = {}
baseline = 0
if len(probnames) == 1:
self.updateStatus("Showing problem %s on %d test runs" %
(probnames[0], len(testruns)))
else:
self.updateStatus("Showing mean over %d problems on %d test runs" %
(len(probnames), len(testruns)))
self.resetAxis()
usenormalization = True
showdualbound = True
xmax = xmin = ymax = ymin = 0
labelorder = []
for testrun in testruns:
testrunname = self.getTestrunName(testrun)
if len(probnames) == 1:
x[testrunname], y[testrunname] = getProcessPlotData(
testrun, probnames[0], usenormalization, access="name")
else:
x[testrunname], y[testrunname] = getMeanIntegral(testrun,
probnames,
access="name")
if not usenormalization and len(probnames) == 1:
baseline = testrun.problemGetOptimalSolution(probnames[0])
y[testrunname] -= baseline
xmax = max(xmax, max(x[testrunname]))
ymax = max(ymax, max(y[testrunname]))
ymin = min(ymin, min(y[testrunname]))
print(y[testrunname])
print(y[testrunname][1:] - y[testrunname][:-1] > 0)
if numpy.any(y[testrunname][1:] - y[testrunname][:-1] > 0):
logging.warn(
"Error: Increasing primal gap function on problems {}".
format(probnames))
if showdualbound:
arguments = {
"historytouse": Key.DualBoundHistory,
"boundkey": Key.DualBound
}
if len(probnames) == 1:
zx[testrunname], z[testrunname] = getProcessPlotData(
testrun,
probnames[0],
usenormalization,
access="name",
**arguments)
else:
zx[testrunname], z[testrunname] = getMeanIntegral(
testrun, probnames, access="name", **arguments)
# normalization requires negative dual gap
if usenormalization:
z[testrunname] = -numpy.array(z[testrunname])
ymin = min(ymin, min(z[testrunname]))
duallinekws[testrunname] = dict(linestyle='dashed')
dualbarkws = dict(alpha=0.1)
# set special key words for the testrun
kws[testrunname] = dict(alpha=0.1)
labelorder.append(testrunname)
# for now, only one color cycle exists
#colormap = cm.get_cmap(name='spectral', lut=128)
#self.axes.set_color_cycle([colormap(i) for i in numpy.linspace(0.1, 0.9, len(self.gui.getTestrunList()))])
# call the plot on the collected data
self.primalpatches, self.primallines, _ = self.axisPlotForTestrunData(
x,
y,
baseline=baseline,
legend=False,
labelsuffix=" (primal)",
plotkw=kws,
barkw=kws,
labelorder=labelorder)
if showdualbound:
__, self.duallines, _ = self.axisPlotForTestrunData(
zx,
z,
step=False,
baseline=0,
legend=False,
labelsuffix=" (dual)",
plotkw=duallinekws,
barkw=dualbarkws,
labelorder=labelorder)
# set a legend and limits
self.sc.axes.legend(fontsize=8)
self.sc.axes.set_xlim(
(xmin - 0.05 * (xmax - xmin), xmax + 0.05 * (xmax - xmin)))
self.sc.axes.set_ylim(
(ymin - 0.1 * (ymax - ymin), ymax + 0.1 * (ymax - ymin)),
emit=True)
self.sc.draw()
def axisPlotForTestrunData(self,
dataX,
dataY,
bars=False,
step=True,
barwidthfactor=1.0,
baseline=0,
testrunnames=None,
legend=True,
labelsuffix="",
colormapname="spectral",
plotkw=None,
barkw=None,
labelorder=[]):
"""
create a plot for your X and Y data. The data can either be specified as matrix, or as a dictionary
specifying containing the labels as keys.
- returns the axes object and a dictionary {label:line} of the line plots that were added
arguments:
-dataX : The X data for the plot, exspected either as
1: A dictionary with the plot labels as keys and some iterable as value-list
OR
2: A list of some iterables which denote the X values.
-dataY : The y plot data of the plot. Must be specified in the same way as the X data
-testrunnames: labels for axis legend. they will overwrite the labels specified by dictionary-organized data.
if testrunnames == None, the primallines will either be labelled from '0' to 'len(dataX)-1',
or inferred from the dataX-keys().
-ax: matplotlib axes object, will be created as new axis if not specified
-legend: specify if legend should be created, default True
-colormapname: name of the colormap to use in case no colors are specified by the 'colors' argument
-kw, other keywords for the plotting function, such as transparency, etc. can be specified for every plot
separately, either as a dictionary with the dataX-keys, or as a kw-list with the same length as the
dataX list
"""
# index everything by labels, either given as dictionary keys, or integer indices ranging from 0 to len(dataX) - 1
assert type(dataX) is type(dataY)
if type(dataX) is dict:
labels = list(dataX.keys())
if testrunnames is None:
testrunnames = {label: label for label in labels}
else:
assert type(dataX) is list
labels = list(range(len(dataX)))
if testrunnames is None:
testrunnames = {label: repr(label) for label in labels}
# init colors if not given
try:
colormap = cm.get_cmap(name=colormapname, lut=128)
except ValueError:
print("Colormap of name ", colormapname, " does not exist")
colormap = cm.get_cmap(name=IpetPbHistoryWindow.default_cmap,
lut=128)
colortransform = numpy.linspace(0.1, 0.9, len(labels))
colors = {
label: colormap(colortransform[index])
for index, label in enumerate(labels)
}
patches = {}
lines = {}
if not labelorder:
labelorder = sorted(labels)
for label in labelorder:
# retrieve special key words, or use the entire keyword dictionary
if plotkw is not None:
linekw = plotkw.get(testrunnames[label], plotkw)
else:
linekw = {}
if barkw is not None:
bkw = barkw.get(testrunnames[label], barkw)
else:
bkw = {}
x = dataX[label]
y = dataY[label]
idd = testrunnames[label]
if bars:
patches[idd] = self.sc.axes.bar(x[:-1],
y[:-1],
width=barwidthfactor *
(x[1:] - x[:-1]),
bottom=baseline,
color=colors[label],
linewidth=0,
**bkw)
else:
patches[idd] = []
# # use step functions for primal and plot for dual plots
plotlabel = idd + labelsuffix
if step:
#lines[idd], = ax.step(x, y + baseline, color=colors[label], label=idd, where='post')
lines[idd], = self.sc.axes.step(x,
y,
label=plotlabel,
where='post')
else:
#lines[idd], = ax.plot(x, y + baseline, color=colors[label], label=idd, **linekw)
lines[idd], = self.sc.axes.plot(x,
y + baseline,
label=plotlabel,
**linekw)
if len(labels) > 0 and legend:
self.sc.axes.legend(fontsize=8)
return (patches, lines, self.sc.axes)
def resetAxis(self):
"""
reset axis by removing all primallines and primalpatches previously drawn.
"""
self.sc.axes.cla()
def about(self):
QtGui.QMessageBox.about(
self, "About", """embedding_in_qt4.py example
Copyright 2005 Florent Rougon, 2006 Darren Dale
This program is a simple example of a Qt4 application embedding matplotlib
canvases.
It may be used and modified with no restriction; raw copies as well as
modified versions may be distributed without limitation.""")
