jf-ruby-opencv 0.0.18

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 init_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,178 @@
+/************************************************************
+
+   cvhumoments.cpp -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2011 ser1zw
+
+************************************************************/
+#include "cvhumoments.h"
+
+/*
+ * Document-class: OpenCV::CvHuMoments
+ *
+ * Hu invariants
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHUMOMENTS
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvHuMoments *ptr;
+  return Data_Make_Struct(klass, CvHuMoments, 0, -1, ptr);
+}
+
+/*
+ * Calculates the seven Hu invariants
+ * @overload new(src_moments)
+ * @param src_moments [CvMoment] Input moments
+ * @return [CvHuMoments] Output Hu invariants
+ * @opencv_func cvGetHuMoments
+ * @see http://en.wikipedia.org/wiki/Image_moment#Rotation_invariant_moments
+ */
+VALUE
+rb_initialize(VALUE self, VALUE src_moments)
+{
+  try {
+    cvGetHuMoments(CVMOMENTS(src_moments), CVHUMOMENTS(self));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;      
+}
+
+/*
+ * Returns hu1 invariant
+ * @overload hu1
+ * @return [Number] hu1 invariant
+ */
+VALUE rb_hu1(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu1);
+}
+
+/*
+ * Returns hu2 invariant
+ * @overload hu2
+ * @return [Number] hu2 invariant
+ */
+VALUE rb_hu2(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu2);
+}
+
+/*
+ * Returns hu3 invariant
+ * @overload hu3
+ * @return [Number] hu3 invariant
+ */
+VALUE rb_hu3(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu3);
+}
+
+/*
+ * Returns hu4 invariant
+ * @overload hu4
+ * @return [Number] hu4 invariant
+ */
+VALUE rb_hu4(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu4);
+}
+
+/*
+ * Returns hu5 invariant
+ * @overload hu5
+ * @return [Number] hu5 invariant
+ */
+VALUE rb_hu5(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu5);
+}
+
+/*
+ * Returns hu6 invariant
+ * @overload hu6
+ * @return [Number] hu6 invariant
+ */
+VALUE rb_hu6(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu6);
+}
+
+/*
+ * Returns hu7 invariant
+ * @overload hu7
+ * @return [Number] hu7 invariant
+ */
+VALUE rb_hu7(VALUE self) {
+  return DBL2NUM(CVHUMOMENTS(self)->hu7);
+}
+
+/*
+ * Returns hu invaliants as an Array
+ * @overload to_ary
+ * @return [Array<Number>] Hu invaliants
+ */
+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;
+}
+
+void
+init_ruby_class()
+{
+#if 0
+  // For documentation using YARD
+  VALUE opencv = rb_define_module("OpenCV");
+#endif
+
+  if (rb_klass)
+    return;
+  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_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");
+
+  rb_define_method(rb_klass, "hu1", RUBY_METHOD_FUNC(rb_hu1), 0);
+  rb_define_method(rb_klass, "hu2", RUBY_METHOD_FUNC(rb_hu2), 0);
+  rb_define_method(rb_klass, "hu3", RUBY_METHOD_FUNC(rb_hu3), 0);
+  rb_define_method(rb_klass, "hu4", RUBY_METHOD_FUNC(rb_hu4), 0);
+  rb_define_method(rb_klass, "hu5", RUBY_METHOD_FUNC(rb_hu5), 0);
+  rb_define_method(rb_klass, "hu6", RUBY_METHOD_FUNC(rb_hu6), 0);
+  rb_define_method(rb_klass, "hu7", RUBY_METHOD_FUNC(rb_hu7), 0);
+}
+
+__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 init_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,159 @@
+/************************************************************
+
+   cvline.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvline.h"
+/*
+ * Document-class: OpenCV::CvLine
+ *
+ * Line parameters represented by a two-element (rho, theta)
+ * for CvMat#hough_lines
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVLINE
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvLine *ptr;
+  return Data_Make_Struct(klass, CvLine, 0, -1, ptr);
+}
+
+/*
+ * Returns distance from the coordinate origin (0, 0)
+ * @overload rho
+ * @return [Number] Distance from the coordinate origin
+ */
+VALUE
+rb_rho(VALUE self)
+{
+  return rb_float_new(CVLINE(self)->rho);
+}
+
+/*
+ * Set distance from the coordinate origin (0, 0)
+ * @overload rho=(value)
+ * @param value [Number] Distance from the coordinate origin
+ */
+VALUE
+rb_set_rho(VALUE self, VALUE rho)
+{
+  CVLINE(self)->rho = NUM2DBL(rho);
+  return self;
+}
+
+/*
+ * Returns line rotation angle in radians
+ * @overload theta
+ * @return [Number] Line rotation angle in radians
+ */
+VALUE
+rb_theta(VALUE self)
+{
+  return rb_float_new(CVLINE(self)->theta);
+}
+
+/*
+ * Set line rotation angle in radians
+ * @overload theta=(value)
+ * @param value [Number] Line rotation angle
+ */
+VALUE
+rb_set_theta(VALUE self, VALUE theta)
+{
+  CVLINE(self)->theta = NUM2DBL(theta);
+  return self;
+}
+
+/*
+ * Returns value of rho, theta
+ * @overload [](index)
+ * @param index [Integer] Index
+ * @return [Number] If index = 0, returns rho, else if index = 1, returns theta.
+ */
+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;
+}
+
+/*
+ * Set value of rho, theta
+ * @overload []=(index, value)
+ * @param index [Integer] Index
+ * @param value [Number] Value
+ * @return [Number] If index = 0, set rho, else if index = 1, set theta.
+ */
+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;
+}
+
+void
+init_ruby_class()
+{
+#if 0
+  // For documentation using YARD
+  VALUE opencv = rb_define_module("OpenCV");
+#endif
+
+  if (rb_klass)
+    return;
+  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);
+}
+
+__NAMESPACE_END_CVLINE
+__NAMESPACE_END_OPENCV