ruby-opencv 0.0.9-x86-mswin32

data/ext/opencv/cvbox2d.h

@@ -0,0 +1,61 @@
+/************************************************************
+
+   cvbox2d.h -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#ifndef RUBY_OPENCV_CVBOX2D_H
+#define RUBY_OPENCV_CVBOX2D_H
+
+#include "opencv.h"
+
+#define __NAMESPACE_BEGIN_CVBOX2D namespace cCvBox2D {
+#define __NAMESPACE_END_CVBOX2D }
+
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVBOX2D
+
+VALUE rb_class();
+
+void define_ruby_class();
+
+VALUE rb_allocate(VALUE klass);
+
+VALUE rb_initialize(int argc, VALUE *argv, VALUE self);
+VALUE rb_center(VALUE self);
+VALUE rb_set_center(VALUE self, VALUE value);
+VALUE rb_size(VALUE self);
+VALUE rb_set_size(VALUE self, VALUE value);
+VALUE rb_angle(VALUE self);
+VALUE rb_set_angle(VALUE self, VALUE value);
+VALUE rb_points(VALUE self);
+
+VALUE new_object();
+VALUE new_object(CvBox2D box);
+
+__NAMESPACE_END_CVBOX2D
+
+inline CvBox2D*
+CVBOX2D(VALUE object){
+  CvBox2D *ptr;
+  Data_Get_Struct(object, CvBox2D, ptr);
+  return ptr;
+}
+
+inline CvBox2D
+VALUE_TO_CVBOX2D(VALUE object){
+  if (rb_obj_is_kind_of(object, cCvBox2D::rb_class())) {
+    return *CVBOX2D(object);
+  }
+  else {
+    raise_typeerror(object, cCvBox2D::rb_class());
+  }
+  throw "Should never reach here";
+}
+
+__NAMESPACE_END_OPENCV
+
+#endif // RUBY_OPENCV_CVBOX2D_H

data/ext/opencv/cvcapture.cpp

@@ -0,0 +1,506 @@
+/************************************************************
+
+   cvcapture.cpp -
+
+   $Author: lsxi $
+
+   Copyright (C) 2005-2006 Masakazu Yonekura
+
+************************************************************/
+#include "cvcapture.h"
+/*
+ * Document-class: OpenCV::CvCapture
+ *
+ * Capture image from video stream.
+ *
+ */
+__NAMESPACE_BEGIN_OPENCV
+__NAMESPACE_BEGIN_CVCAPTURE
+
+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, "CvCapture", rb_cData);
+  
+  VALUE video_interface = rb_hash_new();
+  /* {:any, :mil, :vfw, :v4l, :v4l2, :fireware, :ieee1394, :dc1394, :cmu1394, :stereo,
+     :tyzx, :tyzx_left, :tyzx_right, :tyzx_color, :tyzx_z, :qt, :qtuicktime}: video source */
+  rb_define_const(rb_klass, "INTERFACE", video_interface);
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("any")), INT2FIX(CV_CAP_ANY));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("mil")), INT2FIX(CV_CAP_MIL));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("vfw")), INT2FIX(CV_CAP_VFW));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("v4l")), INT2FIX(CV_CAP_V4L));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("v4l2")), INT2FIX(CV_CAP_V4L2));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("fireware")), INT2FIX(CV_CAP_FIREWARE));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("ieee1394")), INT2FIX(CV_CAP_IEEE1394));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("dc1394")), INT2FIX(CV_CAP_DC1394));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("cmu1394")), INT2FIX(CV_CAP_CMU1394));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("stereo")), INT2FIX(CV_CAP_STEREO));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("tyzx")), INT2FIX(CV_CAP_TYZX));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("tyzx_left")), INT2FIX(CV_TYZX_LEFT));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("tyzx_right")), INT2FIX(CV_TYZX_RIGHT));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("tyzx_color")), INT2FIX(CV_TYZX_COLOR));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("tyzx_z")), INT2FIX(CV_TYZX_Z));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("qt")), INT2FIX(CV_CAP_QT));
+  rb_hash_aset(video_interface, ID2SYM(rb_intern("quicktime")), INT2FIX(CV_CAP_QT));
+  
+  rb_define_singleton_method(rb_klass, "open", RUBY_METHOD_FUNC(rb_open), -1);
+  
+  rb_define_method(rb_klass, "grab", RUBY_METHOD_FUNC(rb_grab), 0);
+  rb_define_method(rb_klass, "retrieve", RUBY_METHOD_FUNC(rb_retrieve), 0);
+  rb_define_method(rb_klass, "query", RUBY_METHOD_FUNC(rb_query), 0);
+  rb_define_method(rb_klass, "millisecond", RUBY_METHOD_FUNC(rb_get_millisecond), 0);
+  rb_define_method(rb_klass, "millisecond=", RUBY_METHOD_FUNC(rb_set_millisecond), 1);
+  rb_define_method(rb_klass, "frames", RUBY_METHOD_FUNC(rb_get_frames), 0);
+  rb_define_method(rb_klass, "frames=", RUBY_METHOD_FUNC(rb_set_frames), 1);
+  rb_define_method(rb_klass, "avi_ratio", RUBY_METHOD_FUNC(rb_get_avi_ratio), 0);
+  rb_define_method(rb_klass, "avi_ratio=", RUBY_METHOD_FUNC(rb_set_avi_ratio), 1);
+  rb_define_method(rb_klass, "size", RUBY_METHOD_FUNC(rb_get_size), 0);
+  rb_define_method(rb_klass, "size=", RUBY_METHOD_FUNC(rb_set_size), 1);
+  rb_define_method(rb_klass, "width", RUBY_METHOD_FUNC(rb_get_width), 0);
+  rb_define_method(rb_klass, "width=", RUBY_METHOD_FUNC(rb_set_width), 1);
+  rb_define_method(rb_klass, "height", RUBY_METHOD_FUNC(rb_get_height), 0);
+  rb_define_method(rb_klass, "height=", RUBY_METHOD_FUNC(rb_set_height), 1);
+  rb_define_method(rb_klass, "fps", RUBY_METHOD_FUNC(rb_get_fps), 0);
+  rb_define_method(rb_klass, "fps=", RUBY_METHOD_FUNC(rb_set_fps), 1);
+  rb_define_method(rb_klass, "fourcc", RUBY_METHOD_FUNC(rb_get_fourcc), 0);
+  rb_define_method(rb_klass, "frame_count", RUBY_METHOD_FUNC(rb_get_frame_count), 0);
+  rb_define_method(rb_klass, "format", RUBY_METHOD_FUNC(rb_get_format), 0);
+  rb_define_method(rb_klass, "mode", RUBY_METHOD_FUNC(rb_get_mode), 0);
+  rb_define_method(rb_klass, "brightness", RUBY_METHOD_FUNC(rb_get_brightness), 0);
+  rb_define_method(rb_klass, "contrast", RUBY_METHOD_FUNC(rb_get_contrast), 0);
+  rb_define_method(rb_klass, "saturation", RUBY_METHOD_FUNC(rb_get_saturation), 0);
+  rb_define_method(rb_klass, "hue", RUBY_METHOD_FUNC(rb_get_hue), 0);
+  rb_define_method(rb_klass, "gain", RUBY_METHOD_FUNC(rb_get_gain), 0);
+  rb_define_method(rb_klass, "exposure", RUBY_METHOD_FUNC(rb_get_exposure), 0);
+  rb_define_method(rb_klass, "convert_rgb", RUBY_METHOD_FUNC(rb_get_convert_rgb), 0);
+  rb_define_method(rb_klass, "rectification", RUBY_METHOD_FUNC(rb_get_rectification), 0);
+}
+
+void
+cvcapture_free(void *ptr)
+{ 
+  if (ptr)
+    cvReleaseCapture((CvCapture**)&ptr);
+}
+
+/*
+ * call-seq:
+ *   CvCapture.open(<i>[dev = -1]</i>)
+ *
+ * Reading video stream from the specified file or camera device.
+ * If <i>dev</i> is string (i.e "stream.avi"), reading video stream from file.
+ * If <i>dev</i> is number or symbol(include CvCapture::INTERFACE),
+ * reading video stream from camera. 
+ * Currently two camera interfaces can be used on Windows:
+ * * Video for Windows(VFW)
+ * * Matrox Imaging Library(MIL)
+ * and two on Linux
+ * * V4L
+ * * FireWire(IEEE1394).
+ * If there is only one camera or it does not matter what camera to use <i>nil</i> may be passed.
+ */
+VALUE
+rb_open(int argc, VALUE *argv, VALUE self)
+{
+  VALUE device;
+  rb_scan_args(argc, argv, "01", &device);
+  CvCapture *capture = 0;
+  try {
+    switch (TYPE(device)) {
+    case T_STRING:
+      capture = cvCaptureFromFile(StringValueCStr(device));
+      break;
+    case T_FIXNUM:
+      capture = cvCaptureFromCAM(FIX2INT(device));
+      break;
+    case T_SYMBOL: {
+      VALUE cap_index = rb_hash_lookup(rb_const_get(rb_class(), rb_intern("INTERFACE")), device);
+      if (NIL_P(cap_index))
+        rb_raise(rb_eArgError, "undefined interface.");
+      capture = cvCaptureFromCAM(NUM2INT(cap_index));
+      break;
+    }
+    case T_NIL:
+      capture = cvCaptureFromCAM(CV_CAP_ANY);
+      break;
+    }
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  if (!capture)
+    rb_raise(rb_eStandardError, "Invalid capture format.");
+  return Data_Wrap_Struct(rb_klass, 0, cvcapture_free, capture);
+}
+
+
+/*
+ * call-seq:
+ *   grab -> true or false
+ *
+ * Grabbed frame is stored internally. To grab frame
+ * <i>fast</i> that is important for syncronization in case of reading from
+ * several cameras simultaneously. The grabbed frames are not exposed because
+ * they may be stored in compressed format (as defined by camera/driver).
+ * To retrieve the grabbed frame, retrieve should be used.
+ *
+ * If grabbed frame was success, return true. Otherwise return false.
+ */
+VALUE
+rb_grab(VALUE self)
+{
+  int grab = 0;
+  try {
+    grab = cvGrabFrame(CVCAPTURE(self));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return grab ? Qtrue : Qfalse;
+}
+
+/*
+ * call-seq:
+ *   retrieve -> IplImage or nil
+ *
+ * Gets the image grabbed with grab.
+ */
+VALUE
+rb_retrieve(VALUE self)
+{
+  VALUE image = Qnil;
+  IplImage *frame = NULL;
+  try {
+    if (!(frame = cvRetrieveFrame(CVCAPTURE(self)))) {
+      return Qnil;
+    }
+    image = cIplImage::new_object(frame->width, frame->height,
+				  CV_MAKETYPE(IPL2CV_DEPTH(frame->depth), frame->nChannels));
+    if (frame->origin == IPL_ORIGIN_TL) {
+      cvCopy(frame, CVARR(image));
+    }
+    else {
+      cvFlip(frame, CVARR(image));
+    }
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return image;
+
+}
+
+/*
+ * call-seq:
+ *   query -> IplImage or nil
+ *
+ * Grabs and returns a frame camera or file. Just a combination of grab and retrieve in one call.
+ */
+VALUE
+rb_query(VALUE self)
+{
+  VALUE image = Qnil;
+  IplImage *frame = NULL;
+  try {
+    if (!(frame = cvQueryFrame(CVCAPTURE(self)))) {
+      return Qnil;
+    }
+    image = cIplImage::new_object(frame->width, frame->height,
+				  CV_MAKETYPE(IPL2CV_DEPTH(frame->depth), frame->nChannels));
+    if (frame->origin == IPL_ORIGIN_TL) {
+      cvCopy(frame, CVARR(image));
+    }
+    else {
+      cvFlip(frame, CVARR(image));
+    }
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return image;
+}
+
+VALUE
+rb_get_capture_property(VALUE self, int id)
+{
+  double result = 0;
+  try {
+    result = cvGetCaptureProperty(CVCAPTURE(self), id);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return rb_float_new(result);
+}
+
+VALUE
+rb_set_capture_property(VALUE self, int id, VALUE value)
+{
+  double result = 0;
+  try {
+    result = cvSetCaptureProperty(CVCAPTURE(self), id, NUM2DBL(value));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return rb_float_new(result);
+}
+
+/*
+ * Get film current position in milliseconds or video capture timestamp.
+ */
+VALUE
+rb_get_millisecond(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_POS_MSEC);
+}
+/*
+ * Set film current position in milliseconds or video capture timestamp.
+ */
+VALUE
+rb_set_millisecond(VALUE self, VALUE value)
+{
+  return rb_set_capture_property(self, CV_CAP_PROP_POS_MSEC, value);
+}
+/*
+ * Get 0-based index of the frame to be decoded/captured next
+ */
+VALUE
+rb_get_frames(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_POS_FRAMES);
+}
+/*
+ * Set 0-based index of the frame to be decoded/captured next
+ */
+VALUE
+rb_set_frames(VALUE self, VALUE value)
+{
+  return rb_set_capture_property(self, CV_CAP_PROP_POS_FRAMES, value);
+}
+/*
+ * Get relative position of video file (0 - start of the film, 1 - end of the film)
+ */
+VALUE
+rb_get_avi_ratio(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_POS_AVI_RATIO);
+}
+/*
+ * Set relative position of video file (0 - start of the film, 1 - end of the film)
+ */
+VALUE
+rb_set_avi_ratio(VALUE self, VALUE value)
+{
+  return rb_set_capture_property(self, CV_CAP_PROP_POS_AVI_RATIO, value);
+}
+/*
+ * Get size of frames in the video stream.
+ */
+VALUE
+rb_get_size(VALUE self)
+{
+  CvSize size;
+  try {
+    CvCapture* self_ptr = CVCAPTURE(self);
+    size = cvSize((int)cvGetCaptureProperty(self_ptr, CV_CAP_PROP_FRAME_WIDTH),
+		  (int)cvGetCaptureProperty(self_ptr, CV_CAP_PROP_FRAME_HEIGHT));
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return cCvSize::new_object(size);
+}
+/*
+ * Set size of frames in the video stream.
+ */
+VALUE
+rb_set_size(VALUE self, VALUE value)
+{
+  double result = 0;
+  CvSize size = VALUE_TO_CVSIZE(value);
+  try {
+    CvCapture* self_ptr = CVCAPTURE(self);
+    cvSetCaptureProperty(self_ptr, CV_CAP_PROP_FRAME_WIDTH, size.width);
+    result = cvSetCaptureProperty(self_ptr, CV_CAP_PROP_FRAME_HEIGHT, size.height);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return DBL2NUM(result);
+}
+/*
+ * Get width of frames in the video stream.
+ */
+VALUE
+rb_get_width(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_FRAME_WIDTH);
+}
+/*
+ * Set width of frames in the video stream.
+ */
+VALUE
+rb_set_width(VALUE self, VALUE value)
+{
+  return rb_set_capture_property(self, CV_CAP_PROP_FRAME_WIDTH, value);
+}
+/*
+ * Get height of frames in the video stream.
+ */
+VALUE
+rb_get_height(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_FRAME_HEIGHT);
+}
+/*
+ * Set height of frames in the video stream.
+ */
+VALUE
+rb_set_height(VALUE self, VALUE value)
+{
+  return rb_set_capture_property(self, CV_CAP_PROP_FRAME_HEIGHT, value);
+}
+/*
+ * Get frame rate
+ */
+VALUE
+rb_get_fps(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_FPS);
+}
+/*
+ * Set frame rate
+ */
+VALUE
+rb_set_fps(VALUE self, VALUE value)
+{
+  return rb_set_capture_property(self, CV_CAP_PROP_FPS, value);
+}
+/*
+ * Get 4character code of codec. see http://www.fourcc.org/
+ */
+VALUE
+rb_get_fourcc(VALUE self)
+{
+  char str[4];
+  double fourcc = cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FOURCC);
+  sprintf(str, "%s", (char*)&fourcc);
+  return rb_str_new2(str);
+}
+/*
+ * Get number of frames in video file.
+ */
+VALUE
+rb_get_frame_count(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_FRAME_COUNT);
+}
+/*
+ * Get the format of the Mat objects returned by CvCapture#retrieve
+ */
+VALUE
+rb_get_format(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_FORMAT);
+}
+/*
+ * Get a backend-specific value indicating the current capture mode
+ */
+VALUE
+rb_get_mode(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_MODE);
+}
+/*
+ * Get brightness of the image (only for cameras)
+ */
+VALUE
+rb_get_brightness(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_BRIGHTNESS);
+}
+/*
+ * Get contrast of the image (only for cameras)
+ */
+VALUE
+rb_get_contrast(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_CONTRAST);
+}
+/*
+ * Get saturation of the image (only for cameras)
+ */
+VALUE
+rb_get_saturation(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_SATURATION);
+}
+/*
+ * Get hue of the image (only for cameras)
+ */
+VALUE
+rb_get_hue(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_HUE);
+}
+/*
+ * Get gain of the image (only for cameras)
+ */
+VALUE
+rb_get_gain(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_GAIN);
+}
+/*
+ * Get exposure (only for cameras)
+ */
+VALUE
+rb_get_exposure(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_EXPOSURE);
+}
+/*
+ * Get boolean flags indicating whether images should be converted to RGB
+ */
+VALUE
+rb_get_convert_rgb(VALUE self)
+{
+  int flag = 0;
+  try {
+    flag = (int)cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_CONVERT_RGB);
+  }
+  catch (cv::Exception& e) {
+    raise_cverror(e);
+  }
+  return flag ? Qtrue : Qfalse;
+}
+/*
+ * Get TOWRITE (note: only supported by DC1394 v 2.x backend currently)
+ */
+VALUE
+rb_get_rectification(VALUE self)
+{
+  return rb_get_capture_property(self, CV_CAP_PROP_RECTIFICATION);
+}
+__NAMESPACE_END_CVCAPTURE
+__NAMESPACE_END_OPENCV
+