mexopencv  0.1
mex interface for opencv library
FREAK_.cpp
Go to the documentation of this file.
1 
8 #include <typeinfo>
9 #include "mexopencv.hpp"
10 #include "mexopencv_features2d.hpp"
11 #include "opencv2/xfeatures2d.hpp"
12 using namespace std;
13 using namespace cv;
14 using namespace cv::xfeatures2d;
15 
16 // Persistent objects
17 namespace {
19 int last_id = 0;
21 map<int,Ptr<FREAK> > obj_;
22 }
23 
31 void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
32 {
33  // Check the number of arguments
34  nargchk(nrhs>=2 && nlhs<=2);
35 
36  // Argument vector
37  vector<MxArray> rhs(prhs, prhs+nrhs);
38  int id = rhs[0].toInt();
39  string method(rhs[1].toString());
40 
41  // Constructor is called. Create a new object from argument
42  if (method == "new") {
43  nargchk(nrhs>=2 && nlhs<=1);
44  obj_[++last_id] = createFREAK(rhs.begin() + 2, rhs.end());
45  plhs[0] = MxArray(last_id);
46  return;
47  }
48 
49  // Big operation switch
50  Ptr<FREAK> obj = obj_[id];
51  if (method == "delete") {
52  nargchk(nrhs==2 && nlhs==0);
53  obj_.erase(id);
54  }
55  else if (method == "typeid") {
56  nargchk(nrhs==2 && nlhs<=1);
57  plhs[0] = MxArray(string(typeid(*obj).name()));
58  }
59  else if (method == "clear") {
60  nargchk(nrhs==2 && nlhs==0);
61  obj->clear();
62  }
63  else if (method == "load") {
64  nargchk(nrhs>=3 && (nrhs%2)==1 && nlhs==0);
65  string objname;
66  bool loadFromString = false;
67  for (int i=3; i<nrhs; i+=2) {
68  string key(rhs[i].toString());
69  if (key == "ObjName")
70  objname = rhs[i+1].toString();
71  else if (key == "FromString")
72  loadFromString = rhs[i+1].toBool();
73  else
74  mexErrMsgIdAndTxt("mexopencv:error",
75  "Unrecognized option %s", key.c_str());
76  }
77  obj_[id] = (loadFromString ?
78  Algorithm::loadFromString<FREAK>(rhs[2].toString(), objname) :
79  Algorithm::load<FREAK>(rhs[2].toString(), objname));
80  }
81  else if (method == "save") {
82  nargchk(nrhs==3 && nlhs==0);
83  obj->save(rhs[2].toString());
84  }
85  else if (method == "empty") {
86  nargchk(nrhs==2 && nlhs<=1);
87  plhs[0] = MxArray(obj->empty());
88  }
89  else if (method == "getDefaultName") {
90  nargchk(nrhs==2 && nlhs<=1);
91  plhs[0] = MxArray(obj->getDefaultName());
92  }
93  else if (method == "defaultNorm") {
94  nargchk(nrhs==2 && nlhs<=1);
95  plhs[0] = MxArray(NormTypeInv[obj->defaultNorm()]);
96  }
97  else if (method == "descriptorSize") {
98  nargchk(nrhs==2 && nlhs<=1);
99  plhs[0] = MxArray(obj->descriptorSize());
100  }
101  else if (method == "descriptorType") {
102  nargchk(nrhs==2 && nlhs<=1);
103  plhs[0] = MxArray(ClassNameInvMap[obj->descriptorType()]);
104  }
105  else if (method == "compute") {
106  nargchk(nrhs==4 && nlhs<=2);
107  if (rhs[2].isNumeric()) { // first variant that accepts an image
108  Mat image(rhs[2].toMat(rhs[2].isUint16() ? CV_16U : CV_8U)),
109  descriptors;
110  vector<KeyPoint> keypoints(rhs[3].toVector<KeyPoint>());
111  obj->compute(image, keypoints, descriptors);
112  plhs[0] = MxArray(descriptors);
113  if (nlhs > 1)
114  plhs[1] = MxArray(keypoints);
115  }
116  else if (rhs[2].isCell()) { // second variant that accepts an image set
117  //vector<Mat> images(rhs[2].toVector<Mat>());
118  vector<Mat> images, descriptors;
119  {
120  vector<MxArray> arr(rhs[2].toVector<MxArray>());
121  images.reserve(arr.size());
122  for (vector<MxArray>::const_iterator it = arr.begin(); it != arr.end(); ++it)
123  images.push_back(it->toMat(it->isUint16() ? CV_16U : CV_8U));
124  }
125  vector<vector<KeyPoint> > keypoints(rhs[3].toVector(
126  const_mem_fun_ref_t<vector<KeyPoint>, MxArray>(
127  &MxArray::toVector<KeyPoint>)));
128  obj->compute(images, keypoints, descriptors);
129  plhs[0] = MxArray(descriptors);
130  if (nlhs > 1)
131  plhs[1] = MxArray(keypoints);
132  }
133  else
134  mexErrMsgIdAndTxt("mexopencv:error", "Invalid arguments");
135  }
136  //else if (method == "detect")
137  //else if (method == "detectAndCompute")
138  else
139  mexErrMsgIdAndTxt("mexopencv:error",
140  "Unrecognized operation %s",method.c_str());
141 }
const ConstMap< int, std::string > NormTypeInv
Inverse norm type map for option processing.
Definition: mexopencv.hpp:145
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
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
Main entry called from Matlab.
Definition: FREAK_.cpp:31
Common definitions for the features2d and xfeatures2d modules.
const ConstMap< int, std::string > ClassNameInvMap
Translates data type definition used in OpenCV to that of MATLAB.
Definition: mexopencv.hpp:42
Global constant definitions.