Skip to main content

docling_core/
document.rs

1//! The unified document representation.
2
3use crate::markdown::{to_markdown, to_markdown_images};
4use crate::ImageMode;
5
6/// The unified, format-agnostic document produced by every backend.
7///
8/// This is the heart of docling: backends parse their source format into a
9/// `DoclingDocument`, and serializers turn it back into Markdown, HTML, JSON,
10/// etc. Phase 0 uses a flat sequence of [`Node`]s; the production schema will
11/// match docling-core's body-tree-with-references layout.
12#[derive(Debug, Clone, PartialEq)]
13pub struct DoclingDocument {
14    /// Logical document name (usually the input file stem).
15    pub name: String,
16    /// Top-level content, in reading order.
17    pub nodes: Vec<Node>,
18    /// Default Markdown export mode for [`Self::export_to_markdown`]. `false`
19    /// (the default) reproduces docling's legacy output byte-for-byte; `true`
20    /// emits cleaner, more conformant Markdown. Set by `DocumentConverter`.
21    pub strict_markdown: bool,
22    /// Emit tables in the compact `| a | b |` / `| - | - |` form rather than
23    /// docling-core's width-padded GitHub serializer. The PDF backend sets this
24    /// (its committed groundtruth corpus predates the padded serializer); DOCX/HTML
25    /// leave it `false` to match current published docling.
26    pub compact_tables: bool,
27    /// Text the Markdown export inserts between two pages — docling-core's
28    /// `MarkdownParams.page_break_placeholder` (e.g. `"<!-- page break -->"`).
29    /// `None` (the default) omits page breaks from Markdown, as docling does.
30    /// Set by `DocumentConverter::page_break_placeholder`. See
31    /// [`crate::markdown`] for where a break lands (only between two rendered
32    /// blocks that sit on different pages, never leading or trailing).
33    pub page_break_placeholder: Option<String>,
34    /// Hyperlinks recovered from the source, as `(anchor_text, href)` pairs in
35    /// document order. docling's standard pipeline drops PDF link annotations, so
36    /// these are rendered as Markdown `[anchor](href)` **only in strict mode**
37    /// (legacy/docling output is left byte-for-byte unchanged). The PDF backend
38    /// populates this from pdfium link annotations; other backends leave it empty.
39    pub links: Vec<(String, String)>,
40    /// Conversion-confidence report (#183), populated by the PDF/image ML
41    /// pipeline; `None` for declarative conversions. Deliberately **not**
42    /// part of any document export (docling keeps it on the conversion
43    /// result, outside the document schema) — docling-serve surfaces it in
44    /// the HTTP response instead.
45    pub confidence: Option<crate::confidence::ConfidenceReport>,
46    /// docling's item tree, when the backend built one (the HTML backend
47    /// does): the JSON export serializes it instead of deriving a tree from
48    /// `nodes`, so the JSON carries upstream's exact parent/child structure,
49    /// item numbering, inline groups, formatting and content layers. Every
50    /// other serializer reads `nodes`. See [`crate::tree`].
51    pub tree: Option<crate::tree::ItemTree>,
52    /// Rendered page images by 1-based page number — docling's
53    /// `PageItem.image`, filled by the PDF/image pipeline only when page
54    /// images are requested (docling's `generate_page_images`, #520). The
55    /// JSON export writes each as the page's `image`, so docling-core's
56    /// `TableItem.get_image` / `FormulaItem.get_image` can crop from it.
57    /// Empty otherwise; no other export reads it.
58    pub page_images: std::collections::BTreeMap<usize, PictureImage>,
59}
60
61/// A single piece of document content.
62#[derive(Debug, Clone, PartialEq)]
63pub enum Node {
64    /// A heading. `level` is 1-6.
65    Heading { level: u8, text: String },
66    /// A run of body text.
67    Paragraph { text: String },
68    /// A form checkbox (docling's `checkbox_selected`/`checkbox_unselected`): its
69    /// clean label `text` with the checked state. DocLang emits a `<checkbox>`
70    /// element head; Markdown/JSON render the task-list form (`- [x] `/`- [ ] `).
71    CheckboxItem { checked: bool, text: String },
72    /// A single list item at the given nesting `level` (0 = top). For ordered
73    /// items, `number` is the display number (honoring the list's `start`); it
74    /// is unused for unordered items. `first_in_list` marks the first item of a
75    /// list so the serializer can blank-line-separate adjacent sibling lists.
76    ///
77    /// `marker` is the DocLang enumeration marker (`"1."`, `"1.1."`, …) when the
78    /// backend provides one — HTML and DOCX set it for enumerated items, so
79    /// DocLang emits `<ldiv><marker>…</marker></ldiv>`; Markdown and the other
80    /// declarative backends leave it `None`, giving a bare `<ldiv/>` (matching
81    /// docling, whose Markdown backend passes no marker).
82    ListItem {
83        ordered: bool,
84        number: u64,
85        first_in_list: bool,
86        text: String,
87        level: u8,
88        marker: Option<String>,
89        /// Optional layout provenance (`x0,y0,x1,y1`, normalized to 0–511): the
90        /// four DocLang `<location>` values emitted inside the `<list>` right
91        /// after the item's `<ldiv>`. Set only by backends with real geometry
92        /// (e.g. PPTX shapes); `None` for the declarative backends. Kept on the
93        /// item itself (rather than a [`Node::Located`] wrapper) so consecutive
94        /// items still group into one `<list>`.
95        location: Option<[u16; 4]>,
96        /// DocLang-only override for items whose DocLang form diverges from their
97        /// flat Markdown `text`. Markdown/JSON always render the fields above; the
98        /// DocLang serializer, when this is `Some`, takes the list kind, marker,
99        /// and content from here instead. Used for docx multilevel numbering
100        /// (Markdown shows `- 1.1. x`, DocLang an ordered `<marker>1.1.</marker>`
101        /// with clean text) and inline equations/formatting in list items.
102        dclx: Option<ListItemDclx>,
103        /// The item's hyperlink target, when its content is a link — docling's
104        /// HTML backend emits it as an `<href uri=…/>` in the item head, and the
105        /// anchor's Markdown link markup is stripped from the rendered content.
106        /// `None` for a plain item; ignored by Markdown/JSON.
107        href: Option<String>,
108        /// Non-body content layer (docling's HTML site chrome before the first
109        /// heading → `furniture`). DocLang emits a `<layer value=…/>` in the item
110        /// head; Markdown/JSON drop a non-body item entirely.
111        layer: Option<ContentLayer>,
112    },
113    /// A fenced code block.
114    Code {
115        language: Option<String>,
116        text: String,
117        /// The original (pre-enrichment) text when the CodeFormula model
118        /// rewrote `text`: docling keeps the raw extraction in the JSON `orig`
119        /// field while `text` carries the model output. `None` → `orig == text`.
120        orig: Option<String>,
121        /// A line-preserving rendering, when the backend can reconstruct one
122        /// but docling's own output for the format cannot. The PDF pipeline
123        /// sets it (docling-parse joins code lines with single spaces, so
124        /// `text` carries that flat docling-parity form): **strict** Markdown
125        /// prefers `pretty`, every byte-conformance surface (legacy Markdown,
126        /// JSON, DocLang, chunks) serializes `text`.
127        pretty: Option<String>,
128    },
129    /// A table. The first row is treated as the header.
130    Table(Table),
131    /// A picture/figure, with an optional caption and (when a backend extracts
132    /// it) the embedded image itself.
133    Picture {
134        caption: Option<String>,
135        /// Hyperlink annotation on the caption (docling's caption text item
136        /// `hyperlink`): the HTML backend sets it when an `<a href>` wraps the
137        /// image whose `alt` became the caption. DocLang emits the block-form
138        /// `<caption>` with an `<href uri=…/>` head; JSON puts `hyperlink` on
139        /// the caption item; Markdown and LaTeX print the plain caption text,
140        /// as docling does.
141        caption_href: Option<String>,
142        image: Option<PictureImage>,
143        /// DocumentPictureClassifier predictions (all classes, descending
144        /// confidence), when the picture-classification enrichment ran.
145        /// Serialized as docling's `classification` annotation + `meta` field
146        /// on the JSON picture item; Markdown/DocLang output is unaffected.
147        classification: Option<Vec<PictureClass>>,
148        /// Where the caption item hangs in the JSON tree (#390); see
149        /// [`CaptionParent`]. Markdown, DocLang and LaTeX ignore it.
150        caption_parent: CaptionParent,
151    },
152    /// A display-math formula item decoded by the CodeFormula enrichment:
153    /// `latex` is the model's LaTeX (no `$$` wrapping), `orig` the raw glyph
154    /// text extracted from the PDF. Markdown renders `$$latex$$`; JSON emits a
155    /// `formula` text item (docling's un-enriched pipeline instead emits a
156    /// placeholder paragraph — see the PDF assembler).
157    Formula {
158        latex: String,
159        orig: String,
160        location: Option<[u16; 4]>,
161    },
162    /// A standalone caption item (docling's `DocItemLabel.CAPTION` text that
163    /// no picture or table claims): the HTML backend emits a `<figure>`'s
164    /// `<figcaption>` this way when the figure produced no picture and its
165    /// first item is not a table (docling#4050). `href` is the caption's
166    /// hyperlink annotation (the first link inside the figcaption); Markdown
167    /// renders it as `[text](href)`, JSON puts `hyperlink` on the caption
168    /// item, DocLang emits the block-form `<caption>` with an `<href>` head.
169    Caption { text: String, href: Option<String> },
170    /// A chart (docling's `PictureItem` classified as a chart, carrying a
171    /// `PictureTabularChartData` annotation). Markdown and JSON render it exactly
172    /// like a [`Node::Picture`] placeholder (an `<!-- image -->` / `picture`
173    /// item); the DocLang serializer emits `<picture class="chart">` with a
174    /// `<label value="{kind}"/>` and the data `table` as a `<tabular>`.
175    Chart {
176        /// docling's classification label, e.g. `bar_chart`, `line_chart`.
177        kind: String,
178        /// The chart's data grid (row 0 is the header band).
179        table: Table,
180        /// The chart title (docling's caption item on the picture).
181        caption: Option<String>,
182        /// DocLang `<location>` provenance for the picture element.
183        location: Option<[u16; 4]>,
184    },
185    /// A logical grouping of child nodes (e.g. a list, a section, a spreadsheet
186    /// sheet). `name` is docling's group name when it differs from the label —
187    /// an xlsx sheet group is `label: "sheet"`, `name: Some("Sheet1")`. `layer`
188    /// puts the group *and* everything it contains on a non-body content layer
189    /// (a hidden sheet is `invisible`), which is how the serializers that only
190    /// render body content know to skip it; DocLang stamps each child with the
191    /// layer token, exactly as a [`Node::Furniture`] wrapper on each would.
192    /// DocLang has no group element, so a group is transparent there.
193    Group {
194        label: String,
195        name: Option<String>,
196        layer: Option<ContentLayer>,
197        children: Vec<Node>,
198    },
199    /// A form key-value region (docling's `field_region`): a set of form fields,
200    /// each pairing an optional marker, key, and value. Backends detect these
201    /// from form structure (e.g. HTML's `keyN` / `keyN_valueM` / `keyN_marker`
202    /// `id`-convention); the serializers render each item's parts as separate
203    /// labelled texts (`marker` / `field_key` / `field_value`).
204    FieldRegion { items: Vec<FieldItem> },
205    /// docling's `KeyValueItem`: a graph of key and value cells and the links
206    /// between them (docling-core's `GraphData`). The XBRL backend builds one
207    /// from an instance's numeric facts and the presentation and calculation
208    /// hierarchies of its taxonomy. Only the JSON export carries the graph:
209    /// docling's Markdown serializer has no rendering for the item and writes
210    /// its `<!-- missing-key-value-item -->` placeholder, which the Markdown
211    /// export reproduces; the other serializers omit it.
212    KeyValueGraph {
213        cells: Vec<GraphCell>,
214        links: Vec<GraphLink>,
215    },
216    /// Rich inline content — docling's `InlineGroup`: a run of styled text
217    /// segments that a backend captured with formatting (`<bold>`, `<italic>`,
218    /// `<underline>`, `<strikethrough>`, sub/superscript, inline `<code>`) the
219    /// flat Markdown text cannot represent. Markdown/JSON render this exactly
220    /// like `Paragraph { text: md_text }` (so their output is unchanged); the
221    /// DocLang serializer uses the structured `runs`. `unwrapped` is set when the
222    /// group's docling parent is a heading/text (no enclosing `<text>` wrapper).
223    InlineGroup {
224        unwrapped: bool,
225        runs: Vec<InlineRun>,
226        md_text: String,
227    },
228    /// A node in a non-body content layer — `furniture` (page headers/footers,
229    /// the HTML `<title>`, site navigation/chrome) or `notes` (docx comments).
230    /// Markdown and JSON omit these layers by default; DocLang renders the wrapped
231    /// node with a `<layer value="{layer}"/>` head.
232    Furniture {
233        layer: ContentLayer,
234        inner: Box<Node>,
235    },
236    /// One reviewer comment: docling's notes-layer `comment_section` group
237    /// holding a single text item. `name` is docling's own — `comment-{id}` for
238    /// a docx `w:comment`, `comment-{sheet}-{cell}` for a spreadsheet cell note.
239    /// JSON emits the group plus its notes-layer text; DocLang emits the flat
240    /// `<text><layer value="notes"/>…</text>` upstream writes (its DocLang
241    /// carries no group for comments); Markdown and LaTeX omit the notes layer.
242    ///
243    /// `refs_note_text` picks what a [`Node::Commented`] annotation points at,
244    /// mirroring an upstream asymmetry: docling-core's `add_comment` appends the
245    /// **note text**'s ref to each target (which is what the xlsx backend gets),
246    /// while the docx backend overwrites that with the **group**'s ref so a
247    /// comment's replies group together.
248    ///
249    /// `grouped` is whether docling wraps the note in a `comment_section`
250    /// group at all: the docx and spreadsheet backends do, while backends that
251    /// call docling-core's `add_comment` directly (Pages, #383) get a bare
252    /// notes-layer text item under the body — JSON then emits no group and
253    /// `name` is unused.
254    CommentSection {
255        name: String,
256        text: String,
257        refs_note_text: bool,
258        grouped: bool,
259    },
260    /// A body item annotated by reviewer comments: `comments` are indices into
261    /// the document's [`Node::CommentSection`] nodes, in document order. JSON
262    /// emits docling's `comments: [{"$ref": …}]` on the item, each ref pointing
263    /// where the section says (see [`Node::CommentSection::refs_note_text`]);
264    /// every other serializer renders `inner` unchanged.
265    Commented {
266        comments: Vec<usize>,
267        inner: Box<Node>,
268    },
269    /// A node carrying layout provenance — the four DocLang `<location>` values
270    /// (`x0,y0,x1,y1`, normalized to 0–511) docling attaches to elements from
271    /// backends with real geometry (e.g. the slide shapes in PPTX). Markdown and
272    /// JSON render the wrapped node unchanged; DocLang emits the `<location>`
273    /// tokens as the element's first children.
274    Located {
275        location: [u16; 4],
276        inner: Box<Node>,
277    },
278    /// Exact page provenance for a backend whose geometry already *is* the
279    /// page's coordinate system — an XLSX item's cell-index box on its sheet,
280    /// which docling writes verbatim (`bbox` in a top-left origin, the
281    /// `charspan` the backend chose) and sizes the page from. The 0–511 grid
282    /// of a [`Node::Located`] cannot round-trip such integers exactly, so the
283    /// JSON export reads this wrapper; every other serializer renders `inner`
284    /// unchanged (DocLang keeps taking its `<location>` tokens from the grid).
285    Prov {
286        page_no: usize,
287        /// `[l, t, r, b]`, page units, top-left origin.
288        bbox: [f32; 4],
289        /// docling's `charspan` for the item (`[0, 0]` for an XLSX table).
290        charspan: [usize; 2],
291        /// The item's creation rank among its siblings, when that differs
292        /// from the node order: docling numbers `#/tables/N` / `#/texts/N` /
293        /// `#/pictures/N` in the order it *creates* items (a sheet's tables,
294        /// then its images, then its charts) and only afterwards sorts the
295        /// container's children by position. The JSON export adds siblings in
296        /// this order and lays their refs out in node order; `None` when the
297        /// two orders coincide.
298        seq: Option<usize>,
299        inner: Box<Node>,
300    },
301    /// A PDF page header or footer (docling's `page_header`/`page_footer`
302    /// furniture): DocLang emits `<page_header>`/`<page_footer>` with a
303    /// `<layer value="furniture"/>` head, the four `<location>` tokens, then the
304    /// text. The JSON writes it as a body-parented `page_header`/`page_footer`
305    /// text item on the furniture layer; Markdown omits it.
306    PageFurniture {
307        footer: bool,
308        location: [u16; 4],
309        text: String,
310    },
311    /// A furniture-layer text item with an explicit docling label — a page
312    /// header / footer paragraph (`page_header` / `page_footer`) or a note
313    /// body (`footnote`) from a backend that builds no item tree (#535: the
314    /// Word 97 `.doc` stories). The JSON writes a body-parented item of that
315    /// label on the furniture layer; DocLang `<{label}>` with a `<layer
316    /// value="furniture"/>` head (docling-core's own shape); Markdown, LaTeX
317    /// and the chunker leave it out like any furniture.
318    FurnitureText { label: String, text: String },
319    /// The text a PDF picture contains, kept the way docling keeps it: every
320    /// regular layout cluster > 80 % inside a picture cluster is that
321    /// `PictureItem`'s child (`LayoutPostprocessor._set_cluster_children`,
322    /// `ReadingOrderModel._add_child_elements`). Emitted right after its
323    /// picture node. The JSON export writes these nodes as items parented to
324    /// that picture, after its caption; docling's Markdown and LaTeX picture
325    /// serializers print only the caption and the image, so every other
326    /// serializer skips it — except the Markdown export asked to
327    /// `traverse_pictures` ([`MarkdownExportOptions`], #599), which renders
328    /// them after the picture like upstream's item walk does.
329    PictureChildren(Vec<Node>),
330    /// A page boundary — docling's implicit page break between pages. The PPTX
331    /// backend emits one between consecutive slides. DocLang renders it as
332    /// `<page_break/>`; Markdown and JSON omit it (matching docling's default
333    /// exports, which carry page breaks only in the document model).
334    PageBreak,
335    /// An invisible page marker — the first node of every page the PDF paths
336    /// assemble: the 1-based page number and the page size in PDF points. It
337    /// carries exactly what the JSON export needs to populate docling's
338    /// `pages` map and to denormalize the 0–511 `<location>` grid back into
339    /// BOTTOMLEFT point bboxes for per-item `prov` (#171). Every other
340    /// serializer skips it, so Markdown / DocLang / DocTags output is
341    /// byte-for-byte unchanged.
342    PageInfo {
343        /// 1-based page number (0 = "not yet numbered": the assembler emits
344        /// the marker, the document-level collector stamps the real number).
345        page_no: usize,
346        /// Page width in PDF points.
347        width: f32,
348        /// Page height in PDF points.
349        height: f32,
350    },
351    /// A node docling keeps in the document model (and DocLang) but leaves out
352    /// of the Markdown and JSON exports — e.g. an ODF *presentation*'s pictures
353    /// and charts, which appear in the `.dclx` body but not in its `.md`/`.json`.
354    /// DocLang renders the wrapped node in place; Markdown and JSON skip it.
355    DoclangOnly(Box<Node>),
356    /// A verbatim plain-text dump — docling's plain-text backend emits the whole
357    /// file as a single text item (used for legacy USPTO APS `.txt` grants, which
358    /// docling routes to plain text rather than its APS parser). The stored string
359    /// is the file body, one record per line. Markdown/JSON render it as one text
360    /// block; the DocLang serializer reproduces minidom's per-line layout, CDATA-
361    /// escaping only the lines that need it (see `emit_text_dump`).
362    TextDump(String),
363}
364
365/// Vertical text position of an [`InlineRun`] — docling's `Script`.
366#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
367pub enum Script {
368    #[default]
369    Baseline,
370    Sub,
371    Super,
372}
373
374/// One styled segment of a [`Node::InlineGroup`] — the docling.rs analogue of a
375/// `TextItem` inside an `InlineGroup`, carrying the ancestor formatting docling
376/// tracks. `text` is already whitespace-normalized/trimmed (one segment per
377/// source text node). A hyperlink is intentionally not stored: DocLang drops the
378/// target inside inline scope, keeping only the anchor text.
379#[derive(Debug, Clone, PartialEq, Eq, Default)]
380pub struct InlineRun {
381    pub text: String,
382    pub bold: bool,
383    pub italic: bool,
384    pub underline: bool,
385    pub strike: bool,
386    pub script: Script,
387    pub code: bool,
388    /// An inline equation (`text` holds LaTeX): DocLang renders `<formula>…`,
389    /// Markdown/JSON keep the `$…$` already baked into the group's `md_text`.
390    pub formula: bool,
391}
392
393/// A DocLang content layer other than the default `body` (see [`Node::Furniture`]).
394#[derive(Debug, Clone, Copy, PartialEq, Eq)]
395pub enum ContentLayer {
396    /// Page headers/footers, HTML `<title>`, site navigation/chrome.
397    Furniture,
398    /// Editorial notes (docx reviewer comments).
399    Notes,
400    /// Invisible content (hidden spreadsheet sheets).
401    Invisible,
402}
403
404impl ContentLayer {
405    /// The `<layer value="…"/>` token value.
406    pub fn value(self) -> &'static str {
407        match self {
408            ContentLayer::Furniture => "furniture",
409            ContentLayer::Notes => "notes",
410            ContentLayer::Invisible => "invisible",
411        }
412    }
413}
414
415/// A set of content layers, `body` included — docling-core's
416/// `set[ContentLayer]` (`HTMLParams.layers`,
417/// `export_to_html(included_content_layers=…)`), where `body` is a member
418/// like any other. [`Default`] is docling's `DEFAULT_CONTENT_LAYERS`: body
419/// only, so an export built with it is unchanged from before layers could
420/// be chosen (#499).
421#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
422pub struct ContentLayers {
423    /// The main content (items without an explicit layer).
424    pub body: bool,
425    /// [`ContentLayer::Furniture`]: page headers/footers, navigation chrome.
426    pub furniture: bool,
427    /// [`ContentLayer::Notes`]: reviewer comments and other editorial notes.
428    pub notes: bool,
429    /// [`ContentLayer::Invisible`]: hidden content (hidden sheets, …).
430    pub invisible: bool,
431}
432
433impl Default for ContentLayers {
434    fn default() -> Self {
435        Self::BODY
436    }
437}
438
439impl ContentLayers {
440    /// Body only — docling's `DEFAULT_CONTENT_LAYERS`.
441    pub const BODY: Self = Self {
442        body: true,
443        furniture: false,
444        notes: false,
445        invisible: false,
446    };
447    /// Every layer — Python's `set(ContentLayer)`.
448    pub const ALL: Self = Self {
449        body: true,
450        furniture: true,
451        notes: true,
452        invisible: true,
453    };
454    /// No layer at all (nothing renders); build a set from it with
455    /// [`with`](Self::with) / [`with_body`](Self::with_body).
456    pub const NONE: Self = Self {
457        body: false,
458        furniture: false,
459        notes: false,
460        invisible: false,
461    };
462
463    /// The set plus `layer`.
464    pub const fn with(mut self, layer: ContentLayer) -> Self {
465        match layer {
466            ContentLayer::Furniture => self.furniture = true,
467            ContentLayer::Notes => self.notes = true,
468            ContentLayer::Invisible => self.invisible = true,
469        }
470        self
471    }
472
473    /// The set plus the body layer.
474    pub const fn with_body(mut self) -> Self {
475        self.body = true;
476        self
477    }
478
479    /// Whether `layer` is in the set; `None` is the body layer.
480    pub fn contains(&self, layer: Option<ContentLayer>) -> bool {
481        match layer {
482            None => self.body,
483            Some(ContentLayer::Furniture) => self.furniture,
484            Some(ContentLayer::Notes) => self.notes,
485            Some(ContentLayer::Invisible) => self.invisible,
486        }
487    }
488
489    /// Whether the layer spelled `name` (`body`, `furniture`, `notes`,
490    /// `invisible` — the JSON `content_layer` values) is in the set; an
491    /// unknown name is not.
492    pub fn contains_name(&self, name: &str) -> bool {
493        match name {
494            "body" => self.body,
495            "furniture" => self.furniture,
496            "notes" => self.notes,
497            "invisible" => self.invisible,
498            _ => false,
499        }
500    }
501
502    /// Parse a comma-separated list of layer names (`body,furniture`;
503    /// whitespace around a name is ignored, `all` is every layer). An
504    /// unknown name is the error. docling-core 2.99's `background` layer
505    /// (watermarks) has no items in this model, so the name is accepted and
506    /// adds nothing — a set written for Python keeps parsing.
507    pub fn parse_list(list: &str) -> Result<Self, String> {
508        let mut set = Self::NONE;
509        for name in list.split(',').map(str::trim).filter(|n| !n.is_empty()) {
510            match name {
511                "body" => set.body = true,
512                "furniture" => set.furniture = true,
513                "notes" => set.notes = true,
514                "invisible" => set.invisible = true,
515                "background" => {}
516                "all" => set = Self::ALL,
517                other => {
518                    return Err(format!(
519                        "unknown content layer `{other}` (expected body, furniture, notes, invisible or all)"
520                    ))
521                }
522            }
523        }
524        Ok(set)
525    }
526}
527
528/// Options of the HTML export ([`DoclingDocument::export_to_html_with`]):
529/// docling-core's `HTMLParams` subset the port honours. [`Default`] is
530/// upstream's default export — placeholder images, `artifacts` as the
531/// referenced-image directory, the body layer only.
532#[derive(Debug, Clone, PartialEq, Eq)]
533pub struct HtmlExportOptions {
534    /// How pictures render (`HTMLParams.image_mode`): nothing but captions
535    /// and meta for [`ImageMode::Placeholder`], `data:` URIs when embedded,
536    /// `<img src>` paths under [`artifacts_dir`](Self::artifacts_dir) when
537    /// referenced.
538    pub image_mode: ImageMode,
539    /// The directory referenced images are named under, the Markdown
540    /// export's convention (`<artifacts_dir>/image_NNNNNN.<ext>`).
541    pub artifacts_dir: String,
542    /// The content layers rendered (`HTMLParams.layers`, #499): an item on a
543    /// layer outside the set is skipped, its children still walked — exactly
544    /// how docling-core's `get_excluded_refs` reads the set.
545    pub layers: ContentLayers,
546}
547
548impl Default for HtmlExportOptions {
549    fn default() -> Self {
550        Self {
551            image_mode: ImageMode::Placeholder,
552            artifacts_dir: "artifacts".to_string(),
553            layers: ContentLayers::BODY,
554        }
555    }
556}
557
558/// Options of the Markdown export
559/// ([`DoclingDocument::export_to_markdown_with_options`], #599): the
560/// docling-core `MarkdownParams` the port honours beyond the image mode.
561/// [`Default`] is upstream's default export — placeholder images,
562/// `artifacts` as the referenced-image directory, the body layer only, no
563/// picture traversal, HTML and underscore escaping on, `<!-- image -->` —
564/// so a document exported with it is byte-identical to
565/// [`DoclingDocument::export_to_markdown`].
566#[derive(Debug, Clone, PartialEq, Eq)]
567pub struct MarkdownExportOptions {
568    /// How pictures render (`MarkdownParams.image_mode`): the placeholder,
569    /// `data:` URIs when embedded, `![Image](<artifacts_dir>/image_NNNNNN.<ext>)`
570    /// when referenced (the bytes come back for the caller to write).
571    pub image_mode: ImageMode,
572    /// The directory referenced images are named under.
573    pub artifacts_dir: String,
574    /// The content layers rendered (`MarkdownParams.layers`,
575    /// `export_to_markdown(included_content_layers=…)`): an item on a layer
576    /// outside the set is skipped, its children still walked. The default
577    /// body-only set drops page headers/footers (`furniture`), reviewer
578    /// comments (`notes`) and hidden sheets (`invisible`), as upstream does; a
579    /// scanned form's running header reads out with
580    /// `ContentLayers::BODY.with(ContentLayer::Furniture)`. The extra items
581    /// render through the body's serializers — a page header is a paragraph.
582    pub layers: ContentLayers,
583    /// `CommonParams.traverse_pictures`: render the text items nested in a
584    /// picture (the PDF pipeline's picture children — a bordered form laid
585    /// out as one picture holds every field as a child) after the picture's
586    /// own caption and image, as upstream yields them. Off, a picture prints
587    /// only its caption and image.
588    pub traverse_pictures: bool,
589    /// `MarkdownParams.escape_html`: `&`, `<` and `>` in text as `&amp;`,
590    /// `&lt;`, `&gt;` (Python's `html.escape(quote=False)`). Off, `R&D` stays
591    /// `R&D`.
592    pub escape_html: bool,
593    /// `MarkdownParams.escape_underscores`: `_` in text as `\_`. Off, the
594    /// text keeps its underscores.
595    pub escape_underscores: bool,
596    /// `MarkdownParams.image_placeholder`: what a picture prints as when it
597    /// renders no image data (the placeholder mode, or a picture without a
598    /// payload); upstream's `<!-- image -->`.
599    pub image_placeholder: String,
600}
601
602impl Default for MarkdownExportOptions {
603    fn default() -> Self {
604        Self {
605            image_mode: ImageMode::Placeholder,
606            artifacts_dir: "artifacts".to_string(),
607            layers: ContentLayers::BODY,
608            traverse_pictures: false,
609            escape_html: true,
610            escape_underscores: true,
611            image_placeholder: "<!-- image -->".to_string(),
612        }
613    }
614}
615
616/// DocLang-only content for a [`Node::ListItem`] whose DocLang form differs from
617/// its flat Markdown `text` (see [`Node::ListItem::dclx`]). `ordered` picks the
618/// enclosing `<list>` kind, `marker` the `<ldiv><marker>`; content is `runs`
619/// (structured equations/formatting) when non-empty, else `text` re-parsed for
620/// inline markers.
621#[derive(Debug, Clone, PartialEq, Eq, Default)]
622pub struct ListItemDclx {
623    pub ordered: bool,
624    pub marker: Option<String>,
625    pub text: String,
626    pub runs: Vec<InlineRun>,
627}
628
629impl InlineRun {
630    /// A run with no active formatting (renders as bare inline text).
631    pub fn is_plain(&self) -> bool {
632        !self.bold
633            && !self.italic
634            && !self.underline
635            && !self.strike
636            && !self.code
637            && !self.formula
638            && self.script == Script::Baseline
639    }
640}
641
642/// Build the [`Node`] for a paragraph of inline content from its structured
643/// `runs` and Markdown text, applying docling's `InlineGroup` boundary:
644///
645/// * a single plain run (or none) → a plain [`Node::Paragraph`] (which the
646///   serializers render as `<text>…</text>`, and a lone hyperlink via `<href>`);
647/// * a single uniformly-formatted run, or two or more runs → a
648///   [`Node::InlineGroup`]. `unwrapped` (the group's docling parent is a
649///   heading, so no enclosing `<text>`) only applies to multi-run groups.
650///
651/// Markdown/JSON render the group's `md_text`, so their output is identical to
652/// emitting a `Paragraph` — the structured runs are DocLang-only.
653pub fn inline_paragraph_node(md_text: String, runs: Vec<InlineRun>, unwrapped: bool) -> Node {
654    let single_plain = runs.len() <= 1 && runs.first().is_none_or(|r| r.is_plain());
655    if single_plain {
656        Node::Paragraph { text: md_text }
657    } else {
658        Node::InlineGroup {
659            unwrapped: unwrapped && runs.len() >= 2,
660            runs,
661            md_text,
662        }
663    }
664}
665
666/// One entry of a [`Node::FieldRegion`]: a marker/key/value triple, any of which
667/// may be absent. Mirrors docling's `field_item` with its `marker` / `field_key`
668/// / `field_value` child texts.
669#[derive(Debug, Clone, PartialEq, Default)]
670pub struct FieldItem {
671    pub marker: Option<String>,
672    pub key: Option<String>,
673    pub value: Option<String>,
674    /// docling's `kind` on the `field_value` item — `read_only` or
675    /// `fillable` (the value is or holds a form control); JSON-only.
676    pub value_kind: Option<String>,
677}
678
679/// One node of a [`Node::KeyValueGraph`] — docling-core's `GraphCell`.
680#[derive(Debug, Clone, PartialEq)]
681pub struct GraphCell {
682    /// docling's `GraphCellLabel` value: `key` or `value` (it also knows
683    /// `unspecified` and `checkbox`).
684    pub label: String,
685    /// The cell's number within its graph, what the links refer to.
686    pub cell_id: usize,
687    pub text: String,
688    /// The text before any cleanup — for an XBRL fact key, the concept's
689    /// qualified name where `text` is its local name.
690    pub orig: String,
691}
692
693/// One edge of a [`Node::KeyValueGraph`] — docling-core's `GraphLink`.
694#[derive(Debug, Clone, PartialEq)]
695pub struct GraphLink {
696    /// docling's `GraphLinkLabel` value: `to_value`, `to_child`, `to_parent`
697    /// or `to_key`.
698    pub label: String,
699    pub source_cell_id: usize,
700    pub target_cell_id: usize,
701}
702
703/// One DocumentPictureClassifier prediction — docling-core's
704/// `PictureClassificationClass` (`class_name` + `confidence`).
705#[derive(Debug, Clone, PartialEq)]
706pub struct PictureClass {
707    /// e.g. `bar_chart`, `logo`, `signature` (the classifier's 26-label set).
708    pub class_name: String,
709    pub confidence: f32,
710}
711
712/// An extracted picture's raw encoded bytes plus its mimetype and pixel size —
713/// the docling.rs analogue of docling-core's `ImageRef`.
714#[derive(Debug, Clone, PartialEq)]
715pub struct PictureImage {
716    /// e.g. `image/png`, `image/jpeg`.
717    pub mimetype: String,
718    pub width: u32,
719    pub height: u32,
720    /// The image file bytes, exactly as embedded (PNG/JPEG/…).
721    pub data: Vec<u8>,
722    /// Pixels per inch of the image relative to the page it came from —
723    /// docling-core's `ImageRef.dpi`, which consumers use to map pixels back
724    /// to points. A crop rendered from a PDF page at `s` px/pt is `72·s`
725    /// (docling: `int(72 * images_scale)`, #519); an image embedded in an
726    /// office file has no render scale and keeps docling's `72`
727    /// ([`Self::DEFAULT_DPI`]).
728    pub dpi: u32,
729}
730
731impl PictureImage {
732    /// docling's `ImageRef` default when no render scale applies.
733    pub const DEFAULT_DPI: u32 = 72;
734
735    /// The `dpi` of an image rendered at `scale` pixels per PDF point
736    /// (`72·scale`, rounded — `int()` would floor 143.99… to 143).
737    pub fn dpi_for_scale(scale: f32) -> u32 {
738        (72.0 * scale).round().max(1.0) as u32
739    }
740
741    /// A `data:` URI for the image (`data:<mimetype>;base64,<…>`).
742    pub fn data_uri(&self) -> String {
743        format!(
744            "data:{};base64,{}",
745            self.mimetype,
746            crate::base64::encode(&self.data)
747        )
748    }
749}
750
751/// One table cell as a first-class object (#240) — the Rust counterpart of
752/// docling's `TableCell`: its text, page geometry, grid rectangle and header
753/// roles. Produced by the PDF TableFormer paths from the predicted OTSL
754/// structure; `bbox` is `[l, t, r, b]` in page points with a top-left origin.
755#[derive(Debug, Clone, PartialEq)]
756pub struct TableCell {
757    pub text: String,
758    /// `[l, t, r, b]`, page points, top-left origin; `None` without geometry.
759    pub bbox: Option<[f32; 4]>,
760    /// Anchor grid position (0-based row/column offsets).
761    pub start_row: usize,
762    pub start_col: usize,
763    /// Span extents (≥ 1); the covered grid positions repeat the cell's text
764    /// in [`Table::rows`].
765    pub row_span: usize,
766    pub col_span: usize,
767    /// OTSL `ched` — a column-header cell.
768    pub column_header: bool,
769    /// OTSL `rhed` — a row-header cell.
770    pub row_header: bool,
771    /// OTSL `srow` — a section-row cell.
772    pub row_section: bool,
773}
774
775/// Where a picture's or table's caption text item hangs in the docling-JSON
776/// tree (#390). docling's backends do not agree, and the JSON structure is
777/// the only surface that shows it (Markdown, DocLang and LaTeX place a
778/// caption by its item, whatever its parent): the PDF pipeline parents a
779/// layout caption to the picture or table it belongs to, while every
780/// declarative backend creates the caption with `doc.add_text(label=CAPTION)`
781/// and no parent — the document body — even when the item itself sits in a
782/// group or under a section header (JATS, LaTeX, HTML, Markdown, EPUB,
783/// AsciiDoc all do). The office backends hang a chart's title caption off the
784/// chart's container (the sheet group, the slide, docx's current parent,
785/// docling#4190) — that is [`Node::Chart`]'s own path, not this choice. A
786/// backend that adds a new captioned item picks the variant matching
787/// upstream's `add_text` call for that format; the default is upstream's
788/// default.
789#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
790pub enum CaptionParent {
791    /// `#/body`, listed after the enclosing top-level item — docling's
792    /// `add_text` default, which every declarative backend leaves alone.
793    #[default]
794    Body,
795    /// The item's own container (its `parent`), listed ahead of the item:
796    /// the caption is created first, as docling's office backends do.
797    Container,
798    /// The item's own container, listed *after* the item: docling's HTML
799    /// backend adds a `<figure>`-wrapped table, then its `<figcaption>` under
800    /// `self.parents[self.level]` (docling#4050).
801    ContainerAfter,
802    /// The item itself — the caption is the picture's or table's first
803    /// child, as docling's PDF pipeline attaches a layout caption.
804    Item,
805}
806
807/// A simple row-major table. By default `rows[0]` is the header row; a
808/// [`TableStructure`] overlay overrides that and adds column spans.
809#[derive(Debug, Clone, PartialEq, Default)]
810pub struct Table {
811    pub rows: Vec<Vec<String>>,
812    /// Optional layout provenance: the four DocLang `<location>` values
813    /// (`x0,y0,x1,y1`, each already normalized to the 0–511 resolution) emitted
814    /// before the table's cells. Set only by backends with real geometry (e.g.
815    /// the spreadsheet backend, whose cell grid yields a bounding box); left
816    /// `None` by declarative backends, which have no coordinates.
817    pub location: Option<[u16; 4]>,
818    /// Optional OTSL structure overlay for backends that parse real table
819    /// geometry (USPTO CALS): explicit header-row count and horizontal-span
820    /// continuations. `None` → the default (row 0 is the header, no spans).
821    /// `rows` still carries the full text grid (span text replicated) for
822    /// Markdown/JSON; DocLang uses this overlay to emit `<ched/>`/`<lcel/>`.
823    pub structure: Option<TableStructure>,
824    /// Optional per-cell block content, parallel to `rows`. A *rich* cell (an
825    /// ODF cell holding a list, several paragraphs, or a nested table) carries
826    /// its DocLang blocks here; the DocLang serializer emits them after the
827    /// cell token instead of the flat `rows` text. Markdown/JSON ignore this
828    /// and render `rows`, so their output is unchanged. `None` (or an empty
829    /// `Vec` for a given cell) → the flat text is used everywhere.
830    pub cell_blocks: Option<Vec<Vec<Vec<Node>>>>,
831    /// Optional caption (docling's `TableItem.captions`): the JATS
832    /// `<table-wrap>` label+caption, an HTML `<caption>`, etc. Markdown renders
833    /// it as a text line *before* the grid; JSON emits a caption text item the
834    /// table references; DocLang emits a `<caption>` as the table's first child.
835    /// `None` → the table has no caption.
836    pub caption: Option<String>,
837    /// Where the caption item hangs in the JSON tree (#390); see
838    /// [`CaptionParent`]. Only the JSON export reads it.
839    pub caption_parent: CaptionParent,
840    /// Optional per-cell bounding boxes, same shape as [`Self::rows`]: `[l, t,
841    /// r, b]` in page points with a **top-left** origin (the PDF pipeline's
842    /// native space). Set by the ML pipeline's TableFormer paths — a spanned
843    /// cell repeats its anchor's box across the covered grid positions — and
844    /// First-class cells (#240): the authoritative per-cell records —
845    /// text, page geometry, spans and header roles — when the backend
846    /// produces them (the PDF TableFormer paths do; declarative backends
847    /// leave `None`). [`Self::rows`] stays the dense text grid every
848    /// serializer renders (a spanning cell's text is replicated across its
849    /// covered positions there); JSON serializes these cells verbatim when
850    /// present, and the DocLang structure overlay is derived from them.
851    pub cells: Option<Vec<TableCell>>,
852}
853
854impl Table {
855    /// A cell's text at a grid position, `None` outside the grid.
856    pub fn cell_text(&self, row: usize, col: usize) -> Option<&str> {
857        self.rows.get(row)?.get(col).map(String::as_str)
858    }
859
860    /// Replace the text at a grid position; `false` (and no change) outside
861    /// the grid. When a first-class cell covers the position, the whole
862    /// cell is updated: its record text and every grid position its span
863    /// covers, so the repair shows once in Markdown, not once per covered
864    /// column.
865    pub fn set_cell_text(&mut self, row: usize, col: usize, text: impl Into<String>) -> bool {
866        if self.rows.get(row).and_then(|r| r.get(col)).is_none() {
867            return false;
868        }
869        let text = text.into();
870        let covering = self.cells.as_mut().and_then(|cells| {
871            cells.iter_mut().find(|c| {
872                (c.start_row..c.start_row + c.row_span).contains(&row)
873                    && (c.start_col..c.start_col + c.col_span).contains(&col)
874            })
875        });
876        if let Some(cell) = covering {
877            cell.text = text.clone();
878            let (r0, r1) = (cell.start_row, cell.start_row + cell.row_span);
879            let (c0, c1) = (cell.start_col, cell.start_col + cell.col_span);
880            for r in self.rows.iter_mut().take(r1).skip(r0) {
881                for slot in r.iter_mut().take(c1).skip(c0) {
882                    *slot = text.clone();
883                }
884            }
885        } else {
886            self.rows[row][col] = text;
887        }
888        true
889    }
890
891    /// Derive first-class cells (#240) from the dense grid plus the
892    /// [`TableStructure`] overlay — how declarative tables (DOCX/XLSX merged
893    /// regions, HTML `th`/spans, ODF covered cells, USPTO CALS) get real
894    /// `TableCell` records without page geometry. Anchors are the positions
895    /// not marked as span continuations; extents scan the continuation grids
896    /// right/down (matching the DocLang `lcel`/`ucel` reading). Header roles
897    /// come from the per-cell `col_header`/`row_header` grids when present,
898    /// else the `header_row` band, else docling's declarative default (row 0
899    /// is the header). Without any overlay every position is a 1×1 cell.
900    pub fn derive_cells(&self) -> Vec<TableCell> {
901        let s = self.structure.as_ref();
902        let flag = |grid: Option<&Vec<Vec<bool>>>, r: usize, c: usize| {
903            grid.and_then(|g| g.get(r))
904                .and_then(|row| row.get(c))
905                .copied()
906                .unwrap_or(false)
907        };
908        let col_cont = |r: usize, c: usize| flag(s.map(|s| &s.col_continuation), r, c);
909        let row_cont = |r: usize, c: usize| flag(s.map(|s| &s.row_continuation), r, c);
910        let is_col_header = |r: usize, c: usize| match s {
911            Some(st) if !st.col_header.is_empty() => flag(Some(&st.col_header), r, c),
912            Some(st) if !st.header_row.is_empty() => st.header_row.get(r).copied().unwrap_or(false),
913            _ => r == 0,
914        };
915        let mut cells = Vec::new();
916        for (r, row) in self.rows.iter().enumerate() {
917            for (c, text) in row.iter().enumerate() {
918                if col_cont(r, c) || row_cont(r, c) {
919                    continue; // covered by a span anchor
920                }
921                let mut col_span = 1;
922                while c + col_span < row.len() && col_cont(r, c + col_span) {
923                    col_span += 1;
924                }
925                let mut row_span = 1;
926                while r + row_span < self.rows.len() && row_cont(r + row_span, c) {
927                    row_span += 1;
928                }
929                cells.push(TableCell {
930                    text: text.clone(),
931                    bbox: None,
932                    start_row: r,
933                    start_col: c,
934                    row_span,
935                    col_span,
936                    column_header: is_col_header(r, c),
937                    row_header: flag(s.map(|s| &s.row_header), r, c),
938                    row_section: false,
939                });
940            }
941        }
942        cells
943    }
944
945    /// The number of leading grid rows that form the column header —
946    /// docling-core's `_count_header_rows` (docling-core#723, 2.96) shared by
947    /// the Markdown serializer and the chunker's dataframe view: a row counts
948    /// only when a `column_header` cell *starts* on it (a header spanning
949    /// several rows is replicated into each row it covers, and counting those
950    /// would pull the data rows beneath it into the header block). Two
951    /// special cases: `1` when no cell carries the flag at all, so tables from
952    /// backends that never set it keep row 0 as the header; `0` when flags
953    /// exist but none starts on row 0 — then nothing is promotable and every
954    /// row stays in the body. Uses the first-class [`Self::cells`] when
955    /// present (the PDF pipeline's TableFormer flags), else the cells derived
956    /// from the structure overlay.
957    ///
958    /// A pivot table's row headers no longer disturb this: docling#4216 flags
959    /// a spanning `<th>` row as `row_header`, not `column_header`, so the
960    /// first data row is no longer folded into the header block and the
961    /// deviation this port carried for docling-core#765 is gone.
962    ///
963    /// A row also stops the block when it *carries body text* (#604,
964    /// docling-core#766, 2.97): a cell starting on it that is not a column
965    /// header, has non-blank text and does not start in the table's first
966    /// column. A scanned form flags its first-column labels `column_header`
967    /// on every row; without the check each such row folded its values into
968    /// the header, and a table labelled on every row lost its whole body.
969    /// When row 0 itself is stopped that way and no later row carries a
970    /// flag, row 0 is still the header (`1`), as for an unflagged table.
971    pub fn header_row_count(&self) -> usize {
972        if self.rows.is_empty() {
973            return 0;
974        }
975        let derived;
976        let cells: &[TableCell] = match &self.cells {
977            Some(c) if !c.is_empty() => c,
978            _ => {
979                derived = self.derive_cells();
980                &derived
981            }
982        };
983        let first_col = cells.iter().map(|c| c.start_col).min().unwrap_or(0);
984        let is_header_row = |r: usize| {
985            let mut flagged = false;
986            for c in cells.iter().filter(|c| c.start_row == r) {
987                if c.column_header {
988                    flagged = true;
989                } else if c.start_col != first_col && !c.text.trim().is_empty() {
990                    return false;
991                }
992            }
993            flagged
994        };
995        let count = (0..self.rows.len())
996            .take_while(|&r| is_header_row(r))
997            .count();
998        // Upstream's row-0 fallback: no flag on any grid row below the first
999        // (a spanning header cell counts on every row it covers) → row 0.
1000        if count == 0
1001            && !cells
1002                .iter()
1003                .any(|c| c.column_header && c.start_row + c.row_span.max(1) > 1)
1004        {
1005            return 1;
1006        }
1007        count
1008    }
1009
1010    /// The first-class cell covering a grid position, if any.
1011    pub fn cell_at(&self, row: usize, col: usize) -> Option<&TableCell> {
1012        self.cells.as_ref()?.iter().find(|c| {
1013            (c.start_row..c.start_row + c.row_span).contains(&row)
1014                && (c.start_col..c.start_col + c.col_span).contains(&col)
1015        })
1016    }
1017
1018    /// A cell's bounding box (`[l, t, r, b]`, page points, top-left origin);
1019    /// `None` when no cell with geometry covers the position.
1020    pub fn cell_bbox(&self, row: usize, col: usize) -> Option<[f32; 4]> {
1021        self.cell_at(row, col)?.bbox
1022    }
1023
1024    /// Set (or replace) the bounding box of the cell covering a grid
1025    /// position; `false` outside the text grid. A table without first-class
1026    /// cells materializes them first (one 1×1 cell per grid position, texts
1027    /// from the grid), so declarative tables can be annotated too.
1028    pub fn set_cell_bbox(&mut self, row: usize, col: usize, bbox: [f32; 4]) -> bool {
1029        if self.rows.get(row).and_then(|r| r.get(col)).is_none() {
1030            return false;
1031        }
1032        let rows = &self.rows;
1033        let cells = self.cells.get_or_insert_with(|| {
1034            rows.iter()
1035                .enumerate()
1036                .flat_map(|(r, cols)| {
1037                    cols.iter().enumerate().map(move |(c, text)| TableCell {
1038                        text: text.clone(),
1039                        bbox: None,
1040                        start_row: r,
1041                        start_col: c,
1042                        row_span: 1,
1043                        col_span: 1,
1044                        column_header: false,
1045                        row_header: false,
1046                        row_section: false,
1047                    })
1048                })
1049                .collect()
1050        });
1051        match cells.iter_mut().find(|c| {
1052            (c.start_row..c.start_row + c.row_span).contains(&row)
1053                && (c.start_col..c.start_col + c.col_span).contains(&col)
1054        }) {
1055            Some(cell) => {
1056                cell.bbox = Some(bbox);
1057                true
1058            }
1059            None => {
1060                cells.push(TableCell {
1061                    text: self.rows[row][col].clone(),
1062                    bbox: Some(bbox),
1063                    start_row: row,
1064                    start_col: col,
1065                    row_span: 1,
1066                    col_span: 1,
1067                    column_header: false,
1068                    row_header: false,
1069                    row_section: false,
1070                });
1071                true
1072            }
1073        }
1074    }
1075
1076    /// The anchor position of the cell whose box overlaps `bbox` best
1077    /// (largest intersection-over-union), ties resolved in cell order.
1078    /// `None` when nothing overlaps or the table carries no geometry. This
1079    /// is the lookup half of the repair workflow: find the cell an external
1080    /// OCR box refers to, then [`Self::set_cell_text`] it.
1081    pub fn find_cell_by_bbox(&self, bbox: [f32; 4]) -> Option<(usize, usize)> {
1082        let area = |b: &[f32; 4]| ((b[2] - b[0]) * (b[3] - b[1])).max(0.0);
1083        let mut best: Option<(f32, (usize, usize))> = None;
1084        for cell in self.cells.as_deref()?.iter() {
1085            let Some(cb) = cell.bbox else { continue };
1086            let iw = (bbox[2].min(cb[2]) - bbox[0].max(cb[0])).max(0.0);
1087            let ih = (bbox[3].min(cb[3]) - bbox[1].max(cb[1])).max(0.0);
1088            let inter = iw * ih;
1089            if inter <= 0.0 {
1090                continue;
1091            }
1092            let iou = inter / (area(&bbox) + area(&cb) - inter).max(f32::EPSILON);
1093            if best.is_none_or(|(b, _)| iou > b) {
1094                best = Some((iou, (cell.start_row, cell.start_col)));
1095            }
1096        }
1097        best.map(|(_, pos)| pos)
1098    }
1099
1100    /// Locate the cell overlapping `bbox` best and replace its text — the
1101    /// one-call form of the OCR-repair loop. Returns the updated anchor.
1102    pub fn update_cell_by_bbox(
1103        &mut self,
1104        bbox: [f32; 4],
1105        text: impl Into<String>,
1106    ) -> Option<(usize, usize)> {
1107        let (row, col) = self.find_cell_by_bbox(bbox)?;
1108        self.set_cell_text(row, col, text);
1109        Some((row, col))
1110    }
1111}
1112
1113/// OTSL structure overlay for a [`Table`], parallel to [`Table::rows`].
1114#[derive(Debug, Clone, PartialEq, Default)]
1115pub struct TableStructure {
1116    /// Per-row: `true` if the row's non-empty cells are column headers
1117    /// (emitted as `<ched/>` rather than `<fcel/>`).
1118    pub header_row: Vec<bool>,
1119    /// Same shape as [`Table::rows`]; `true` where a cell continues a
1120    /// horizontal span from its left neighbour (emitted as `<lcel/>`).
1121    pub col_continuation: Vec<Vec<bool>>,
1122    /// Same shape as [`Table::rows`]; `true` where a cell continues a
1123    /// vertical span from the cell above (emitted as `<ucel/>`). Empty or all
1124    /// `false` when the backend has no vertical spans (e.g. USPTO CALS).
1125    pub row_continuation: Vec<Vec<bool>>,
1126    /// Same shape as [`Table::rows`]; `true` where a non-empty cell is a row
1127    /// header (emitted as `<rhed/>`) — a chart's category column. Empty when
1128    /// the table has no row headers.
1129    pub row_header: Vec<Vec<bool>>,
1130    /// Same shape as [`Table::rows`]; `true` where a cell is a *column header*
1131    /// cell (an HTML `<th>`). When non-empty this per-cell grid supersedes the
1132    /// per-row [`Self::header_row`] for `<ched/>` emission, matching docling's
1133    /// cell-level `column_header` flag; the chunker derives its header-row
1134    /// count from it.
1135    pub col_header: Vec<Vec<bool>>,
1136}
1137
1138impl DoclingDocument {
1139    /// Create an empty document with the given name.
1140    pub fn new(name: impl Into<String>) -> Self {
1141        Self {
1142            name: name.into(),
1143            nodes: Vec::new(),
1144            strict_markdown: false,
1145            compact_tables: false,
1146            page_break_placeholder: None,
1147            links: Vec::new(),
1148            confidence: None,
1149            tree: None,
1150            page_images: std::collections::BTreeMap::new(),
1151        }
1152    }
1153
1154    /// Append a node.
1155    /// The document's top-level tables in reading order — the read half of
1156    /// the post-extraction table API (#238). [`Node::Located`] wrappers (the
1157    /// PDF pipeline attaches layout provenance that way) are looked through;
1158    /// tables nested inside rich table cells (`Table::cell_blocks`) are not
1159    /// traversed.
1160    pub fn tables(&self) -> impl Iterator<Item = &Table> {
1161        fn unwrap_table(n: &Node) -> Option<&Table> {
1162            match n {
1163                Node::Table(t) => Some(t),
1164                Node::Located { inner, .. } | Node::Prov { inner, .. } => unwrap_table(inner),
1165                _ => None,
1166            }
1167        }
1168        self.nodes.iter().filter_map(unwrap_table)
1169    }
1170
1171    /// Mutable access to the document's top-level tables, for repair
1172    /// workflows (#238): locate a cell via [`Table::find_cell_by_bbox`], fix
1173    /// its text with [`Table::set_cell_text`], then re-export — every
1174    /// serializer reads the same grid.
1175    pub fn tables_mut(&mut self) -> impl Iterator<Item = &mut Table> {
1176        fn unwrap_table(n: &mut Node) -> Option<&mut Table> {
1177            match n {
1178                Node::Table(t) => Some(t),
1179                Node::Located { inner, .. } | Node::Prov { inner, .. } => unwrap_table(inner),
1180                _ => None,
1181            }
1182        }
1183        self.nodes.iter_mut().filter_map(unwrap_table)
1184    }
1185
1186    pub fn push(&mut self, node: Node) {
1187        self.nodes.push(node);
1188    }
1189
1190    /// Convenience: append a heading.
1191    pub fn add_heading(&mut self, level: u8, text: impl Into<String>) {
1192        self.push(Node::Heading {
1193            level,
1194            text: text.into(),
1195        });
1196    }
1197
1198    /// Convenience: append a paragraph.
1199    pub fn add_paragraph(&mut self, text: impl Into<String>) {
1200        self.push(Node::Paragraph { text: text.into() });
1201    }
1202
1203    /// Serialize the document to Markdown.
1204    ///
1205    /// The Rust equivalent of docling-core's
1206    /// `DoclingDocument.export_to_markdown()`. Uses [`Self::strict_markdown`] to
1207    /// pick between docling-legacy output (default) and the cleaner, more
1208    /// conformant variant.
1209    pub fn export_to_markdown(&self) -> String {
1210        to_markdown(self, self.strict_markdown)
1211    }
1212
1213    /// Serialize to Markdown, explicitly choosing the mode regardless of
1214    /// [`Self::strict_markdown`]. `strict = true` produces cleaner, more
1215    /// conformant Markdown (code-fence languages preserved, no inline-run
1216    /// spacing artifacts); `strict = false` reproduces docling's legacy output.
1217    pub fn export_to_markdown_with(&self, strict: bool) -> String {
1218        to_markdown(self, strict)
1219    }
1220
1221    /// Markdown for this document as the *content of a rich table cell*
1222    /// (docling-core's `in_table_cell` serialization, docling-core#540):
1223    /// headings render as plain text since Markdown tables can't hold them.
1224    /// Backends build a sub-document per rich cell and flatten this into the
1225    /// cell text; no trailing newline.
1226    pub fn export_to_table_cell_markdown(&self) -> String {
1227        crate::markdown::to_markdown_table_cell(self, self.strict_markdown)
1228    }
1229
1230    /// Serialize to docling-core's native JSON wire format (`DoclingDocument`
1231    /// schema), pretty-printed — the Rust equivalent of
1232    /// `DoclingDocument.export_to_dict()` / `save_as_json()`. The output loads
1233    /// back into Python docling-core and round-trips to the same Markdown.
1234    pub fn export_to_json(&self) -> String {
1235        serde_json::to_string_pretty(&self.export_to_json_value())
1236            .expect("DoclingDocument JSON is always serializable")
1237    }
1238
1239    /// The same JSON wire format as [`Self::export_to_json`], as a
1240    /// `serde_json::Value` — for callers that append response-level extras
1241    /// (docling-serve adds the confidence report, #183) before serializing.
1242    pub fn export_to_json_value(&self) -> serde_json::Value {
1243        crate::json::to_json(self)
1244    }
1245
1246    /// Serialize to a complete LaTeX document — the Rust counterpart of
1247    /// docling-core's `LaTeXDocSerializer` with default parameters (docling
1248    /// 2.124's `--to latex`, #317). No trailing newline, like the upstream
1249    /// CLI's `<stem>.tex`.
1250    pub fn export_to_latex(&self) -> String {
1251        crate::latex::to_latex(self)
1252    }
1253
1254    /// Serialize to Pandoc's AST as JSON (`pandoc -f json`, #515) with the
1255    /// default options: pictures as captioned figures without image data,
1256    /// body layer only, the [`PANDOC_API_VERSION`](crate::pandoc::PANDOC_API_VERSION)
1257    /// API. One line, no trailing newline. See [`crate::pandoc`].
1258    pub fn export_to_pandoc_json(&self) -> String {
1259        crate::pandoc::to_pandoc(self, &crate::pandoc::PandocExportOptions::default())
1260            .expect("the default Pandoc API version is always supported")
1261            .0
1262    }
1263
1264    /// Pandoc JSON per `options` (image mode, artifacts directory, content
1265    /// layers, required API version). Returns the JSON and, for
1266    /// [`ImageMode::Referenced`], the `(path, bytes)` image files; an
1267    /// unsupported `api_version` is an error.
1268    pub fn export_to_pandoc_json_with(
1269        &self,
1270        options: &crate::pandoc::PandocExportOptions,
1271    ) -> Result<crate::pandoc::PandocOutput, crate::pandoc::PandocError> {
1272        crate::pandoc::to_pandoc(self, options)
1273    }
1274
1275    /// Serialize to DocLang XML (`<doclang version="0.7">…`), the markup that
1276    /// lives inside a `.dclx` archive — the Rust counterpart of docling-core's
1277    /// `export_to_doclang()` with default parameters. No trailing newline; the
1278    /// archive writer appends exactly one.
1279    pub fn export_to_doclang(&self) -> String {
1280        crate::doclang::export_to_doclang(&self.nodes)
1281    }
1282
1283    /// [`Self::export_to_doclang`] plus the picture assets its
1284    /// `<src uri="assets/image_….png"/>` references name, as `(path, bytes)`
1285    /// in document order — the parts docling's `save_as_doclang_archive`
1286    /// stores next to `document.xml`. PNG and JPEG pictures come back as PNG
1287    /// (a JPEG re-encoded from its libjpeg-decoded pixels); other encodings
1288    /// keep their own bytes for the archive writer to convert.
1289    pub fn export_to_doclang_with_assets(&self) -> (String, Vec<(String, Vec<u8>)>) {
1290        crate::doclang::export_to_doclang_with_assets(&self.nodes)
1291    }
1292
1293    /// Serialize to Markdown with an explicit picture [`ImageMode`] (mirrors
1294    /// docling's `image_mode`). Returns the Markdown and, for
1295    /// [`ImageMode::Referenced`], the `(relative-path, bytes)` of each image the
1296    /// caller should write next to the Markdown file. `artifacts_dir` is the
1297    /// directory name used in referenced links.
1298    pub fn export_to_markdown_with_images(
1299        &self,
1300        image_mode: ImageMode,
1301        artifacts_dir: &str,
1302    ) -> (String, Vec<(String, Vec<u8>)>) {
1303        to_markdown_images(self, self.strict_markdown, image_mode, artifacts_dir)
1304    }
1305
1306    /// Markdown per `options` (#599) — docling-core's
1307    /// `export_to_markdown(included_content_layers=…, image_mode=…,
1308    /// image_placeholder=…, escape_html=…, escape_underscores=…,
1309    /// traverse_pictures=…)`: the content layers rendered, whether a
1310    /// picture's nested text items print, HTML and underscore escaping, the
1311    /// image placeholder, and the image mode with its referenced-image
1312    /// directory. [`Self::strict_markdown`] still picks the Markdown flavour.
1313    /// Returns the Markdown and, for [`ImageMode::Referenced`], the
1314    /// `(path, bytes)` artifacts. With [`MarkdownExportOptions::default`]
1315    /// the Markdown is [`Self::export_to_markdown`]'s byte for byte.
1316    pub fn export_to_markdown_with_options(
1317        &self,
1318        options: &MarkdownExportOptions,
1319    ) -> (String, Vec<(String, Vec<u8>)>) {
1320        crate::markdown::to_markdown_with_options(self, self.strict_markdown, options)
1321    }
1322
1323    /// A complete HTML document — docling-core's `HTMLDocSerializer` with its
1324    /// defaults (#492): pictures stay out (`ImageRefMode.PLACEHOLDER`, only
1325    /// their captions and meta render). See [`export_to_html_with_images`]
1326    /// for the embedded / referenced modes and [`crate::html`]'s module docs
1327    /// for what is reproduced.
1328    ///
1329    /// [`export_to_html_with_images`]: Self::export_to_html_with_images
1330    pub fn export_to_html(&self) -> String {
1331        crate::html::to_html(self, &HtmlExportOptions::default()).0
1332    }
1333
1334    /// HTML with pictures per `image_mode`: `data:` URIs when embedded, and
1335    /// when referenced `<img src>` paths under `artifacts_dir` whose bytes come
1336    /// back as `(path, bytes)` for the caller to write — the Markdown export's
1337    /// contract and file names.
1338    pub fn export_to_html_with_images(
1339        &self,
1340        image_mode: ImageMode,
1341        artifacts_dir: &str,
1342    ) -> (String, Vec<(String, Vec<u8>)>) {
1343        crate::html::to_html(
1344            self,
1345            &HtmlExportOptions {
1346                image_mode,
1347                artifacts_dir: artifacts_dir.to_string(),
1348                ..HtmlExportOptions::default()
1349            },
1350        )
1351    }
1352
1353    /// HTML rendering the content `layers` — docling-core's
1354    /// `export_to_html(included_content_layers=…)` (#499). The default export
1355    /// is body only; `ContentLayers::BODY.with(ContentLayer::Furniture)` adds
1356    /// page headers/footers, `.with(ContentLayer::Notes)` reviewer comments,
1357    /// [`ContentLayers::ALL`] is Python's `set(ContentLayer)`. An item on a
1358    /// layer outside the set is skipped while its children are still walked,
1359    /// and the items that do render go through the same serializers as the
1360    /// body (a page header is a `<p>`, a comment a `<p>`, a header table a
1361    /// `<table>`), exactly as upstream's `HTMLParams.layers` behaves. Pictures
1362    /// stay placeholders; see [`export_to_html_with`] for the full option set.
1363    ///
1364    /// [`export_to_html_with`]: Self::export_to_html_with
1365    pub fn export_to_html_with_layers(&self, layers: ContentLayers) -> String {
1366        crate::html::to_html(
1367            self,
1368            &HtmlExportOptions {
1369                layers,
1370                ..HtmlExportOptions::default()
1371            },
1372        )
1373        .0
1374    }
1375
1376    /// HTML per `options` — image mode, referenced-image directory and content
1377    /// layers in one call; the other `export_to_html*` methods are
1378    /// shorthands for it. Returns the HTML and, for
1379    /// [`ImageMode::Referenced`], the `(path, bytes)` artifacts.
1380    pub fn export_to_html_with(
1381        &self,
1382        options: &HtmlExportOptions,
1383    ) -> (String, Vec<(String, Vec<u8>)>) {
1384        crate::html::to_html(self, options)
1385    }
1386}
1387
1388#[cfg(test)]
1389mod table_api_tests {
1390    use super::*;
1391
1392    fn cell(
1393        text: &str,
1394        bbox: [f32; 4],
1395        (start_row, start_col): (usize, usize),
1396        (row_span, col_span): (usize, usize),
1397    ) -> TableCell {
1398        TableCell {
1399            text: text.into(),
1400            bbox: Some(bbox),
1401            start_row,
1402            start_col,
1403            row_span,
1404            col_span,
1405            column_header: false,
1406            row_header: false,
1407            row_section: false,
1408        }
1409    }
1410
1411    fn table() -> Table {
1412        Table {
1413            rows: vec![
1414                vec!["Year".into(), "Ducks".into()],
1415                vec!["2019".into(), "120".into()],
1416            ],
1417            cells: Some(vec![
1418                cell("Year", [0.0, 0.0, 50.0, 10.0], (0, 0), (1, 1)),
1419                cell("Ducks", [50.0, 0.0, 100.0, 10.0], (0, 1), (1, 1)),
1420                cell("2019", [0.0, 10.0, 50.0, 20.0], (1, 0), (1, 1)),
1421                cell("120", [50.0, 10.0, 100.0, 20.0], (1, 1), (1, 1)),
1422            ]),
1423            ..Default::default()
1424        }
1425    }
1426
1427    /// Declarative tables derive first-class cells from the structure
1428    /// overlay: continuation grids become span extents, `col_header` (or the
1429    /// row-0 fallback) becomes the header role — the XLSX/DOCX/HTML merge
1430    /// path into real `TableCell`s (#240).
1431    #[test]
1432    fn derive_cells_reads_spans_and_headers_from_structure() {
1433        let t = Table {
1434            rows: vec![
1435                vec!["Wide".into(), "Wide".into(), "C".into()],
1436                vec!["a".into(), "b".into(), "c".into()],
1437            ],
1438            structure: Some(TableStructure {
1439                header_row: vec![true, false],
1440                col_continuation: vec![vec![false, true, false], vec![false; 3]],
1441                row_continuation: vec![vec![false; 3], vec![false; 3]],
1442                row_header: Vec::new(),
1443                col_header: Vec::new(),
1444            }),
1445            ..Default::default()
1446        };
1447        let cells = t.derive_cells();
1448        assert_eq!(cells.len(), 5, "two anchors in row 0, three in row 1");
1449        let wide = &cells[0];
1450        assert_eq!((wide.col_span, wide.row_span), (2, 1));
1451        assert!(wide.column_header, "header_row band");
1452        assert!(cells.iter().skip(2).all(|c| !c.column_header));
1453
1454        // Without any overlay: every position 1x1, row 0 the header
1455        // (docling's declarative default — the old JSON synthesis).
1456        let plain = Table {
1457            rows: vec![vec!["h".into()], vec!["x".into()]],
1458            ..Default::default()
1459        };
1460        let cells = plain.derive_cells();
1461        assert_eq!(cells.len(), 2);
1462        assert!(cells[0].column_header && !cells[1].column_header);
1463    }
1464
1465    /// A spanning cell updates once: the record text and every covered grid
1466    /// position — a repair shows once in Markdown, not once per column.
1467    #[test]
1468    fn span_repair_updates_the_whole_cell() {
1469        let mut t = Table {
1470            rows: vec![
1471                vec!["Wide".into(), "Wide".into(), "C".into()],
1472                vec!["a".into(), "b".into(), "c".into()],
1473            ],
1474            cells: Some(vec![
1475                cell("Wide", [0.0, 0.0, 100.0, 10.0], (0, 0), (1, 2)),
1476                cell("C", [100.0, 0.0, 150.0, 10.0], (0, 2), (1, 1)),
1477            ]),
1478            ..Default::default()
1479        };
1480        // Update through the covered (non-anchor) position.
1481        assert!(t.set_cell_text(0, 1, "Fixed"));
1482        assert_eq!(
1483            t.rows[0],
1484            vec!["Fixed".to_string(), "Fixed".into(), "C".into()]
1485        );
1486        assert_eq!(t.cell_at(0, 1).unwrap().text, "Fixed");
1487        assert_eq!(t.cell_at(0, 1).unwrap().col_span, 2);
1488    }
1489
1490    /// The OCR-repair loop (#238): locate a cell by an external box (best
1491    /// IoU), replace its text, and see the fix in the export — the grid is
1492    /// the single source of truth for every serializer.
1493    #[test]
1494    fn bbox_lookup_and_repair_flow_into_exports() {
1495        let mut doc = DoclingDocument::new("t");
1496        doc.push(Node::Table(table()));
1497        assert_eq!(doc.tables().count(), 1);
1498
1499        let t = doc.tables_mut().next().unwrap();
1500        // A slightly-off OCR box still lands on the (1,1) cell.
1501        assert_eq!(t.find_cell_by_bbox([52.0, 11.0, 98.0, 19.0]), Some((1, 1)));
1502        assert_eq!(
1503            t.update_cell_by_bbox([52.0, 11.0, 98.0, 19.0], "125"),
1504            Some((1, 1))
1505        );
1506        assert_eq!(t.cell_text(1, 1), Some("125"));
1507        assert!(doc.export_to_markdown().contains("125"));
1508
1509        // No overlap → no match, nothing changed.
1510        let t = doc.tables_mut().next().unwrap();
1511        assert_eq!(t.find_cell_by_bbox([500.0, 500.0, 600.0, 600.0]), None);
1512    }
1513
1514    #[test]
1515    fn cell_accessors_bound_check_and_geometry_materializes() {
1516        let mut t = table();
1517        assert_eq!(t.cell_text(0, 0), Some("Year"));
1518        assert_eq!(t.cell_text(5, 0), None);
1519        assert!(!t.set_cell_text(0, 9, "x"), "outside the grid");
1520        assert_eq!(t.cell_bbox(1, 0), Some([0.0, 10.0, 50.0, 20.0]));
1521
1522        // A geometry-less table materializes its box grid on first set.
1523        let mut plain = Table {
1524            rows: vec![vec!["a".into(), "b".into()]],
1525            ..Default::default()
1526        };
1527        assert_eq!(plain.cell_bbox(0, 1), None);
1528        assert!(!plain.set_cell_bbox(0, 5, [0.0; 4]), "outside the grid");
1529        assert!(plain.set_cell_bbox(0, 1, [1.0, 2.0, 3.0, 4.0]));
1530        assert_eq!(plain.cell_bbox(0, 1), Some([1.0, 2.0, 3.0, 4.0]));
1531        assert_eq!(plain.find_cell_by_bbox([1.5, 2.5, 2.5, 3.5]), Some((0, 1)));
1532    }
1533}