class MPL_Widget(QtGui.QWidget):
def __init__(self, parent = None):
QtGui.QWidget.__init__(self, parent)
self.canvas = MyMplCanvas()
#self.toolbar = MyNavigationToolbar(self.canvas, self.canvas, direction = 'v')
self.toolbar = NavigationToolbar(self.canvas, self.canvas)
#self.toolbar.hide()
self.hbox = QtGui.QHBoxLayout()
#self.hbox.addWidget(self.toolbar)
self.hbox.addWidget(self.canvas)
self.setLayout(self.hbox)
##########################
self.hZoom = QtGui.QAction("Zoom", self)
self.hZoom.setShortcut("Ctrl+Z")
self.addAction(self.hZoom)
QtCore.QObject.connect(self.hZoom,QtCore.SIGNAL("triggered()"), self.ZoomToggle)
self.actionAutoScale = QtGui.QAction("AutoScale", self)#self.MainWindow)
self.actionAutoScale.setShortcut("Ctrl+A")
self.addAction(self.actionAutoScale)
QtCore.QObject.connect(self.actionAutoScale,QtCore.SIGNAL("triggered()"), self.autoscale_plot)
def ZoomToggle(self):
self.toolbar.zoom()
def autoscale_plot(self):
self.toolbar.home()
class MPL_Widget(QtGui.QWidget):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.canvas = MyMplCanvas()
#self.toolbar = MyNavigationToolbar(self.canvas, self.canvas, direction = 'v')
self.toolbar = NavigationToolbar(self.canvas, self.canvas)
#self.toolbar.hide()
self.hbox = QtGui.QHBoxLayout()
#self.hbox.addWidget(self.toolbar)
self.hbox.addWidget(self.canvas)
self.setLayout(self.hbox)
##########################
self.hZoom = QtGui.QAction("Zoom", self)
self.hZoom.setShortcut("Ctrl+Z")
self.addAction(self.hZoom)
QtCore.QObject.connect(self.hZoom, QtCore.SIGNAL("triggered()"),
self.ZoomToggle)
self.actionAutoScale = QtGui.QAction("AutoScale",
self) #self.MainWindow)
self.actionAutoScale.setShortcut("Ctrl+A")
self.addAction(self.actionAutoScale)
QtCore.QObject.connect(self.actionAutoScale,
QtCore.SIGNAL("triggered()"),
self.autoscale_plot)
def ZoomToggle(self):
self.toolbar.zoom()
def autoscale_plot(self):
self.toolbar.home()
class MPL_Widget(QtGui.QWidget):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.canvas = MyMplCanvas()
#self.toolbar = MyNavigationToolbar(self.canvas, self.canvas, direction = 'v')
self.toolbar = NavigationToolbar(self.canvas, self.canvas)
self.toolbar.hide()
self.hbox = QtGui.QHBoxLayout()
#self.hbox.addWidget(self.toolbar)
self.hbox.addWidget(self.canvas)
self.setLayout(self.hbox)
##########################
self.hZoom = QtGui.QAction("Zoom", self)
self.hZoom.setShortcut("Ctrl+Z")
self.addAction(self.hZoom)
QtCore.QObject.connect(self.hZoom, QtCore.SIGNAL("triggered()"),
self.ZoomToggle)
self.actionAutoScale = QtGui.QAction("AutoScale",
self) #self.MainWindow)
self.actionAutoScale.setShortcut("Ctrl+A")
self.addAction(self.actionAutoScale)
QtCore.QObject.connect(self.actionAutoScale,
QtCore.SIGNAL("triggered()"),
self.autoscale_plot)
self.installEventFilter(EventFilter(self))
###########SAVING FIGURE TO CLIPBOARD##########
self.cb = None #will be used for the clipboard
self.tempPath = getHomeDir()
self.tempPath = os.path.join(self.tempPath, 'tempMPL.png')
def ZoomToggle(self):
self.toolbar.zoom()
def autoscale_plot(self):
self.toolbar.home()
def mpl2Clip(self):
try:
self.canvas.fig.savefig(self.tempPath)
tempImg = QtGui.QImage(self.tempPath)
self.cb = QtGui.QApplication.clipboard()
self.cb.setImage(tempImg)
except:
print 'Error copying figure to clipboard'
errorMsg = "Sorry: %s\n\n:%s\n" % (sys.exc_type, sys.exc_value)
print errorMsg
def focusEvent(self, event):
# self.enableAutoScale()
# self.enableZoom()
self.enableClip()
# self.enableCSV()
#print "Focus In %s"%self.canvas.plotTitle
def lossFocusEvent(self, event):
# self.disableAutoScale()
# self.disableZoom()
self.disableClip()
# self.disableCSV()
#print "Focus Out %s"%self.canvas.plotTitle
def enableClip(self):
self.mpl2ClipAction = QtGui.QAction("Save to Clipboard", self)
self.mpl2ClipAction.setShortcut("Ctrl+C")
self.addAction(self.mpl2ClipAction)
QtCore.QObject.connect(self.mpl2ClipAction,
QtCore.SIGNAL("triggered()"), self.mpl2Clip)
def disableClip(self):
QtCore.QObject.disconnect(self.mpl2ClipAction,
QtCore.SIGNAL("triggered()"), self.mpl2Clip)
self.removeAction(self.mpl2ClipAction)
def home(self, *args):
NavigationToolbar2QT.home(self,*args)
self.homeClicked.emit()
class MPL_Widget(QtGui.QWidget):
def __init__(self, parent = None):
QtGui.QWidget.__init__(self, parent)
self.canvas = MyMplCanvas()
#self.toolbar = MyNavigationToolbar(self.canvas, self.canvas, direction = 'v')
self.toolbar = NavigationToolbar(self.canvas, self.canvas)
self.toolbar.hide()
self.hbox = QtGui.QHBoxLayout()
#self.hbox.addWidget(self.toolbar)
self.hbox.addWidget(self.canvas)
self.setLayout(self.hbox)
##########################
self.hZoom = QtGui.QAction("Zoom", self)
self.hZoom.setShortcut("Ctrl+Z")
self.addAction(self.hZoom)
QtCore.QObject.connect(self.hZoom,QtCore.SIGNAL("triggered()"), self.ZoomToggle)
self.actionAutoScale = QtGui.QAction("AutoScale", self)#self.MainWindow)
self.actionAutoScale.setShortcut("Ctrl+A")
self.addAction(self.actionAutoScale)
QtCore.QObject.connect(self.actionAutoScale,QtCore.SIGNAL("triggered()"), self.autoscale_plot)
self.installEventFilter(EventFilter(self))
###########SAVING FIGURE TO CLIPBOARD##########
self.cb = None #will be used for the clipboard
self.tempPath = getHomeDir()
self.tempPath = os.path.join(self.tempPath,'tempMPL.png')
def ZoomToggle(self):
self.toolbar.zoom()
def autoscale_plot(self):
self.toolbar.home()
def mpl2Clip(self):
try:
self.canvas.fig.savefig(self.tempPath)
tempImg = QtGui.QImage(self.tempPath)
self.cb = QtGui.QApplication.clipboard()
self.cb.setImage(tempImg)
except:
print 'Error copying figure to clipboard'
errorMsg = "Sorry: %s\n\n:%s\n"%(sys.exc_type, sys.exc_value)
print errorMsg
def focusEvent(self, event):
# self.enableAutoScale()
# self.enableZoom()
self.enableClip()
# self.enableCSV()
#print "Focus In %s"%self.canvas.plotTitle
def lossFocusEvent(self, event):
# self.disableAutoScale()
# self.disableZoom()
self.disableClip()
# self.disableCSV()
#print "Focus Out %s"%self.canvas.plotTitle
def enableClip(self):
self.mpl2ClipAction = QtGui.QAction("Save to Clipboard", self)
self.mpl2ClipAction.setShortcut("Ctrl+C")
self.addAction(self.mpl2ClipAction)
QtCore.QObject.connect(self.mpl2ClipAction,QtCore.SIGNAL("triggered()"), self.mpl2Clip)
def disableClip(self):
QtCore.QObject.disconnect(self.mpl2ClipAction,QtCore.SIGNAL("triggered()"), self.mpl2Clip)
self.removeAction(self.mpl2ClipAction)
class MPL_Widget(QtGui.QWidget):
def __init__(self,
parent=None,
enableAutoScale=True,
enableCSV=False,
enableEdit=True,
doublePlot=False,
layoutDir='h'):
QtGui.QWidget.__init__(self, parent)
if doublePlot:
self.canvas = DoubleMyMplCanvas()
else:
self.canvas = MyMplCanvas()
self.toolbar = NavigationToolbar(self.canvas, self.canvas)
# self.toolbar.hide()
self.installEventFilter(EventFilter(self))
if layoutDir == 'v':
self.toolbar.setOrientation(QtCore.Qt.Vertical)
self.layout = QtGui.QHBoxLayout()
elif layoutDir == 'h':
self.layout = QtGui.QVBoxLayout()
# self.layout.addWidget(self.canvas)
# self.layout.addWidget(self.toolbar)
self.layout.addWidget(self.toolbar)
self.layout.addWidget(self.canvas)
self.setLayout(self.layout)
##########################
self.dataLabels = None
###########SAVING FIGURE TO CLIPBOARD##########
self.cb = None #will be used for the clipboard
self.tempPath = getHomeDir()
self.tempPath = os.path.join(self.tempPath, 'tempMPL.png')
if enableEdit:
self.enableEdit()
self.lineDict = None
self.addLegend = False
self.x = None #used for picker
self.y = None
#######SAVING FIGURE DATA############################
if enableCSV:
self.enableCSV()
########### HELPER FUNCTIONS #########################
self.clearPlotAction = QtGui.QAction("Clear Plot", self)
self.addAction(self.clearPlotAction)
QtCore.QObject.connect(self, QtCore.SIGNAL("triggered()"),
self.clearPlot)
# def focusOutEvent(self, event):
# print "Focus Out"
def clearPlot(self):
print "Clear Plot"
self.canvas.ax.cla()
self.canvas.format_labels()
self.canvas.draw()
# def __initContextMenus__(self):
# self.clearPlotAction = QtGui.QAction("Clear Plot", self)
# self.addAction(self.clearPlotAction)
# QtCore.QObject.connect(self, QtCore.SIGNAL("triggered()"), self.clearPlot)
#
# self.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
# self.connect(self, QtCore.SIGNAL("customContextMenuRequested(QPoint)"), self.plotTabContext)
# def plotTabContext(self, point):
# '''Create a menu for mainTabWidget'''
# plotCT_menu = QtGui.QMenu("Menu", self)
# plotCT_menu.addAction(self.Zoom)
# plotCT_menu.addAction(self.actionAutoScale)
# plotCT_menu.addSeparator()
# plotCT_menu.addAction(self.clearPlotAction)
# plotCT_menu.exec_(self.mapToGlobal(point))
def focusEvent(self, event):
self.enableAutoScale()
self.enableZoom()
self.enableClip()
# self.enableCSV()
#print "Focus In %s"%self.canvas.plotTitle
def lossFocusEvent(self, event):
self.disableAutoScale()
self.disableZoom()
self.disableClip()
# self.disableCSV()
#print "Focus Out %s"%self.canvas.plotTitle
def enableClip(self):
self.mpl2ClipAction = QtGui.QAction("Save to Clipboard", self)
self.mpl2ClipAction.setShortcut("Ctrl+C")
self.addAction(self.mpl2ClipAction)
QtCore.QObject.connect(self.mpl2ClipAction,
QtCore.SIGNAL("triggered()"), self.mpl2Clip)
def disableClip(self):
QtCore.QObject.disconnect(self.mpl2ClipAction,
QtCore.SIGNAL("triggered()"), self.mpl2Clip)
self.removeAction(self.mpl2ClipAction)
def enableEdit(self):
self.editAction = QtGui.QAction("Edit Line Properties", self)
self.editAction.setShortcut("Ctrl+Alt+E")
self.addAction(self.editAction)
QtCore.QObject.connect(self.editAction, QtCore.SIGNAL("triggered()"),
self.editPlotProperties)
def disableEdit(self):
if self.editAction:
QtCore.QObject.disconnect(self.editAction,
QtCore.SIGNAL("triggered()"),
self.editPlotProperties)
self.removeAction(self.editAction)
def enableZoom(self):
self.Zoom = QtGui.QAction("Zoom", self)
self.Zoom.setShortcut("Ctrl+Z")
self.addAction(self.Zoom)
QtCore.QObject.connect(self.Zoom, QtCore.SIGNAL("triggered()"),
self.ZoomToggle)
def disableZoom(self):
if self.Zoom != None:
QtCore.QObject.disconnect(self.Zoom, QtCore.SIGNAL("triggered()"),
self.ZoomToggle)
self.removeAction(self.Zoom)
def disableAutoScale(self):
if self.actionAutoScale != None:
QtCore.QObject.disconnect(self.actionAutoScale,
QtCore.SIGNAL("triggered()"),
self.autoscale_plot)
self.removeAction(self.actionAutoScale)
def enableAutoScale(self):
self.actionAutoScale = QtGui.QAction("AutoScale",
self) #self.MainWindow)
self.actionAutoScale.setShortcut("Ctrl+A")
self.addAction(self.actionAutoScale)
QtCore.QObject.connect(self.actionAutoScale,
QtCore.SIGNAL("triggered()"),
self.autoscale_plot)
def enableCSV(self):
self.saveCSVAction = QtGui.QAction("Save to CSV", self)
self.saveCSVAction.setShortcut("Ctrl+Alt+S")
self.addAction(self.saveCSVAction)
QtCore.QObject.connect(self.saveCSVAction,
QtCore.SIGNAL("triggered()"), self.save2CSV)
def disableCSV(self):
QtCore.QObject.disconnect(self.saveCSVAction,
QtCore.SIGNAL("triggered()"), self.save2CSV)
self.removeAction(self.saveCSVAction)
def setLineDict(self):
self.lineDict = {}
lineList = self.canvas.ax.get_lines()
if lineList > 0:
for line in lineList:
self.lineDict[line.get_label()] = line
def editPlotProperties(self):
print "Edit Enabled"
self.setLineDict()
# if len(self.lineDict) > 0:
curAx = self.canvas.ax
if POD(self.lineDict, curAx=curAx, parent=self).exec_():
if self.addLegend:
curAx.legend(borderaxespad=0.03, axespad=0.25)
self.canvas.format_labels()
self.canvas.draw()
else:
print "Cancel"
def ZoomToggle(self):
self.toolbar.zoom() #this implements the classic zoom
# if self.hZoom:
# self.hZoom = False
# self.span.visible = False
# else:
# self.hZoom = True
# self.span.visible = True
def autoscale_plot(self):
self.toolbar.home() #implements the classic return to home
# self.canvas.ax.autoscale_view(tight = False, scalex=True, scaley=True)
# self.canvas.draw()
# self.emit(QtCore.SIGNAL("autoScaleAxis(bool)"),True)
# def onclick(self, event):
# #sets up the maximum Y level to be displayed after the zoom.
# #if not set then it maxes to the largest point in the data
# #not necessarily the local max
# if event.ydata != None:
# self.localYMax = int(event.ydata)
#
# def onselect(self, xmin, xmax):
# #print xmin, xmax
# if self.hZoom:
# self.canvas.ax.set_ylim(ymax = self.localYMax)
# self.canvas.ax.set_xlim(xmin, xmax)
# self.canvas.draw()
def save2CSV(self):
path = self.SFDialog()
if path != None:
try:
lines = self.canvas.ax.get_lines()
data2write = []
for line in lines:
data2write.append(line.get_data()[0])
data2write.append(line.get_data()[1])
print data2write
data2write = N.array(data2write)
data2write.dtype = N.float32
N.savetxt(str(path),
N.transpose(data2write),
delimiter=',',
fmt='%.4f')
except:
try:
#this is for the case where the data may not be in float format?
N.savetxt(str(path),
N.transpose(data2write),
delimiter=',')
except:
print 'Error saving figure data'
errorMsg = "Sorry: %s\n\n:%s\n" % (sys.exc_type,
sys.exc_value)
print errorMsg
def SFDialog(self):
fileName = QtGui.QFileDialog.getSaveFileName(self,
"Select File to Save", "",
"csv Files (*.csv)")
if not fileName.isEmpty():
print fileName
return fileName
else:
return None
def removePicker(self):
try:
self.handleA.remove()
except:
print ""
def setData(self, x, y):
if x != None:
self.x = x
if y != None:
self.y = y
def addPicker(self):
'''Sets up the plot variables used for interaction'''
self.handleA, = self.canvas.ax.plot([0], [0], 'o',\
ms=8, alpha=.5, color='yellow', visible=True, label = '_nolegend_')
self.is_hZoom = False
self.canvas.mpl_connect('pick_event', self.OnPickPlot)
def OnPickPlot(self, event):
self.pickIndex = event.ind[0]
try:
self.textHandle.remove()
except:
pass
self.curText = event.artist.get_label()
if self.x != None and self.y != None:
#I'd rather do the following ALWAYS but it seems difficult to do in terms of keeping track of which arrays were plotted when multiple plots are present
paramVal = N.take(self.y, [self.pickIndex])
self.handleA.set_data(N.take(self.x, [self.pickIndex]), [paramVal])
else:
paramVal = event.mouseevent.ydata
self.handleA.set_data([event.mouseevent.xdata], [paramVal])
showText = '%s: %s' % (self.dataLabels[self.pickIndex], paramVal[0])
# showText = '%s: %s'%self.dataLabels[self.pickIndex]
self.textHandle = self.canvas.ax.text(0.03, 0.95, showText, fontsize=9,\
bbox=dict(facecolor='yellow', alpha=0.1),\
transform=self.canvas.ax.transAxes, va='top')
self.handleA.set_visible(True)
self.canvas.draw()
def mpl2Clip(self):
try:
self.canvas.fig.savefig(self.tempPath)
tempImg = QtGui.QImage(self.tempPath)
self.cb = QtGui.QApplication.clipboard()
self.cb.setImage(tempImg)
except:
print 'Error copying figure to clipboard'
errorMsg = "Sorry: %s\n\n:%s\n" % (sys.exc_type, sys.exc_value)
print errorMsg
def home(self, *args): NavigationToolbar2QT.home(self,*args) self.homeClicked.emit()
class MPL_Widget(QtGui.QWidget):
def __init__(self, parent = None, enableAutoScale = True, enableCSV = False, enableEdit = True, doublePlot = False, layoutDir = 'h'):
QtGui.QWidget.__init__(self, parent)
if doublePlot:
self.canvas = DoubleMyMplCanvas()
else:
self.canvas = MyMplCanvas()
self.toolbar = NavigationToolbar(self.canvas, self.canvas)
# self.toolbar.hide()
self.installEventFilter(EventFilter(self))
if layoutDir == 'v':
self.toolbar.setOrientation(QtCore.Qt.Vertical)
self.layout = QtGui.QHBoxLayout()
elif layoutDir == 'h':
self.layout = QtGui.QVBoxLayout()
# self.layout.addWidget(self.canvas)
# self.layout.addWidget(self.toolbar)
self.layout.addWidget(self.toolbar)
self.layout.addWidget(self.canvas)
self.setLayout(self.layout)
##########################
self.dataLabels = None
###########SAVING FIGURE TO CLIPBOARD##########
self.cb = None #will be used for the clipboard
self.tempPath = getHomeDir()
self.tempPath = os.path.join(self.tempPath,'tempMPL.png')
if enableEdit:
self.enableEdit()
self.lineDict = None
self.addLegend = False
self.x = None#used for picker
self.y = None
#######SAVING FIGURE DATA############################
if enableCSV:
self.enableCSV()
########### HELPER FUNCTIONS #########################
self.clearPlotAction = QtGui.QAction("Clear Plot", self)
self.addAction(self.clearPlotAction)
QtCore.QObject.connect(self, QtCore.SIGNAL("triggered()"), self.clearPlot)
# def focusOutEvent(self, event):
# print "Focus Out"
def clearPlot(self):
print "Clear Plot"
self.canvas.ax.cla()
self.canvas.format_labels()
self.canvas.draw()
# def __initContextMenus__(self):
# self.clearPlotAction = QtGui.QAction("Clear Plot", self)
# self.addAction(self.clearPlotAction)
# QtCore.QObject.connect(self, QtCore.SIGNAL("triggered()"), self.clearPlot)
#
# self.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
# self.connect(self, QtCore.SIGNAL("customContextMenuRequested(QPoint)"), self.plotTabContext)
# def plotTabContext(self, point):
# '''Create a menu for mainTabWidget'''
# plotCT_menu = QtGui.QMenu("Menu", self)
# plotCT_menu.addAction(self.Zoom)
# plotCT_menu.addAction(self.actionAutoScale)
# plotCT_menu.addSeparator()
# plotCT_menu.addAction(self.clearPlotAction)
# plotCT_menu.exec_(self.mapToGlobal(point))
def focusEvent(self, event):
self.enableAutoScale()
self.enableZoom()
self.enableClip()
# self.enableCSV()
#print "Focus In %s"%self.canvas.plotTitle
def lossFocusEvent(self, event):
self.disableAutoScale()
self.disableZoom()
self.disableClip()
# self.disableCSV()
#print "Focus Out %s"%self.canvas.plotTitle
def enableClip(self):
self.mpl2ClipAction = QtGui.QAction("Save to Clipboard", self)
self.mpl2ClipAction.setShortcut("Ctrl+C")
self.addAction(self.mpl2ClipAction)
QtCore.QObject.connect(self.mpl2ClipAction,QtCore.SIGNAL("triggered()"), self.mpl2Clip)
def disableClip(self):
QtCore.QObject.disconnect(self.mpl2ClipAction,QtCore.SIGNAL("triggered()"), self.mpl2Clip)
self.removeAction(self.mpl2ClipAction)
def enableEdit(self):
self.editAction = QtGui.QAction("Edit Line Properties", self)
self.editAction.setShortcut("Ctrl+Alt+E")
self.addAction(self.editAction)
QtCore.QObject.connect(self.editAction,QtCore.SIGNAL("triggered()"), self.editPlotProperties)
def disableEdit(self):
if self.editAction:
QtCore.QObject.disconnect(self.editAction,QtCore.SIGNAL("triggered()"), self.editPlotProperties)
self.removeAction(self.editAction)
def enableZoom(self):
self.Zoom = QtGui.QAction("Zoom", self)
self.Zoom.setShortcut("Ctrl+Z")
self.addAction(self.Zoom)
QtCore.QObject.connect(self.Zoom, QtCore.SIGNAL("triggered()"), self.ZoomToggle)
def disableZoom(self):
if self.Zoom != None:
QtCore.QObject.disconnect(self.Zoom,QtCore.SIGNAL("triggered()"), self.ZoomToggle)
self.removeAction(self.Zoom)
def disableAutoScale(self):
if self.actionAutoScale != None:
QtCore.QObject.disconnect(self.actionAutoScale,QtCore.SIGNAL("triggered()"), self.autoscale_plot)
self.removeAction(self.actionAutoScale)
def enableAutoScale(self):
self.actionAutoScale = QtGui.QAction("AutoScale", self)#self.MainWindow)
self.actionAutoScale.setShortcut("Ctrl+A")
self.addAction(self.actionAutoScale)
QtCore.QObject.connect(self.actionAutoScale,QtCore.SIGNAL("triggered()"), self.autoscale_plot)
def enableCSV(self):
self.saveCSVAction = QtGui.QAction("Save to CSV", self)
self.saveCSVAction.setShortcut("Ctrl+Alt+S")
self.addAction(self.saveCSVAction)
QtCore.QObject.connect(self.saveCSVAction,QtCore.SIGNAL("triggered()"), self.save2CSV)
def disableCSV(self):
QtCore.QObject.disconnect(self.saveCSVAction,QtCore.SIGNAL("triggered()"), self.save2CSV)
self.removeAction(self.saveCSVAction)
def setLineDict(self):
self.lineDict = {}
lineList = self.canvas.ax.get_lines()
if lineList > 0:
for line in lineList:
self.lineDict[line.get_label()]=line
def editPlotProperties(self):
print "Edit Enabled"
self.setLineDict()
# if len(self.lineDict) > 0:
curAx = self.canvas.ax
if POD(self.lineDict, curAx = curAx, parent = self).exec_():
if self.addLegend:
curAx.legend(borderaxespad = 0.03, axespad=0.25)
self.canvas.format_labels()
self.canvas.draw()
else:
print "Cancel"
def ZoomToggle(self):
self.toolbar.zoom() #this implements the classic zoom
# if self.hZoom:
# self.hZoom = False
# self.span.visible = False
# else:
# self.hZoom = True
# self.span.visible = True
def autoscale_plot(self):
self.toolbar.home() #implements the classic return to home
# self.canvas.ax.autoscale_view(tight = False, scalex=True, scaley=True)
# self.canvas.draw()
# self.emit(QtCore.SIGNAL("autoScaleAxis(bool)"),True)
# def onclick(self, event):
# #sets up the maximum Y level to be displayed after the zoom.
# #if not set then it maxes to the largest point in the data
# #not necessarily the local max
# if event.ydata != None:
# self.localYMax = int(event.ydata)
#
# def onselect(self, xmin, xmax):
# #print xmin, xmax
# if self.hZoom:
# self.canvas.ax.set_ylim(ymax = self.localYMax)
# self.canvas.ax.set_xlim(xmin, xmax)
# self.canvas.draw()
def save2CSV(self):
path = self.SFDialog()
if path != None:
try:
lines = self.canvas.ax.get_lines()
data2write = []
for line in lines:
data2write.append(line.get_data()[0])
data2write.append(line.get_data()[1])
print data2write
data2write = N.array(data2write)
data2write.dtype = N.float32
N.savetxt(str(path), N.transpose(data2write), delimiter = ',', fmt='%.4f')
except:
try:
#this is for the case where the data may not be in float format?
N.savetxt(str(path), N.transpose(data2write), delimiter = ',')
except:
print 'Error saving figure data'
errorMsg = "Sorry: %s\n\n:%s\n"%(sys.exc_type, sys.exc_value)
print errorMsg
def SFDialog(self):
fileName = QtGui.QFileDialog.getSaveFileName(self,
"Select File to Save",
"",
"csv Files (*.csv)")
if not fileName.isEmpty():
print fileName
return fileName
else:
return None
def removePicker(self):
try:
self.handleA.remove()
except:
print ""
def setData(self, x, y):
if x != None:
self.x = x
if y != None:
self.y = y
def addPicker(self):
'''Sets up the plot variables used for interaction'''
self.handleA, = self.canvas.ax.plot([0], [0], 'o',\
ms=8, alpha=.5, color='yellow', visible=True, label = '_nolegend_')
self.is_hZoom = False
self.canvas.mpl_connect('pick_event', self.OnPickPlot)
def OnPickPlot(self, event):
self.pickIndex = event.ind[0]
try:
self.textHandle.remove()
except:
pass
self.curText = event.artist.get_label()
if self.x != None and self.y != None:
#I'd rather do the following ALWAYS but it seems difficult to do in terms of keeping track of which arrays were plotted when multiple plots are present
paramVal = N.take(self.y, [self.pickIndex])
self.handleA.set_data(N.take(self.x, [self.pickIndex]), [paramVal])
else:
paramVal = event.mouseevent.ydata
self.handleA.set_data([event.mouseevent.xdata], [paramVal])
showText = '%s: %s'%(self.dataLabels[self.pickIndex],paramVal[0])
# showText = '%s: %s'%self.dataLabels[self.pickIndex]
self.textHandle = self.canvas.ax.text(0.03, 0.95, showText, fontsize=9,\
bbox=dict(facecolor='yellow', alpha=0.1),\
transform=self.canvas.ax.transAxes, va='top')
self.handleA.set_visible(True)
self.canvas.draw()
def mpl2Clip(self):
try:
self.canvas.fig.savefig(self.tempPath)
tempImg = QtGui.QImage(self.tempPath)
self.cb = QtGui.QApplication.clipboard()
self.cb.setImage(tempImg)
except:
print 'Error copying figure to clipboard'
errorMsg = "Sorry: %s\n\n:%s\n"%(sys.exc_type, sys.exc_value)
print errorMsg
