pyx/font/font.py

   1 # -*- encoding: utf-8 -*-
   2 #
   3 #
   4 # Copyright (C) 2005-2011 Jörg Lehmann <joergl@users.sourceforge.net>
   5 # Copyright (C) 2006-2011 Michael Schindler <m-schindler@users.sourceforge.net>
   6 # Copyright (C) 2005-2011 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
  23
  24 import logging
  25 from pyx import bbox, baseclasses, deco, path, pswriter, pdfwriter, trafo, unit
  26 from . import t1file, afmfile
  27
  28 logger = logging.getLogger("pyx")
  29
  30 ##############################################################################
  31 # PS resources
  32 ##############################################################################
  33
  34 class PST1file(pswriter.PSresource):
  35
  36     """ PostScript font definition included in the prolog """
  37
  38     def __init__(self, t1file, glyphnames, charcodes):
  39         """ include type 1 font t1file stripped to the given glyphnames"""
  40         self.type = "t1file"
  41         self.t1file = t1file
  42         self.id = t1file.name
  43         self.glyphnames = set(glyphnames)
  44         self.charcodes = set(charcodes)
  45
  46     def merge(self, other):
  47         self.glyphnames.update(other.glyphnames)
  48         self.charcodes.update(other.charcodes)
  49
  50     def output(self, file, writer, registry):
  51         file.write("%%%%BeginFont: %s\n" % self.t1file.name)
  52         if writer.strip_fonts:
  53             if self.glyphnames:
  54                 file.write("%%Included glyphs: %s\n" % " ".join(self.glyphnames))
  55             if self.charcodes:
  56                 file.write("%%Included charcodes: %s\n" % " ".join([str(charcode) for charcode in self.charcodes]))
  57             self.t1file.getstrippedfont(self.glyphnames, self.charcodes).outputPS(file, writer)
  58         else:
  59             self.t1file.outputPS(file, writer)
  60         file.write("\n%%EndFont\n")
  61
  62
  63 _ReEncodeFont = pswriter.PSdefinition("ReEncodeFont", b"""{
  64   5 dict
  65   begin
  66     /newencoding exch def
  67     /newfontname exch def
  68     /basefontname exch def
  69     /basefontdict basefontname findfont def
  70     /newfontdict basefontdict maxlength dict def
  71     basefontdict {
  72       exch dup dup /FID ne exch /Encoding ne and
  73       { exch newfontdict 3 1 roll put }
  74       { pop pop }
  75       ifelse
  76     } forall
  77     newfontdict /FontName newfontname put
  78     newfontdict /Encoding newencoding put
  79     newfontname newfontdict definefont pop
  80   end
  81 }""")
  82
  83
  84 class PSreencodefont(pswriter.PSresource):
  85
  86     """ reencoded PostScript font"""
  87
  88     def __init__(self, basefontname, newfontname, encoding):
  89         """ reencode the font """
  90
  91         self.type = "reencodefont"
  92         self.basefontname = basefontname
  93         self.id = self.newfontname = newfontname
  94         self.encoding = encoding
  95
  96     def output(self, file, writer, registry):
  97         file.write("%%%%BeginResource: %s\n" % self.newfontname)
  98         file.write("/%s /%s\n[" % (self.basefontname, self.newfontname))
  99         vector = [None] * len(self.encoding)
 100         for glyphname, charcode in list(self.encoding.items()):
 101             vector[charcode] = glyphname
 102         for i, glyphname in enumerate(vector):
 103             if i:
 104                 if not (i % 8):
 105                     file.write("\n")
 106                 else:
 107                     file.write(" ")
 108             file.write("/%s" % glyphname)
 109         file.write("]\n")
 110         file.write("ReEncodeFont\n")
 111         file.write("%%EndResource\n")
 112
 113
 114 _ChangeFontMatrix = pswriter.PSdefinition("ChangeFontMatrix", b"""{
 115   5 dict
 116   begin
 117     /newfontmatrix exch def
 118     /newfontname exch def
 119     /basefontname exch def
 120     /basefontdict basefontname findfont def
 121     /newfontdict basefontdict maxlength dict def
 122     basefontdict {
 123       exch dup dup /FID ne exch /FontMatrix ne and
 124       { exch newfontdict 3 1 roll put }
 125       { pop pop }
 126       ifelse
 127     } forall
 128     newfontdict /FontName newfontname put
 129     newfontdict /FontMatrix newfontmatrix readonly put
 130     newfontname newfontdict definefont pop
 131   end
 132 }""")
 133
 134
 135 class PSchangefontmatrix(pswriter.PSresource):
 136
 137     """ change font matrix of a PostScript font"""
 138
 139     def __init__(self, basefontname, newfontname, newfontmatrix):
 140         """ change the font matrix """
 141
 142         self.type = "changefontmatrix"
 143         self.basefontname = basefontname
 144         self.id = self.newfontname = newfontname
 145         self.newfontmatrix = newfontmatrix
 146
 147     def output(self, file, writer, registry):
 148         file.write("%%%%BeginResource: %s\n" % self.newfontname)
 149         file.write("/%s /%s\n" % (self.basefontname, self.newfontname))
 150         file.write(str(self.newfontmatrix))
 151         file.write("\nChangeFontMatrix\n")
 152         file.write("%%EndResource\n")
 153
 154
 155 ##############################################################################
 156 # PDF resources
 157 ##############################################################################
 158
 159 class PDFfont(pdfwriter.PDFobject):
 160
 161     def __init__(self, fontname, basefontname, charcodes, fontdescriptor, encoding, metric):
 162         pdfwriter.PDFobject.__init__(self, "font", fontname)
 163
 164         self.fontname = fontname
 165         self.basefontname = basefontname
 166         self.charcodes = set(charcodes)
 167         self.fontdescriptor = fontdescriptor
 168         self.encoding = encoding
 169         self.metric = metric
 170
 171     def merge(self, other):
 172         self.charcodes.update(other.charcodes)
 173
 174     def write(self, file, writer, registry):
 175         file.write("<<\n"
 176                    "/Type /Font\n"
 177                    "/Subtype /Type1\n")
 178         file.write("/Name /%s\n" % self.fontname)
 179         file.write("/BaseFont /%s\n" % self.basefontname)
 180         firstchar = min(self.charcodes)
 181         lastchar = max(self.charcodes)
 182         file.write("/FirstChar %d\n" % firstchar)
 183         file.write("/LastChar %d\n" % lastchar)
 184         file.write("/Widths\n"
 185                    "[")
 186         if self.encoding:
 187             encoding = self.encoding.getvector()
 188         else:
 189             if self.fontdescriptor.fontfile.t1file.encoding is None:
 190                 self.fontdescriptor.fontfile.t1file._encoding()
 191             encoding = self.fontdescriptor.fontfile.t1file.encoding
 192         for i in range(firstchar, lastchar+1):
 193             if i:
 194                 if not (i % 8):
 195                     file.write("\n")
 196                 else:
 197                     file.write(" ")
 198             if i in self.charcodes:
 199                 if self.metric is not None:
 200                     file.write("%i" % self.metric.width_ds(encoding[i]))
 201                 else:
 202                     file.write("%i" % self.fontdescriptor.fontfile.t1file.getglyphinfo(encoding[i])[0])
 203             else:
 204                 file.write("0")
 205         file.write(" ]\n")
 206         file.write("/FontDescriptor %d 0 R\n" % registry.getrefno(self.fontdescriptor))
 207         if self.encoding:
 208             file.write("/Encoding %d 0 R\n" % registry.getrefno(self.encoding))
 209         file.write(">>\n")
 210
 211
 212 class PDFstdfont(pdfwriter.PDFobject):
 213
 214     def __init__(self, basename):
 215         pdfwriter.PDFobject.__init__(self, "font", "stdfont-%s" % basename)
 216         self.name = basename # name is ignored by acroread
 217         self.basename = basename
 218
 219     def write(self, file, writer, registry):
 220         file.write("<</BaseFont /%s\n" % self.basename)
 221         file.write("/Name /%s\n" % self.name)
 222         file.write("/Type /Font\n")
 223         file.write("/Subtype /Type1\n")
 224         file.write(">>\n")
 225
 226 # the 14 standard fonts that are always available in PDF
 227 PDFTimesRoman           = PDFstdfont("Times-Roman")
 228 PDFTimesBold            = PDFstdfont("Times-Bold")
 229 PDFTimesItalic          = PDFstdfont("Times-Italic")
 230 PDFTimesBoldItalic      = PDFstdfont("Times-BoldItalic")
 231 PDFHelvetica            = PDFstdfont("Helvetica")
 232 PDFHelveticaBold        = PDFstdfont("Helvetica-Bold")
 233 PDFHelveticaOblique     = PDFstdfont("Helvetica-Oblique")
 234 PDFHelveticaBoldOblique = PDFstdfont("Helvetica-BoldOblique")
 235 PDFCourier              = PDFstdfont("Courier")
 236 PDFCourierBold          = PDFstdfont("Courier-Bold")
 237 PDFCourierOblique       = PDFstdfont("Courier-Oblique")
 238 PDFCourierBoldOblique   = PDFstdfont("Courier-BoldOblique")
 239 PDFSymbol               = PDFstdfont("Symbol")
 240 PDFZapfDingbats         = PDFstdfont("ZapfDingbats")
 241
 242
 243 class PDFfontdescriptor(pdfwriter.PDFobject):
 244
 245     def __init__(self, fontname, fontfile, metric):
 246         pdfwriter.PDFobject.__init__(self, "fontdescriptor", fontname)
 247         self.fontname = fontname
 248         self.fontfile = fontfile
 249         self.metric = metric
 250
 251     def write(self, file, writer, registry):
 252         file.write("<<\n"
 253                    "/Type /FontDescriptor\n"
 254                    "/FontName /%s\n" % self.fontname)
 255         if self.metric is not None:
 256             self.metric.writePDFfontinfo(file)
 257         else:
 258             self.fontfile.t1file.writePDFfontinfo(file)
 259         if self.fontfile is not None:
 260             file.write("/FontFile %d 0 R\n" % registry.getrefno(self.fontfile))
 261         file.write(">>\n")
 262
 263
 264 class PDFfontfile(pdfwriter.PDFobject):
 265
 266     def __init__(self, t1file, glyphnames, charcodes):
 267         pdfwriter.PDFobject.__init__(self, "fontfile", t1file.name)
 268         self.t1file = t1file
 269         self.glyphnames = set(glyphnames)
 270         self.charcodes = set(charcodes)
 271
 272     def merge(self, other):
 273         self.glyphnames.update(other.glyphnames)
 274         self.charcodes.update(other.charcodes)
 275
 276     def write(self, file, writer, registry):
 277         if writer.strip_fonts:
 278             self.t1file.getstrippedfont(self.glyphnames, self.charcodes).outputPDF(file, writer)
 279         else:
 280             self.t1file.outputPDF(file, writer)
 281
 282
 283 class PDFencoding(pdfwriter.PDFobject):
 284
 285     def __init__(self, encoding, name):
 286         pdfwriter.PDFobject.__init__(self, "encoding", name)
 287         self.encoding = encoding
 288
 289     def getvector(self):
 290         # As self.encoding might be appended after the constructur has set it,
 291         # we need to defer the calculation until the whole content was constructed.
 292         vector = [None] * len(self.encoding)
 293         for glyphname, charcode in list(self.encoding.items()):
 294             vector[charcode] = glyphname
 295         return vector
 296
 297     def write(self, file, writer, registry):
 298         file.write("<<\n"
 299                    "/Type /Encoding\n"
 300                    "/Differences\n"
 301                    "[0")
 302         for i, glyphname in enumerate(self.getvector()):
 303             if i:
 304                 if not (i % 8):
 305                     file.write("\n")
 306                 else:
 307                     file.write(" ")
 308             file.write("/%s" % glyphname)
 309         file.write("]\n"
 310                    ">>\n")
 311
 312
 313 ##############################################################################
 314 # basic PyX text output
 315 ##############################################################################
 316
 317 class font:
 318
 319     def text(self, x, y, charcodes, size_pt, **kwargs):
 320         return self.text_pt(unit.topt(x), unit.topt(y), charcodes, size_pt, **kwargs)
 321
 322
 323 class T1font(font):
 324
 325     def __init__(self, t1file, metric=None):
 326         self.t1file = t1file
 327         self.name = t1file.name
 328         self.metric = metric
 329
 330     def text_pt(self, x, y, charcodes, size_pt, **kwargs):
 331         return T1text_pt(self, x, y, charcodes, size_pt, **kwargs)
 332
 333
 334 class T1builtinfont(T1font):
 335
 336     def __init__(self, name, metric):
 337         self.name = name
 338         self.t1file = None
 339         self.metric = metric
 340
 341
 342 class selectedfont:
 343
 344     def __init__(self, name, size_pt):
 345         self.name = name
 346         self.size_pt = size_pt
 347
 348     def __ne__(self, other):
 349         return self.name != other.name or self.size_pt != other.size_pt
 350
 351     def outputPS(self, file, writer):
 352         file.write("/%s %f selectfont\n" % (self.name, self.size_pt))
 353
 354     def outputPDF(self, file, writer):
 355         file.write("/%s %f Tf\n" % (self.name, self.size_pt))
 356
 357
 358 class text_pt(baseclasses.canvasitem):
 359
 360     def requiretextregion(self):
 361         return True
 362
 363
 364 class T1text_pt(text_pt):
 365
 366     def __init__(self, font, x_pt, y_pt, charcodes, size_pt, decoding=afmfile.unicodestring, slant=None, ignorebbox=False, kerning=False, ligatures=False, spaced_pt=0):
 367         if decoding is not None:
 368             self.glyphnames = [decoding[character] for character in charcodes]
 369             self.decode = True
 370         else:
 371             self.charcodes = charcodes
 372             self.decode = False
 373         self.font = font
 374         self.x_pt = x_pt
 375         self.y_pt = y_pt
 376         self.size_pt = size_pt
 377         self.slant = slant
 378         self.ignorebbox = ignorebbox
 379         self.kerning = kerning
 380         self.ligatures = ligatures
 381         self.spaced_pt = spaced_pt
 382         self._textpath = None
 383
 384         if self.kerning and not self.decode:
 385             raise ValueError("decoding required for font metric access (kerning)")
 386         if self.ligatures and not self.decode:
 387             raise ValueError("decoding required for font metric access (ligatures)")
 388         if self.ligatures:
 389             self.glyphnames = self.font.metric.resolveligatures(self.glyphnames)
 390
 391     def bbox(self):
 392         if self.font.metric is None:
 393             logger.warning("We are about to extract the bounding box from the path of the text. This is slow and differs from the font metric information. You should provide an afm file whenever possible.")
 394             return self.textpath().bbox()
 395         if not self.decode:
 396             raise ValueError("decoding required for font metric access (bbox)")
 397         return bbox.bbox_pt(self.x_pt,
 398                             self.y_pt+self.font.metric.depth_pt(self.glyphnames, self.size_pt),
 399                             self.x_pt+self.font.metric.width_pt(self.glyphnames, self.size_pt),
 400                             self.y_pt+self.font.metric.height_pt(self.glyphnames, self.size_pt))
 401
 402     def getencodingname(self, encodings):
 403         """returns the name of the encoding (in encodings) mapping self.glyphnames to codepoints
 404         If no such encoding can be found or extended, a new encoding is added to encodings
 405         """
 406         glyphnames = set(self.glyphnames)
 407         if len(glyphnames) > 256:
 408             raise ValueError("glyphs do not fit into one single encoding")
 409         for encodingname, encoding in list(encodings.items()):
 410             glyphsmissing = []
 411             for glyphname in glyphnames:
 412                 if glyphname not in list(encoding.keys()):
 413                     glyphsmissing.append(glyphname)
 414
 415             if len(glyphsmissing) + len(encoding) < 256:
 416                 # new glyphs fit in existing encoding which will thus be extended
 417                 for glyphname in glyphsmissing:
 418                     encoding[glyphname] = len(encoding)
 419                 return encodingname
 420         # create a new encoding for the glyphnames
 421         encodingname = "encoding%d" % len(encodings)
 422         encodings[encodingname] = dict([(glyphname, i) for i, glyphname in enumerate(glyphnames)])
 423         return encodingname
 424
 425     def textpath(self):
 426         if self._textpath is None:
 427             if self.decode:
 428                 if self.kerning:
 429                     data = self.font.metric.resolvekernings(self.glyphnames, self.size_pt)
 430                 else:
 431                     data = self.glyphnames
 432             else:
 433                 data = self.charcodes
 434             self._textpath = path.path()
 435             x_pt = self.x_pt
 436             y_pt = self.y_pt
 437             for i, value in enumerate(data):
 438                 if self.kerning and i % 2:
 439                     if value is not None:
 440                         x_pt += value
 441                 else:
 442                     if i:
 443                         x_pt += self.spaced_pt
 444                     glyphpath = self.font.t1file.getglyphpath_pt(x_pt, y_pt, value, self.size_pt, convertcharcode=not self.decode)
 445                     self._textpath += glyphpath.path
 446                     x_pt += glyphpath.wx_pt
 447                     y_pt += glyphpath.wy_pt
 448         return self._textpath
 449
 450     def processPS(self, file, writer, context, registry, bbox):
 451         if not self.ignorebbox:
 452             bbox += self.bbox()
 453
 454         if writer.text_as_path:
 455             deco.decoratedpath(self.textpath(), fillstyles=[]).processPS(file, writer, context, registry, bbox)
 456         else:
 457             # register resources
 458             if self.font.t1file is not None:
 459                 if self.decode:
 460                     registry.add(PST1file(self.font.t1file, self.glyphnames, []))
 461                 else:
 462                     registry.add(PST1file(self.font.t1file, [], self.charcodes))
 463
 464             fontname = self.font.name
 465             if self.decode:
 466                 encodingname = self.getencodingname(writer.encodings.setdefault(self.font.name, {}))
 467                 encoding = writer.encodings[self.font.name][encodingname]
 468                 newfontname = "%s-%s" % (fontname, encodingname)
 469                 registry.add(_ReEncodeFont)
 470                 registry.add(PSreencodefont(fontname, newfontname, encoding))
 471                 fontname = newfontname
 472
 473             if self.slant:
 474                 newfontmatrix = trafo.trafo_pt(matrix=((1, self.slant), (0, 1)))
 475                 if self.font.t1file is not None:
 476                     newfontmatrix = newfontmatrix * self.font.t1file.fontmatrix
 477                 newfontname = "%s-slant%f" % (fontname, self.slant)
 478                 registry.add(_ChangeFontMatrix)
 479                 registry.add(PSchangefontmatrix(fontname, newfontname, newfontmatrix))
 480                 fontname = newfontname
 481
 482             # select font if necessary
 483             sf = selectedfont(fontname, self.size_pt)
 484             if context.selectedfont is None or sf != context.selectedfont:
 485                 context.selectedfont = sf
 486                 sf.outputPS(file, writer)
 487
 488             file.write("%f %f moveto (" % (self.x_pt, self.y_pt))
 489             if self.decode:
 490                 if self.kerning:
 491                     data = self.font.metric.resolvekernings(self.glyphnames, self.size_pt)
 492                 else:
 493                     data = self.glyphnames
 494             else:
 495                 data = self.charcodes
 496             for i, value in enumerate(data):
 497                 if self.kerning and i % 2:
 498                     if value is not None:
 499                         file.write(") show\n%f 0 rmoveto (" % (value+self.spaced_pt))
 500                     elif self.spaced_pt:
 501                         file.write(") show\n%f 0 rmoveto (" % self.spaced_pt)
 502                 else:
 503                     if i and not self.kerning and self.spaced_pt:
 504                         file.write(") show\n%f 0 rmoveto (" % self.spaced_pt)
 505                     if self.decode:
 506                         value = encoding[value]
 507                     if 32 < value < 127 and chr(value) not in "()[]<>\\":
 508                         file.write("%s" % chr(value))
 509                     else:
 510                         file.write("\\%03o" % value)
 511             file.write(") show\n")
 512
 513     def processPDF(self, file, writer, context, registry, bbox):
 514         if not self.ignorebbox:
 515             bbox += self.bbox()
 516
 517         if writer.text_as_path:
 518             deco.decoratedpath(self.textpath(), fillstyles=[]).processPDF(file, writer, context, registry, bbox)
 519         else:
 520             if self.decode:
 521                 encodingname = self.getencodingname(writer.encodings.setdefault(self.font.name, {}))
 522                 encoding = writer.encodings[self.font.name][encodingname]
 523                 charcodes = [encoding[glyphname] for glyphname in self.glyphnames]
 524             else:
 525                 charcodes = self.charcodes
 526
 527             # create resources
 528             fontname = self.font.name
 529             if self.decode:
 530                 newfontname = "%s-%s" % (fontname, encodingname)
 531                 _encoding = PDFencoding(encoding, newfontname)
 532                 fontname = newfontname
 533             else:
 534                 _encoding = None
 535             if self.font.t1file is not None:
 536                 if self.decode:
 537                     fontfile = PDFfontfile(self.font.t1file, self.glyphnames, [])
 538                 else:
 539                     fontfile = PDFfontfile(self.font.t1file, [], self.charcodes)
 540             else:
 541                 fontfile = None
 542             fontdescriptor = PDFfontdescriptor(self.font.name, fontfile, self.font.metric)
 543             font = PDFfont(fontname, self.font.name, charcodes, fontdescriptor, _encoding, self.font.metric)
 544
 545             # register resources
 546             if fontfile is not None:
 547                 registry.add(fontfile)
 548             registry.add(fontdescriptor)
 549             if _encoding is not None:
 550                 registry.add(_encoding)
 551             registry.add(font)
 552
 553             registry.addresource("Font", fontname, font, procset="Text")
 554
 555             if self.slant is None:
 556                 slantvalue = 0
 557             else:
 558                 slantvalue = self.slant
 559
 560             # select font if necessary
 561             sf = selectedfont(fontname, self.size_pt)
 562             if context.selectedfont is None or sf != context.selectedfont:
 563                 context.selectedfont = sf
 564                 sf.outputPDF(file, writer)
 565
 566             if self.kerning:
 567                 file.write("1 0 %f 1 %f %f Tm [(" % (slantvalue, self.x_pt, self.y_pt))
 568             else:
 569                 file.write("1 0 %f 1 %f %f Tm (" % (slantvalue, self.x_pt, self.y_pt))
 570             if self.decode:
 571                 if self.kerning:
 572                     data = self.font.metric.resolvekernings(self.glyphnames)
 573                 else:
 574                     data = self.glyphnames
 575             else:
 576                 data = self.charcodes
 577             for i, value in enumerate(data):
 578                 if self.kerning and i % 2:
 579                     if value is not None:
 580                         file.write(")%f(" % (-value-self.spaced_pt))
 581                     elif self.spaced_pt:
 582                         file.write(")%f(" % (-self.spaced_pt))
 583                 else:
 584                     if i and not self.kerning and self.spaced_pt:
 585                         file.write(")%f(" % (-self.spaced_pt))
 586                     if self.decode:
 587                         value = encoding[value]
 588                     if 32 <= value <= 127 and chr(value) not in "()[]<>\\":
 589                         file.write("%s" % chr(value))
 590                     else:
 591                         file.write("\\%03o" % value)
 592             if self.kerning:
 593                 file.write(")] TJ\n")
 594             else:
 595                 file.write(") Tj\n")