1 files changed, 50 insertions, 0 deletions
diff --git a/include/recognizer.h b/include/recognizer.h
new file mode 100644
index 0000000..57747bb
--- /dev/null
+++ b/include/recognizer.h
@@ -0,0 +1,50 @@
+#pragma once
+#include <iostream>
+#include <leptonica/allheaders.h>
+#include <tesseract/baseapi.h>
+
+#include "./image.h"
+
+// OCR configuration
+#define CONFIDENCE_THRESHOLD 25.
+#define MIN_CHARACTER_COUNT 3
+const tesseract::PageIteratorLevel ITER_LEVEL = tesseract::RIL_WORD;
+
+// NOTE: Remember to update size and alignment in ocr hs module on change
+struct OCRMatch {
+  int startX, startY;
+  int endX, endY;
+  const char *text;
+};
+
+class Recognizer {
+  tesseract::TessBaseAPI *tesseract;
+  int x, y, width, height;
+  bool failed = false;
+
+public:
+  Recognizer(int x, int y, int width, int height)
+      : x(x), y(y), width(width), height(height) {
+    initializeTesseract();
+  }
+
+  ~Recognizer() { tesseract->End(); }
+
+  void fail(const char *msg);
+
+  void recognize(Pix *image);
+
+  OCRMatch *fetchMatch(tesseract::ResultIterator *iterator);
+
+  std::vector<OCRMatch> getResults();
+
+private:
+  void initializeTesseract();
+};
+
+inline void printMatch(const OCRMatch &match) {
+  std::cout << "Text: " << match.text << "; Position: (" << match.startX << ","
+            << match.startY << ") -> (" << match.endX << "," << match.endY
+            << ")" << std::endl
+            << std::endl;
+}