--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/eric6/Graphics/PixmapDiagram.py Sun Apr 14 15:09:21 2019 +0200 @@ -0,0 +1,406 @@ +# -*- coding: utf-8 -*- + +# Copyright (c) 2004 - 2019 Detlev Offenbach <detlev@die-offenbachs.de> +# + +""" +Module implementing a dialog showing a pixmap. +""" + +from __future__ import unicode_literals + +from PyQt5.QtCore import Qt, QSize, QEvent +from PyQt5.QtGui import QPalette, QImage, QPixmap, QPainter, QFont, QColor +from PyQt5.QtWidgets import QLabel, QSizePolicy, QScrollArea, QAction, QMenu, \ + QToolBar +from PyQt5.QtPrintSupport import QPrinter, QPrintDialog + +from E5Gui import E5MessageBox +from E5Gui.E5MainWindow import E5MainWindow +from E5Gui.E5ZoomWidget import E5ZoomWidget + +import UI.Config + +import Preferences +from Globals import qVersionTuple + + +class PixmapDiagram(E5MainWindow): + """ + Class implementing a dialog showing a pixmap. + """ + ZoomLevels = [ + 1, 3, 5, 7, 9, + 10, 20, 30, 50, 67, 80, 90, + 100, + 110, 120, 133, 150, 170, 200, 240, 300, 400, + 500, 600, 700, 800, 900, 1000, + ] + ZoomLevelDefault = 100 + + def __init__(self, pixmap, parent=None, name=None): + """ + Constructor + + @param pixmap filename of a graphics file to show (string) + @param parent parent widget of the view (QWidget) + @param name name of the view widget (string) + """ + super(PixmapDiagram, self).__init__(parent) + if name: + self.setObjectName(name) + else: + self.setObjectName("PixmapDiagram") + self.setWindowTitle(self.tr("Pixmap-Viewer")) + + self.pixmapLabel = QLabel() + self.pixmapLabel.setObjectName("pixmapLabel") + self.pixmapLabel.setBackgroundRole(QPalette.Base) + self.pixmapLabel.setSizePolicy( + QSizePolicy.Ignored, QSizePolicy.Ignored) + self.pixmapLabel.setScaledContents(True) + + self.pixmapView = QScrollArea() + self.pixmapView.setObjectName("pixmapView") + self.pixmapView.setBackgroundRole(QPalette.Dark) + self.pixmapView.setWidget(self.pixmapLabel) + + self.setCentralWidget(self.pixmapView) + + self.__zoomWidget = E5ZoomWidget( + UI.PixmapCache.getPixmap("zoomOut.png"), + UI.PixmapCache.getPixmap("zoomIn.png"), + UI.PixmapCache.getPixmap("zoomReset.png"), self) + self.statusBar().addPermanentWidget(self.__zoomWidget) + self.__zoomWidget.setMapping( + PixmapDiagram.ZoomLevels, PixmapDiagram.ZoomLevelDefault) + self.__zoomWidget.valueChanged.connect(self.__doZoom) + + # polish up the dialog + self.resize(QSize(800, 600).expandedTo(self.minimumSizeHint())) + + self.pixmapfile = pixmap + self.status = self.__showPixmap(self.pixmapfile) + + self.__initActions() + self.__initContextMenu() + self.__initToolBars() + + self.grabGesture(Qt.PinchGesture) + + def __initActions(self): + """ + Private method to initialize the view actions. + """ + self.closeAct = \ + QAction(UI.PixmapCache.getIcon("close.png"), + self.tr("Close"), self) + self.closeAct.triggered.connect(self.close) + + self.printAct = \ + QAction(UI.PixmapCache.getIcon("print.png"), + self.tr("Print"), self) + self.printAct.triggered.connect(self.__printDiagram) + + self.printPreviewAct = \ + QAction(UI.PixmapCache.getIcon("printPreview.png"), + self.tr("Print Preview"), self) + self.printPreviewAct.triggered.connect(self.__printPreviewDiagram) + + def __initContextMenu(self): + """ + Private method to initialize the context menu. + """ + self.__menu = QMenu(self) + self.__menu.addAction(self.closeAct) + self.__menu.addSeparator() + self.__menu.addAction(self.printPreviewAct) + self.__menu.addAction(self.printAct) + + self.setContextMenuPolicy(Qt.CustomContextMenu) + self.customContextMenuRequested.connect(self.__showContextMenu) + + def __showContextMenu(self, coord): + """ + Private slot to show the context menu of the listview. + + @param coord the position of the mouse pointer (QPoint) + """ + self.__menu.popup(self.mapToGlobal(coord)) + + def __initToolBars(self): + """ + Private method to populate the toolbars with our actions. + """ + self.windowToolBar = QToolBar(self.tr("Window"), self) + self.windowToolBar.setIconSize(UI.Config.ToolBarIconSize) + self.windowToolBar.addAction(self.closeAct) + + self.graphicsToolBar = QToolBar(self.tr("Graphics"), self) + self.graphicsToolBar.setIconSize(UI.Config.ToolBarIconSize) + self.graphicsToolBar.addAction(self.printPreviewAct) + self.graphicsToolBar.addAction(self.printAct) + + self.addToolBar(Qt.TopToolBarArea, self.windowToolBar) + self.addToolBar(Qt.TopToolBarArea, self.graphicsToolBar) + + def __showPixmap(self, filename): + """ + Private method to show a file. + + @param filename name of the file to be shown (string) + @return flag indicating success (boolean) + """ + image = QImage(filename) + if image.isNull(): + E5MessageBox.warning( + self, + self.tr("Pixmap-Viewer"), + self.tr( + """<p>The file <b>{0}</b> cannot be displayed.""" + """ The format is not supported.</p>""").format(filename)) + return False + + self.pixmapLabel.setPixmap(QPixmap.fromImage(image)) + self.pixmapLabel.adjustSize() + return True + + def getDiagramName(self): + """ + Public method to retrieve a name for the diagram. + + @return name for the diagram + """ + return self.pixmapfile + + def getStatus(self): + """ + Public method to retrieve the status of the canvas. + + @return flag indicating a successful pixmap loading (boolean) + """ + return self.status + + def wheelEvent(self, evt): + """ + Protected method to handle wheel events. + + @param evt reference to the wheel event (QWheelEvent) + """ + if evt.modifiers() & Qt.ControlModifier: + if qVersionTuple() >= (5, 0, 0): + delta = evt.angleDelta().y() + else: + delta = evt.delta() + if delta < 0: + self.__zoomOut() + elif delta > 0: + self.__zoomIn() + evt.accept() + return + + super(PixmapDiagram, self).wheelEvent(evt) + + def event(self, evt): + """ + Public method handling events. + + @param evt reference to the event (QEvent) + @return flag indicating, if the event was handled (boolean) + """ + if evt.type() == QEvent.Gesture: + self.gestureEvent(evt) + return True + + return super(PixmapDiagram, self).event(evt) + + def gestureEvent(self, evt): + """ + Protected method handling gesture events. + + @param evt reference to the gesture event (QGestureEvent + """ + pinch = evt.gesture(Qt.PinchGesture) + if pinch: + if pinch.state() == Qt.GestureStarted: + pinch.setTotalScaleFactor(self.__zoom() / 100) + elif pinch.state() == Qt.GestureUpdated: + self.__doZoom(int(pinch.totalScaleFactor() * 100)) + evt.accept() + + ########################################################################### + ## Private menu handling methods below. + ########################################################################### + + def __adjustScrollBar(self, scrollBar, factor): + """ + Private method to adjust a scrollbar by a certain factor. + + @param scrollBar reference to the scrollbar object (QScrollBar) + @param factor factor to adjust by (float) + """ + scrollBar.setValue(int(factor * scrollBar.value() + + ((factor - 1) * scrollBar.pageStep() / 2))) + + def __levelForZoom(self, zoom): + """ + Private method determining the zoom level index given a zoom factor. + + @param zoom zoom factor (integer) + @return index of zoom factor (integer) + """ + try: + index = PixmapDiagram.ZoomLevels.index(zoom) + except ValueError: + for index in range(len(PixmapDiagram.ZoomLevels)): + if zoom <= PixmapDiagram.ZoomLevels[index]: + break + return index + + def __doZoom(self, value): + """ + Private method to set the zoom value in percent. + + @param value zoom value in percent (integer) + """ + oldValue = self.__zoom() + if value != oldValue: + self.pixmapLabel.resize( + value / 100 * self.pixmapLabel.pixmap().size()) + + factor = value / oldValue + self.__adjustScrollBar( + self.pixmapView.horizontalScrollBar(), factor) + self.__adjustScrollBar( + self.pixmapView.verticalScrollBar(), factor) + + self.__zoomWidget.setValue(value) + + def __zoomIn(self): + """ + Private method to zoom into the pixmap. + """ + index = self.__levelForZoom(self.__zoom()) + if index < len(PixmapDiagram.ZoomLevels) - 1: + self.__doZoom(PixmapDiagram.ZoomLevels[index + 1]) + + def __zoomOut(self): + """ + Private method to zoom out of the pixmap. + """ + index = self.__levelForZoom(self.__zoom()) + if index > 0: + self.__doZoom(PixmapDiagram.ZoomLevels[index - 1]) + + def __zoomReset(self): + """ + Private method to reset the zoom value. + """ + self.__doZoom(PixmapDiagram.ZoomLevels[PixmapDiagram.ZoomLevelDefault]) + + def __zoom(self): + """ + Private method to get the current zoom factor in percent. + + @return current zoom factor in percent (integer) + """ + return int(self.pixmapLabel.width() / + self.pixmapLabel.pixmap().width() * 100.0) + + def __printDiagram(self): + """ + Private slot called to print the diagram. + """ + printer = QPrinter(mode=QPrinter.ScreenResolution) + printer.setFullPage(True) + if Preferences.getPrinter("ColorMode"): + printer.setColorMode(QPrinter.Color) + else: + printer.setColorMode(QPrinter.GrayScale) + if Preferences.getPrinter("FirstPageFirst"): + printer.setPageOrder(QPrinter.FirstPageFirst) + else: + printer.setPageOrder(QPrinter.LastPageFirst) + printer.setPrinterName(Preferences.getPrinter("PrinterName")) + + printDialog = QPrintDialog(printer, self) + if printDialog.exec_(): + self.__print(printer) + + def __printPreviewDiagram(self): + """ + Private slot called to show a print preview of the diagram. + """ + from PyQt5.QtPrintSupport import QPrintPreviewDialog + + printer = QPrinter(mode=QPrinter.ScreenResolution) + printer.setFullPage(True) + if Preferences.getPrinter("ColorMode"): + printer.setColorMode(QPrinter.Color) + else: + printer.setColorMode(QPrinter.GrayScale) + if Preferences.getPrinter("FirstPageFirst"): + printer.setPageOrder(QPrinter.FirstPageFirst) + else: + printer.setPageOrder(QPrinter.LastPageFirst) + printer.setPageMargins( + Preferences.getPrinter("LeftMargin") * 10, + Preferences.getPrinter("TopMargin") * 10, + Preferences.getPrinter("RightMargin") * 10, + Preferences.getPrinter("BottomMargin") * 10, + QPrinter.Millimeter + ) + printer.setPrinterName(Preferences.getPrinter("PrinterName")) + + preview = QPrintPreviewDialog(printer, self) + preview.paintRequested[QPrinter].connect(self.__print) + preview.exec_() + + def __print(self, printer): + """ + Private slot to the actual printing. + + @param printer reference to the printer object (QPrinter) + """ + painter = QPainter() + painter.begin(printer) + + # calculate margin and width of printout + font = QFont("times", 10) + painter.setFont(font) + fm = painter.fontMetrics() + fontHeight = fm.lineSpacing() + marginX = printer.pageRect().x() - printer.paperRect().x() + marginX = Preferences.getPrinter("LeftMargin") * \ + int(printer.resolution() / 2.54) - marginX + marginY = printer.pageRect().y() - printer.paperRect().y() + marginY = Preferences.getPrinter("TopMargin") * \ + int(printer.resolution() / 2.54) - marginY + + width = printer.width() - marginX - \ + Preferences.getPrinter("RightMargin") * \ + int(printer.resolution() / 2.54) + height = printer.height() - fontHeight - 4 - marginY - \ + Preferences.getPrinter("BottomMargin") * \ + int(printer.resolution() / 2.54) + + # write a foot note + s = self.tr("Diagram: {0}").format(self.getDiagramName()) + tc = QColor(50, 50, 50) + painter.setPen(tc) + painter.drawRect(marginX, marginY, width, height) + painter.drawLine(marginX, marginY + height + 2, + marginX + width, marginY + height + 2) + painter.setFont(font) + painter.drawText(marginX, marginY + height + 4, width, + fontHeight, Qt.AlignRight, s) + + # render the diagram + size = self.pixmapLabel.pixmap().size() + size.scale(QSize(width - 10, height - 10), # 5 px inner margin + Qt.KeepAspectRatio) + painter.setViewport(marginX + 5, marginY + 5, + size.width(), size.height()) + painter.setWindow(self.pixmapLabel.pixmap().rect()) + painter.drawPixmap(0, 0, self.pixmapLabel.pixmap()) + painter.end()