1 # -*- coding: ISO-8859-1 -*-
4 # Copyright (C) 2005-2006 Jörg Lehmann <joergl@users.sourceforge.net>
5 # Copyright (C) 2005-2006 André Wobst <wobsta@users.sourceforge.net>
7 # This file is part of PyX (http://pyx.sourceforge.net/).
9 # PyX is free software; you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation; either version 2 of the License, or
12 # (at your option) any later version.
14 # PyX is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 # GNU General Public License for more details.
19 # You should have received a copy of the GNU General Public License
20 # along with PyX; if not, write to the Free Software
21 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
23 import cStringIO
, copy
, time
, math
24 import bbox
, config
, style
, version
, unit
, trafo
30 # in order to keep a consistent order of the registered resources we
31 # not only store them in a hash but also keep an ordered list (up to a
32 # possible merging of resources, in which case the first instance is
34 self
.resourceshash
= {}
35 self
.resourceslist
= []
37 def add(self
, resource
):
38 rkey
= (resource
.type, resource
.id)
39 if self
.resourceshash
.has_key(rkey
):
40 self
.resourceshash
[rkey
].merge(resource
)
42 self
.resourceshash
[rkey
] = resource
43 self
.resourceslist
.append(resource
)
45 def mergeregistry(self
, registry
):
46 for resource
in registry
.resources
:
49 def output(self
, file, writer
):
50 """ write all PostScript code of the prolog resources """
51 for resource
in self
.resourceslist
:
52 resource
.output(file, writer
, self
)
60 """ a PostScript resource """
62 def __init__(self
, type, id):
63 # Every PSresource has to have a type and a unique id.
64 # Resources with the same type and id will be merged
65 # when they are registered in the PSregistry
69 def merge(self
, other
):
70 """ merge self with other, which has to be a resource of the same type and with
74 def output(self
, file, writer
, registry
):
75 raise NotImplementedError("output not implemented for %s" % repr(self
))
77 class PSdefinition(PSresource
):
79 """ PostScript function definition included in the prolog """
81 def __init__(self
, id, body
):
82 self
.type = "definition"
86 def output(self
, file, writer
, registry
):
87 file.write("%%%%BeginRessource: %s\n" % self
.id)
88 file.write("%(body)s /%(id)s exch def\n" % self
.__dict
__)
89 file.write("%%EndRessource\n")
97 def __init__(self
, title
=None, stripfonts
=True, textaspath
=False):
100 self
.stripfonts
= stripfonts
101 self
.textaspath
= textaspath
103 # dictionary mapping font names to dictionaries mapping encoding names to encodings
104 # encodings themselves are mappings from glyphnames to codepoints
107 def writeinfo(self
, file):
108 file.write("%%%%Creator: PyX %s\n" % version
.version
)
109 if self
.title
is not None:
110 file.write("%%%%Title: %s\n" % self
.title
)
111 file.write("%%%%CreationDate: %s\n" %
112 time
.asctime(time
.localtime(time
.time())))
114 def getfontmap(self
):
115 if self
._fontmap
is None:
116 # late import due to cyclic dependency
117 from pyx
.dvi
import mapfile
118 fontmapfiles
= config
.getlist("text", "psfontmaps", ["psfonts.map"])
119 self
._fontmap
= mapfile
.readfontmap(fontmapfiles
)
123 class EPSwriter(_PSwriter
):
125 def __init__(self
, document
, file, **kwargs
):
126 _PSwriter
.__init
__(self
, **kwargs
)
128 if len(document
.pages
) != 1:
129 raise ValueError("EPS file can be constructed out of a single page document only")
130 page
= document
.pages
[0]
133 pagefile
= cStringIO
.StringIO()
134 registry
= PSregistry()
136 pagebbox
= bbox
.empty()
138 page
.processPS(pagefile
, self
, acontext
, registry
, pagebbox
)
140 file.write("%!PS-Adobe-3.0 EPSF-3.0\n")
142 file.write("%%%%BoundingBox: %d %d %d %d\n" % pagebbox
.lowrestuple_pt())
143 file.write("%%%%HiResBoundingBox: %g %g %g %g\n" % pagebbox
.highrestuple_pt())
145 file.write("%%EndComments\n")
147 file.write("%%BeginProlog\n")
148 registry
.output(file, self
)
149 file.write("%%EndProlog\n")
151 file.write(pagefile
.getvalue())
154 file.write("showpage\n")
155 file.write("%%Trailer\n")
156 file.write("%%EOF\n")
159 class PSwriter(_PSwriter
):
161 def __init__(self
, document
, file, writebbox
=0, **kwargs
):
162 _PSwriter
.__init
__(self
, **kwargs
)
164 # We first have to process the content of the pages, writing them into the stream pagesfile
165 # Doing so, we fill the registry and also calculate the page bounding boxes, which are
166 # stored in page._bbox for every page
167 pagesfile
= cStringIO
.StringIO()
168 registry
= PSregistry()
170 # calculated bounding boxes of the whole document
171 documentbbox
= bbox
.empty()
173 for nr
, page
in enumerate(document
.pages
):
174 # process contents of page
175 pagefile
= cStringIO
.StringIO()
177 pagebbox
= bbox
.empty()
178 page
.processPS(pagefile
, self
, acontext
, registry
, pagebbox
)
180 documentbbox
+= pagebbox
182 pagesfile
.write("%%%%Page: %s %d\n" % (page
.pagename
is None and str(nr
+1) or page
.pagename
, nr
+1))
184 pagesfile
.write("%%%%PageMedia: %s\n" % page
.paperformat
.name
)
185 pagesfile
.write("%%%%PageOrientation: %s\n" % (page
.rotated
and "Landscape" or "Portrait"))
186 if pagebbox
and writebbox
:
187 pagesfile
.write("%%%%PageBoundingBox: %d %d %d %d\n" % pagebbox
.lowrestuple_pt())
190 pagesfile
.write("%%BeginPageSetup\n")
191 pagesfile
.write("/pgsave save def\n")
193 pagesfile
.write("%%EndPageSetup\n")
194 pagesfile
.write(pagefile
.getvalue())
196 pagesfile
.write("pgsave restore\n")
197 pagesfile
.write("showpage\n")
198 pagesfile
.write("%%PageTrailer\n")
200 file.write("%!PS-Adobe-3.0\n")
201 if documentbbox
and writebbox
:
202 file.write("%%%%BoundingBox: %d %d %d %d\n" % documentbbox
.lowrestuple_pt())
203 file.write("%%%%HiResBoundingBox: %g %g %g %g\n" % documentbbox
.highrestuple_pt())
206 # required paper formats
208 for page
in document
.pages
:
210 paperformats
[page
.paperformat
] = page
.paperformat
213 for paperformat
in paperformats
.values():
215 file.write("%%DocumentMedia: ")
219 file.write("%s %d %d 75 white ()\n" % (paperformat
.name
,
220 unit
.topt(paperformat
.width
),
221 unit
.topt(paperformat
.height
)))
223 # file.write(%%DocumentNeededResources: ") # register not downloaded fonts here
225 file.write("%%%%Pages: %d\n" % len(document
.pages
))
226 file.write("%%PageOrder: Ascend\n")
227 file.write("%%EndComments\n")
229 # document defaults section
230 #file.write("%%BeginDefaults\n")
231 #file.write("%%EndDefaults\n")
233 # document prolog section
234 file.write("%%BeginProlog\n")
235 registry
.output(file, self
)
236 file.write("%%EndProlog\n")
238 # document setup section
239 #file.write("%%BeginSetup\n")
240 #file.write("%%EndSetup\n")
242 file.write(pagesfile
.getvalue())
245 file.write("%%Trailer\n")
246 file.write("%%EOF\n")
252 self
.linewidth_pt
= None
253 self
.colorspace
= None
254 self
.selectedfont
= None
256 def __call__(self
, **kwargs
):
257 newcontext
= copy
.copy(self
)
258 for key
, value
in kwargs
.items():
259 setattr(newcontext
, key
, value
)