kreuzberg 4.8.4

High-performance document intelligence library for Rust. Extract text, metadata, and structured data from PDFs, Office documents, images, and 91+ formats and 248 programming languages via tree-sitter code intelligence with async/sync APIs.
Documentation 
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

//! Image extraction using the pdf_oxide backend.
//!
//! Extracts embedded image positions from PDF pages via pdf_oxide for use
//! in the assembly pipeline.

use super::OxideDocument;
use crate::pdf::error::{PdfError, Result};

/// Extract image positions from all pages for interleaving into the assembly pipeline.
///
/// Calls `doc.doc.extract_images(page_idx)` for each page and creates an
/// `(page_number, image_index)` pair for each image found. These positions
/// are used by the assembly pipeline to insert image placeholders into the
/// document structure.
///
/// # Arguments
///
/// * `doc` - Mutable reference to the oxide document
///
/// # Returns
///
/// A `Vec<(usize, usize)>` of (1-indexed page number, global image index) pairs.
pub(crate) fn extract_image_positions(doc: &mut OxideDocument) -> Result<Vec<(usize, usize)>> {
    let page_count = doc
        .doc
        .page_count()
        .map_err(|e| PdfError::MetadataExtractionFailed(format!("pdf_oxide: failed to get page count: {e}")))?;

    let mut positions = Vec::new();
    let mut global_index = 0usize;

    for page_idx in 0..page_count {
        let oxide_images = match doc.doc.extract_images(page_idx) {
            Ok(images) => images,
            Err(e) => {
                tracing::debug!(
                    page = page_idx,
                    "pdf_oxide: failed to extract images for positions: {e}"
                );
                continue;
            }
        };

        let page_number = page_idx + 1; // Kreuzberg uses 1-indexed page numbers

        for _img in &oxide_images {
            positions.push((page_number, global_index));
            global_index += 1;
        }
    }

    Ok(positions)
}