pyx/pswriter.py

   1 #!/usr/bin/env python
   2 # -*- coding: ISO-8859-1 -*-
   3 #
   4 #
   5 # Copyright (C) 2005 Jörg Lehmann <joergl@users.sourceforge.net>
   6 # Copyright (C) 2005 André Wobst <wobsta@users.sourceforge.net>
   7 #
   8 # This file is part of PyX (http://pyx.sourceforge.net/).
   9 #
  10 # PyX is free software; you can redistribute it and/or modify
  11 # it under the terms of the GNU General Public License as published by
  12 # the Free Software Foundation; either version 2 of the License, or
  13 # (at your option) any later version.
  14 #
  15 # PyX is distributed in the hope that it will be useful,
  16 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 # GNU General Public License for more details.
  19 #
  20 # You should have received a copy of the GNU General Public License
  21 # along with PyX; if not, write to the Free Software
  22 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  23
  24 import copy, time, math
  25 import style, version, type1font, unit
  26
  27
  28 class PSregistry:
  29
  30     def __init__(self):
  31         # in order to keep a consistent order of the registered resources we
  32         # not only store them in a hash but also keep an ordered list (up to a
  33         # possible merging of resources, in which case the first instance is
  34         # kept)
  35         self.resourceshash = {}
  36         self.resourceslist = []
  37
  38     def add(self, resource):
  39         rkey = (resource.type, resource.id)
  40         if self.resourceshash.has_key(rkey):
  41            self.resourceshash[rkey].merge(resource)
  42         else:
  43            self.resourceshash[rkey] = resource
  44            self.resourceslist.append(resource)
  45
  46     def outputPS(self, file, writer):
  47         """ write all PostScript code of the prolog resources """
  48         for resource in self.resourceslist:
  49             resource.outputPS(file, writer, self)
  50
  51 #
  52 # Abstract base class
  53 #
  54
  55 class PSresource:
  56
  57     """ a PostScript resource """
  58
  59     def __init__(self, type, id):
  60         # Every PSresource has to have a type and a unique id.
  61         # Resources with the same type and id will be merged
  62         # when they are registered in the PSregistry
  63         self.type = type
  64         self.id = id
  65
  66     def merge(self, other):
  67         """ merge self with other, which has to be a resource of the same type and with
  68         the same id"""
  69         pass
  70
  71     def outputPS(self, file, writer, registry):
  72         raise NotImplementedError("outputPS not implemented for %s" % repr(self))
  73
  74 #
  75 # Different variants of prolog items
  76 #
  77
  78 class PSdefinition(PSresource):
  79
  80     """ PostScript function definition included in the prolog """
  81
  82     def __init__(self, id, body):
  83         self.type = "definition"
  84         self.id = id
  85         self.body = body
  86
  87     def outputPS(self, file, writer, registry):
  88         file.write("%%%%BeginRessource: %s\n" % self.id)
  89         file.write("%(body)s /%(id)s exch def\n" % self.__dict__)
  90         file.write("%%EndRessource\n")
  91
  92
  93 class PSfont:
  94
  95     def __init__(self, font, chars, registry):
  96         if font.filename:
  97             registry.add(PSfontfile(font.basefontname,
  98                                     font.filename,
  99                                     font.encoding,
 100                                     chars))
 101         if font.encoding:
 102             registry.add(_ReEncodeFont)
 103             registry.add(PSfontencoding(font.encoding))
 104             registry.add(PSfontreencoding(font.name,
 105                                           font.basefontname,
 106                                           font.encoding.name))
 107
 108
 109 class PSfontfile(PSresource):
 110
 111     """ PostScript font definition included in the prolog """
 112
 113     def __init__(self, name, filename, encoding, chars):
 114         """ include type 1 font defined by the following parameters
 115
 116         - name:        name of the PostScript font
 117         - filename:    name (without path) of file containing the font definition
 118         - encfilename: name (without path) of file containing used encoding of font
 119                        or None (if no encoding file used)
 120         - chars:       character list to fill usedchars
 121
 122         """
 123
 124         # Note that here we only need the encoding for selecting the used glyphs!
 125
 126         self.type = "fontfile"
 127         self.id = self.name = name
 128         self.filename = filename
 129         if encoding is None:
 130             self.encodingfilename = None
 131         else:
 132             self.encodingfilename = encoding.filename
 133         self.usedchars = {}
 134         for char in chars:
 135             self.usedchars[char] = 1
 136
 137     def merge(self, other):
 138         if self.encodingfilename == other.encodingfilename:
 139             self.usedchars.update(other.usedchars)
 140         else:
 141             self.usedchars = None # stripping of font not possible
 142
 143     def outputPS(self, file, writer, registry):
 144         fontfile = type1font.fontfile(self.name, self.filename, self.usedchars, self.encodingfilename)
 145         fontfile.outputPS(file, writer, registry)
 146
 147
 148 class PSfontencoding(PSresource):
 149
 150     """ PostScript font encoding vector included in the prolog """
 151
 152     def __init__(self, encoding):
 153         """ include font encoding vector specified by encoding """
 154
 155         self.type = "fontencoding"
 156         self.id = encoding.name
 157         self.encoding = encoding
 158
 159     def outputPS(self, file, writer, registry):
 160         encodingfile = type1font.encodingfile(self.encoding.name, self.encoding.filename)
 161         encodingfile.outputPS(file, writer, registry)
 162
 163
 164 class PSfontreencoding(PSresource):
 165
 166     """ PostScript font re-encoding directive included in the prolog """
 167
 168     def __init__(self, fontname, basefontname, encodingname):
 169         """ include font re-encoding directive specified by
 170
 171         - fontname:     PostScript FontName of the new reencoded font
 172         - basefontname: PostScript FontName of the original font
 173         - encname:      name of the encoding
 174         - font:         a reference to the font instance (temporarily added for pdf support)
 175
 176         Before being able to reencode a font, you have to include the
 177         encoding via a fontencoding prolog item with name=encname
 178
 179         """
 180
 181         self.type = "fontreencoding"
 182         self.id = self.fontname = fontname
 183         self.basefontname = basefontname
 184         self.encodingname = encodingname
 185
 186     def outputPS(self, file, writer, registry):
 187         file.write("%%%%BeginProcSet: %s\n" % self.fontname)
 188         file.write("/%s /%s %s ReEncodeFont\n" % (self.basefontname, self.fontname, self.encodingname))
 189         file.write("%%EndProcSet\n")
 190
 191
 192 _ReEncodeFont = PSdefinition("ReEncodeFont", """{
 193   5 dict
 194   begin
 195     /newencoding exch def
 196     /newfontname exch def
 197     /basefontname exch def
 198     /basefontdict basefontname findfont def
 199     /newfontdict basefontdict maxlength dict def
 200     basefontdict {
 201       exch dup dup /FID ne exch /Encoding ne and
 202       { exch newfontdict 3 1 roll put }
 203       { pop pop }
 204       ifelse
 205     } forall
 206     newfontdict /FontName newfontname put
 207     newfontdict /Encoding newencoding put
 208     newfontname newfontdict definefont pop
 209   end
 210 }""")
 211
 212
 213 class epswriter:
 214
 215     def __init__(self, document, filename):
 216         if len(document.pages) != 1:
 217             raise ValueError("EPS file can be construced out of a single page document only")
 218         page = document.pages[0]
 219         canvas = page.canvas
 220
 221         if filename[-4:] != ".eps":
 222             filename = filename + ".eps"
 223         try:
 224             file = open(filename, "w")
 225         except IOError:
 226             raise IOError("cannot open output file")
 227
 228         bbox = canvas.bbox()
 229         bbox.enlarge(page.bboxenlarge)
 230         pagetrafo = page.pagetrafo(bbox)
 231
 232         # if a page transformation is necessary, we have to adjust the bounding box
 233         # accordingly
 234         if pagetrafo is not None:
 235             bbox.transform(pagetrafo)
 236
 237         file.write("%!PS-Adobe-3.0 EPSF-3.0\n")
 238         bbox.outputPS(file, self)
 239         file.write("%%%%Creator: PyX %s\n" % version.version)
 240         file.write("%%%%Title: %s\n" % filename)
 241         file.write("%%%%CreationDate: %s\n" %
 242                    time.asctime(time.localtime(time.time())))
 243         file.write("%%EndComments\n")
 244
 245         file.write("%%BeginProlog\n")
 246         registry = PSregistry()
 247         canvas.registerPS(registry)
 248         registry.outputPS(file, self)
 249         file.write("%%EndProlog\n")
 250
 251         acontext = context()
 252         # apply a possible page transformation
 253         if pagetrafo is not None:
 254             pagetrafo.outputPS(file, self, acontext)
 255
 256         style.linewidth.normal.outputPS(file, self, acontext)
 257
 258         # here comes the canvas content
 259         canvas.outputPS(file, self, acontext)
 260
 261         file.write("showpage\n")
 262         file.write("%%Trailer\n")
 263         file.write("%%EOF\n")
 264
 265
 266 class pswriter:
 267
 268     def __init__(self, document, filename, insertpagebbox=0):
 269         if filename[-4:] != ".ps":
 270             filename = filename + ".ps"
 271         try:
 272             file = open(filename, "w")
 273         except IOError:
 274             raise IOError("cannot open output file")
 275
 276         # calculated bounding boxes of separate pages and the bounding box of the whole document
 277         documentbbox = None
 278         for page in document.pages:
 279             canvas = page.canvas
 280             page.bbox = canvas.bbox()
 281             page.bbox.enlarge(page.bboxenlarge)
 282             page._pagetrafo = page.pagetrafo(page.bbox)
 283             # if a page transformation is necessary, we have to adjust the bounding box
 284             # accordingly
 285             if page._pagetrafo is not None and page.bbox is not None:
 286                 page.bbox.transform(page._pagetrafo)
 287             if documentbbox is None:
 288                 documentbbox = page.bbox.enlarged(0)
 289             elif page.bbox is not None:
 290                 documentbbox += page.bbox
 291
 292         file.write("%!PS-Adobe-3.0\n")
 293         documentbbox.outputPS(file, self)
 294         file.write("%%%%Creator: PyX %s\n" % version.version)
 295         file.write("%%%%Title: %s\n" % filename)
 296         file.write("%%%%CreationDate: %s\n" %
 297                    time.asctime(time.localtime(time.time())))
 298
 299         # required paper formats
 300         paperformats = {}
 301         for page in document.pages:
 302             paperformats[page.paperformat] = page.paperformat
 303
 304         first = 1
 305         for paperformat in paperformats.values():
 306             if first:
 307                 file.write("%%DocumentMedia: ")
 308                 first = 0
 309             else:
 310                 file.write("%%+ ")
 311             file.write("%s %d %d 75 white ()\n" % (paperformat.name,
 312                                                    unit.topt(paperformat.width),
 313                                                    unit.topt(paperformat.height)))
 314
 315         # file.write(%%DocumentNeededResources: ") # register not downloaded fonts here
 316
 317         file.write("%%%%Pages: %d\n" % len(document.pages))
 318         file.write("%%PageOrder: Ascend\n")
 319         file.write("%%EndComments\n")
 320
 321         # document defaults section
 322         #file.write("%%BeginDefaults\n")
 323         #file.write("%%EndDefaults\n")
 324
 325         # document prolog section
 326         file.write("%%BeginProlog\n")
 327         registry = PSregistry()
 328         for page in document.pages:
 329             page.canvas.registerPS(registry)
 330         registry.outputPS(file, self)
 331         file.write("%%EndProlog\n")
 332
 333         # document setup section
 334         #file.write("%%BeginSetup\n")
 335         #file.write("%%EndSetup\n")
 336
 337         # pages section
 338         for nr, page in enumerate(document.pages):
 339             file.write("%%%%Page: %s %d\n" % (page.pagename is None and str(nr+1) or page.pagename, nr+1))
 340             file.write("%%%%PageMedia: %s\n" % page.paperformat.name)
 341             file.write("%%%%PageOrientation: %s\n" % (page.rotated and "Landscape" or "Portrait"))
 342             if insertpagebbox:
 343                 file.write("%%%%PageBoundingBox: %d %d %d %d\n" % (math.floor(page.bbox.llx_pt), math.floor(page.bbox.lly_pt),
 344                                                                    math.ceil(page.bbox.urx_pt), math.ceil(page.bbox.ury_pt)))
 345
 346             # page setup section
 347             file.write("%%BeginPageSetup\n")
 348             file.write("/pgsave save def\n")
 349
 350             acontext = context()
 351             # apply a possible page transformation
 352             if page._pagetrafo is not None:
 353                 page._pagetrafo.outputPS(file, self, acontext)
 354
 355             style.linewidth.normal.outputPS(file, self, acontext)
 356             file.write("%%EndPageSetup\n")
 357
 358             # here comes the actual content
 359             page.canvas.outputPS(file, self, acontext)
 360             file.write("pgsave restore\n")
 361             file.write("showpage\n")
 362             file.write("%%PageTrailer\n")
 363
 364         file.write("%%Trailer\n")
 365         file.write("%%EOF\n")
 366
 367 class context:
 368
 369     def __init__(self):
 370         self.linewidth_pt = None
 371         self.colorspace = None
 372         self.font = None
 373
 374     def __call__(self, **kwargs):
 375         newcontext = copy.copy(self)
 376         for key, value in kwargs.items():
 377             setattr(newcontext, key, value)
 378         return newcontext