jf-ruby-opencv 0.0.18

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 init_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 *data_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,75 @@
+/************************************************************
+
+   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
+init_ruby_class()
+{
+#if 0
+  // For documentation using YARD
+  VALUE opencv = rb_define_module("OpenCV");
+  VALUE alghorithm = rb_define_class_under(opencv, "Algorithm", rb_cObject);
+  VALUE face_recognizer = rb_define_class_under(opencv, "FaceRecognizer", alghorithm);
+#endif
+
+  if (rb_klass)
+    return;
+  /* 
+   * opencv = rb_define_module("OpenCV");
+   * 
+   * note: this comment is used by rdoc.
+   */
+  VALUE opencv = rb_module_opencv();
+  VALUE face_recognizer = cFaceRecognizer::rb_class();
+  rb_klass = rb_define_class_under(opencv, "FisherFaces", face_recognizer);
+  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 init_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/gui.cpp

@@ -0,0 +1,71 @@
+/************************************************************
+
+  gui.cpp -
+
+  $Author: lsxi $
+
+  Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "gui.h"
+/*
+ * Document-module: OpenCV::GUI
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_GUI
+    
+VALUE rb_module;
+
+VALUE
+rb_module_GUI()
+{
+  return rb_module;
+}
+
+/*
+ * Waits for a pressed key.
+ *
+ * @overload wait_key(delay = 0)
+ *   @param delay [Integer] Delay in milliseconds. 0 is the special value that means "forever".
+ * @return [Number] The code of the pressed key or <tt>nil</tt> if no key was pressed
+ *   before the specified time had elapsed.
+ * @opencv_func cvWaitKey
+ */
+VALUE
+rb_wait_key(int argc, VALUE *argv, VALUE self)
+{
+  VALUE delay;
+  rb_scan_args(argc, argv, "01", &delay);
+  int keycode = 0;
+  try {
+    keycode = cvWaitKey(IF_INT(delay, 0));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return (keycode < 0) ? Qnil : INT2NUM(keycode);
+}
+
+void
+init_ruby_module()
+{
+#if 0
+  // For documentation using YARD
+  VALUE opencv = rb_define_module("OpenCV");
+#endif
+
+  if (rb_module)
+    return;
+  /* 
+   * opencv = rb_define_module("OpenCV");      
+   *
+   * note: this comment is used by rdoc.
+   */
+  VALUE opencv = rb_module_opencv();
+  rb_module = rb_define_module_under(opencv, "GUI");
+  rb_define_singleton_method(rb_module, "wait_key", RUBY_METHOD_FUNC(rb_wait_key), -1);
+}
+
+__NAMESPACE_END_GUI
+__NAMESPACE_END_OPENCV
+

data/ext/opencv/gui.h

@@ -0,0 +1,30 @@
+/************************************************************
+
+   gui.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_GUI_H
+#define RUBY_OPENCV_GUI_H
+
+#define __NAMESPACE_BEGIN_GUI namespace mGUI {
+#define __NAMESPACE_END_GUI }
+
+#include "opencv.h"
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_GUI
+
+VALUE rb_module_GUI();
+void init_ruby_module();
+
+VALUE rb_wait_key(int argc, VALUE *argv, VALUE self);
+
+__NAMESPACE_END_GUI
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_GUI_H
+

data/ext/opencv/iplconvkernel.cpp

@@ -0,0 +1,198 @@
+/************************************************************
+
+   iplconvkernel.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "iplconvkernel.h"
+/*
+ * Document-class: OpenCV::IplConvKernel
+ *
+ * Structuring element can be used in the morphological operations.
+ *
+ * CvMat#erode, CvMat#dilate, CvMat#morphology_open, CvMat#morphology_close,
+ * CvMat#morphology_gradient, CvMat#morphology_tophat, CvMat#morphology_blackhat
+ *
+ * Create by IplConvKernel.new or CvMat#to_IplConvKernel
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_IPLCONVKERNEL
+
+VALUE rb_klass;
+
+VALUE
+rb_class()
+{
+  return rb_klass;
+}
+
+VALUE
+rb_allocate(VALUE klass)
+{
+  return IPLCONVKERNEL_OBJECT(klass, 0);
+}
+
+/*
+ * call-seq:
+ *   IplConvKernel.new(cols, rows, anchor_x, anchor_y, shape [,values = nil])
+ *
+ * Creates structuring element.
+ *  cols
+ *    Number of columns in the structuring element. 
+ *  rows
+ *    Number of rows in the structuring element. 
+ *  anchor_x
+ *    Relative horizontal offset of the anchor point. 
+ *  anchor_y
+ *    Relative vertical offset of the anchor point.
+ *  shape
+ *    Shape of the structuring element; may have the following values:
+ *     :rect
+ *     :cross
+ *     :ellipse
+ *     :custom
+ *
+ */
+VALUE
+rb_initialize(int argc, VALUE *argv, VALUE self)
+{
+  VALUE shape, rows, cols, anchor_x, anchor_y, values;
+  rb_scan_args(argc, argv, "51", &cols, &rows, &anchor_x, &anchor_y, &shape, &values);
+  int shape_type;
+  int _cols = NUM2INT(cols);
+  int _rows = NUM2INT(rows);
+  int num_values;
+  int *_values = NULL;
+  const int INVALID_SHAPE = -1;
+  
+  shape_type = CVMETHOD("STRUCTURING_ELEMENT_SHAPE", shape, INVALID_SHAPE);
+  if (shape_type == INVALID_SHAPE)
+    rb_raise(rb_eTypeError, "argument 1 (shape) should be :rect or :cross or :ellipse or :custom.");
+  if (shape_type == CV_SHAPE_CUSTOM) {
+    if (NIL_P(values))
+      rb_raise(rb_eArgError, "argument 6 (values) should not be nil when the shape is :custom.");
+    num_values = RARRAY_LEN(values);
+    _values = RB_ALLOC_N(int, num_values);
+    VALUE *values_ptr = RARRAY_PTR(values);
+    for (int i = 0; i < num_values; ++i)
+      _values[i] = NUM2INT(values_ptr[i]);
+  }
+  try {
+    DATA_PTR(self) = rb_cvCreateStructuringElementEx(_cols, _rows, NUM2INT(anchor_x), NUM2INT(anchor_y),
+						     shape_type, _values);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+/*
+ * call-seq:
+ *   size -> cvsize
+ *
+ * Return the structuring element's size.
+ */
+VALUE
+rb_size(VALUE self)
+{
+  IplConvKernel *kernel = IPLCONVKERNEL(self);
+  return cCvSize::new_object(cvSize(kernel->nCols, kernel->nRows));
+}
+
+/*
+ * call-seq:
+ *   cols -> int
+ *
+ * Return number of columns in the structuring element.
+ */
+VALUE
+rb_cols(VALUE self)
+{
+  return INT2NUM(IPLCONVKERNEL(self)->nCols);
+}
+
+/*
+ * call-seq:
+ *   rows -> int
+ *
+ * Return number of rows in the structuring element.
+ */
+VALUE
+rb_rows(VALUE self)
+{
+  return INT2NUM(IPLCONVKERNEL(self)->nRows);
+}
+
+/*
+ * call-seq:
+ *   anchor -> cvpoint
+ *
+ * Return anchor of the structuring element.
+ */
+VALUE
+rb_anchor(VALUE self)
+{
+  IplConvKernel *kernel = IPLCONVKERNEL(self);
+  return cCvPoint::new_object(cvPoint(kernel->anchorX, kernel->anchorY));
+}
+
+/*
+ * call-seq:
+ *   anchor_x -> int
+ *
+ * Return relative horizontal offset of the anchor point.
+ */
+VALUE
+rb_anchor_x(VALUE self)
+{
+  return INT2NUM(IPLCONVKERNEL(self)->anchorX);
+}
+
+/*
+ * call-seq:
+ *   anchor_y -> int
+ *
+ * Return relative vertical offset of the anchor point.
+ */
+VALUE
+rb_anchor_y(VALUE self)
+{
+  return INT2NUM(IPLCONVKERNEL(self)->anchorY);
+}
+
+void
+init_ruby_class()
+{
+#if 0
+  // For documentation using YARD
+  VALUE opencv = rb_define_module("OpenCV");
+#endif
+
+  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, "IplConvKernel", 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, "size", RUBY_METHOD_FUNC(rb_size), 0);
+  rb_define_method(rb_klass, "cols", RUBY_METHOD_FUNC(rb_cols), 0);
+  rb_define_alias(rb_klass, "columns", "cols");
+  rb_define_method(rb_klass, "rows", RUBY_METHOD_FUNC(rb_rows), 0);
+  rb_define_method(rb_klass, "anchor", RUBY_METHOD_FUNC(rb_anchor), 0);
+  rb_define_method(rb_klass, "anchor_x", RUBY_METHOD_FUNC(rb_anchor_x), 0);
+  rb_define_method(rb_klass, "anchor_y", RUBY_METHOD_FUNC(rb_anchor_y), 0);
+}
+
+__NAMESPACE_END_IPLCONVKERNEL
+__NAMESPACE_END_OPENCV
+