mexopencv  0.1
mex interface for opencv library
textureFlattening.cpp
Go to the documentation of this file.
1 
8 #include "mexopencv.hpp"
9 #include "opencv2/photo.hpp"
10 using namespace std;
11 using namespace cv;
12 
20 void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
21 {
22  // Check the number of arguments
23  nargchk(nrhs>=2 && (nrhs%2)==0 && nlhs<=1);
24 
25  // Argument vector
26  vector<MxArray> rhs(prhs, prhs+nrhs);
27 
28  // Option processing
29  float low_threshold = 30;
30  float high_threshold = 45;
31  int kernel_size = 3;
32  bool flip = true;
33  for (int i=2; i<nrhs; i+=2) {
34  string key(rhs[i].toString());
35  if (key == "LowThreshold")
36  low_threshold = rhs[i+1].toFloat();
37  else if (key == "HighThreshold")
38  high_threshold = rhs[i+1].toFloat();
39  else if (key == "KernelSize")
40  kernel_size = rhs[i+1].toInt();
41  else if (key == "FlipChannels")
42  flip = rhs[i+1].toBool();
43  else
44  mexErrMsgIdAndTxt("mexopencv:error",
45  "Unrecognized option %s", key.c_str());
46  }
47 
48  // Process
49  Mat src(rhs[0].toMat(CV_8U)),
50  mask(rhs[1].toMat(CV_8U)),
51  dst;
52  // MATLAB's default is RGB while OpenCV's is BGR
53  if (flip && src.channels() == 3)
54  cvtColor(src, src, cv::COLOR_RGB2BGR);
55  if (flip && mask.channels() == 3)
56  cvtColor(mask, mask, cv::COLOR_RGB2BGR);
57  textureFlattening(src, mask, dst, low_threshold, high_threshold, kernel_size);
58  // OpenCV's default is BGR while MATLAB's is RGB
59  if (flip && dst.channels() == 3)
60  cvtColor(dst, dst, cv::COLOR_BGR2RGB);
61  plhs[0] = MxArray(dst);
62 }
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
Main entry called from Matlab.
mxArray object wrapper for data conversion and manipulation.
Definition: MxArray.hpp:123
void nargchk(bool cond)
Alias for input/ouput arguments number check.
Definition: mexopencv.hpp:166
Global constant definitions.