ruby-opencv 0.0.8.pre

data/ext/opencv/cvhistogram.h

@@ -0,0 +1,73 @@
+/************************************************************
+
+   cvhistogram.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2008 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVHISTOGRAM_H
+#define RUBY_OPENCV_CVHISTOGRAM_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVHISTOGRAM namespace cCvHistogram {
+#define __NAMESPACE_END_CVHISTOGRAM }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHISTOGRAM
+
+VALUE rb_class();
+VALUE rb_allocate(VALUE klass);
+void define_ruby_class();
+
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_is_uniform(VALUE self);
+VALUE rb_is_sparse(VALUE self);
+VALUE rb_has_range(VALUE self);
+VALUE rb_calc_hist(int argc, VALUE* argv, VALUE self);
+VALUE rb_calc_hist_bang(int argc, VALUE* argv, VALUE self);
+VALUE rb_aref(VALUE self, VALUE args);
+VALUE rb_min_max_value(VALUE self);
+VALUE rb_dims(VALUE self);
+VALUE rb_copy_hist(VALUE self);
+
+VALUE rb_clear_hist(VALUE self);
+VALUE rb_clear_hist_bang(VALUE self);
+
+VALUE rb_normalize_hist(VALUE self, VALUE factor);
+VALUE rb_normalize_hist_bang(VALUE self, VALUE factor);
+VALUE rb_thresh_hist(VALUE self, VALUE threshold);
+VALUE rb_thresh_hist_bang(VALUE self, VALUE threshold);
+
+VALUE rb_set_hist_bin_ranges(int argc, VALUE* argv, VALUE self);
+VALUE rb_set_hist_bin_ranges_bang(int argc, VALUE* argv, VALUE self);
+
+VALUE rb_calc_back_project(VALUE self, VALUE image);
+VALUE rb_calc_back_project_patch(VALUE self, VALUE image, VALUE patch_size, VALUE method, VALUE factor);
+
+VALUE rb_compare_hist(VALUE self, VALUE hist1, VALUE hist2, VALUE method);
+VALUE rb_calc_prob_density(int argc, VALUE* argv, VALUE self);
+
+__NAMESPACE_END_CVHISTOGRAM
+
+inline CvHistogram*
+CVHISTOGRAM(VALUE object)
+{
+  CvHistogram* ptr;
+  Data_Get_Struct(object, CvHistogram, ptr);
+  return ptr;
+}
+
+inline CvHistogram*
+CVHISTOGRAM_WITH_CHECK(VALUE object)
+{
+  if (!rb_obj_is_kind_of(object, cCvHistogram::rb_class()))
+    raise_typeerror(object, cCvHistogram::rb_class());
+  return CVHISTOGRAM(object);
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVHISTOGRAM_H

data/ext/opencv/cvhumoments.cpp

@@ -0,0 +1,139 @@
+/************************************************************
+
+   cvhumoments.cpp -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2011 ser1zw
+
+************************************************************/
+#include "cvhumoments.h"
+
+/*
+ * Document-class: OpenCV::CvHuMoments
+ *
+ * moments
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHUMOMENTS
+
+#define DEFINE_CVHUMOMENTS_GETTER(elem)					\
+  rb_define_method(rb_klass, #elem, RUBY_METHOD_FUNC(rb_##elem), 0)
+
+#define CVHUMOMENTS_GETTER(elem)					\
+  VALUE rb_##elem(VALUE self) { return DBL2NUM(CVHUMOMENTS(self)->elem); }
+
+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, "CvHuMoments", 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, "to_ary", RUBY_METHOD_FUNC(rb_to_ary), 0);
+  rb_define_alias(rb_klass, "to_a", "to_ary");
+  
+  DEFINE_CVHUMOMENTS_GETTER(hu1);
+  DEFINE_CVHUMOMENTS_GETTER(hu2);
+  DEFINE_CVHUMOMENTS_GETTER(hu3);
+  DEFINE_CVHUMOMENTS_GETTER(hu4);
+  DEFINE_CVHUMOMENTS_GETTER(hu5);
+  DEFINE_CVHUMOMENTS_GETTER(hu6);
+  DEFINE_CVHUMOMENTS_GETTER(hu7);
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvHuMoments *ptr;
+  return Data_Make_Struct(klass, CvHuMoments, 0, -1, ptr);
+}
+
+/*
+ * call-seq:
+ *   CvHuMoments.new(<i>src_moments</i>)
+ *
+ * Calculates the seven Hu invariants.
+ * <i>src_moments</i> The input moments
+ *
+ * seven Hu invariants that are defined as:
+ *   h1=η20+η02
+ *   h2=(η20-η02)²+4η11²
+ *   h3=(η30-3η12)²+ (3η21-η03)²
+ *   h4=(η30+η12)²+ (η21+η03)²
+ *   h5=(η30-3η12)(η30+η12)[(η30+η12)²-3(η21+η03)²]+(3η21-η03)(η21+η03)[3(η30+η12)²-(η21+η03)²]
+ *   h6=(η20-η02)[(η30+η12)²- (η21+η03)²]+4η11(η30+η12)(η21+η03)
+ *   h7=(3η21-η03)(η21+η03)[3(η30+η12)²-(η21+η03)²]-(η30-3η12)(η21+η03)[3(η30+η12)²-(η21+η03)²]
+ * where ηi,j are normalized central moments of 2-nd and 3-rd orders. The computed values are proved to be invariant to the image scaling, rotation, and reflection except the seventh one, whose sign is changed by reflection. 
+ */
+VALUE
+rb_initialize(VALUE self, VALUE src_moments)
+{
+  try {
+    cvGetHuMoments(CVMOMENTS(src_moments), CVHUMOMENTS(self));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;      
+}
+
+CVHUMOMENTS_GETTER(hu1);
+CVHUMOMENTS_GETTER(hu2);
+CVHUMOMENTS_GETTER(hu3);
+CVHUMOMENTS_GETTER(hu4);
+CVHUMOMENTS_GETTER(hu5);
+CVHUMOMENTS_GETTER(hu6);
+CVHUMOMENTS_GETTER(hu7);
+
+/*
+ * call-seq:
+ *   to_ary -> [hu1, hu2, ... ,hu7]
+ *
+ * Return values by Array.
+ */
+VALUE
+rb_to_ary(VALUE self)
+{
+  CvHuMoments *hu_moments = CVHUMOMENTS(self);
+  return rb_ary_new3(7,
+		     rb_float_new(hu_moments->hu1),
+		     rb_float_new(hu_moments->hu2),
+		     rb_float_new(hu_moments->hu3),
+		     rb_float_new(hu_moments->hu4),
+		     rb_float_new(hu_moments->hu5),
+		     rb_float_new(hu_moments->hu6),
+		     rb_float_new(hu_moments->hu7));
+}
+
+VALUE
+new_object(CvMoments *src_moments)
+{
+  VALUE object = rb_allocate(rb_klass);
+  try {
+    cvGetHuMoments(src_moments, CVHUMOMENTS(object));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return object;
+}
+
+__NAMESPACE_END_CVHUMOMENTS
+__NAMESPACE_END_OPENCV

data/ext/opencv/cvhumoments.h

@@ -0,0 +1,51 @@
+/************************************************************
+
+   cvhumoments.h -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2011 ser1zw
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVHUMOMENTS_H
+#define RUBY_OPENCV_CVHUMOMENTS_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVHUMOMENTS namespace cCvHuMoments {
+#define __NAMESPACE_END_CVHUMOMENTS }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHUMOMENTS
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+VALUE rb_initialize(VALUE src_moments, VALUE self);
+
+VALUE rb_hu1(VALUE self);
+VALUE rb_hu2(VALUE self);
+VALUE rb_hu3(VALUE self);
+VALUE rb_hu4(VALUE self);
+VALUE rb_hu5(VALUE self);
+VALUE rb_hu6(VALUE self);
+VALUE rb_hu7(VALUE self);
+VALUE rb_to_ary(VALUE self);
+
+VALUE new_object(CvMoments *cvmoments);
+
+__NAMESPACE_END_CVHUMOMENTS
+
+inline CvHuMoments*
+CVHUMOMENTS(VALUE object)
+{
+  CvHuMoments *ptr;
+  Data_Get_Struct(object, CvHuMoments, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVHUMOMENTS_H

data/ext/opencv/cvline.cpp

@@ -0,0 +1,154 @@
+/************************************************************
+
+   cvline.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvline.h"
+/*
+ * Document-class: OpenCV::CvLine
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVLINE
+
+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, "CvLine", rb_cObject);
+  rb_define_alloc_func(rb_klass, rb_allocate);
+  rb_define_method(rb_klass, "rho", RUBY_METHOD_FUNC(rb_rho), 0);
+  rb_define_method(rb_klass, "rho=", RUBY_METHOD_FUNC(rb_set_rho), 1);
+  rb_define_method(rb_klass, "theta", RUBY_METHOD_FUNC(rb_theta), 0);
+  rb_define_method(rb_klass, "theta=", RUBY_METHOD_FUNC(rb_set_theta), 1);
+  rb_define_method(rb_klass, "[]", RUBY_METHOD_FUNC(rb_aref), 1);
+  rb_define_method(rb_klass, "[]=", RUBY_METHOD_FUNC(rb_aset), 2);
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvLine *ptr;
+  return Data_Make_Struct(klass, CvLine, 0, -1, ptr);
+}
+
+/*
+ * Return parameter on rho.
+ */
+VALUE
+rb_rho(VALUE self)
+{
+  return rb_float_new(CVLINE(self)->rho);
+}
+
+/*
+ * call-seq:
+ *   rho = val
+ *
+ * Set rho parameter, return self.
+ */
+VALUE
+rb_set_rho(VALUE self, VALUE rho)
+{
+  CVLINE(self)->rho = NUM2DBL(rho);
+  return self;
+}
+
+/*
+ * Return parameter on theta.
+ */
+VALUE
+rb_theta(VALUE self)
+{
+  return rb_float_new(CVLINE(self)->theta);
+}
+
+/*
+ * call-seq:
+ *   y = val
+ *
+ * Set theta parameter, return self.
+ */
+VALUE
+rb_set_theta(VALUE self, VALUE theta)
+{
+  CVLINE(self)->theta = NUM2DBL(theta);
+  return self;
+}
+
+/*
+ * call-seq:
+ *   [<i>index</i>]
+ *
+ * Return value of <i>index</i> dimension.
+ */
+VALUE
+rb_aref(VALUE self, VALUE index)
+{
+  switch (NUM2INT(index)) {
+  case 0:
+    return DBL2NUM(CVLINE(self)->rho);
+    break;
+  case 1:
+    return DBL2NUM(CVLINE(self)->theta);
+    break;
+  default:
+    rb_raise(rb_eIndexError, "index should be 0...2");
+    break;
+  }
+  return Qnil;
+}
+
+/*
+ * call-seq:
+ *   [<i>index</i>] = <i>value</i>
+ *
+ * Set value of <i>index</i> dimension to <i>value</i>
+ */
+VALUE
+rb_aset(VALUE self, VALUE index, VALUE value)
+{
+  switch (NUM2INT(index)) {
+  case 0:
+    CVLINE(self)->rho = NUM2DBL(value);
+    break;
+  case 1:
+    CVLINE(self)->theta = NUM2DBL(value);
+    break;
+  default:
+    rb_raise(rb_eIndexError, "index should be 0...2");
+    break;
+  }
+  return value;
+}
+
+VALUE
+new_object(CvLine line)
+{
+  VALUE object = rb_allocate(rb_klass);
+  *CVLINE(object) = line;
+  return object;
+}
+
+__NAMESPACE_END_CVLINE
+__NAMESPACE_END_OPENCV

data/ext/opencv/cvline.h

@@ -0,0 +1,54 @@
+/************************************************************
+
+   cvline.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2007 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVLINE_H
+#define RUBY_OPENCV_CVLINE_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVLINE namespace cCvLine {
+#define __NAMESPACE_END_CVLINE }
+
+__NAMESPACE_BEGIN_OPENCV
+
+typedef struct CvLine {
+  float rho;
+  float theta;
+} CvLine;
+
+__NAMESPACE_BEGIN_CVLINE
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_rho(VALUE self);
+VALUE rb_set_rho(VALUE self, VALUE rho);
+VALUE rb_theta(VALUE self);
+VALUE rb_set_theta(VALUE self, VALUE theta);
+VALUE rb_aref(VALUE self, VALUE index);
+VALUE rb_aset(VALUE self, VALUE index, VALUE value);
+
+VALUE new_object(CvLine line);
+
+__NAMESPACE_END_CVLINE
+
+inline CvLine*
+CVLINE(VALUE object)
+{
+  CvLine *ptr;
+  Data_Get_Struct(object, CvLine, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVLINE_H