bantic-ruby-opencv 0.0.8

data/ext/cvhaarclassifiercascade.cpp

@@ -0,0 +1,159 @@
+/************************************************************
+
+  cvhaarclassifercascade.cpp -
+
+  $Author: lsxi $
+
+  Copyright (C) 2005-2007 Masakazu Yonekura
+
+************************************************************/
+#include "cvhaarclassifiercascade.h"
+/*
+ * Document-class: OpenCV::CvHaarClassifierCascade
+ *
+ * CvHaarClassifierCascade object is "fast-object-detector".
+ * This detector can discover object (e.g. human's face) from image.
+ *
+ * Find face-area from picture "lena"...
+ * link:../images/face_detect_from_lena.jpg
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVHAARCLASSIFERCASCADE
+
+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, "CvHaarClassifierCascade", rb_cObject);
+  rb_define_alloc_func(rb_klass, rb_allocate);
+  rb_define_singleton_method(rb_klass, "load", RUBY_METHOD_FUNC(rb_load), 1);
+  rb_define_method(rb_klass, "detect_objects", RUBY_METHOD_FUNC(rb_detect_objects), -1);
+  rb_define_method(rb_klass, "detect_objects_with_pruning", RUBY_METHOD_FUNC(rb_detect_objects_with_pruning), -1);
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  return OPENCV_OBJECT(klass, 0);
+}
+
+/*
+ * call-seq:
+ *   CvHaarClassiferCascade.load(<i>path</i>) -> object-detector
+ * 
+ * Load trained cascade of haar classifers from file.
+ * Object detection classifiers are stored in XML or YAML files.
+ * sample of object detection classifier files is included by OpenCV.
+ *
+ * You can found these at
+ *    C:\Program Files\OpenCV\data\haarcascades\*.xml (Windows, default install path)
+ *
+ * e.g. you want to try to detect human's face.
+ *    detector = CvHaarClassiferCascade.load("haarcascade_frontalface_alt.xml")
+ */
+VALUE
+rb_load(VALUE klass, VALUE path)
+{
+  CvHaarClassifierCascade *cascade = (CvHaarClassifierCascade*)cvLoad(StringValueCStr(path), 0, 0, 0);
+  if(!CV_IS_HAAR_CLASSIFIER(cascade))
+    rb_raise(rb_eTypeError, "invalid format haar classifier cascade file.");
+  return OPENCV_OBJECT(rb_klass, cascade);
+}
+
+VALUE
+rb_save(VALUE self, VALUE path)
+{
+  rb_raise(rb_eNotImpError, "");
+}
+
+/*
+ * call-seq:
+ *   detect_objects(image[,scale_factor = 1.1, min_neighbor = 3, min_size = CvSize.new(0,0)]) -> cvseq(include CvAvgComp object)
+ *   detect_objects(image[,scale_factor = 1.1, min_neighbor = 3, min_size = CvSize.new(0,0)]){|cmp| ... } -> cvseq(include CvAvgComp object)
+ *
+ * Detects objects in the image. This method finds rectangular regions in the
+ * given image that are likely to contain objects the cascade has been trained
+ * for and return those regions as a sequence of rectangles.
+ *
+ * * scale_factor (should be > 1.0)
+ *     The factor by which the search window is scaled between the subsequent scans, for example, 1.1 mean increasing window by 10%.
+ * * min_neighbors
+ *     Minimum number (minus 1) of neighbor rectangles that makes up an object.
+ *     All the groups of a smaller number of rectangles than min_neighbors - 1 are rejected.
+ *     If min_neighbors is 0, the function does not any grouping at all and returns all the detected
+ *     candidate rectangles, whitch many be useful if the user wants to apply a customized grouping procedure.
+ * * min_size
+ *     Minimum window size. By default, it is set to size of samples the classifier has been trained on.
+ */
+VALUE 
+rb_detect_objects(int argc, VALUE *argv, VALUE self)
+{ 
+  VALUE image, storage, scale_factor, min_neighbors, min_size, result;
+  rb_scan_args(argc, argv, "14", &image, &storage, &scale_factor, &min_neighbors, &min_size);
+  if (!rb_obj_is_kind_of(image, cCvMat::rb_class()))
+    rb_raise(rb_eTypeError, "argument 1(target-image) should be %s.", rb_class2name(cCvMat::rb_class()));
+  double scale = IF_DBL(scale_factor, 1.1);
+  if (!(scale > 1.0))
+    rb_raise(rb_eArgError, "argument 2 (scale factor) must > 1.0.");
+  storage = CHECK_CVMEMSTORAGE(storage);
+  CvSeq *seq = cvHaarDetectObjects(CVMAT(image), CVHAARCLASSIFIERCASCADE(self), CVMEMSTORAGE(storage),
+                                   scale, IF_INT(min_neighbors, 3), 0, NIL_P(min_size) ? cvSize(0,0) : VALUE_TO_CVSIZE(min_size));
+  result = cCvSeq::new_sequence(cCvSeq::rb_class(), seq, cCvAvgComp::rb_class(), storage);
+  if (rb_block_given_p()) {
+    for(int i = 0; i < seq->total; i++)
+      rb_yield(REFER_OBJECT(cCvAvgComp::rb_class(), cvGetSeqElem(seq, i), storage));      
+  }
+  return result;
+}
+
+/*
+ * call-seq:
+ *   detect_objects_with_pruning(image[,scale_factor = 1.1, min_neighbor = 3, min_size = CvSize.new(0,0)]) -> cvseq(include CvAvgComp object)
+ *   detect_objects_with_pruning(image[,scale_factor = 1.1, min_neighbor = 3, min_size = CvSize.new(0,0)]){|cmp| ... } -> cvseq(include CvAvgComp object)
+ *
+ * Almost same to #detect_objects (Return detected objects).
+ * 
+ * Before scanning to image, Canny edge detector to reject some image regions
+ * that contain too few or too much edges, and thus can not contain the searched object.
+ *
+ *   note: The particular threshold values are tuned for face detection.
+ *         And in this case the pruning speeds up the processing.
+ */ 
+VALUE
+rb_detect_objects_with_pruning(int argc, VALUE *argv, VALUE self)
+{
+  VALUE image, storage, scale_factor, min_neighbors, min_size, result;
+  rb_scan_args(argc, argv, "14", &image, &storage, &scale_factor, &min_neighbors, &min_size);
+  if (!rb_obj_is_kind_of(image, cCvMat::rb_class()))
+    rb_raise(rb_eTypeError, "argument 1(target-image) should be %s.", rb_class2name(cCvMat::rb_class()));
+  double scale = IF_DBL(scale_factor, 1.1);
+  if (!(scale > 1.0))
+    rb_raise(rb_eArgError, "argument 2 (scale factor) must > 1.0.");
+  storage = CHECK_CVMEMSTORAGE(storage);
+  CvSeq *seq = cvHaarDetectObjects(CVMAT(image), CVHAARCLASSIFIERCASCADE(self), CVMEMSTORAGE(storage),
+                                   scale, IF_INT(min_neighbors, 3), CV_HAAR_DO_CANNY_PRUNING, NIL_P(min_size) ? cvSize(0,0) : VALUE_TO_CVSIZE(min_size));
+  result = cCvSeq::new_sequence(cCvSeq::rb_class(), seq, cCvAvgComp::rb_class(), storage);
+  if (rb_block_given_p()) {
+    for(int i = 0; i < seq->total; i++)
+      rb_yield(REFER_OBJECT(cCvAvgComp::rb_class(), cvGetSeqElem(seq, i), storage));      
+  }
+  return result;
+}
+
+__NAMESPACE_END_CVHAARCLASSIFERCASCADE
+__NAMESPACE_END_OPENCV

data/ext/cvhaarclassifiercascade.h

@@ -0,0 +1,41 @@
+/************************************************************
+
+   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_initialize(int argc, VALUE *argv, VALUE self);
+
+VALUE rb_load(VALUE klass, VALUE path);
+VALUE rb_save(VALUE self, VALUE name);
+VALUE rb_detect_objects(int argc, VALUE *argv, VALUE self);
+VALUE rb_detect_objects_with_pruning(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/cvhistogram.cpp

@@ -0,0 +1,200 @@
+/************************************************************
+
+   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_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, "normalize", RUBY_METHOD_FUNC(rb_normalize), 1);
+  rb_define_method(rb_klass, "normalize!", RUBY_METHOD_FUNC(rb_normalize_bang), 1);
+  rb_define_method(rb_klass, "thresh", RUBY_METHOD_FUNC(rb_thresh), 1);
+  rb_define_alias(rb_klass, "threshold", "thresh");
+  rb_define_method(rb_klass, "thresh!", RUBY_METHOD_FUNC(rb_thresh_bang), 1);
+  rb_define_alias(rb_klass, "threshold!", "thresh!");
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  // not yet
+  return Qnil;
+}
+
+/*
+ * 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;
+}
+
+/*
+ * call-seq:
+ *   dims -> [int[,int...]]
+ */
+VALUE
+rb_dims(VALUE self)
+{
+  int size[CV_MAX_DIM];
+  int dims = cvGetDims(CVHISTOGRAM(self)->bins, size);
+  VALUE result = rb_ary_new2(dims);
+  for(int i = 0; i < dims; i++){
+    rb_ary_store(result, i, INT2FIX(size[i]));
+  }
+  return result;
+}
+
+/*
+ * call-seq:
+ *   bins -> cvmatnd or cvsparsemat
+ */
+VALUE
+rb_bins(VALUE self)
+{
+  CvHistogram *hist = CVHISTOGRAM(self);
+  return REFER_OBJECT(CV_IS_SPARSE_HIST(hist) ? cCvSparseMat::rb_class() : cCvMatND::rb_class(), hist->bins, self);
+}
+
+/*
+ * call-seq:
+ *   copy -> cvhist
+ *
+ * Clone histogram.
+ */
+VALUE
+rb_copy(VALUE self)
+{
+  VALUE dest = 0;
+  CvHistogram *hist = CVHISTOGRAM(dest);
+  cvCopyHist(CVHISTOGRAM(self), &hist);
+  return dest;
+}
+
+/*
+ * call-seq:
+ *   clear!
+ *
+ * Sets all histogram bins to 0 in case of dense histogram and removes all histogram bins in case of sparse array.
+ */
+VALUE
+rb_clear_bang(VALUE self)
+{
+  cvClearHist(CVHISTOGRAM(self));
+  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(VALUE self, VALUE factor)
+{
+  return rb_normalize_bang(rb_copy(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_bang(VALUE self, VALUE factor)
+{
+  cvNormalizeHist(CVHISTOGRAM(self), NUM2DBL(factor));
+  return self;
+}
+
+/*
+ * call-seq:
+ *   thresh(<i>factor</i>) -> cvhist
+ *
+ * Return cleared histogram bins that are below the specified threshold.
+ */
+VALUE
+rb_thresh(VALUE self, VALUE factor)
+{
+  return rb_thresh_bang(rb_copy(self), factor);
+}
+
+/*
+ * call-seq:
+ *   thresh!(<i>factor</i>) -> self
+ *
+ * Cleares histogram bins that are below the specified threshold.
+ */
+VALUE
+rb_thresh_bang(VALUE self, VALUE factor)
+{
+  cvThreshHist(CVHISTOGRAM(self), NUM2DBL(factor));
+  return self;
+}
+
+
+__NAMESPACE_END_CVHISTOGRAM
+__NAMESPACE_END_OPENCV

data/ext/cvhistogram.h

@@ -0,0 +1,51 @@
+/************************************************************
+
+   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();
+
+void define_ruby_class();
+
+VALUE rb_is_uniform(VALUE self);
+VALUE rb_is_sparse(VALUE self);
+VALUE rb_has_range(VALUE self);
+VALUE rb_dims(VALUE self);
+VALUE rb_bins(VALUE self);
+
+VALUE rb_clear(VALUE self);
+VALUE rb_clear_bang(VALUE self);
+
+VALUE rb_normalize(VALUE self, VALUE factor);
+VALUE rb_normalize_bang(VALUE self, VALUE factor);
+VALUE rb_thresh(VALUE self, VALUE factor);
+VALUE rb_thresh_bang(VALUE self, VALUE factor);
+
+__NAMESPACE_END_CVHISTOGRAM
+
+inline CvHistogram
+*CVHISTOGRAM(VALUE object)
+{
+  CvHistogram *ptr;
+  Data_Get_Struct(object, CvHistogram, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVHISTOGRAM_H