ruby-opencv 0.0.10 → 0.0.11.pre

data/ext/opencv/eigenfaces.cpp

@@ -0,0 +1,67 @@
+/************************************************************
+
+   eigenfaces.cpp -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#include <stdio.h>
+#include "eigenfaces.h"
+/*
+ * Document-class: OpenCV::EigenFaces
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_EIGENFACES
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+/* 
+ * call-seq:
+ *   EigenFaces.new(num_components=0, threshold=DBL_MAX)
+ */
+VALUE
+rb_initialize(int argc, VALUE argv[], VALUE self)
+{
+  VALUE num_components_val, threshold_val;
+  rb_scan_args(argc, argv, "02", &num_components_val, &threshold_val);
+
+  int num_components = NIL_P(num_components_val) ? 0 : NUM2INT(num_components_val);
+  double threshold = NIL_P(threshold_val) ? DBL_MAX : NUM2DBL(threshold_val);
+
+  free(DATA_PTR(self));
+  cv::Ptr<cv::FaceRecognizer> ptr = cv::createEigenFaceRecognizer(num_components, threshold);
+  DATA_PTR(self) = ptr;
+
+  cFaceRecognizer::guard_facerecognizer(DATA_PTR(self), ptr);
+
+  return self;
+}
+
+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, "EigenFaces", cFaceRecognizer::rb_class());
+  rb_define_alloc_func(rb_klass, cFaceRecognizer::allocate_facerecognizer);
+  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
+}
+
+__NAMESPACE_END_EIGENFACES
+__NAMESPACE_END_OPENCV
+

data/ext/opencv/eigenfaces.h

@@ -0,0 +1,30 @@
+/************************************************************
+
+   eigenfaces.h
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#ifndef RUBY_OPENCV_EIGENFACES_H
+#define RUBY_OPENCV_EIGENFACES_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_EIGENFACES namespace cEigenFaces {
+#define __NAMESPACE_END_EIGENFACES }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_EIGENFACES
+
+VALUE rb_class();
+
+void define_ruby_class();
+VALUE rb_initialize(int argc, VALUE argv[], VALUE self);
+
+__NAMESPACE_END_EIGENFACES
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_EIGENFACES_H
+

data/ext/opencv/facerecognizer.cpp

@@ -0,0 +1,174 @@
+/************************************************************
+
+   facerecognizer.cpp -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#include <stdio.h>
+#include "facerecognizer.h"
+/*
+ * Document-class: OpenCV::FaceRecognizer
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_FACERECOGNIZER
+
+VALUE rb_klass;
+
+std::map< long, cv::Ptr<cv::FaceRecognizer> > ptr_guard_map;
+
+void
+guard_facerecognizer(void* data_ptr, cv::Ptr<cv::FaceRecognizer> ptr)
+{
+  ptr_guard_map[(long)data_ptr] = ptr;
+}
+
+void
+release_facerecognizer(void *ptr)
+{
+  long key = (long)ptr;
+  ptr_guard_map[key].release();
+  ptr_guard_map.erase(key);
+}
+
+VALUE
+allocate_facerecognizer(VALUE klass)
+{
+  return Data_Wrap_Struct(klass, 0, release_facerecognizer, NULL);
+}
+
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+/*
+ * call-seq:
+ *   train(src, labels)
+ *
+ * Trains a FaceRecognizer with given data and associated labels.
+ */
+VALUE
+rb_train(VALUE self, VALUE src, VALUE labels)
+{
+  Check_Type(src, T_ARRAY);
+  Check_Type(labels, T_ARRAY);
+
+  VALUE *src_ptr = RARRAY_PTR(src);
+  int src_size = RARRAY_LEN(src);
+  std::vector<cv::Mat> images;
+  for (int i = 0; i < src_size; i++) {
+    images.push_back(cv::Mat(CVMAT_WITH_CHECK(src_ptr[i])));
+  }
+
+  VALUE *labels_ptr = RARRAY_PTR(labels);
+  int labels_size = RARRAY_LEN(labels);
+  std::vector<int> local_labels;
+  for (int i = 0; i < labels_size; i++) {
+    local_labels.push_back(NUM2INT(labels_ptr[i]));
+  }
+
+  cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
+  try {
+    self_ptr->train(images, local_labels);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return Qnil;
+}
+
+/*
+ * call-seq:
+ *   predict(src)
+ * 
+ * Predicts a label and associated confidence (e.g. distance) for a given input image.
+ */
+VALUE
+rb_predict(VALUE self, VALUE src)
+{
+  cv::Mat mat = cv::Mat(CVMAT_WITH_CHECK(src));
+  cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
+  int label;
+  double confidence;
+  try {
+    self_ptr->predict(mat, label, confidence);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return rb_ary_new3(2, INT2NUM(label), DBL2NUM(confidence));
+}
+
+
+/*
+ * call-seq:
+ *   save(filename)
+ *
+ * Saves this model to a given filename, either as XML or YAML.
+ */
+VALUE
+rb_save(VALUE self, VALUE filename)
+{
+  Check_Type(filename, T_STRING);
+  cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
+  try {
+    char* s = StringValueCStr(filename);
+    self_ptr->save(std::string(s));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return Qnil;
+}
+
+/*
+ * call-seq:
+ *   load(filename)
+ *
+ * Loads a FaceRecognizer and its model state.
+ */
+VALUE
+rb_load(VALUE self, VALUE filename)
+{
+  Check_Type(filename, T_STRING);
+  cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
+  try {
+    char* s = StringValueCStr(filename);
+    self_ptr->load(std::string(s));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+
+  return Qnil;
+}
+
+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, "FaceRecognizer", cAlgorithm::rb_class());
+  rb_define_method(rb_klass, "train", RUBY_METHOD_FUNC(rb_train), 2);
+  rb_define_method(rb_klass, "predict", RUBY_METHOD_FUNC(rb_predict), 1);
+  rb_define_method(rb_klass, "save", RUBY_METHOD_FUNC(rb_save), 1);
+  rb_define_method(rb_klass, "load", RUBY_METHOD_FUNC(rb_load), 1);
+}
+
+__NAMESPACE_END_FACERECOGNIZER
+__NAMESPACE_END_OPENCV

data/ext/opencv/facerecognizer.h

@@ -0,0 +1,46 @@
+/************************************************************
+
+   facerecognizer.h
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#ifndef RUBY_OPENCV_FACERECOGNIZER_H
+#define RUBY_OPENCV_FACERECOGNIZER_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_FACERECOGNIZER namespace cFaceRecognizer {
+#define __NAMESPACE_END_FACERECOGNIZER }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_FACERECOGNIZER
+
+VALUE rb_class();
+
+void define_ruby_class();
+VALUE rb_train(VALUE self, VALUE src, VALUE labels);
+VALUE rb_predict(VALUE self, VALUE src);
+VALUE rb_save(VALUE self, VALUE filename);
+VALUE rb_load(VALUE self, VALUE filename);
+
+void guard_facerecognizer(void* data_ptr, cv::Ptr<cv::FaceRecognizer> ptr);
+void release_facerecognizer(void *ptr);
+VALUE allocate_facerecognizer(VALUE klass);
+
+__NAMESPACE_END_FACERECOGNIZER
+
+inline cv::FaceRecognizer*
+FACERECOGNIZER(VALUE object)
+{
+  cv::FaceRecognizer *ptr;
+  Data_Get_Struct(object, cv::FaceRecognizer, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_FACERECOGNIZER_H
+

data/ext/opencv/fisherfaces.cpp

@@ -0,0 +1,67 @@
+/************************************************************
+
+   fisherfaces.cpp -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#include <stdio.h>
+#include "fisherfaces.h"
+/*
+ * Document-class: OpenCV::FisherFaces
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_FISHERFACES
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+/* 
+ * call-seq:
+ *   FisherFaces.new(num_components=0, threshold=DBL_MAX)
+ */
+VALUE
+rb_initialize(int argc, VALUE argv[], VALUE self)
+{
+  VALUE num_components_val, threshold_val;
+  rb_scan_args(argc, argv, "02", &num_components_val, &threshold_val);
+
+  int num_components = NIL_P(num_components_val) ? 0 : NUM2INT(num_components_val);
+  double threshold = NIL_P(threshold_val) ? DBL_MAX : NUM2DBL(threshold_val);
+
+  free(DATA_PTR(self));
+  cv::Ptr<cv::FaceRecognizer> ptr = cv::createFisherFaceRecognizer(num_components, threshold);
+  DATA_PTR(self) = ptr;
+
+  cFaceRecognizer::guard_facerecognizer(DATA_PTR(self), ptr);
+
+  return self;
+}
+
+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, "FisherFaces", cFaceRecognizer::rb_class());
+  rb_define_alloc_func(rb_klass, cFaceRecognizer::allocate_facerecognizer);
+  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
+}
+
+__NAMESPACE_END_FISHERFACES
+__NAMESPACE_END_OPENCV
+

data/ext/opencv/fisherfaces.h

@@ -0,0 +1,30 @@
+/************************************************************
+
+   fisherfaces.h
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#ifndef RUBY_OPENCV_FISHERFACES_H
+#define RUBY_OPENCV_FISHERFACES_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_FISHERFACES namespace cFisherFaces {
+#define __NAMESPACE_END_FISHERFACES }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_FISHERFACES
+
+VALUE rb_class();
+
+void define_ruby_class();
+VALUE rb_initialize(int argc, VALUE argv[], VALUE self);
+
+__NAMESPACE_END_FISHERFACES
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_FISHERFACES_H
+

data/ext/opencv/lbph.cpp

@@ -0,0 +1,70 @@
+/************************************************************
+
+   lbph.cpp -
+
+   $Author: ser1zw $
+
+   Copyright (C) 2013 ser1zw
+
+************************************************************/
+#include <stdio.h>
+#include "lbph.h"
+/*
+ * Document-class: OpenCV::LBPH
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_LBPH
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+/* 
+ * call-seq:
+ *   LBPH.new(radius=1, neighbors=8, grid_x=8, grid_y=8, threshold=DBL_MAX) -> cvmat
+ */
+VALUE
+rb_initialize(int argc, VALUE argv[], VALUE self)
+{
+  VALUE radius_val, neighbors_val, grid_x_val, grid_y_val, threshold_val;
+  rb_scan_args(argc, argv, "05", &radius_val, &neighbors_val, &grid_x_val, &grid_y_val, &threshold_val);
+
+  int radius = NIL_P(radius_val) ? 1 : NUM2INT(radius_val);
+  int neighbors = NIL_P(neighbors_val) ? 8 : NUM2INT(neighbors_val);
+  int grid_x = NIL_P(grid_x_val) ? 8 : NUM2INT(grid_x_val);
+  int grid_y = NIL_P(grid_y_val) ? 8 : NUM2INT(grid_y_val);
+  double threshold = NIL_P(threshold_val) ? DBL_MAX : NUM2INT(threshold_val);
+
+  free(DATA_PTR(self));
+  cv::Ptr<cv::FaceRecognizer> ptr = cv::createLBPHFaceRecognizer(radius, neighbors, grid_x, grid_y, threshold);
+  DATA_PTR(self) = ptr;
+
+  cFaceRecognizer::guard_facerecognizer(DATA_PTR(self), ptr);
+
+  return self;
+}
+
+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, "LBPH", cFaceRecognizer::rb_class());
+  rb_define_alloc_func(rb_klass, cFaceRecognizer::allocate_facerecognizer);
+  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
+}
+
+__NAMESPACE_END_LBPH
+__NAMESPACE_END_OPENCV
+