ruby-opencv 0.0.9.pre2-mingw32

data/test/test_cvcircle32f.rb

@@ -0,0 +1,41 @@
+#!/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 OpenCV::CvCircle32f
+class TestCvCircle32f < OpenCVTestCase
+  def setup
+    @circle = CvCircle32f.new
+  end
+  
+  def test_initialize
+    assert_equal(CvCircle32f, @circle.class)
+  end
+
+  def test_center
+    assert_equal(CvPoint2D32f, @circle.center.class)
+  end
+
+  def test_radius
+    assert_equal(Float, @circle.radius.class)
+  end
+
+  def test_aref
+    assert_equal(Float, @circle[0].class)
+    assert_equal(Float, @circle[1].class)
+    assert_equal(Float, @circle[2].class)
+    assert_raise(IndexError) {
+      @circle[3]
+    }
+  end
+
+  def test_to_ary
+    assert_equal(Array, @circle.to_ary.class)
+    assert_equal(2, @circle.to_ary.size)
+  end
+end
+

data/test/test_cvconnectedcomp.rb

@@ -0,0 +1,61 @@
+#!/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 OpenCV::CvConnectedComp
+class TestCvConnectedComp < OpenCVTestCase
+  def setup
+    @connected_comp = CvConnectedComp.new(9216, CvScalar.new(1, 2, 3, 4),
+                                          CvRect.new(1, 2, 3, 4), CvSeq.new(CvPoint))
+  end
+  
+  def test_initialize
+    connected_comp = CvConnectedComp.new
+    assert_equal(CvConnectedComp, connected_comp.class)
+    assert_not_nil(connected_comp.area)
+    assert_not_nil(connected_comp.value)
+    assert_not_nil(connected_comp.rect)
+    assert_not_nil(connected_comp.contour)
+
+    connected_comp = CvConnectedComp.new(100, CvScalar.new(1, 2, 3, 4),
+                                         CvRect.new(1, 2, 3, 4), CvSeq.new(CvPoint))
+    assert_equal(CvConnectedComp, connected_comp.class)
+    assert_not_nil(connected_comp.area)
+    assert_not_nil(connected_comp.value)
+    assert_not_nil(connected_comp.rect)
+    assert_not_nil(connected_comp.contour)
+  end
+
+  def test_area
+    assert_in_delta(9216.0, @connected_comp.area, 0.01)
+  end
+
+  def test_value
+    assert_equal(CvScalar, @connected_comp.value.class)
+    assert_cvscalar_equal(CvScalar.new(1, 2, 3, 4), @connected_comp.value)
+  end
+
+  def test_rect
+    assert_equal(CvRect, @connected_comp.rect.class)
+    assert_equal(1, @connected_comp.rect.x)
+    assert_equal(2, @connected_comp.rect.y)
+    assert_equal(3, @connected_comp.rect.width)
+    assert_equal(4, @connected_comp.rect.height)
+
+    @connected_comp.rect = CvRect.new(10, 20, 30, 40);
+    assert_equal(10, @connected_comp.rect.x)
+    assert_equal(20, @connected_comp.rect.y)
+    assert_equal(30, @connected_comp.rect.width)
+    assert_equal(40, @connected_comp.rect.height)
+  end
+
+  def test_contour
+    assert_equal(CvContour, @connected_comp.contour.class)
+    assert_not_nil(@connected_comp.contour)
+  end
+end
+

data/test/test_cvcontour.rb

@@ -0,0 +1,150 @@
+#!/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 OpenCV::CvContour
+class TestCvContour < OpenCVTestCase
+  def test_APPROX_OPTION
+    assert_equal(0, CvContour::APPROX_OPTION[:method])
+    assert_equal(1.0, CvContour::APPROX_OPTION[:accuracy])
+    assert_false(CvContour::APPROX_OPTION[:recursive])
+  end
+  
+  def test_initialize
+    contour = CvContour.new
+    assert_not_nil(contour)
+    assert_equal(CvContour, contour.class)
+    assert(contour.is_a? CvSeq)
+  end
+
+  def test_rect
+    contour = CvContour.new
+    assert_not_nil(contour.rect)
+    assert_equal(CvRect, contour.rect.class)
+  end
+
+  def test_color
+    contour = CvContour.new
+    assert_equal(0, contour.color)
+    contour.color = 1
+    assert_equal(1, contour.color)
+  end
+
+  def test_reserved
+    reserved = CvContour.new.reserved
+    assert_equal(Array, reserved.class)
+    assert_equal(3, reserved.size)
+  end
+
+  def test_approx_poly
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contours = mat0.find_contours(:mode => CV_RETR_EXTERNAL)
+
+    poly = contours.approx_poly
+    assert_equal(CvContour, poly.class)
+    assert(poly.size > 0)
+    assert(poly.all? { |c| c.class == CvPoint })
+
+    poly = contours.approx_poly(:method => :dp)
+    assert_equal(CvContour, poly.class)
+    assert(poly.size > 0)
+    assert(poly.all? { |c| c.class == CvPoint })
+
+    poly = contours.approx_poly(:accuracy => 2.0)
+    assert_equal(CvContour, poly.class)
+    assert(poly.size > 0)
+    assert(poly.all? { |c| c.class == CvPoint })
+
+    [true, false, 1, 0].each { |recursive|
+      poly = contours.approx_poly(:recursive => recursive)
+      assert_equal(CvContour, poly.class)
+      assert(poly.size > 0)
+      assert(poly.all? { |c| c.class == CvPoint })
+    }
+
+    poly = contours.approx_poly(:method => :dp, :accuracy => 2.0, :recursive => false)
+    assert_equal(CvContour, poly.class)
+    assert(poly.size > 0)
+    assert(poly.all? { |c| c.class == CvPoint })
+
+    # Uncomment the following lines to show the result
+    # poly = contours.approx_poly(:accuracy => 3.0)
+    # dst = mat0.clone.zero
+    # begin
+    #   dst.draw_contours!(poly, CvColor::White, CvColor::Black, 2,
+    #                      :thickness => 1, :line_type => :aa)
+    # end while (poly = poly.h_next)
+    # snap dst
+  end
+  
+  def test_bounding_rect
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contours = mat0.find_contours
+    rect = contours.bounding_rect
+    assert_equal(CvRect, rect.class)
+    assert_equal(32, rect.x)
+    assert_equal(32, rect.y)
+    assert_equal(65, rect.width)
+    assert_equal(65, rect.height)
+  end
+
+  def test_create_tree
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contour = mat0.find_contours
+    tree = contour.create_tree
+    assert_equal(CvContourTree, tree.class)
+    assert_equal(34, tree.p1.x)
+    assert_equal(53, tree.p1.y)
+    assert_equal(0, tree.p2.x)
+    assert_equal(0, tree.p2.y)
+  end
+
+  def test_in
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contour = mat0.find_contours
+    assert(contour.in? CvPoint.new(64, 64))
+    assert_false(contour.in? CvPoint.new(0, 0))
+    assert_nil(contour.in? CvPoint.new(64, 32))
+  end
+
+  def test_measure_distance
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contour = mat0.find_contours
+    assert_in_delta(-0.7071, contour.measure_distance(CvPoint.new(63, 32)), 0.01)
+    assert_in_delta(31.01, contour.measure_distance(CvPoint.new(64, 64)), 0.01)
+  end
+
+  def test_point_polygon_test
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contour = mat0.find_contours
+
+    assert_equal(1, contour.point_polygon_test(CvPoint.new(64, 64), 0))
+    assert_equal(1, contour.point_polygon_test(CvPoint.new(64, 64), false))
+    assert_equal(-1, contour.point_polygon_test(CvPoint.new(0, 0), 0))
+    assert_equal(-1, contour.point_polygon_test(CvPoint.new(0, 0), false))
+    assert_equal(0, contour.point_polygon_test(CvPoint.new(64, 32), 0))
+    assert_equal(0, contour.point_polygon_test(CvPoint.new(64, 32), false))
+
+    assert_in_delta(-0.7071, contour.point_polygon_test(CvPoint.new(63, 32), 1), 0.01)
+    assert_in_delta(-0.7071, contour.point_polygon_test(CvPoint.new(63, 32), true), 0.01)
+    assert_in_delta(31.01, contour.point_polygon_test(CvPoint.new(64, 64), 1), 0.01)
+    assert_in_delta(31.01, contour.point_polygon_test(CvPoint.new(64, 64), true), 0.01)
+  end
+end
+

data/test/test_cvcontourtree.rb

@@ -0,0 +1,43 @@
+#!/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 OpenCV::CvContourTree
+class TestCvContourTree < OpenCVTestCase
+  def setup
+    @tree = CvContourTree.new(CvPoint)
+  end
+
+  def test_initialize
+    tree = CvContourTree.new(CvPoint)
+    assert_equal(CvContourTree, tree.class)
+    assert(tree.is_a? CvSeq)
+  end
+
+  def test_p1
+    assert_equal(CvPoint, @tree.p1.class)
+  end
+
+  def test_p2
+    assert_equal(CvPoint, @tree.p2.class)
+  end
+
+  def test_contour
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    contour = mat0.find_contours
+    tree = contour.create_tree
+    contour = tree.contour(CvTermCriteria.new(100, 0.01))
+    assert_equal(CvContour, contour.class)
+
+    assert_raise(CvStsBadArg) {
+      tree.contour(CvTermCriteria.new(0, 0))
+    }
+  end
+end
+

data/test/test_cverror.rb

@@ -0,0 +1,50 @@
+#!/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 OpenCV::CvError
+class TestCvError < OpenCVTestCase
+  def test_cverror
+    assert_equal(StandardError, CvError.superclass)
+    begin
+      raise CvError
+    rescue
+      assert_equal(CvError, $!.class)
+    end
+  end
+  
+  def test_subclass
+    errors = [CvStsBackTrace, CvStsError, CvStsInternal, CvStsNoMem, CvStsBadArg, CvStsBadFunc, CvStsNoConv, CvStsAutoTrace, CvHeaderIsNull, CvBadImageSize, CvBadOffset, CvBadDataPtr, CvBadStep, CvBadModelOrChSeq, CvBadNumChannels, CvBadNumChannel1U, CvBadDepth, CvBadAlphaChannel, CvBadOrder, CvBadOrigin, CvBadAlign, CvBadCallBack, CvBadTileSize, CvBadCOI, CvBadROISize, CvMaskIsTiled, CvStsNullPtr, CvStsVecLengthErr, CvStsFilterStructContentErr, CvStsKernelStructContentErr, CvStsFilterOffsetErr, CvStsBadSize, CvStsDivByZero, CvStsInplaceNotSupported, CvStsObjectNotFound, CvStsUnmatchedFormats, CvStsBadFlag, CvStsBadPoint, CvStsBadMask, CvStsUnmatchedSizes, CvStsUnsupportedFormat, CvStsOutOfRange, CvStsParseError, CvStsNotImplemented, CvStsBadMemBlock, CvStsAssert, CvGpuNotSupported, CvGpuApiCallError]
+
+    errors.each { |err|
+      assert_equal(CvError, err.superclass)
+
+      begin
+        raise err
+      rescue err
+        assert_equal(err, $!.class)
+      rescue
+        flunk("Failed to catch #{err}")
+      end
+
+      begin
+        raise err
+      rescue CvError
+        assert_equal(err, $!.class)
+      rescue
+        flunk("Failed to catch #{err}")
+      end
+
+      begin
+        raise err
+      rescue
+        assert_equal(err, $!.class)
+      end
+    }
+  end
+end
+

data/test/test_cvfeaturetree.rb

@@ -0,0 +1,65 @@
+#!/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 OpenCV::CvFeatureTree
+class TestCvFeatureTree < OpenCVTestCase
+  def test_initialize
+    desc = CvMat.new(1, 1, :cv32f, 1)
+    ft = CvFeatureTree.new(desc)
+    assert_equal(CvFeatureTree, ft.class)
+    
+    assert_raise(TypeError) {
+      CvFeatureTree.new(DUMMY_OBJ)
+    }
+  end
+
+  def test_find_feature
+    dim = 2
+    points = []
+    points << [99, 51]
+    points << [52, 57]
+    points << [57, 42]
+    points << [13, 39]
+    points << [15, 68]
+    points << [75, 11]
+    points << [69, 62]
+    points << [52, 46]
+    points << [0, 64]
+    points << [67, 16]
+
+    desc1 = CvMat.new(points.size, dim, :cv32f, 1)
+    desc1.set_data(points)
+
+    pt = [[50, 50], [11, 40]]
+    desc2 = CvMat.new(pt.size, dim, :cv32f, 1)
+    desc2.set_data(pt)
+
+    ft = CvFeatureTree.new(desc1)
+    results, dist = ft.find_features(desc2, 1, 10)
+
+    assert_equal(CvMat, results.class)
+    assert_equal(CvMat, dist.class)
+
+    assert_equal(7, results[0][0].to_i)
+    assert_in_delta(4.472, dist[0][0], 0.001)
+
+    assert_equal(3, results[1][0].to_i)
+    assert_in_delta(2.236, dist[1][0], 0.001)
+
+    assert_raise(TypeError) {
+      ft.find_features(DUMMY_OBJ, 1, 10)
+    }
+    assert_raise(TypeError) {
+      ft.find_features(desc2, DUMMY_OBJ, 10)
+    }
+    assert_raise(TypeError) {
+      ft.find_features(desc2, 1, DUMMY_OBJ)
+    }
+  end
+end
+

data/test/test_cvfont.rb

@@ -0,0 +1,58 @@
+#!/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 OpenCV::CvFont
+class TestCvFont < OpenCVTestCase
+  def test_FACE
+    assert_equal(0, CvFont::FACE[:simplex])
+    assert_equal(1, CvFont::FACE[:plain])
+    assert_equal(2, CvFont::FACE[:duplex])
+    assert_equal(4, CvFont::FACE[:triplex])
+    assert_equal(5, CvFont::FACE[:complex_small])
+    assert_equal(6, CvFont::FACE[:script_simplex])
+    assert_equal(7, CvFont::FACE[:script_complex])
+  end
+
+  def test_FONT_OPTION
+    assert_equal(1.0, CvFont::FONT_OPTION[:hscale])
+    assert_equal(1.0, CvFont::FONT_OPTION[:vscale])
+    assert_equal(0, CvFont::FONT_OPTION[:shear])
+    assert_equal(1, CvFont::FONT_OPTION[:thickness])
+    assert_equal(8, CvFont::FONT_OPTION[:line_type])
+  end
+
+  def test_initialize
+    font = CvFont.new(:simplex)
+    assert_equal(0, font.face)
+    assert_equal(1.0, font.hscale)
+    assert_equal(1.0, font.vscale)
+    assert_equal(0, font.shear)
+    assert_equal(1, font.thickness)
+    assert_equal(8, font.line_type)
+    assert_false(font.italic)
+
+    font = CvFont.new(:plain, :hscale => 2.5, :vscale => 3.5,
+                      :shear => 0.5, :thickness => 3, :line_type => 2, :italic => false)
+    assert_equal(1, font.face)
+    assert_equal(2.5, font.hscale)
+    assert_equal(3.5, font.vscale)
+    assert_equal(0.5, font.shear)
+    assert_equal(3, font.thickness)
+    assert_equal(2, font.line_type)
+    assert_false(font.italic)
+
+    font = CvFont.new(:simplex, :italic => true)
+    assert_equal(16, font.face)
+    assert(font.italic)
+
+    assert_raise(ArgumentError) {
+      CvFont.new(:foo)
+    }
+  end
+end
+