Add mask parameter to matrix.matchTemplate

examples/match-template-mask.js

@@ -0,0 +1,22 @@
+// see https://github.com/Itseez/opencv/blob/master/samples/cpp/mask_tmpl.cpp
+const cv = require('../lib/opencv');
+
+const TM_CCORR_NORMED = 3
+const imageFilename = 'files/lena_tmpl.jpg';
+const templateFilename = 'files/tmpl.png';
+const maskFilename = 'files/mask.png';
+const outFilename = 'tmp/match-template-mask.png';
+
+cv.readImage(imageFilename, function(err, im) {
+  if (err) throw err;
+  cv.readImage(templateFilename, function (err, tmpl) {
+    if (err) throw err;
+    const res = im.matchTemplate(templateFilename, TM_CCORR_NORMED,  maskFilename);
+    const minMax = res.minMaxLoc();
+    const topLeft = minMax.maxLoc;
+    im.rectangle([topLeft.x, topLeft.y], [tmpl.width(), tmpl.height()], [0, 255,0], 2);
+
+    im.save(outFilename);
+    console.log('Image saved to ' + outFilename);
+  });
+});

examples/match-template.js

@@ -0,0 +1,20 @@
+const cv = require('../lib/opencv');
+
+const TM_CCORR_NORMED = 3;
+const imageFilename = 'files/stuff.png';
+const templateFilename = 'files/stuff-template.png';
+const outFilename = 'tmp/match-template.png';
+
+cv.readImage(imageFilename, function (err, im) {
+  if (err) throw err;
+  cv.readImage(templateFilename, function (err, tmpl) {
+    if (err) throw err;
+    const res = im.matchTemplate(templateFilename, TM_CCORR_NORMED);
+    const minMax = res.minMaxLoc();
+    const topLeft = minMax.maxLoc;
+    im.rectangle([topLeft.x, topLeft.y], [tmpl.width(), tmpl.height()], [0, 255,0], 2);
+
+    im.save(outFilename);
+    console.log('Image saved to ' + outFilename);
+  });
+});

src/Matrix.cc

@@ -2200,7 +2200,7 @@ NAN_METHOD(Matrix::TemplateMatches) {
 
 // @author ytham
 // Match Template filter
-// Usage: output = input.matchTemplate("templateFileString", method);
+// Usage: output = input.matchTemplate("templateFileString"[, "method"][, "maskPath"]);
 NAN_METHOD(Matrix::MatchTemplate) {
   Nan::HandleScope scope;
 
@@ -2229,6 +2229,19 @@ NAN_METHOD(Matrix::MatchTemplate) {
   int method = (info.Length() < 2) ? (int)cv::TM_CCORR_NORMED : info[1]->Uint32Value();
   cv::matchTemplate(self->mat, templ, m_out->mat, method);
 
+  if (info.Length() < 3) {
+    cv::matchTemplate(self->mat, templ, m_out->mat, method);
+  } else {
+    v8::String::Utf8Value args2(info[2]->ToString());
+    std::string maskFilename = std::string(*args2);
+    cv::Mat mask;
+    mask = cv::imread(maskFilename, CV_8S);
+    if (mask.size().height == 0 && mask.size().width == 0) {
+      throw std::runtime_error(("Failed to load mask file: " + maskFilename).c_str());
+    }
+    cv::matchTemplate(self->mat, templ, m_out->mat, method, mask);
+  }
+
   info.GetReturnValue().Set(out);
 }