Class: Sqed::Extractor

Inherits:
Object
  • Object
show all
Defined in:
lib/sqed/extractor.rb

Overview

An Extractor takes Boundaries object and a metadata_map and returns a Sqed::Result

Defined Under Namespace

Classes: Error

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(**opts) ⇒ Extractor

Returns a new instance of Extractor.

Raises:



21
22
23
24
25
26
27
28
29
 
# File 'lib/sqed/extractor.rb', line 21

def initialize(**opts)
  @metadata_map = opts[:metadata_map]
  @boundaries = opts[:boundaries]
  @image = opts[:image]

  raise Error, 'boundaries not provided or provided boundary is not a Sqed::Boundaries' if boundaries.nil? || !boundaries.class == Sqed::Boundaries
  raise Error, 'metadata_map not provided or metadata_map not a Hash' if metadata_map.nil? || !metadata_map.class == Hash
  raise Error, 'image not provided' if image.nil? || !image.class.name == 'Magick::Image'
end

Instance Attribute Details

#boundaries ⇒ Object

a Sqed::Boundaries instance



12
13
14
 
# File 'lib/sqed/extractor.rb', line 12

def boundaries
  @boundaries
end

#image ⇒ Magick::Image file

Returns:

  • (Magick::Image file) 


19
20
21
 
# File 'lib/sqed/extractor.rb', line 19

def image
  @image
end

#metadata_map ⇒ Hash

a metadata_map hash from EXTRACTION_PATTERNS like:

Returns:

  • (Hash) —

    like ‘{0 => :annotated_specimen, 1 => :identifier, 2 => :image_registration }`



16
17
18
 
# File 'lib/sqed/extractor.rb', line 16

def metadata_map
  @metadata_map
end

Instance Method Details

#extract_image(coords) ⇒ Object

crop takes x, y, width, height



64
65
66
 
# File 'lib/sqed/extractor.rb', line 64

def extract_image(coords)
  @image.crop(*coords, true)
end

#result ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/sqed/extractor.rb', line 31

def result
  r = Sqed::Result.new

  r.sections = metadata_map.values.sort

  # assign the images to the result
  boundaries.each do |section_index, coords|
    section_type = metadata_map[section_index]

    r.send("#{section_type}_image=", extract_image(coords))
    r.boundary_coordinates[section_type] = coords
  end

  # assign the metadata to the result
  metadata_map.each do |section_index, section_type|
    # only extract data if a parser exists
    if parsers = SqedConfig::SECTION_PARSERS[section_type]
      section_image = r.send("#{section_type}_image")
      updated = r.send(section_type)

      parsers.each do |p|
        parsed_result = p.new(section_image).get_text(section_type: section_type)
        updated[p::TYPE] = parsed_result if parsed_result && parsed_result.length > 0
      end

      r.send("#{section_type}=", updated)
    end
  end

  r
end