afeld-opencv 0.0.8

data/ext/opencv/cvvideowriter.cpp

@@ -0,0 +1,137 @@
+/************************************************************
+
+   cvvideowriter.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvvideowriter.h"
+/*
+ * Document-class: OpenCV::CvVideoWriter
+ *
+ * Create video stream from images.
+ *
+ * C structure is "black box".
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVVIDEOWRITER
+
+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, "CvVideoWriter", rb_cObject);
+  rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
+  rb_define_method(rb_klass, "write", RUBY_METHOD_FUNC(rb_write), 1);
+  rb_define_method(rb_klass, "close", RUBY_METHOD_FUNC(rb_close), 0);
+}
+
+/*
+ * call-seq:
+ *   CvVideoWriter.new(<i>filname, fourcc, fps, size[, is_color]</i>) -> cvvideowriter
+ *   CvVideoWriter.new(<i>filname, fourcc, fps, size[, is_color]</i>){|vw| ... } -> nil
+ *
+ * Open new video writer. If block given, writer is closed automatically when end of block.
+ * 
+ * note: if <i>fourcc</i> is nil, popup codec select dialog (Windows only).
+ */
+VALUE
+rb_initialize(int argc, VALUE *argv, VALUE self)
+{
+  VALUE filename, fourcc, fps, size, is_color_val;
+  rb_scan_args(argc, argv, "41", &filename, &fourcc, &fps, &size, &is_color_val);
+  char codec[4] = {' ', ' ', ' ', ' '};
+  int codec_number;
+  Check_Type(filename, T_STRING);
+  if (RSTRING_LEN(filename) == 0)
+    rb_raise(rb_eArgError, "argument 1 (file name) dose not given");
+  if (NIL_P(fourcc))
+    codec_number = -1;
+  else {
+    Check_Type(fourcc, T_STRING);
+    if (RSTRING_LEN(fourcc) > 4)
+      rb_raise(rb_eStandardError, "argument 2 (fourcc) should be specific 4-character. (i.e \"PIM1\",\"MJPG\")");
+    else {
+      int len = RSTRING_LEN(fourcc);
+      for (int i = 0; i < len; ++i)
+        codec[i] = RSTRING_PTR(fourcc)[i];
+      codec_number = CV_FOURCC(codec[0], codec[1], codec[2], codec[3]);
+    }
+  }
+  int is_color;
+  if (NIL_P(is_color_val))
+    is_color = 1;
+  else
+    is_color = (is_color_val == Qtrue) ? 1 : 0;
+  try {
+    DATA_PTR(self) = cvCreateVideoWriter(StringValueCStr(filename), codec_number,
+					 NUM2DBL(fps), VALUE_TO_CVSIZE(size), is_color);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  if (rb_block_given_p()) {
+    rb_yield(self);
+    rb_close(self);
+    return Qnil;
+  }
+  else
+    return self;
+}
+
+/*
+ * call-seq:
+ *   write(<i>frame</i>)
+ *
+ * Write image as frame of video stream.
+ * <i>frame</i> should be IplImage
+ */
+VALUE
+rb_write(VALUE self, VALUE frame)
+{
+  try {
+    cvWriteFrame(CVVIDEOWRITER(self), IPLIMAGE_WITH_CHECK(frame));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return self;
+}
+
+/*
+ * Close vidoe writer.
+ */
+VALUE
+rb_close(VALUE self)
+{
+  CvVideoWriter *writer = CVVIDEOWRITER(self);
+  try {
+    if (writer)
+      cvReleaseVideoWriter(&writer);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return Qnil;
+}
+
+
+__NAMESPACE_END_CVVIDEOWRITER
+__NAMESPACE_END_OPENCV

data/ext/opencv/cvvideowriter.h

@@ -0,0 +1,43 @@
+/************************************************************
+
+   cvvideowriter.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVVIDEOWRITER_H
+#define RUBY_OPENCV_CVVIDEOWRITER_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVVIDEOWRITER namespace cCvVideoWriter {
+#define __NAMESPACE_END_CVVIDEOWRITER }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVVIDEOWRITER
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_write(VALUE self, VALUE frame);
+VALUE rb_close(VALUE self);
+
+__NAMESPACE_END_CVVIDEOWRITER
+
+inline CvVideoWriter*
+CVVIDEOWRITER(VALUE object)
+{
+  // CvVideoWriter is 
+  // CvVideoWriter *ptr;
+  // Data_Get_Struct(object, CvVideoWriter, ptr);
+  // return ptr;
+  return (CvVideoWriter*)DATA_PTR(object);
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVVIDEOWRITER_H

data/ext/opencv/gui.cpp

@@ -0,0 +1,68 @@
+/************************************************************
+
+  gui.cpp -
+
+  $Author: lsxi $
+
+  Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "gui.h"
+/*
+ * Document-module: OpenCV::GUI
+ *
+ * OpenCV simple GUI interface. Provides Window, Trackbar.
+ * This GUI work on GTK or Microsoft Windows native GUI.
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_GUI
+    
+VALUE rb_module;
+
+VALUE
+rb_module_GUI()
+{
+  return rb_module;
+}
+
+void
+define_ruby_module()
+{
+  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);
+}
+
+/*
+ * call-seq:
+ *   wait_key(<i>[delay]</i>) -> int or nil
+ *
+ * Waits for a pressed key each GUI wedget.
+ * Return the code of the pressed key or nil if no key were pressed until the specified timeout has elapsed.
+ * <i>delay</i> should be Fixnum. Wait <i>delay</i> millisecond.
+ */
+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);
+}
+
+__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 define_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,192 @@
+/************************************************************
+
+   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;
+}
+
+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, "IplConvKernel", 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, "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);
+}
+
+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 = ALLOCA_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);
+}
+
+__NAMESPACE_END_IPLCONVKERNEL
+__NAMESPACE_END_OPENCV