Spaces:
Sleeping
Sleeping
File size: 1,775 Bytes
754d92a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | from collections.abc import Iterable
from typing import Any
from babeldoc.pdfminer.pdfparser import PDFSyntaxError
from babeldoc.pdfminer.pdftypes import dict_value
from babeldoc.pdfminer.pdftypes import int_value
from babeldoc.pdfminer.pdftypes import list_value
from babeldoc.pdfminer.utils import choplist
from babeldoc.pdfminer import settings
class NumberTree:
"""A PDF number tree.
See Section 3.8.6 of the PDF Reference.
"""
def __init__(self, obj: Any):
self._obj = dict_value(obj)
self.nums: Iterable[Any] | None = None
self.kids: Iterable[Any] | None = None
self.limits: Iterable[Any] | None = None
if "Nums" in self._obj:
self.nums = list_value(self._obj["Nums"])
if "Kids" in self._obj:
self.kids = list_value(self._obj["Kids"])
if "Limits" in self._obj:
self.limits = list_value(self._obj["Limits"])
def _parse(self) -> list[tuple[int, Any]]:
items = []
if self.nums: # Leaf node
for k, v in choplist(2, self.nums):
items.append((int_value(k), v))
if self.kids: # Root or intermediate node
for child_ref in self.kids:
items += NumberTree(child_ref)._parse()
return items
values: list[tuple[int, Any]] # workaround decorators unsupported by mypy
@property # type: ignore[no-redef,misc]
def values(self) -> list[tuple[int, Any]]:
values = self._parse()
if settings.STRICT:
if not all(a[0] <= b[0] for a, b in zip(values, values[1:], strict=False)):
raise PDFSyntaxError("Number tree elements are out of order")
else:
values.sort(key=lambda t: t[0])
return values
|