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
230
ext/opencv/cvcondensation.cpp
Normal file
230
ext/opencv/cvcondensation.cpp
Normal file
|
|
@ -0,0 +1,230 @@
|
|||
/************************************************************
|
||||
|
||||
cvcondensation.cpp -
|
||||
|
||||
$Author: lsxi $
|
||||
|
||||
Copyright (C) 2005-2006 Masakazu Yonekura
|
||||
|
||||
************************************************************/
|
||||
#include "cvcondensation.h"
|
||||
/*
|
||||
* Document-class: OpenCV::CvConDensation
|
||||
*
|
||||
*/
|
||||
__NAMESPACE_BEGIN_OPENCV
|
||||
__NAMESPACE_BEGIN_CVCONDENSATION
|
||||
|
||||
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, "CvConDensation", rb_cObject);
|
||||
rb_define_method(rb_klass, "dp", RUBY_METHOD_FUNC(rb_dp), 0);
|
||||
rb_define_method(rb_klass, "mp", RUBY_METHOD_FUNC(rb_mp), 0);
|
||||
rb_define_method(rb_klass, "dynamic_matrix", RUBY_METHOD_FUNC(rb_dynamic_matrix), 0);
|
||||
rb_define_method(rb_klass, "confidence", RUBY_METHOD_FUNC(rb_confidence), 0);
|
||||
rb_define_method(rb_klass, "cumulative", RUBY_METHOD_FUNC(rb_cumulative), 0);
|
||||
rb_define_method(rb_klass, "state", RUBY_METHOD_FUNC(rb_state), 0);
|
||||
rb_define_method(rb_klass, "samples_num", RUBY_METHOD_FUNC(rb_samples_num), 0);
|
||||
rb_define_method(rb_klass, "init_sample_set", RUBY_METHOD_FUNC(rb_init_sample_set), 2);
|
||||
rb_define_method(rb_klass, "update_by_time", RUBY_METHOD_FUNC(rb_update_by_time), 0);
|
||||
rb_define_alias(rb_klass, "update", "update_by_time");
|
||||
rb_define_method(rb_klass, "each_sample", RUBY_METHOD_FUNC(rb_each_sample), 0);
|
||||
rb_define_method(rb_klass, "calculate_confidence", RUBY_METHOD_FUNC(rb_calculate_confidence), 0);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* dp -> int
|
||||
*
|
||||
* Return dimension of state vector
|
||||
*/
|
||||
VALUE
|
||||
rb_dp(VALUE self)
|
||||
{
|
||||
return INT2FIX(CVCONDENSATION(self)->DP);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* mp -> int
|
||||
*
|
||||
* Return demension of measurement vector.
|
||||
*/
|
||||
VALUE
|
||||
rb_mp(VALUE self)
|
||||
{
|
||||
return INT2FIX(CVCONDENSATION(self)->MP);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* dynamic_matrix -> mat
|
||||
*
|
||||
* Return matrix of the linear Dynamics system.
|
||||
*/
|
||||
VALUE
|
||||
rb_dynamic_matrix(VALUE self)
|
||||
{
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
return DEPEND_OBJECT(cCvMat::rb_class(), cvInitMatHeader(ALLOC(CvMat), cd->DP, cd->DP, CV_MAKETYPE(CV_32F, 1), cd->DynamMatr), self);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* confidence -> mat
|
||||
*
|
||||
* Return confidence for each sample.
|
||||
*/
|
||||
VALUE
|
||||
rb_confidence(VALUE self)
|
||||
{
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
return DEPEND_OBJECT(cCvMat::rb_class(), cvInitMatHeader(ALLOC(CvMat), cd->SamplesNum, 1, CV_MAKETYPE(CV_32F, 1), cd->flConfidence), self);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* cumulative -> mat
|
||||
*
|
||||
* Return cumulative confidence.
|
||||
*/
|
||||
VALUE
|
||||
rb_cumulative(VALUE self)
|
||||
{
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
return DEPEND_OBJECT(cCvMat::rb_class(), cvInitMatHeader(ALLOC(CvMat), cd->SamplesNum, 1, CV_MAKETYPE(CV_32F, 1), cd->flCumulative), self);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* state -> mat
|
||||
*
|
||||
* Return vector of state
|
||||
*/
|
||||
VALUE
|
||||
rb_state(VALUE self)
|
||||
{
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
return DEPEND_OBJECT(cCvMat::rb_class(), cvInitMatHeader(ALLOC(CvMat), cd->DP, 1, CV_MAKETYPE(CV_32F, 1), cd->State), self);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* samples_num -> int
|
||||
*
|
||||
* Return number of the samples
|
||||
*/
|
||||
VALUE
|
||||
rb_samples_num(VALUE self)
|
||||
{
|
||||
return INT2FIX(CVCONDENSATION(self)->SamplesNum);
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* init_sample_set(upper, lower)
|
||||
*
|
||||
* Initializes sample set for ConDensation algorithm.
|
||||
* Fills the samples with values within specified(lower to upper) ranges.
|
||||
*/
|
||||
VALUE
|
||||
rb_init_sample_set(VALUE self, VALUE lower, VALUE upper)
|
||||
{
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
CvMat *lower_bound = CVMAT(lower), *upper_bound = CVMAT(upper), lb_stub, ub_stub;
|
||||
int lower_type = lower_bound->type, upper_type = lower_bound->type;
|
||||
if (lower_type != CV_32FC1 || lower_bound->cols != 1) {
|
||||
if (CV_MAT_DEPTH(lower_type) == CV_32F) {
|
||||
lower_bound = cvReshape(lower_bound, &lb_stub, 1, lower_bound->rows * lower_bound->cols);
|
||||
} else {
|
||||
lower = cCvMat::new_object(cvSize(lower_bound->rows * lower_bound->cols, 1), CV_MAKETYPE(CV_32S, 1));
|
||||
cvConvertScale(lower_bound, CVARR(lower));
|
||||
lower_bound = CVMAT(lower);
|
||||
}
|
||||
}
|
||||
if (upper_type != CV_32FC1 || upper_bound->cols != 1) {
|
||||
if (CV_MAT_DEPTH(upper_type) == CV_32F) {
|
||||
upper_bound = cvReshape(upper_bound, &ub_stub, 1, upper_bound->rows * upper_bound->cols);
|
||||
} else {
|
||||
upper = cCvMat::new_object(cvSize(upper_bound->rows * upper_bound->cols, 1), CV_MAKETYPE(CV_32F, 1));
|
||||
cvConvertScale(upper_bound, CVARR(upper));
|
||||
upper_bound = CVMAT(upper);
|
||||
}
|
||||
}
|
||||
if (lower_bound->rows != cd->DP || upper_bound->rows != cd->DP) {
|
||||
rb_raise(rb_eTypeError, "sample matrix step unmatch.");
|
||||
}
|
||||
cvConDensInitSampleSet(cd, lower_bound, upper_bound);
|
||||
return self;
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* update_by_time
|
||||
*
|
||||
* Estimates subsequent model state.
|
||||
*/
|
||||
VALUE
|
||||
rb_update_by_time(VALUE self)
|
||||
{
|
||||
cvConDensUpdateByTime(CVCONDENSATION(self));
|
||||
return self;
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* each_sample{|mat| ... }
|
||||
*
|
||||
* Evaluate each sample by given block.
|
||||
*/
|
||||
VALUE
|
||||
rb_each_sample(VALUE self)
|
||||
{
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
if (rb_block_given_p()) {
|
||||
for (int i = 0; i < cd->SamplesNum; i++) {
|
||||
rb_yield(DEPEND_OBJECT(cCvMat::rb_class(), cvInitMatHeader(ALLOC(CvMat), cd->DP, 1, CV_MAKETYPE(CV_32F, 1), cd->flSamples[i]), self));
|
||||
}
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
/*
|
||||
* call-seq:
|
||||
* calculate_confidence{|value| ... }
|
||||
*
|
||||
* Evalute each sample by given block, then return value set to confidence.
|
||||
*/
|
||||
VALUE
|
||||
rb_calculate_confidence(VALUE self)
|
||||
{
|
||||
VALUE value;
|
||||
CvConDensation *cd = CVCONDENSATION(self);
|
||||
if (rb_block_given_p()) {
|
||||
for (int i = 0; i < cd->SamplesNum; i++) {
|
||||
value = rb_yield(DEPEND_OBJECT(cCvMat::rb_class(), cvInitMatHeader(ALLOC(CvMat), cd->DP, 1, CV_MAKETYPE(CV_32F, 1), cd->flSamples[i]), self));
|
||||
cd->flConfidence[i] = NUM2DBL(value);
|
||||
}
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
__NAMESPACE_END_CVCONDENSATION
|
||||
__NAMESPACE_END_OPENCV
|
||||
Loading…
Add table
Add a link
Reference in a new issue