mirror of
https://github.com/pyqt/examples.git
synced 2025-08-23 01:57:10 +00:00
54f8b69baa
The commits since then moved many files into the src/ subdirectory. This would have broken existing links on the internet. To solve this, we leave the master branch as-is and make such breaking changes on a different branch.
322 lines
11 KiB
Python
322 lines
11 KiB
Python
#!/usr/bin/env python
|
|
|
|
|
|
#############################################################################
|
|
##
|
|
## Copyright (C) 2013 Riverbank Computing Limited.
|
|
## Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
|
|
## All rights reserved.
|
|
##
|
|
## This file is part of the examples of PyQt.
|
|
##
|
|
## $QT_BEGIN_LICENSE:BSD$
|
|
## You may use this file under the terms of the BSD license as follows:
|
|
##
|
|
## "Redistribution and use in source and binary forms, with or without
|
|
## modification, are permitted provided that the following conditions are
|
|
## met:
|
|
## * Redistributions of source code must retain the above copyright
|
|
## notice, this list of conditions and the following disclaimer.
|
|
## * Redistributions in binary form must reproduce the above copyright
|
|
## notice, this list of conditions and the following disclaimer in
|
|
## the documentation and/or other materials provided with the
|
|
## distribution.
|
|
## * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor
|
|
## the names of its contributors may be used to endorse or promote
|
|
## products derived from this software without specific prior written
|
|
## permission.
|
|
##
|
|
## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
|
|
## $QT_END_LICENSE$
|
|
##
|
|
#############################################################################
|
|
|
|
|
|
from PyQt5.QtCore import QDir, QPoint, QRect, QSize, Qt
|
|
from PyQt5.QtGui import QImage, QImageWriter, QPainter, QPen, qRgb
|
|
from PyQt5.QtWidgets import (QAction, QApplication, QColorDialog, QFileDialog,
|
|
QInputDialog, QMainWindow, QMenu, QMessageBox, QWidget)
|
|
from PyQt5.QtPrintSupport import QPrintDialog, QPrinter
|
|
|
|
|
|
class ScribbleArea(QWidget):
|
|
def __init__(self, parent=None):
|
|
super(ScribbleArea, self).__init__(parent)
|
|
|
|
self.setAttribute(Qt.WA_StaticContents)
|
|
self.modified = False
|
|
self.scribbling = False
|
|
self.myPenWidth = 1
|
|
self.myPenColor = Qt.blue
|
|
self.image = QImage()
|
|
self.lastPoint = QPoint()
|
|
|
|
def openImage(self, fileName):
|
|
loadedImage = QImage()
|
|
if not loadedImage.load(fileName):
|
|
return False
|
|
|
|
newSize = loadedImage.size().expandedTo(self.size())
|
|
self.resizeImage(loadedImage, newSize)
|
|
self.image = loadedImage
|
|
self.modified = False
|
|
self.update()
|
|
return True
|
|
|
|
def saveImage(self, fileName, fileFormat):
|
|
visibleImage = self.image
|
|
self.resizeImage(visibleImage, self.size())
|
|
|
|
if visibleImage.save(fileName, fileFormat):
|
|
self.modified = False
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def setPenColor(self, newColor):
|
|
self.myPenColor = newColor
|
|
|
|
def setPenWidth(self, newWidth):
|
|
self.myPenWidth = newWidth
|
|
|
|
def clearImage(self):
|
|
self.image.fill(qRgb(255, 255, 255))
|
|
self.modified = True
|
|
self.update()
|
|
|
|
def mousePressEvent(self, event):
|
|
if event.button() == Qt.LeftButton:
|
|
self.lastPoint = event.pos()
|
|
self.scribbling = True
|
|
|
|
def mouseMoveEvent(self, event):
|
|
if (event.buttons() & Qt.LeftButton) and self.scribbling:
|
|
self.drawLineTo(event.pos())
|
|
|
|
def mouseReleaseEvent(self, event):
|
|
if event.button() == Qt.LeftButton and self.scribbling:
|
|
self.drawLineTo(event.pos())
|
|
self.scribbling = False
|
|
|
|
def paintEvent(self, event):
|
|
painter = QPainter(self)
|
|
dirtyRect = event.rect()
|
|
painter.drawImage(dirtyRect, self.image, dirtyRect)
|
|
|
|
def resizeEvent(self, event):
|
|
if self.width() > self.image.width() or self.height() > self.image.height():
|
|
newWidth = max(self.width() + 128, self.image.width())
|
|
newHeight = max(self.height() + 128, self.image.height())
|
|
self.resizeImage(self.image, QSize(newWidth, newHeight))
|
|
self.update()
|
|
|
|
super(ScribbleArea, self).resizeEvent(event)
|
|
|
|
def drawLineTo(self, endPoint):
|
|
painter = QPainter(self.image)
|
|
painter.setPen(QPen(self.myPenColor, self.myPenWidth, Qt.SolidLine,
|
|
Qt.RoundCap, Qt.RoundJoin))
|
|
painter.drawLine(self.lastPoint, endPoint)
|
|
self.modified = True
|
|
|
|
rad = self.myPenWidth / 2 + 2
|
|
self.update(QRect(self.lastPoint, endPoint).normalized().adjusted(-rad, -rad, +rad, +rad))
|
|
self.lastPoint = QPoint(endPoint)
|
|
|
|
def resizeImage(self, image, newSize):
|
|
if image.size() == newSize:
|
|
return
|
|
|
|
newImage = QImage(newSize, QImage.Format_RGB32)
|
|
newImage.fill(qRgb(255, 255, 255))
|
|
painter = QPainter(newImage)
|
|
painter.drawImage(QPoint(0, 0), image)
|
|
self.image = newImage
|
|
|
|
def print_(self):
|
|
printer = QPrinter(QPrinter.HighResolution)
|
|
|
|
printDialog = QPrintDialog(printer, self)
|
|
if printDialog.exec_() == QPrintDialog.Accepted:
|
|
painter = QPainter(printer)
|
|
rect = painter.viewport()
|
|
size = self.image.size()
|
|
size.scale(rect.size(), Qt.KeepAspectRatio)
|
|
painter.setViewport(rect.x(), rect.y(), size.width(), size.height())
|
|
painter.setWindow(self.image.rect())
|
|
painter.drawImage(0, 0, self.image)
|
|
painter.end()
|
|
|
|
def isModified(self):
|
|
return self.modified
|
|
|
|
def penColor(self):
|
|
return self.myPenColor
|
|
|
|
def penWidth(self):
|
|
return self.myPenWidth
|
|
|
|
|
|
class MainWindow(QMainWindow):
|
|
def __init__(self):
|
|
super(MainWindow, self).__init__()
|
|
|
|
self.saveAsActs = []
|
|
|
|
self.scribbleArea = ScribbleArea()
|
|
self.setCentralWidget(self.scribbleArea)
|
|
|
|
self.createActions()
|
|
self.createMenus()
|
|
|
|
self.setWindowTitle("Scribble")
|
|
self.resize(500, 500)
|
|
|
|
def closeEvent(self, event):
|
|
if self.maybeSave():
|
|
event.accept()
|
|
else:
|
|
event.ignore()
|
|
|
|
def open(self):
|
|
if self.maybeSave():
|
|
fileName, _ = QFileDialog.getOpenFileName(self, "Open File",
|
|
QDir.currentPath())
|
|
if fileName:
|
|
self.scribbleArea.openImage(fileName)
|
|
|
|
def save(self):
|
|
action = self.sender()
|
|
fileFormat = action.data()
|
|
self.saveFile(fileFormat)
|
|
|
|
def penColor(self):
|
|
newColor = QColorDialog.getColor(self.scribbleArea.penColor())
|
|
if newColor.isValid():
|
|
self.scribbleArea.setPenColor(newColor)
|
|
|
|
def penWidth(self):
|
|
newWidth, ok = QInputDialog.getInt(self, "Scribble",
|
|
"Select pen width:", self.scribbleArea.penWidth(), 1, 50, 1)
|
|
if ok:
|
|
self.scribbleArea.setPenWidth(newWidth)
|
|
|
|
def about(self):
|
|
QMessageBox.about(self, "About Scribble",
|
|
"<p>The <b>Scribble</b> example shows how to use "
|
|
"QMainWindow as the base widget for an application, and how "
|
|
"to reimplement some of QWidget's event handlers to receive "
|
|
"the events generated for the application's widgets:</p>"
|
|
"<p> We reimplement the mouse event handlers to facilitate "
|
|
"drawing, the paint event handler to update the application "
|
|
"and the resize event handler to optimize the application's "
|
|
"appearance. In addition we reimplement the close event "
|
|
"handler to intercept the close events before terminating "
|
|
"the application.</p>"
|
|
"<p> The example also demonstrates how to use QPainter to "
|
|
"draw an image in real time, as well as to repaint "
|
|
"widgets.</p>")
|
|
|
|
def createActions(self):
|
|
self.openAct = QAction("&Open...", self, shortcut="Ctrl+O",
|
|
triggered=self.open)
|
|
|
|
for format in QImageWriter.supportedImageFormats():
|
|
format = str(format)
|
|
|
|
text = format.upper() + "..."
|
|
|
|
action = QAction(text, self, triggered=self.save)
|
|
action.setData(format)
|
|
self.saveAsActs.append(action)
|
|
|
|
self.printAct = QAction("&Print...", self,
|
|
triggered=self.scribbleArea.print_)
|
|
|
|
self.exitAct = QAction("E&xit", self, shortcut="Ctrl+Q",
|
|
triggered=self.close)
|
|
|
|
self.penColorAct = QAction("&Pen Color...", self,
|
|
triggered=self.penColor)
|
|
|
|
self.penWidthAct = QAction("Pen &Width...", self,
|
|
triggered=self.penWidth)
|
|
|
|
self.clearScreenAct = QAction("&Clear Screen", self, shortcut="Ctrl+L",
|
|
triggered=self.scribbleArea.clearImage)
|
|
|
|
self.aboutAct = QAction("&About", self, triggered=self.about)
|
|
|
|
self.aboutQtAct = QAction("About &Qt", self,
|
|
triggered=QApplication.instance().aboutQt)
|
|
|
|
def createMenus(self):
|
|
self.saveAsMenu = QMenu("&Save As", self)
|
|
for action in self.saveAsActs:
|
|
self.saveAsMenu.addAction(action)
|
|
|
|
fileMenu = QMenu("&File", self)
|
|
fileMenu.addAction(self.openAct)
|
|
fileMenu.addMenu(self.saveAsMenu)
|
|
fileMenu.addAction(self.printAct)
|
|
fileMenu.addSeparator()
|
|
fileMenu.addAction(self.exitAct)
|
|
|
|
optionMenu = QMenu("&Options", self)
|
|
optionMenu.addAction(self.penColorAct)
|
|
optionMenu.addAction(self.penWidthAct)
|
|
optionMenu.addSeparator()
|
|
optionMenu.addAction(self.clearScreenAct)
|
|
|
|
helpMenu = QMenu("&Help", self)
|
|
helpMenu.addAction(self.aboutAct)
|
|
helpMenu.addAction(self.aboutQtAct)
|
|
|
|
self.menuBar().addMenu(fileMenu)
|
|
self.menuBar().addMenu(optionMenu)
|
|
self.menuBar().addMenu(helpMenu)
|
|
|
|
def maybeSave(self):
|
|
if self.scribbleArea.isModified():
|
|
ret = QMessageBox.warning(self, "Scribble",
|
|
"The image has been modified.\n"
|
|
"Do you want to save your changes?",
|
|
QMessageBox.Save | QMessageBox.Discard |
|
|
QMessageBox.Cancel)
|
|
if ret == QMessageBox.Save:
|
|
return self.saveFile('png')
|
|
elif ret == QMessageBox.Cancel:
|
|
return False
|
|
|
|
return True
|
|
|
|
def saveFile(self, fileFormat):
|
|
initialPath = QDir.currentPath() + '/untitled.' + fileFormat
|
|
|
|
fileName, _ = QFileDialog.getSaveFileName(self, "Save As", initialPath,
|
|
"%s Files (*.%s);;All Files (*)" % (fileFormat.upper(), fileFormat))
|
|
if fileName:
|
|
return self.scribbleArea.saveImage(fileName, fileFormat)
|
|
|
|
return False
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
import sys
|
|
|
|
app = QApplication(sys.argv)
|
|
window = MainWindow()
|
|
window.show()
|
|
sys.exit(app.exec_())
|