kreuzberg 4.0.0.pre.rc.29 → 4.0.0.rc1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (321) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +0 -6
  3. data/.rubocop.yaml +534 -1
  4. data/Gemfile +2 -1
  5. data/Gemfile.lock +28 -116
  6. data/README.md +269 -629
  7. data/Rakefile +0 -9
  8. data/Steepfile +4 -8
  9. data/examples/async_patterns.rb +58 -1
  10. data/ext/kreuzberg_rb/extconf.rb +5 -35
  11. data/ext/kreuzberg_rb/native/Cargo.toml +16 -55
  12. data/ext/kreuzberg_rb/native/build.rs +14 -12
  13. data/ext/kreuzberg_rb/native/include/ieeefp.h +1 -1
  14. data/ext/kreuzberg_rb/native/include/msvc_compat/strings.h +1 -1
  15. data/ext/kreuzberg_rb/native/include/strings.h +2 -2
  16. data/ext/kreuzberg_rb/native/include/unistd.h +1 -1
  17. data/ext/kreuzberg_rb/native/src/lib.rs +34 -897
  18. data/extconf.rb +6 -38
  19. data/kreuzberg.gemspec +20 -114
  20. data/lib/kreuzberg/api_proxy.rb +18 -2
  21. data/lib/kreuzberg/cache_api.rb +0 -22
  22. data/lib/kreuzberg/cli.rb +10 -2
  23. data/lib/kreuzberg/cli_proxy.rb +10 -0
  24. data/lib/kreuzberg/config.rb +22 -274
  25. data/lib/kreuzberg/errors.rb +7 -73
  26. data/lib/kreuzberg/extraction_api.rb +8 -237
  27. data/lib/kreuzberg/mcp_proxy.rb +11 -2
  28. data/lib/kreuzberg/ocr_backend_protocol.rb +73 -0
  29. data/lib/kreuzberg/post_processor_protocol.rb +71 -0
  30. data/lib/kreuzberg/result.rb +33 -151
  31. data/lib/kreuzberg/setup_lib_path.rb +2 -22
  32. data/lib/kreuzberg/validator_protocol.rb +73 -0
  33. data/lib/kreuzberg/version.rb +1 -1
  34. data/lib/kreuzberg.rb +13 -27
  35. data/pkg/kreuzberg-4.0.0.rc1.gem +0 -0
  36. data/sig/kreuzberg.rbs +12 -105
  37. data/spec/binding/cache_spec.rb +22 -22
  38. data/spec/binding/cli_proxy_spec.rb +4 -2
  39. data/spec/binding/cli_spec.rb +11 -12
  40. data/spec/binding/config_spec.rb +0 -74
  41. data/spec/binding/config_validation_spec.rb +6 -100
  42. data/spec/binding/error_handling_spec.rb +97 -283
  43. data/spec/binding/plugins/ocr_backend_spec.rb +8 -8
  44. data/spec/binding/plugins/postprocessor_spec.rb +11 -11
  45. data/spec/binding/plugins/validator_spec.rb +13 -12
  46. data/spec/examples.txt +104 -0
  47. data/spec/fixtures/config.toml +1 -0
  48. data/spec/fixtures/config.yaml +1 -0
  49. data/spec/fixtures/invalid_config.toml +1 -0
  50. data/spec/smoke/package_spec.rb +3 -2
  51. data/spec/spec_helper.rb +3 -1
  52. data/vendor/kreuzberg/Cargo.toml +67 -192
  53. data/vendor/kreuzberg/README.md +9 -97
  54. data/vendor/kreuzberg/build.rs +194 -516
  55. data/vendor/kreuzberg/src/api/handlers.rs +9 -130
  56. data/vendor/kreuzberg/src/api/mod.rs +3 -18
  57. data/vendor/kreuzberg/src/api/server.rs +71 -236
  58. data/vendor/kreuzberg/src/api/types.rs +7 -43
  59. data/vendor/kreuzberg/src/bin/profile_extract.rs +455 -0
  60. data/vendor/kreuzberg/src/cache/mod.rs +3 -27
  61. data/vendor/kreuzberg/src/chunking/mod.rs +79 -1705
  62. data/vendor/kreuzberg/src/core/batch_mode.rs +0 -60
  63. data/vendor/kreuzberg/src/core/config.rs +23 -905
  64. data/vendor/kreuzberg/src/core/extractor.rs +106 -403
  65. data/vendor/kreuzberg/src/core/io.rs +2 -4
  66. data/vendor/kreuzberg/src/core/mime.rs +12 -2
  67. data/vendor/kreuzberg/src/core/mod.rs +3 -22
  68. data/vendor/kreuzberg/src/core/pipeline.rs +78 -395
  69. data/vendor/kreuzberg/src/embeddings.rs +21 -169
  70. data/vendor/kreuzberg/src/error.rs +2 -2
  71. data/vendor/kreuzberg/src/extraction/archive.rs +31 -36
  72. data/vendor/kreuzberg/src/extraction/docx.rs +1 -365
  73. data/vendor/kreuzberg/src/extraction/email.rs +11 -12
  74. data/vendor/kreuzberg/src/extraction/excel.rs +129 -138
  75. data/vendor/kreuzberg/src/extraction/html.rs +170 -1447
  76. data/vendor/kreuzberg/src/extraction/image.rs +14 -138
  77. data/vendor/kreuzberg/src/extraction/libreoffice.rs +3 -13
  78. data/vendor/kreuzberg/src/extraction/mod.rs +5 -21
  79. data/vendor/kreuzberg/src/extraction/office_metadata/mod.rs +0 -2
  80. data/vendor/kreuzberg/src/extraction/pandoc/batch.rs +275 -0
  81. data/vendor/kreuzberg/src/extraction/pandoc/mime_types.rs +178 -0
  82. data/vendor/kreuzberg/src/extraction/pandoc/mod.rs +491 -0
  83. data/vendor/kreuzberg/src/extraction/pandoc/server.rs +496 -0
  84. data/vendor/kreuzberg/src/extraction/pandoc/subprocess.rs +1188 -0
  85. data/vendor/kreuzberg/src/extraction/pandoc/version.rs +162 -0
  86. data/vendor/kreuzberg/src/extraction/pptx.rs +94 -196
  87. data/vendor/kreuzberg/src/extraction/structured.rs +4 -5
  88. data/vendor/kreuzberg/src/extraction/table.rs +1 -2
  89. data/vendor/kreuzberg/src/extraction/text.rs +10 -18
  90. data/vendor/kreuzberg/src/extractors/archive.rs +0 -22
  91. data/vendor/kreuzberg/src/extractors/docx.rs +148 -69
  92. data/vendor/kreuzberg/src/extractors/email.rs +9 -37
  93. data/vendor/kreuzberg/src/extractors/excel.rs +40 -81
  94. data/vendor/kreuzberg/src/extractors/html.rs +173 -182
  95. data/vendor/kreuzberg/src/extractors/image.rs +8 -32
  96. data/vendor/kreuzberg/src/extractors/mod.rs +10 -171
  97. data/vendor/kreuzberg/src/extractors/pandoc.rs +201 -0
  98. data/vendor/kreuzberg/src/extractors/pdf.rs +64 -329
  99. data/vendor/kreuzberg/src/extractors/pptx.rs +34 -79
  100. data/vendor/kreuzberg/src/extractors/structured.rs +0 -16
  101. data/vendor/kreuzberg/src/extractors/text.rs +7 -30
  102. data/vendor/kreuzberg/src/extractors/xml.rs +8 -27
  103. data/vendor/kreuzberg/src/keywords/processor.rs +1 -9
  104. data/vendor/kreuzberg/src/keywords/rake.rs +1 -0
  105. data/vendor/kreuzberg/src/language_detection/mod.rs +51 -94
  106. data/vendor/kreuzberg/src/lib.rs +5 -17
  107. data/vendor/kreuzberg/src/mcp/mod.rs +1 -4
  108. data/vendor/kreuzberg/src/mcp/server.rs +21 -145
  109. data/vendor/kreuzberg/src/ocr/mod.rs +0 -2
  110. data/vendor/kreuzberg/src/ocr/processor.rs +8 -19
  111. data/vendor/kreuzberg/src/ocr/tesseract_backend.rs +0 -2
  112. data/vendor/kreuzberg/src/pdf/error.rs +1 -93
  113. data/vendor/kreuzberg/src/pdf/metadata.rs +100 -263
  114. data/vendor/kreuzberg/src/pdf/mod.rs +2 -33
  115. data/vendor/kreuzberg/src/pdf/rendering.rs +12 -12
  116. data/vendor/kreuzberg/src/pdf/table.rs +64 -61
  117. data/vendor/kreuzberg/src/pdf/text.rs +24 -416
  118. data/vendor/kreuzberg/src/plugins/extractor.rs +8 -40
  119. data/vendor/kreuzberg/src/plugins/mod.rs +0 -3
  120. data/vendor/kreuzberg/src/plugins/ocr.rs +14 -22
  121. data/vendor/kreuzberg/src/plugins/processor.rs +1 -10
  122. data/vendor/kreuzberg/src/plugins/registry.rs +0 -15
  123. data/vendor/kreuzberg/src/plugins/validator.rs +8 -20
  124. data/vendor/kreuzberg/src/stopwords/mod.rs +2 -2
  125. data/vendor/kreuzberg/src/text/mod.rs +0 -8
  126. data/vendor/kreuzberg/src/text/quality.rs +15 -28
  127. data/vendor/kreuzberg/src/text/string_utils.rs +10 -22
  128. data/vendor/kreuzberg/src/text/token_reduction/core.rs +50 -86
  129. data/vendor/kreuzberg/src/text/token_reduction/filters.rs +16 -37
  130. data/vendor/kreuzberg/src/text/token_reduction/simd_text.rs +1 -2
  131. data/vendor/kreuzberg/src/types.rs +67 -907
  132. data/vendor/kreuzberg/src/utils/mod.rs +0 -14
  133. data/vendor/kreuzberg/src/utils/quality.rs +3 -12
  134. data/vendor/kreuzberg/tests/api_tests.rs +0 -506
  135. data/vendor/kreuzberg/tests/archive_integration.rs +0 -2
  136. data/vendor/kreuzberg/tests/batch_orchestration.rs +12 -57
  137. data/vendor/kreuzberg/tests/batch_processing.rs +8 -32
  138. data/vendor/kreuzberg/tests/chunking_offset_demo.rs +92 -0
  139. data/vendor/kreuzberg/tests/concurrency_stress.rs +8 -40
  140. data/vendor/kreuzberg/tests/config_features.rs +1 -33
  141. data/vendor/kreuzberg/tests/config_loading_tests.rs +39 -16
  142. data/vendor/kreuzberg/tests/core_integration.rs +9 -35
  143. data/vendor/kreuzberg/tests/csv_integration.rs +81 -71
  144. data/vendor/kreuzberg/tests/docx_metadata_extraction_test.rs +25 -23
  145. data/vendor/kreuzberg/tests/email_integration.rs +1 -3
  146. data/vendor/kreuzberg/tests/error_handling.rs +34 -43
  147. data/vendor/kreuzberg/tests/format_integration.rs +1 -7
  148. data/vendor/kreuzberg/tests/helpers/mod.rs +0 -60
  149. data/vendor/kreuzberg/tests/image_integration.rs +0 -2
  150. data/vendor/kreuzberg/tests/mime_detection.rs +16 -17
  151. data/vendor/kreuzberg/tests/ocr_configuration.rs +0 -4
  152. data/vendor/kreuzberg/tests/ocr_errors.rs +0 -22
  153. data/vendor/kreuzberg/tests/ocr_quality.rs +0 -2
  154. data/vendor/kreuzberg/tests/pandoc_integration.rs +503 -0
  155. data/vendor/kreuzberg/tests/pdf_integration.rs +0 -2
  156. data/vendor/kreuzberg/tests/pipeline_integration.rs +2 -36
  157. data/vendor/kreuzberg/tests/plugin_ocr_backend_test.rs +0 -5
  158. data/vendor/kreuzberg/tests/plugin_postprocessor_test.rs +1 -17
  159. data/vendor/kreuzberg/tests/plugin_system.rs +0 -6
  160. data/vendor/kreuzberg/tests/registry_integration_tests.rs +22 -2
  161. data/vendor/kreuzberg/tests/security_validation.rs +1 -13
  162. data/vendor/kreuzberg/tests/test_fastembed.rs +23 -45
  163. metadata +25 -171
  164. data/.rubocop.yml +0 -543
  165. data/ext/kreuzberg_rb/native/.cargo/config.toml +0 -23
  166. data/ext/kreuzberg_rb/native/Cargo.lock +0 -7619
  167. data/lib/kreuzberg/error_context.rb +0 -136
  168. data/lib/kreuzberg/types.rb +0 -170
  169. data/lib/libpdfium.so +0 -0
  170. data/spec/binding/async_operations_spec.rb +0 -473
  171. data/spec/binding/batch_operations_spec.rb +0 -595
  172. data/spec/binding/batch_spec.rb +0 -359
  173. data/spec/binding/config_result_spec.rb +0 -377
  174. data/spec/binding/embeddings_spec.rb +0 -816
  175. data/spec/binding/error_recovery_spec.rb +0 -488
  176. data/spec/binding/font_config_spec.rb +0 -220
  177. data/spec/binding/images_spec.rb +0 -738
  178. data/spec/binding/keywords_extraction_spec.rb +0 -600
  179. data/spec/binding/metadata_types_spec.rb +0 -1228
  180. data/spec/binding/pages_extraction_spec.rb +0 -471
  181. data/spec/binding/tables_spec.rb +0 -641
  182. data/spec/unit/config/chunking_config_spec.rb +0 -213
  183. data/spec/unit/config/embedding_config_spec.rb +0 -343
  184. data/spec/unit/config/extraction_config_spec.rb +0 -438
  185. data/spec/unit/config/font_config_spec.rb +0 -285
  186. data/spec/unit/config/hierarchy_config_spec.rb +0 -314
  187. data/spec/unit/config/image_extraction_config_spec.rb +0 -209
  188. data/spec/unit/config/image_preprocessing_config_spec.rb +0 -249
  189. data/spec/unit/config/keyword_config_spec.rb +0 -229
  190. data/spec/unit/config/language_detection_config_spec.rb +0 -258
  191. data/spec/unit/config/ocr_config_spec.rb +0 -171
  192. data/spec/unit/config/page_config_spec.rb +0 -221
  193. data/spec/unit/config/pdf_config_spec.rb +0 -267
  194. data/spec/unit/config/postprocessor_config_spec.rb +0 -290
  195. data/spec/unit/config/tesseract_config_spec.rb +0 -181
  196. data/spec/unit/config/token_reduction_config_spec.rb +0 -251
  197. data/test/metadata_types_test.rb +0 -959
  198. data/vendor/Cargo.toml +0 -61
  199. data/vendor/kreuzberg/examples/bench_fixes.rs +0 -71
  200. data/vendor/kreuzberg/examples/test_pdfium_fork.rs +0 -62
  201. data/vendor/kreuzberg/src/chunking/processor.rs +0 -219
  202. data/vendor/kreuzberg/src/core/batch_optimizations.rs +0 -385
  203. data/vendor/kreuzberg/src/core/config_validation.rs +0 -949
  204. data/vendor/kreuzberg/src/core/formats.rs +0 -235
  205. data/vendor/kreuzberg/src/core/server_config.rs +0 -1220
  206. data/vendor/kreuzberg/src/extraction/capacity.rs +0 -263
  207. data/vendor/kreuzberg/src/extraction/markdown.rs +0 -216
  208. data/vendor/kreuzberg/src/extraction/office_metadata/odt_properties.rs +0 -284
  209. data/vendor/kreuzberg/src/extractors/bibtex.rs +0 -470
  210. data/vendor/kreuzberg/src/extractors/docbook.rs +0 -504
  211. data/vendor/kreuzberg/src/extractors/epub.rs +0 -696
  212. data/vendor/kreuzberg/src/extractors/fictionbook.rs +0 -492
  213. data/vendor/kreuzberg/src/extractors/jats.rs +0 -1054
  214. data/vendor/kreuzberg/src/extractors/jupyter.rs +0 -368
  215. data/vendor/kreuzberg/src/extractors/latex.rs +0 -653
  216. data/vendor/kreuzberg/src/extractors/markdown.rs +0 -701
  217. data/vendor/kreuzberg/src/extractors/odt.rs +0 -628
  218. data/vendor/kreuzberg/src/extractors/opml.rs +0 -635
  219. data/vendor/kreuzberg/src/extractors/orgmode.rs +0 -529
  220. data/vendor/kreuzberg/src/extractors/rst.rs +0 -577
  221. data/vendor/kreuzberg/src/extractors/rtf.rs +0 -809
  222. data/vendor/kreuzberg/src/extractors/security.rs +0 -484
  223. data/vendor/kreuzberg/src/extractors/security_tests.rs +0 -367
  224. data/vendor/kreuzberg/src/extractors/typst.rs +0 -651
  225. data/vendor/kreuzberg/src/language_detection/processor.rs +0 -218
  226. data/vendor/kreuzberg/src/ocr/language_registry.rs +0 -520
  227. data/vendor/kreuzberg/src/panic_context.rs +0 -154
  228. data/vendor/kreuzberg/src/pdf/bindings.rs +0 -306
  229. data/vendor/kreuzberg/src/pdf/bundled.rs +0 -408
  230. data/vendor/kreuzberg/src/pdf/fonts.rs +0 -358
  231. data/vendor/kreuzberg/src/pdf/hierarchy.rs +0 -903
  232. data/vendor/kreuzberg/src/text/quality_processor.rs +0 -231
  233. data/vendor/kreuzberg/src/text/utf8_validation.rs +0 -193
  234. data/vendor/kreuzberg/src/utils/pool.rs +0 -503
  235. data/vendor/kreuzberg/src/utils/pool_sizing.rs +0 -364
  236. data/vendor/kreuzberg/src/utils/string_pool.rs +0 -761
  237. data/vendor/kreuzberg/tests/api_embed.rs +0 -360
  238. data/vendor/kreuzberg/tests/api_extract_multipart.rs +0 -52
  239. data/vendor/kreuzberg/tests/api_large_pdf_extraction.rs +0 -471
  240. data/vendor/kreuzberg/tests/api_large_pdf_extraction_diagnostics.rs +0 -289
  241. data/vendor/kreuzberg/tests/batch_pooling_benchmark.rs +0 -154
  242. data/vendor/kreuzberg/tests/bibtex_parity_test.rs +0 -421
  243. data/vendor/kreuzberg/tests/config_integration_test.rs +0 -753
  244. data/vendor/kreuzberg/tests/data/hierarchy_ground_truth.json +0 -294
  245. data/vendor/kreuzberg/tests/docbook_extractor_tests.rs +0 -500
  246. data/vendor/kreuzberg/tests/docx_vs_pandoc_comparison.rs +0 -370
  247. data/vendor/kreuzberg/tests/epub_native_extractor_tests.rs +0 -275
  248. data/vendor/kreuzberg/tests/fictionbook_extractor_tests.rs +0 -228
  249. data/vendor/kreuzberg/tests/html_table_test.rs +0 -551
  250. data/vendor/kreuzberg/tests/instrumentation_test.rs +0 -139
  251. data/vendor/kreuzberg/tests/jats_extractor_tests.rs +0 -639
  252. data/vendor/kreuzberg/tests/jupyter_extractor_tests.rs +0 -704
  253. data/vendor/kreuzberg/tests/latex_extractor_tests.rs +0 -496
  254. data/vendor/kreuzberg/tests/markdown_extractor_tests.rs +0 -490
  255. data/vendor/kreuzberg/tests/ocr_language_registry.rs +0 -191
  256. data/vendor/kreuzberg/tests/odt_extractor_tests.rs +0 -674
  257. data/vendor/kreuzberg/tests/opml_extractor_tests.rs +0 -616
  258. data/vendor/kreuzberg/tests/orgmode_extractor_tests.rs +0 -822
  259. data/vendor/kreuzberg/tests/page_markers.rs +0 -297
  260. data/vendor/kreuzberg/tests/pdf_hierarchy_detection.rs +0 -301
  261. data/vendor/kreuzberg/tests/pdf_hierarchy_quality.rs +0 -589
  262. data/vendor/kreuzberg/tests/pdf_ocr_triggering.rs +0 -301
  263. data/vendor/kreuzberg/tests/pdf_text_merging.rs +0 -475
  264. data/vendor/kreuzberg/tests/pdfium_linking.rs +0 -340
  265. data/vendor/kreuzberg/tests/rst_extractor_tests.rs +0 -694
  266. data/vendor/kreuzberg/tests/rtf_extractor_tests.rs +0 -775
  267. data/vendor/kreuzberg/tests/typst_behavioral_tests.rs +0 -1260
  268. data/vendor/kreuzberg/tests/typst_extractor_tests.rs +0 -648
  269. data/vendor/kreuzberg-ffi/Cargo.toml +0 -67
  270. data/vendor/kreuzberg-ffi/README.md +0 -851
  271. data/vendor/kreuzberg-ffi/benches/result_view_benchmark.rs +0 -227
  272. data/vendor/kreuzberg-ffi/build.rs +0 -168
  273. data/vendor/kreuzberg-ffi/cbindgen.toml +0 -37
  274. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +0 -12
  275. data/vendor/kreuzberg-ffi/kreuzberg.h +0 -3012
  276. data/vendor/kreuzberg-ffi/src/batch_streaming.rs +0 -588
  277. data/vendor/kreuzberg-ffi/src/config.rs +0 -1341
  278. data/vendor/kreuzberg-ffi/src/error.rs +0 -901
  279. data/vendor/kreuzberg-ffi/src/extraction.rs +0 -555
  280. data/vendor/kreuzberg-ffi/src/helpers.rs +0 -879
  281. data/vendor/kreuzberg-ffi/src/lib.rs +0 -977
  282. data/vendor/kreuzberg-ffi/src/memory.rs +0 -493
  283. data/vendor/kreuzberg-ffi/src/mime.rs +0 -329
  284. data/vendor/kreuzberg-ffi/src/panic_shield.rs +0 -265
  285. data/vendor/kreuzberg-ffi/src/plugins/document_extractor.rs +0 -442
  286. data/vendor/kreuzberg-ffi/src/plugins/mod.rs +0 -14
  287. data/vendor/kreuzberg-ffi/src/plugins/ocr_backend.rs +0 -628
  288. data/vendor/kreuzberg-ffi/src/plugins/post_processor.rs +0 -438
  289. data/vendor/kreuzberg-ffi/src/plugins/validator.rs +0 -329
  290. data/vendor/kreuzberg-ffi/src/result.rs +0 -510
  291. data/vendor/kreuzberg-ffi/src/result_pool.rs +0 -639
  292. data/vendor/kreuzberg-ffi/src/result_view.rs +0 -773
  293. data/vendor/kreuzberg-ffi/src/string_intern.rs +0 -568
  294. data/vendor/kreuzberg-ffi/src/types.rs +0 -363
  295. data/vendor/kreuzberg-ffi/src/util.rs +0 -210
  296. data/vendor/kreuzberg-ffi/src/validation.rs +0 -848
  297. data/vendor/kreuzberg-ffi/tests.disabled/README.md +0 -48
  298. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +0 -299
  299. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +0 -346
  300. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +0 -232
  301. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +0 -470
  302. data/vendor/kreuzberg-tesseract/.commitlintrc.json +0 -13
  303. data/vendor/kreuzberg-tesseract/.crate-ignore +0 -2
  304. data/vendor/kreuzberg-tesseract/Cargo.lock +0 -2933
  305. data/vendor/kreuzberg-tesseract/Cargo.toml +0 -57
  306. data/vendor/kreuzberg-tesseract/LICENSE +0 -22
  307. data/vendor/kreuzberg-tesseract/README.md +0 -399
  308. data/vendor/kreuzberg-tesseract/build.rs +0 -1127
  309. data/vendor/kreuzberg-tesseract/patches/README.md +0 -71
  310. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +0 -199
  311. data/vendor/kreuzberg-tesseract/src/api.rs +0 -1371
  312. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +0 -77
  313. data/vendor/kreuzberg-tesseract/src/enums.rs +0 -297
  314. data/vendor/kreuzberg-tesseract/src/error.rs +0 -81
  315. data/vendor/kreuzberg-tesseract/src/lib.rs +0 -145
  316. data/vendor/kreuzberg-tesseract/src/monitor.rs +0 -57
  317. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +0 -197
  318. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +0 -253
  319. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +0 -286
  320. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +0 -183
  321. data/vendor/kreuzberg-tesseract/tests/integration_test.rs +0 -211
@@ -1,286 +0,0 @@
1
- use crate::api::TessDeleteText;
2
- use crate::enums::TessPageIteratorLevel;
3
- use crate::error::{Result, TesseractError};
4
- use std::ffi::CStr;
5
- use std::os::raw::{c_char, c_float, c_int, c_void};
6
- use std::sync::{Arc, Mutex};
7
-
8
- pub struct ResultIterator {
9
- pub handle: Arc<Mutex<*mut c_void>>,
10
- }
11
-
12
- unsafe impl Send for ResultIterator {}
13
- unsafe impl Sync for ResultIterator {}
14
-
15
- impl ResultIterator {
16
- /// Creates a new instance of the ResultIterator.
17
- ///
18
- /// # Arguments
19
- ///
20
- /// * `handle` - Pointer to the ResultIterator.
21
- ///
22
- /// # Returns
23
- ///
24
- /// Returns the new instance of the ResultIterator.
25
- pub fn new(handle: *mut c_void) -> Self {
26
- ResultIterator {
27
- handle: Arc::new(Mutex::new(handle)),
28
- }
29
- }
30
-
31
- /// Gets the UTF-8 text of the current iterator.
32
- ///
33
- /// # Arguments
34
- ///
35
- /// * `level` - Level of the text.
36
- ///
37
- /// # Returns
38
- ///
39
- /// Returns the UTF-8 text as a `String` if successful, otherwise returns an error.
40
- pub fn get_utf8_text(&self, level: TessPageIteratorLevel) -> Result<String> {
41
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
42
- let text_ptr = unsafe { TessResultIteratorGetUTF8Text(*handle, level as c_int) };
43
- if text_ptr.is_null() {
44
- return Err(TesseractError::NullPointerError);
45
- }
46
- let c_str = unsafe { CStr::from_ptr(text_ptr) };
47
- let result = c_str.to_str()?.to_owned();
48
- unsafe { TessDeleteText(text_ptr as *mut c_char) };
49
- Ok(result)
50
- }
51
-
52
- /// Gets the confidence of the current iterator.
53
- ///
54
- /// # Arguments
55
- ///
56
- /// * `level` - Level of the confidence.
57
- ///
58
- /// # Returns
59
- ///
60
- /// Returns the confidence as a `f32`.
61
- pub fn confidence(&self, level: TessPageIteratorLevel) -> Result<f32> {
62
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
63
- Ok(unsafe { TessResultIteratorConfidence(*handle, level as c_int) })
64
- }
65
-
66
- /// Gets the recognition language of the current iterator.
67
- ///
68
- /// # Returns
69
- ///
70
- /// Returns the recognition language as a `String` if successful, otherwise returns an error.
71
- pub fn word_recognition_language(&self) -> Result<String> {
72
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
73
- let lang_ptr = unsafe { TessResultIteratorWordRecognitionLanguage(*handle) };
74
- if lang_ptr.is_null() {
75
- return Err(TesseractError::NullPointerError);
76
- }
77
- let c_str = unsafe { CStr::from_ptr(lang_ptr) };
78
- Ok(c_str.to_str()?.to_owned())
79
- }
80
-
81
- /// Gets the font attributes of the current iterator.
82
- ///
83
- /// # Returns
84
- ///
85
- /// Returns the font attributes as a tuple if successful, otherwise returns an error.
86
- pub fn word_font_attributes(&self) -> Result<(bool, bool, bool, bool, bool, bool, i32, i32)> {
87
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
88
- let mut is_bold = 0;
89
- let mut is_italic = 0;
90
- let mut is_underlined = 0;
91
- let mut is_monospace = 0;
92
- let mut is_serif = 0;
93
- let mut is_smallcaps = 0;
94
- let mut pointsize = 0;
95
- let mut font_id = 0;
96
-
97
- let result = unsafe {
98
- TessResultIteratorWordFontAttributes(
99
- *handle,
100
- &mut is_bold,
101
- &mut is_italic,
102
- &mut is_underlined,
103
- &mut is_monospace,
104
- &mut is_serif,
105
- &mut is_smallcaps,
106
- &mut pointsize,
107
- &mut font_id,
108
- )
109
- };
110
-
111
- if result == 0 {
112
- Err(TesseractError::InvalidParameterError)
113
- } else {
114
- Ok((
115
- is_bold != 0,
116
- is_italic != 0,
117
- is_underlined != 0,
118
- is_monospace != 0,
119
- is_serif != 0,
120
- is_smallcaps != 0,
121
- pointsize,
122
- font_id,
123
- ))
124
- }
125
- }
126
-
127
- /// Checks if the current iterator is from the dictionary.
128
- ///
129
- /// # Returns
130
- ///
131
- /// Returns `true` if the current iterator is from the dictionary, otherwise returns `false`.
132
- pub fn word_is_from_dictionary(&self) -> Result<bool> {
133
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
134
- Ok(unsafe { TessResultIteratorWordIsFromDictionary(*handle) != 0 })
135
- }
136
-
137
- /// Checks if the current iterator is numeric.
138
- ///
139
- /// # Returns
140
- ///
141
- /// Returns `true` if the current iterator is numeric, otherwise returns `false`.
142
- pub fn word_is_numeric(&self) -> Result<bool> {
143
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
144
- Ok(unsafe { TessResultIteratorWordIsNumeric(*handle) != 0 })
145
- }
146
-
147
- /// Checks if the current iterator is superscript.
148
- ///
149
- /// # Returns
150
- ///
151
- /// Returns `true` if the current iterator is superscript, otherwise returns `false`.
152
- pub fn symbol_is_superscript(&self) -> Result<bool> {
153
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
154
- Ok(unsafe { TessResultIteratorSymbolIsSuperscript(*handle) != 0 })
155
- }
156
-
157
- /// Checks if the current iterator is subscript.
158
- ///
159
- /// # Returns
160
- ///
161
- /// Returns `true` if the current iterator is subscript, otherwise returns `false`.
162
- pub fn symbol_is_subscript(&self) -> Result<bool> {
163
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
164
- Ok(unsafe { TessResultIteratorSymbolIsSubscript(*handle) != 0 })
165
- }
166
-
167
- /// Checks if the current iterator is dropcap.
168
- ///
169
- /// # Returns
170
- ///
171
- /// Returns `true` if the current iterator is dropcap, otherwise returns `false`.
172
- pub fn symbol_is_dropcap(&self) -> Result<bool> {
173
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
174
- Ok(unsafe { TessResultIteratorSymbolIsDropcap(*handle) != 0 })
175
- }
176
-
177
- /// Moves to the next iterator.
178
- ///
179
- /// # Arguments
180
- ///
181
- /// * `level` - Level of the next iterator.
182
- ///
183
- /// # Returns
184
- ///
185
- /// Returns `true` if the next iterator exists, otherwise returns `false`.
186
- pub fn next(&self, level: TessPageIteratorLevel) -> Result<bool> {
187
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
188
- Ok(unsafe { TessResultIteratorNext(*handle, level as c_int) != 0 })
189
- }
190
-
191
- /// Gets the current word from the iterator with its bounding box and confidence.
192
- ///
193
- /// # Returns
194
- ///
195
- /// Returns a tuple of (text, left, top, right, bottom, confidence) if successful
196
- pub fn get_word_with_bounds(&self) -> Result<(String, i32, i32, i32, i32, f32)> {
197
- let text = self.get_utf8_text(TessPageIteratorLevel::RIL_WORD)?;
198
- let (left, top, right, bottom) = self.get_bounding_box(TessPageIteratorLevel::RIL_WORD)?;
199
- let confidence = self.confidence(TessPageIteratorLevel::RIL_WORD)?;
200
-
201
- Ok((text, left, top, right, bottom, confidence))
202
- }
203
-
204
- /// Advances the iterator to the next word.
205
- ///
206
- /// # Returns
207
- ///
208
- /// Returns true if successful, false if there are no more words
209
- pub fn next_word(&self) -> Result<bool> {
210
- self.next(TessPageIteratorLevel::RIL_WORD)
211
- }
212
-
213
- /// Gets the word information for the current position in the iterator.
214
- /// Should be called before next() to ensure valid data.
215
- ///
216
- /// # Returns
217
- /// Returns a tuple of (text, left, top, right, bottom, confidence) if successful
218
- pub fn get_current_word(&self) -> Result<(String, i32, i32, i32, i32, f32)> {
219
- let text = self.get_utf8_text(TessPageIteratorLevel::RIL_WORD)?;
220
- let (left, top, right, bottom) = self.get_bounding_box(TessPageIteratorLevel::RIL_WORD)?;
221
- let confidence = self.confidence(TessPageIteratorLevel::RIL_WORD)?;
222
-
223
- Ok((text, left, top, right, bottom, confidence))
224
- }
225
-
226
- /// Gets the bounding box for the current element.
227
- pub fn get_bounding_box(&self, level: TessPageIteratorLevel) -> Result<(i32, i32, i32, i32)> {
228
- let mut left = 0;
229
- let mut top = 0;
230
- let mut right = 0;
231
- let mut bottom = 0;
232
-
233
- let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
234
-
235
- let result = unsafe {
236
- TessPageIteratorBoundingBox(*handle, level as c_int, &mut left, &mut top, &mut right, &mut bottom)
237
- };
238
-
239
- if result == 0 {
240
- Err(TesseractError::InvalidParameterError)
241
- } else {
242
- Ok((left, top, right, bottom))
243
- }
244
- }
245
- }
246
-
247
- impl Drop for ResultIterator {
248
- fn drop(&mut self) {
249
- if let Ok(handle) = self.handle.lock() {
250
- unsafe { TessResultIteratorDelete(*handle) };
251
- }
252
- }
253
- }
254
-
255
- #[cfg(feature = "build-tesseract")]
256
- unsafe extern "C" {
257
- pub fn TessResultIteratorDelete(handle: *mut c_void);
258
- pub fn TessResultIteratorGetUTF8Text(handle: *mut c_void, level: c_int) -> *mut c_char;
259
- pub fn TessResultIteratorConfidence(handle: *mut c_void, level: c_int) -> c_float;
260
- pub fn TessResultIteratorWordRecognitionLanguage(handle: *mut c_void) -> *const c_char;
261
- pub fn TessResultIteratorWordFontAttributes(
262
- handle: *mut c_void,
263
- is_bold: *mut c_int,
264
- is_italic: *mut c_int,
265
- is_underlined: *mut c_int,
266
- is_monospace: *mut c_int,
267
- is_serif: *mut c_int,
268
- is_smallcaps: *mut c_int,
269
- pointsize: *mut c_int,
270
- font_id: *mut c_int,
271
- ) -> c_int;
272
- pub fn TessResultIteratorWordIsFromDictionary(handle: *mut c_void) -> c_int;
273
- pub fn TessResultIteratorWordIsNumeric(handle: *mut c_void) -> c_int;
274
- pub fn TessResultIteratorSymbolIsSuperscript(handle: *mut c_void) -> c_int;
275
- pub fn TessResultIteratorSymbolIsSubscript(handle: *mut c_void) -> c_int;
276
- pub fn TessResultIteratorSymbolIsDropcap(handle: *mut c_void) -> c_int;
277
- pub fn TessResultIteratorNext(handle: *mut c_void, level: c_int) -> c_int;
278
- pub fn TessPageIteratorBoundingBox(
279
- handle: *mut c_void,
280
- level: c_int,
281
- left: *mut c_int,
282
- top: *mut c_int,
283
- right: *mut c_int,
284
- bottom: *mut c_int,
285
- ) -> c_int;
286
- }
@@ -1,183 +0,0 @@
1
- use crate::TesseractAPI;
2
- use crate::error::{Result, TesseractError};
3
- use std::ffi::{CStr, CString};
4
- use std::os::raw::{c_char, c_int, c_void};
5
- use std::sync::Arc;
6
- use std::sync::Mutex;
7
-
8
- pub struct TessResultRenderer {
9
- handle: Arc<Mutex<*mut c_void>>,
10
- }
11
-
12
- unsafe impl Send for TessResultRenderer {}
13
- unsafe impl Sync for TessResultRenderer {}
14
-
15
- impl TessResultRenderer {
16
- /// Creates a new instance of the TessResultRenderer.
17
- ///
18
- /// # Arguments
19
- ///
20
- /// * `outputbase` - Output base path.
21
- ///
22
- /// # Returns
23
- ///
24
- /// Returns the new instance of the TessResultRenderer.
25
- pub fn new_text_renderer(outputbase: &str) -> Result<Self> {
26
- let outputbase = CString::new(outputbase).unwrap();
27
- let handle = unsafe { TessTextRendererCreate(outputbase.as_ptr()) };
28
- if handle.is_null() {
29
- Err(TesseractError::NullPointerError)
30
- } else {
31
- Ok(TessResultRenderer {
32
- handle: Arc::new(Mutex::new(handle)),
33
- })
34
- }
35
- }
36
-
37
- /// Creates a new instance of the TessResultRenderer for HOCR.
38
- ///
39
- /// # Arguments
40
- ///
41
- /// * `outputbase` - Output base path.
42
- ///
43
- /// # Returns
44
- ///
45
- /// Returns the new instance of the TessResultRenderer.
46
- pub fn new_hocr_renderer(outputbase: &str) -> Result<Self> {
47
- let outputbase = CString::new(outputbase).unwrap();
48
- let handle = unsafe { TessHOcrRendererCreate(outputbase.as_ptr()) };
49
- if handle.is_null() {
50
- Err(TesseractError::NullPointerError)
51
- } else {
52
- Ok(TessResultRenderer {
53
- handle: Arc::new(Mutex::new(handle)),
54
- })
55
- }
56
- }
57
-
58
- /// Creates a new instance of the TessResultRenderer for PDF.
59
- ///
60
- /// # Arguments
61
- ///
62
- /// * `outputbase` - Output base path.
63
- /// * `datadir` - Data directory path.
64
- /// * `textonly` - Whether to include text only.
65
- ///
66
- /// # Returns
67
- ///
68
- /// Returns the new instance of the TessResultRenderer.
69
- pub fn new_pdf_renderer(outputbase: &str, datadir: &str, textonly: bool) -> Result<Self> {
70
- let outputbase = CString::new(outputbase).unwrap();
71
- let datadir = CString::new(datadir).unwrap();
72
- let handle = unsafe { TessPDFRendererCreate(outputbase.as_ptr(), datadir.as_ptr(), textonly as c_int) };
73
- if handle.is_null() {
74
- Err(TesseractError::NullPointerError)
75
- } else {
76
- Ok(TessResultRenderer {
77
- handle: Arc::new(Mutex::new(handle)),
78
- })
79
- }
80
- }
81
-
82
- /// Begins a new document.
83
- ///
84
- /// # Arguments
85
- ///
86
- /// * `title` - Title of the document.
87
- ///
88
- /// # Returns
89
- ///
90
- /// Returns `true` if the document was created successfully, otherwise returns `false`.
91
- pub fn begin_document(&self, title: &str) -> bool {
92
- let title = CString::new(title).unwrap();
93
- let handle = self.handle.lock().unwrap();
94
- unsafe { TessResultRendererBeginDocument(*handle, title.as_ptr()) != 0 }
95
- }
96
-
97
- /// Adds an image to the document.
98
- ///
99
- /// # Arguments
100
- ///
101
- /// * `api` - The TesseractAPI instance.
102
- ///
103
- /// # Returns
104
- ///
105
- /// Returns `true` if the image was added successfully, otherwise returns `false`.
106
- pub fn add_image(&self, api: &TesseractAPI) -> bool {
107
- let api_handle = api.handle.lock().unwrap();
108
- let handle = self.handle.lock().unwrap();
109
- unsafe { TessResultRendererAddImage(*handle, *api_handle) != 0 }
110
- }
111
-
112
- /// Ends the document.
113
- ///
114
- /// # Returns
115
- ///
116
- /// Returns `true` if the document was ended successfully, otherwise returns `false`.
117
- pub fn end_document(&self) -> bool {
118
- let handle = self.handle.lock().unwrap();
119
- unsafe { TessResultRendererEndDocument(*handle) != 0 }
120
- }
121
-
122
- /// Gets the extension of the document.
123
- ///
124
- /// # Returns
125
- ///
126
- /// Returns the extension as a `String` if successful, otherwise returns an error.
127
- pub fn get_extension(&self) -> Result<String> {
128
- let handle = self.handle.lock().unwrap();
129
- let ext_ptr = unsafe { TessResultRendererExtention(*handle) };
130
- if ext_ptr.is_null() {
131
- Err(TesseractError::NullPointerError)
132
- } else {
133
- let c_str = unsafe { CStr::from_ptr(ext_ptr) };
134
- Ok(c_str.to_str()?.to_owned())
135
- }
136
- }
137
-
138
- /// Gets the title of the document.
139
- ///
140
- /// # Returns
141
- ///
142
- /// Returns the title as a `String` if successful, otherwise returns an error.
143
- pub fn get_title(&self) -> Result<String> {
144
- let handle = self.handle.lock().unwrap();
145
- let title_ptr = unsafe { TessResultRendererTitle(*handle) };
146
- if title_ptr.is_null() {
147
- Err(TesseractError::NullPointerError)
148
- } else {
149
- let c_str = unsafe { CStr::from_ptr(title_ptr) };
150
- Ok(c_str.to_str()?.to_owned())
151
- }
152
- }
153
-
154
- /// Gets the number of images in the document.
155
- ///
156
- /// # Returns
157
- ///
158
- /// Returns the number of images as an `i32`.
159
- pub fn get_image_num(&self) -> i32 {
160
- let handle = self.handle.lock().unwrap();
161
- unsafe { TessResultRendererImageNum(*handle) }
162
- }
163
- }
164
-
165
- impl Drop for TessResultRenderer {
166
- fn drop(&mut self) {
167
- let handle = self.handle.lock().unwrap();
168
- unsafe { TessDeleteResultRenderer(*handle) };
169
- }
170
- }
171
-
172
- unsafe extern "C" {
173
- pub fn TessTextRendererCreate(outputbase: *const c_char) -> *mut c_void;
174
- pub fn TessHOcrRendererCreate(outputbase: *const c_char) -> *mut c_void;
175
- pub fn TessPDFRendererCreate(outputbase: *const c_char, datadir: *const c_char, textonly: c_int) -> *mut c_void;
176
- pub fn TessDeleteResultRenderer(renderer: *mut c_void);
177
- pub fn TessResultRendererBeginDocument(renderer: *mut c_void, title: *const c_char) -> c_int;
178
- pub fn TessResultRendererAddImage(renderer: *mut c_void, api: *mut c_void) -> c_int;
179
- pub fn TessResultRendererEndDocument(renderer: *mut c_void) -> c_int;
180
- pub fn TessResultRendererExtention(renderer: *mut c_void) -> *const c_char;
181
- pub fn TessResultRendererTitle(renderer: *mut c_void) -> *const c_char;
182
- pub fn TessResultRendererImageNum(renderer: *mut c_void) -> c_int;
183
- }
@@ -1,211 +0,0 @@
1
- use kreuzberg_tesseract::TesseractAPI;
2
- use std::path::{Path, PathBuf};
3
-
4
- fn get_default_tessdata_dir() -> PathBuf {
5
- if cfg!(target_os = "macos") {
6
- let home_dir = std::env::var("HOME").expect("HOME environment variable not set");
7
- PathBuf::from(home_dir)
8
- .join("Library")
9
- .join("Application Support")
10
- .join("tesseract-rs")
11
- .join("tessdata")
12
- } else if cfg!(target_os = "linux") {
13
- let system_paths = [
14
- PathBuf::from("/usr/share/tesseract-ocr/5/tessdata"),
15
- PathBuf::from("/usr/share/tesseract-ocr/tessdata"),
16
- ];
17
- for path in &system_paths {
18
- if path.exists() {
19
- return path.clone();
20
- }
21
- }
22
- let home_dir = std::env::var("HOME").expect("HOME environment variable not set");
23
- PathBuf::from(home_dir).join(".tesseract-rs").join("tessdata")
24
- } else if cfg!(target_os = "windows") {
25
- PathBuf::from(std::env::var("APPDATA").expect("APPDATA environment variable not set"))
26
- .join("tesseract-rs")
27
- .join("tessdata")
28
- } else {
29
- panic!("Unsupported operating system");
30
- }
31
- }
32
-
33
- fn get_tessdata_dir() -> PathBuf {
34
- match std::env::var("TESSDATA_PREFIX") {
35
- Ok(dir) => {
36
- let prefix_path = PathBuf::from(dir);
37
- let tessdata_path = if prefix_path.ends_with("tessdata") {
38
- prefix_path
39
- } else {
40
- prefix_path.join("tessdata")
41
- };
42
- println!("Using TESSDATA_PREFIX directory: {:?}", tessdata_path);
43
- tessdata_path
44
- }
45
- Err(_) => {
46
- let default_dir = get_default_tessdata_dir();
47
- println!("TESSDATA_PREFIX not set, using default directory: {:?}", default_dir);
48
- default_dir
49
- }
50
- }
51
- }
52
-
53
- fn ensure_eng_traineddata_exists(tessdata_dir: &Path) {
54
- let eng_traineddata = tessdata_dir.join("eng.traineddata");
55
- assert!(
56
- eng_traineddata.exists(),
57
- "eng.traineddata not found in {}. Set TESSDATA_PREFIX or install English tessdata.",
58
- tessdata_dir.display()
59
- );
60
- }
61
-
62
- fn repo_root() -> PathBuf {
63
- PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("..").join("..")
64
- }
65
-
66
- fn load_test_image(relative: &str) -> Result<(Vec<u8>, u32, u32), Box<dyn std::error::Error>> {
67
- let mut path = repo_root();
68
- path.push("test_documents");
69
- path.push(relative);
70
-
71
- let img = image::open(&path)
72
- .map_err(|e| format!("Failed to open test image {}: {}", path.display(), e))?
73
- .to_rgb8();
74
- let (width, height) = img.dimensions();
75
- Ok((img.into_raw(), width, height))
76
- }
77
-
78
- #[test]
79
- fn test_ocr_on_hello_world_image() {
80
- let tessdata_dir = get_tessdata_dir();
81
- ensure_eng_traineddata_exists(&tessdata_dir);
82
-
83
- let api = TesseractAPI::new();
84
- api.init(tessdata_dir.to_str().unwrap(), "eng")
85
- .expect("Failed to initialize Tesseract");
86
-
87
- let (image_data, width, height) =
88
- load_test_image("images/test_hello_world.png").expect("Failed to load test image");
89
- api.set_image(&image_data, width as i32, height as i32, 3, 3 * width as i32)
90
- .expect("Failed to set image");
91
-
92
- let text = api.get_utf8_text().expect("Failed to perform OCR");
93
- assert!(
94
- text.to_lowercase().contains("hello"),
95
- "Text does not contain expected word. Found: {}",
96
- text
97
- );
98
- }
99
-
100
- #[test]
101
- fn test_ocr_on_table_image() {
102
- let tessdata_dir = get_tessdata_dir();
103
- ensure_eng_traineddata_exists(&tessdata_dir);
104
-
105
- let api = TesseractAPI::new();
106
- api.init(tessdata_dir.to_str().unwrap(), "eng")
107
- .expect("Failed to initialize Tesseract");
108
- api.set_variable("tessedit_pageseg_mode", "1")
109
- .expect("Failed to set PSM");
110
-
111
- let (image_data, width, height) = load_test_image("tables/simple_table.png").expect("Failed to load test image");
112
- api.set_image(&image_data, width as i32, height as i32, 3, 3 * width as i32)
113
- .expect("Failed to set image");
114
-
115
- let text = api.get_utf8_text().expect("Failed to perform OCR");
116
- let lowercase = text.to_lowercase();
117
- assert!(
118
- lowercase.contains("product") && lowercase.contains("price"),
119
- "Table text missing expected words. Found: {}",
120
- text
121
- );
122
- }
123
-
124
- #[test]
125
- fn test_invalid_language_code() {
126
- let tessdata_dir = get_tessdata_dir();
127
- ensure_eng_traineddata_exists(&tessdata_dir);
128
-
129
- let api = TesseractAPI::new();
130
-
131
- let result = api.init(tessdata_dir.to_str().unwrap(), "invalid_lang");
132
- assert!(result.is_err());
133
- }
134
-
135
- #[test]
136
- fn test_empty_image_data() {
137
- let tessdata_dir = get_tessdata_dir();
138
- ensure_eng_traineddata_exists(&tessdata_dir);
139
-
140
- let api = TesseractAPI::new();
141
- api.init(tessdata_dir.to_str().unwrap(), "eng")
142
- .expect("Failed to initialize Tesseract");
143
-
144
- let empty_data: Vec<u8> = Vec::new();
145
- let res = api.set_image(&empty_data, 100, 100, 3, 300);
146
- assert!(res.is_err());
147
- }
148
-
149
- #[test]
150
- fn test_invalid_image_parameters() {
151
- let tessdata_dir = get_tessdata_dir();
152
- ensure_eng_traineddata_exists(&tessdata_dir);
153
-
154
- let api = TesseractAPI::new();
155
- api.init(tessdata_dir.to_str().unwrap(), "eng")
156
- .expect("Failed to initialize Tesseract");
157
-
158
- let (image_data, width, height) =
159
- load_test_image("images/test_hello_world.png").expect("Failed to load test image");
160
-
161
- let res = api.set_image(&image_data, -1, height as i32, 3, 3 * width as i32);
162
- assert!(res.is_err());
163
-
164
- let res = api.set_image(&image_data, width as i32, 0, 3, 3 * width as i32);
165
- assert!(res.is_err());
166
-
167
- let res = api.set_image(&image_data, width as i32, height as i32, 0, 3 * width as i32);
168
- assert!(res.is_err());
169
-
170
- let res = api.set_image(&image_data, width as i32, height as i32, 3, width as i32);
171
- assert!(res.is_err());
172
- }
173
-
174
- #[test]
175
- fn test_variable_setting() {
176
- let tessdata_dir = get_tessdata_dir();
177
- ensure_eng_traineddata_exists(&tessdata_dir);
178
-
179
- let api = TesseractAPI::new();
180
- api.init(tessdata_dir.to_str().unwrap(), "eng")
181
- .expect("Failed to initialize Tesseract");
182
-
183
- let res = api.set_variable("invalid_variable_name", "1");
184
- assert!(res.is_err());
185
-
186
- let res = api.set_variable("tessedit_char_whitelist", "");
187
- assert!(res.is_ok());
188
-
189
- assert!(api.set_variable("tessedit_pageseg_mode", "1").is_ok());
190
- assert!(api.set_variable("tessedit_ocr_engine_mode", "1").is_ok());
191
- }
192
-
193
- #[test]
194
- fn test_multiple_operations() {
195
- let tessdata_dir = get_tessdata_dir();
196
- ensure_eng_traineddata_exists(&tessdata_dir);
197
-
198
- let api = TesseractAPI::new();
199
- api.init(tessdata_dir.to_str().unwrap(), "eng")
200
- .expect("Failed to initialize Tesseract");
201
-
202
- let (image_data, width, height) =
203
- load_test_image("images/test_hello_world.png").expect("Failed to load test image");
204
-
205
- for _ in 0..3 {
206
- let res = api.set_image(&image_data, width as i32, height as i32, 3, 3 * width as i32);
207
- assert!(res.is_ok());
208
- let text = api.get_utf8_text().expect("Failed to perform OCR");
209
- assert!(!text.is_empty());
210
- }
211
- }