afeld-opencv 0.0.8

data/test/test_cvmat_drawing.rb

@@ -0,0 +1,318 @@
+#!/usr/bin/env ruby
+# -*- mode: ruby; coding: utf-8-unix -*- 
+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_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
+
+

data/test/test_cvmat_dxt.rb

@@ -0,0 +1,147 @@
+#!/usr/bin/env ruby
+# -*- mode: ruby; coding: utf-8-unix -*- 
+require 'test/unit'
+require 'opencv'
+require File.expand_path(File.dirname(__FILE__)) + '/helper'
+
+include OpenCV
+
+# Tests for dft and dct functions of OpenCV::CvMat
+class TestCvMat_dxt < OpenCVTestCase
+  def test_dft_1D
+    n = 32
+    w = 2 * Math::PI / n
+    
+    mat0 = create_cvmat(n, 1, :cv32f, 2) { |j, i, c|
+      s = Math.sin(c * w)
+      CvScalar.new(s, s)
+    }
+
+    mat1 = mat0.dft(:forward)
+    mat2 = mat0.dft(:forward, :scale)
+    mat3 = mat0.dft(:forward, :scale).dft(:inverse)
+    n.times { |j|
+      if j == 1
+        assert_in_delta(n / 2, mat1[j, 0][0], 0.001)
+        assert_in_delta(-n / 2, mat1[j, 0][1], 0.001)
+        assert_in_delta(0.5, mat2[j, 0][0], 0.001)
+        assert_in_delta(-0.5, mat2[j, 0][1], 0.001)
+      elsif j == n - 1
+        assert_in_delta(-n / 2, mat1[j, 0][0], 0.001)
+        assert_in_delta(n / 2, mat1[j, 0][1], 0.001)
+        assert_in_delta(-0.5, mat2[j, 0][0], 0.001)
+        assert_in_delta(0.5, mat2[j, 0][1], 0.001)
+      else
+        assert_in_delta(0, mat1[j, 0][0], 0.001)
+        assert_in_delta(0, mat1[j, 0][1], 0.001)
+        assert_in_delta(0, mat2[j, 0][0], 0.001)
+        assert_in_delta(0, mat2[j, 0][1], 0.001)
+      end
+      assert_in_delta(mat0[j, 0][0], mat3[j, 0][0], 0.001)
+      assert_in_delta(mat0[j, 0][1], mat3[j, 0][1], 0.001)
+    }
+
+    assert_raise(TypeError) {
+      mat0.dft(DUMMY_OBJ)
+    }
+  end
+
+  def test_dft_2D
+    n = 32
+    w = 2 * Math::PI / n
+    c = 0
+    mat0 = CvMat.new(n, n, :cv32f, 2)
+    n.times { |j|
+      s = Math.sin(c * w)
+      n.times { |i|
+        mat0[j, i] = CvScalar.new(s, s)
+      }
+      c += 1
+    }
+
+    mat1 = mat0.dft(:forward)
+    mat2 = mat0.dft(:forward, :scale)
+    mat3 = mat0.dft(:forward, :scale).dft(:inverse)
+    n.times { |j|
+      n.times { |i|
+        if i == 0 and j == 1
+          assert_in_delta(n * n / 2, mat1[j, i][0], 0.001)
+          assert_in_delta(-n * n / 2, mat1[j, i][1], 0.001)
+          assert_in_delta(0.5, mat2[j, i][0], 0.001)
+          assert_in_delta(-0.5, mat2[j, i][1], 0.001)
+        elsif i == 0 and j == n - 1
+          assert_in_delta(-n * n / 2, mat1[j, i][0], 0.001)
+          assert_in_delta(n * n / 2, mat1[j, i][1], 0.001)
+          assert_in_delta(-0.5, mat2[j, i][0], 0.001)
+          assert_in_delta(0.5, mat2[j, i][1], 0.001)
+        else
+          assert_in_delta(0, mat1[j, i][0], 0.001)
+          assert_in_delta(0, mat1[j, i][1], 0.001)
+          assert_in_delta(0, mat2[j, i][0], 0.001)
+          assert_in_delta(0, mat2[j, i][1], 0.001)
+        end
+        assert_in_delta(mat0[j, 0][0], mat3[j, i][0], 0.001)
+        assert_in_delta(mat0[j, 0][1], mat3[j, i][1], 0.001)
+      }
+    }
+  end
+
+  def test_dct_1D
+    n = 8
+    w = 2 * Math::PI / n
+    
+    mat0 = create_cvmat(n, 1, :cv32f, 1) { |j, i, c|
+      s = Math.sin(c * w)
+      CvScalar.new(s)
+    }
+
+    mat1 = mat0.dct
+    mat2 = mat0.dct(:forward).dct(:inverse)
+    expected1 = [0, 1.599647, -0.765367, -0.906127, 0, -0.180240, 0, -0.042290]
+    n.times { |j|
+      assert_in_delta(expected1[j], mat1[j, 0][0], 0.001)
+      assert_in_delta(mat0[j, 0][0], mat2[j, 0][0], 0.001)
+    }
+
+    assert_raise(TypeError) {
+      mat0.dct(DUMMY_OBJ)
+    }
+  end
+
+  def test_dct_2D
+    n = 8
+    w = 2 * Math::PI / n
+    c = 0
+    mat0 = CvMat.new(n, n, :cv32f, 1)
+    n.times { |j|
+      s = Math.sin(c * w)
+      n.times { |i|
+        mat0[j, i] = CvScalar.new(s, s)
+      }
+      c += 1
+    }
+
+    mat1 = mat0.dct(:forward)
+    mat2 = mat0.dct(:forward).dct(:inverse)
+    n.times { |j|
+      n.times { |i|
+        if i == 0 and j == 1
+          assert_in_delta(4.524486, mat1[j, i][0], 0.001)
+        elsif i == 0 and j == 2
+          assert_in_delta(-2.164784, mat1[j, i][0], 0.001)
+        elsif i == 0 and j == 3
+          assert_in_delta(-2.562915, mat1[j, i][0], 0.001)
+        elsif i == 0 and j == 5
+          assert_in_delta(-0.509796, mat1[j, i][0], 0.001)
+        elsif i == 0 and j == 7
+          assert_in_delta(-0.119615, mat1[j, i][0], 0.001)
+        else
+          assert_in_delta(0, mat1[j, i][0], 0.001)
+        end
+        assert_in_delta(mat0[j, i][0], mat2[j, i][0], 0.001)
+      }
+    }
+  end
+end
+
+