Source code for qpageview.page

# -*- coding: utf-8 -*-
#
# This file is part of the qpageview package.
#
# Copyright (c) 2016 - 2019 by Wilbert Berendsen
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
# See http://www.gnu.org/licenses/ for more information.

"""
A Page is responsible for drawing a page inside a PageLayout.

"""

import weakref

from PyQt5.QtCore import QBuffer, QPointF, QRect, QRectF, QSizeF, Qt
from PyQt5.QtGui import (
    QColor, QImage, QPageSize, QPainter, QPdfWriter, QPixmap, QTransform)
from PyQt5.QtSvg import QSvgGenerator

from . import util
from .constants import Rotate_0


# a cache to store "owned" copies
_copycache = weakref.WeakKeyDictionary()


[docs]class AbstractPage(util.Rectangular): """A Page is a rectangle that is positioned in a PageLayout. A Page represents one page, added to a PageLayout that is displayed in a View. Although there is no mechanism to enforce it, a Page is normally only used in one PageLayout at a time. A Page has instance attributes: * that normally do not change during its lifetime: `pageWidth` the original width (by default in points, `dpi` is 72.0 `pageHeight` the original height but can be changed at class level) * that can be modified by the user (having defaults at the class level): `scaleX` the scale in X-direction of the original page (1.0) `scaleY` the scale in Y-direction of the original page (1.0) `rotation` the rotation (Rotate_0) `z` the z-index (0) (only relevant when pages overlap) `paperColor` the paper color (None). If None, the renderer's paperColor is used. * and that are set by the layout when computing the size and positioning the pages: `x` the position x-coordinate `y` the position y-coordinate `width` the width in pixels `height` the height in pixels `computedRotation` the rotation in which finally to render The class variable `dpi` is 72.0 by default but can be set to a different value depending on the page type. E.g. for Svg pages 90 or 96 makes sense. """ renderer = None dpi = 72.0 pageWidth = 595.28 # default to A4 pageHeight = 841.89 z = 0 rotation = Rotate_0 computedRotation = Rotate_0 scaleX = 1.0 scaleY = 1.0 paperColor = None
[docs] @classmethod def load(cls, filename, renderer=None): """Implement this to yield one or more pages by reading the file. The renderer may be None, and not all page types use a renderer. The filename may be a string or a QByteArray object containing the data. """ pass
[docs] @classmethod def loadFiles(cls, filenames, renderer=None): """Load multiple files, yielding Page instances of this type.""" for f in filenames: for page in cls.load(f, renderer): yield page
[docs] def copy(self, owner=None, matrix=None): """Return a copy of the page with the same instance attributes. If owner is specified, the copy is weakly cached for that owner and returned next time. All instance attribute will be updated each time. If matrix is specified, it should be a QTransform, and it will be used to map the geometry of the original to the (cached) copy before it is returned. """ cls = type(self) if owner: try: page = _copycache[owner][self] except KeyError: page = cls.__new__(cls) _copycache.setdefault(owner, weakref.WeakKeyDictionary())[self] = page else: page.__dict__.clear() else: page = cls.__new__(cls) page.__dict__.update(self.__dict__) if matrix: page.setGeometry(matrix.mapRect(self.geometry())) return page
[docs] def setPageSize(self, sizef): """Set our natural page size (QSizeF). Normally this is done in the constructor, based on the page we need to render. By default the page size is assumed to be in points, 1/72 of an inch. You can set the `dpi` class variable to use a different unit. """ self.pageWidth = sizef.width() self.pageHeight = sizef.height()
[docs] def pageSize(self): """Return our natural page size (QSizeF). By default the page size is assumed to be in points, 1/72 of an inch. You can set the `dpi` class variable to use a different unit. """ return QSizeF(self.pageWidth, self.pageHeight)
[docs] def pageRect(self): """Return QRectF(0, 0, pageWidth, pageHeight).""" return QRectF(0, 0, self.pageWidth, self.pageHeight)
[docs] def transform(self, width=None, height=None): """Return a QTransform, converting an original area to page coordinates. The `width` and `height` refer to the original (unrotated) width and height of the page's contents, and default to pageWidth and pageHeight. """ if width is None: width = self.pageWidth if height is None: height = self.pageHeight t = QTransform() t.scale(self.width, self.height) t.translate(.5, .5) t.rotate(self.computedRotation * 90) t.translate(-.5, -.5) t.scale(1 / width, 1 / height) return t
[docs] def defaultSize(self): """Return the pageSize() scaled and rotated (if needed). Based on scaleX, scaleY, and computedRotation attributes. """ s = QSizeF(self.pageWidth * self.scaleX, self.pageHeight * self.scaleY) if self.computedRotation & 1: s.transpose() return s
[docs] def updateSize(self, dpiX, dpiY, zoomFactor): """Set the width and height attributes of the page. This size is computed based on the page's natural size, dpi, scale and computedRotation attribute; and the supplied dpiX, dpiY, and zoomFactor. """ s = self.defaultSize() # now handle dpi, scale and zoom self.width = round(s.width() * dpiX / self.dpi * zoomFactor) self.height = round(s.height() * dpiY / self.dpi * zoomFactor)
[docs] def zoomForWidth(self, width, rotation, dpiX): """Return the zoom we need to display ourselves at the given width.""" width = max(width, 1) if (self.rotation + rotation) & 1: w = self.pageHeight / self.scaleY else: w = self.pageWidth / self.scaleX return width * self.dpi / dpiX / w
[docs] def zoomForHeight(self, height, rotation, dpiY): """Return the zoom we need to display ourselves at the given height.""" height = max(height, 1) if (self.rotation + rotation) & 1: h = self.pageWidth / self.scaleX else: h = self.pageHeight / self.scaleY return height * self.dpi / dpiY / h
[docs] def paint(self, painter, rect, callback=None): """Implement this to paint our Page. The View calls this method in the paint event. If you can't paint quickly, just return and schedule an image to be rendered in the background. If a callback is specified, it is called when the image is ready with the page as argument. """ pass
[docs] def print(self, painter, rect=None, paperColor=None): """Implement this to paint a page for printing. The difference with paint() and image() is that the rect (QRectF) supplied to print() is not in the Page coordinates, but in the original pageSize() and unrotated. The painter has been prepared for scale and rotation. If rect is None, the full pageRect() is used. """ pass
[docs] def output(self, device, rect=None, paperColor=None): """Paint specified rectangle (or the whole page) to the paint device. The page is rotated and scaled, and the resolution of the paint device is used in case pixelbased images need to be generated. But where possible, vector painting is used. This method uses :meth:`print` to do the actual painting to the paint device. If paperColor is not given, no background is printed normally. """ if rect is None: rect = self.pageRect() painter = QPainter(device) painter.scale(device.logicalDpiX() / self.dpi, device.logicalDpiY() / self.dpi) util.rotate(painter, self.computedRotation, rect.width(), rect.height()) painter.scale(self.scaleX, self.scaleY) self.print(painter, rect, paperColor) return painter.end()
[docs] def image(self, rect=None, dpiX=None, dpiY=None, paperColor=None): """Implement this to return a QImage of the specified rectangle. The rectangle is relative to our top-left position. dpiX defaults to our default dpi and dpiY defaults to dpiX. """ pass
[docs] def pdf(self, filename, rect=None, resolution=72.0, paperColor=None): """Create a PDF file for the selected rect or the whole page. The filename may be a string or a QIODevice object. The rectangle is relative to our top-left position. Normally vector graphics are rendered, but in cases where that is not possible, the resolution will be used to determine the DPI for the generated rendering. """ # map to the original page source = self.pageRect() if rect is None else self.mapFromPage().rect(rect) # scale to target size w = source.width() * self.scaleX h = source.height() * self.scaleY if self.computedRotation & 1: w, h = h, w targetSize = QSizeF(w, h) pdf = QPdfWriter(filename) pdf.setCreator("qpageview") pdf.setResolution(int(resolution)) layout = pdf.pageLayout() layout.setMode(layout.FullPageMode) layout.setPageSize(QPageSize(targetSize * 72.0 / self.dpi, QPageSize.Point)) pdf.setPageLayout(layout) return self.output(pdf, source, paperColor)
[docs] def eps(self, filename, rect=None, resolution=72.0, paperColor=None): """Create a EPS (Encapsulated Postscript) file for the selected rect or the whole page. This needs the popplerqt5 module. The filename may be a string or a QIODevice object. The rectangle is relative to our top-left position. Normally vector graphics are rendered, but in cases where that is not possible, the resolution will be used to determine the DPI for the generated rendering. """ buf = QBuffer() buf.open(QBuffer.WriteOnly) success = self.pdf(buf, rect, resolution, paperColor) buf.close() if success: from . import poppler for pdf in poppler.PopplerPage.load(buf.data()): ps = pdf.document.psConverter() ps.setPageList([pdf.pageNumber+1]) if isinstance(filename, str): ps.setOutputFileName(filename) else: ps.setOutputDevice(filename) try: ps.setPSOptions(ps.PSOption(ps.Printing | ps.StrictMargins)) ps.setPSOptions(ps.PSOption(ps.Printing | ps.StrictMargins | ps.PrintToEPS)) except AttributeError: pass ps.setVDPI(resolution) ps.setHDPI(resolution) return ps.convert() return False
[docs] def svg(self, filename, rect=None, resolution=72.0, paperColor=None): """Create a SVG file for the selected rect or the whole page. The filename may be a string or a QIODevice object. The rectangle is relative to our top-left position. Normally vector graphics are rendered, but in cases where that is not possible, the resolution will be used to determine the DPI for the generated rendering. """ # map to the original page source = self.pageRect() if rect is None else self.mapFromPage().rect(rect) # scale to target size w = source.width() * self.scaleX h = source.height() * self.scaleY if self.computedRotation & 1: w, h = h, w targetSize = QSizeF(w, h) * resolution / self.dpi svg = QSvgGenerator() if isinstance(filename, str): svg.setFileName(filename) else: svg.setOutputDevice(filename) svg.setResolution(int(resolution)) svg.setSize(targetSize.toSize()) svg.setViewBox(QRectF(0, 0, targetSize.width(), targetSize.height())) return self.output(svg, source, paperColor)
[docs] def pixmap(self, rect=None, size=100, paperColor=None): """Return a QPixmap, scaled so that width or height doesn't exceed size. Uses the :meth:`image` method to get the image, and converts that to a QPixmap. """ s = self.defaultSize() w, h = s.width(), s.height() if rect is not None: w *= rect.width() / self.width h *= rect.height() / self.height l = max(w, h) dpi = size / l * self.dpi return QPixmap.fromImage(self.image(rect, dpi, dpi, paperColor))
[docs] def mutex(self): """Return an object that should be locked when rendering the page. Page are guaranteed not to be rendered at the same time when they return the same mutex object. By default, None is returned. """
[docs] def group(self): """Return the group the page belongs to. This could be some document structure, so that different Page objects could refer to the same graphical contents, preventing double caching. This object is used together with the value returned by ident() as a key to cache the page. The idea is that the contents of the page are uniquely identified by the objects returned by group() and ident(). This way, when the same document is opened in multiple page instances, only one copy resides in the (global) cache. By default, the page object itself is returned. """ return self
[docs] def ident(self): """Return a value that identifies the page within the group returned by group(). By default, None is returned. """ return None
[docs] def mapToPage(self, width=None, height=None): """Return a MapToPage object, that can map original to Page coordinates. The `width` and `height` refer to the original (unrotated) width and height of the page's contents, and default to pageWidth and pageHeight. """ return util.MapToPage(self.transform(width, height))
[docs] def mapFromPage(self, width=None, height=None): """Return a MapFromPage object, that can map Page to original coordinates. The `width` and `height` refer to the original (unrotated) width and height of the page's contents, and default to pageWidth and pageHeight. """ return util.MapFromPage(self.transform(width, height).inverted()[0])
[docs] def text(self, rect): """Implement this method to get the text at the specified rectangle. The rectangle should be in page coordinates. The default implementation simply returns an empty string. """ return ""
[docs] def linksAt(self, point): """Return a list of zero or more links touched by QPoint point. The point is in page coordinates. The list is sorted with the smallest rectangle first. """ # Link objects have their area ranging # in width and height from 0.0 to 1.0 ... pos = self.mapFromPage(1, 1).point(point) links = self.links() return sorted(links.at(pos.x(), pos.y()), key=links.width)
[docs] def linksIn(self, rect): """Return an unordered set of links enclosed in rectangle. The rectangle is in page coordinates. """ return self.links().inside(*self.mapFromPage(1, 1).rect(rect).getCoords())
[docs] def linkRect(self, link): """Return a QRect encompassing the linkArea of a link in coordinates of our page.""" return self.mapToPage(1, 1).rect(link.rect())
[docs]class AbstractRenderedPage(AbstractPage): """A Page that has a renderer that performs caching and painting. The renderer lives in the renderer attribute. """ def __init__(self, renderer=None): if renderer is not None: self.renderer = renderer
[docs] def paint(self, painter, rect, callback=None): """Reimplement this to paint our Page. The View calls this method in the paint event. If you can't paint quickly, just return and schedule an image to be rendered in the background. If a callback is specified, it is called when the image is ready with the page as argument. By default, this method calls the renderer's :meth:`~.render.AbstractRenderer.paint` method. """ if rect: self.renderer.paint(self, painter, rect, callback)
[docs] def print(self, painter, rect=None, paperColor=None): """Paint a page for printing. The difference with :meth:`paint` and :meth:`image` is that the rect (QRectF) supplied to print() is not in the Page coordinates, but in the original pageSize() and unrotated. The painter has been prepared for scale and rotation. If rect is None, the full pageRect() is used. This method calls the renderer's draw() method. """ if rect is None: rect = self.pageRect() else: rect = rect & self.pageRect() from . import render k = render.Key(self.group(), self.ident(), 0, self.pageWidth, self.pageHeight) t = render.Tile(*rect.normalized().getRect()) self.renderer.draw(self, painter, k, t, paperColor)
[docs] def image(self, rect=None, dpiX=None, dpiY=None, paperColor=None): """Returns a QImage of the specified rectangle. The rectangle is relative to our top-left position. dpiX defaults to our default dpi and dpiY defaults to dpiX. This implementation calls the renderer to generate the image. The image is not cached. """ if rect is None: rect = self.rect() if dpiX is None: dpiX = self.dpi if dpiY is None: dpiY = dpiX return self.renderer.image(self, rect, dpiX, dpiY, paperColor)
[docs]class BlankPage(AbstractPage): """A blank page."""
[docs] def paint(self, painter, rect, callback=None): """Paint blank page in the View.""" painter.fillRect(rect, self.paperColor or Qt.white)
[docs] def print(self, painter, rect=None, paperColor=None): """Paint blank page for printing.""" if rect is None: rect = self.pageRect() else: rect = rect & self.pageRect() painter.fillRect(rect, paperColor or Qt.white)
[docs] def image(self, rect=None, dpiX=None, dpiY=None, paperColor=None): """Return a blank image.""" if rect is None: rect = self.rect() if dpiX is None: dpiX = self.dpi if dpiY is None: dpiY = dpiX s = self.defaultSize() width = s.width() * dpiX / self.dpi height = s.height() * dpiY / self.dpi image = QImage(width, height, QImage.Format_ARGB32_Premultiplied) image.fill(paperColor or Qt.white) return image
[docs]class ImagePrintPageMixin: """A Page mixin that implements print() using the image() method. This can be used e.g. for compositing pages, which does not work well when painting to a PDF, a printer or a SVG generator. """
[docs] def print(self, painter, rect=None, paperColor=None): """Print using the image() method.""" if rect is None: rect = self.pageRect() else: rect = rect & self.pageRect() # Find the rectangle on the Page in page coordinates target = self.mapToPage().rect(rect) # Make an image exactly in the printer's resolution m = painter.transform() r = m.mapRect(rect) # see where the rect ends up w, h = r.width(), r.height() if m.m11() == 0: w, h = h, w # swap if rotation & 1 :-) # now we know the scale from our dpi to the paintdevice's logicalDpi! hscale = w / rect.width() vscale = h / rect.height() dpiX = self.dpi * hscale dpiY = self.dpi * vscale image = self.image(target, dpiX, dpiY, paperColor) painter.translate(-rect.topLeft()) painter.drawImage(rect, image)