kotovalexarian-likes-github/ruby-opencv
1
0
Fork 0
mirror of https://github.com/ruby-opencv/ruby-opencv synced 2023-03-27 23:22:12 -04:00
Code Releases Activity

add cv::setIdentity

Browse source
This commit is contained in:
ser1zw 2016-04-30 16:49:12 +09:00
parent cee1b67433
commit 7b34f21e4f
2 changed files with 37 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
ext/opencv/mat.cpp
View file

@ -933,6 +933,29 @@ namespace rubyopencv {
return mat2obj(dstptr, CLASS_OF(self));
}
/*
* Initializes a scaled identity matrix.
*
* @overload set_identity(s = Scalar.new(1))
* @param s [Scalar] Value to assign to diagonal elements.
* @return [Mat] +self+
* @opencv_func cv::setIdentity
*/
VALUE rb_set_identity(int argc, VALUE *argv, VALUE self) {
VALUE s = Qnil;
rb_scan_args(argc, argv, "01", &s);
cv::Mat* selfptr = obj2mat(self);
cv::Scalar s_value = NIL_P(s) ? cv::Scalar(1) : *(Scalar::obj2scalar(s));
try {
cv::setIdentity(*selfptr, s_value);
}
catch (cv::Exception& e) {
Error::raise(e);
}
return self;
}
void init() {
VALUE opencv = rb_define_module("Cv");
@ -993,6 +1016,7 @@ namespace rubyopencv {
rb_define_method(rb_klass, "convert_scale_abs", RUBY_METHOD_FUNC(rb_convert_scale_abs), -1);
rb_define_method(rb_klass, "convert_to", RUBY_METHOD_FUNC(rb_convert_to), -1);
rb_define_method(rb_klass, "set_identity", RUBY_METHOD_FUNC(rb_set_identity), -1);
}
}
}

13
test/test_mat.rb
View file

@ -566,4 +566,17 @@ class TestMat < OpenCVTestCase
m.set_to(s1, DUMMY_OBJ)
}
end
def test_set_identity
m = Mat::zeros(3, 3, CV_8U)
m.set_identity(Scalar.new(10))
expected = "<Cv::Mat:3x3,depth=0,channels=1,\n[ 10, 0, 0;\n 0, 10, 0;\n 0, 0, 10]>"
assert_equal(expected, m.to_s)
m.set_identity
expected = "<Cv::Mat:3x3,depth=0,channels=1,\n[ 1, 0, 0;\n 0, 1, 0;\n 0, 0, 1]>"
assert_equal(expected, m.to_s)
assert_raise(TypeError) {
m.set_identity(DUMMY_OBJ)
}
end
end