Contents

# frozen_string_literal: true

class RTesseract
  module Box
    extend RTesseract::Base

    class << self
      def run(source, errors, options)
        options = options.merge({ tessedit_create_hocr: 1 })

        RTesseract::Command.new(source, temp_file_path, errors, options).run do |output_path|
          parse(File.read("#{output_path}.hocr"))
        end
      end

      def parse(content)
        content.lines.map { |line| parse_line(line) }.compact
      end

      def parse_line(line)
        return unless line.match?(/oc(rx|r)_word/)

        word = line.to_s.scan(/>(.*)</).flatten.first.to_s

        return if word.strip == ''

        word_info(word, parse_position(line), parse_confidence(line))
      end

      def word_info(word, positions, confidence)
        {
          word: word,
          confidence: confidence[-1].to_i,
          x_start: positions[1].to_i,
          y_start: positions[2].to_i,
          x_end: positions[3].to_i,
          y_end: positions[4].to_i
        }
      end

      def parse_position(line)
        line.match(/(?<=title)(.*?)(?=;)/).to_s.split
      end

      def parse_confidence(line)
        line.match(/(?<=;)(.*?)(?=')/).to_s.split
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rtesseract-3.1.3	lib/rtesseract/box.rb