ruby-opencv 0.0.9-x86-mswin32

data/examples/snake.rb

@@ -0,0 +1,43 @@
+#!/usr/bin/env ruby
+# snake.rb
+require "rubygems"
+require "opencv"
+include OpenCV
+
+puts <<USAGE
+usage:
+  left-click: set a 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.circle!(mouse, 1, :color => CvColor::Red, :thickness => 2)
+    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
+    snake_points = image.snake_image(points, 1.0, 0.5, 1.5, CvSize.new(3, 3), 100)
+    display = image.GRAY2BGR
+    display.poly_line!([snake_points], :color => CvColor::Red, :is_closed => true, :thickness => 2)
+    window.show display
+    points.clear
+  end
+}
+
+GUI::wait_key
+

data/ext/opencv/curve.cpp

@@ -0,0 +1,112 @@
+/************************************************************
+
+   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, "closed?", 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);
+  double length = 0;
+  try {
+    length = cvArcLength(CVARR(self),
+			 NIL_P(slice) ? CV_WHOLE_SEQ : VALUE_TO_CVSLICE(slice),
+			 TRUE_OR_FALSE(is_closed, -1));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return rb_float_new(length);
+}
+
+__NAMESPACE_END_CURVE
+__NAMESPACE_END_OPENCV
+

data/ext/opencv/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/opencv/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 INT2NUM(CVAVGCOMP(self)->neighbors);
+}
+
+__NAMESPACE_END_AVGCOMP
+__NAMESPACE_END_OPENCV

data/ext/opencv/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/opencv/cvbox2d.cpp

@@ -0,0 +1,197 @@
+/************************************************************
+
+   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_alloc_func(rb_klass, rb_allocate);
+  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
+  rb_define_method(rb_klass, "center", RUBY_METHOD_FUNC(rb_center), 0);
+  rb_define_method(rb_klass, "center=", RUBY_METHOD_FUNC(rb_set_center), 1);
+  rb_define_method(rb_klass, "size", RUBY_METHOD_FUNC(rb_size), 0);
+  rb_define_method(rb_klass, "size=", RUBY_METHOD_FUNC(rb_set_size), 1);
+  rb_define_method(rb_klass, "angle", RUBY_METHOD_FUNC(rb_angle), 0);
+  rb_define_method(rb_klass, "angle=", RUBY_METHOD_FUNC(rb_set_angle), 1);
+  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);
+}
+
+/* 
+ * call-seq:
+ *   CvBox2D.new(<i>[center][, size][, angle]</i>) -> cvbox2d
+ *
+ * Create a box
+ */
+VALUE
+rb_initialize(int argc, VALUE *argv, VALUE self)
+{
+  VALUE center, size, angle;
+  CvBox2D* self_ptr = CVBOX2D(self);
+  rb_scan_args(argc, argv, "03", &center, &size, &angle);
+  
+  if (!NIL_P(center)) {
+    self_ptr->center = VALUE_TO_CVPOINT2D32F(center);
+  }
+  if (!NIL_P(size)) {
+    self_ptr->size = VALUE_TO_CVSIZE2D32F(size);
+    self_ptr->angle = NUM2DBL(angle);
+  }
+  
+  return self;
+}
+
+/*
+ * call-seq:
+ *   center -> cvpoint2d32f
+ * Return center point of box as CvPoint2D32f.
+ */
+VALUE
+rb_center(VALUE self)
+{
+  return REFER_OBJECT(cCvPoint2D32f::rb_class(), &CVBOX2D(self)->center, self);
+}
+
+/*
+ * call-seq:
+ *   center = <i>value</i>
+ *
+ * Set center to <i>value</i>
+ */
+VALUE
+rb_set_center(VALUE self, VALUE value)
+{
+  CVBOX2D(self)->center = VALUE_TO_CVPOINT2D32F(value);
+  return self;
+}
+
+/*
+ * call-seq:
+ *   size -> cvsize2d32f
+ * Return size of box as CvSize2D32f.
+ */
+VALUE
+rb_size(VALUE self)
+{
+  return REFER_OBJECT(cCvSize2D32f::rb_class(), &CVBOX2D(self)->size, self);
+}
+
+/*
+ * call-seq:
+ *   size = <i>value</i>
+ *
+ * Set size to <i>value</i>
+ */
+VALUE
+rb_set_size(VALUE self, VALUE value)
+{
+  CVBOX2D(self)->size = VALUE_TO_CVSIZE2D32F(value);
+  return self;
+}
+
+/*
+ * call-seq:
+ *   angle -> float
+ *
+ * Return angle of box as Float.
+ */
+VALUE
+rb_angle(VALUE self)
+{
+  return rb_float_new(CVBOX2D(self)->angle);
+}
+
+/*
+ * call-seq:
+ *   angle = <i>value</i>
+ *
+ * Set angle to <i>value</i>
+ */
+VALUE
+rb_set_angle(VALUE self, VALUE value)
+{
+  CVBOX2D(self)->angle = NUM2DBL(value);
+  return self;
+}
+
+/*
+ * call-seq:
+ *   points -> array(include cvpoint2d32f)
+ * Find box vertices. Return Array contain 4 CvPoint2D32f.
+ */
+VALUE
+rb_points(VALUE self)
+{
+  const int n = 4;
+  CvPoint2D32f p[n];
+  try {
+    cvBoxPoints(*CVBOX2D(self), p);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  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