afeld-opencv 0.0.8

data/ext/opencv/window.h

@@ -0,0 +1,66 @@
+/************************************************************
+
+   window.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_GUI_H
+#include "gui.h"
+#endif
+
+#ifndef RUBY_OPENCV_GUI_WINDOW_H
+#define RUBY_OPENCV_GUI_WINDOW_H
+
+#include "opencv.h"
+#define __NAMESPACE_BEGIN_WINDOW namespace cWindow {
+#define __NAMESPACE_END_WINDOW }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_GUI
+
+typedef struct Window {
+  VALUE name;
+  VALUE image;
+  VALUE trackbars;
+  VALUE blocks;
+} Window;
+
+__NAMESPACE_BEGIN_WINDOW
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+
+void window_mark(void *ptr);
+void window_free(void *ptr);
+VALUE rb_alive_q(VALUE self);
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_destroy(VALUE self);
+VALUE rb_destroy_all(VALUE klass);
+VALUE rb_resize(int argc, VALUE *argv, VALUE self);
+VALUE rb_move(int argc, VALUE *argv, VALUE self);
+VALUE rb_show_image(VALUE self, VALUE img);
+VALUE rb_set_trackbar(int argc, VALUE *argv, VALUE self);
+VALUE rb_set_mouse_callback(int argc, VALUE* argv, VALUE self);
+
+inline Window*
+WINDOW(VALUE object) {
+  Window *ptr;
+  Data_Get_Struct(object, Window, ptr);
+  return ptr;
+}
+
+inline const char*
+GET_WINDOW_NAME(VALUE object) {
+  return StringValueCStr(WINDOW(object)->name);
+}
+
+__NAMESPACE_END_WINDOW
+__NAMESPACE_END_GUI
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_GUI_WINDOW_H
+

data/extconf.rb

@@ -0,0 +1,65 @@
+#!/usr/bin/env ruby
+=begin
+create Makefile script for Ruby/OpenCV
+
+usage : ruby extconf.rb
+        make && make install
+
+VC : ruby extconf.rb
+     nmake
+	 nmake install
+=end
+require "mkmf"
+
+dir_config("opencv", "/usr/local/include", "/usr/local/lib")
+dir_config("libxml2", "/usr/include", "/usr/lib")
+
+opencv_headers = ["opencv2/core/core_c.h", "opencv2/core/core.hpp", "opencv2/imgproc/imgproc_c.h",
+                  "opencv2/imgproc/imgproc.hpp", "opencv2/video/tracking.hpp", "opencv2/features2d/features2d.hpp",
+                  "opencv2/flann/flann.hpp", "opencv2/calib3d/calib3d.hpp", "opencv2/objdetect/objdetect.hpp",
+                  "opencv2/legacy/compat.hpp", "opencv2/legacy/legacy.hpp", "opencv2/highgui/highgui_c.h",
+                  "opencv2/highgui/highgui.hpp", "opencv2/photo/photo.hpp", "opencv2/nonfree/nonfree.hpp"]
+
+opencv_libraries = ["opencv_calib3d", "opencv_contrib", "opencv_core", "opencv_features2d",
+                    "opencv_flann", "opencv_gpu", "opencv_highgui", "opencv_imgproc",
+                    "opencv_legacy", "opencv_ml", "opencv_objdetect", "opencv_video",
+                    "opencv_photo", "opencv_nonfree"]
+
+
+puts ">> Check the required libraries..."
+
+OPENCV_VERSION_SUFFIX = '240'
+case CONFIG["arch"]
+when /mswin32/
+  opencv_libraries.map! {|lib| lib + OPENCV_VERSION_SUFFIX }
+  have_library("msvcrt")
+  opencv_libraries.each {|lib|
+    raise "#{lib}.lib not found." unless have_library(lib)
+  }
+  $CFLAGS << ' /EHsc'
+when /mingw32/
+  opencv_libraries.map! {|lib| lib + OPENCV_VERSION_SUFFIX }
+  have_library("msvcrt")
+  opencv_libraries.each {|lib|
+    raise "lib#{lib} not found." unless have_library(lib)
+  }
+else
+  opencv_libraries.each {|lib|
+    raise "lib#{lib} not found." unless have_library(lib)
+  }
+  have_library("stdc++")
+end
+
+# Check the required headers
+puts ">> Check the required headers..."
+opencv_headers.each {|header|
+  raise "#{header} not found." unless have_header(header)
+}
+have_header("stdarg.h")
+
+# Quick fix for 1.8.7
+$CFLAGS << " -I#{File.dirname(__FILE__)}/ext/opencv"
+
+# Create Makefile
+create_makefile("opencv", "./ext/opencv")
+

data/test/helper.rb

@@ -0,0 +1,146 @@
+#!/usr/bin/env ruby
+# -*- mode: ruby; coding: utf-8-unix -*- 
+require 'test/unit'
+require 'digest/md5'
+require 'opencv'
+include OpenCV
+
+class OpenCVTestCase < Test::Unit::TestCase
+  SAMPLE_DIR = File.expand_path(File.dirname(__FILE__)) + '/samples/'
+  FILENAME_CAT = SAMPLE_DIR + 'cat.jpg'
+  FILENAME_LENA256x256 = SAMPLE_DIR + 'lena-256x256.jpg'
+  FILENAME_LENA32x32 = SAMPLE_DIR + 'lena-32x32.jpg'
+  FILENAME_LENA_EYES = File.expand_path(File.dirname(__FILE__)) + '/samples/lena-eyes.jpg'
+  FILENAME_FRUITS = SAMPLE_DIR + 'fruits.jpg'
+  FILENAME_CONTOURS = File.expand_path(File.dirname(__FILE__)) + '/samples/contours.jpg'
+  HAARCASCADE_FRONTALFACE_ALT = SAMPLE_DIR + 'haarcascade_frontalface_alt.xml.gz'
+  AVI_SAMPLE = SAMPLE_DIR + 'movie_sample.avi'
+
+  DUMMY_OBJ = Digest::MD5.new # dummy object for argument type check test
+
+  def snap(*images)
+    n = -1
+    images.map! { |val|
+      n += 1
+      if val.is_a? Hash
+        val
+      elsif val.is_a? Array
+        {:title => val[0], :image => val[1] }
+      else
+        {:title => "snap-#{n}", :image => val }
+      end
+    }
+
+    pos = CvPoint.new(0, 0)
+    images.each { |img|
+      w = GUI::Window.new(img[:title])
+      w.show(img[:image])
+      w.move(pos)
+      pos.x += img[:image].width
+      if pos.x > 800
+        pos.y += img[:image].height
+        pos.x = 0
+      end
+    }
+    
+    GUI::wait_key
+    GUI::Window::destroy_all
+  end
+
+  def hash_img(img)
+    # Compute a hash for an image, useful for image comparisons
+    Digest::MD5.hexdigest(img.data)
+  end
+
+  unless Test::Unit::TestCase.instance_methods.map {|m| m.to_sym }.include? :assert_false
+    def assert_false(actual, message = nil)
+      assert_equal(false, actual, message)
+    end
+  end
+
+  alias original_assert_in_delta assert_in_delta
+
+  def assert_cvscalar_equal(expected, actual, message = nil)
+    assert_equal(CvScalar, actual.class, message)
+    assert_array_equal(expected.to_ary, actual.to_ary, message)
+  end
+
+  def assert_array_equal(expected, actual, message = nil)
+    assert_equal(expected.size, actual.size, message)
+    expected.zip(actual) { |e, a|
+      assert_equal(e, a, message)
+    }
+  end
+  
+  def assert_in_delta(expected, actual, delta)
+    if expected.is_a? CvScalar or actual.is_a? CvScalar
+      expected = expected.to_ary if expected.is_a? CvScalar
+      actual = actual.to_ary if actual.is_a? CvScalar
+      assert_in_delta(expected, actual ,delta)
+    elsif expected.is_a? Array and actual.is_a? Array
+      assert_equal(expected.size, actual.size)
+      expected.zip(actual) { |e, a|
+        original_assert_in_delta(e, a, delta)
+      }
+    else
+      original_assert_in_delta(expected, actual, delta)
+    end
+  end
+
+  def create_cvmat(height, width, depth = :cv8u, channel = 4, &block)
+    m = CvMat.new(height, width, depth, channel)
+    block = lambda { |j, i, c| CvScalar.new(*([c + 1] * channel)) } unless block_given?
+    count = 0
+    height.times { |j|
+      width.times { |i|
+        m[j, i] = block.call(j, i, count)
+        count += 1
+      }
+    }
+    m
+  end
+
+  def create_iplimage(width, height, depth = :cv8u, channel = 4, &block)
+    m = IplImage.new(width, height, depth, channel)
+    block = lambda { |j, i, c| CvScalar.new(*([c + 1] * channel)) } unless block_given?
+    count = 0
+    height.times { |j|
+      width.times { |i|
+        m[j, i] = block.call(j, i, count)
+        count += 1
+      }
+    }
+    m
+  end
+
+  def assert_each_cvscalar(actual, delta = 0, &block)
+    raise unless block_given?
+    count = 0
+    actual.height.times { |j|
+      actual.width.times { |i|
+        expected = block.call(j, i, count)
+        if delta == 0
+          expected = expected.to_ary if expected.is_a? CvScalar
+          assert_array_equal(expected, actual[j, i].to_ary)
+        else
+          assert_in_delta(expected, actual[j, i], delta)
+        end
+        count += 1
+      }
+    }
+  end
+
+  def print_cvmat(mat)
+    s = []
+    mat.height.times { |j|
+      a = []
+      mat.width.times { |i|
+        tmp = mat[j, i].to_ary.map {|m| m.to_f.round(2) }.join(',')
+        a << "[#{tmp}]"
+      }
+      s << a.join(' ')
+    }
+    puts s.join("\n")
+  end
+end
+

data/test/runner.rb

@@ -0,0 +1,30 @@
+#!/usr/bin/env ruby
+# -*- mode: ruby; coding: utf-8-unix -*- 
+require 'test/unit'
+
+src_testdir = File.dirname(File.expand_path(__FILE__))
+if Test::Unit.methods.include? :setup_argv
+  srcdir = File.dirname(src_testdir)
+  Test::Unit.setup_argv {|files|
+    if files.empty?
+      [src_testdir]
+    else
+      files.map {|f|
+        if File.exist? "#{src_testdir}/#{f}"
+          "#{src_testdir}/#{f}"
+        elsif File.exist? "#{srcdir}/#{f}"
+          "#{srcdir}/#{f}"
+        elsif File.exist? f
+          f
+        else
+          raise ArgumentError, "not found: #{f}"
+        end
+      }
+    end
+  }
+elsif Test::Unit.constants.map {|c| c.to_sym }.include? :AutoRunner
+  Test::Unit::AutoRunner.run(true, src_testdir)
+else
+  raise 'Test runner not found'
+end
+

data/test/test_curve.rb

@@ -0,0 +1,43 @@
+#!/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::Curve
+class TestCurve < OpenCVTestCase
+  def setup
+    @contour1 = CvContour.new
+    
+    mat0 = create_cvmat(128, 128, :cv8u, 1) { |j, i|
+      (j - 64) ** 2 + (i - 64) ** 2 <= (32 ** 2) ? CvColor::White : CvColor::Black
+    }
+    @contour2 = mat0.find_contours
+  end
+
+  def test_closed
+    assert_false(@contour1.closed?)
+    assert(@contour2.closed?)
+  end
+
+  def test_convex
+    assert_false(@contour1.convex?)
+  end
+
+  def test_hole
+    assert_false(@contour1.hole?)
+  end
+
+  def test_simple
+    assert(@contour1.simple?)
+  end
+
+  def test_arc_length
+    assert_in_delta(211.480, @contour2.arc_length, 0.001)
+    assert_in_delta(32.181, @contour2.arc_length(CvSlice.new(0, 9), true), 0.001)
+    assert_in_delta(32.181, @contour2.arc_length(0..10, true), 0.001)
+  end
+end
+