def printPlot(self):
"""Open the print dialog and print the plot.
Use :meth:`Plot.saveGraph` to print the plot.
:return: True if successful
"""
# Init printer and start printer dialog
dialog = qt.QPrintDialog(self.getPrinter(), self.plot)
dialog.setWindowTitle('Print Plot')
if not dialog.exec_():
return False
# Save Plot as PNG and make a pixmap from it with default dpi
pngData = _plotAsPNG(self.plot)
pixmap = qt.QPixmap()
pixmap.loadFromData(pngData, 'png')
xScale = self.getPrinter().pageRect().width() / pixmap.width()
yScale = self.getPrinter().pageRect().height() / pixmap.height()
scale = min(xScale, yScale)
# Draw pixmap with painter
painter = qt.QPainter()
if not painter.begin(self.getPrinter()):
return False
painter.drawPixmap(0, 0,
pixmap.width() * scale,
pixmap.height() * scale,
pixmap)
painter.end()
return True
def _triggered(self, checked=False):
plot3d = self.getPlot3DWidget()
if plot3d is None:
_logger.error('Cannot print widget, no associated Plot3DWidget')
else:
printer = self.getPrinter()
dialog = qt.QPrintDialog(printer, plot3d)
dialog.setWindowTitle('Print Plot3D snapshot')
if not dialog.exec_():
return
image = plot3d.grabGL()
# Draw pixmap with painter
painter = qt.QPainter()
if not painter.begin(printer):
return
if (printer.pageRect().width() < image.width()
or printer.pageRect().height() < image.height()):
# Downscale to page
xScale = printer.pageRect().width() / image.width()
yScale = printer.pageRect().height() / image.height()
scale = min(xScale, yScale)
else:
scale = 1.
rect = qt.QRectF(0, 0, scale * image.width(),
scale * image.height())
painter.drawImage(rect, image)
painter.end()
def printPlotAsWidget(self):
"""Open the print dialog and print the plot.
Use :meth:`QWidget.render` to print the plot
:return: True if successful
"""
dialog = qt.QPrintDialog(self.getPrinter(), self.plot)
dialog.setWindowTitle('Print Plot')
if not dialog.exec_():
return False
# Print a snapshot of the plot widget at the top of the page
widget = self.plot.centralWidget()
painter = qt.QPainter()
if not painter.begin(self.getPrinter()):
return False
pageRect = self.getPrinter().pageRect()
xScale = pageRect.width() / widget.width()
yScale = pageRect.height() / widget.height()
scale = min(xScale, yScale)
painter.translate(pageRect.width() / 2., 0.)
painter.scale(scale, scale)
painter.translate(-widget.width() / 2., 0.)
widget.render(painter)
painter.end()
return True
def setup(self):
"""Open a print dialog to ensure the :attr:`printer` is set.
If the setting fails or is cancelled, :attr:`printer` is reset to
*None*.
"""
if self.printer is None:
self.printer = qt.QPrinter()
if self.printDialog is None:
self.printDialog = qt.QPrintDialog(self.printer, self)
if self.printDialog.exec_():
if self.printer.width() <= 0 or self.printer.height() <= 0:
self.message = qt.QMessageBox(self)
self.message.setIcon(qt.QMessageBox.Critical)
self.message.setText(
"Unknown library error \non printer initialization")
self.message.setWindowTitle("Library Error")
self.message.setModal(0)
self.printer = None
return
self.printer.setFullPage(True)
self._updatePrinter()
else:
# printer setup cancelled, check for a possible previous configuration
if self.page is None:
# not initialized
self.printer = None
