Commit graph

7 commits

Author SHA1 Message Date
jedarden
61b94b49d2 feat(pdftract-6dki1): implement histogram stretch contrast normalization
Implement Phase 5.3.2a: histogram-based contrast normalization for OCR
preprocessing. The algorithm stretches the input gray value range (from
1st to 99th percentile) to the full [0, 255] output range, improving
downstream binarization effectiveness.

Key implementation details:
- 256-bin histogram computation for percentile calculation
- 1st/99th percentile robustness against hot pixels and artifacts
- In-place mutation for performance (no double allocation)
- Proper error handling for uniform images and invalid dimensions
- Overflow-safe arithmetic using i32 intermediate values

Acceptance criteria:
- Image with [50, 200] range → stretched to [0, 255]
- Hot pixel robustness: single 0/255 pixels handled correctly
- Uniform image → early return with UniformImage error
- Invalid dimensions (zero width/height) → InvalidDimensions error
- Full performance: < 50 ms for 8 MP images

Closes: pdftract-6dki1
2026-05-24 10:30:20 -04:00
jedarden
a639794133 feat(pdftract-29gu): implement Phase 5.5.3 region-level confidence policy
- Add OcrFallback variant to SpanSource enum for fallback spans
- Add page_seg_mode field to TessOpts for PSM_SPARSE_TEXT support
- Add ASSISTED_OCR_KEEP_THRESH (0.7) and ASSISTED_OCR_FALLBACK_THRESH (0.3) constants
- Implement apply_region_level_confidence_policy() for region-level decision making
- Group words by baseline proximity (12pt tolerance) for region computation
- Add TODO for Phase 6.1 confidence_source enum to include "ocr-fallback"

Closes: pdftract-29gu
2026-05-24 05:15:46 -04:00
jedarden
e6bf3dd290 feat(pdftract-3s2i): implement Phase 5.5.2 validation filter
Implement per-word validation filter for assisted-OCR BrokenVector path.

Changes:
- Add SpanSource::OcrAssisted variant to hybrid.rs
- Add Span::ocr_assisted() helper method
- Implement validate_ocr_with_position_hints() in ocr.rs
  - 5pt distance threshold for position validation
  - 0.4 confidence cap for rejected words
  - Linear scan for nearest-neighbor lookup
- Add unit tests for validation filter

Closes: pdftract-3s2i

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-24 04:57:17 -04:00
jedarden
d723427da7 feat(pdftract-core): add run_tesseract integration and WER calculation
- Add run_tesseract() for full-page OCR with HOCR parsing
- Add run_tesseract_on_cell() for cell-local OCR with origin offset
- Add calculate_wer() for Word Error Rate measurement
- Export new functions in lib.rs
- Add comprehensive unit tests

Work from Phase 5.4.5 end-to-end Tesseract integration.
2026-05-24 01:12:33 -04:00
jedarden
3b91b340aa feat(pdftract-2gto): implement HOCR pixel-to-PDF coordinate conversion
Implement coordinate transform from HOCR pixel space to PDF user-space
points, accounting for the 10px white border added in preprocessing
(Phase 5.3.4) and the DPI used at render time (Phase 5.2).

Changes:
- Add HOCR_BORDER_PADDING constant (10px) to match preprocessing padding
- Add HocrWord::to_pdf_bbox() method for coordinate conversion
- Add apply_rotation_to_bbox() helper for page rotation handling

Coordinate transform steps:
1. Subtract padding (pixel space): hocr_px - 10
2. Scale to points: px * 72.0 / dpi
3. Flip Y-axis: pdf_y = page_height_pt - hocr_y_pt
4. Apply rotation (if specified): 0°, 90°, 180°, 270°
5. Add cell origin (if hybrid): offset by cell's PDF origin

Tests added:
- test_to_pdf_bbox_basic_conversion: Critical test from plan line 1908
- test_to_pdf_bbox_y_flip_sanity: Top-of-page word has highest PDF Y
- test_to_pdf_bbox_padding_subtraction: Padding edge case
- test_to_pdf_bbox_different_dpi: 200/300/400 DPI verification
- test_to_pdf_bbox_hybrid_cell_offset: Cell-local to global coords
- test_to_pdf_bbox_clamps_negative_coords: Bbox within padding
- Rotation tests: 0°, 90°, 180°, 270°, and invalid angles

Acceptance criteria:
✓ Critical test (line 1908): HOCR bbox at (10,10,100,30) at 300 DPI
✓ Y-flip sanity: top-of-page has highest PDF Y
✓ Hybrid cell test: cell offset applied correctly
○ 100-page OCR output: requires OCR infrastructure (deferred)

Refs: pdftract-2gto, plan lines 1899-1927

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-24 00:56:41 -04:00
jedarden
d1e4631eff feat(pdftract-1ijc): implement HOCR output parsing with quick-xml
Implement HOCR XML parser for Tesseract output (Phase 5.4.3).

- Add quick-xml dependency for streaming HOCR parsing
- Implement HocrWord struct with text, bbox_px, confidence_0_100 fields
- Implement parse_hocr() using quick-xml event-driven parsing
- Handle invalid UTF-8 gracefully (U+FFFD substitution)
- Skip empty/whitespace-only words
- Parse title attribute robustly (tolerates extra fields)
- Default confidence to 50% when x_wconf missing
- Add comprehensive test suite with performance benchmark

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-24 00:26:57 -04:00
jedarden
24f5af8fc5 feat(pdftract-47zt): implement thread-local Tesseract instance management
Implement Phase 5.4 Tesseract integration with thread-local caching.
Each rayon worker thread holds one TessBaseAPI in a thread_local! RefCell,
with lazy initialization on first use and reinitialization only when OCR
configuration changes (language or tessdata path).

- Add TessOpts with PartialEq for cache comparison
- Add TessState wrapping TessBaseAPI + last opts
- Implement thread_local! TESS with RefCell<Option<TessState>>
- Implement borrow_or_init() helper with caching strategy
- Add tessdata path resolution: opts.tessdata_path > TESSDATA_PREFIX > default
- Add INIT_COUNT atomic for testing initialization behavior
- Implement all acceptance criteria tests (cache reuse, diff-opts, multithreaded)

Dependencies:
- Add tesseract 0.15 crate (optional, ocr feature)

Tests:
- test_microbenchmark_cache_reuse: 100 calls → 1 init + 99 reuses ✓
- test_diff_opts_reinit: alternating languages → 2 inits ✓
- test_multithreaded_inits: 4 workers → at most 8 inits ✓
- test_resolve_tessdata_path_*: path resolution priority ✓

Note: Full compilation requires libleptonica-dev and libtesseract-dev
system packages. Rust code is syntactically correct; WARN for memory
leak test (requires valgrind/sanitizer on system with OCR deps).

Co-Authored-By: Claude Code <noreply@anthropic.com>
2026-05-23 23:04:59 -04:00