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

modified CvMat#initialize to run GC and retry memory allocation when it is failed

Browse source
This commit is contained in:
ser1zw 2011-04-23 22:34:47 +09:00
parent 3bba97baf1
commit 882b778e94
1 changed files with 24 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

26
ext/opencv/cvmat.cpp
View file

@ -434,8 +434,30 @@ rb_initialize(int argc, VALUE *argv, VALUE self)
VALUE row, column, depth, channel;
rb_scan_args(argc, argv, "22", &row, &column, &depth, &channel);
DATA_PTR(self) = cvCreateMat(FIX2INT(row), FIX2INT(column),
CV_MAKETYPE(CVMETHOD("DEPTH", depth, CV_8U), argc < 4 ? 3 : FIX2INT(channel)));
CvMat *ptr;
try {
ptr = cvCreateMat(FIX2INT(row), FIX2INT(column),
CV_MAKETYPE(CVMETHOD("DEPTH", depth, CV_8U), argc < 4 ? 3 : FIX2INT(channel)));
}
catch(cv::Exception& e) {
if (e.code != CV_StsNoMem) {
rb_raise(rb_eRuntimeError, "%s", e.what());
}
// When memory allocation is failed, run GC and retry it
rb_gc_start();
try {
ptr = cvCreateMat(FIX2INT(row), FIX2INT(column),
CV_MAKETYPE(CVMETHOD("DEPTH", depth, CV_8U), argc < 4 ? 3 : FIX2INT(channel)));
}
catch (...) {
fprintf(stderr, "[FATAL] failed to allocate memory\n");
exit(EXIT_FAILURE);
}
}
free(DATA_PTR(self));
DATA_PTR(self) = ptr;
return self;
}