opencv 0.0.6

data/examples/snake.rb

@@ -0,0 +1,43 @@
+#!/bin/env ruby
+# snake.rb
+require "rubygems"
+gem "opencv"
+require "opencv"
+include OpenCV
+
+puts <<USAGE
+usage:
+  left-click: set point
+  right-click: do snake
+USAGE
+
+window = GUI::Window.new "snake demo"
+image = CvMat.new(256, 256, :cv8u, 1).clear!
+image.circle!(CvPoint.new(128,128), 40, :color => CvColor::White, :thickness =>  -1)
+display = image.GRAY2BGR
+
+window.show display
+
+points = []
+
+window.on_mouse{|mouse|
+  case mouse.event
+  when :left_button_down
+    display[mouse.x, mouse.y] = CvColor::Red
+    puts "set point (#{mouse.x},#{mouse.y})"
+    points << CvPoint.new(mouse.x, mouse.y)
+    window.show display
+  when :right_button_down
+    if points.length < 3
+      puts "please set more point!"
+      next
+    end
+    points = image.snake_image(points, 1.0, 0.5, 1.5, CvSize.new(3, 3), 100)
+    display = image.GRAY2BGR
+    display.poly_line! points, :color => CvColor::Red, :is_closed => true
+    window.show display
+  end
+}
+
+GUI::wait_key
+

data/ext/curve.cpp

@@ -0,0 +1,103 @@
+/************************************************************
+
+   curve.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005 Masakazu Yonekura
+
+************************************************************/
+#include"curve.h"
+/*
+ * Document-class: OpenCV::Curve
+ *
+ * Curve sequence.
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CURVE
+
+VALUE module;
+    
+VALUE
+rb_module()
+{
+  return module;
+}
+
+void
+define_ruby_module()
+{
+  if(module)
+    return;
+  /* 
+   * opencv = rb_define_module("OpenCV");
+   *
+   * note: this comment is used by rdoc.
+   */
+  VALUE opencv = rb_module_opencv();
+  module = rb_define_module_under(opencv, "Curve");
+  rb_define_method(module, "close?", RUBY_METHOD_FUNC(rb_closed_q), 0);
+  rb_define_method(module, "convex?", RUBY_METHOD_FUNC(rb_convex_q), 0);
+  rb_define_method(module, "hole?", RUBY_METHOD_FUNC(rb_hole_q), 0);
+  rb_define_method(module, "simple?", RUBY_METHOD_FUNC(rb_simple_q), 0);
+  rb_define_method(module, "arc_length", RUBY_METHOD_FUNC(rb_arc_length), -1);
+}
+
+/*
+ * If curve is closed, return true. Otherwise return false.
+ */
+VALUE
+rb_closed_q(VALUE self)
+{
+  return CV_IS_SEQ_CLOSED(CVSEQ(self)) ? Qtrue : Qfalse;
+}
+
+/*
+ * If curve is convex, return true. Otherwise return false.
+ */
+VALUE
+rb_convex_q(VALUE self)
+{
+  return CV_IS_SEQ_CONVEX(CVSEQ(self)) ? Qtrue : Qfalse;
+}
+
+/*
+ * If curve is hole(inner contour), return true. Otherwise return false.
+ */
+VALUE
+rb_hole_q(VALUE self)
+{  
+  return CV_IS_SEQ_HOLE(CVSEQ(self)) ? Qtrue : Qfalse;
+}
+
+/*
+ * no idia.
+ */
+VALUE
+rb_simple_q(VALUE self)
+{
+  return CV_IS_SEQ_SIMPLE(CVSEQ(self)) ? Qtrue : Qfalse;
+}
+
+/*
+ * call-seq:
+ *   arc_length(<i>[slice = nil][,is_closed = nil]</i>) -> float
+ *
+ * Calculates contour perimeter or curve length.
+ * <i>slice</i> is starting and ending points of the curve.
+ * <i>is_closed</i> is indicates whether the curve is closed or not. There are 3 cases:
+ * * is_closed = true  - the curve is assumed to be unclosed.
+ * * is_closed = false - the curve is assumed to be closed.
+ * * is_closed = nil (default) use self#close?
+ */
+VALUE
+rb_arc_length(int argc, VALUE *argv, VALUE self)
+{
+  VALUE slice, is_closed;
+  rb_scan_args(argc, argv, "02", &slice, &is_closed);
+  return rb_float_new(cvArcLength(CVARR(self), NIL_P(slice) ? CV_WHOLE_SEQ : VALUE_TO_CVSLICE(slice), TRUE_OR_FALSE(is_closed, -1)));
+}
+
+__NAMESPACE_END_CURVE
+__NAMESPACE_END_OPENCV
+

data/ext/curve.h

@@ -0,0 +1,34 @@
+/************************************************************
+
+   curve.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVSEQ_CURVE_H
+#define RUBY_OPENCV_CVSEQ_CURVE_H
+
+#include"opencv.h"
+
+#define __NAMESPACE_BEGIN_CURVE namespace mCurve{
+#define __NAMESPACE_END_CURVE }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CURVE
+
+VALUE rb_module();
+
+void define_ruby_module();
+
+VALUE rb_closed_q(VALUE self);
+VALUE rb_convex_q(VALUE self);
+VALUE rb_hole_q(VALUE self);
+VALUE rb_simple_q(VALUE self);
+VALUE rb_arc_length(int argc, VALUE *argv, VALUE self);
+
+__NAMESPACE_END_CURVE
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVSEQ_CURVE_H

data/ext/cvavgcomp.cpp

@@ -0,0 +1,67 @@
+/************************************************************
+
+   cvavgcomp.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvavgcomp.h"
+/*
+ * Document-class: OpenCV::CvAvgComp
+ *
+ * CvRect with parameter "neighbors".
+ * CvHaarClassifierCascade#detect_object. 
+ *
+ *   typedef struct CvAvgComp{
+ *     CvRect rect;
+ *     int neighbors;
+ *   }
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_AVGCOMP
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+void
+define_ruby_class()
+{
+  if (rb_klass)
+    return;
+  /* 
+   * opencv = rb_define_module("OpenCV");
+   * cvrect = rb_define_class_under(opencv, "CvRect", rb_cObject);
+   *
+   * note: this comment is used by rdoc.
+   */
+  VALUE opencv = rb_module_opencv(), cvrect = cCvRect::rb_class();
+  rb_klass = rb_define_class_under(opencv, "CvAvgComp", cvrect);
+  rb_define_alloc_func(rb_klass, rb_allocate);
+  rb_define_method(rb_klass, "neighbors", RUBY_METHOD_FUNC(rb_neighbors), 0);
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvAvgComp *ptr;
+  return Data_Make_Struct(klass, CvAvgComp, 0, -1, ptr);
+}
+
+/*
+ * Return neighbors.
+ */
+VALUE
+rb_neighbors(VALUE self)
+{
+  return INT2FIX(CVAVGCOMP(self)->neighbors);
+}
+
+__NAMESPACE_END_AVGCOMP
+__NAMESPACE_END_OPENCV

data/ext/cvavgcomp.h

@@ -0,0 +1,39 @@
+/**********************************************************************
+
+   cvavgcomp.h
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+**********************************************************************/
+#ifndef RUBY_OPENCV_AVGCOMP_H
+#define RUBY_OPENCV_AVGCOMP_H
+
+#define __NAMESPACE_BEGIN_AVGCOMP namespace cCvAvgComp{
+#define __NAMESPACE_END_AVGCOMP }
+
+#include <opencv.h>
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_AVGCOMP
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+VALUE rb_neighbors(VALUE self);
+
+__NAMESPACE_END_AVGCOMP
+
+inline CvAvgComp *CVAVGCOMP(VALUE object){
+  CvAvgComp *ptr;
+  Data_Get_Struct(object, CvAvgComp, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+
+#endif // RUBY_OPENCV_AVGCOMP_H

data/ext/cvbox2d.cpp

@@ -0,0 +1,114 @@
+/************************************************************
+
+   cvbox2d.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvbox2d.h"
+/*
+ * Document-class: OpenCV::CvBox2D
+ *
+ * C structure is here.
+ *   typdef struct CvBox2D{
+ *     CvPoint2D32f center; // center of the box.
+ *     CvSize2D32f size;    // box width and length
+ *     float angle;         // angle between the horizonal axis and the first side (i.e length) in radians
+ *   }CvBox2D;
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVBOX2D
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+void
+define_ruby_class()
+{
+  if(rb_klass)
+    return;
+  /* 
+   * opencv = rb_define_module("OpenCV");
+   * 
+   * note: this comment is used by rdoc.
+   */
+  VALUE opencv = rb_module_opencv();
+  rb_klass = rb_define_class_under(opencv, "CvBox2D", rb_cObject);
+  rb_define_method(rb_klass, "center", RUBY_METHOD_FUNC(rb_center), 0);
+  rb_define_method(rb_klass, "size", RUBY_METHOD_FUNC(rb_size), 0);
+  rb_define_method(rb_klass, "angle", RUBY_METHOD_FUNC(rb_angle), 0);
+  rb_define_method(rb_klass, "points", RUBY_METHOD_FUNC(rb_points), 0);
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvBox2D *ptr;
+  return Data_Make_Struct(klass, CvBox2D, 0, -1, ptr);
+}
+
+/*
+ * Return center point of box as CvPoint2D32f.
+ */
+VALUE
+rb_center(VALUE self)
+{
+  return REFER_OBJECT(cCvPoint2D32f::rb_class(), &CVBOX2D(self)->center, self);
+}
+
+/*
+ * Return size of box as CvSize2D32f.
+ */
+VALUE
+rb_size(VALUE self)
+{
+  return REFER_OBJECT(cCvSize2D32f::rb_class(), &CVBOX2D(self)->size, self);
+}
+
+/*
+ * Return angle of box as Float.
+ */
+VALUE
+rb_angle(VALUE self)
+{
+  return rb_float_new(CVBOX2D(self)->angle);
+}
+
+/*
+ * Find box vertices. Return Array contain 4 CvPoint2D32f.
+ */
+VALUE
+rb_points(VALUE self)
+{
+  const int n = 4;
+  CvPoint2D32f p[n];
+  cvBoxPoints(*CVBOX2D(self), p);
+  VALUE points = rb_ary_new2(n);
+  for(int i = 0; i < n; i++)
+    rb_ary_store(points, i, cCvPoint2D32f::new_object(p[i]));        
+  return points;
+}
+
+VALUE
+new_object()
+{
+  return rb_allocate(cCvBox2D::rb_class());
+}
+
+VALUE
+new_object(CvBox2D box)
+{
+  VALUE object = rb_allocate(rb_klass);
+  *CVBOX2D(object) = box;
+  return object;
+}
+
+__NAMESPACE_END_CVBOX2D
+__NAMESPACE_END_OPENCV

data/ext/cvbox2d.h

@@ -0,0 +1,53 @@
+/************************************************************
+
+   cvbox2d.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVBOX2D_H
+#define RUBY_OPENCV_CVBOX2D_H
+
+#include"opencv.h"
+
+#define __NAMESPACE_BEGIN_CVBOX2D namespace cCvBox2D{
+#define __NAMESPACE_END_CVBOX2D }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVBOX2D
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+
+VALUE rb_center(VALUE self);
+VALUE rb_size(VALUE self);
+VALUE rb_angle(VALUE self);
+VALUE rb_points(VALUE self);
+
+VALUE new_object();
+VALUE new_object(CvBox2D box);
+
+__NAMESPACE_END_CVBOX2D
+
+inline CvBox2D *CVBOX2D(VALUE object){
+  CvBox2D *ptr;
+  Data_Get_Struct(object, CvBox2D, ptr);
+  return ptr;
+}
+
+inline CvBox2D VALUE_TO_CVBOX2D(VALUE object){
+  if(rb_obj_is_kind_of(object, cCvBox2D::rb_class())) {
+    return *CVBOX2D(object);
+  }else{
+    rb_raise(rb_eTypeError, "require %s or compatible object.", rb_class2name(cCvBox2D::rb_class()));
+  }
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVBOX2D_H