mirror of
https://github.com/ruby-opencv/ruby-opencv
synced 2023-03-27 23:22:12 -04:00
349 lines
12 KiB
Ruby
Executable file
349 lines
12 KiB
Ruby
Executable file
#!/usr/bin/env ruby
|
|
# -*- mode: ruby; coding: utf-8 -*-
|
|
require 'test/unit'
|
|
require 'opencv'
|
|
require File.expand_path(File.dirname(__FILE__)) + '/helper'
|
|
|
|
include OpenCV
|
|
|
|
# Tests for drawing functions of OpenCV::CvMat
|
|
class TestCvMat_drawing < OpenCVTestCase
|
|
def test_DRAWING_OPTION
|
|
CvMat::DRAWING_OPTION[:color].to_ary.each { |c|
|
|
assert_in_delta(0, c, 0.01)
|
|
}
|
|
assert_equal(1, CvMat::DRAWING_OPTION[:thickness])
|
|
assert_equal(8, CvMat::DRAWING_OPTION[:line_type])
|
|
assert_equal(0, CvMat::DRAWING_OPTION[:shift])
|
|
end
|
|
|
|
def test_FLOOD_FILL_OPTION
|
|
assert_equal(4, CvMat::FLOOD_FILL_OPTION[:connectivity])
|
|
assert((not CvMat::FLOOD_FILL_OPTION[:fixed_range]))
|
|
assert((not CvMat::FLOOD_FILL_OPTION[:mask_only]))
|
|
end
|
|
|
|
def test_line
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
m1 = m0.clone
|
|
m2 = m0.line(CvPoint.new(1, 0), CvPoint.new(m0.width - 1, m0.height - 1),
|
|
:color => CvColor::Red, :thickness => 3, :line_type => :aa)
|
|
m1.line!(CvPoint.new(1, 0), CvPoint.new(m0.width - 1, m0.height - 1),
|
|
:color => CvColor::Blue, :thickness => 1, :line_type => :aa)
|
|
|
|
# Uncomment the following line to show the image
|
|
# snap(['Line: Blue, thickness = 1', m1], ['Line: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m0.line(DUMMY_OBJ, CvPoint.new(1, 0))
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.line(CvPoint.new(1, 0), DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.line(CvPoint.new(1, 0), CvPoint.new(1, 1), :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m0.line(CvPoint.new(1, 0), CvPoint.new(1, 1), :thickness => DUMMY_OBJ)
|
|
# }
|
|
# m0.line(CvPoint.new(1, 0), CvPoint.new(1, 1), :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_rectangle
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
m1 = m0.clone
|
|
m2 = m0.rectangle(CvPoint.new(20, 20), CvPoint.new(m0.width - 20, m0.height - 20),
|
|
:color => CvColor::Red, :thickness => 3, :line_type => :aa)
|
|
m1.rectangle!(CvPoint.new(20, 20), CvPoint.new(m0.width - 20, m0.height - 20),
|
|
:color => CvColor::Blue, :thickness => 1, :line_type => :aa)
|
|
|
|
# Uncomment the following line to show the image
|
|
# snap(['Rectangle: Blue, thickness = 1', m1], ['Rectangle: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m0.line(DUMMY_OBJ, CvPoint.new(1, 0))
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.rectangle(CvPoint.new(1, 0), DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.rectangle(CvPoint.new(1, 0), CvPoint.new(1, 1), :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m0.rectangle(CvPoint.new(1, 0), CvPoint.new(1, 1), :thickness => DUMMY_OBJ)
|
|
# }
|
|
# m0.rectangle(CvPoint.new(1, 0), CvPoint.new(1, 1), :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_circle
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
m1 = m0.clone
|
|
m2 = m0.circle(CvPoint.new(m0.width / 2, m0.height / 2), 80,
|
|
:color => CvColor::Red, :thickness => 3, :line_type => :aa)
|
|
m1.circle!(CvPoint.new(m0.width / 2, m0.height / 2), 80,
|
|
:color => CvColor::Blue, :thickness => 1, :line_type => :aa)
|
|
|
|
# Uncomment the following line to show the image
|
|
# snap(['Circle: Blue, thickness = 1', m1], ['Circle: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m0.circle(DUMMY_OBJ, 10)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.circle(CvPoint.new(1, 0), DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.circle(CvPoint.new(1, 0), 10, :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m0.circle(CvPoint.new(1, 0), 10, :thickness => DUMMY_OBJ)
|
|
# }
|
|
m0.circle(CvPoint.new(1, 0), 10, :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_ellipse
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
m1 = m0.clone
|
|
m2 = m0.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, 0, 360,
|
|
:color => CvColor::Red, :thickness => 3, :line_type => :aa)
|
|
m1.ellipse!(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, 0, 360,
|
|
:color => CvColor::Blue, :thickness => 1, :line_type => :aa)
|
|
|
|
# Uncomment the following line to show the image
|
|
# snap(['Ellipse: Blue, thickness = 1', m1], ['Ellipse: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m1.ellipse(DUMMY_OBJ, CvSize.new(100, 60), 30, 0, 360)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), DUMMY_OBJ, 30, 0, 360)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), DUMMY_OBJ, 0, 360)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, DUMMY_OBJ, 360)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, 0, DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, 0, 360,
|
|
:color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, 0, 360,
|
|
# :thickness => DUMMY_OBJ)
|
|
# }
|
|
m1.ellipse(CvPoint.new(m0.width / 2, m0.height / 2), CvSize.new(100, 60), 30, 0, 360,
|
|
:line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_ellipse_box
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
box = CvBox2D.new(CvPoint2D32f.new(m0.width / 2, m0.height / 2), CvSize2D32f.new(120, 160), 30)
|
|
m1 = m0.clone
|
|
m2 = m0.ellipse_box(box, :color => CvColor::Red, :thickness => 3, :line_type => :aa)
|
|
m1.ellipse_box!(box, :color => CvColor::Blue, :thickness => 1, :line_type => :aa)
|
|
|
|
# Uncomment the following line to show the image
|
|
# snap(['Ellipse box: Blue, thickness = 1', m1], ['Ellipse box: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m1.ellipse_box(DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.ellipse_box(box, :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m1.ellipse_box(box, :thickness => DUMMY_OBJ)
|
|
# }
|
|
m1.ellipse_box(box, :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_fill_poly
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
pt = [[CvPoint.new(10, 20), CvPoint.new(10, 150), CvPoint.new(100, 50)],
|
|
[CvPoint.new(200, 10), CvPoint.new(200, 200), CvPoint.new(170, 200)],
|
|
[CvPoint.new(30, 10), CvPoint.new(0, 0), CvPoint.new(90, 150)]]
|
|
|
|
m1 = m0.clone
|
|
m2 = m0.fill_poly(pt, :color => CvColor::Red, :line_type => :aa)
|
|
m1.fill_poly!(pt, :color => CvColor::Blue, :line_type => :aa)
|
|
|
|
# Uncomment the following line to view the image
|
|
# snap(['Fill poly: Blue', m1], ['Fill poly: Red', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m1.fill_poly(DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.fill_poly([DUMMY_OBJ, DUMMY_OBJ])
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.fill_poly(pt, :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m1.fill_poly(pt, :thickness => DUMMY_OBJ)
|
|
# }
|
|
# m1.fill_poly(pt, :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_fill_convex_poly
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
pt = [CvPoint.new(10, 20), CvPoint.new(10, 150), CvPoint.new(100, 50)]
|
|
|
|
m1 = m0.clone
|
|
m2 = m0.fill_convex_poly(pt, :color => CvColor::Red, :line_type => :aa)
|
|
m1.fill_convex_poly!(pt, :color => CvColor::Blue, :line_type => :aa)
|
|
|
|
# Uncomment the following line to view the image
|
|
# snap(['Fill convex poly: Blue', m1], ['Fill convex poly: Red', m2])
|
|
|
|
|
|
assert_raise(TypeError) {
|
|
m1.fill_convex_poly(DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.fill_convex_poly([DUMMY_OBJ, DUMMY_OBJ])
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.fill_convex_poly(pt, :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m1.fill_convex_poly(pt, :thickness => DUMMY_OBJ)
|
|
# }
|
|
# m1.fill_convex_poly(pt, :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_poly_line
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
pt = [[CvPoint.new(10, 20), CvPoint.new(10, 150), CvPoint.new(100, 150), CvPoint.new(10, 20)],
|
|
[CvPoint.new(100, 200), CvPoint.new(200, 190), CvPoint.new(180, 50), CvPoint.new(100, 200)]]
|
|
|
|
m1 = m0.clone
|
|
m2 = m0.poly_line(pt, :color => CvColor::Red, :thickness => 3, :line_type => :aa)
|
|
m1.poly_line!(pt, :color => CvColor::Blue, :thickness => 1, :line_type => :aa)
|
|
|
|
# Uncomment the following line to view the image
|
|
# snap(['Fill poly line: Blue, thickness = 1', m1], ['Fill poly line: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m1.poly_line(DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.poly_line([DUMMY_OBJ, DUMMY_OBJ])
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.poly_line([[DUMMY_OBJ, DUMMY_OBJ], [DUMMY_OBJ, DUMMY_OBJ]])
|
|
}
|
|
assert_raise(TypeError) {
|
|
m1.poly_line(pt, :color => DUMMY_OBJ)
|
|
}
|
|
# assert_raise(CvError) {
|
|
# m1.poly_line(pt, :thickness => DUMMY_OBJ)
|
|
# }
|
|
# m1.poly_line(pt, :line_type => DUMMY_OBJ)
|
|
end
|
|
|
|
def test_draw_contours
|
|
mat0 = CvMat.load(FILENAME_CONTOURS, CV_LOAD_IMAGE_GRAYSCALE)
|
|
|
|
mat0 = mat0.threshold(128, 255, CV_THRESH_BINARY)
|
|
contours = mat0.find_contours(:mode => CV_RETR_TREE, :method => CV_CHAIN_APPROX_SIMPLE)
|
|
dst0 = mat0.clone.clear
|
|
dst1 = mat0.clone.clear.GRAY2BGR
|
|
begin
|
|
dst0 = dst0.draw_contours!(contours, CvColor::Black, CvColor::White, -1)
|
|
dst1.draw_contours!(contours, CvColor::Red, CvColor::Blue, 2,
|
|
:thickness => -1, :line_type => :aa)
|
|
end while (contours = contours.h_next)
|
|
|
|
[dst0, dst1].each { |dst|
|
|
assert_equal(mat0.class, dst.class)
|
|
assert_equal(mat0.rows, dst.rows)
|
|
assert_equal(mat0.cols, dst.cols)
|
|
assert_equal(mat0.depth, dst.depth)
|
|
}
|
|
|
|
assert_raise(TypeError) {
|
|
dst0.draw_contours(DUMMY_OBJ, CvColor::Black, CvColor::White, -1)
|
|
}
|
|
assert_raise(TypeError) {
|
|
dst0.draw_contours(contours, DUMMY_OBJ, CvColor::White, -1)
|
|
}
|
|
assert_raise(TypeError) {
|
|
dst0.draw_contours(contours, CvColor::Black, DUMMY_OBJ, -1)
|
|
}
|
|
assert_raise(TypeError) {
|
|
dst0.draw_contours(contours, CvColor::Black, CvColor::White, DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
dst0.draw_contours(contours, CvColor::Black, CvColor::White, -1, :thickness => DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
dst0.draw_contours(contours, CvColor::Black, CvColor::White, -1, :line_type => DUMMY_OBJ)
|
|
}
|
|
|
|
# Uncomment the following line to show the results
|
|
# snap ['src', mat0], ['result0', dst0], ['result1', dst1]
|
|
end
|
|
|
|
def test_draw_chessboard_corners
|
|
mat0 = CvMat.load(FILENAME_CHESSBOARD, 1)
|
|
mat1 = mat0.clone
|
|
pattern_size = CvSize.new(4, 4)
|
|
|
|
gray = mat1.BGR2GRAY
|
|
corners, found = gray.find_chessboard_corners(pattern_size)
|
|
|
|
mat2 = mat1.draw_chessboard_corners(pattern_size, corners, found)
|
|
mat1.draw_chessboard_corners!(pattern_size, corners, found)
|
|
[mat1, mat2].each { |dst|
|
|
assert_equal(mat0.class, dst.class)
|
|
assert_equal(mat0.rows, dst.rows)
|
|
assert_equal(mat0.cols, dst.cols)
|
|
assert_equal(mat0.depth, dst.depth)
|
|
}
|
|
|
|
assert_raise(TypeError) {
|
|
mat1.draw_chessboard_corners(DUMMY_OBJ, corners, found)
|
|
}
|
|
assert_raise(TypeError) {
|
|
mat1.draw_chessboard_corners(pattern_size, DUMMY_OBJ, found)
|
|
}
|
|
assert_nothing_raised {
|
|
mat1.draw_chessboard_corners(pattern_size, corners, DUMMY_OBJ)
|
|
}
|
|
|
|
# Uncomment the following line to show the results
|
|
# snap mat0, mat1, mat2
|
|
end
|
|
|
|
def test_put_text
|
|
m0 = create_cvmat(240, 320, :cv8u, 3) { CvColor::White }
|
|
m1 = m0.clone
|
|
m1.put_text!('test 1', CvPoint.new(60, 90), CvFont.new(:simplex), CvColor::Blue)
|
|
font = CvFont.new(:plain, :hscale => 5.0, :vscale => 4.5,
|
|
:shear => 1.0, :thickness => 3, :line_type => 5, :italic => true)
|
|
m2 = m0.put_text('test 2', CvPoint.new(30, 80), font, CvColor::Red)
|
|
|
|
# Uncomment the following lines to view the image
|
|
# snap(['Put text: Blue, thickness = 1', m1], ['Put text: Red, thickness = 3', m2])
|
|
|
|
assert_raise(TypeError) {
|
|
m0.put_text(DUMMY_OBJ, CvPoint.new(60, 90), font)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.put_text('test', DUMMY_OBJ, font)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.put_text('test', CvPoint.new(60, 90), DUMMY_OBJ)
|
|
}
|
|
assert_raise(TypeError) {
|
|
m0.put_text('test', CvPoint.new(60, 90), font, DUMMY_OBJ)
|
|
}
|
|
end
|
|
end
|
|
|
|
|