Create a sorted list of ocr matches by position on screen

1 files changed, 12 insertions, 2 deletions

diff --git a/include/libchelleport.h b/include/libchelleport.h
index c022975..d4e202f 100644
--- a/include/libchelleport.h
+++ b/include/libchelleport.h
@@ -19,8 +19,18 @@
 extern "C" OCRMatch *findWordCoordinates(const char *image_path,
                                          /* returns */ int *size);
 
-std::vector<OCRMatch> extractTextMatches(const char *imagePath);
+struct ScreenPositionComparator {
+  bool operator()(const OCRMatch &a, const OCRMatch &b) const {
+    if (abs(a.startY - b.startY) < 5)
+      return a.startX < b.startX;
+    return a.startY < b.startY;
+  }
+};
 
-std::vector<OCRMatch>
+typedef std::set<OCRMatch, ScreenPositionComparator> OCRMatchSet;
+
+OCRMatchSet extractTextMatches(const char *imagePath);
+
+OCRMatchSet
 runRecognizers(std::vector<std::unique_ptr<Recognizer>> &recognizers,
                Pix *image);