otBase.py 52 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458
  1. from fontTools.config import OPTIONS
  2. from fontTools.misc.textTools import Tag, bytesjoin
  3. from .DefaultTable import DefaultTable
  4. from enum import IntEnum
  5. import sys
  6. import array
  7. import struct
  8. import logging
  9. from functools import lru_cache
  10. from typing import Iterator, NamedTuple, Optional, Tuple
  11. log = logging.getLogger(__name__)
  12. have_uharfbuzz = False
  13. try:
  14. import uharfbuzz as hb
  15. have_uharfbuzz = True
  16. except ImportError:
  17. pass
  18. USE_HARFBUZZ_REPACKER = OPTIONS[f"{__name__}:USE_HARFBUZZ_REPACKER"]
  19. class OverflowErrorRecord(object):
  20. def __init__(self, overflowTuple):
  21. self.tableType = overflowTuple[0]
  22. self.LookupListIndex = overflowTuple[1]
  23. self.SubTableIndex = overflowTuple[2]
  24. self.itemName = overflowTuple[3]
  25. self.itemIndex = overflowTuple[4]
  26. def __repr__(self):
  27. return str(
  28. (
  29. self.tableType,
  30. "LookupIndex:",
  31. self.LookupListIndex,
  32. "SubTableIndex:",
  33. self.SubTableIndex,
  34. "ItemName:",
  35. self.itemName,
  36. "ItemIndex:",
  37. self.itemIndex,
  38. )
  39. )
  40. class OTLOffsetOverflowError(Exception):
  41. def __init__(self, overflowErrorRecord):
  42. self.value = overflowErrorRecord
  43. def __str__(self):
  44. return repr(self.value)
  45. class RepackerState(IntEnum):
  46. # Repacking control flow is implemnted using a state machine. The state machine table:
  47. #
  48. # State | Packing Success | Packing Failed | Exception Raised |
  49. # ------------+-----------------+----------------+------------------+
  50. # PURE_FT | Return result | PURE_FT | Return failure |
  51. # HB_FT | Return result | HB_FT | FT_FALLBACK |
  52. # FT_FALLBACK | HB_FT | FT_FALLBACK | Return failure |
  53. # Pack only with fontTools, don't allow sharing between extensions.
  54. PURE_FT = 1
  55. # Attempt to pack with harfbuzz (allowing sharing between extensions)
  56. # use fontTools to attempt overflow resolution.
  57. HB_FT = 2
  58. # Fallback if HB/FT packing gets stuck. Pack only with fontTools, don't allow sharing between
  59. # extensions.
  60. FT_FALLBACK = 3
  61. class BaseTTXConverter(DefaultTable):
  62. """Generic base class for TTX table converters. It functions as an
  63. adapter between the TTX (ttLib actually) table model and the model
  64. we use for OpenType tables, which is necessarily subtly different.
  65. """
  66. def decompile(self, data, font):
  67. """Create an object from the binary data. Called automatically on access."""
  68. from . import otTables
  69. reader = OTTableReader(data, tableTag=self.tableTag)
  70. tableClass = getattr(otTables, self.tableTag)
  71. self.table = tableClass()
  72. self.table.decompile(reader, font)
  73. def compile(self, font):
  74. """Compiles the table into binary. Called automatically on save."""
  75. # General outline:
  76. # Create a top-level OTTableWriter for the GPOS/GSUB table.
  77. # Call the compile method for the the table
  78. # for each 'converter' record in the table converter list
  79. # call converter's write method for each item in the value.
  80. # - For simple items, the write method adds a string to the
  81. # writer's self.items list.
  82. # - For Struct/Table/Subtable items, it add first adds new writer to the
  83. # to the writer's self.items, then calls the item's compile method.
  84. # This creates a tree of writers, rooted at the GUSB/GPOS writer, with
  85. # each writer representing a table, and the writer.items list containing
  86. # the child data strings and writers.
  87. # call the getAllData method
  88. # call _doneWriting, which removes duplicates
  89. # call _gatherTables. This traverses the tables, adding unique occurences to a flat list of tables
  90. # Traverse the flat list of tables, calling getDataLength on each to update their position
  91. # Traverse the flat list of tables again, calling getData each get the data in the table, now that
  92. # pos's and offset are known.
  93. # If a lookup subtable overflows an offset, we have to start all over.
  94. overflowRecord = None
  95. # this is 3-state option: default (None) means automatically use hb.repack or
  96. # silently fall back if it fails; True, use it and raise error if not possible
  97. # or it errors out; False, don't use it, even if you can.
  98. use_hb_repack = font.cfg[USE_HARFBUZZ_REPACKER]
  99. if self.tableTag in ("GSUB", "GPOS"):
  100. if use_hb_repack is False:
  101. log.debug(
  102. "hb.repack disabled, compiling '%s' with pure-python serializer",
  103. self.tableTag,
  104. )
  105. elif not have_uharfbuzz:
  106. if use_hb_repack is True:
  107. raise ImportError("No module named 'uharfbuzz'")
  108. else:
  109. assert use_hb_repack is None
  110. log.debug(
  111. "uharfbuzz not found, compiling '%s' with pure-python serializer",
  112. self.tableTag,
  113. )
  114. if (
  115. use_hb_repack in (None, True)
  116. and have_uharfbuzz
  117. and self.tableTag in ("GSUB", "GPOS")
  118. ):
  119. state = RepackerState.HB_FT
  120. else:
  121. state = RepackerState.PURE_FT
  122. hb_first_error_logged = False
  123. lastOverflowRecord = None
  124. while True:
  125. try:
  126. writer = OTTableWriter(tableTag=self.tableTag)
  127. self.table.compile(writer, font)
  128. if state == RepackerState.HB_FT:
  129. return self.tryPackingHarfbuzz(writer, hb_first_error_logged)
  130. elif state == RepackerState.PURE_FT:
  131. return self.tryPackingFontTools(writer)
  132. elif state == RepackerState.FT_FALLBACK:
  133. # Run packing with FontTools only, but don't return the result as it will
  134. # not be optimally packed. Once a successful packing has been found, state is
  135. # changed back to harfbuzz packing to produce the final, optimal, packing.
  136. self.tryPackingFontTools(writer)
  137. log.debug(
  138. "Re-enabling sharing between extensions and switching back to "
  139. "harfbuzz+fontTools packing."
  140. )
  141. state = RepackerState.HB_FT
  142. except OTLOffsetOverflowError as e:
  143. hb_first_error_logged = True
  144. ok = self.tryResolveOverflow(font, e, lastOverflowRecord)
  145. lastOverflowRecord = e.value
  146. if ok:
  147. continue
  148. if state is RepackerState.HB_FT:
  149. log.debug(
  150. "Harfbuzz packing out of resolutions, disabling sharing between extensions and "
  151. "switching to fontTools only packing."
  152. )
  153. state = RepackerState.FT_FALLBACK
  154. else:
  155. raise
  156. def tryPackingHarfbuzz(self, writer, hb_first_error_logged):
  157. try:
  158. log.debug("serializing '%s' with hb.repack", self.tableTag)
  159. return writer.getAllDataUsingHarfbuzz(self.tableTag)
  160. except (ValueError, MemoryError, hb.RepackerError) as e:
  161. # Only log hb repacker errors the first time they occur in
  162. # the offset-overflow resolution loop, they are just noisy.
  163. # Maybe we can revisit this if/when uharfbuzz actually gives
  164. # us more info as to why hb.repack failed...
  165. if not hb_first_error_logged:
  166. error_msg = f"{type(e).__name__}"
  167. if str(e) != "":
  168. error_msg += f": {e}"
  169. log.warning(
  170. "hb.repack failed to serialize '%s', attempting fonttools resolutions "
  171. "; the error message was: %s",
  172. self.tableTag,
  173. error_msg,
  174. )
  175. hb_first_error_logged = True
  176. return writer.getAllData(remove_duplicate=False)
  177. def tryPackingFontTools(self, writer):
  178. return writer.getAllData()
  179. def tryResolveOverflow(self, font, e, lastOverflowRecord):
  180. ok = 0
  181. if lastOverflowRecord == e.value:
  182. # Oh well...
  183. return ok
  184. overflowRecord = e.value
  185. log.info("Attempting to fix OTLOffsetOverflowError %s", e)
  186. if overflowRecord.itemName is None:
  187. from .otTables import fixLookupOverFlows
  188. ok = fixLookupOverFlows(font, overflowRecord)
  189. else:
  190. from .otTables import fixSubTableOverFlows
  191. ok = fixSubTableOverFlows(font, overflowRecord)
  192. if ok:
  193. return ok
  194. # Try upgrading lookup to Extension and hope
  195. # that cross-lookup sharing not happening would
  196. # fix overflow...
  197. from .otTables import fixLookupOverFlows
  198. return fixLookupOverFlows(font, overflowRecord)
  199. def toXML(self, writer, font):
  200. self.table.toXML2(writer, font)
  201. def fromXML(self, name, attrs, content, font):
  202. from . import otTables
  203. if not hasattr(self, "table"):
  204. tableClass = getattr(otTables, self.tableTag)
  205. self.table = tableClass()
  206. self.table.fromXML(name, attrs, content, font)
  207. self.table.populateDefaults()
  208. def ensureDecompiled(self, recurse=True):
  209. self.table.ensureDecompiled(recurse=recurse)
  210. # https://github.com/fonttools/fonttools/pull/2285#issuecomment-834652928
  211. assert len(struct.pack("i", 0)) == 4
  212. assert array.array("i").itemsize == 4, "Oops, file a bug against fonttools."
  213. class OTTableReader(object):
  214. """Helper class to retrieve data from an OpenType table."""
  215. __slots__ = ("data", "offset", "pos", "localState", "tableTag")
  216. def __init__(self, data, localState=None, offset=0, tableTag=None):
  217. self.data = data
  218. self.offset = offset
  219. self.pos = offset
  220. self.localState = localState
  221. self.tableTag = tableTag
  222. def advance(self, count):
  223. self.pos += count
  224. def seek(self, pos):
  225. self.pos = pos
  226. def copy(self):
  227. other = self.__class__(self.data, self.localState, self.offset, self.tableTag)
  228. other.pos = self.pos
  229. return other
  230. def getSubReader(self, offset):
  231. offset = self.offset + offset
  232. return self.__class__(self.data, self.localState, offset, self.tableTag)
  233. def readValue(self, typecode, staticSize):
  234. pos = self.pos
  235. newpos = pos + staticSize
  236. (value,) = struct.unpack(f">{typecode}", self.data[pos:newpos])
  237. self.pos = newpos
  238. return value
  239. def readArray(self, typecode, staticSize, count):
  240. pos = self.pos
  241. newpos = pos + count * staticSize
  242. value = array.array(typecode, self.data[pos:newpos])
  243. if sys.byteorder != "big":
  244. value.byteswap()
  245. self.pos = newpos
  246. return value.tolist()
  247. def readInt8(self):
  248. return self.readValue("b", staticSize=1)
  249. def readInt8Array(self, count):
  250. return self.readArray("b", staticSize=1, count=count)
  251. def readShort(self):
  252. return self.readValue("h", staticSize=2)
  253. def readShortArray(self, count):
  254. return self.readArray("h", staticSize=2, count=count)
  255. def readLong(self):
  256. return self.readValue("i", staticSize=4)
  257. def readLongArray(self, count):
  258. return self.readArray("i", staticSize=4, count=count)
  259. def readUInt8(self):
  260. return self.readValue("B", staticSize=1)
  261. def readUInt8Array(self, count):
  262. return self.readArray("B", staticSize=1, count=count)
  263. def readUShort(self):
  264. return self.readValue("H", staticSize=2)
  265. def readUShortArray(self, count):
  266. return self.readArray("H", staticSize=2, count=count)
  267. def readULong(self):
  268. return self.readValue("I", staticSize=4)
  269. def readULongArray(self, count):
  270. return self.readArray("I", staticSize=4, count=count)
  271. def readUInt24(self):
  272. pos = self.pos
  273. newpos = pos + 3
  274. (value,) = struct.unpack(">l", b"\0" + self.data[pos:newpos])
  275. self.pos = newpos
  276. return value
  277. def readUInt24Array(self, count):
  278. return [self.readUInt24() for _ in range(count)]
  279. def readTag(self):
  280. pos = self.pos
  281. newpos = pos + 4
  282. value = Tag(self.data[pos:newpos])
  283. assert len(value) == 4, value
  284. self.pos = newpos
  285. return value
  286. def readData(self, count):
  287. pos = self.pos
  288. newpos = pos + count
  289. value = self.data[pos:newpos]
  290. self.pos = newpos
  291. return value
  292. def __setitem__(self, name, value):
  293. state = self.localState.copy() if self.localState else dict()
  294. state[name] = value
  295. self.localState = state
  296. def __getitem__(self, name):
  297. return self.localState and self.localState[name]
  298. def __contains__(self, name):
  299. return self.localState and name in self.localState
  300. class OffsetToWriter(object):
  301. def __init__(self, subWriter, offsetSize):
  302. self.subWriter = subWriter
  303. self.offsetSize = offsetSize
  304. def __eq__(self, other):
  305. if type(self) != type(other):
  306. return NotImplemented
  307. return self.subWriter == other.subWriter and self.offsetSize == other.offsetSize
  308. def __hash__(self):
  309. # only works after self._doneWriting() has been called
  310. return hash((self.subWriter, self.offsetSize))
  311. class OTTableWriter(object):
  312. """Helper class to gather and assemble data for OpenType tables."""
  313. def __init__(self, localState=None, tableTag=None):
  314. self.items = []
  315. self.pos = None
  316. self.localState = localState
  317. self.tableTag = tableTag
  318. self.parent = None
  319. self.name = "<none>"
  320. def __setitem__(self, name, value):
  321. state = self.localState.copy() if self.localState else dict()
  322. state[name] = value
  323. self.localState = state
  324. def __getitem__(self, name):
  325. return self.localState[name]
  326. def __delitem__(self, name):
  327. del self.localState[name]
  328. # assembler interface
  329. def getDataLength(self):
  330. """Return the length of this table in bytes, without subtables."""
  331. l = 0
  332. for item in self.items:
  333. if hasattr(item, "getCountData"):
  334. l += item.size
  335. elif hasattr(item, "subWriter"):
  336. l += item.offsetSize
  337. else:
  338. l = l + len(item)
  339. return l
  340. def getData(self):
  341. """Assemble the data for this writer/table, without subtables."""
  342. items = list(self.items) # make a shallow copy
  343. pos = self.pos
  344. numItems = len(items)
  345. for i in range(numItems):
  346. item = items[i]
  347. if hasattr(item, "subWriter"):
  348. if item.offsetSize == 4:
  349. items[i] = packULong(item.subWriter.pos - pos)
  350. elif item.offsetSize == 2:
  351. try:
  352. items[i] = packUShort(item.subWriter.pos - pos)
  353. except struct.error:
  354. # provide data to fix overflow problem.
  355. overflowErrorRecord = self.getOverflowErrorRecord(
  356. item.subWriter
  357. )
  358. raise OTLOffsetOverflowError(overflowErrorRecord)
  359. elif item.offsetSize == 3:
  360. items[i] = packUInt24(item.subWriter.pos - pos)
  361. else:
  362. raise ValueError(item.offsetSize)
  363. return bytesjoin(items)
  364. def getDataForHarfbuzz(self):
  365. """Assemble the data for this writer/table with all offset field set to 0"""
  366. items = list(self.items)
  367. packFuncs = {2: packUShort, 3: packUInt24, 4: packULong}
  368. for i, item in enumerate(items):
  369. if hasattr(item, "subWriter"):
  370. # Offset value is not needed in harfbuzz repacker, so setting offset to 0 to avoid overflow here
  371. if item.offsetSize in packFuncs:
  372. items[i] = packFuncs[item.offsetSize](0)
  373. else:
  374. raise ValueError(item.offsetSize)
  375. return bytesjoin(items)
  376. def __hash__(self):
  377. # only works after self._doneWriting() has been called
  378. return hash(self.items)
  379. def __ne__(self, other):
  380. result = self.__eq__(other)
  381. return result if result is NotImplemented else not result
  382. def __eq__(self, other):
  383. if type(self) != type(other):
  384. return NotImplemented
  385. return self.items == other.items
  386. def _doneWriting(self, internedTables, shareExtension=False):
  387. # Convert CountData references to data string items
  388. # collapse duplicate table references to a unique entry
  389. # "tables" are OTTableWriter objects.
  390. # For Extension Lookup types, we can
  391. # eliminate duplicates only within the tree under the Extension Lookup,
  392. # as offsets may exceed 64K even between Extension LookupTable subtables.
  393. isExtension = hasattr(self, "Extension")
  394. # Certain versions of Uniscribe reject the font if the GSUB/GPOS top-level
  395. # arrays (ScriptList, FeatureList, LookupList) point to the same, possibly
  396. # empty, array. So, we don't share those.
  397. # See: https://github.com/fonttools/fonttools/issues/518
  398. dontShare = hasattr(self, "DontShare")
  399. if isExtension and not shareExtension:
  400. internedTables = {}
  401. items = self.items
  402. for i, item in enumerate(items):
  403. if hasattr(item, "getCountData"):
  404. items[i] = item.getCountData()
  405. elif hasattr(item, "subWriter"):
  406. item.subWriter._doneWriting(
  407. internedTables, shareExtension=shareExtension
  408. )
  409. # At this point, all subwriters are hashable based on their items.
  410. # (See hash and comparison magic methods above.) So the ``setdefault``
  411. # call here will return the first writer object we've seen with
  412. # equal content, or store it in the dictionary if it's not been
  413. # seen yet. We therefore replace the subwriter object with an equivalent
  414. # object, which deduplicates the tree.
  415. if not dontShare:
  416. items[i].subWriter = internedTables.setdefault(
  417. item.subWriter, item.subWriter
  418. )
  419. self.items = tuple(items)
  420. def _gatherTables(self, tables, extTables, done):
  421. # Convert table references in self.items tree to a flat
  422. # list of tables in depth-first traversal order.
  423. # "tables" are OTTableWriter objects.
  424. # We do the traversal in reverse order at each level, in order to
  425. # resolve duplicate references to be the last reference in the list of tables.
  426. # For extension lookups, duplicate references can be merged only within the
  427. # writer tree under the extension lookup.
  428. done[id(self)] = True
  429. numItems = len(self.items)
  430. iRange = list(range(numItems))
  431. iRange.reverse()
  432. isExtension = hasattr(self, "Extension")
  433. selfTables = tables
  434. if isExtension:
  435. assert (
  436. extTables is not None
  437. ), "Program or XML editing error. Extension subtables cannot contain extensions subtables"
  438. tables, extTables, done = extTables, None, {}
  439. # add Coverage table if it is sorted last.
  440. sortCoverageLast = False
  441. if hasattr(self, "sortCoverageLast"):
  442. # Find coverage table
  443. for i in range(numItems):
  444. item = self.items[i]
  445. if (
  446. hasattr(item, "subWriter")
  447. and getattr(item.subWriter, "name", None) == "Coverage"
  448. ):
  449. sortCoverageLast = True
  450. break
  451. if id(item.subWriter) not in done:
  452. item.subWriter._gatherTables(tables, extTables, done)
  453. else:
  454. # We're a new parent of item
  455. pass
  456. for i in iRange:
  457. item = self.items[i]
  458. if not hasattr(item, "subWriter"):
  459. continue
  460. if (
  461. sortCoverageLast
  462. and (i == 1)
  463. and getattr(item.subWriter, "name", None) == "Coverage"
  464. ):
  465. # we've already 'gathered' it above
  466. continue
  467. if id(item.subWriter) not in done:
  468. item.subWriter._gatherTables(tables, extTables, done)
  469. else:
  470. # Item is already written out by other parent
  471. pass
  472. selfTables.append(self)
  473. def _gatherGraphForHarfbuzz(self, tables, obj_list, done, objidx, virtual_edges):
  474. real_links = []
  475. virtual_links = []
  476. item_idx = objidx
  477. # Merge virtual_links from parent
  478. for idx in virtual_edges:
  479. virtual_links.append((0, 0, idx))
  480. sortCoverageLast = False
  481. coverage_idx = 0
  482. if hasattr(self, "sortCoverageLast"):
  483. # Find coverage table
  484. for i, item in enumerate(self.items):
  485. if getattr(item, "name", None) == "Coverage":
  486. sortCoverageLast = True
  487. if id(item) not in done:
  488. coverage_idx = item_idx = item._gatherGraphForHarfbuzz(
  489. tables, obj_list, done, item_idx, virtual_edges
  490. )
  491. else:
  492. coverage_idx = done[id(item)]
  493. virtual_edges.append(coverage_idx)
  494. break
  495. child_idx = 0
  496. offset_pos = 0
  497. for i, item in enumerate(self.items):
  498. if hasattr(item, "subWriter"):
  499. pos = offset_pos
  500. elif hasattr(item, "getCountData"):
  501. offset_pos += item.size
  502. continue
  503. else:
  504. offset_pos = offset_pos + len(item)
  505. continue
  506. if id(item.subWriter) not in done:
  507. child_idx = item_idx = item.subWriter._gatherGraphForHarfbuzz(
  508. tables, obj_list, done, item_idx, virtual_edges
  509. )
  510. else:
  511. child_idx = done[id(item.subWriter)]
  512. real_edge = (pos, item.offsetSize, child_idx)
  513. real_links.append(real_edge)
  514. offset_pos += item.offsetSize
  515. tables.append(self)
  516. obj_list.append((real_links, virtual_links))
  517. item_idx += 1
  518. done[id(self)] = item_idx
  519. if sortCoverageLast:
  520. virtual_edges.pop()
  521. return item_idx
  522. def getAllDataUsingHarfbuzz(self, tableTag):
  523. """The Whole table is represented as a Graph.
  524. Assemble graph data and call Harfbuzz repacker to pack the table.
  525. Harfbuzz repacker is faster and retain as much sub-table sharing as possible, see also:
  526. https://github.com/harfbuzz/harfbuzz/blob/main/docs/repacker.md
  527. The input format for hb.repack() method is explained here:
  528. https://github.com/harfbuzz/uharfbuzz/blob/main/src/uharfbuzz/_harfbuzz.pyx#L1149
  529. """
  530. internedTables = {}
  531. self._doneWriting(internedTables, shareExtension=True)
  532. tables = []
  533. obj_list = []
  534. done = {}
  535. objidx = 0
  536. virtual_edges = []
  537. self._gatherGraphForHarfbuzz(tables, obj_list, done, objidx, virtual_edges)
  538. # Gather all data in two passes: the absolute positions of all
  539. # subtable are needed before the actual data can be assembled.
  540. pos = 0
  541. for table in tables:
  542. table.pos = pos
  543. pos = pos + table.getDataLength()
  544. data = []
  545. for table in tables:
  546. tableData = table.getDataForHarfbuzz()
  547. data.append(tableData)
  548. return hb.serialize_with_tag(str(tableTag), data, obj_list)
  549. def getAllData(self, remove_duplicate=True):
  550. """Assemble all data, including all subtables."""
  551. if remove_duplicate:
  552. internedTables = {}
  553. self._doneWriting(internedTables)
  554. tables = []
  555. extTables = []
  556. done = {}
  557. self._gatherTables(tables, extTables, done)
  558. tables.reverse()
  559. extTables.reverse()
  560. # Gather all data in two passes: the absolute positions of all
  561. # subtable are needed before the actual data can be assembled.
  562. pos = 0
  563. for table in tables:
  564. table.pos = pos
  565. pos = pos + table.getDataLength()
  566. for table in extTables:
  567. table.pos = pos
  568. pos = pos + table.getDataLength()
  569. data = []
  570. for table in tables:
  571. tableData = table.getData()
  572. data.append(tableData)
  573. for table in extTables:
  574. tableData = table.getData()
  575. data.append(tableData)
  576. return bytesjoin(data)
  577. # interface for gathering data, as used by table.compile()
  578. def getSubWriter(self):
  579. subwriter = self.__class__(self.localState, self.tableTag)
  580. subwriter.parent = (
  581. self # because some subtables have idential values, we discard
  582. )
  583. # the duplicates under the getAllData method. Hence some
  584. # subtable writers can have more than one parent writer.
  585. # But we just care about first one right now.
  586. return subwriter
  587. def writeValue(self, typecode, value):
  588. self.items.append(struct.pack(f">{typecode}", value))
  589. def writeArray(self, typecode, values):
  590. a = array.array(typecode, values)
  591. if sys.byteorder != "big":
  592. a.byteswap()
  593. self.items.append(a.tobytes())
  594. def writeInt8(self, value):
  595. assert -128 <= value < 128, value
  596. self.items.append(struct.pack(">b", value))
  597. def writeInt8Array(self, values):
  598. self.writeArray("b", values)
  599. def writeShort(self, value):
  600. assert -32768 <= value < 32768, value
  601. self.items.append(struct.pack(">h", value))
  602. def writeShortArray(self, values):
  603. self.writeArray("h", values)
  604. def writeLong(self, value):
  605. self.items.append(struct.pack(">i", value))
  606. def writeLongArray(self, values):
  607. self.writeArray("i", values)
  608. def writeUInt8(self, value):
  609. assert 0 <= value < 256, value
  610. self.items.append(struct.pack(">B", value))
  611. def writeUInt8Array(self, values):
  612. self.writeArray("B", values)
  613. def writeUShort(self, value):
  614. assert 0 <= value < 0x10000, value
  615. self.items.append(struct.pack(">H", value))
  616. def writeUShortArray(self, values):
  617. self.writeArray("H", values)
  618. def writeULong(self, value):
  619. self.items.append(struct.pack(">I", value))
  620. def writeULongArray(self, values):
  621. self.writeArray("I", values)
  622. def writeUInt24(self, value):
  623. assert 0 <= value < 0x1000000, value
  624. b = struct.pack(">L", value)
  625. self.items.append(b[1:])
  626. def writeUInt24Array(self, values):
  627. for value in values:
  628. self.writeUInt24(value)
  629. def writeTag(self, tag):
  630. tag = Tag(tag).tobytes()
  631. assert len(tag) == 4, tag
  632. self.items.append(tag)
  633. def writeSubTable(self, subWriter, offsetSize):
  634. self.items.append(OffsetToWriter(subWriter, offsetSize))
  635. def writeCountReference(self, table, name, size=2, value=None):
  636. ref = CountReference(table, name, size=size, value=value)
  637. self.items.append(ref)
  638. return ref
  639. def writeStruct(self, format, values):
  640. data = struct.pack(*(format,) + values)
  641. self.items.append(data)
  642. def writeData(self, data):
  643. self.items.append(data)
  644. def getOverflowErrorRecord(self, item):
  645. LookupListIndex = SubTableIndex = itemName = itemIndex = None
  646. if self.name == "LookupList":
  647. LookupListIndex = item.repeatIndex
  648. elif self.name == "Lookup":
  649. LookupListIndex = self.repeatIndex
  650. SubTableIndex = item.repeatIndex
  651. else:
  652. itemName = getattr(item, "name", "<none>")
  653. if hasattr(item, "repeatIndex"):
  654. itemIndex = item.repeatIndex
  655. if self.name == "SubTable":
  656. LookupListIndex = self.parent.repeatIndex
  657. SubTableIndex = self.repeatIndex
  658. elif self.name == "ExtSubTable":
  659. LookupListIndex = self.parent.parent.repeatIndex
  660. SubTableIndex = self.parent.repeatIndex
  661. else: # who knows how far below the SubTable level we are! Climb back up to the nearest subtable.
  662. itemName = ".".join([self.name, itemName])
  663. p1 = self.parent
  664. while p1 and p1.name not in ["ExtSubTable", "SubTable"]:
  665. itemName = ".".join([p1.name, itemName])
  666. p1 = p1.parent
  667. if p1:
  668. if p1.name == "ExtSubTable":
  669. LookupListIndex = p1.parent.parent.repeatIndex
  670. SubTableIndex = p1.parent.repeatIndex
  671. else:
  672. LookupListIndex = p1.parent.repeatIndex
  673. SubTableIndex = p1.repeatIndex
  674. return OverflowErrorRecord(
  675. (self.tableTag, LookupListIndex, SubTableIndex, itemName, itemIndex)
  676. )
  677. class CountReference(object):
  678. """A reference to a Count value, not a count of references."""
  679. def __init__(self, table, name, size=None, value=None):
  680. self.table = table
  681. self.name = name
  682. self.size = size
  683. if value is not None:
  684. self.setValue(value)
  685. def setValue(self, value):
  686. table = self.table
  687. name = self.name
  688. if table[name] is None:
  689. table[name] = value
  690. else:
  691. assert table[name] == value, (name, table[name], value)
  692. def getValue(self):
  693. return self.table[self.name]
  694. def getCountData(self):
  695. v = self.table[self.name]
  696. if v is None:
  697. v = 0
  698. return {1: packUInt8, 2: packUShort, 4: packULong}[self.size](v)
  699. def packUInt8(value):
  700. return struct.pack(">B", value)
  701. def packUShort(value):
  702. return struct.pack(">H", value)
  703. def packULong(value):
  704. assert 0 <= value < 0x100000000, value
  705. return struct.pack(">I", value)
  706. def packUInt24(value):
  707. assert 0 <= value < 0x1000000, value
  708. return struct.pack(">I", value)[1:]
  709. class BaseTable(object):
  710. """Generic base class for all OpenType (sub)tables."""
  711. def __getattr__(self, attr):
  712. reader = self.__dict__.get("reader")
  713. if reader:
  714. del self.reader
  715. font = self.font
  716. del self.font
  717. self.decompile(reader, font)
  718. return getattr(self, attr)
  719. raise AttributeError(attr)
  720. def ensureDecompiled(self, recurse=False):
  721. reader = self.__dict__.get("reader")
  722. if reader:
  723. del self.reader
  724. font = self.font
  725. del self.font
  726. self.decompile(reader, font)
  727. if recurse:
  728. for subtable in self.iterSubTables():
  729. subtable.value.ensureDecompiled(recurse)
  730. def __getstate__(self):
  731. # before copying/pickling 'lazy' objects, make a shallow copy of OTTableReader
  732. # https://github.com/fonttools/fonttools/issues/2965
  733. if "reader" in self.__dict__:
  734. state = self.__dict__.copy()
  735. state["reader"] = self.__dict__["reader"].copy()
  736. return state
  737. return self.__dict__
  738. @classmethod
  739. def getRecordSize(cls, reader):
  740. totalSize = 0
  741. for conv in cls.converters:
  742. size = conv.getRecordSize(reader)
  743. if size is NotImplemented:
  744. return NotImplemented
  745. countValue = 1
  746. if conv.repeat:
  747. if conv.repeat in reader:
  748. countValue = reader[conv.repeat] + conv.aux
  749. else:
  750. return NotImplemented
  751. totalSize += size * countValue
  752. return totalSize
  753. def getConverters(self):
  754. return self.converters
  755. def getConverterByName(self, name):
  756. return self.convertersByName[name]
  757. def populateDefaults(self, propagator=None):
  758. for conv in self.getConverters():
  759. if conv.repeat:
  760. if not hasattr(self, conv.name):
  761. setattr(self, conv.name, [])
  762. countValue = len(getattr(self, conv.name)) - conv.aux
  763. try:
  764. count_conv = self.getConverterByName(conv.repeat)
  765. setattr(self, conv.repeat, countValue)
  766. except KeyError:
  767. # conv.repeat is a propagated count
  768. if propagator and conv.repeat in propagator:
  769. propagator[conv.repeat].setValue(countValue)
  770. else:
  771. if conv.aux and not eval(conv.aux, None, self.__dict__):
  772. continue
  773. if hasattr(self, conv.name):
  774. continue # Warn if it should NOT be present?!
  775. if hasattr(conv, "writeNullOffset"):
  776. setattr(self, conv.name, None) # Warn?
  777. # elif not conv.isCount:
  778. # # Warn?
  779. # pass
  780. if hasattr(conv, "DEFAULT"):
  781. # OptionalValue converters (e.g. VarIndex)
  782. setattr(self, conv.name, conv.DEFAULT)
  783. def decompile(self, reader, font):
  784. self.readFormat(reader)
  785. table = {}
  786. self.__rawTable = table # for debugging
  787. for conv in self.getConverters():
  788. if conv.name == "SubTable":
  789. conv = conv.getConverter(reader.tableTag, table["LookupType"])
  790. if conv.name == "ExtSubTable":
  791. conv = conv.getConverter(reader.tableTag, table["ExtensionLookupType"])
  792. if conv.name == "FeatureParams":
  793. conv = conv.getConverter(reader["FeatureTag"])
  794. if conv.name == "SubStruct":
  795. conv = conv.getConverter(reader.tableTag, table["MorphType"])
  796. try:
  797. if conv.repeat:
  798. if isinstance(conv.repeat, int):
  799. countValue = conv.repeat
  800. elif conv.repeat in table:
  801. countValue = table[conv.repeat]
  802. else:
  803. # conv.repeat is a propagated count
  804. countValue = reader[conv.repeat]
  805. countValue += conv.aux
  806. table[conv.name] = conv.readArray(reader, font, table, countValue)
  807. else:
  808. if conv.aux and not eval(conv.aux, None, table):
  809. continue
  810. table[conv.name] = conv.read(reader, font, table)
  811. if conv.isPropagated:
  812. reader[conv.name] = table[conv.name]
  813. except Exception as e:
  814. name = conv.name
  815. e.args = e.args + (name,)
  816. raise
  817. if hasattr(self, "postRead"):
  818. self.postRead(table, font)
  819. else:
  820. self.__dict__.update(table)
  821. del self.__rawTable # succeeded, get rid of debugging info
  822. def compile(self, writer, font):
  823. self.ensureDecompiled()
  824. # TODO Following hack to be removed by rewriting how FormatSwitching tables
  825. # are handled.
  826. # https://github.com/fonttools/fonttools/pull/2238#issuecomment-805192631
  827. if hasattr(self, "preWrite"):
  828. deleteFormat = not hasattr(self, "Format")
  829. table = self.preWrite(font)
  830. deleteFormat = deleteFormat and hasattr(self, "Format")
  831. else:
  832. deleteFormat = False
  833. table = self.__dict__.copy()
  834. # some count references may have been initialized in a custom preWrite; we set
  835. # these in the writer's state beforehand (instead of sequentially) so they will
  836. # be propagated to all nested subtables even if the count appears in the current
  837. # table only *after* the offset to the subtable that it is counting.
  838. for conv in self.getConverters():
  839. if conv.isCount and conv.isPropagated:
  840. value = table.get(conv.name)
  841. if isinstance(value, CountReference):
  842. writer[conv.name] = value
  843. if hasattr(self, "sortCoverageLast"):
  844. writer.sortCoverageLast = 1
  845. if hasattr(self, "DontShare"):
  846. writer.DontShare = True
  847. if hasattr(self.__class__, "LookupType"):
  848. writer["LookupType"].setValue(self.__class__.LookupType)
  849. self.writeFormat(writer)
  850. for conv in self.getConverters():
  851. value = table.get(
  852. conv.name
  853. ) # TODO Handle defaults instead of defaulting to None!
  854. if conv.repeat:
  855. if value is None:
  856. value = []
  857. countValue = len(value) - conv.aux
  858. if isinstance(conv.repeat, int):
  859. assert len(value) == conv.repeat, "expected %d values, got %d" % (
  860. conv.repeat,
  861. len(value),
  862. )
  863. elif conv.repeat in table:
  864. CountReference(table, conv.repeat, value=countValue)
  865. else:
  866. # conv.repeat is a propagated count
  867. writer[conv.repeat].setValue(countValue)
  868. try:
  869. conv.writeArray(writer, font, table, value)
  870. except Exception as e:
  871. e.args = e.args + (conv.name + "[]",)
  872. raise
  873. elif conv.isCount:
  874. # Special-case Count values.
  875. # Assumption: a Count field will *always* precede
  876. # the actual array(s).
  877. # We need a default value, as it may be set later by a nested
  878. # table. We will later store it here.
  879. # We add a reference: by the time the data is assembled
  880. # the Count value will be filled in.
  881. # We ignore the current count value since it will be recomputed,
  882. # unless it's a CountReference that was already initialized in a custom preWrite.
  883. if isinstance(value, CountReference):
  884. ref = value
  885. ref.size = conv.staticSize
  886. writer.writeData(ref)
  887. table[conv.name] = ref.getValue()
  888. else:
  889. ref = writer.writeCountReference(table, conv.name, conv.staticSize)
  890. table[conv.name] = None
  891. if conv.isPropagated:
  892. writer[conv.name] = ref
  893. elif conv.isLookupType:
  894. # We make sure that subtables have the same lookup type,
  895. # and that the type is the same as the one set on the
  896. # Lookup object, if any is set.
  897. if conv.name not in table:
  898. table[conv.name] = None
  899. ref = writer.writeCountReference(
  900. table, conv.name, conv.staticSize, table[conv.name]
  901. )
  902. writer["LookupType"] = ref
  903. else:
  904. if conv.aux and not eval(conv.aux, None, table):
  905. continue
  906. try:
  907. conv.write(writer, font, table, value)
  908. except Exception as e:
  909. name = value.__class__.__name__ if value is not None else conv.name
  910. e.args = e.args + (name,)
  911. raise
  912. if conv.isPropagated:
  913. writer[conv.name] = value
  914. if deleteFormat:
  915. del self.Format
  916. def readFormat(self, reader):
  917. pass
  918. def writeFormat(self, writer):
  919. pass
  920. def toXML(self, xmlWriter, font, attrs=None, name=None):
  921. tableName = name if name else self.__class__.__name__
  922. if attrs is None:
  923. attrs = []
  924. if hasattr(self, "Format"):
  925. attrs = attrs + [("Format", self.Format)]
  926. xmlWriter.begintag(tableName, attrs)
  927. xmlWriter.newline()
  928. self.toXML2(xmlWriter, font)
  929. xmlWriter.endtag(tableName)
  930. xmlWriter.newline()
  931. def toXML2(self, xmlWriter, font):
  932. # Simpler variant of toXML, *only* for the top level tables (like GPOS, GSUB).
  933. # This is because in TTX our parent writes our main tag, and in otBase.py we
  934. # do it ourselves. I think I'm getting schizophrenic...
  935. for conv in self.getConverters():
  936. if conv.repeat:
  937. value = getattr(self, conv.name, [])
  938. for i, item in enumerate(value):
  939. conv.xmlWrite(xmlWriter, font, item, conv.name, [("index", i)])
  940. else:
  941. if conv.aux and not eval(conv.aux, None, vars(self)):
  942. continue
  943. value = getattr(
  944. self, conv.name, None
  945. ) # TODO Handle defaults instead of defaulting to None!
  946. conv.xmlWrite(xmlWriter, font, value, conv.name, [])
  947. def fromXML(self, name, attrs, content, font):
  948. try:
  949. conv = self.getConverterByName(name)
  950. except KeyError:
  951. raise # XXX on KeyError, raise nice error
  952. value = conv.xmlRead(attrs, content, font)
  953. # Some manually-written tables have a conv.repeat of ""
  954. # to represent lists. Hence comparing to None here to
  955. # allow those lists to be read correctly from XML.
  956. if conv.repeat is not None:
  957. seq = getattr(self, conv.name, None)
  958. if seq is None:
  959. seq = []
  960. setattr(self, conv.name, seq)
  961. seq.append(value)
  962. else:
  963. setattr(self, conv.name, value)
  964. def __ne__(self, other):
  965. result = self.__eq__(other)
  966. return result if result is NotImplemented else not result
  967. def __eq__(self, other):
  968. if type(self) != type(other):
  969. return NotImplemented
  970. self.ensureDecompiled()
  971. other.ensureDecompiled()
  972. return self.__dict__ == other.__dict__
  973. class SubTableEntry(NamedTuple):
  974. """See BaseTable.iterSubTables()"""
  975. name: str
  976. value: "BaseTable"
  977. index: Optional[int] = None # index into given array, None for single values
  978. def iterSubTables(self) -> Iterator[SubTableEntry]:
  979. """Yield (name, value, index) namedtuples for all subtables of current table.
  980. A sub-table is an instance of BaseTable (or subclass thereof) that is a child
  981. of self, the current parent table.
  982. The tuples also contain the attribute name (str) of the of parent table to get
  983. a subtable, and optionally, for lists of subtables (i.e. attributes associated
  984. with a converter that has a 'repeat'), an index into the list containing the
  985. given subtable value.
  986. This method can be useful to traverse trees of otTables.
  987. """
  988. for conv in self.getConverters():
  989. name = conv.name
  990. value = getattr(self, name, None)
  991. if value is None:
  992. continue
  993. if isinstance(value, BaseTable):
  994. yield self.SubTableEntry(name, value)
  995. elif isinstance(value, list):
  996. yield from (
  997. self.SubTableEntry(name, v, index=i)
  998. for i, v in enumerate(value)
  999. if isinstance(v, BaseTable)
  1000. )
  1001. # instance (not @class)method for consistency with FormatSwitchingBaseTable
  1002. def getVariableAttrs(self):
  1003. return getVariableAttrs(self.__class__)
  1004. class FormatSwitchingBaseTable(BaseTable):
  1005. """Minor specialization of BaseTable, for tables that have multiple
  1006. formats, eg. CoverageFormat1 vs. CoverageFormat2."""
  1007. @classmethod
  1008. def getRecordSize(cls, reader):
  1009. return NotImplemented
  1010. def getConverters(self):
  1011. try:
  1012. fmt = self.Format
  1013. except AttributeError:
  1014. # some FormatSwitchingBaseTables (e.g. Coverage) no longer have 'Format'
  1015. # attribute after fully decompiled, only gain one in preWrite before being
  1016. # recompiled. In the decompiled state, these hand-coded classes defined in
  1017. # otTables.py lose their format-specific nature and gain more high-level
  1018. # attributes that are not tied to converters.
  1019. return []
  1020. return self.converters.get(self.Format, [])
  1021. def getConverterByName(self, name):
  1022. return self.convertersByName[self.Format][name]
  1023. def readFormat(self, reader):
  1024. self.Format = reader.readUShort()
  1025. def writeFormat(self, writer):
  1026. writer.writeUShort(self.Format)
  1027. def toXML(self, xmlWriter, font, attrs=None, name=None):
  1028. BaseTable.toXML(self, xmlWriter, font, attrs, name)
  1029. def getVariableAttrs(self):
  1030. return getVariableAttrs(self.__class__, self.Format)
  1031. class UInt8FormatSwitchingBaseTable(FormatSwitchingBaseTable):
  1032. def readFormat(self, reader):
  1033. self.Format = reader.readUInt8()
  1034. def writeFormat(self, writer):
  1035. writer.writeUInt8(self.Format)
  1036. formatSwitchingBaseTables = {
  1037. "uint16": FormatSwitchingBaseTable,
  1038. "uint8": UInt8FormatSwitchingBaseTable,
  1039. }
  1040. def getFormatSwitchingBaseTableClass(formatType):
  1041. try:
  1042. return formatSwitchingBaseTables[formatType]
  1043. except KeyError:
  1044. raise TypeError(f"Unsupported format type: {formatType!r}")
  1045. # memoize since these are parsed from otData.py, thus stay constant
  1046. @lru_cache()
  1047. def getVariableAttrs(cls: BaseTable, fmt: Optional[int] = None) -> Tuple[str]:
  1048. """Return sequence of variable table field names (can be empty).
  1049. Attributes are deemed "variable" when their otData.py's description contain
  1050. 'VarIndexBase + {offset}', e.g. COLRv1 PaintVar* tables.
  1051. """
  1052. if not issubclass(cls, BaseTable):
  1053. raise TypeError(cls)
  1054. if issubclass(cls, FormatSwitchingBaseTable):
  1055. if fmt is None:
  1056. raise TypeError(f"'fmt' is required for format-switching {cls.__name__}")
  1057. converters = cls.convertersByName[fmt]
  1058. else:
  1059. converters = cls.convertersByName
  1060. # assume if no 'VarIndexBase' field is present, table has no variable fields
  1061. if "VarIndexBase" not in converters:
  1062. return ()
  1063. varAttrs = {}
  1064. for name, conv in converters.items():
  1065. offset = conv.getVarIndexOffset()
  1066. if offset is not None:
  1067. varAttrs[name] = offset
  1068. return tuple(sorted(varAttrs, key=varAttrs.__getitem__))
  1069. #
  1070. # Support for ValueRecords
  1071. #
  1072. # This data type is so different from all other OpenType data types that
  1073. # it requires quite a bit of code for itself. It even has special support
  1074. # in OTTableReader and OTTableWriter...
  1075. #
  1076. valueRecordFormat = [
  1077. # Mask Name isDevice signed
  1078. (0x0001, "XPlacement", 0, 1),
  1079. (0x0002, "YPlacement", 0, 1),
  1080. (0x0004, "XAdvance", 0, 1),
  1081. (0x0008, "YAdvance", 0, 1),
  1082. (0x0010, "XPlaDevice", 1, 0),
  1083. (0x0020, "YPlaDevice", 1, 0),
  1084. (0x0040, "XAdvDevice", 1, 0),
  1085. (0x0080, "YAdvDevice", 1, 0),
  1086. # reserved:
  1087. (0x0100, "Reserved1", 0, 0),
  1088. (0x0200, "Reserved2", 0, 0),
  1089. (0x0400, "Reserved3", 0, 0),
  1090. (0x0800, "Reserved4", 0, 0),
  1091. (0x1000, "Reserved5", 0, 0),
  1092. (0x2000, "Reserved6", 0, 0),
  1093. (0x4000, "Reserved7", 0, 0),
  1094. (0x8000, "Reserved8", 0, 0),
  1095. ]
  1096. def _buildDict():
  1097. d = {}
  1098. for mask, name, isDevice, signed in valueRecordFormat:
  1099. d[name] = mask, isDevice, signed
  1100. return d
  1101. valueRecordFormatDict = _buildDict()
  1102. class ValueRecordFactory(object):
  1103. """Given a format code, this object convert ValueRecords."""
  1104. def __init__(self, valueFormat):
  1105. format = []
  1106. for mask, name, isDevice, signed in valueRecordFormat:
  1107. if valueFormat & mask:
  1108. format.append((name, isDevice, signed))
  1109. self.format = format
  1110. def __len__(self):
  1111. return len(self.format)
  1112. def readValueRecord(self, reader, font):
  1113. format = self.format
  1114. if not format:
  1115. return None
  1116. valueRecord = ValueRecord()
  1117. for name, isDevice, signed in format:
  1118. if signed:
  1119. value = reader.readShort()
  1120. else:
  1121. value = reader.readUShort()
  1122. if isDevice:
  1123. if value:
  1124. from . import otTables
  1125. subReader = reader.getSubReader(value)
  1126. value = getattr(otTables, name)()
  1127. value.decompile(subReader, font)
  1128. else:
  1129. value = None
  1130. setattr(valueRecord, name, value)
  1131. return valueRecord
  1132. def writeValueRecord(self, writer, font, valueRecord):
  1133. for name, isDevice, signed in self.format:
  1134. value = getattr(valueRecord, name, 0)
  1135. if isDevice:
  1136. if value:
  1137. subWriter = writer.getSubWriter()
  1138. writer.writeSubTable(subWriter, offsetSize=2)
  1139. value.compile(subWriter, font)
  1140. else:
  1141. writer.writeUShort(0)
  1142. elif signed:
  1143. writer.writeShort(value)
  1144. else:
  1145. writer.writeUShort(value)
  1146. class ValueRecord(object):
  1147. # see ValueRecordFactory
  1148. def __init__(self, valueFormat=None, src=None):
  1149. if valueFormat is not None:
  1150. for mask, name, isDevice, signed in valueRecordFormat:
  1151. if valueFormat & mask:
  1152. setattr(self, name, None if isDevice else 0)
  1153. if src is not None:
  1154. for key, val in src.__dict__.items():
  1155. if not hasattr(self, key):
  1156. continue
  1157. setattr(self, key, val)
  1158. elif src is not None:
  1159. self.__dict__ = src.__dict__.copy()
  1160. def getFormat(self):
  1161. format = 0
  1162. for name in self.__dict__.keys():
  1163. format = format | valueRecordFormatDict[name][0]
  1164. return format
  1165. def getEffectiveFormat(self):
  1166. format = 0
  1167. for name, value in self.__dict__.items():
  1168. if value:
  1169. format = format | valueRecordFormatDict[name][0]
  1170. return format
  1171. def toXML(self, xmlWriter, font, valueName, attrs=None):
  1172. if attrs is None:
  1173. simpleItems = []
  1174. else:
  1175. simpleItems = list(attrs)
  1176. for mask, name, isDevice, format in valueRecordFormat[:4]: # "simple" values
  1177. if hasattr(self, name):
  1178. simpleItems.append((name, getattr(self, name)))
  1179. deviceItems = []
  1180. for mask, name, isDevice, format in valueRecordFormat[4:8]: # device records
  1181. if hasattr(self, name):
  1182. device = getattr(self, name)
  1183. if device is not None:
  1184. deviceItems.append((name, device))
  1185. if deviceItems:
  1186. xmlWriter.begintag(valueName, simpleItems)
  1187. xmlWriter.newline()
  1188. for name, deviceRecord in deviceItems:
  1189. if deviceRecord is not None:
  1190. deviceRecord.toXML(xmlWriter, font, name=name)
  1191. xmlWriter.endtag(valueName)
  1192. xmlWriter.newline()
  1193. else:
  1194. xmlWriter.simpletag(valueName, simpleItems)
  1195. xmlWriter.newline()
  1196. def fromXML(self, name, attrs, content, font):
  1197. from . import otTables
  1198. for k, v in attrs.items():
  1199. setattr(self, k, int(v))
  1200. for element in content:
  1201. if not isinstance(element, tuple):
  1202. continue
  1203. name, attrs, content = element
  1204. value = getattr(otTables, name)()
  1205. for elem2 in content:
  1206. if not isinstance(elem2, tuple):
  1207. continue
  1208. name2, attrs2, content2 = elem2
  1209. value.fromXML(name2, attrs2, content2, font)
  1210. setattr(self, name, value)
  1211. def __ne__(self, other):
  1212. result = self.__eq__(other)
  1213. return result if result is NotImplemented else not result
  1214. def __eq__(self, other):
  1215. if type(self) != type(other):
  1216. return NotImplemented
  1217. return self.__dict__ == other.__dict__