opencv 0.0.6

data/ext/cvvector.h

@@ -0,0 +1,54 @@
+/************************************************************
+
+   cvvector.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2007 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVVECTOR_H
+#define RUBY_OPENCV_CVVECTOR_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVVECTOR namespace cCvVector{
+#define __NAMESPACE_END_CVVECTOR }
+
+typedef struct CvVector{
+  float x;
+  float y;
+} CvVector;
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVVECTOR
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_compatible_q(VALUE klass, VALUE object);
+
+VALUE rb_allocate(VALUE klass);
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_x(VALUE self);
+VALUE rb_set_x(VALUE self, VALUE x);
+VALUE rb_y(VALUE self);
+VALUE rb_set_y(VALUE self, VALUE y);
+
+VALUE rb_to_s(VALUE self);
+VALUE rb_to_ary(VALUE self);
+
+VALUE new_object(CvVector vector);
+
+__NAMESPACE_END_CVVECTOR
+
+inline CvVector *CVVECTOR(VALUE object){
+  CvVector *ptr;
+  Data_Get_Struct(object, CvVector, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVVECTOR_H

data/ext/cvvideowriter.cpp

@@ -0,0 +1,116 @@
+/************************************************************
+
+   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>filanem, fourcc, fps, size</i>) -> cvvideowriter
+ *   CvVideoWriter.new(<i>filanem, fourcc, fps, size</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;
+  rb_scan_args(argc, argv, "40", &filename, &fourcc, &fps, &size);
+  char codec[4] = {' ', ' ', ' ', ' '};
+  int codec_number;
+  Check_Type(filename, T_STRING);
+  if (RSTRING(filename)->len == 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(fourcc)->len > 4)
+      rb_raise(rb_eStandardError, "argument 2 (fourcc) should be specific 4-character. (i.e \"PIM1\",\"MJPG\")");
+    else {
+      for (int i = 0; i < RSTRING(fourcc)->len; i++)
+        codec[i] = RSTRING(fourcc)->ptr[i];
+      codec_number = CV_FOURCC(codec[0], codec[1], codec[2], codec[3]);
+    }
+  }
+  DATA_PTR(self) = cvCreateVideoWriter(StringValueCStr(filename), codec_number, FIX2INT(fps), VALUE_TO_CVSIZE(size));
+  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 CvMat or subclass.
+ */
+VALUE
+rb_write(VALUE self, VALUE frame)
+{
+  if (!rb_obj_is_kind_of(frame, cCvMat::rb_class()))
+    rb_raise(rb_eTypeError, "argument 1 (frame) should be %s or subclass", rb_class2name(cCvMat::rb_class()));
+  cvWriteFrame(CVVIDEOWRITER(self), IPLIMAGE(frame));
+  return self;
+}
+
+/*
+ * Close vidoe writer.
+ */
+VALUE
+rb_close(VALUE self)
+{
+  CvVideoWriter *writer = CVVIDEOWRITER(self);
+  cvReleaseVideoWriter(&writer);
+  return Qnil;
+}
+
+
+__NAMESPACE_END_CVVIDEOWRITER
+__NAMESPACE_END_OPENCV

data/ext/cvvideowriter.h

@@ -0,0 +1,41 @@
+/************************************************************
+
+   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 *ptr;
+  Data_Get_Struct(object, CvVideoWriter, ptr);
+  return ptr;
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVVIDEOWRITER_H

data/ext/extconf.rb

@@ -0,0 +1,61 @@
+#!/usr/env ruby
+#/usr/local/bin/ruby
+=begin
+create Makefile script for Ruby/OpenCV
+
+usage : ruby extconf.rb
+        make && make install
+
+VC : ruby extconf.rb
+     nmake
+=end
+require "mkmf"
+
+# option "opencv"
+# extconf.rb --with-opencv-lib=/path/to/opencv/lib
+# extconf.rb --with-opencv-include=/path/to/opencv/include
+
+dir_config("opencv", "/usr/local/include/opencv", "/usr/local/lib")
+dir_config("ffcall", "/usr/local/include", "/usr/local/lib")
+
+opencv_libraries = ["cxcore", "cv", "highgui"]
+
+puts ">> check require libraries..."
+case CONFIG["arch"]
+when /mswin32/
+  have_library("msvcrt", nil)
+  opencv_libraries.each{|lib|
+    have_library(lib)
+  }
+else
+  opencv_libraries.each{|lib|
+    raise "lib#{lib} not found." unless have_library(lib)
+  }
+  #have_library("ml")
+  have_library("stdc++")
+end
+
+# check require headers
+puts ">> check require headers..."
+opencv_libraries.map{|lib| "#{lib}.h"}.each{|header|
+  raise "#{header} not found." unless have_header(header)
+}
+#have_header("ml.h")
+have_header("stdarg.h")
+
+# check require functions.
+# todo
+
+# optional libraies check.
+puts ">> ----- optional -----"
+puts ">> check ffcall..."
+# check ffcall
+if have_library("callback") && have_header("callback.h")
+  puts ">> support OpenCV::GUI"
+else
+  puts ">> ! unsupport OpenCV::GUI (if need it. install ffcall)"
+  puts "http://www.haible.de/bruno/packages-ffcall.html"
+end
+
+# step-final. create Makefile
+create_makefile("opencv")

data/ext/gui.cpp

@@ -0,0 +1,65 @@
+/************************************************************
+
+  gui.cpp -
+
+  $Author: lsxi $
+
+  Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifdef HAVE_CALLBACK_H
+
+#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;
+  return ((keycode = cvWaitKey(IF_INT(delay, 0))) < 0) ? Qnil : INT2FIX(keycode);
+}
+
+__NAMESPACE_END_GUI
+__NAMESPACE_END_OPENCV
+
+#endif // HAVE_CALLBACK_H

data/ext/gui.h

@@ -0,0 +1,33 @@
+/************************************************************
+
+   gui.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifdef HAVE_CALLBACK_H
+
+#define __NAMESPACE_BEGIN_GUI namespace mGUI{
+#define __NAMESPACE_END_GUI }
+
+#ifndef RUBY_OPENCV_GUI_H
+#define RUBY_OPENCV_GUI_H
+
+#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
+
+#endif // HAVE_CALLBACK_H

data/ext/iplconvkernel.cpp

@@ -0,0 +1,177 @@
+/************************************************************
+
+   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 OPENCV_OBJECT(klass, 0);
+}
+
+/*
+ * call-seq:
+ *   IplConvKernel.new(shape, rows, cols, anchor_x, anchor_y)
+ *
+ * Creates structuring element.
+ *  shape
+ *    Shape of the structuring element; may have the following values:
+ *     :rect
+ *     :cross
+ *     :ellipse
+ *  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.
+ *
+ */
+VALUE
+rb_initialize(int argc, VALUE *argv, VALUE self)
+{
+  VALUE shape, rows, cols, anchor_x, anchor_y;
+  rb_scan_args(argc, argv, "50", &shape, &rows, &cols, &anchor_x, &anchor_y);
+  Check_Type(shape, T_SYMBOL);
+  const char *shape_name = rb_id2name(SYM2ID(shape));
+  int shape_type = 0;
+  if (!strcmp(shape_name, "rect"))
+    shape_type = CV_SHAPE_RECT;
+  else if (!strcmp(shape_name, "corss"))
+    shape_type = CV_SHAPE_CROSS;
+  else if (!strcmp(shape_name, "ellipse"))
+    shape_type = CV_SHAPE_ELLIPSE;
+  else
+    rb_raise(rb_eTypeError, "argument 1 (shape) should be :rect or :cross or :ellipse.");
+  DATA_PTR(self) = cvCreateStructuringElementEx(NUM2INT(cols), NUM2INT(rows), NUM2INT(anchor_x), NUM2INT(anchor_y), shape_type);
+  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