# -*- coding: ISO-8859-1 -*-
#
#
# Copyright (C) 2002-2006 Jrg Lehmann <joergl@users.sourceforge.net>
# Copyright (C) 2002-2006 Andr Wobst <wobsta@users.sourceforge.net>
#
# This file is part of PyX (http://pyx.sourceforge.net/).
#
# PyX 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.
#
# PyX 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 PyX; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
"""The canvas module provides a PostScript canvas class and related classes
A canvas holds a collection of all elements and corresponding attributes to be
displayed. """
#
# canvas item
#
from __future__ import nested_scopes
import os
class canvasitem:
"""Base class for everything which can be inserted into a canvas"""
def bbox(self):
"""return bounding box of canvasitem"""
pass
def processPS(self, file, writer, context, registry, bbox):
"""process canvasitem by writing the corresponding PS code to file and
by updating context, registry as well as bbox
- the PS code corresponding to the canvasitem has to be written in the
stream file, which provides a write(string) method
- writer is the PSwriter used for the output
- context is an instance of pswriter.context which is used for keeping
track of the graphics state (current linewidth, colorspace and font))
- registry is used for tracking resources needed by the canvasitem
- bbox has to be updated to include the bounding box of the canvasitem
"""
raise NotImplementedError()
def processPDF(self, file, writer, context, registry, bbox):
"""process canvasitem by writing the corresponding PDF code to file and
by updating context, registry as well as bbox
- the PDF code corresponding to the canvasitem has to be written in the
stream file, which provides a write(string) method
- writer is the PDFwriter used for the output, which contains properties
like whether streamcompression is used
- context is an instance of pdfwriter.context which is used for keeping
track of the graphics state, in particular for the emulation of PS
behaviour regarding fill and stroke styles, for keeping track of the
currently selected font as well as of text regions.
- registry is used for tracking resources needed by the canvasitem
- bbox has to be updated to include the bounding box of the canvasitem
"""
raise NotImplementedError()
import attr, deco, deformer, document, style, trafo, type1font
import bbox as bboxmodule
#
# clipping class
#
class clip(canvasitem):
"""class for use in canvas constructor which clips to a path"""
def __init__(self, path):
"""construct a clip instance for a given path"""
self.path = path
def bbox(self):
# as a canvasitem a clipping path has NO influence on the bbox...
return bboxmodule.empty()
def clipbbox(self):
# ... but for clipping, we nevertheless need the bbox
return self.path.bbox()
def processPS(self, file, writer, context, registry, bbox):
file.write("newpath\n")
self.path.outputPS(file, writer)
file.write("clip\n")
def processPDF(self, file, writer, context, registry, bbox):
self.path.outputPDF(file, writer)
file.write("W n\n")
#
# general canvas class
#
class _canvas(canvasitem):
"""a canvas holds a collection of canvasitems"""
def __init__(self, attrs=[], texrunner=None):
"""construct a canvas
The canvas can be modfied by supplying args, which have
to be instances of one of the following classes:
- trafo.trafo (leading to a global transformation of the canvas)
- canvas.clip (clips the canvas)
- style.strokestyle, style.fillstyle (sets some global attributes of the canvas)
Note that, while the first two properties are fixed for the
whole canvas, the last one can be changed via canvas.set().
The texrunner instance used for the text method can be specified
using the texrunner argument. It defaults to text.defaulttexrunner
"""
self.items = []
self.trafo = trafo.trafo()
self.clipbbox = None
if texrunner is not None:
self.texrunner = texrunner
else:
# prevent cyclic imports
import text
self.texrunner = text.defaulttexrunner
attr.checkattrs(attrs, [trafo.trafo_pt, clip, style.strokestyle, style.fillstyle])
# We have to reverse the trafos such that the PostScript concat operators
# are in the right order. Correspondingly, we below multiply the current self.trafo
# from the right.
# Note that while for the stroke and fill styles the order doesn't matter at all,
# this is not true for the clip operation.
attrs = attrs[:]
attrs.reverse()
for aattr in attrs:
if isinstance(aattr, trafo.trafo_pt):
self.trafo = self.trafo * aattr
elif isinstance(aattr, clip):
if self.clipbbox is None:
self.clipbbox = aattr.clipbbox().transformed(self.trafo)
else:
self.clippbox *= aattr.clipbbox().transformed(self.trafo)
self.items.append(aattr)
def __len__(self):
return len(self.items)
def __getitem__(self, i):
return self.items[i]
def bbox(self):
"""returns bounding box of canvas
Note that this bounding box doesn't take into account the linewidths, so
is less accurate than the one used when writing the output to a file.
"""
obbox = bboxmodule.empty()
for cmd in self.items:
obbox += cmd.bbox()
# transform according to our global transformation and
# intersect with clipping bounding box (which has already been
# transformed in canvas.__init__())
obbox.transform(self.trafo)
if self.clipbbox is not None:
obbox *= self.clipbbox
return obbox
def processPS(self, file, writer, context, registry, bbox):
context = context()
if self.items:
file.write("gsave\n")
nbbox = bboxmodule.empty()
for item in self.items:
item.processPS(file, writer, context, registry, nbbox)
# update bounding bbox
nbbox.transform(self.trafo)
if self.clipbbox is not None:
nbbox *= self.clipbbox
bbox += nbbox
file.write("grestore\n")
def processPDF(self, file, writer, context, registry, bbox):
context = context()
if self.items:
file.write("q\n") # gsave
nbbox = bboxmodule.empty()
for item in self.items:
if isinstance(item, type1font.text_pt):
if not context.textregion:
file.write("BT\n")
context.textregion = 1
else:
if context.textregion:
file.write("ET\n")
context.textregion = 0
context.font = None
item.processPDF(file, writer, context, registry, nbbox)
if context.textregion:
file.write("ET\n")
context.textregion = 0
context.font = None
# update bounding bbox
nbbox.transform(self.trafo)
if self.clipbbox is not None:
nbbox *= self.clipbbox
bbox += nbbox
file.write("Q\n") # grestore
def insert(self, item, attrs=None):
"""insert item in the canvas.
If attrs are passed, a canvas containing the item is inserted applying attrs.
returns the item
"""
if not isinstance(item, canvasitem):
raise RuntimeError("only instances of base.canvasitem can be inserted into a canvas")
if attrs:
sc = _canvas(attrs)
sc.insert(item)
self.items.append(sc)
else:
self.items.append(item)
return item
def draw(self, path, attrs):
"""draw path on canvas using the style given by args
The argument attrs consists of PathStyles, which modify
the appearance of the path, PathDecos, which add some new
visual elements to the path, or trafos, which are applied
before drawing the path.
"""
attrs = attr.mergeattrs(attrs)
attr.checkattrs(attrs, [deco.deco, deformer.deformer, style.fillstyle, style.strokestyle])
for adeformer in attr.getattrs(attrs, [deformer.deformer]):
path = adeformer.deform(path)
styles = attr.getattrs(attrs, [style.fillstyle, style.strokestyle])
dp = deco.decoratedpath(path, styles=styles)
# add path decorations and modify path accordingly
for adeco in attr.getattrs(attrs, [deco.deco]):
adeco.decorate(dp, self.texrunner)
self.insert(dp)
def stroke(self, path, attrs=[]):
"""stroke path on canvas using the style given by args
The argument attrs consists of PathStyles, which modify
the appearance of the path, PathDecos, which add some new
visual elements to the path, or trafos, which are applied
before drawing the path.
"""
self.draw(path, [deco.stroked]+list(attrs))
def fill(self, path, attrs=[]):
"""fill path on canvas using the style given by args
The argument attrs consists of PathStyles, which modify
the appearance of the path, PathDecos, which add some new
visual elements to the path, or trafos, which are applied
before drawing the path.
"""
self.draw(path, [deco.filled]+list(attrs))
def settexrunner(self, texrunner):
"""sets the texrunner to be used to within the text and text_pt methods"""
self.texrunner = texrunner
def text(self, x, y, atext, *args, **kwargs):
"""insert a text into the canvas
inserts a textbox created by self.texrunner.text into the canvas
returns the inserted textbox"""
return self.insert(self.texrunner.text(x, y, atext, *args, **kwargs))
def text_pt(self, x, y, atext, *args):
"""insert a text into the canvas
inserts a textbox created by self.texrunner.text_pt into the canvas
returns the inserted textbox"""
return self.insert(self.texrunner.text_pt(x, y, atext, *args))
#
# user canvas class which adds a few convenience methods for single page output
#
def _wrappedindocument(method):
def wrappedindocument(self, file, *args, **kwargs):
d = document.document([document.page(self, *args, **kwargs)])
self.__name__ = method.__name__
self.__doc__ = method.__doc__
return method(d, file)
return wrappedindocument
class canvas(_canvas):
"""a canvas holds a collection of canvasitems"""
writeEPSfile = _wrappedindocument(document.document.writeEPSfile)
writePSfile = _wrappedindocument(document.document.writePSfile)
writePDFfile = _wrappedindocument(document.document.writePDFfile)
writetofile = _wrappedindocument(document.document.writetofile)
def pipeGS(self, filename="-", device=None, resolution=100,
gscommand="gs", gsoptions="",
textalphabits=4, graphicsalphabits=4,
**kwargs):
if device is None:
if filename.endswith(".png"):
device = "png16m"
if filename.endswith(".jpg"):
device = "jpeg"
gscommand += " -dEPSCrop -dNOPAUSE -dQUIET -dBATCH -r%i -sDEVICE=%s -sOutputFile=%s" % (resolution, device, filename)
if gsoptions:
gscommand += " %s" % gsoptions
if textalphabits is not None:
gscommand += " -dTextAlphaBits=%i" % textalphabits
if graphicsalphabits is not None:
gscommand += " -dGraphicsAlphaBits=%i" % graphicsalphabits
gscommand += " -"
input = os.popen(gscommand, "w")
self.writeEPSfile(input, **kwargs)
|