feat(propose): the PDF shipped a structure index and the build discarded it unopened
`okf build` recovers a PDF's boundaries from the shape of its page text and never opens the file's own `/Outlines` bookmark tree. On a 701-page process code whose publisher also ships a NISO-STS structure for it, measured outside this repository and reproduced here exactly: the shipped default finds 1967 of 2761 titled sections, 0 of its 28 chapters, and 794 of 794 misses have their heading text PRESENT in the extracted text. The line was read; the boundary was never opened. The same file's bookmark tree matches 2761 of 2761 of those titles exactly after normalisation. `--pdf-outline`, OFF, cuts a PDF at the boundaries its tree declares. boundaries 1967 of 2761 -> 2759 of 2761 (gate was 2700) depth 1 0 of 28 -> 28 of 28 titles identical to source -- -> 2761 of 2761 false positives 163 of 2182 -> 3 of 2762 directories with two files 132 of 2050 -> 2 of 2738 front-matter concepts 72 of 2182 -> 2 of 2762 consumption fasit present 4 of 7 -> 7 of 7 hit@1 / hit@8 / hit@50 1/6 2/6 4/6 -> 3/6 5/6 6/6 It is a SEGMENTATION arm, not a reader option: the extracted text is byte for byte the same either way. A PDF with no tree builds byte-identically with the flag on -- `diff -r` empty across the pre-change tree, the arm off and the arm on. An unresolvable `/Dest` is dropped and COUNTED, never fabricated into a boundary and never a refusal of the file. The bridge from (page, y) to a line index is the whole risk, so both routes are measured. `extract_text_lines` splits lines identically to `extract_text` on 701 of 701 pages, and is CHECKED per page rather than assumed. The y route and the title route disagree on 0 of 2762 nodes, flat from a 0pt tolerance to 8pt and collapsing at 12pt, so the rule ships with no tolerance constant. The naive "nearest line" rule was wrong on 1840 of 2762, one line early every time. The orphan check is not applied to a bookmark mark: it asks whether anything stands under a candidate's first line, which is the right question for a heuristic's guess and the wrong one for a publisher's declaration. 683 of 2762 marks are container sections; applying it scores 2079 instead of 2759. No new dependency and no second parse of the pages: `pdfminer.six` already ships under `pdfplumber` in `[extract]`. 119.22s -> 183.31s wall, peak RSS 3252 -> 3251 MiB. The default does not move; 1 of the 8 reference PDFs carries a usable tree at all. `.pdf` also gains its `_EVIDENCE` row, as `measured` -- it was the row with the most measurement behind it and no entry in the table. Report: docs/2026-09-10-k3-runde12-pdf-outlines.md Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
parent
8e1917946a
commit
e1f4faa098
12 changed files with 1261 additions and 15 deletions
|
|
@ -106,7 +106,18 @@ _PANDOC_FORMATS: dict[str, str] = {
|
|||
# the honesty limit that travels with it: the 828 files are ONE product in ONE
|
||||
# format from ONE publisher, and the file boundaries and `<h1>`s are a
|
||||
# generator's cut of that document, not 828 documents anyone wrote.
|
||||
#
|
||||
# `.pdf` JOINED THE TABLE 2026-09-10, as `measured`, and it enters on the
|
||||
# strongest evidence of any row here: eight real corpus PDFs with a fasit the
|
||||
# operator hand-counted document by document, plus a 701-page process code
|
||||
# whose PUBLISHER also ships a NISO-STS structure for it -- 2 761 titled
|
||||
# sections, written for their own purposes and long before any lookup of ours.
|
||||
# The honesty limit that travels with it: those 2 761 rows are ONE product in
|
||||
# ONE format from ONE publisher, its structure is a strict numbered hierarchy
|
||||
# on 2 739 of 2 761 titles, and a running prose document would measure
|
||||
# something else entirely.
|
||||
_EVIDENCE: dict[str, str] = {
|
||||
".pdf": "measured",
|
||||
".docx": "measured",
|
||||
".xlsx": "measured",
|
||||
".pptx": "constructed",
|
||||
|
|
@ -638,6 +649,202 @@ def _pdf_pages(
|
|||
return tuple((number, page) for number, page in enumerate(pages, start=1) if page)
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class OutlineMark:
|
||||
"""One `/Outlines` node, placed on a LINE of the string `extract_text` returns.
|
||||
|
||||
`level` is what the TREE declares, not a distance normalised against
|
||||
anything: a document whose outline carries its own root node puts its
|
||||
chapters at level 2, and rewriting that here would state a structure the
|
||||
publisher did not. Measured on a 701-page process code -- the tree's levels
|
||||
2..8 hold 28/118/500/1141/872/93/9 nodes against the publisher's own
|
||||
NISO-STS depths 1..7 at 28/118/500/1141/868/97/9, so the mapping is level
|
||||
minus one on five rows and the publisher disagrees with the publisher on
|
||||
four nodes. That disagreement is data, and it survives only if the level is
|
||||
reported rather than fixed up.
|
||||
"""
|
||||
|
||||
line: int
|
||||
level: int
|
||||
title: str
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class PdfOutline:
|
||||
"""The bookmark tree, bridged onto lines -- with what did not bridge counted.
|
||||
|
||||
`unresolved` is not decoration. A `/Dest` that names an object which is not
|
||||
a page, or a page that produced no text, has to be DROPPED: fabricating a
|
||||
boundary from it would put a heading somewhere the document never had one,
|
||||
and raising would refuse a file over a defect in one of its bookmarks.
|
||||
Dropping silently is the third option this library refuses everywhere else,
|
||||
so the count is part of the return value.
|
||||
"""
|
||||
|
||||
marks: tuple[OutlineMark, ...]
|
||||
unresolved: int
|
||||
|
||||
|
||||
def _outline_page_and_top(doc: object, dest: object, action: object) -> tuple[object, float | None]:
|
||||
"""`(page reference, /XYZ top)` from a bookmark's destination, or `(None, None)`.
|
||||
|
||||
Four shapes reach here and all four are in the wild: an explicit array, a
|
||||
NAMED destination resolved through the document's name tree, a `GoTo`
|
||||
action carrying either, and an indirect reference to any of them.
|
||||
"""
|
||||
target = dest
|
||||
if target is None and action is not None:
|
||||
resolved = action.resolve() if hasattr(action, "resolve") else action
|
||||
if isinstance(resolved, dict):
|
||||
target = resolved.get("D")
|
||||
if isinstance(target, (bytes, str)) or hasattr(target, "name"):
|
||||
name = target.name if hasattr(target, "name") else target
|
||||
try:
|
||||
target = doc.get_dest(name) # type: ignore[attr-defined]
|
||||
except Exception:
|
||||
return (None, None)
|
||||
if hasattr(target, "resolve"):
|
||||
try:
|
||||
target = target.resolve()
|
||||
except Exception:
|
||||
return (None, None)
|
||||
if isinstance(target, dict):
|
||||
target = target.get("D")
|
||||
if not isinstance(target, list) or not target:
|
||||
return (None, None)
|
||||
top: float | None = None
|
||||
if len(target) > 3 and getattr(target[1], "name", None) == "XYZ":
|
||||
candidate = target[3]
|
||||
if isinstance(candidate, (int, float)):
|
||||
top = float(candidate)
|
||||
return (target[0], top)
|
||||
|
||||
|
||||
def pdf_outline(
|
||||
name: str, data: bytes, *, pdf_headings: bool = False, ocr: bool = False
|
||||
) -> PdfOutline:
|
||||
"""`pdf`: the file's own `/Outlines` tree, as marks on the extracted text.
|
||||
|
||||
THE BRIDGE IS THE WHOLE PROBLEM, and both routes are measured rather than
|
||||
argued. A bookmark states a PAGE and a y position; a candidate needs a LINE
|
||||
index. On the 701-page document this was built against, 2 706 of 2 761
|
||||
bookmarks share a destination page with another bookmark, so the page alone
|
||||
is never a cut point.
|
||||
|
||||
Y ROUTE (primary). `page.extract_text_lines()` carries a `top` per line,
|
||||
and the mark takes the FIRST line at or below the destination. It needs
|
||||
the line splitting to be the one `page.extract_text()` produced -- an
|
||||
assumption, so it is CHECKED per page and the route is used only where
|
||||
the two strings are identical. Measured: 701 of 701 pages, and the
|
||||
resulting index agrees with the title route on 2 762 of 2 762 nodes,
|
||||
flat from a 0 pt tolerance to 8 pt and collapsing at 12 (the line
|
||||
spacing). It therefore ships with NO tolerance constant at all.
|
||||
|
||||
TITLE ROUTE (fallback). The bookmark's title, normalised, searched in the
|
||||
destination page's own lines. It resolved 2 762 of 2 763 on that
|
||||
document, and its weakness is real: a title like `Armering` occurs nine
|
||||
times in that structure, so it is scoped to the destination page and is
|
||||
never asked a question the y route already answered.
|
||||
|
||||
`pdf_headings` and `ocr` are passed through so the line indices address the
|
||||
SAME rendering the caller extracted. They are not options of this arm: a
|
||||
plan indexes one exact string, and marks computed against another one point
|
||||
at the right words in the wrong places.
|
||||
"""
|
||||
if Path(name).suffix.lower() != ".pdf":
|
||||
return PdfOutline((), 0)
|
||||
try:
|
||||
import pdfplumber
|
||||
except ImportError as exc:
|
||||
raise _extra_missing(".pdf") from exc
|
||||
from pdfminer.pdfdocument import PDFNoOutlines
|
||||
from pdfminer.pdfpage import PDFPage
|
||||
|
||||
rendered = _pdf_pages(data, pdf_headings, ocr)
|
||||
starts: dict[int, int] = {}
|
||||
page_lines: dict[int, list[str]] = {}
|
||||
offset = 0
|
||||
for number, page_text in rendered:
|
||||
starts[number] = offset
|
||||
page_lines[number] = page_text.split("\n")
|
||||
offset += len(page_lines[number]) + 1
|
||||
|
||||
unresolved = 0
|
||||
placed: dict[int, OutlineMark] = {}
|
||||
with pdfplumber.open(io.BytesIO(data)) as pdf:
|
||||
try:
|
||||
nodes = list(pdf.doc.get_outlines())
|
||||
except PDFNoOutlines:
|
||||
# NOT an error, and not zero concepts either: this file simply
|
||||
# carries no index, which is the common case and the one the
|
||||
# byte-identical guarantee below rests on.
|
||||
return PdfOutline((), 0)
|
||||
except Exception as exc:
|
||||
raise ExtractionError(
|
||||
f"the PDF parser failed reading /Outlines: {exc}",
|
||||
code="extractor_pdf_error",
|
||||
) from exc
|
||||
numbers = {
|
||||
page.pageid: index + 1 for index, page in enumerate(PDFPage.create_pages(pdf.doc))
|
||||
}
|
||||
wanted: dict[int, list[tuple[int, str, float | None]]] = {}
|
||||
for level, title, dest, action, _ in nodes:
|
||||
reference, top = _outline_page_and_top(pdf.doc, dest, action)
|
||||
page_number = numbers.get(getattr(reference, "objid", None))
|
||||
if page_number is None or page_number not in starts:
|
||||
unresolved += 1
|
||||
continue
|
||||
wanted.setdefault(page_number, []).append((int(level), str(title), top))
|
||||
# Geometry is read only for the pages that carry a bookmark, because
|
||||
# `extract_text_lines` costs a second render of every page it is asked
|
||||
# about -- 78 s over 701 pages, and nothing at all over the pages no
|
||||
# bookmark points at.
|
||||
for page in pdf.pages:
|
||||
number = page.page_number
|
||||
group = wanted.get(number)
|
||||
if not group:
|
||||
continue
|
||||
lines = page_lines[number]
|
||||
tops: list[float] | None = None
|
||||
geometry = page.extract_text_lines()
|
||||
if [str(entry["text"]) for entry in geometry] == lines:
|
||||
tops = [float(entry["top"]) for entry in geometry]
|
||||
height = float(page.height)
|
||||
for level, title, top in group:
|
||||
index: int | None = None
|
||||
if tops is not None and top is not None:
|
||||
want = height - top
|
||||
index = next(
|
||||
(position for position, value in enumerate(tops) if value >= want),
|
||||
len(tops) - 1,
|
||||
)
|
||||
else:
|
||||
target = _normalise_outline(title)
|
||||
joined = ""
|
||||
bounds: list[int] = []
|
||||
for rendered_line in lines:
|
||||
bounds.append(len(joined))
|
||||
joined += _normalise_outline(rendered_line)
|
||||
found = joined.find(target)
|
||||
if found >= 0:
|
||||
index = max(position for position, at in enumerate(bounds) if at <= found)
|
||||
if index is None:
|
||||
unresolved += 1
|
||||
continue
|
||||
at = starts[number] + index
|
||||
# FIRST in tree order wins a shared line. Two marks on one line
|
||||
# would give the second an empty span, and the orphan check
|
||||
# deletes an empty span silently -- the same trap the proposer
|
||||
# documents for a second-pass candidate list.
|
||||
placed.setdefault(at, OutlineMark(line=at, level=level, title=title))
|
||||
return PdfOutline(tuple(placed[at] for at in sorted(placed)), unresolved)
|
||||
|
||||
|
||||
def _normalise_outline(value: str) -> str:
|
||||
"""Whitespace out, case folded -- the form the title route compares on."""
|
||||
return re.sub(r"\s+", "", value).lower()
|
||||
|
||||
|
||||
def _extract_pdf(data: bytes, *, headings: bool = False, ocr: bool = False) -> str:
|
||||
"""`pdf`: page text via `pdfplumber`, in page order, pages separated by a
|
||||
blank line.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue