CODEWITHSUNDEEP
c++opencv
Dat Chu
Dat Chu

Reputation: 11140

Construct OpenCV element type in runtime

In OpenCV, when I need to create a cv::Mat, I will need to do something along the line of

cv::Mat new_mat(width, height, CV_32FC3)

What happens if I only know that I need the elements to be either float or double + whether I need 1/2/3 channels in runtime?

In other words, given the element type (float) and number of channel (int), how can I construct the term: CV_32FC3?

Upvotes: 3

Views: 2436

Answers (1)

Reinderien
Reinderien

Reputation: 15233

Read the source for cxtypes.h. It contains lines like the following:

#define CV_32FC1 CV_MAKETYPE(CV_32F,1)
#define CV_32FC2 CV_MAKETYPE(CV_32F,2)
#define CV_32FC3 CV_MAKETYPE(CV_32F,3)
#define CV_32FC4 CV_MAKETYPE(CV_32F,4)
#define CV_32FC(n) CV_MAKETYPE(CV_32F,(n))

#define CV_64FC1 CV_MAKETYPE(CV_64F,1)
#define CV_64FC2 CV_MAKETYPE(CV_64F,2)
#define CV_64FC3 CV_MAKETYPE(CV_64F,3)
#define CV_64FC4 CV_MAKETYPE(CV_64F,4)
#define CV_64FC(n) CV_MAKETYPE(CV_64F,(n))

CV_MAKETYPE is defined as:

#define CV_MAKETYPE(depth,cn) (CV_MAT_DEPTH(depth) + (((cn)-1) << CV_CN_SHIFT))

This suggests the following code:

bool isdouble;
int nchannels;
// ...
if (isdouble) typeflag = CV_64FC(nchannels);
else typeflag = CV_32FC(nchannels);

I haven't tested this; let me know if it works. Also: I hate opencv's terrible type safety.

Upvotes: 6

Related Questions