mexopencv  0.1
mex interface for opencv library
MSDDetector_.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<MSDDetector> > 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<=1);
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] = createMSDDetector(rhs.begin() + 2, rhs.end());
45  plhs[0] = MxArray(last_id);
46  return;
47  }
48 
49  // Big operation switch
50  Ptr<MSDDetector> 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<MSDDetector>(rhs[2].toString(), objname) :
79  Algorithm::load<MSDDetector>(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 == "detect") {
94  nargchk(nrhs>=3 && (nrhs%2)==1 && nlhs<=1);
95  if (rhs[2].isNumeric()) { // first variant that accepts an image
96  Mat mask;
97  for (int i=3; i<nrhs; i+=2) {
98  string key(rhs[i].toString());
99  if (key == "Mask")
100  mask = rhs[i+1].toMat(CV_8U);
101  else
102  mexErrMsgIdAndTxt("mexopencv:error",
103  "Unrecognized option %s", key.c_str());
104  }
105  Mat image(rhs[2].toMat(CV_8U));
106  vector<KeyPoint> keypoints;
107  obj->detect(image, keypoints, mask);
108  plhs[0] = MxArray(keypoints);
109  }
110  else if (rhs[2].isCell()) { // second variant that accepts an image set
111  vector<Mat> masks;
112  for (int i=3; i<nrhs; i+=2) {
113  string key(rhs[i].toString());
114  if (key == "Mask") {
115  //masks = rhs[i+1].toVector<Mat>();
116  vector<MxArray> arr(rhs[i+1].toVector<MxArray>());
117  masks.clear();
118  masks.reserve(arr.size());
119  for (vector<MxArray>::const_iterator it = arr.begin(); it != arr.end(); ++it)
120  masks.push_back(it->toMat(CV_8U));
121  }
122  else
123  mexErrMsgIdAndTxt("mexopencv:error",
124  "Unrecognized option %s", key.c_str());
125  }
126  //vector<Mat> images(rhs[2].toVector<Mat>());
127  vector<Mat> images;
128  {
129  vector<MxArray> arr(rhs[2].toVector<MxArray>());
130  images.reserve(arr.size());
131  for (vector<MxArray>::const_iterator it = arr.begin(); it != arr.end(); ++it)
132  images.push_back(it->toMat(CV_8U));
133  }
134  vector<vector<KeyPoint> > keypoints;
135  obj->detect(images, keypoints, masks);
136  plhs[0] = MxArray(keypoints);
137  }
138  else
139  mexErrMsgIdAndTxt("mexopencv:error", "Invalid arguments");
140  }
141  //else if (method == "defaultNorm")
142  //else if (method == "descriptorSize")
143  //else if (method == "descriptorType")
144  //else if (method == "compute")
145  //else if (method == "detectAndCompute")
146  else
147  mexErrMsgIdAndTxt("mexopencv:error",
148  "Unrecognized operation %s",method.c_str());
149 }
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
Common definitions for the features2d and xfeatures2d modules.
Global constant definitions.