Add SuperpixelSLIC functionality

Author: grassias

binding.gyp

@@ -42,6 +42,7 @@
 			"cc/modules/imgproc/Moments.cc",
 			"cc/modules/ximgproc/ximgproc.cc",
 			"cc/modules/ximgproc/SuperpixelSEEDS.cc",
+			"cc/modules/ximgproc/SuperpixelSLIC.cc",			
 			"cc/modules/objdetect/objdetect.cc",
 			"cc/modules/objdetect/CascadeClassifier.cc",
 			"cc/modules/objdetect/HOGDescriptor.cc",

cc/modules/ximgproc/SuperpixelSLIC.cc

@@ -0,0 +1,61 @@
+#ifdef HAVE_XIMGPROC
+
+#include "SuperpixelSLIC.h"
+
+Nan::Persistent<v8::FunctionTemplate> SuperpixelSLIC::constructor;
+
+NAN_MODULE_INIT(SuperpixelSLIC::Init) {
+  v8::Local<v8::FunctionTemplate> ctor = Nan::New<v8::FunctionTemplate>(SuperpixelSLIC::New);
+	v8::Local<v8::ObjectTemplate> instanceTemplate = ctor->InstanceTemplate();
+  constructor.Reset(ctor);
+  instanceTemplate->SetInternalFieldCount(1);
+  ctor->SetClassName(Nan::New("SuperpixelSLIC").ToLocalChecked());
+
+  Nan::SetAccessor(instanceTemplate, Nan::New("img").ToLocalChecked(), SuperpixelSLIC::GetImg);
+	Nan::SetAccessor(instanceTemplate, Nan::New("labels").ToLocalChecked(), SuperpixelSLIC::GetLabels);
+	Nan::SetAccessor(instanceTemplate, Nan::New("labelContourMask").ToLocalChecked(), SuperpixelSLIC::GetLabelContourMask);  
+	Nan::SetAccessor(instanceTemplate, Nan::New("algorithm").ToLocalChecked(), SuperpixelSLIC::GetAlgorithm);
+	Nan::SetAccessor(instanceTemplate, Nan::New("regionSize").ToLocalChecked(), SuperpixelSLIC::GetRegionSize);
+	Nan::SetAccessor(instanceTemplate, Nan::New("ruler").ToLocalChecked(), SuperpixelSLIC::GetRuler);
+	Nan::SetAccessor(instanceTemplate, Nan::New("numCalculatedSuperpixels").ToLocalChecked(), SuperpixelSLIC::GetNumCalculatedSuperpixels);
+	
+	Nan::SetPrototypeMethod(ctor, "iterate", SuperpixelSLIC::Iterate);
+
+  target->Set(Nan::New("SuperpixelSLIC").ToLocalChecked(), ctor->GetFunction());
+};
+
+NAN_METHOD(SuperpixelSLIC::New) {
+	FF_METHOD_CONTEXT("SuperpixelSLIC::New");
+	if (!info.IsConstructCall()) {
+		FF_THROW("expected new key word");
+	}
+	SuperpixelSLIC* self = new SuperpixelSLIC();
+
+	FF_ARG_INSTANCE(0, self->img, Mat::constructor, FF_UNWRAP_MAT_AND_GET);
+	FF_ARG_INT(1, self->algorithm);
+  FF_ARG_INT_IFDEF(2, self->regionSize, 10);
+  FF_ARG_INT_IFDEF(3, self->ruler, 10.0);
+	
+	self->Wrap(info.Holder());
+	self->superpixelSlic = cv::ximgproc::createSuperpixelSLIC(
+		self->img,
+		self->algorithm, 
+		self->regionSize, 
+		self->ruler
+	);
+  info.GetReturnValue().Set(info.Holder());
+}
+
+NAN_METHOD(SuperpixelSLIC::Iterate) {
+	FF_METHOD_CONTEXT("SuperpixelSLIC::Iterate");
+
+	FF_ARG_UINT_IFDEF(0, uint iterations, 10);
+
+	SuperpixelSLIC* self = FF_UNWRAP(info.This(), SuperpixelSLIC);
+	self->superpixelSlic->iterate((int)iterations);
+	self->superpixelSlic->getLabels(self->labels);
+	self->numCalculatedSuperpixels = self->superpixelSlic->getNumberOfSuperpixels();
+	self->superpixelSlic->getLabelContourMask(self->labelContourMask, false);
+}
+
+#endif // HAVE_XIMGPROC

cc/modules/ximgproc/SuperpixelSLIC.h

@@ -0,0 +1,34 @@
+#include <opencv2/ximgproc.hpp>
+#include "Mat.h"
+#include "macros.h"
+
+#ifndef __FF_SUPERPIXELSLIC_H__
+#define __FF_SUPERPIXELSLIC_H__
+
+class SuperpixelSLIC : public Nan::ObjectWrap {
+public:
+	cv::Ptr<cv::ximgproc::SuperpixelSLIC> superpixelSlic;
+	cv::Mat img;
+	cv::Mat labels;
+	cv::Mat labelContourMask;
+	int algorithm;
+	int regionSize = 10;
+	float ruler = 10.0;
+	int numCalculatedSuperpixels = 0;
+
+  static NAN_MODULE_INIT(Init); 
+  static NAN_METHOD(New);
+	static NAN_METHOD(Iterate);
+
+	static FF_GETTER_JSOBJ(SuperpixelSLIC, GetImg, img, FF_UNWRAP_MAT_AND_GET, Mat::constructor);
+	static FF_GETTER_JSOBJ(SuperpixelSLIC, GetLabels, labels, FF_UNWRAP_MAT_AND_GET, Mat::constructor);
+	static FF_GETTER_JSOBJ(SuperpixelSLIC, GetLabelContourMask, labelContourMask, FF_UNWRAP_MAT_AND_GET, Mat::constructor);
+	static FF_GETTER(SuperpixelSLIC, GetAlgorithm, algorithm);
+	static FF_GETTER(SuperpixelSLIC, GetRegionSize, regionSize);
+	static FF_GETTER(SuperpixelSLIC, GetRuler, ruler);
+	static FF_GETTER(SuperpixelSLIC, GetNumCalculatedSuperpixels, numCalculatedSuperpixels);
+
+  static Nan::Persistent<v8::FunctionTemplate> constructor;
+};
+
+#endif

cc/modules/ximgproc/ximgproc.cc

@@ -2,9 +2,11 @@
 
 #include "ximgproc.h"
 #include "SuperpixelSEEDS.h"
+#include "SuperpixelSLIC.h"
 
 NAN_MODULE_INIT(XImgproc::Init) {
 	SuperpixelSEEDS::Init(target);
+	SuperpixelSLIC::Init(target);
 }
 
 #endif // HAVE_XIMGPROC