2 # -*- coding: ISO-8859-1 -*-
5 # Copyright (C) 2005-2006 Jörg Lehmann <joergl@users.sourceforge.net>
6 # Copyright (C) 2005-2006 André Wobst <wobsta@users.sourceforge.net>
8 # This file is part of PyX (http://pyx.sourceforge.net/).
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.
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.
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
24 import cStringIO
, copy
, warnings
, time
31 import bbox
, unit
, style
, type1font
, version
36 # fallback implementation for Python 2.2 and below
38 return zip(xrange(len(list)), list)
43 # fallback implementation for Python 2.1
46 for key
, value
in list:
55 # we want to keep the order of the resources
58 self
.procsets
= {"PDF": 1}
61 def add(self
, object):
62 """ register object, merging it with an already registered object of the same type and id """
63 sameobjects
= self
.types
.setdefault(object.type, {})
64 if sameobjects
.has_key(object.id):
65 sameobjects
[object.id].merge(object)
67 self
.objects
.append(object)
68 sameobjects
[object.id] = object
70 def getrefno(self
, object):
72 return self
.merged
.getrefno(object)
74 return self
.types
[object.type][object.id].refno
76 def mergeregistry(self
, registry
):
77 for object in registry
.objects
:
79 registry
.merged
= self
81 def write(self
, file, writer
, catalog
):
82 # first we set all refnos
84 for object in self
.objects
:
88 # second, all objects are written, keeping the positions in the output file
90 for object in self
.objects
:
91 fileposes
.append(file.tell())
92 file.write("%i 0 obj\n" % object.refno
)
93 object.write(file, writer
, self
)
94 file.write("endobj\n")
100 "0000000000 65535 f \n" % refno
)
102 for filepos
in fileposes
:
103 file.write("%010i 00000 n \n" % filepos
)
106 file.write("trailer\n"
108 "/Size %i\n" % refno
)
109 file.write("/Root %i 0 R\n" % self
.getrefno(catalog
))
110 file.write("/Info %i 0 R\n" % self
.getrefno(catalog
.PDFinfo
))
114 file.write("%%EOF\n")
116 def addresource(self
, resourcetype
, resourcename
, object, procset
=None):
117 self
.resources
.setdefault(resourcetype
, {})[resourcename
] = object
119 self
.procsets
[procset
] = 1
121 def writeresources(self
, file):
122 file.write("/Resources <<\n")
123 file.write("/ProcSet [ %s ]\n" % " ".join(["/%s" % p
for p
in self
.procsets
.keys()]))
125 for resourcetype
, resources
in self
.resources
.items():
126 file.write("/%s <<\n%s\n>>\n" % (resourcetype
, "\n".join(["/%s %i 0 R" % (name
, self
.getrefno(object))
127 for name
, object in resources
.items()])))
133 def __init__(self
, type, _id
=None):
134 """create a PDFobject
135 - type has to be a string describing the type of the object
136 - _id is a unique identification used for the object if it is not None.
137 Otherwise id(self) is used
145 def merge(self
, other
):
148 def write(self
, file, writer
, registry
):
149 raise NotImplementedError("write method has to be provided by PDFobject subclass")
152 class PDFcatalog(PDFobject
):
154 def __init__(self
, document
, writer
, registry
):
155 PDFobject
.__init
__(self
, "catalog")
156 self
.PDFpages
= PDFpages(document
, writer
, registry
)
157 registry
.add(self
.PDFpages
)
158 self
.PDFinfo
= PDFinfo()
159 registry
.add(self
.PDFinfo
)
161 def write(self
, file, writer
, registry
):
164 "/Pages %i 0 R\n" % registry
.getrefno(self
.PDFpages
))
165 if writer
.fullscreen
:
166 file.write("/PageMode /FullScreen\n")
170 class PDFinfo(PDFobject
):
173 PDFobject
.__init
__(self
, "info")
175 def write(self
, file, writer
, registry
):
176 if time
.timezone
< 0:
177 # divmod on positive numbers, otherwise the minutes have a different sign from the hours
178 timezone
= "-%02i'%02i'" % divmod(-time
.timezone
/60, 60)
179 elif time
.timezone
> 0:
180 timezone
= "+%02i'%02i'" % divmod(time
.timezone
/60, 60)
187 if 32 <= ord(c
) <= 127 and c
not in "()[]<>\\":
190 r
+= "\\%03o" % ord(c
)
195 file.write("/Title (%s)\n" % pdfstring(writer
.title
))
197 file.write("/Author (%s)\n" % pdfstring(writer
.author
))
199 file.write("/Subject (%s)\n" % pdfstring(writer
.subject
))
201 file.write("/Keywords (%s)\n" % pdfstring(writer
.keywords
))
202 file.write("/Creator (PyX %s)\n" % version
.version
)
203 file.write("/CreationDate (D:%s%s)\n" % (time
.strftime("%Y%m%d%H%M"), timezone
))
207 class PDFpages(PDFobject
):
209 def __init__(self
, document
, writer
, registry
):
210 PDFobject
.__init
__(self
, "pages")
211 self
.PDFpagelist
= []
212 for pageno
, page
in enumerate(document
.pages
):
213 page
= PDFpage(page
, pageno
, self
, writer
, registry
)
215 self
.PDFpagelist
.append(page
)
217 def write(self
, file, writer
, registry
):
222 ">>\n" % (" ".join(["%i 0 R" % registry
.getrefno(page
)
223 for page
in self
.PDFpagelist
]),
224 len(self
.PDFpagelist
)))
227 class PDFpage(PDFobject
):
229 def __init__(self
, page
, pageno
, PDFpages
, writer
, registry
):
230 PDFobject
.__init
__(self
, "page")
231 self
.PDFpages
= PDFpages
234 # every page uses its own registry in order to find out which
235 # resources are used within the page. However, the
236 # pageregistry is also merged in the global registry
237 self
.pageregistry
= PDFregistry()
239 self
.PDFcontent
= PDFcontent(page
, writer
, self
.pageregistry
)
240 self
.pageregistry
.add(self
.PDFcontent
)
241 registry
.mergeregistry(self
.pageregistry
)
243 def write(self
, file, writer
, registry
):
246 "/Parent %i 0 R\n" % registry
.getrefno(self
.PDFpages
))
247 paperformat
= self
.page
.paperformat
249 file.write("/MediaBox [0 0 %f %f]\n" % (unit
.topt(paperformat
.width
), unit
.topt(paperformat
.height
)))
251 file.write("/MediaBox [%f %f %f %f]\n" % self
.PDFcontent
.bbox
.highrestuple_pt())
252 if self
.PDFcontent
.bbox
and writer
.writebbox
:
253 file.write("/CropBox [%f %f %f %f]\n" % self
.PDFcontent
.bbox
.highrestuple_pt())
254 if self
.page
.rotated
:
255 file.write("/Rotate 90\n")
256 file.write("/Contents %i 0 R\n" % registry
.getrefno(self
.PDFcontent
))
257 self
.pageregistry
.writeresources(file)
261 class PDFcontent(PDFobject
):
263 def __init__(self
, page
, writer
, registry
):
264 PDFobject
.__init
__(self
, registry
, "content")
265 contentfile
= cStringIO
.StringIO()
266 self
.bbox
= bbox
.empty()
268 page
.processPDF(contentfile
, writer
, acontext
, registry
, self
.bbox
)
269 self
.content
= contentfile
.getvalue()
272 def write(self
, file, writer
, registry
):
274 content
= zlib
.compress(self
.content
)
276 content
= self
.content
278 "/Length %i\n" % len(content
))
280 file.write("/Filter /FlateDecode\n")
284 file.write("endstream\n")
287 class PDFfont(PDFobject
):
289 def __init__(self
, font
, chars
, writer
, registry
):
290 PDFobject
.__init
__(self
, "font", font
.name
)
291 registry
.addresource("Font", font
.name
, self
, procset
="Text")
293 self
.fontdescriptor
= PDFfontdescriptor(font
, chars
, writer
, registry
)
294 registry
.add(self
.fontdescriptor
)
297 self
.encoding
= PDFencoding(font
.encoding
, writer
, registry
)
298 registry
.add(self
.encoding
)
302 self
.name
= font
.name
303 self
.basefontname
= font
.basefontname
304 self
.metric
= font
.metric
306 def write(self
, file, writer
, registry
):
310 file.write("/Name /%s\n" % self
.name
)
311 file.write("/BaseFont /%s\n" % self
.basefontname
)
312 if self
.fontdescriptor
.fontfile
is not None and self
.fontdescriptor
.fontfile
.usedchars
is not None:
313 usedchars
= self
.fontdescriptor
.fontfile
.usedchars
314 firstchar
= min(usedchars
.keys())
315 lastchar
= max(usedchars
.keys())
316 file.write("/FirstChar %d\n" % firstchar
)
317 file.write("/LastChar %d\n" % lastchar
)
318 file.write("/Widths\n"
320 for i
in range(firstchar
, lastchar
+1):
321 if i
and not (i
% 8):
325 if usedchars
.has_key(i
):
326 file.write("%f" % self
.metric
.getwidth_ds(i
))
331 file.write("/FirstChar 0\n"
336 if i
and not (i
% 8):
341 width
= self
.metric
.getwidth_ds(i
)
342 except (IndexError, AttributeError):
344 file.write("%f" % width
)
346 file.write("/FontDescriptor %d 0 R\n" % registry
.getrefno(self
.fontdescriptor
))
348 file.write("/Encoding %d 0 R\n" % registry
.getrefno(self
.encoding
))
352 class PDFfontdescriptor(PDFobject
):
354 def __init__(self
, font
, chars
, writer
, registry
):
355 PDFobject
.__init
__(self
, "fontdescriptor", font
.basefontname
)
357 if font
.filename
is None:
360 self
.fontfile
= PDFfontfile(font
.basefontname
, font
.filename
, font
.encoding
, chars
, writer
, registry
)
361 registry
.add(self
.fontfile
)
363 self
.name
= font
.basefontname
364 self
.fontinfo
= font
.metric
.fontinfo()
366 def write(self
, file, writer
, registry
):
368 "/Type /FontDescriptor\n"
369 "/FontName /%s\n" % self
.name
)
370 if self
.fontfile
is None:
371 file.write("/Flags 32\n")
373 file.write("/Flags %d\n" % self
.fontfile
.getflags())
374 file.write("/FontBBox [%d %d %d %d]\n" % self
.fontinfo
.fontbbox
)
375 file.write("/ItalicAngle %d\n" % self
.fontinfo
.italicangle
)
376 file.write("/Ascent %d\n" % self
.fontinfo
.ascent
)
377 file.write("/Descent %d\n" % self
.fontinfo
.descent
)
378 file.write("/CapHeight %d\n" % self
.fontinfo
.capheight
)
379 file.write("/StemV %d\n" % self
.fontinfo
.vstem
)
380 if self
.fontfile
is not None:
381 file.write("/FontFile %d 0 R\n" % registry
.getrefno(self
.fontfile
))
385 class PDFfontfile(PDFobject
):
387 def __init__(self
, name
, filename
, encoding
, chars
, writer
, registry
):
388 PDFobject
.__init
__(self
, "fontfile", filename
)
390 self
.filename
= filename
392 self
.encodingfilename
= None
394 self
.encodingfilename
= encoding
.filename
397 self
.usedchars
[char
] = 1
402 def merge(self
, other
):
403 if self
.encodingfilename
== other
.encodingfilename
:
404 self
.usedchars
.update(other
.usedchars
)
406 # TODO: need to resolve the encoding when several encodings are in the play
409 def mkfontfile(self
):
411 self
.font
= font
.t1font
.T1pfbfont(self
.filename
)
414 if self
.font
is None:
416 return self
.font
.getflags()
418 def write(self
, file, writer
, registry
):
419 if self
.font
is None:
422 # XXX: access to the encoding file
423 if self
.encodingfilename
:
424 encodingfile
= type1font
.encodingfile(self
.encodingfilename
, self
.encodingfilename
)
425 usedglyphs
= dict([(encodingfile
.decode(char
)[1:], 1) for char
in self
.usedchars
.keys()])
427 self
.font
._encoding
()
428 usedglyphs
= dict([(self
.font
.encoding
.decode(char
), 1) for char
in self
.usedchars
.keys()])
429 strippedfont
= self
.font
.getstrippedfont(usedglyphs
)
431 strippedfont
= self
.font
432 strippedfont
.outputPDF(file, writer
)
435 class PDFencoding(PDFobject
):
437 def __init__(self
, encoding
, writer
, registry
):
438 PDFobject
.__init
__(self
, "encoding", encoding
.name
)
439 self
.encoding
= encoding
441 def write(self
, file, writer
, registry
):
442 encodingfile
= type1font
.encodingfile(self
.encoding
.name
, self
.encoding
.filename
)
443 encodingfile
.outputPDF(file, writer
)
448 def __init__(self
, document
, filename
,
449 title
=None, author
=None, subject
=None, keywords
=None,
450 fullscreen
=0, writebbox
=0, compress
=1, compresslevel
=6):
454 if not filename
.endswith(".pdf"):
455 filename
= filename
+ ".pdf"
457 file = open(filename
, "wb")
459 raise IOError("cannot open output file")
465 self
.subject
= subject
466 self
.keywords
= keywords
467 self
.fullscreen
= fullscreen
468 self
.writebbox
= writebbox
469 if compress
and not haszlib
:
471 warnings
.warn("compression disabled due to missing zlib module")
472 self
.compress
= compress
473 self
.compresslevel
= compresslevel
475 # the PDFcatalog class automatically builds up the pdfobjects from a document
476 registry
= PDFregistry()
477 catalog
= PDFcatalog(document
, self
, registry
)
478 registry
.add(catalog
)
480 file.write("%%PDF-1.4\n%%%s%s%s%s\n" % (chr(195), chr(182), chr(195), chr(169)))
481 registry
.write(file, self
, catalog
)
488 self
.linewidth_pt
= None
489 # XXX there are both stroke and fill color spaces
490 self
.colorspace
= None
496 def __call__(self
, **kwargs
):
497 newcontext
= copy
.copy(self
)
498 for key
, value
in kwargs
.items():
499 setattr(newcontext
, key
, value
)