def export_plot(plt, filename, width=None, height=None):
""" Export an existing plot to an image and save with given filename.
Optionally, image width and height in pixels can be specified. """
e = ImageExporter(plt.plotItem)
if width is not None: e.parameters()['width'] = width
if height is not None: e.parameters()['height'] = height
e.export(filename)
def save_plot(self):
from PySide2.QtWidgets import QFileDialog
(fileName, selectedFilter) = QFileDialog.getSaveFileName(
None, "Open Image", path.expanduser("~"),
"Image Files (*.png *.jpg *.bmp)")
if fileName:
from pyqtgraph.exporters import ImageExporter
exporter = ImageExporter(self.fftGraphicsView.plotItem)
exporter.export(fileName)
def export(self, fname, export_type="image"):
"""
Save the item as an image
"""
if export_type == "image":
exporter = ImageExporter(self.scene())
elif export_type == "svg":
exporter = SVGExporter(self.scene())
exporter.export(fname)
del exporter
def test_ImageExporter_toBytes():
p = pg.plot()
p.hideAxis('bottom')
p.hideAxis('left')
exp = ImageExporter(p.getPlotItem())
qimg = exp.export(toBytes=True)
qimg = qimg.convertToFormat(QtGui.QImage.Format.Format_RGBA8888)
data = fn.qimage_to_ndarray(qimg)
black = (0, 0, 0, 255)
assert np.all(data == black), "Exported image should be entirely black."
def saveFig(): # Get pix map of window and save png
dirName = "/home"
fname = "sampleWindow.png"
fname2 = "samplePlotWindow.png"
dataPath = dirName + fname
dataPath2 = dirName + fname2
locObj = QtGui.QPixmap.grabWindow(glw.winId())
locObj.save(dataPath, 'png')
exporter = IE.ImageExporter(plt)
exporter.export(dataPath2)
def __init__(self, parent, name, plot, statusbar=None):
super(SaveChartDialog, self).__init__(parent)
self.setupUi(self)
self.name = name
self.plot = plot
self.exporter = ImageExporter(self.plot.getPlotItem())
self.__x = self.plot.size().width()
self.__y = self.plot.size().height()
self.__aspectRatio = self.__x / self.__y
self.widthPixels.setValue(self.__x)
self.heightPixels.setValue(self.__y)
self.statusbar = statusbar
self.__dialog = QFileDialog(parent=self)
def draw_image(self, index, color, x, y, split_name_list):
self.plt.plot(x, y, pen=pg.mkPen(QColor(color[0], color[1], color[2])))
pltItem = self.plt.getPlotItem()
left_axis = pltItem.getAxis("left")
left_axis.enableAutoSIPrefix(False)
font = QFont()
font.setPixelSize(16)
left_axis.tickFont = font
bottom_axis = pltItem.getAxis("bottom")
bottom_axis.tickFont = font
exporter = ImageExporter(self.plt.getPlotItem())
exporter.parameters()['width'] = 800
exporter.parameters()['height'] = 600
split_name = 'tmp/' + 'split_' + str(index) + '.png'
exporter.export(split_name)
split_name_list.append(split_name)
def save(self):
self.timer.stop()
self.deque = deque()
firstName = self.plots.lineName.text()
firstName = firstName[0].upper() + firstName[1:]
lastName = self.plots.lineSurname.text().split(' ')
lastName[-1] = lastName[-1][0].upper() + lastName[-1][1:]
lastName = ' '.join(lastName)
name = '{} {}'.format(firstName, lastName)
self.globalScores[name] = [
int(self.plots.lineAge.text()), self.maxPower, self.maxFrequency
]
self.localScores[name] = [
int(self.plots.lineAge.text()), self.maxPower, self.maxFrequency
]
# update highscores
self.dumpScores()
# update scores window
self.updateScoresSignal.emit()
pen = pg.mkPen(color=(0, 0, 0), width=2)
pg.setConfigOption('background', 'w')
timePlot = pg.PlotWidget()
timePlot.setRange(yRange=[1.2 * -2**15, 1.2 * 2**15])
timePlot.setBackground(None)
timePlot.hideAxis('left')
timePlot.hideAxis('bottom')
timePlot.resize(800, 200)
timeCurve = timePlot.plot(self.timeAxis,
self.audioData[::self.DECIMATION],
pen=pen)
# create rotated version of spectrum
# labelStyle = {'color': '#000', 'font-size': '16px'}
# spectrumPlot = pg.PlotWidget()
# # spectrumPlot.setTitle(title='Power Spectral Density', **labelStyle)
# # spectrumPlot.setLabel('bottom', 'frequency (Hz)', **labelStyle)
# # spectrumPlot.setLabel('left', 'power spectral density (dB)', **labelStyle)
# spectrumPlot.getAxis("left").setWidth(200)
# spectrumPlot.getAxis('left').setPen(pen)
# spectrumPlot.getAxis('bottom').setPen(pen)
# #spectrumPlot.setBackground(None)
# spectrumPlot.getPlotItem().setLogMode(True, False)
# spectrumPlot.getPlotItem().setRange(
# xRange=[1, np.log10(self.SAMPLE_RATE // 2)],
# yRange=[0, self.SPECTRUM_MAX])
# spectrumPlot.getAxis('left').setTicks(
# [[(x, str(x)) for x in range(self.SPECTRUM_MIN, self.SPECTRUM_MAX, 20)]])
# spectrumPlot.getAxis('bottom').setTicks(
# [[(x, str(int(10**x))) for x in [1, 2, 3, 4, 5]]])
# spectrumCurve = spectrumPlot.plot(self.spectrumScale, self.maxSpectrum, pen=pen)
# export plots to png
# exporter = ImageExporter(spectrumPlot.plotItem)
# exporter.parameters()['width'] = 2000
# exporter.export('images/spectrum.png')
exporter = ImageExporter(timePlot.plotItem)
exporter.parameters()['width'] = 2000
exporter.export('images/timeseries.png')
exporter = ImageExporter(self.spectrogramImage)
exporter.parameters()['width'] = 2000
exporter.export('images/spectrogram.png')
diplomaPath = createDiploma(firstName, lastName, self.maxPower,
self.maxFrequency)
time.sleep(0.5)
def plot(self, dataToPlot, times=None, save=False):
if not dataToPlot.equals(self.data):
self.minYLeft, self.maxYLeft, self.minYRight, self.maxYRight = None, None, None, None
self.data = dataToPlot
self.data.to_clipboard()
self.plotItem.clear()
try:
self.plotItem.scene().removeItem(self.plot2)
except AttributeError:
pass
x1 = self.data.iloc[:, 0]
y1 = self.data.iloc[:, 1]
y2 = self.data.iloc[:, 2]
x2 = self.data.iloc[:, 3]
y3 = self.data.iloc[:, 4]
for g in (x1, y1, y2, x2, y3):
g.dropna(inplace=True)
x1 = [i.to_pydatetime().timestamp() for i in x1.to_list()]
x2 = [i.to_pydatetime().timestamp() for i in x2.to_list()]
self.plotItem.addLegend().clear()
self.plotItem.plot(x1, y1, pen=pg.mkPen('g', width=3), name='Pressure')
self.plotItem.plot(x1,
y2,
pen=pg.mkPen('r', width=3),
name="Temperature")
self.plot2.clear()
self.plotItem.scene().addItem(self.plot2)
self.plotItem.getAxis('right').linkToView(self.plot2)
self.plot2.setXLink(self.plotItem)
self.updateViews()
self.plotItem.vb.sigResized.connect(self.updateViews)
curve3 = pg.PlotDataItem(x2,
y3,
name='Depth',
pen=pg.mkPen('b', width=3))
self.plot2.addItem(curve3)
self.plotItem.addLegend().addItem(curve3, 'Depth')
self.plotItem.getViewBox().autoRange()
range = self.plotItem.getViewBox().viewRange()[0]
minXFirstLeft, maxXFirstLeft = range[0], range[1]
self.plot2.autoRange()
range = self.plot2.viewRange()[0]
minXFirstRight, maxXFirstRight = range[0], range[1]
self.plot2.setXRange(min(minXFirstRight, minXFirstLeft),
max(maxXFirstRight, maxXFirstLeft))
if (self.minYLeft, self.maxYLeft, self.minYRight,
self.maxYRight) == (None, None, None, None):
range = self.plotItem.getViewBox().viewRange()[1]
self.minYLeft, self.maxYLeft = range[0], range[1]
range = self.plot2.viewRange()[1]
self.minYRight, self.maxYRight = range[0], range[1]
if save:
exporter = ImageExporter(self.plotItem)
exporter.parameters()['width'] = 580
exporter.parameters()['height'] = 450
buffer = QBuffer()
buffer.open(QIODevice.ReadWrite)
exporter.export(toBytes=True).save(buffer, "PNG")
buffer.seek(0)
fig = BytesIO(buffer.readAll())
return fig
else:
if times is not None:
self.button1.setVisible(True)
self.button2.setVisible(True)
vLines1, vLines2 = [self.makeInfLine(time, 0) for time in times[0]], \
[self.makeInfLine(time, 1) for time in times[1]]
for line in chain(vLines1, vLines2):
self.plotItem.addItem(line)
line.infLineSignal.connect(self.emittingToMain)
def test_ImageExporter_filename_dialog():
"""Tests ImageExporter code path that opens a file dialog. Regression test
for pull request 1133."""
p = pg.plot()
exp = ImageExporter(p.getPlotItem())
exp.export()
def save_plot_to_clipboard_as_png(self, plot_item):
self.img_exporter = ImageExporter(plot_item)
self.img_exporter.export(copy=True)
def saveFig():
dirName = "/mnt/odrive/HPD/Cooldown150605/"
fname = "test.png"
dataPath = dirName + fname
exporter = IE.ImageExporter(plt)
exporter.export(dataPath)