percept 0.0.1 → 3.0.0

data/spec/percept/line_detector/detect_lines_spec.rb

@@ -0,0 +1,93 @@
+describe Percept::LineDetector, '#detect_lines' do
+
+  let(:line_detector) { Percept::LineDetector.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  it 'returns a collection of lines with the appropriate coordinates' do
+    image = image_class.new(File.join(fixture_path, 'low_line.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 1
+    line = lines.first
+    expect(line.length).to eq 32
+    expect(line.start_y).to eq 14
+    expect(line.end_y).to eq 15
+    expect(line.start_x).to eq 4
+    expect(line.end_x).to eq 36
+  end
+
+  it 'detects lines with divots' do
+    image = image_class.new(File.join(fixture_path, 'divots.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 1
+    line = lines.first
+    expect(line.length).to eq 32
+    expect(line.start_y).to eq 11
+    expect(line.end_y).to eq 14
+    expect(line.start_x).to eq 4
+    expect(line.end_x).to eq 36
+  end
+
+  it 'detects lines with nubs' do
+    image = image_class.new(File.join(fixture_path, 'nubs.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 1
+    line = lines.first
+    expect(line.length).to eq 32
+    expect(line.start_y).to eq 12
+    expect(line.end_y).to eq 14
+    expect(line.start_x).to eq 4
+    expect(line.end_x).to eq 36
+  end
+
+  it 'detects lines that are sloped to the left' do
+    image = image_class.new(File.join(fixture_path, 'slanted_left.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 1
+    line = lines.first
+    expect(line.length).to eq 91
+    expect(line.start_y).to eq 21
+    expect(line.end_y).to eq 23
+    expect(line.start_x).to eq 5
+    expect(line.end_x).to eq 96
+  end
+
+  it 'detects lines that are sloped to the right' do
+    image = image_class.new(File.join(fixture_path, 'slanted_right.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 1
+    line = lines.first
+    expect(line.length).to eq 91
+    expect(line.start_y).to eq 21
+    expect(line.end_y).to eq 23
+    expect(line.start_x).to eq 5
+    expect(line.end_x).to eq 96
+  end
+
+  it 'detects lines in images with gray filler' do
+    image = image_class.new(File.join(fixture_path, 'gray_image_section.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 2
+    line = lines.first
+    expect(line.length).to eq 180
+    expect(line.start_y).to eq 16
+    expect(line.end_y).to eq 19
+    expect(line.start_x).to eq 20
+    expect(line.end_x).to eq 200
+    line = lines.last
+    expect(line.length).to eq 180
+    expect(line.start_y).to eq 68
+    expect(line.end_y).to eq 72
+    expect(line.start_x).to eq 20
+    expect(line.end_x).to eq 200
+  end
+
+  it 'it detects lines in a lighter gray image' do
+    image = image_class.new(File.join(fixture_path, 'light_gray.png'))
+    lines = line_detector.detect_lines(image)
+    expect(lines.size).to eq 1
+  end
+
+end

data/spec/percept/line_detector/scaled_color_spec.rb

@@ -0,0 +1,14 @@
+describe Percept::LineDetector, '#scaled_color' do
+
+  let(:detector) { Percept::LineDetector.new }
+
+  before(:each) do
+    Percept.config.color_tolerance = 150
+  end
+
+  it 'multiplies the value by 256' do
+    expect(detector.scaled_color(1)).to eq 255
+    expect(detector.scaled_color(256)).to eq 65_535
+  end
+
+end

data/spec/percept/line_filter/filter_line_predicate_spec.rb

@@ -0,0 +1,29 @@
+describe Percept::LineFilter, '#filter_line?' do
+
+  let(:filter) { Percept::LineFilter.new }
+
+  context 'when a line is the tops of letters' do
+    it 'returns true' do
+      image = image_class.new(File.join(fixture_path, 'text_no_line.png'))
+      line = image.lines.first
+      expect(filter.filter_line?(line)).to be true
+    end
+  end
+
+  context 'when a line is the bottom of a table cell' do
+    it 'returns false' do
+      image = image_class.new(File.join(fixture_path, 'gray_cell_bottom.png'))
+      line = image.lines.first
+      expect(filter.filter_line?(line)).to be false
+    end
+  end
+
+  context 'when a line is too thick' do
+    it 'returns true' do
+      image = image_class.new(File.join(fixture_path, 'bumpy_no_line.png'))
+      line = image.lines.first
+      expect(filter.filter_line?(line)).to be true
+    end
+  end
+
+end

data/spec/percept/line_filter/left_bottom_corner_predicate_spec.rb

@@ -0,0 +1,49 @@
+describe Percept::LineFilter, '#left_bottom_corner?' do
+
+  let(:corners_path) { File.join(fixture_path, 'corners.png') }
+  let(:image_1) { image_class.new(corners_path) }
+  let(:lines_1) { image_1.lines }
+
+  let(:more_corners_path) { File.join(fixture_path, 'more_corners.png') }
+  let(:image_2) { image_class.new(more_corners_path) }
+  let(:lines_2) { image_2.lines }
+
+  let(:filter) { Percept::LineFilter.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  context 'when the line is the left bottom corner of a box' do
+    it 'returns true' do
+      expect(filter.left_bottom_corner?(lines_1[5])).to eq true
+      expect(filter.left_bottom_corner?(lines_1[6])).to eq true
+      expect(filter.left_bottom_corner?(lines_1[8])).to eq true
+
+      expect(filter.left_bottom_corner?(lines_2[8])).to eq true
+    end
+  end
+
+  context 'when the line is not the left bottom corner of a box' do
+    it 'returns false' do
+      expect(filter.left_bottom_corner?(lines_1[0])).to eq false
+      expect(filter.left_bottom_corner?(lines_1[1])).to eq false
+      expect(filter.left_bottom_corner?(lines_1[2])).to eq false
+      expect(filter.left_bottom_corner?(lines_1[3])).to eq false
+      expect(filter.left_bottom_corner?(lines_1[4])).to eq false
+      expect(filter.left_bottom_corner?(lines_1[7])).to eq false
+      expect(filter.left_bottom_corner?(lines_1[9])).to eq false
+
+      expect(filter.left_bottom_corner?(lines_2[0])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[1])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[2])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[3])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[4])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[5])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[6])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[7])).to eq false
+      expect(filter.left_bottom_corner?(lines_2[9])).to eq false
+    end
+  end
+
+end

data/spec/percept/line_filter/left_top_corner_predicate_spec.rb

@@ -0,0 +1,50 @@
+describe Percept::LineFilter, '#left_top_corner?' do
+
+  let(:corners_path) { File.join(fixture_path, 'corners.png') }
+  let(:image_1) { image_class.new(corners_path) }
+  let(:lines_1) { image_1.lines }
+
+  let(:more_corners_path) { File.join(fixture_path, 'more_corners.png') }
+  let(:image_2) { image_class.new(more_corners_path) }
+  let(:lines_2) { image_2.lines }
+
+  let(:filter) { Percept::LineFilter.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  context 'when the line is the left top corner of a box' do
+    it 'returns true' do
+      expect(lines_1.size).to eq 10
+      expect(filter.left_top_corner?(lines_1[2])).to eq true
+      expect(filter.left_top_corner?(lines_1[2])).to eq true
+      expect(filter.left_top_corner?(lines_1[3])).to eq true
+    end
+  end
+
+  context 'when the line is not the left top corner of a box' do
+    it 'returns false' do
+      expect(filter.left_top_corner?(lines_1[0])).to eq false
+      expect(filter.left_top_corner?(lines_1[1])).to eq false
+      expect(filter.left_top_corner?(lines_1[4])).to eq false
+      expect(filter.left_top_corner?(lines_1[5])).to eq false
+      expect(filter.left_top_corner?(lines_1[6])).to eq false
+      expect(filter.left_top_corner?(lines_1[7])).to eq false
+      expect(filter.left_top_corner?(lines_1[8])).to eq false
+      expect(filter.left_top_corner?(lines_1[9])).to eq false
+
+      expect(filter.left_top_corner?(lines_2[0])).to eq false
+      expect(filter.left_top_corner?(lines_2[1])).to eq false
+      expect(filter.left_top_corner?(lines_2[2])).to eq false
+      expect(filter.left_top_corner?(lines_2[3])).to eq false
+      expect(filter.left_top_corner?(lines_2[4])).to eq false
+      expect(filter.left_top_corner?(lines_2[5])).to eq false
+      expect(filter.left_top_corner?(lines_2[6])).to eq false
+      expect(filter.left_top_corner?(lines_2[7])).to eq false
+      expect(filter.left_top_corner?(lines_2[8])).to eq false
+      expect(filter.left_top_corner?(lines_2[9])).to eq false
+    end
+  end
+
+end

data/spec/percept/line_filter/right_bottom_corner_predicate_spec.rb

@@ -0,0 +1,49 @@
+describe Percept::LineFilter, '#right_bottom_corner?' do
+
+  let(:corners_path) { File.join(fixture_path, 'corners.png') }
+  let(:image_1) { image_class.new(corners_path) }
+  let(:lines_1) { image_1.lines }
+
+  let(:more_corners_path) { File.join(fixture_path, 'more_corners.png') }
+  let(:image_2) { image_class.new(more_corners_path) }
+  let(:lines_2) { image_2.lines }
+
+  let(:filter) { Percept::LineFilter.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  context 'when the line is the right bottom corner of a box' do
+    it 'returns true' do
+      expect(filter.right_bottom_corner?(lines_1[4])).to eq true
+      expect(filter.right_bottom_corner?(lines_1[6])).to eq true
+      expect(filter.right_bottom_corner?(lines_1[7])).to eq true
+
+      expect(filter.right_bottom_corner?(lines_2[9])).to eq true
+    end
+  end
+
+  context 'when the line is not the right bottom corner of a box' do
+    it 'returns false' do
+      expect(filter.right_bottom_corner?(lines_1[0])).to eq false
+      expect(filter.right_bottom_corner?(lines_1[1])).to eq false
+      expect(filter.right_bottom_corner?(lines_1[2])).to eq false
+      expect(filter.right_bottom_corner?(lines_1[3])).to eq false
+      expect(filter.right_bottom_corner?(lines_1[5])).to eq false
+      expect(filter.right_bottom_corner?(lines_1[8])).to eq false
+      expect(filter.right_bottom_corner?(lines_1[9])).to eq false
+
+      expect(filter.right_bottom_corner?(lines_2[0])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[1])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[2])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[3])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[4])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[5])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[6])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[7])).to eq false
+      expect(filter.right_bottom_corner?(lines_2[8])).to eq false
+    end
+  end
+
+end

data/spec/percept/line_filter/right_top_corner_predicate_spec.rb

@@ -0,0 +1,49 @@
+describe Percept::LineFilter, '#right_top_corner?' do
+
+  let(:corners_path) { File.join(fixture_path, 'corners.png') }
+  let(:image_1) { image_class.new(corners_path) }
+  let(:lines_1) { image_1.lines }
+
+  let(:more_corners_path) { File.join(fixture_path, 'more_corners.png') }
+  let(:image_2) { image_class.new(more_corners_path) }
+  let(:lines_2) { image_2.lines }
+
+  let(:filter) { Percept::LineFilter.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  context 'when the line is the right top corner of a box' do
+    it 'returns true' do
+      expect(lines_1.size).to eq 10
+      expect(filter.right_top_corner?(lines_1[1])).to eq true
+      expect(filter.right_top_corner?(lines_1[3])).to eq true
+    end
+  end
+
+  context 'when the line is not the right top corner of a box' do
+    it 'returns false' do
+      expect(filter.right_top_corner?(lines_1[0])).to eq false
+      expect(filter.right_top_corner?(lines_1[2])).to eq false
+      expect(filter.right_top_corner?(lines_1[4])).to eq false
+      expect(filter.right_top_corner?(lines_1[5])).to eq false
+      expect(filter.right_top_corner?(lines_1[6])).to eq false
+      expect(filter.right_top_corner?(lines_1[7])).to eq false
+      expect(filter.right_top_corner?(lines_1[8])).to eq false
+      expect(filter.right_top_corner?(lines_1[9])).to eq false
+
+      expect(filter.right_top_corner?(lines_2[0])).to eq false
+      expect(filter.right_top_corner?(lines_2[1])).to eq false
+      expect(filter.right_top_corner?(lines_2[2])).to eq false
+      expect(filter.right_top_corner?(lines_2[3])).to eq false
+      expect(filter.right_top_corner?(lines_2[4])).to eq false
+      expect(filter.right_top_corner?(lines_2[5])).to eq false
+      expect(filter.right_top_corner?(lines_2[6])).to eq false
+      expect(filter.right_top_corner?(lines_2[7])).to eq false
+      expect(filter.right_top_corner?(lines_2[8])).to eq false
+      expect(filter.right_top_corner?(lines_2[9])).to eq false
+    end
+  end
+
+end

data/spec/percept/line_filter/table_bottom_predicate_spec.rb

@@ -0,0 +1,48 @@
+describe Percept::LineFilter, '#table_bottom?' do
+
+  let(:corners_path) { File.join(fixture_path, 'corners.png') }
+  let(:image_1) { image_class.new(corners_path) }
+  let(:lines_1) { image_1.lines }
+
+  let(:more_corners_path) { File.join(fixture_path, 'more_corners.png') }
+  let(:image_2) { image_class.new(more_corners_path) }
+  let(:lines_2) { image_2.lines }
+
+  let(:filter) { Percept::LineFilter.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  context 'when the line has both left and right bottom corners' do
+    it 'returns true' do
+      expect(filter.table_bottom?(lines_1[6])).to eq true
+    end
+  end
+
+  context 'when the line does not have left and right bottom corners' do
+    it 'returns false' do
+      expect(filter.table_bottom?(lines_1[0])).to eq false
+      expect(filter.table_bottom?(lines_1[1])).to eq false
+      expect(filter.table_bottom?(lines_1[2])).to eq false
+      expect(filter.table_bottom?(lines_1[3])).to eq false
+      expect(filter.table_bottom?(lines_1[4])).to eq false
+      expect(filter.table_bottom?(lines_1[5])).to eq false
+      expect(filter.table_bottom?(lines_1[7])).to eq false
+      expect(filter.table_bottom?(lines_1[8])).to eq false
+      expect(filter.table_bottom?(lines_1[9])).to eq false
+
+      expect(filter.table_bottom?(lines_2[0])).to eq false
+      expect(filter.table_bottom?(lines_2[1])).to eq false
+      expect(filter.table_bottom?(lines_2[2])).to eq false
+      expect(filter.table_bottom?(lines_2[3])).to eq false
+      expect(filter.table_bottom?(lines_2[4])).to eq false
+      expect(filter.table_bottom?(lines_2[5])).to eq false
+      expect(filter.table_bottom?(lines_2[6])).to eq false
+      expect(filter.table_bottom?(lines_2[7])).to eq false
+      expect(filter.table_bottom?(lines_2[8])).to eq false
+      expect(filter.table_bottom?(lines_2[9])).to eq false
+    end
+  end
+
+end

data/spec/percept/line_filter/table_top_predicate_spec.rb

@@ -0,0 +1,48 @@
+describe Percept::LineFilter, '#table_top?' do
+
+  let(:corners_path) { File.join(fixture_path, 'corners.png') }
+  let(:image_1) { image_class.new(corners_path) }
+  let(:lines_1) { image_1.lines }
+
+  let(:more_corners_path) { File.join(fixture_path, 'more_corners.png') }
+  let(:image_2) { image_class.new(more_corners_path) }
+  let(:lines_2) { image_2.lines }
+
+  let(:filter) { Percept::LineFilter.new }
+
+  before(:each) do
+    Percept.config.line_length = 30
+  end
+
+  context 'when the line has both left and right top corners' do
+    it 'returns true' do
+      expect(filter.table_top?(lines_1[3])).to eq true
+    end
+  end
+
+  context 'when the line does not have left and right top corners' do
+    it 'returns false' do
+      expect(filter.table_top?(lines_1[0])).to eq false
+      expect(filter.table_top?(lines_1[1])).to eq false
+      expect(filter.table_top?(lines_1[2])).to eq false
+      expect(filter.table_top?(lines_1[4])).to eq false
+      expect(filter.table_top?(lines_1[5])).to eq false
+      expect(filter.table_top?(lines_1[6])).to eq false
+      expect(filter.table_top?(lines_1[7])).to eq false
+      expect(filter.table_top?(lines_1[8])).to eq false
+      expect(filter.table_top?(lines_1[9])).to eq false
+
+      expect(filter.table_top?(lines_2[0])).to eq false
+      expect(filter.table_top?(lines_2[1])).to eq false
+      expect(filter.table_top?(lines_2[2])).to eq false
+      expect(filter.table_top?(lines_2[3])).to eq false
+      expect(filter.table_top?(lines_2[4])).to eq false
+      expect(filter.table_top?(lines_2[5])).to eq false
+      expect(filter.table_top?(lines_2[6])).to eq false
+      expect(filter.table_top?(lines_2[7])).to eq false
+      expect(filter.table_top?(lines_2[8])).to eq false
+      expect(filter.table_top?(lines_2[9])).to eq false
+    end
+  end
+
+end

data/spec/percept/pixel/blackish_predicate_spec.rb

@@ -0,0 +1,36 @@
+describe Percept::Pixel, '#blackish?' do
+
+  let(:pixel) { Percept::Pixel.new({ red: 10, green: 10, blue: 10 }) }
+
+  before(:each) do
+    Percept.config.color_tolerance = 150
+  end
+
+  context 'if the given object has all colors below the tolerance' do
+    it 'returns true' do
+      expect(pixel.blackish?).to eq true
+    end
+  end
+
+  context 'if red is greater than the tolerance' do
+    it 'returns false' do
+      expect(pixel).to receive(:red).and_return(45_000)
+      expect(pixel.blackish?).to eq false
+    end
+  end
+
+  context 'if blue is greater than the tolerance' do
+    it 'returns false' do
+      expect(pixel).to receive(:blue).and_return(45_000)
+      expect(pixel.blackish?).to eq false
+    end
+  end
+
+  context 'if green is greater than the tolerance' do
+    it 'returns false' do
+      expect(pixel).to receive(:green).and_return(45_000)
+      expect(pixel.blackish?).to eq false
+    end
+  end
+
+end

data/spec/percept/rmagick_image/interface_spec.rb

@@ -0,0 +1,11 @@
+describe Percept::RMagickImage do
+
+  it_behaves_like 'image interface' do
+    let(:image) { Percept::RMagickImage.new(image_path) }
+
+    before(:each) do
+      Percept.config.image_class = Percept::RMagickImage
+    end
+  end
+
+end

data/spec/percept/runner/color_fields_spec.rb

@@ -0,0 +1,28 @@
+describe Percept::Runner, '#color_fields' do
+
+  let(:image_path) { File.join(fixture_path, 'black_lines.png') }
+  let(:runner) { Percept::Runner.new }
+  let(:image) { image_class.new(image_path) }
+
+  before(:each) do
+    Percept.config.line_length = 30
+    Percept.config.min_field_height = 10
+  end
+
+  def red?(pixel)
+    pixel.red > 0 && pixel.blue == 0 && pixel.green == 0
+  end
+
+  it 'colors a red box above the lines' do
+    black_pixels = image.pixels.select(&:blackish?)
+    runner.color_fields(image_path)
+    line = image.lines.first
+    field = Percept::FieldDetector.new.detect_field(line, image)
+    output_image = image_class.new(output_path)
+    output_black_pixels = output_image.pixels.select(&:blackish?)
+    expect(output_black_pixels.length).to eq black_pixels.length
+    output_red_pixels = output_image.pixels.select { |p| red?(p) }
+    expect(output_red_pixels.length).to eq field.pixels.length
+  end
+
+end

data/spec/percept/runner/color_lines_spec.rb

@@ -0,0 +1,27 @@
+describe Percept::Runner, '#color_lines' do
+
+  let(:image_path) { File.join(fixture_path, 'black_lines.png') }
+  let(:runner) { Percept::Runner.new }
+  let(:image) { image_class.new(image_path) }
+
+  before(:each) do
+    Percept.config.line_length = 30
+    Percept.config.color_tolerance = 150
+  end
+
+  def red?(pixel)
+    pixel.red > 0 && pixel.blue == 0 && pixel.green == 0
+  end
+
+  it 'colors lines red that are greater than or equal to the line length' do
+    black_pixels = image.pixels.select(&:blackish?)
+    runner.color_lines(image_path)
+    output_image = image_class.new(output_path)
+    output_black_pixels = output_image.pixels.select(&:blackish?)
+    output_red_pixels = output_image.pixels.select { |p| red?(p) }
+    expect(output_red_pixels.length).to eq 77
+    expected_length = black_pixels.length - output_red_pixels.length
+    expect(output_black_pixels.length).to eq expected_length
+  end
+
+end