Class: OCR::Scan

Inherits:
Object
  • Object
show all
Defined in:
lib/simple-ocr/scan.rb

Constant Summary collapse

EXTENS = 
%w{pdf}

Instance Method Summary collapse

Constructor Details

#initialize(input_file, output_file, options, type) ⇒ Scan

Initialize your Input File, Output File, Options, Type.



11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/simple-ocr/scan.rb', line 11

def initialize(input_file, output_file, options, type)
  @output_file = output_file
  @options = options
  @type = handle_output_type(type)
  @input_file = input_file
  if OCR::Path.new(input_file).name_exten[1] == OCR::Path::EXTENS[:pdf]
    @image = OCR::Path.new(input_file).image_path
    convert_to_img
  else
    @image = input_file
  end
  @clean_image = OCR::Path.new(input_file).clean_image_path
end

Instance Method Details

#clean_img ⇒ Object

Shell Script for cleaning the Image.



53
54
55
56
57
 
# File 'lib/simple-ocr/scan.rb', line 53

def clean_img
  name = 'simple-ocr'
  g = Gem::Specification.find_by_name(name)
  `sh #{File.join(g.full_gem_path, 'lib/textcleaner')} -g -e stretch -f 25 -o 20 -t 30 -u -s 1 -T -p 20 '#{@image}' '#{@clean_image}'`
end

#convert_to_img ⇒ Object

Conversion of PDF to Image



36
37
38
 
# File 'lib/simple-ocr/scan.rb', line 36

def convert_to_img
  `gs -sDEVICE=png16m '-r#{OCR::MIN_DENSITY}' -o '#{@image}' '#{@input_file}'`
end

#delete_files ⇒ Object

Deleting unnecessary files after processing.



60
61
62
63
 
# File 'lib/simple-ocr/scan.rb', line 60

def delete_files
  FileUtils.rm_rf(@clean_image)
  FileUtils.rm_rf(@image) if OCR::Path.new(@input_file).name_exten[1] == "pdf"
end

#exec_command(command) ⇒ Object

Execute Command



48
49
50
 
# File 'lib/simple-ocr/scan.rb', line 48

def exec_command(command)
  Open3.popen3(command)
end

#handle_output_type(type) ⇒ Object 



25
26
27
28
29
30
31
32
33
 
# File 'lib/simple-ocr/scan.rb', line 25

def handle_output_type(type)
  if type == :pdf
    "pdf"
  elsif type == :hocr
    "hocr"
  else
    nil.to_s
  end
end

#scan_img ⇒ Object

OCR of Input



41
42
43
44
45
 
# File 'lib/simple-ocr/scan.rb', line 41

def scan_img
  clean_img
  `tesseract '#{@clean_image}' #{@options} '#{@output_file}' #{@type}`
  delete_files
end