tesseract_ffi 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 618b8a311591353b1d164091aaf4cf793241a4f47eadbd3413a4c44c61c5fe7a
-  data.tar.gz: 33a199c65cfbe8f734c9b103ce9639068b5572ff805f2fa991ec6bd748108315
+  metadata.gz: c5a9c0c8d2c8b8df8ac6796658f09879e752b87a0942310489e52f0eb7e86150
+  data.tar.gz: 72a6acdefeba8c4c021f9f031d2c7523ae1450f883eff21429e36aa4795b610b
 SHA512:
-  metadata.gz: ddda303d8de39d73cf353c5265acaf690ec4551aea60e8ec88c1720cfb38d2f5f637d02db13ed54068e95342c7fa2f5a3f8b5a1b613146ae120e2bf9905ef7af
-  data.tar.gz: 74db6585a8eae9546b7bdd969572c789e548bc465db393af29ed6ceebe47baf2d9c8f7369f5df2d0d2bff2c30ae0214737781bd64d074aedb80c32dd8d8b3001
+  metadata.gz: bdc7ad7ab5f9fd23141097b9b0a924aabcc3891331481563c6f0bdfaba0398688a8954ae31cbdc5b779379546007374c156c98eef6ce067c4ef2ea462e67642b
+  data.tar.gz: 0fc5c10d36c8822d3604f46ed9687adc4cbbb1e4d76229c0a04632af54bf75e4461947c49bffc327fb69728ba1fbededfafe84cdefdda269e1e24cf29eb029ba

data/lib/tesseract_ffi.rb

@@ -2,8 +2,9 @@
 
 require 'ffi'
 require 'tesseract_ffi/version'
-require 'tesseract_ffi/conf_vars' # mix-in to tesseract
-require 'tesseract_ffi/oem' # mix-in to tesseract
+require 'tesseract_ffi/conf_vars'   # mix-in to tesseract
+require 'tesseract_ffi/oem'         # mix-in to tesseract
+require 'tesseract_ffi/rectangles'  # mix-in to tesseract
 require 'tesseract_ffi/tesseract'
 require 'tesseract_ffi/tess_exception'
 require 'tesseract_ffi/quick'

data/lib/tesseract_ffi/rectangles.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+module TesseractFFI
+  # module Rectangles mixin for recognizing text blocks defined by rectangles
+  module Rectangles
+    def set_rectangle(x_coord, y_coord, width, height)
+      tess_set_rectangle(@handle, x_coord, y_coord, width, height)
+    end
+
+    def recognize_rectangle(x_coord, y_coord, width, height)
+      setup do
+        set_rectangle(x_coord, y_coord, width, height)
+        ocr
+      end
+    end
+
+    # rubocop:disable Metrics/MethodLength
+    def recognize_rectangles(rectangle_list)
+      @texts = []
+      if valid_rectangle_list? rectangle_list
+        @rectangle_list = rectangle_list
+        setup do
+          @rectangle_list.each do |r|
+            set_rectangle(r[0], r[1], r[2], r[3])
+            ocr
+            @texts << @utf8_text.strip
+          end
+        end
+      else
+        @rectangle_list = nil
+      end
+      @texts
+    end
+
+    def valid_rectangle_list?(list)
+      if list.is_a?(Array) && list.all? { |r| valid_rectangle?(r) }
+        true
+      else
+        msg = 'Tess Error Argument must be a list'
+        # copy the error message as we are not going to Setup
+        @errors << msg
+        raise TessException.new(error_msg: msg)
+      end
+    end
+
+    def valid_rectangle?(rectangle)
+      if rectangle.is_a?(Array) &&
+         rectangle.length == 4 &&
+         rectangle.all? { |r| r.is_a?(Integer) }
+        true
+      else
+        msg = 'Tesseract Error Argument must be array of 4 Integer'
+        @errors << msg
+        raise TessException.new(error_msg: msg)
+      end
+    end
+    # rubocop:enable Metrics/MethodLength
+  end
+end

data/lib/tesseract_ffi/tesseract.rb

@@ -6,6 +6,7 @@ module TesseractFFI
     include TesseractFFI
     include ConfVars
     include OEM
+    include Rectangles
 
     attr_accessor :language, :file_name, :source_resolution
     attr_reader :utf8_text, :hocr_text, :errors
@@ -48,9 +49,7 @@ module TesseractFFI
       tess_set_source_resolution(@handle, @source_resolution)
       raise TessException.new(error_msg: 'Recognition Error') if tess_recognize(@handle, 0) != 0
 
-      @utf8_text = ''
-      text = tess_get_utf8(@handle, 0)
-      @utf8_text = text.encode('UTF-8') if text
+      @utf8_text = tess_get_utf8(@handle, 0)
       @hocr_text = tess_get_hocr(@handle, 0)
     end
 
@@ -67,42 +66,5 @@ module TesseractFFI
         TesseractFFI.tess_process_pages(@handle, @file_name, nil, 5000, pdf_renderer)
       end
     end
-
-    def set_rectangle(x_coord, y_coord, width, height)
-      tess_set_rectangle(@handle, x_coord, y_coord, width, height)
-    end
-
-    def recognize_rectangle(x_coord, y_coord, width, height)
-      setup do
-        set_rectangle(x_coord, y_coord, width, height)
-        ocr
-      end
-    end
-
-    # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
-    def recognize_rectangles(rectangle_list)
-      unless rectangle_list.is_a?(Array) && rectangle_list.length.positive?
-        msg = 'Tess Error Argument must be a list'
-        # copy the error message as we are not going to Setup
-        @errors << msg
-        raise TessException.new(error_msg: msg)
-      end
-
-      texts = []
-      setup do
-        rectangle_list.each do |r|
-          unless r.is_a?(Array) && rectangle_list.length > 3
-            msg = 'Argument must be a list of 4-arrays'
-            raise TessException.new(error_msg: msg)
-          end
-
-          set_rectangle(r[0], r[1], r[2], r[3])
-          ocr
-          texts << @utf8_text.strip
-        end
-      end
-      texts
-    end
-    # rubocop:enable Metrics/AbcSize, Metrics/MethodLength
   end
 end

data/lib/tesseract_ffi/version.rb

@@ -2,5 +2,5 @@
 
 # module with version
 module TesseractFFI
-  VERSION = '0.3.0'
+  VERSION = '0.4.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tesseract_ffi
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - David Verrier
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-08-14 00:00:00.000000000 Z
+date: 2020-08-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -121,6 +121,7 @@ files:
 - lib/tesseract_ffi/conf_vars.rb
 - lib/tesseract_ffi/oem.rb
 - lib/tesseract_ffi/quick.rb
+- lib/tesseract_ffi/rectangles.rb
 - lib/tesseract_ffi/tess_exception.rb
 - lib/tesseract_ffi/tesseract.rb
 - lib/tesseract_ffi/version.rb