mirror of
https://github.com/ruby-opencv/ruby-opencv
synced 2023-03-27 23:22:12 -04:00
466 lines
13 KiB
C++
466 lines
13 KiB
C++
/************************************************************
|
|
|
|
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, "white_balance", RUBY_METHOD_FUNC(rb_get_white_balance), 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, i;
|
|
rb_scan_args(argc, argv, "01", &device);
|
|
CvCapture *capture = 0;
|
|
switch (TYPE(device)) {
|
|
case T_STRING:
|
|
capture = cvCaptureFromFile(StringValueCStr(device));
|
|
break;
|
|
case T_FIXNUM:
|
|
capture = cvCaptureFromCAM(FIX2INT(device));
|
|
break;
|
|
case T_SYMBOL:
|
|
i = rb_hash_aref(rb_const_get(rb_class(), rb_intern("INTERFACE")), device);
|
|
if (NIL_P(i))
|
|
rb_raise(rb_eArgError, "undefined interface.");
|
|
capture = cvCaptureFromCAM(NUM2INT(i));
|
|
break;
|
|
case T_NIL:
|
|
capture = cvCaptureFromCAM(CV_CAP_ANY);
|
|
break;
|
|
}
|
|
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)
|
|
{
|
|
return cvGrabFrame(CVCAPTURE(self)) ? Qtrue : Qfalse;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* retrieve -> IplImage or nil
|
|
*
|
|
* Gets the image grabbed with grab.
|
|
*/
|
|
VALUE
|
|
rb_retrieve(VALUE self)
|
|
{
|
|
IplImage *frame = cvRetrieveFrame(CVCAPTURE(self));
|
|
if (!frame)
|
|
return Qnil;
|
|
VALUE image = cIplImage::new_object(cvSize(frame->width, frame->height), CV_MAKETYPE(CV_8U, frame->nChannels));
|
|
if (frame->origin == IPL_ORIGIN_TL)
|
|
cvCopy(frame, CVARR(image));
|
|
else
|
|
cvFlip(frame, CVARR(image));
|
|
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)
|
|
{
|
|
IplImage *frame = cvQueryFrame(CVCAPTURE(self));
|
|
if (!frame)
|
|
return Qnil;
|
|
VALUE image = cIplImage::new_object(cvSize(frame->width, frame->height), CV_MAKETYPE(CV_8U, frame->nChannels));
|
|
if (frame->origin == IPL_ORIGIN_TL)
|
|
cvCopy(frame, CVARR(image));
|
|
else
|
|
cvFlip(frame, CVARR(image));
|
|
return image;
|
|
}
|
|
|
|
/*
|
|
* Get film current position in milliseconds or video capture timestamp.
|
|
*/
|
|
VALUE
|
|
rb_get_millisecond(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(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_float_new(cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_POS_MSEC, NUM2DBL(value)));
|
|
}
|
|
|
|
/*
|
|
* Get 0-based index of the frame to be decoded/captured next
|
|
*/
|
|
VALUE
|
|
rb_get_frames(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(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_float_new(cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_POS_FRAMES, NUM2DBL(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_float_new(cvGetCaptureProperty(CVCAPTURE(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_float_new(cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_POS_AVI_RATIO, NUM2DBL(value)));
|
|
}
|
|
|
|
/*
|
|
* Get size of frames in the video stream.
|
|
*/
|
|
VALUE
|
|
rb_get_size(VALUE self)
|
|
{
|
|
return cCvSize::new_object(cvSize((int)cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_WIDTH),
|
|
(int)cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_HEIGHT)));
|
|
}
|
|
|
|
/*
|
|
* Set size of frames in the video stream.
|
|
*/
|
|
VALUE
|
|
rb_set_size(VALUE self, VALUE value)
|
|
{
|
|
double result;
|
|
if (cCvSize::rb_compatible_q(cCvSize::rb_class(), value)) {
|
|
int width = NUM2INT(rb_funcall(value, rb_intern("width"), 0));
|
|
int height = NUM2INT(rb_funcall(value, rb_intern("height"), 0));
|
|
cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_WIDTH, width);
|
|
result = cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_HEIGHT, height);
|
|
}
|
|
else
|
|
rb_raise(rb_eTypeError, "require %s or compatible object.", rb_class2name(cCvSize::rb_class()));
|
|
return DBL2NUM(result);
|
|
}
|
|
|
|
/*
|
|
* Get width of frames in the video stream.
|
|
*/
|
|
VALUE
|
|
rb_get_width(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_WIDTH));
|
|
}
|
|
|
|
/*
|
|
* Set width of frames in the video stream.
|
|
*/
|
|
VALUE
|
|
rb_set_width(VALUE self, VALUE value)
|
|
{
|
|
return rb_float_new(cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_WIDTH, NUM2DBL(value)));
|
|
}
|
|
|
|
/*
|
|
* Get height of frames in the video stream.
|
|
*/
|
|
VALUE
|
|
rb_get_height(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_HEIGHT));
|
|
}
|
|
|
|
/*
|
|
* Set height of frames in the video stream.
|
|
*/
|
|
VALUE
|
|
rb_set_height(VALUE self, VALUE value)
|
|
{
|
|
return rb_float_new(cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FRAME_HEIGHT, NUM2DBL(value)));
|
|
}
|
|
|
|
/*
|
|
* Get frame rate
|
|
*/
|
|
VALUE
|
|
rb_get_fps(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FPS));
|
|
}
|
|
|
|
/*
|
|
* Set frame rate
|
|
*/
|
|
VALUE
|
|
rb_set_fps(VALUE self, VALUE value)
|
|
{
|
|
return rb_float_new(cvSetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FPS, NUM2DBL(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_float_new(cvGetCaptureProperty(CVCAPTURE(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_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_FORMAT));
|
|
}
|
|
|
|
/*
|
|
* Get a backend-specific value indicating the current capture mode
|
|
*/
|
|
VALUE
|
|
rb_get_mode(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_MODE));
|
|
}
|
|
|
|
/*
|
|
* Get brightness of the image (only for cameras)
|
|
*/
|
|
VALUE
|
|
rb_get_brightness(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_BRIGHTNESS));
|
|
}
|
|
|
|
/*
|
|
* Get contrast of the image (only for cameras)
|
|
*/
|
|
VALUE
|
|
rb_get_contrast(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_CONTRAST));
|
|
}
|
|
|
|
/*
|
|
* Get saturation of the image (only for cameras)
|
|
*/
|
|
VALUE
|
|
rb_get_saturation(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_SATURATION));
|
|
}
|
|
|
|
/*
|
|
* Get hue of the image (only for cameras)
|
|
*/
|
|
VALUE
|
|
rb_get_hue(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_HUE));
|
|
}
|
|
|
|
/*
|
|
* Get gain of the image (only for cameras)
|
|
*/
|
|
VALUE
|
|
rb_get_gain(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_GAIN));
|
|
}
|
|
|
|
/*
|
|
* Get exposure (only for cameras)
|
|
*/
|
|
VALUE
|
|
rb_get_exposure(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(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 = (int)cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_CONVERT_RGB);
|
|
return (flag == 1) ? Qtrue : Qfalse;
|
|
}
|
|
|
|
/*
|
|
* Currently unsupported
|
|
*/
|
|
VALUE
|
|
rb_get_white_balance(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_WHITE_BALANCE));
|
|
}
|
|
|
|
/*
|
|
* Get TOWRITE (note: only supported by DC1394 v 2.x backend currently)
|
|
*/
|
|
VALUE
|
|
rb_get_rectification(VALUE self)
|
|
{
|
|
return rb_float_new(cvGetCaptureProperty(CVCAPTURE(self), CV_CAP_PROP_RECTIFICATION));
|
|
}
|
|
|
|
__NAMESPACE_END_CVCAPTURE
|
|
__NAMESPACE_END_OPENCV
|