ruby-opencv 0.0.8.pre

data/ext/opencv/cvcircle32f.h

@@ -0,0 +1,52 @@
+/************************************************************
+
+   cvcircle32f.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2007 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVCIRCLE32F_H
+#define RUBY_OPENCV_CVCIRCLE32F_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVCIRCLE32F namespace cCvCircle32f {
+#define __NAMESPACE_END_CVCIRCLE32F }
+
+__NAMESPACE_BEGIN_OPENCV
+
+typedef struct CvCircle32f {
+  CvPoint2D32f center;
+  float radius;
+} CvCircle32f;
+
+__NAMESPACE_BEGIN_CVCIRCLE32F
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_center(VALUE self);
+VALUE rb_radius(VALUE self);
+VALUE rb_aref(VALUE self, VALUE index);
+VALUE rb_to_ary(VALUE self);
+
+VALUE new_object(CvCircle32f circle32f);
+
+__NAMESPACE_END_CVCIRCLE32F
+
+inline CvCircle32f*
+CVCIRCLE32F(VALUE object)
+{
+  CvCircle32f *ptr;
+  Data_Get_Struct(object, CvCircle32f, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVCIRCLE32F_H

data/ext/opencv/cvcondensation.cpp

@@ -0,0 +1,282 @@
+/************************************************************
+
+   cvcondensation.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvcondensation.h"
+/*
+ * Document-class: OpenCV::CvConDensation
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVCONDENSATION
+
+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, "CvConDensation", rb_cObject);
+  rb_define_method(rb_klass, "dp", RUBY_METHOD_FUNC(rb_dp), 0);
+  rb_define_method(rb_klass, "mp", RUBY_METHOD_FUNC(rb_mp), 0);
+  rb_define_method(rb_klass, "dynamic_matrix", RUBY_METHOD_FUNC(rb_dynamic_matrix), 0);
+  rb_define_method(rb_klass, "confidence", RUBY_METHOD_FUNC(rb_confidence), 0);
+  rb_define_method(rb_klass, "cumulative", RUBY_METHOD_FUNC(rb_cumulative), 0);
+  rb_define_method(rb_klass, "state", RUBY_METHOD_FUNC(rb_state), 0);
+  rb_define_method(rb_klass, "samples_num", RUBY_METHOD_FUNC(rb_samples_num), 0);
+  rb_define_method(rb_klass, "init_sample_set", RUBY_METHOD_FUNC(rb_init_sample_set), 2);
+  rb_define_method(rb_klass, "update_by_time", RUBY_METHOD_FUNC(rb_update_by_time), 0);
+  rb_define_alias(rb_klass, "update", "update_by_time");
+  rb_define_method(rb_klass, "each_sample", RUBY_METHOD_FUNC(rb_each_sample), 0);
+  rb_define_method(rb_klass, "calculate_confidence", RUBY_METHOD_FUNC(rb_calculate_confidence), 0);
+}
+
+/*
+ * call-seq:
+ *   dp -> int
+ *
+ * Return dimension of state vector
+ */
+VALUE
+rb_dp(VALUE self)
+{
+  return INT2NUM(CVCONDENSATION(self)->DP);
+}
+
+/*
+ * call-seq:
+ *   mp -> int
+ *
+ * Return demension of measurement vector.
+ */
+VALUE
+rb_mp(VALUE self)
+{
+  return INT2NUM(CVCONDENSATION(self)->MP);
+}
+
+/*
+ * call-seq:
+ *   dynamic_matrix -> mat
+ *
+ * Return matrix of the linear Dynamics system.
+ */
+VALUE
+rb_dynamic_matrix(VALUE self)
+{
+  CvConDensation *cd = CVCONDENSATION(self);
+  CvMat* mat = NULL;
+  try {
+    mat = cvInitMatHeader(ALLOC(CvMat), cd->DP, cd->DP, CV_MAKETYPE(CV_32F, 1), cd->DynamMatr);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return DEPEND_OBJECT(cCvMat::rb_class(), mat, self);
+}
+
+/*
+ * call-seq:
+ *   confidence -> mat
+ *
+ * Return confidence for each sample.
+ */
+VALUE
+rb_confidence(VALUE self)
+{
+  CvConDensation *cd = CVCONDENSATION(self);
+  CvMat* mat = NULL;
+  try {
+    mat = cvInitMatHeader(ALLOC(CvMat), cd->SamplesNum, 1, CV_MAKETYPE(CV_32F, 1), cd->flConfidence);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return DEPEND_OBJECT(cCvMat::rb_class(), mat, self);
+}
+
+/*
+ * call-seq:
+ *   cumulative -> mat
+ *
+ * Return cumulative confidence.
+ */
+VALUE
+rb_cumulative(VALUE self)
+{
+  CvConDensation *cd = CVCONDENSATION(self);
+  CvMat* mat = NULL;
+  try {
+    mat = cvInitMatHeader(ALLOC(CvMat), cd->SamplesNum, 1, CV_MAKETYPE(CV_32F, 1), cd->flCumulative);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return DEPEND_OBJECT(cCvMat::rb_class(), mat, self);
+}
+
+/*
+ * call-seq:
+ *   state -> mat
+ *
+ * Return vector of state
+ */
+VALUE
+rb_state(VALUE self)
+{
+  CvConDensation *cd = CVCONDENSATION(self);
+  CvMat* mat = NULL;
+  try {
+    mat = cvInitMatHeader(ALLOC(CvMat), cd->DP, 1, CV_MAKETYPE(CV_32F, 1), cd->State);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return DEPEND_OBJECT(cCvMat::rb_class(), mat, self);  
+}
+
+/*
+ * call-seq:
+ *   samples_num -> int
+ *
+ * Return number of the samples
+ */
+VALUE
+rb_samples_num(VALUE self)
+{
+  return INT2NUM(CVCONDENSATION(self)->SamplesNum);
+}
+
+/*
+ * call-seq:
+ *   init_sample_set(upper, lower)
+ *
+ * Initializes sample set for ConDensation algorithm.
+ * Fills the samples with values within specified(lower to upper) ranges.
+ */
+VALUE
+rb_init_sample_set(VALUE self, VALUE lower, VALUE upper)
+{
+  CvConDensation *cd = CVCONDENSATION(self);
+  CvMat *lower_bound = CVMAT_WITH_CHECK(lower), *upper_bound = CVMAT_WITH_CHECK(upper), lb_stub, ub_stub;
+  int lower_type = lower_bound->type, upper_type = lower_bound->type;
+  try {
+    if (lower_type != CV_32FC1 || lower_bound->cols != 1) {
+      if (CV_MAT_DEPTH(lower_type) == CV_32F) {
+	lower_bound = cvReshape(lower_bound, &lb_stub, 1, lower_bound->rows * lower_bound->cols);
+      }
+      else {
+	lower = cCvMat::new_object(cvSize(lower_bound->rows * lower_bound->cols, 1), CV_MAKETYPE(CV_32S, 1));
+	cvConvertScale(lower_bound, CVARR(lower));
+	lower_bound = CVMAT(lower);      
+      }
+    }
+    if (upper_type != CV_32FC1 || upper_bound->cols != 1) {
+      if (CV_MAT_DEPTH(upper_type) == CV_32F) {
+	upper_bound = cvReshape(upper_bound, &ub_stub, 1, upper_bound->rows * upper_bound->cols);
+      }
+      else {
+	upper = cCvMat::new_object(cvSize(upper_bound->rows * upper_bound->cols, 1), CV_MAKETYPE(CV_32F, 1));
+	cvConvertScale(upper_bound, CVARR(upper));
+	upper_bound = CVMAT(upper);      
+      }
+    }
+    if (lower_bound->rows != cd->DP || upper_bound->rows != cd->DP) {
+      rb_raise(rb_eTypeError, "sample matrix step unmatch.");
+    }
+    cvConDensInitSampleSet(cd, lower_bound, upper_bound);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+/*
+ * call-seq:
+ *   update_by_time
+ *
+ * Estimates subsequent model state.
+ */
+VALUE
+rb_update_by_time(VALUE self)
+{
+  try {
+    cvConDensUpdateByTime(CVCONDENSATION(self));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+/*
+ * call-seq:
+ *   each_sample {|mat| ... }
+ *
+ * Evaluate each sample by given block.
+ */
+VALUE
+rb_each_sample(VALUE self)
+{
+  CvConDensation *cd = CVCONDENSATION(self);
+  if (rb_block_given_p()) {
+    try {
+      for (int i = 0; i < cd->SamplesNum; ++i) {
+	CvMat* mat = cvInitMatHeader(ALLOC(CvMat), cd->DP, 1, CV_MAKETYPE(CV_32F, 1), cd->flSamples[i]);
+	rb_yield(DEPEND_OBJECT(cCvMat::rb_class(), mat, self));
+      }
+    }
+    catch (cv::Exception& e) {
+      raise_cverror(e);
+    }
+  }
+  return self;
+}
+
+/*
+ * call-seq:
+ *   calculate_confidence {|value| ... }
+ *
+ * Evalute each sample by given block, then return value set to confidence.
+ */
+VALUE
+rb_calculate_confidence(VALUE self)
+{
+  VALUE value;
+  CvConDensation *cd = CVCONDENSATION(self);
+  if (rb_block_given_p()) {
+    try {
+      for (int i = 0; i < cd->SamplesNum; ++i) {
+	CvMat* mat = cvInitMatHeader(ALLOC(CvMat), cd->DP, 1, CV_MAKETYPE(CV_32F, 1), cd->flSamples[i]);
+	value = rb_yield(DEPEND_OBJECT(cCvMat::rb_class(), mat, self));
+	cd->flConfidence[i] = NUM2DBL(value);
+      }
+    }
+    catch (cv::Exception& e) {
+      raise_cverror(e);
+    }
+  }
+  return self;
+}
+
+__NAMESPACE_END_CVCONDENSATION
+__NAMESPACE_END_OPENCV

data/ext/opencv/cvcondensation.h

@@ -0,0 +1,49 @@
+/************************************************************
+
+   cvcondensation.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVCONDENSATION_H
+#define RUBY_OPENCV_CVCONDENSATION_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVCONDENSATION namespace cCvConDensation {
+#define __NAMESPACE_END_CVCONDENSATION }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVCONDENSATION
+
+VALUE rb_class();
+void define_ruby_class();
+
+VALUE rb_dp(VALUE self);
+VALUE rb_mp(VALUE self);
+VALUE rb_dynamic_matrix(VALUE self);
+VALUE rb_state(VALUE self);
+VALUE rb_confidence(VALUE self);
+VALUE rb_cumulative(VALUE self);
+VALUE rb_samples_num(VALUE self);
+
+VALUE rb_init_sample_set(VALUE self, VALUE lower, VALUE upper);
+VALUE rb_update_by_time(VALUE self);
+VALUE rb_each_sample(VALUE self);
+VALUE rb_calculate_confidence(VALUE self);
+
+__NAMESPACE_END_CVCONDENSATION
+
+inline CvConDensation*
+CVCONDENSATION(VALUE object)
+{
+  CvConDensation *ptr;
+  Data_Get_Struct(object, CvConDensation, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVCONDENSATION_H

data/ext/opencv/cvconnectedcomp.cpp

@@ -0,0 +1,143 @@
+/************************************************************
+
+   cvconnectedcomp.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2007 Masakazu Yonekura
+
+************************************************************/
+#include "cvconnectedcomp.h"
+/*
+ * Document-class: OpenCV::CvConnectedComp
+ *
+ * see CvMat#flood_fill
+ *
+ * C structure is here.
+ *   typedef struct CvConnectedComp
+ *   {
+ *     double area;    
+ *     CvScalar value; 
+ *     CvRect rect;    
+ *     CvSeq* contour;
+ *   } CvConnectedComp;
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVCONNECTEDCOMP
+
+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, "CvConnectedComp", 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, "area", RUBY_METHOD_FUNC(rb_area), 0);
+  rb_define_method(rb_klass, "value", RUBY_METHOD_FUNC(rb_value), 0);
+  rb_define_method(rb_klass, "rect", RUBY_METHOD_FUNC(rb_rect), 0);
+  rb_define_method(rb_klass, "rect=", RUBY_METHOD_FUNC(rb_set_rect), 1);
+  rb_define_method(rb_klass, "contour", RUBY_METHOD_FUNC(rb_contour), 0);
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvConnectedComp *ptr;
+  return Data_Make_Struct(klass, CvConnectedComp, 0, -1, ptr);
+}
+
+VALUE
+rb_initialize(int argc, VALUE *argv, VALUE self)
+{
+  VALUE area, value, rect, contour;
+  rb_scan_args(argc, argv, "04", &area, &value, &rect, &contour);
+
+  if (!NIL_P(area))
+    CVCONNECTEDCOMP(self)->area = NUM2DBL(area);
+  if (!NIL_P(value))
+    CVCONNECTEDCOMP(self)->value = *CVSCALAR(value);
+  if (!NIL_P(rect))
+    CVCONNECTEDCOMP(self)->rect = *CVRECT(rect);
+  if (!NIL_P(contour))
+    CVCONNECTEDCOMP(self)->contour = CVSEQ(contour);
+  return self;
+}
+
+/*
+ * Return area of connected component.
+ */
+VALUE
+rb_area(VALUE self)
+{
+  return rb_float_new(CVCONNECTEDCOMP(self)->area);
+}
+
+/*
+ * Return average color of the connected component.
+ */
+VALUE
+rb_value(VALUE self)
+{
+  return REFER_OBJECT(cCvScalar::rb_class(), &CVCONNECTEDCOMP(self)->value, self);
+}
+
+/*
+ * Return ROI of the component.
+ */
+VALUE
+rb_rect(VALUE self)
+{
+  return REFER_OBJECT(cCvRect::rb_class(), &CVCONNECTEDCOMP(self)->rect, self);
+}
+
+/*
+ * Set ROI of the component.
+ */
+VALUE
+rb_set_rect(VALUE self, VALUE rect)
+{
+  CVCONNECTEDCOMP(self)->rect = VALUE_TO_CVRECT(rect);
+  return self;
+}
+
+/*
+ * Return optional component boundary
+ */
+VALUE
+rb_contour(VALUE self)
+{
+  return REFER_OBJECT(cCvContour::rb_class(), &CVCONNECTEDCOMP(self)->contour, self);
+}
+
+VALUE
+new_object()
+{
+  return rb_allocate(rb_klass);
+}
+
+VALUE
+new_object(CvConnectedComp comp)
+{
+  VALUE object = rb_allocate(rb_klass);
+  *CVCONNECTEDCOMP(object) = comp;
+  return object;
+}
+
+__NAMESPACE_END_CVCONNECTEDCOMP
+__NAMESPACE_END_OPENCV