bonanza-ruby-opencv 0.0.13.20140330211753

data/test/test_cvchain.rb

@@ -0,0 +1,108 @@
+#!/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::CvChain
+class TestCvChain < OpenCVTestCase
+  def test_APPROX_OPTION
+    assert_equal(:approx_simple, CvChain::APPROX_CHAIN_OPTION[:method])
+    assert_equal(0, CvChain::APPROX_CHAIN_OPTION[:parameter])
+    assert_equal(0, CvChain::APPROX_CHAIN_OPTION[:minimal_perimeter])
+    assert_false(CvChain::APPROX_CHAIN_OPTION[:recursive])
+  end
+  
+  def test_initialize
+    chain = CvChain.new
+    assert_not_nil(chain)
+    assert_equal(CvChain, chain.class)
+    assert(chain.is_a? CvSeq)
+  end
+
+  def test_origin
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    chain = mat0.find_contours(mode: CV_RETR_EXTERNAL, method: CV_CHAIN_CODE)
+    assert_equal(CvChain, chain.class)
+    assert_equal(64, chain.origin.x)
+    assert_equal(32, chain.origin.y)
+
+    chain.origin = CvPoint.new(32, 64)
+    assert_equal(32, chain.origin.x)
+    assert_equal(64, chain.origin.y)
+  end
+
+  def test_codes
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    chain = mat0.find_contours(mode: CV_RETR_EXTERNAL, method: CV_CHAIN_CODE)
+    assert_equal(Array, chain.codes.class)
+    assert(chain.codes.all? { |a| (a.class == Fixnum) and (a >= 0 and a <= 7) })
+  end
+
+  def test_points
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    chain = mat0.find_contours(mode: CV_RETR_EXTERNAL, method: CV_CHAIN_CODE)
+    assert_equal(Array, chain.points.class)
+    assert(chain.points.all? { |a| a.class == CvPoint })
+  end
+
+  def test_approx_chains
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    chain = mat0.find_contours(mode: CV_RETR_EXTERNAL, method: CV_CHAIN_CODE)
+
+    contours = chain.approx_chains
+    assert_equal(CvChain, contours.class)
+    assert(contours.size > 0)
+    assert(contours.all? { |c| c.class == CvPoint })
+
+    [CV_CHAIN_APPROX_NONE, CV_CHAIN_APPROX_SIMPLE,
+     CV_CHAIN_APPROX_TC89_L1, CV_CHAIN_APPROX_TC89_KCOS,
+     :approx_none, :approx_simple, :approx_tc89_l1, :approx_tc89_kcos].each { |method|
+      contours = chain.approx_chains(method: method)
+      assert_equal(CvChain, contours.class)
+      assert(contours.size > 0)
+      assert(contours.all? { |c| c.class == CvPoint })
+    }
+
+    contours = chain.approx_chains(minimal_parameter: 10)
+    assert_equal(CvChain, contours.class)
+    assert(contours.size > 0)
+    assert(contours.all? { |c| c.class == CvPoint })
+
+    contours = chain.approx_chains(minimal_perimeter: (32 * 2 * Math::PI).ceil)
+    assert_nil(contours)
+
+    [true, false].each { |recursive|
+      contours = chain.approx_chains(recursive: recursive)
+      assert_equal(CvChain, contours.class)
+      assert(contours.size > 0)
+      assert(contours.all? { |c| c.class == CvPoint })
+    }
+
+    contours = chain.approx_chains(method: :approx_simple,
+                                  minimal_parameter: 100, recursive: false)
+    assert_equal(CvChain, contours.class)
+    assert(contours.size > 0)
+    assert(contours.all? { |c| c.class == CvPoint })
+
+    # Uncomment the following lines to show the result
+    # contours = chain.approx_chains
+    # dst = mat0.clone.zero
+    # begin
+    #   dst.draw_contours!(contours, CvColor::White, CvColor::Black, 2,
+    #                      thickness: 1, line_type: :aa)
+    # end while (contours = contours.h_next)
+    # snap dst
+  end
+end
+

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
+