Merge pull request #358 from ScottRocha/master

Implement 'setTo()' function (Fixes #357)

Author: justadudewhohacks

cc/core/Mat.cc

@@ -28,6 +28,8 @@ NAN_MODULE_INIT(Mat::Init) {
   Nan::SetPrototypeMethod(ctor, "at", At);
   Nan::SetPrototypeMethod(ctor, "atRaw", AtRaw);
   Nan::SetPrototypeMethod(ctor, "set", Set);
+  Nan::SetPrototypeMethod(ctor, "setTo", SetTo);
+  Nan::SetPrototypeMethod(ctor, "setToAsync", SetToAsync);
   Nan::SetPrototypeMethod(ctor, "push_back", PushBack);
   Nan::SetPrototypeMethod(ctor, "push_backAsync", PushBackAsync);
   Nan::SetPrototypeMethod(ctor, "pushBack", PushBack);
@@ -295,6 +297,22 @@ NAN_METHOD(Mat::Set) {
   }
 }
 
+NAN_METHOD(Mat::SetTo) {
+  FF::SyncBinding(
+    std::make_shared<MatBindings::SetToWorker>(Mat::Converter::unwrap(info.This())),
+    "Mat::SetTo",
+    info
+  );
+}
+
+NAN_METHOD(Mat::SetToAsync) {
+  FF::AsyncBinding(
+    std::make_shared<MatBindings::SetToWorker>(Mat::Converter::unwrap(info.This())),
+    "Mat::SetToAsync",
+    info
+  );
+}
+
 NAN_METHOD(Mat::GetDataAsArray) {
   cv::Mat mat = FF_UNWRAP_MAT_AND_GET(info.This());
   FF_ARR rowArray = FF_NEW_ARRAY(mat.size[0]);

cc/core/Mat.h

@@ -64,6 +64,8 @@ class Mat : public Nan::ObjectWrap {
   static NAN_METHOD(At);
   static NAN_METHOD(AtRaw);
   static NAN_METHOD(Set);
+  static NAN_METHOD(SetTo);
+  static NAN_METHOD(SetToAsync);
   static NAN_METHOD(GetDataAsArray);
   static NAN_METHOD(GetRegion);
   static NAN_METHOD(Norm);

cc/core/MatBindings.h

@@ -5,6 +5,69 @@
 
 namespace MatBindings {
 
+  struct SetToWorker : public CatchCvExceptionWorker {
+  public:
+    cv::Mat self;
+    SetToWorker(cv::Mat self) {
+      this->self = self;
+    }
+
+    cv::Mat mask = cv::noArray().getMat();
+
+    double newVal1;
+    cv::Vec2d newVal2;
+    cv::Vec3d newVal3;
+    cv::Vec4d newVal4;
+
+    cv::Mat returnValue;
+
+    std::string executeCatchCvExceptionWorker() {
+      switch (self.channels()) {
+        case 2:
+          returnValue = self.setTo(newVal2, mask);
+          break;
+        case 3:
+          returnValue = self.setTo(newVal3, mask);
+          break;
+        case 4:
+          returnValue = self.setTo(newVal4, mask);
+          break;
+        default:
+          returnValue = self.setTo(newVal1, mask);
+          break;
+      }
+
+      return "";
+    }
+
+    FF_VAL getReturnValue() {
+      return Mat::Converter::wrap(returnValue);
+    }
+
+    bool unwrapRequiredArgs(Nan::NAN_METHOD_ARGS_TYPE info) {
+      switch (self.channels()) {
+        case 2:
+          return (Vec2::Converter::arg(0, &newVal2, info));
+          break;
+        case 3:
+          return (Vec3::Converter::arg(0, &newVal3, info));
+          break;
+        case 4:
+          return (Vec4::Converter::arg(0, &newVal4, info));
+          break;
+        default:
+          return (DoubleConverter::arg(0, &newVal1, info));
+          break;
+      }
+    }
+
+    bool unwrapOptionalArgs(Nan::NAN_METHOD_ARGS_TYPE info) {
+      return (
+        Mat::Converter::optArg(1, &mask, info)
+      );
+    }
+  };
+
   struct PushBackWorker : public CatchCvExceptionWorker {
   public:
     cv::Mat self;

lib/typings/Mat.d.ts

@@ -264,6 +264,14 @@ export class Mat {
   set(row: number, col: number, value: Vec2): void;
   set(row: number, col: number, value: Vec3): void;
   set(row: number, col: number, value: Vec4): void;
+  setTo(value: number, mask?: Mat): Mat;
+  setTo(value: Vec2, mask?: Mat): Mat;
+  setTo(value: Vec3, mask?: Mat): Mat;
+  setTo(value: Vec4, mask?: Mat): Mat;
+  setToAsync(value: number, mask?: Mat): Promise<Mat>;
+  setToAsync(value: Vec2, mask?: Mat): Promise<Mat>;
+  setToAsync(value: Vec3, mask?: Mat): Promise<Mat>;
+  setToAsync(value: Vec4, mask?: Mat): Promise<Mat>;
   sobel(ddepth: number, dx: number, dy: number, ksize?: number, scale?: number, delta?: number, borderType?: number): Mat;
   sobelAsync(ddepth: number, dx: number, dy: number, ksize?: number, scale?: number, delta?: number, borderType?: number): Promise<Mat>;
   split(): Mat[];