mirror of
https://github.com/ruby-opencv/ruby-opencv
synced 2023-03-27 23:22:12 -04:00
Added gemspec to be able to bundle install from git.
Added IplImage#smoothness that returns :smooth, :messy, or :blank. Rake compile will now compile the C extension code.
This commit is contained in:
Patrick Ting
parent
9f801045b8
commit
08a9655dc1
127 changed files with 826 additions and 540 deletions
243
ext/opencv/cvmoments.cpp
Normal file
243
ext/opencv/cvmoments.cpp
Normal file
|
|
@ -0,0 +1,243 @@
|
|||
/************************************************************
|
||||
|
||||
cvmoments.cpp -
|
||||
|
||||
$Author: lsxi $
|
||||
|
||||
Copyright (C) 2007 Masakazu Yonekura
|
||||
|
||||
************************************************************/
|
||||
#include "cvmoments.h"
|
||||
/*
|
||||
* Document-class: OpenCV::CvMoments
|
||||
*
|
||||
* moments
|
||||
*/
|
||||
__NAMESPACE_BEGIN_OPENCV
|
||||
__NAMESPACE_BEGIN_CVMOMENTS
|
||||
|
||||
#define DEFINE_CVMOMENTS_ACCESSOR(elem) \
|
||||
rb_define_method(rb_klass, #elem, RUBY_METHOD_FUNC(rb_##elem), 0)
|
||||
|
||||
#define CVMOMENTS_ACCESSOR(elem) \
|
||||
VALUE rb_##elem(VALUE self) { return DBL2NUM(CVMOMENTS(self)->elem); }
|
||||
|
||||
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, "CvMoments", 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, "spatial", RUBY_METHOD_FUNC(rb_spatial), 2);
|
||||
rb_define_method(rb_klass, "central", RUBY_METHOD_FUNC(rb_central), 2);
|
||||
rb_define_method(rb_klass, "normalized_central", RUBY_METHOD_FUNC(rb_normalized_central), 2);
|
||||
rb_define_method(rb_klass, "hu", RUBY_METHOD_FUNC(rb_hu), 0);
|
||||
rb_define_method(rb_klass, "gravity_center", RUBY_METHOD_FUNC(rb_gravity_center), 0);
|
||||
rb_define_method(rb_klass, "angle", RUBY_METHOD_FUNC(rb_angle), 0);
|
||||
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m00);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m10);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m01);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m20);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m11);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m02);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m30);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m21);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m12);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(m03);
|
||||
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu20);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu11);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu02);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu30);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu21);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu12);
|
||||
DEFINE_CVMOMENTS_ACCESSOR(mu03);
|
||||
|
||||
DEFINE_CVMOMENTS_ACCESSOR(inv_sqrt_m00);
|
||||
}
|
||||
|
||||
VALUE
|
||||
rb_allocate(VALUE klass)
|
||||
{
|
||||
CvMoments *ptr;
|
||||
return Data_Make_Struct(klass, CvMoments, 0, -1, ptr);
|
||||
}
|
||||
/*
|
||||
* call-seq:
|
||||
* CvMoments.new(<i>src[,is_binary = nil]</i>)
|
||||
*
|
||||
* Calculates all moments up to third order of a polygon or rasterized shape.
|
||||
* <i>src</i> should be CvMat or CvPolygon.
|
||||
*
|
||||
* If is_binary = true, all the zero pixel values are treated as zeroes, all the others are treated as 1's.
|
||||
*/
|
||||
VALUE
|
||||
rb_initialize(int argc, VALUE *argv, VALUE self)
|
||||
{
|
||||
VALUE src, is_binary;
|
||||
rb_scan_args(argc, argv, "02", &src, &is_binary);
|
||||
if (!NIL_P(src)) {
|
||||
if (rb_obj_is_kind_of(src, cCvMat::rb_class()) || rb_obj_is_kind_of(src, cCvSeq::rb_class()))
|
||||
cvMoments(CVARR(src), CVMOMENTS(self), TRUE_OR_FALSE(is_binary, 0));
|
||||
else
|
||||
rb_raise(rb_eTypeError, "argument 1 (src) should be %s or %s.",
|
||||
rb_class2name(cCvMat::rb_class()), rb_class2name(cCvSeq::rb_class()));
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
CVMOMENTS_ACCESSOR(m00);
|
||||
CVMOMENTS_ACCESSOR(m10);
|
||||
CVMOMENTS_ACCESSOR(m01);
|
||||
CVMOMENTS_ACCESSOR(m20);
|
||||
CVMOMENTS_ACCESSOR(m11);
|
||||
CVMOMENTS_ACCESSOR(m02);
|
||||
CVMOMENTS_ACCESSOR(m30);
|
||||
CVMOMENTS_ACCESSOR(m21);
|
||||
CVMOMENTS_ACCESSOR(m12);
|
||||
CVMOMENTS_ACCESSOR(m03);
|
||||
|
||||
CVMOMENTS_ACCESSOR(mu20);
|
||||
CVMOMENTS_ACCESSOR(mu11);
|
||||
CVMOMENTS_ACCESSOR(mu02);
|
||||
CVMOMENTS_ACCESSOR(mu30);
|
||||
CVMOMENTS_ACCESSOR(mu21);
|
||||
CVMOMENTS_ACCESSOR(mu12);
|
||||
CVMOMENTS_ACCESSOR(mu03);
|
||||
|
||||
CVMOMENTS_ACCESSOR(inv_sqrt_m00);
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* spatial -> float
|
||||
*
|
||||
* Retrieves spatial moment.
|
||||
*
|
||||
* which in case of image moments is defined as:
|
||||
* Mx_order,y_order=sumx,y(I(x,y)*xx_order*yy_order)
|
||||
* where I(x,y) is the intensity of the pixel (x, y).
|
||||
*/
|
||||
VALUE
|
||||
rb_spatial(VALUE self, VALUE x_order, VALUE y_order)
|
||||
{
|
||||
return rb_float_new(cvGetSpatialMoment(CVMOMENTS(self), NUM2INT(x_order), NUM2INT(y_order)));
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* central -> float
|
||||
*
|
||||
* Retrieves central moment.
|
||||
*
|
||||
* which in case of image moments is defined as:
|
||||
* μx_order,y_order=sumx,y(I(x,y)*(x-xc)x_order*(y-yc)y_order),
|
||||
* where xc=M10/M00, yc=M01/M00 - coordinates of the gravity center
|
||||
*/
|
||||
VALUE
|
||||
rb_central(VALUE self, VALUE x_order, VALUE y_order)
|
||||
{
|
||||
return rb_float_new(cvGetCentralMoment(CVMOMENTS(self), NUM2INT(x_order), NUM2INT(y_order)));
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* normalized_central -> float
|
||||
*
|
||||
* Retrieves normalized central moment.
|
||||
*
|
||||
* ηx_order,y_order= μx_order,y_order/M00((y_order+x_order)/2+1)
|
||||
*/
|
||||
VALUE
|
||||
rb_normalized_central(VALUE self, VALUE x_order, VALUE y_order)
|
||||
{
|
||||
return rb_float_new(cvGetNormalizedCentralMoment(CVMOMENTS(self), NUM2INT(x_order), NUM2INT(y_order)));
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* hu -> cvhumoments
|
||||
*
|
||||
* Calculates seven Hu invariants.
|
||||
*
|
||||
* seven Hu invariants that are defined as:
|
||||
* h1=η20+η02
|
||||
* h2=(η20-η02)²+4η11²
|
||||
* h3=(η30-3η12)²+ (3η21-η03)²
|
||||
* h4=(η30+η12)²+ (η21+η03)²
|
||||
* h5=(η30-3η12)(η30+η12)[(η30+η12)²-3(η21+η03)²]+(3η21-η03)(η21+η03)[3(η30+η12)²-(η21+η03)²]
|
||||
* h6=(η20-η02)[(η30+η12)²- (η21+η03)²]+4η11(η30+η12)(η21+η03)
|
||||
* h7=(3η21-η03)(η21+η03)[3(η30+η12)²-(η21+η03)²]-(η30-3η12)(η21+η03)[3(η30+η12)²-(η21+η03)²]
|
||||
* where ηi,j are normalized central moments of 2-nd and 3-rd orders. The computed values are proved to be invariant to the image scaling, rotation, and reflection except the seventh one, whose sign is changed by reflection.
|
||||
*/
|
||||
VALUE
|
||||
rb_hu(VALUE self)
|
||||
{
|
||||
return cCvHuMoments::new_object(CVMOMENTS(self));
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* gravity_center -> cvpoint2d32f
|
||||
*
|
||||
* Return gravity center.
|
||||
*/
|
||||
VALUE
|
||||
rb_gravity_center(VALUE self)
|
||||
{
|
||||
CvMoments *moments = CVMOMENTS(self);
|
||||
double
|
||||
m00 = cvGetSpatialMoment(moments, 0, 0),
|
||||
m10 = cvGetSpatialMoment(moments, 1, 0),
|
||||
m01 = cvGetSpatialMoment(moments, 0, 1);
|
||||
return cCvPoint2D32f::new_object(cvPoint2D32f(m10 / m00, m01 / m00));
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* angle -> float
|
||||
*
|
||||
* Return angle.
|
||||
*/
|
||||
VALUE
|
||||
rb_angle(VALUE self)
|
||||
{
|
||||
CvMoments *moments = CVMOMENTS(self);
|
||||
double
|
||||
m11 = cvGetCentralMoment(moments, 1, 1),
|
||||
m20 = cvGetCentralMoment(moments, 2, 0),
|
||||
m02 = cvGetCentralMoment(moments, 0, 2),
|
||||
mangle = 0.5 * atan(2 * m11 / (m20 - m02));
|
||||
if(cvIsNaN(mangle) || cvIsInf(mangle))
|
||||
return Qnil;
|
||||
else
|
||||
return rb_float_new(mangle);
|
||||
}
|
||||
|
||||
VALUE
|
||||
new_object(CvArr *arr, int is_binary = 0)
|
||||
{
|
||||
VALUE object = rb_allocate(rb_class());
|
||||
cvMoments(arr, CVMOMENTS(object), is_binary);
|
||||
return object;
|
||||
}
|
||||
|
||||
__NAMESPACE_END_CVMOMENTS
|
||||
__NAMESPACE_END_OPENCV
|
||||
Loading…
Add table
Add a link
Reference in a new issue