ruby-opencv 0.0.8.pre

data/test/test_cvrect.rb

@@ -0,0 +1,144 @@
+#!/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 OpenCV::CvRect
+class TestCvRect < OpenCVTestCase
+  class MyRect; end
+
+  def test_x
+    rect = CvRect.new
+    rect.x = 100
+    assert_equal(100, rect.x)
+    rect.x = 200
+    assert_equal(200, rect.x)
+  end
+
+  def test_y
+    rect = CvRect.new
+    rect.y = 100
+    assert_equal(100, rect.y)
+    rect.y = 200
+    assert_equal(200, rect.y)
+  end
+
+  def test_width
+    rect = CvRect.new
+    rect.width = 100
+    assert_equal(100, rect.width)
+    rect.width = 200
+    assert_equal(200, rect.width)
+  end
+
+  def test_height
+    rect = CvRect.new
+    rect.height = 100
+    assert_equal(100, rect.height)
+    rect.height = 200
+    assert_equal(200, rect.height)
+  end
+
+  def test_compatible
+    assert(!(CvRect.compatible? MyRect.new))
+    MyRect.class_eval { def x; end }
+    assert(!(CvRect.compatible? MyRect.new))
+    MyRect.class_eval { def y; end }
+    assert(!(CvRect.compatible? MyRect.new))
+    MyRect.class_eval { def width; end }
+    assert(!(CvRect.compatible? MyRect.new))
+    MyRect.class_eval { def height; end }
+    assert(CvRect.compatible? MyRect.new)
+    assert(CvRect.compatible? CvRect.new)
+  end
+
+  def test_initialize
+    rect = CvRect.new
+    assert_equal(0, rect.x)
+    assert_equal(0, rect.y)
+    assert_equal(0, rect.width)
+    assert_equal(0, rect.height)
+
+    rect = CvRect.new(10, 20, 30, 40)
+    assert_equal(10, rect.x)
+    assert_equal(20, rect.y)
+    assert_equal(30, rect.width)
+    assert_equal(40, rect.height)
+
+    rect = CvRect.new(CvRect.new(10, 20, 30, 40))
+    assert_equal(10, rect.x)
+    assert_equal(20, rect.y)
+    assert_equal(30, rect.width)
+    assert_equal(40, rect.height)
+
+    assert_raise(TypeError) {
+      CvRect.new(DUMMY_OBJ)
+    }
+    assert_raise(ArgumentError) {
+      CvRect.new(1, 2)
+    }
+    assert_raise(ArgumentError) {
+      CvRect.new(1, 2, 3)
+    }
+    assert_raise(ArgumentError) {
+      CvRect.new(1, 2, 3, 4, 5)
+    }
+  end
+
+  def test_center
+    center = CvRect.new(10, 20, 35, 45).center
+    assert_in_delta(27.5, center.x, 0.01)
+    assert_in_delta(42.5, center.y, 0.01)
+  end
+
+  def test_points
+    points = CvRect.new(10, 20, 35, 45).points
+    assert_equal(4, points.size)
+    assert_in_delta(10, points[0].x, 0.01)
+    assert_in_delta(20, points[0].y, 0.01)
+    assert_in_delta(10, points[1].x, 0.01)
+    assert_in_delta(65, points[1].y, 0.01)
+    assert_in_delta(45, points[2].x, 0.01)
+    assert_in_delta(65, points[2].y, 0.01)
+    assert_in_delta(45, points[3].x, 0.01)
+    assert_in_delta(20, points[3].y, 0.01)
+  end
+
+  def test_top_left
+    tl = CvRect.new(10, 20, 35, 45).top_left
+    assert_equal(10, tl.x)
+    assert_equal(20, tl.y)
+  end
+
+  def test_top_right
+    tr = CvRect.new(10, 20, 35, 45).top_right
+    assert_equal(45, tr.x)
+    assert_equal(20, tr.y)
+  end
+
+  def test_bottom_left
+    bl = CvRect.new(10, 20, 35, 45).bottom_left
+    assert_equal(10, bl.x)
+    assert_equal(65, bl.y)
+  end
+
+  def test_bottom_right
+    br = CvRect.new(10, 20, 35, 45).bottom_right
+    assert_equal(45, br.x)
+    assert_equal(65, br.y)
+  end
+
+  def test_max_rect
+    rect1 = CvRect.new(10, 20, 30, 40)
+    rect2 = CvRect.new(30, 40, 70, 80)
+    rect3 = CvRect.max_rect(rect1, rect2)
+    assert_equal(10, rect3.x)
+    assert_equal(20, rect3.y)
+    assert_equal(90, rect3.width)
+    assert_equal(100, rect3.height)
+  end
+end
+

data/test/test_cvscalar.rb

@@ -0,0 +1,113 @@
+#!/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 OpenCV::CvScalar
+class TestCvScalar < OpenCVTestCase
+  def test_initialize
+    s = CvScalar.new
+    assert_in_delta([0, 0, 0, 0], s, 0.01)
+
+    s = CvScalar.new(1.1)
+    assert_in_delta([1.1, 0, 0, 0], s, 0.01)
+
+    s = CvScalar.new(1.1, 2.2)
+    assert_in_delta([1.1, 2.2, 0, 0], s, 0.01)
+
+    s = CvScalar.new(1.1, 2.2, 3.3)
+    assert_in_delta([1.1, 2.2, 3.3, 0], s, 0.01)
+
+    s = CvScalar.new(1.1, 2.2, 3.3, 4.4)
+    assert_in_delta([1.1, 2.2, 3.3, 4.4], s, 0.01)
+  end
+  
+  def test_aref
+    assert_in_delta([0, 0, 0, 0], CvScalar.new, 0.01)
+    assert_in_delta([10, 20, 30, 40], CvScalar.new(10, 20, 30, 40), 0.01)
+    assert_in_delta([0.1, 0.2, 0.3, 0.4], CvScalar.new(0.1, 0.2, 0.3, 0.4), 0.01)
+  end
+
+  def test_aset
+    s = CvScalar.new
+    [10, 20, 30, 40].each_with_index { |x, i|
+      s[i] = x
+    }
+    assert_in_delta([10, 20, 30, 40], s, 0.01)
+
+    s = CvScalar.new
+    [0.1, 0.2, 0.3, 0.4].each_with_index { |x, i|
+      s[i] = x
+    }
+    assert_in_delta([0.1, 0.2, 0.3, 0.4], s, 0.01)
+  end
+
+  def test_sub
+    s1 = CvScalar.new(10, 20, 30, 40)
+    s2 = CvScalar.new(2, 4, 6, 8)
+    [s1.sub(s2), s1 - s2].each { |s|
+      assert_in_delta([8, 16, 24, 32], s, 0.01)
+    }
+
+    s3 = CvScalar.new(0.2, 0.4, 0.6, 0.8)
+    [s2.sub(s3), s2 - s3].each { |s|
+      assert_in_delta([1.8, 3.6, 5.4, 7.2], s, 0.01)
+    }
+
+    mat = CvMat.new(5, 5)
+    mask = CvMat.new(5, 5, :cv8u, 1)
+    mat.height.times { |j|
+      mat.width.times { |i|
+        mat[i, j] = CvScalar.new(1.5)
+        mask[i, j] = (i < 2 and j < 3) ? 1 : 0
+      }
+    }
+    mat = CvScalar.new(0.1).sub(mat, mask)
+
+    [CvMat.new(5, 5, :cv16u, 1), CvMat.new(5, 5, :cv8u, 3)].each { |msk|
+      assert_raise(TypeError) {
+        CvScalar.new.sub(mat, msk)
+      }
+    }
+  end
+
+  def test_to_s
+    assert_equal("<OpenCV::CvScalar:10,20,30,40>", CvScalar.new(10, 20, 30, 40).to_s)
+    assert_equal("<OpenCV::CvScalar:0.1,0.2,0.3,0.4>", CvScalar.new(0.1, 0.2, 0.3, 0.4).to_s)
+  end
+
+  def test_to_ary
+    [[10, 20, 30, 40], [0.1, 0.2, 0.3, 0.4]].each { |a|
+      s = CvScalar.new(*a)
+      b = s.to_ary
+      c = s.to_a # Alias
+      [b, c].each { |x|
+        assert_equal(Array, x.class)
+        assert_in_delta(a, x, 0.01)
+      }
+    }
+  end
+
+  def test_cvcolor
+    assert_cvscalar_equal(CvColor::Black, CvScalar.new(0x0, 0x0, 0x0, 0))
+    assert_cvscalar_equal(CvColor::Silver, CvScalar.new(0x0c, 0x0c, 0x0c, 0))
+    assert_cvscalar_equal(CvColor::Gray, CvScalar.new(0x80, 0x80, 0x80, 0))
+    assert_cvscalar_equal(CvColor::White, CvScalar.new(0xff, 0xff, 0xff, 0))
+    assert_cvscalar_equal(CvColor::Maroon, CvScalar.new(0x0, 0x0, 0x80, 0))
+    assert_cvscalar_equal(CvColor::Red, CvScalar.new(0x0, 0x0, 0xff, 0))
+    assert_cvscalar_equal(CvColor::Purple, CvScalar.new(0x80, 0x0, 0x80, 0))
+    assert_cvscalar_equal(CvColor::Fuchsia, CvScalar.new(0xff, 0x0, 0xff, 0))
+    assert_cvscalar_equal(CvColor::Green, CvScalar.new(0x0, 0x80, 0x0, 0))
+    assert_cvscalar_equal(CvColor::Lime, CvScalar.new(0x0, 0xff, 0x0, 0))
+    assert_cvscalar_equal(CvColor::Olive, CvScalar.new(0x0, 0x80, 0x80, 0))
+    assert_cvscalar_equal(CvColor::Yellow, CvScalar.new(0x0, 0xff, 0xff, 0))
+    assert_cvscalar_equal(CvColor::Navy, CvScalar.new(0x80, 0x0, 0x0, 0))
+    assert_cvscalar_equal(CvColor::Blue, CvScalar.new(0xff, 0x0, 0x0, 0))
+    assert_cvscalar_equal(CvColor::Teal, CvScalar.new(0x80, 0x80, 0x0, 0))
+    assert_cvscalar_equal(CvColor::Aqua, CvScalar.new(0xff, 0xff, 0x0, 0))
+  end
+end
+

data/test/test_cvseq.rb

@@ -0,0 +1,295 @@
+#!/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 OpenCV::CvSeq
+class TestCvSeq < OpenCVTestCase
+  def test_initialize
+    # assert_not_nil(CvSeq.new(CvIndex))
+    assert_not_nil(CvSeq.new(Fixnum))
+    assert_not_nil(CvSeq.new(CvPoint))
+    assert_not_nil(CvSeq.new(CvPoint2D32f))
+    assert_not_nil(CvSeq.new(CvPoint3D32f))
+
+    assert(CvSeq.new(CvPoint).is_a? Enumerable)
+  end
+
+  def test_total
+    seq1 = CvSeq.new(CvPoint)
+    assert_equal(0, seq1.total)
+
+    seq1.push(CvPoint.new(1, 2))
+    assert_equal(1, seq1.total)
+
+    seq1.push(CvPoint.new(3, 4))
+    assert_equal(2, seq1.total)
+    # Alias
+    assert_equal(2, seq1.length)
+    assert_equal(2, seq1.size)
+  end
+
+  def test_empty
+    assert(CvSeq.new(CvPoint).empty?)
+    assert(CvSeq.new(CvPoint2D32f).empty?)
+    assert(CvSeq.new(CvPoint3D32f).empty?)
+  end
+  
+  def test_aref
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40), CvPoint.new(50, 60))
+
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(10, seq1[0].x)
+    assert_equal(20, seq1[0].y)
+    assert_equal(30, seq1[1].x)
+    assert_equal(40, seq1[1].y)
+    assert_equal(50, seq1[2].x)
+    assert_equal(60, seq1[2].y)
+    
+    seq2 = CvSeq.new(Fixnum).push(10, 20, 30)
+
+    assert_equal(Fixnum, seq2[0].class)
+    assert_equal(10, seq2[0])
+    assert_equal(20, seq2[1])
+    assert_equal(30, seq2[2])
+  end
+
+  def test_push
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40))
+    
+    assert_equal(2, seq1.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(10, seq1[0].x)
+    assert_equal(20, seq1[0].y)
+    assert_equal(CvPoint, seq1[1].class)
+    assert_equal(30, seq1[1].x)
+    assert_equal(40, seq1[1].y)
+
+    seq2 = CvSeq.new(CvPoint).push(CvPoint.new(50, 60))
+    seq2.push(seq1)
+    assert_equal(3, seq2.total)
+    assert_equal(CvPoint, seq2[0].class)
+    assert_equal(50, seq2[0].x)
+    assert_equal(60, seq2[0].y)
+    assert_equal(CvPoint, seq2[1].class)
+    assert_equal(10, seq2[1].x)
+    assert_equal(20, seq2[1].y)
+    assert_equal(CvPoint, seq2[2].class)
+    assert_equal(30, seq2[2].x)
+    assert_equal(40, seq2[2].y)
+
+    seq3 = CvSeq.new(Fixnum).push(10)
+    seq4 = CvSeq.new(Fixnum).push(20, 30)
+    seq3.push(seq4)
+    assert_equal(3, seq3.total)
+    assert_equal(Fixnum, seq3[0].class)
+    assert_equal(10, seq3[0])
+    assert_equal(20, seq3[1])
+    assert_equal(30, seq3[2])
+
+    assert_raise(TypeError) {
+      seq1.push(CvPoint2D32f.new(55.5, 66.6))
+    }
+
+    assert_raise(TypeError) {
+      seq3 = CvSeq.new(CvPoint2D32f).push(CvPoint2D32f.new(55.5, 66.6))
+      seq1.push(seq3)
+    }
+  end
+
+  def test_pop
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40))
+    point1 = seq1.pop
+    assert_equal(CvPoint, point1.class)
+    assert_equal(30, point1.x)
+    assert_equal(40, point1.y)
+
+    assert_equal(1, seq1.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(10, seq1[0].x)
+    assert_equal(20, seq1[0].y)
+
+    assert_nil(CvSeq.new(CvPoint).pop)
+
+    seq2 = CvSeq.new(Fixnum).push(10, 20, 30)
+    assert_equal(30, seq2.pop)
+    assert_equal(20, seq2.pop)
+    assert_equal(10, seq2.pop)
+  end
+
+  def test_clear
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40))
+    seq1.clear
+    assert_not_nil(seq1)
+    assert_equal(0, seq1.total)
+  end
+
+  def test_unshift
+    seq1 = CvSeq.new(CvPoint).unshift(CvPoint.new(10, 20), CvPoint.new(30, 40))
+    
+    assert_equal(2, seq1.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(30, seq1[0].x)
+    assert_equal(40, seq1[0].y)
+    assert_equal(CvPoint, seq1[1].class)
+    assert_equal(10, seq1[1].x)
+    assert_equal(20, seq1[1].y)
+
+    seq2 = CvSeq.new(CvPoint).unshift(CvPoint.new(50, 60))
+    seq2.unshift(seq1)
+    assert_equal(3, seq2.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(30, seq1[0].x)
+    assert_equal(40, seq1[0].y)
+    assert_equal(CvPoint, seq1[1].class)
+    assert_equal(10, seq1[1].x)
+    assert_equal(20, seq1[1].y)
+    assert_equal(CvPoint, seq2[2].class)
+    assert_equal(50, seq2[2].x)
+    assert_equal(60, seq2[2].y)
+
+    seq3 = CvSeq.new(Fixnum).unshift(10, 20, 30)
+    assert_equal(3, seq3.total)
+    assert_equal(30, seq3[0])
+    assert_equal(20, seq3[1])
+    assert_equal(10, seq3[2])
+
+    assert_raise(TypeError) {
+      seq1.unshift(CvPoint2D32f.new(55.5, 66.6))
+    }
+
+    assert_raise(TypeError) {
+      seq3 = CvSeq.new(CvPoint2D32f).push(CvPoint2D32f.new(55.5, 66.6))
+      seq1.unshift(seq3)
+    }
+  end
+
+  def test_shift
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40))
+    point1 = seq1.shift
+    assert_equal(CvPoint, point1.class)
+    assert_equal(10, point1.x)
+    assert_equal(20, point1.y)
+
+    assert_equal(1, seq1.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(30, seq1[0].x)
+    assert_equal(40, seq1[0].y)
+
+    seq2 = CvSeq.new(Fixnum).push(10, 20, 30)
+    assert_equal(10, seq2.shift)
+    assert_equal(20, seq2.shift)
+    assert_equal(30, seq2.shift)
+
+    assert_nil(CvSeq.new(CvPoint).shift)
+  end
+
+  def test_first
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40), CvPoint.new(50, 60))
+    point1 = seq1.first
+    assert_equal(CvPoint, point1.class)
+    assert_equal(10, point1.x)
+    assert_equal(20, point1.y)
+
+    seq2 = CvSeq.new(Fixnum).push(10, 20, 30)
+    assert_equal(10, seq2.first)
+  end
+
+  def test_last
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40), CvPoint.new(50, 60))
+    point1 = seq1.last
+    assert_equal(CvPoint, point1.class)
+    assert_equal(50, point1.x)
+    assert_equal(60, point1.y)
+
+    seq2 = CvSeq.new(Fixnum).push(10, 20, 30)
+    assert_equal(30, seq2.last)
+  end
+
+  def test_each
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40), CvPoint.new(50, 60))
+    i = 0
+    seq1.each { |s|
+      assert_equal(CvPoint, s.class)
+      assert_equal(seq1[i].x, s.x)
+      assert_equal(seq1[i].y, s.y)
+      i += 1
+    }
+    assert_equal(3, i)
+
+    seq2 = CvSeq.new(Fixnum).push(10, 20, 30)
+    i = 0
+    seq2.each { |s|
+      assert_equal(seq2[i], s)
+      i += 1
+    }
+    assert_equal(3, i)
+  end
+
+  def test_each_index
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40), CvPoint.new(50, 60))
+    n = 0
+    seq1.each_index { |i|
+      assert_equal(n, i)
+      n += 1
+    }
+    assert_equal(3, n)
+  end
+
+  def test_insert
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40))
+    seq1.insert(1, CvPoint.new(50, 60))
+    assert_equal(3, seq1.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(10, seq1[0].x)
+    assert_equal(20, seq1[0].y)
+    assert_equal(CvPoint, seq1[1].class)
+    assert_equal(50, seq1[1].x)
+    assert_equal(60, seq1[1].y)
+    assert_equal(CvPoint, seq1[2].class)
+    assert_equal(30, seq1[2].x)
+    assert_equal(40, seq1[2].y)
+
+    seq2 = CvSeq.new(Fixnum).push(10, 20)
+    seq2.insert(1, 15)
+    assert_equal(3, seq2.total)
+    assert_equal(10, seq2[0])
+    assert_equal(15, seq2[1])
+    assert_equal(20, seq2[2])
+  end
+
+  def test_remove
+    seq1 = CvSeq.new(CvPoint).push(CvPoint.new(10, 20), CvPoint.new(30, 40), CvPoint.new(50, 60))
+
+    seq1.remove(1)
+    assert_equal(2, seq1.total)
+    assert_equal(CvPoint, seq1[0].class)
+    assert_equal(10, seq1[0].x)
+    assert_equal(20, seq1[0].y)
+    assert_equal(CvPoint, seq1[1].class)
+    assert_equal(50, seq1[1].x)
+    assert_equal(60, seq1[1].y)
+  end
+
+  # These methods are tested in TestCvMat_imageprocessing#test_find_contours
+  # (test_cvmat_imageprocessing.rb)
+  # def test_h_prev
+  #   flunk('FIXME: CvSeq#h_prev is not tested yet.')
+  # end
+
+  # def test_h_next
+  #   flunk('FIXME: CvSeq#h_next is not tested yet.')
+  # end
+
+  # def test_v_prev
+  #   flunk('FIXME: CvSeq#v_prev is not tested yet.')
+  # end
+
+  # def test_v_next
+  #   flunk('FIXME: CvSeq#v_next is not tested yet.')
+  # end
+end
+