afeld-opencv 0.0.8

data/ext/opencv/cvhaarclassifiercascade.h

@@ -0,0 +1,39 @@
+/************************************************************
+
+   cvhaarclassifiercascade.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVHAARCLASSIFIERCASCADE_H
+#define RUBY_OPENCV_CVHAARCLASSIFIERCASCADE_H
+
+#define __NAMESPACE_BEGIN_CVHAARCLASSIFERCASCADE namespace cCvHaarClassifierCascade {
+#define __NAMESPACE_END_CVHAARCLASSIFERCASCADE }
+
+#include "opencv.h"
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHAARCLASSIFERCASCADE
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+
+VALUE rb_load(VALUE klass, VALUE path);
+VALUE rb_detect_objects(int argc, VALUE *argv, VALUE self);
+
+__NAMESPACE_END_CVHAARCLASSIFERCASCADE
+inline CvHaarClassifierCascade*
+CVHAARCLASSIFIERCASCADE(VALUE object) {
+  CvHaarClassifierCascade *ptr;
+  Data_Get_Struct(object, CvHaarClassifierCascade, ptr);
+  return ptr;
+}
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVHAARCLASSIFIERCASCADE_H

data/ext/opencv/cvhistogram.cpp

@@ -0,0 +1,546 @@
+/************************************************************
+
+   cvhistogram.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2008 Masakazu Yonekura
+
+************************************************************/
+#include "cvhistogram.h"
+/*
+ * Document-class: OpenCV::CvHistogram
+ *
+ * Muti-dimensional histogram.
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHISTOGRAM
+
+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, "CvHistogram", 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, "is_uniform?", RUBY_METHOD_FUNC(rb_is_uniform), 0);
+  rb_define_method(rb_klass, "is_sparse?", RUBY_METHOD_FUNC(rb_is_sparse), 0);
+  rb_define_method(rb_klass, "has_range?", RUBY_METHOD_FUNC(rb_has_range), 0);
+  rb_define_method(rb_klass, "dims", RUBY_METHOD_FUNC(rb_dims), 0);
+  rb_define_method(rb_klass, "calc_hist", RUBY_METHOD_FUNC(rb_calc_hist), -1);
+  rb_define_method(rb_klass, "calc_hist!", RUBY_METHOD_FUNC(rb_calc_hist_bang), -1);
+  rb_define_method(rb_klass, "[]", RUBY_METHOD_FUNC(rb_aref), -2);
+  rb_define_alias(rb_klass, "query_hist_value", "[]");
+  rb_define_method(rb_klass, "min_max_value", RUBY_METHOD_FUNC(rb_min_max_value), 0);
+  rb_define_method(rb_klass, "copy_hist", RUBY_METHOD_FUNC(rb_copy_hist), 0);
+
+  rb_define_method(rb_klass, "clear_hist", RUBY_METHOD_FUNC(rb_clear_hist), 0);
+  rb_define_alias(rb_klass, "clear", "clear_hist");
+  rb_define_method(rb_klass, "clear_hist!", RUBY_METHOD_FUNC(rb_clear_hist_bang), 0);
+  rb_define_alias(rb_klass, "clear!", "clear_hist!");
+
+  rb_define_method(rb_klass, "normalize_hist", RUBY_METHOD_FUNC(rb_normalize_hist), 1);
+  rb_define_alias(rb_klass, "normalize", "normalize_hist");
+  rb_define_method(rb_klass, "normalize_hist!", RUBY_METHOD_FUNC(rb_normalize_hist_bang), 1);
+  rb_define_alias(rb_klass, "normalize!", "normalize_hist!");
+
+  rb_define_method(rb_klass, "thresh_hist", RUBY_METHOD_FUNC(rb_thresh_hist), 1);
+  rb_define_alias(rb_klass, "thresh", "thresh_hist");
+  rb_define_method(rb_klass, "thresh_hist!", RUBY_METHOD_FUNC(rb_thresh_hist_bang), 1);
+  rb_define_alias(rb_klass, "thresh!", "thresh_hist!");
+
+  rb_define_method(rb_klass, "set_hist_bin_ranges", RUBY_METHOD_FUNC(rb_set_hist_bin_ranges), -1);
+  rb_define_method(rb_klass, "set_hist_bin_ranges!", RUBY_METHOD_FUNC(rb_set_hist_bin_ranges_bang), -1);
+
+  rb_define_method(rb_klass, "calc_back_project", RUBY_METHOD_FUNC(rb_calc_back_project), 1);
+  rb_define_method(rb_klass, "calc_back_project_patch", RUBY_METHOD_FUNC(rb_calc_back_project_patch), 4);
+
+  rb_define_singleton_method(rb_klass, "calc_prob_density", RUBY_METHOD_FUNC(rb_calc_prob_density), -1);
+  rb_define_singleton_method(rb_klass, "compare_hist", RUBY_METHOD_FUNC(rb_compare_hist), 3);
+}
+
+void
+release_hist(void* ptr)
+{
+  if (ptr) {
+    try {
+      cvReleaseHist((CvHistogram**)&ptr);
+    }
+    catch (cv::Exception& e) {
+      raise_cverror(e);
+    }
+  }
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  CvHistogram* ptr = NULL;
+  return Data_Wrap_Struct(klass, 0, release_hist, ptr);
+}
+
+float*
+ary2fltptr(VALUE ary, float* buff)
+{
+  Check_Type(ary, T_ARRAY);
+  int size = RARRAY_LEN(ary);
+  VALUE* ary_ptr = RARRAY_PTR(ary);
+  for (int i = 0; i < size; ++i) {
+    buff[i] = NUM2DBL(ary_ptr[i]);
+  }
+  return buff;
+}
+
+int*
+ary2intptr(VALUE ary, int* buff)
+{
+  Check_Type(ary, T_ARRAY);
+  int size = RARRAY_LEN(ary);
+  VALUE* ary_ptr = RARRAY_PTR(ary);
+  for (int i = 0; i < size; ++i) {
+    buff[i] = NUM2INT(ary_ptr[i]);
+  }
+  return buff;
+}
+
+VALUE
+rb_initialize(int argc, VALUE *argv, VALUE self)
+{
+  VALUE _dims, _sizes, _type, _ranges, _uniform;
+  int dims, type, uniform;
+  int* sizes;
+  float** ranges = NULL;
+
+  rb_scan_args(argc, argv, "32", &_dims, &_sizes, &_type, &_ranges, &_uniform);
+  int sizes_len = RARRAY_LEN(_sizes);
+  sizes = ALLOCA_N(int, sizes_len);
+
+  if (NIL_P(_ranges)) {
+    sizes = ary2intptr(_sizes, sizes);
+    ranges = NULL;
+  }
+  else {
+    ranges = ALLOCA_N(float*, sizes_len);
+    VALUE* range_ptr = RARRAY_PTR(_ranges);
+    int i;
+    for (i = 0; i < sizes_len; i++) {
+      sizes[i] = NUM2INT(RARRAY_PTR(_sizes)[i]);
+      ranges[i] = ary2fltptr(range_ptr[i], ALLOCA_N(float, 2));
+    }
+  }
+  uniform = TRUE_OR_FALSE(_uniform, 1);
+
+  try {
+    DATA_PTR(self) = cvCreateHist(NUM2INT(_dims), sizes, NUM2INT(_type), ranges, uniform);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return self;
+}
+
+/*
+ * call-seq:
+ *   is_uniform? -> true or false
+ *
+ */
+VALUE
+rb_is_uniform(VALUE self)
+{
+  return CV_IS_UNIFORM_HIST(CVHISTOGRAM(self)) ? Qtrue : Qfalse;
+}
+
+/*
+ * call-seq:
+ *   is_sparse? -> true or false
+ *
+ */
+VALUE
+rb_is_sparse(VALUE self)
+{
+  return CV_IS_SPARSE_HIST(CVHISTOGRAM(self)) ? Qtrue : Qfalse;
+}
+
+/*
+ * call-seq:
+ *   has_range? -> true or false
+ */
+VALUE
+rb_has_range(VALUE self)
+{
+  return CV_HIST_HAS_RANGES(CVHISTOGRAM(self)) ? Qtrue : Qfalse;
+}
+
+VALUE
+rb_calc_hist(int argc, VALUE* argv, VALUE self)
+{
+  return rb_calc_hist_bang(argc, argv, rb_copy_hist(self));
+}
+
+VALUE
+rb_calc_hist_bang(int argc, VALUE* argv, VALUE self)
+{
+  VALUE images, accumulate, mask;
+  rb_scan_args(argc, argv, "12", &images, &accumulate, &mask);
+  Check_Type(images, T_ARRAY);
+  int num_images = RARRAY_LEN(images);
+  IplImage** img = ALLOCA_N(IplImage*, num_images);
+  VALUE* images_ptr = RARRAY_PTR(images);
+  for (int i = 0; i < num_images; i++) {
+    img[i] = IPLIMAGE_WITH_CHECK(images_ptr[i]);
+  }
+  CvMat* m = NIL_P(mask) ? NULL : CVMAT_WITH_CHECK(mask);
+  try {
+    cvCalcHist(img, CVHISTOGRAM(self), TRUE_OR_FALSE(accumulate, 0), m);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  
+  return self;
+}
+
+/*
+ * call-seq:
+ *   [<i>idx1[,idx2]...</i>]
+ */
+VALUE
+rb_aref(VALUE self, VALUE args)
+{
+  int num_idx = RARRAY_LEN(args);
+  int* idx = ALLOCA_N(int, num_idx);
+  VALUE* args_ptr = RARRAY_PTR(args);
+  for (int i = 0; i < num_idx; i++) {
+    idx[i] = NUM2INT(args_ptr[i]);
+  }
+  
+  float value = 0.0;
+  CvHistogram* self_ptr = CVHISTOGRAM(self);
+  try {
+    switch (num_idx) {
+    case 1:
+      value = cvQueryHistValue_1D(self_ptr, idx[0]);
+      break;
+    case 2:
+      value = cvQueryHistValue_2D(self_ptr, idx[0], idx[1]);
+      break;
+    case 3:
+      value = cvQueryHistValue_3D(self_ptr, idx[0], idx[1], idx[2]);
+      break;
+    default:
+      value = cvQueryHistValue_nD(self_ptr, idx);
+      break;
+    }
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  
+  return rb_float_new((double)value);
+}
+
+VALUE
+rb_min_max_value(VALUE self)
+{
+  CvHistogram* self_ptr = CVHISTOGRAM(self);
+  int dims = 0;
+  float min_value = 0.0, max_value = 0.0;
+  int *min_idx = NULL;
+  int *max_idx = NULL;
+  try {
+    dims = cvGetDims(self_ptr->bins, NULL);
+    min_idx = ALLOCA_N(int, dims);
+    max_idx = ALLOCA_N(int, dims);
+    cvGetMinMaxHistValue(CVHISTOGRAM(self), &min_value, &max_value, min_idx, max_idx);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  VALUE _min_idx = rb_ary_new2(dims);
+  VALUE _max_idx = rb_ary_new2(dims);
+  for (int i = 0; i < dims; i++) {
+    rb_ary_store(_min_idx, i, INT2NUM(min_idx[i]));
+    rb_ary_store(_max_idx, i, INT2NUM(max_idx[i]));
+  }
+
+  return rb_ary_new3(4, rb_float_new((double)min_value), rb_float_new((double)max_value),
+  		     _min_idx, _max_idx);
+}
+
+/*
+ * call-seq:
+ *   dims -> [int[,int...]]
+ */
+VALUE
+rb_dims(VALUE self)
+{
+  VALUE _sizes = Qnil;
+  int size[CV_MAX_DIM];
+  int dims = 0;
+  try {
+    dims = cvGetDims(CVHISTOGRAM(self)->bins, size);  
+    _sizes = rb_ary_new2(dims);
+    for (int i = 0; i < dims; ++i) {
+      rb_ary_store(_sizes, i, INT2NUM(size[i]));
+    }
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return rb_assoc_new(INT2NUM(dims), _sizes);
+}
+
+/*
+ * call-seq:
+ *   copy_hist -> cvhist
+ *
+ * Clone histogram.
+ */
+VALUE
+rb_copy_hist(VALUE self)
+{
+  CvHistogram* hist = NULL;
+  try {
+    cvCopyHist(CVHISTOGRAM(self), &hist);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return Data_Wrap_Struct(rb_klass, 0, release_hist, hist);
+}
+
+/*
+ * call-seq:
+ *   clear_hist
+ */
+VALUE
+rb_clear_hist(VALUE self)
+{
+  return rb_clear_hist_bang(rb_copy_hist(self));
+}
+
+/*
+ * call-seq:
+ *   clear_hist!
+ *
+ * Sets all histogram bins to 0 in case of dense histogram and removes all histogram bins in case of sparse array.
+ */
+VALUE
+rb_clear_hist_bang(VALUE self)
+{
+  try {
+    cvClearHist(CVHISTOGRAM(self));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+/*
+ * call-seq:
+ *   normalize(<i>factor</i>) -> cvhist
+ *
+ * Return normalized the histogram bins by scaling them, such that the sum of the bins becomes equal to <i>factor</i>.
+ */
+VALUE
+rb_normalize_hist(VALUE self, VALUE factor)
+{
+  return rb_normalize_hist_bang(rb_copy_hist(self), factor);
+}
+
+/*
+ * call-seq:
+ *   normalize!(<i>factor</i>) -> self
+ *
+ * normalizes the histogram bins by scaling them, such that the sum of the bins becomes equal to <i>factor</i>.
+ */
+VALUE
+rb_normalize_hist_bang(VALUE self, VALUE factor)
+{
+  try {
+    cvNormalizeHist(CVHISTOGRAM(self), NUM2DBL(factor));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+/*
+ * call-seq:
+ *   thresh_hist(<i>threshold</i>) -> cvhist
+ *
+ * Return cleared histogram bins that are below the specified threshold.
+ */
+VALUE
+rb_thresh_hist(VALUE self, VALUE threshold)
+{
+  return rb_thresh_hist_bang(rb_copy_hist(self), threshold);
+}
+
+/*
+ * call-seq:
+ *   thresh_hist!(<i>threshold</i>) -> self
+ *
+ * Cleares histogram bins that are below the specified threshold.
+ */
+VALUE
+rb_thresh_hist_bang(VALUE self, VALUE threshold)
+{
+  try {
+    cvThreshHist(CVHISTOGRAM(self), NUM2DBL(threshold));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+VALUE
+rb_set_hist_bin_ranges(int argc, VALUE* argv, VALUE self)
+{
+  return rb_set_hist_bin_ranges_bang(argc, argv, rb_copy_hist(self));
+}
+
+VALUE
+rb_set_hist_bin_ranges_bang(int argc, VALUE* argv, VALUE self)
+{
+  VALUE _ranges, _uniform;
+  rb_scan_args(argc, argv, "11", &_ranges, &_uniform);
+  Check_Type(_ranges, T_ARRAY);
+
+  int ranges_size = RARRAY_LEN(_ranges);
+  float** ranges = ALLOCA_N(float*, ranges_size);
+  VALUE* range_ptr = RARRAY_PTR(_ranges);
+  for (int i = 0; i < ranges_size; ++i) {
+    ranges[i] = ary2fltptr(range_ptr[i], ALLOCA_N(float, 2));
+  }
+  int uniform = TRUE_OR_FALSE(_uniform, 1);
+
+  try {
+    cvSetHistBinRanges(CVHISTOGRAM(self), ranges, uniform);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return self;
+}
+
+VALUE
+rb_calc_back_project(VALUE self, VALUE image)
+{
+  Check_Type(image, T_ARRAY);
+  int num_images = RARRAY_LEN(image);
+  if (num_images == 0) {
+    return Qnil;
+  }
+  
+  IplImage** img = ALLOCA_N(IplImage*, num_images);
+  VALUE* image_ptr = RARRAY_PTR(image);
+  for (int i = 0; i < num_images; ++i) {
+    img[i] = IPLIMAGE_WITH_CHECK(image_ptr[i]);
+  }
+  
+  CvSize size;
+  size.width = img[0]->width;
+  size.height = img[0]->height;
+  VALUE back_project = cCvMat::new_mat_kind_object(size, image_ptr[0]);
+  try {
+    cvCalcBackProject(img, CVARR(back_project), CVHISTOGRAM(self));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  
+  return back_project;
+}
+
+VALUE
+rb_calc_back_project_patch(VALUE self, VALUE image, VALUE patch_size, VALUE method, VALUE factor)
+{
+  Check_Type(image, T_ARRAY);
+  int num_images = RARRAY_LEN(image);
+  if (num_images == 0) {
+    return Qnil;
+  }
+  
+  IplImage** img = ALLOCA_N(IplImage*, num_images);
+  VALUE* image_ptr = RARRAY_PTR(image);
+  for (int i = 0; i < num_images; ++i) {
+    img[i] = IPLIMAGE_WITH_CHECK(image_ptr[i]);
+  }
+
+  CvSize patchsize = VALUE_TO_CVSIZE(patch_size);
+  CvSize dst_size;
+  dst_size.width = img[0]->width - patchsize.width + 1;
+  dst_size.height = img[0]->height - patchsize.height + 1;
+
+  VALUE dst = cCvMat::new_mat_kind_object(dst_size, image_ptr[0], CV_32F, 1);
+  try {
+    cvCalcBackProjectPatch(img, CVARR(dst), patchsize, CVHISTOGRAM(self),
+			   NUM2INT(method), NUM2DBL(factor));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  
+  return dst;
+}
+
+VALUE
+rb_compare_hist(VALUE self, VALUE hist1, VALUE hist2, VALUE method)
+{
+  double result = 0;
+  try {
+    result = cvCompareHist(CVHISTOGRAM_WITH_CHECK(hist1), CVHISTOGRAM_WITH_CHECK(hist2),
+			   NUM2INT(method));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return rb_float_new(result);
+}
+
+VALUE
+rb_calc_prob_density(int argc, VALUE* argv, VALUE self)
+{
+  VALUE hist1, hist2, scale;
+  rb_scan_args(argc, argv, "21", &hist1, &hist2, &scale);
+  double s = NIL_P(scale) ? 255 : NUM2DBL(scale);
+
+  CvHistogram* hist1_ptr = CVHISTOGRAM_WITH_CHECK(hist1);
+  VALUE dst_hist = rb_allocate(rb_klass);
+  try {
+    cvCopyHist(hist1_ptr, (CvHistogram**)&(DATA_PTR(dst_hist)));
+    cvCalcProbDensity(hist1_ptr, CVHISTOGRAM_WITH_CHECK(hist2), CVHISTOGRAM(dst_hist), s);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return dst_hist;
+}
+
+__NAMESPACE_END_CVHISTOGRAM
+__NAMESPACE_END_OPENCV