kreuzberg 4.0.0.pre.rc.7 → 4.0.0.pre.rc.11

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 (76) hide show
  1. checksums.yaml +4 -4
  2. data/Gemfile.lock +13 -12
  3. data/README.md +22 -0
  4. data/ext/kreuzberg_rb/native/.cargo/config.toml +1 -1
  5. data/ext/kreuzberg_rb/native/Cargo.lock +397 -183
  6. data/ext/kreuzberg_rb/native/Cargo.toml +3 -3
  7. data/ext/kreuzberg_rb/native/src/lib.rs +36 -13
  8. data/kreuzberg.gemspec +34 -2
  9. data/lib/kreuzberg/cache_api.rb +35 -0
  10. data/lib/kreuzberg/error_context.rb +49 -1
  11. data/lib/kreuzberg/extraction_api.rb +255 -0
  12. data/lib/kreuzberg/version.rb +1 -1
  13. data/lib/kreuzberg.rb +6 -0
  14. data/lib/libpdfium.dylib +0 -0
  15. data/sig/kreuzberg.rbs +9 -0
  16. data/vendor/Cargo.toml +44 -0
  17. data/vendor/kreuzberg/Cargo.toml +65 -35
  18. data/vendor/kreuzberg/README.md +50 -0
  19. data/vendor/kreuzberg/build.rs +548 -190
  20. data/vendor/kreuzberg/src/api/mod.rs +0 -2
  21. data/vendor/kreuzberg/src/core/pipeline.rs +13 -0
  22. data/vendor/kreuzberg/src/embeddings.rs +71 -3
  23. data/vendor/kreuzberg/src/error.rs +1 -1
  24. data/vendor/kreuzberg/src/extraction/docx.rs +1 -1
  25. data/vendor/kreuzberg/src/extraction/html.rs +37 -5
  26. data/vendor/kreuzberg/src/extractors/pdf.rs +99 -47
  27. data/vendor/kreuzberg/src/mcp/mod.rs +3 -2
  28. data/vendor/kreuzberg/src/mcp/server.rs +106 -0
  29. data/vendor/kreuzberg/src/pdf/bindings.rs +44 -0
  30. data/vendor/kreuzberg/src/pdf/bundled.rs +346 -0
  31. data/vendor/kreuzberg/src/pdf/metadata.rs +2 -2
  32. data/vendor/kreuzberg/src/pdf/mod.rs +6 -0
  33. data/vendor/kreuzberg/src/pdf/rendering.rs +2 -2
  34. data/vendor/kreuzberg/src/pdf/table.rs +3 -0
  35. data/vendor/kreuzberg/src/pdf/text.rs +2 -2
  36. data/vendor/kreuzberg/src/text/quality_processor.rs +1 -1
  37. data/vendor/kreuzberg/tests/concurrency_stress.rs +1 -1
  38. data/vendor/kreuzberg/tests/format_integration.rs +4 -1
  39. data/vendor/kreuzberg/tests/pdfium_linking.rs +374 -0
  40. data/vendor/kreuzberg-ffi/Cargo.toml +63 -0
  41. data/vendor/kreuzberg-ffi/README.md +851 -0
  42. data/vendor/kreuzberg-ffi/build.rs +176 -0
  43. data/vendor/kreuzberg-ffi/cbindgen.toml +27 -0
  44. data/vendor/kreuzberg-ffi/kreuzberg-ffi-install.pc +12 -0
  45. data/vendor/kreuzberg-ffi/kreuzberg-ffi.pc.in +12 -0
  46. data/vendor/kreuzberg-ffi/kreuzberg.h +1087 -0
  47. data/vendor/kreuzberg-ffi/src/lib.rs +3616 -0
  48. data/vendor/kreuzberg-ffi/src/panic_shield.rs +247 -0
  49. data/vendor/kreuzberg-ffi/tests.disabled/README.md +48 -0
  50. data/vendor/kreuzberg-ffi/tests.disabled/config_loading_tests.rs +299 -0
  51. data/vendor/kreuzberg-ffi/tests.disabled/config_tests.rs +346 -0
  52. data/vendor/kreuzberg-ffi/tests.disabled/extractor_tests.rs +232 -0
  53. data/vendor/kreuzberg-ffi/tests.disabled/plugin_registration_tests.rs +470 -0
  54. data/vendor/kreuzberg-tesseract/.commitlintrc.json +13 -0
  55. data/vendor/kreuzberg-tesseract/.crate-ignore +2 -0
  56. data/vendor/kreuzberg-tesseract/Cargo.lock +2933 -0
  57. data/vendor/kreuzberg-tesseract/Cargo.toml +48 -0
  58. data/vendor/kreuzberg-tesseract/LICENSE +22 -0
  59. data/vendor/kreuzberg-tesseract/README.md +399 -0
  60. data/vendor/kreuzberg-tesseract/build.rs +1354 -0
  61. data/vendor/kreuzberg-tesseract/patches/README.md +71 -0
  62. data/vendor/kreuzberg-tesseract/patches/tesseract.diff +199 -0
  63. data/vendor/kreuzberg-tesseract/src/api.rs +1371 -0
  64. data/vendor/kreuzberg-tesseract/src/choice_iterator.rs +77 -0
  65. data/vendor/kreuzberg-tesseract/src/enums.rs +297 -0
  66. data/vendor/kreuzberg-tesseract/src/error.rs +81 -0
  67. data/vendor/kreuzberg-tesseract/src/lib.rs +145 -0
  68. data/vendor/kreuzberg-tesseract/src/monitor.rs +57 -0
  69. data/vendor/kreuzberg-tesseract/src/mutable_iterator.rs +197 -0
  70. data/vendor/kreuzberg-tesseract/src/page_iterator.rs +253 -0
  71. data/vendor/kreuzberg-tesseract/src/result_iterator.rs +286 -0
  72. data/vendor/kreuzberg-tesseract/src/result_renderer.rs +183 -0
  73. data/vendor/kreuzberg-tesseract/tests/integration_test.rs +211 -0
  74. data/vendor/rb-sys/src/lib.rs +1 -0
  75. metadata +41 -3
  76. data/vendor/rb-sys/bin/release.sh +0 -22
@@ -0,0 +1,77 @@
1
+ use crate::api::TessDeleteText;
2
+ use crate::error::{Result, TesseractError};
3
+ use std::ffi::CStr;
4
+ use std::os::raw::{c_char, c_float, c_int, c_void};
5
+ use std::sync::{Arc, Mutex};
6
+
7
+ pub struct ChoiceIterator {
8
+ handle: Arc<Mutex<*mut c_void>>,
9
+ }
10
+
11
+ unsafe impl Send for ChoiceIterator {}
12
+ unsafe impl Sync for ChoiceIterator {}
13
+
14
+ impl ChoiceIterator {
15
+ /// Creates a new instance of the ChoiceIterator.
16
+ ///
17
+ /// # Arguments
18
+ ///
19
+ /// * `handle` - Pointer to the ChoiceIterator.
20
+ pub fn new(handle: *mut c_void) -> Self {
21
+ ChoiceIterator {
22
+ handle: Arc::new(Mutex::new(handle)),
23
+ }
24
+ }
25
+
26
+ /// Gets the next choice.
27
+ ///
28
+ /// # Returns
29
+ ///
30
+ /// Returns `true` if the next choice is successful, otherwise returns `false`.
31
+ pub fn next(&self) -> Result<bool> {
32
+ let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
33
+ Ok(unsafe { TessChoiceIteratorNext(*handle) != 0 })
34
+ }
35
+
36
+ /// Gets the UTF-8 text for the current choice.
37
+ ///
38
+ /// # Returns
39
+ ///
40
+ /// Returns the UTF-8 text as a `String` if successful, otherwise returns an error.
41
+ pub fn get_utf8_text(&self) -> Result<String> {
42
+ let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
43
+ let text_ptr = unsafe { TessChoiceIteratorGetUTF8Text(*handle) };
44
+ if text_ptr.is_null() {
45
+ return Err(TesseractError::NullPointerError);
46
+ }
47
+ let c_str = unsafe { CStr::from_ptr(text_ptr) };
48
+ let result = c_str.to_str()?.to_owned();
49
+ unsafe { TessDeleteText(text_ptr) };
50
+ Ok(result)
51
+ }
52
+
53
+ /// Gets the confidence of the current choice.
54
+ ///
55
+ /// # Returns
56
+ ///
57
+ /// Returns the confidence as a `f32`.
58
+ pub fn confidence(&self) -> Result<f32> {
59
+ let handle = self.handle.lock().map_err(|_| TesseractError::MutexLockError)?;
60
+ Ok(unsafe { TessChoiceIteratorConfidence(*handle) })
61
+ }
62
+ }
63
+
64
+ impl Drop for ChoiceIterator {
65
+ fn drop(&mut self) {
66
+ if let Ok(handle) = self.handle.lock() {
67
+ unsafe { TessChoiceIteratorDelete(*handle) };
68
+ }
69
+ }
70
+ }
71
+
72
+ unsafe extern "C" {
73
+ fn TessChoiceIteratorDelete(handle: *mut c_void);
74
+ fn TessChoiceIteratorNext(handle: *mut c_void) -> c_int;
75
+ fn TessChoiceIteratorGetUTF8Text(handle: *mut c_void) -> *mut c_char;
76
+ fn TessChoiceIteratorConfidence(handle: *mut c_void) -> c_float;
77
+ }
@@ -0,0 +1,297 @@
1
+ #[repr(C)]
2
+ #[derive(Debug, Clone, Copy, PartialEq)]
3
+ #[allow(non_camel_case_types)]
4
+ pub enum TessPageSegMode {
5
+ PSM_OSD_ONLY = 0,
6
+ PSM_AUTO_OSD = 1,
7
+ PSM_AUTO_ONLY = 2,
8
+ PSM_AUTO = 3,
9
+ PSM_SINGLE_COLUMN = 4,
10
+ PSM_SINGLE_BLOCK_VERT_TEXT = 5,
11
+ PSM_SINGLE_BLOCK = 6,
12
+ PSM_SINGLE_LINE = 7,
13
+ PSM_SINGLE_WORD = 8,
14
+ PSM_CIRCLE_WORD = 9,
15
+ PSM_SINGLE_CHAR = 10,
16
+ PSM_SPARSE_TEXT = 11,
17
+ PSM_SPARSE_TEXT_OSD = 12,
18
+ PSM_RAW_LINE = 13,
19
+ PSM_COUNT = 14,
20
+ }
21
+
22
+ impl TessPageSegMode {
23
+ pub fn from_int(value: i32) -> Self {
24
+ match value {
25
+ 0 => TessPageSegMode::PSM_OSD_ONLY,
26
+ 1 => TessPageSegMode::PSM_AUTO_OSD,
27
+ 2 => TessPageSegMode::PSM_AUTO_ONLY,
28
+ 3 => TessPageSegMode::PSM_AUTO,
29
+ 4 => TessPageSegMode::PSM_SINGLE_COLUMN,
30
+ 5 => TessPageSegMode::PSM_SINGLE_BLOCK_VERT_TEXT,
31
+ 6 => TessPageSegMode::PSM_SINGLE_BLOCK,
32
+ 7 => TessPageSegMode::PSM_SINGLE_LINE,
33
+ 8 => TessPageSegMode::PSM_SINGLE_WORD,
34
+ 9 => TessPageSegMode::PSM_CIRCLE_WORD,
35
+ 10 => TessPageSegMode::PSM_SINGLE_CHAR,
36
+ 11 => TessPageSegMode::PSM_SPARSE_TEXT,
37
+ 12 => TessPageSegMode::PSM_SPARSE_TEXT_OSD,
38
+ 13 => TessPageSegMode::PSM_RAW_LINE,
39
+ 14 => TessPageSegMode::PSM_COUNT,
40
+ _ => TessPageSegMode::PSM_AUTO,
41
+ }
42
+ }
43
+ }
44
+
45
+ #[repr(C)]
46
+ #[derive(Debug, Clone, Copy, PartialEq)]
47
+ #[allow(non_camel_case_types)]
48
+ pub enum TessPageIteratorLevel {
49
+ RIL_BLOCK = 0,
50
+ RIL_PARA = 1,
51
+ RIL_TEXTLINE = 2,
52
+ RIL_WORD = 3,
53
+ RIL_SYMBOL = 4,
54
+ }
55
+
56
+ impl TessPageIteratorLevel {
57
+ pub fn from_int(value: i32) -> Self {
58
+ match value {
59
+ 0 => TessPageIteratorLevel::RIL_BLOCK,
60
+ 1 => TessPageIteratorLevel::RIL_PARA,
61
+ 2 => TessPageIteratorLevel::RIL_TEXTLINE,
62
+ 3 => TessPageIteratorLevel::RIL_WORD,
63
+ 4 => TessPageIteratorLevel::RIL_SYMBOL,
64
+ _ => TessPageIteratorLevel::RIL_BLOCK,
65
+ }
66
+ }
67
+ }
68
+
69
+ #[repr(C)]
70
+ #[derive(Debug, Clone, Copy, PartialEq)]
71
+ #[allow(non_camel_case_types)]
72
+ pub enum TessPolyBlockType {
73
+ PT_UNKNOWN = 0,
74
+ PT_FLOWING_TEXT = 1,
75
+ PT_HEADING_TEXT = 2,
76
+ PT_PULLOUT_TEXT = 3,
77
+ PT_EQUATION = 4,
78
+ PT_INLINE_EQUATION = 5,
79
+ PT_TABLE = 6,
80
+ PT_VERTICAL_TEXT = 7,
81
+ PT_CAPTION_TEXT = 8,
82
+ PT_FLOWING_IMAGE = 9,
83
+ PT_HEADING_IMAGE = 10,
84
+ PT_PULLOUT_IMAGE = 11,
85
+ PT_HORZ_LINE = 12,
86
+ PT_VERT_LINE = 13,
87
+ PT_NOISE = 14,
88
+ PT_COUNT = 15,
89
+ }
90
+
91
+ impl TessPolyBlockType {
92
+ pub fn from_int(value: i32) -> Self {
93
+ match value {
94
+ 0 => TessPolyBlockType::PT_UNKNOWN,
95
+ 1 => TessPolyBlockType::PT_FLOWING_TEXT,
96
+ 2 => TessPolyBlockType::PT_HEADING_TEXT,
97
+ 3 => TessPolyBlockType::PT_PULLOUT_TEXT,
98
+ 4 => TessPolyBlockType::PT_EQUATION,
99
+ 5 => TessPolyBlockType::PT_INLINE_EQUATION,
100
+ 6 => TessPolyBlockType::PT_TABLE,
101
+ 7 => TessPolyBlockType::PT_VERTICAL_TEXT,
102
+ 8 => TessPolyBlockType::PT_CAPTION_TEXT,
103
+ 9 => TessPolyBlockType::PT_FLOWING_IMAGE,
104
+ 10 => TessPolyBlockType::PT_HEADING_IMAGE,
105
+ 11 => TessPolyBlockType::PT_PULLOUT_IMAGE,
106
+ 12 => TessPolyBlockType::PT_HORZ_LINE,
107
+ 13 => TessPolyBlockType::PT_VERT_LINE,
108
+ 14 => TessPolyBlockType::PT_NOISE,
109
+ 15 => TessPolyBlockType::PT_COUNT,
110
+ _ => TessPolyBlockType::PT_UNKNOWN,
111
+ }
112
+ }
113
+ }
114
+
115
+ #[repr(C)]
116
+ #[derive(Debug, Clone, Copy, PartialEq)]
117
+ #[allow(non_camel_case_types)]
118
+ pub enum TessOrientation {
119
+ ORIENTATION_PAGE_UP = 0,
120
+ ORIENTATION_PAGE_RIGHT = 1,
121
+ ORIENTATION_PAGE_DOWN = 2,
122
+ ORIENTATION_PAGE_LEFT = 3,
123
+ }
124
+
125
+ impl TessOrientation {
126
+ pub fn from_int(value: i32) -> Self {
127
+ match value {
128
+ 0 => TessOrientation::ORIENTATION_PAGE_UP,
129
+ 1 => TessOrientation::ORIENTATION_PAGE_RIGHT,
130
+ 2 => TessOrientation::ORIENTATION_PAGE_DOWN,
131
+ 3 => TessOrientation::ORIENTATION_PAGE_LEFT,
132
+ _ => TessOrientation::ORIENTATION_PAGE_UP,
133
+ }
134
+ }
135
+ }
136
+
137
+ #[repr(C)]
138
+ #[derive(Debug, Clone, Copy, PartialEq)]
139
+ #[allow(non_camel_case_types)]
140
+ pub enum TessParagraphJustification {
141
+ JUSTIFICATION_UNKNOWN = 0,
142
+ JUSTIFICATION_LEFT = 1,
143
+ JUSTIFICATION_CENTER = 2,
144
+ JUSTIFICATION_RIGHT = 3,
145
+ }
146
+
147
+ impl TessParagraphJustification {
148
+ pub fn from_int(value: i32) -> Self {
149
+ match value {
150
+ 0 => TessParagraphJustification::JUSTIFICATION_UNKNOWN,
151
+ 1 => TessParagraphJustification::JUSTIFICATION_LEFT,
152
+ 2 => TessParagraphJustification::JUSTIFICATION_CENTER,
153
+ 3 => TessParagraphJustification::JUSTIFICATION_RIGHT,
154
+ _ => TessParagraphJustification::JUSTIFICATION_UNKNOWN,
155
+ }
156
+ }
157
+ }
158
+
159
+ #[repr(C)]
160
+ #[derive(Debug, Clone, Copy, PartialEq)]
161
+ #[allow(non_camel_case_types)]
162
+ pub enum TessWritingDirection {
163
+ WRITING_DIRECTION_LEFT_TO_RIGHT = 0,
164
+ WRITING_DIRECTION_RIGHT_TO_LEFT = 1,
165
+ WRITING_DIRECTION_TOP_TO_BOTTOM = 2,
166
+ }
167
+
168
+ impl TessWritingDirection {
169
+ pub fn from_int(value: i32) -> Self {
170
+ match value {
171
+ 0 => TessWritingDirection::WRITING_DIRECTION_LEFT_TO_RIGHT,
172
+ 1 => TessWritingDirection::WRITING_DIRECTION_RIGHT_TO_LEFT,
173
+ 2 => TessWritingDirection::WRITING_DIRECTION_TOP_TO_BOTTOM,
174
+ _ => TessWritingDirection::WRITING_DIRECTION_LEFT_TO_RIGHT,
175
+ }
176
+ }
177
+ }
178
+
179
+ #[repr(C)]
180
+ #[derive(Debug, Clone, Copy, PartialEq)]
181
+ #[allow(non_camel_case_types)]
182
+ pub enum TessTextlineOrder {
183
+ TEXTLINE_ORDER_LEFT_TO_RIGHT = 0,
184
+ TEXTLINE_ORDER_RIGHT_TO_LEFT = 1,
185
+ TEXTLINE_ORDER_TOP_TO_BOTTOM = 2,
186
+ }
187
+
188
+ impl TessTextlineOrder {
189
+ pub fn from_int(value: i32) -> Self {
190
+ match value {
191
+ 0 => TessTextlineOrder::TEXTLINE_ORDER_LEFT_TO_RIGHT,
192
+ 1 => TessTextlineOrder::TEXTLINE_ORDER_RIGHT_TO_LEFT,
193
+ 2 => TessTextlineOrder::TEXTLINE_ORDER_TOP_TO_BOTTOM,
194
+ _ => TessTextlineOrder::TEXTLINE_ORDER_LEFT_TO_RIGHT,
195
+ }
196
+ }
197
+ }
198
+
199
+ #[cfg(test)]
200
+ mod tests {
201
+ use super::*;
202
+
203
+ #[test]
204
+ fn test_page_seg_mode_from_int() {
205
+ assert_eq!(TessPageSegMode::from_int(0), TessPageSegMode::PSM_OSD_ONLY);
206
+ assert_eq!(TessPageSegMode::from_int(3), TessPageSegMode::PSM_AUTO);
207
+ assert_eq!(TessPageSegMode::from_int(10), TessPageSegMode::PSM_SINGLE_CHAR);
208
+ assert_eq!(TessPageSegMode::from_int(999), TessPageSegMode::PSM_AUTO);
209
+ }
210
+
211
+ #[test]
212
+ fn test_page_seg_mode_conversion() {
213
+ let mode = TessPageSegMode::PSM_SINGLE_LINE;
214
+ assert_eq!(mode as i32, 7);
215
+ }
216
+
217
+ #[test]
218
+ fn test_page_iterator_level_from_int() {
219
+ assert_eq!(TessPageIteratorLevel::from_int(0), TessPageIteratorLevel::RIL_BLOCK);
220
+ assert_eq!(TessPageIteratorLevel::from_int(3), TessPageIteratorLevel::RIL_WORD);
221
+ assert_eq!(TessPageIteratorLevel::from_int(-1), TessPageIteratorLevel::RIL_BLOCK);
222
+ }
223
+
224
+ #[test]
225
+ fn test_poly_block_type_from_int() {
226
+ assert_eq!(TessPolyBlockType::from_int(1), TessPolyBlockType::PT_FLOWING_TEXT);
227
+ assert_eq!(TessPolyBlockType::from_int(6), TessPolyBlockType::PT_TABLE);
228
+ assert_eq!(TessPolyBlockType::from_int(100), TessPolyBlockType::PT_UNKNOWN);
229
+ }
230
+
231
+ #[test]
232
+ fn test_orientation_from_int() {
233
+ assert_eq!(TessOrientation::from_int(0), TessOrientation::ORIENTATION_PAGE_UP);
234
+ assert_eq!(TessOrientation::from_int(2), TessOrientation::ORIENTATION_PAGE_DOWN);
235
+ assert_eq!(TessOrientation::from_int(5), TessOrientation::ORIENTATION_PAGE_UP);
236
+ }
237
+
238
+ #[test]
239
+ fn test_paragraph_justification_from_int() {
240
+ assert_eq!(
241
+ TessParagraphJustification::from_int(1),
242
+ TessParagraphJustification::JUSTIFICATION_LEFT
243
+ );
244
+ assert_eq!(
245
+ TessParagraphJustification::from_int(3),
246
+ TessParagraphJustification::JUSTIFICATION_RIGHT
247
+ );
248
+ assert_eq!(
249
+ TessParagraphJustification::from_int(-1),
250
+ TessParagraphJustification::JUSTIFICATION_UNKNOWN
251
+ );
252
+ }
253
+
254
+ #[test]
255
+ fn test_writing_direction_from_int() {
256
+ assert_eq!(
257
+ TessWritingDirection::from_int(0),
258
+ TessWritingDirection::WRITING_DIRECTION_LEFT_TO_RIGHT
259
+ );
260
+ assert_eq!(
261
+ TessWritingDirection::from_int(1),
262
+ TessWritingDirection::WRITING_DIRECTION_RIGHT_TO_LEFT
263
+ );
264
+ assert_eq!(
265
+ TessWritingDirection::from_int(10),
266
+ TessWritingDirection::WRITING_DIRECTION_LEFT_TO_RIGHT
267
+ );
268
+ }
269
+
270
+ #[test]
271
+ fn test_textline_order_from_int() {
272
+ assert_eq!(
273
+ TessTextlineOrder::from_int(0),
274
+ TessTextlineOrder::TEXTLINE_ORDER_LEFT_TO_RIGHT
275
+ );
276
+ assert_eq!(
277
+ TessTextlineOrder::from_int(2),
278
+ TessTextlineOrder::TEXTLINE_ORDER_TOP_TO_BOTTOM
279
+ );
280
+ assert_eq!(
281
+ TessTextlineOrder::from_int(99),
282
+ TessTextlineOrder::TEXTLINE_ORDER_LEFT_TO_RIGHT
283
+ );
284
+ }
285
+
286
+ #[test]
287
+ fn test_enums_are_copy() {
288
+ fn assert_copy<T: Copy>() {}
289
+ assert_copy::<TessPageSegMode>();
290
+ assert_copy::<TessPageIteratorLevel>();
291
+ assert_copy::<TessPolyBlockType>();
292
+ assert_copy::<TessOrientation>();
293
+ assert_copy::<TessParagraphJustification>();
294
+ assert_copy::<TessWritingDirection>();
295
+ assert_copy::<TessTextlineOrder>();
296
+ }
297
+ }
@@ -0,0 +1,81 @@
1
+ use std::str::Utf8Error;
2
+ use thiserror::Error;
3
+
4
+ /// Errors that can occur when using the Tesseract API.
5
+ #[derive(Error, Debug)]
6
+ pub enum TesseractError {
7
+ #[error("Failed to initialize Tesseract")]
8
+ InitError,
9
+ #[error("Failed to set image")]
10
+ SetImageError,
11
+ #[error("OCR operation failed")]
12
+ OcrError,
13
+ #[error("Invalid UTF-8 in Tesseract output")]
14
+ Utf8Error(#[from] Utf8Error),
15
+ #[error("Failed to lock mutex")]
16
+ MutexLockError,
17
+ #[error("Failed to set variable")]
18
+ SetVariableError,
19
+ #[error("Failed to get variable")]
20
+ GetVariableError,
21
+ #[error("Null pointer error")]
22
+ NullPointerError,
23
+ #[error("Invalid parameter")]
24
+ InvalidParameterError,
25
+ #[error("Layout analysis failed")]
26
+ AnalyseLayoutError,
27
+ #[error("Page processing failed")]
28
+ ProcessPagesError,
29
+ #[error("I/O error")]
30
+ IoError,
31
+ #[error("Mutex error")]
32
+ MutexError,
33
+ #[error("Invalid dimensions")]
34
+ InvalidDimensions,
35
+ #[error("Invalid bytes per pixel")]
36
+ InvalidBytesPerPixel,
37
+ #[error("Invalid bytes per line")]
38
+ InvalidBytesPerLine,
39
+ #[error("Invalid image data")]
40
+ InvalidImageData,
41
+ #[error("Uninitialized error")]
42
+ UninitializedError,
43
+ }
44
+
45
+ /// Result type for Tesseract operations.
46
+ pub type Result<T> = std::result::Result<T, TesseractError>;
47
+
48
+ #[cfg(test)]
49
+ mod tests {
50
+ use super::*;
51
+
52
+ #[test]
53
+ fn test_error_display() {
54
+ let error = TesseractError::InitError;
55
+ assert_eq!(error.to_string(), "Failed to initialize Tesseract");
56
+
57
+ let error = TesseractError::SetImageError;
58
+ assert_eq!(error.to_string(), "Failed to set image");
59
+
60
+ let error = TesseractError::OcrError;
61
+ assert_eq!(error.to_string(), "OCR operation failed");
62
+ }
63
+
64
+ #[test]
65
+ fn test_utf8_error_conversion() {
66
+ let invalid_utf8 = vec![0xFF, 0xFE];
67
+ let utf8_error = std::str::from_utf8(&invalid_utf8).unwrap_err();
68
+ let tess_error: TesseractError = utf8_error.into();
69
+
70
+ match tess_error {
71
+ TesseractError::Utf8Error(_) => {}
72
+ _ => panic!("Expected Utf8Error variant"),
73
+ }
74
+ }
75
+
76
+ #[test]
77
+ fn test_error_is_send_sync() {
78
+ fn assert_send_sync<T: Send + Sync>() {}
79
+ assert_send_sync::<TesseractError>();
80
+ }
81
+ }
@@ -0,0 +1,145 @@
1
+ #![cfg_attr(not(feature = "build-tesseract"), allow(unused_variables, dead_code))]
2
+ #![allow(clippy::arc_with_non_send_sync)]
3
+ #![allow(clippy::missing_transmute_annotations)]
4
+ #![allow(clippy::type_complexity)]
5
+ #![allow(clippy::new_without_default)]
6
+ #![allow(clippy::not_unsafe_ptr_arg_deref)]
7
+ #![allow(clippy::cmp_null)]
8
+
9
+ //! # tesseract-rs
10
+ //!
11
+ //! `tesseract-rs` provides safe Rust bindings for Tesseract OCR with built-in compilation
12
+ //! of Tesseract and Leptonica libraries. This crate aims to make OCR functionality
13
+ //! easily accessible in Rust projects while handling the complexity of interfacing
14
+ //! with the underlying C++ libraries.
15
+ //!
16
+ //! ## Usage
17
+ //!
18
+ //! Here's a basic example of how to use `tesseract-rs`:
19
+ //!
20
+ //! ```rust
21
+ //! use std::path::PathBuf;
22
+ //! use std::error::Error;
23
+ //! use kreuzberg_tesseract::TesseractAPI;
24
+ //!
25
+ //! fn get_default_tessdata_dir() -> PathBuf {
26
+ //! if cfg!(target_os = "macos") {
27
+ //! let home_dir = std::env::var("HOME").expect("HOME environment variable not set");
28
+ //! PathBuf::from(home_dir)
29
+ //! .join("Library")
30
+ //! .join("Application Support")
31
+ //! .join("tesseract-rs")
32
+ //! .join("tessdata")
33
+ //! } else if cfg!(target_os = "linux") {
34
+ //! let home_dir = std::env::var("HOME").expect("HOME environment variable not set");
35
+ //! PathBuf::from(home_dir)
36
+ //! .join(".tesseract-rs")
37
+ //! .join("tessdata")
38
+ //! } else if cfg!(target_os = "windows") {
39
+ //! PathBuf::from(std::env::var("APPDATA").expect("APPDATA environment variable not set"))
40
+ //! .join("tesseract-rs")
41
+ //! .join("tessdata")
42
+ //! } else {
43
+ //! panic!("Unsupported operating system");
44
+ //! }
45
+ //! }
46
+ //!
47
+ //! fn get_tessdata_dir() -> PathBuf {
48
+ //! match std::env::var("TESSDATA_PREFIX") {
49
+ //! Ok(dir) => {
50
+ //! let path = PathBuf::from(dir);
51
+ //! println!("Using TESSDATA_PREFIX directory: {:?}", path);
52
+ //! path
53
+ //! }
54
+ //! Err(_) => {
55
+ //! let default_dir = get_default_tessdata_dir();
56
+ //! println!(
57
+ //! "TESSDATA_PREFIX not set, using default directory: {:?}",
58
+ //! default_dir
59
+ //! );
60
+ //! default_dir
61
+ //! }
62
+ //! }
63
+ //! }
64
+ //!
65
+ //! fn main() -> Result<(), Box<dyn Error>> {
66
+ //! let api = TesseractAPI::new();
67
+ //!
68
+ //! // Get tessdata directory (uses default location or TESSDATA_PREFIX if set)
69
+ //! let tessdata_dir = get_tessdata_dir();
70
+ //! api.init(tessdata_dir.to_str().unwrap(), "eng")?;
71
+ //!
72
+ //! let width = 24;
73
+ //! let height = 24;
74
+ //! let bytes_per_pixel = 1;
75
+ //! let bytes_per_line = width * bytes_per_pixel;
76
+ //!
77
+ //! // Initialize image data with all white pixels
78
+ //! let mut image_data = vec![255u8; width * height];
79
+ //!
80
+ //! // Draw number 9 with clearer distinction
81
+ //! for y in 4..19 {
82
+ //! for x in 7..17 {
83
+ //! // Top bar
84
+ //! if y == 4 && x >= 8 && x <= 15 {
85
+ //! image_data[y * width + x] = 0;
86
+ //! }
87
+ //! // Top curve left side
88
+ //! if y >= 4 && y <= 10 && x == 7 {
89
+ //! image_data[y * width + x] = 0;
90
+ //! }
91
+ //! // Top curve right side
92
+ //! if y >= 4 && y <= 11 && x == 16 {
93
+ //! image_data[y * width + x] = 0;
94
+ //! }
95
+ //! // Middle bar
96
+ //! if y == 11 && x >= 8 && x <= 15 {
97
+ //! image_data[y * width + x] = 0;
98
+ //! }
99
+ //! // Bottom right vertical line
100
+ //! if y >= 11 && y <= 18 && x == 16 {
101
+ //! image_data[y * width + x] = 0;
102
+ //! }
103
+ //! // Bottom bar
104
+ //! if y == 18 && x >= 8 && x <= 15 {
105
+ //! image_data[y * width + x] = 0;
106
+ //! }
107
+ //! }
108
+ //! }
109
+ //!
110
+ //! // Set the image data
111
+ //! api.set_image(&image_data, width.try_into().unwrap(), height.try_into().unwrap(), bytes_per_pixel.try_into().unwrap(), bytes_per_line.try_into().unwrap())?;
112
+ //!
113
+ //! // Set whitelist for digits only
114
+ //! api.set_variable("tessedit_char_whitelist", "0123456789")?;
115
+ //!
116
+ //! // Set PSM mode to single character
117
+ //! api.set_variable("tessedit_pageseg_mode", "10")?;
118
+ //!
119
+ //! // Get the recognized text
120
+ //! let text = api.get_utf8_text()?;
121
+ //! println!("Recognized text: {}", text.trim());
122
+ //!
123
+ //! Ok(())
124
+ //! }
125
+ //! ```
126
+ pub use error::{Result, TesseractError};
127
+ mod error;
128
+ mod page_iterator;
129
+ pub use page_iterator::PageIterator;
130
+ mod result_iterator;
131
+ pub use result_iterator::ResultIterator;
132
+ mod choice_iterator;
133
+ pub use choice_iterator::ChoiceIterator;
134
+ mod monitor;
135
+ pub use monitor::TessMonitor;
136
+ mod result_renderer;
137
+ pub use result_renderer::TessResultRenderer;
138
+ mod mutable_iterator;
139
+ pub use mutable_iterator::MutableIterator;
140
+ mod enums;
141
+ pub use enums::{
142
+ TessOrientation, TessPageIteratorLevel, TessPageSegMode, TessPolyBlockType, TessTextlineOrder, TessWritingDirection,
143
+ };
144
+ mod api;
145
+ pub use api::TesseractAPI;
@@ -0,0 +1,57 @@
1
+ use std::os::raw::{c_int, c_void};
2
+ use std::sync::{Arc, Mutex};
3
+
4
+ pub struct TessMonitor {
5
+ handle: Arc<Mutex<*mut c_void>>,
6
+ }
7
+
8
+ unsafe impl Send for TessMonitor {}
9
+ unsafe impl Sync for TessMonitor {}
10
+
11
+ impl TessMonitor {
12
+ /// Creates a new instance of the TessMonitor.
13
+ ///
14
+ /// # Returns
15
+ ///
16
+ /// Returns the new instance of the TessMonitor.
17
+ pub fn new() -> Self {
18
+ let handle = unsafe { TessMonitorCreate() };
19
+ TessMonitor {
20
+ handle: Arc::new(Mutex::new(handle)),
21
+ }
22
+ }
23
+
24
+ /// Sets the deadline for the monitor.
25
+ ///
26
+ /// # Arguments
27
+ ///
28
+ /// * `deadline` - Deadline in milliseconds.
29
+ pub fn set_deadline(&self, deadline: i32) {
30
+ let handle = self.handle.lock().unwrap();
31
+ unsafe { TessMonitorSetDeadlineMSecs(*handle, deadline) };
32
+ }
33
+
34
+ /// Gets the progress of the monitor.
35
+ ///
36
+ /// # Returns
37
+ ///
38
+ /// Returns the progress as an `i32`.
39
+ pub fn get_progress(&self) -> i32 {
40
+ let handle = self.handle.lock().unwrap();
41
+ unsafe { TessMonitorGetProgress(*handle) }
42
+ }
43
+ }
44
+
45
+ impl Drop for TessMonitor {
46
+ fn drop(&mut self) {
47
+ let handle = self.handle.lock().unwrap();
48
+ unsafe { TessMonitorDelete(*handle) };
49
+ }
50
+ }
51
+
52
+ unsafe extern "C" {
53
+ pub fn TessMonitorCreate() -> *mut c_void;
54
+ pub fn TessMonitorDelete(monitor: *mut c_void);
55
+ pub fn TessMonitorSetDeadlineMSecs(monitor: *mut c_void, deadline: c_int);
56
+ pub fn TessMonitorGetProgress(monitor: *mut c_void) -> c_int;
57
+ }