From b17fceca72176afa596743055d495d185abd34c7 Mon Sep 17 00:00:00 2001
From: Peter Braden <peterbraden@peterbraden.co.uk>
Date: Wed, 13 Mar 2013 16:15:29 -0700
Subject: [PATCH] Initial work on face recognition using the FaceRecognizer
 class from opencv 2.4.4

---
 binding.gyp           |  1 +
 smoke/smoketest.js    |  7 +++++-
 src/FaceRecognizer.cc | 52 +++++++++++++++++++++++++++++++++++++++++++
 src/FaceRecognizer.h  | 28 +++++++++++++++++++++++
 src/init.cc           |  7 +++++-
 5 files changed, 93 insertions(+), 2 deletions(-)
 create mode 100644 src/FaceRecognizer.cc
 create mode 100644 src/FaceRecognizer.h

diff --git a/binding.gyp b/binding.gyp
index dd37c53..c4053a1 100755
--- a/binding.gyp
+++ b/binding.gyp
@@ -11,6 +11,7 @@
         , "src/VideoCaptureWrap.cc"
         , "src/CamShift.cc"
         , "src/HighGUI.cc"
+        , "src/FaceRecognizer.cc"
         ]
       , 'libraries': [
           '<!@(pkg-config --libs opencv)'
diff --git a/smoke/smoketest.js b/smoke/smoketest.js
index 6f9c3d9..3f79b5f 100755
--- a/smoke/smoketest.js
+++ b/smoke/smoketest.js
@@ -1,5 +1,10 @@
 var cv = require('../lib/opencv')
 
+
+
+var facerec = new cv.FaceRecognizer(); 
+
+/*
 var win = new cv.NamedWindow("foo");
 cv.readImage('./examples/stuff.png', function(e, im){
   im.pyrDown();
@@ -8,7 +13,7 @@ cv.readImage('./examples/stuff.png', function(e, im){
   win.destroy();
 
 })
-
+*/
 
 
 /*
diff --git a/src/FaceRecognizer.cc b/src/FaceRecognizer.cc
new file mode 100644
index 0000000..c37bb03
--- /dev/null
+++ b/src/FaceRecognizer.cc
@@ -0,0 +1,52 @@
+#include "FaceRecognizer.h"
+#include "OpenCV.h"
+
+#if CV_MAJOR_VERSION >= 2 && CV_MINOR_VERSION >=4
+
+#include "Matrix.h"
+
+void AsyncPredict(uv_work_t *req);
+void AfterAsyncPredict(uv_work_t *req);
+
+Persistent<FunctionTemplate> FaceRecognizerWrap::constructor;
+
+void
+FaceRecognizerWrap::Init(Handle<Object> target) {
+    HandleScope scope;
+
+    // Constructor
+      constructor = Persistent<FunctionTemplate>::New(FunctionTemplate::New(FaceRecognizerWrap::New));
+      constructor->InstanceTemplate()->SetInternalFieldCount(1);
+      constructor->SetClassName(String::NewSymbol("FaceRecognizer"));
+ 
+    NODE_SET_PROTOTYPE_METHOD(constructor, "train", Train);
+
+
+    target->Set(String::NewSymbol("FaceRecognizer"), constructor->GetFunction());
+};    
+
+Handle<Value>
+FaceRecognizerWrap::New(const Arguments &args) {
+  HandleScope scope;
+
+  if (args.This()->InternalFieldCount() == 0)
+    return v8::ThrowException(v8::Exception::TypeError(v8::String::New("Cannot Instantiate without new")));
+
+
+  // By default initialize eigenface
+  cv::FaceRecognizer *f = cv::createEigenFaceRecognizer();  
+  FaceRecognizerWrap *pt = new FaceRecognizerWrap(f);
+
+  pt->Wrap(args.This());
+  return args.This();
+}
+
+FaceRecognizerWrap::FaceRecognizerWrap(cv::FaceRecognizer* f){
+}
+
+Handle<Value>
+FaceRecognizerWrap::Train(const Arguments& args){
+	SETUP_FUNCTION(Matrix)
+
+}
+#endif // End version > 2.4
diff --git a/src/FaceRecognizer.h b/src/FaceRecognizer.h
new file mode 100644
index 0000000..17df540
--- /dev/null
+++ b/src/FaceRecognizer.h
@@ -0,0 +1,28 @@
+#include "OpenCV.h"
+
+#if CV_MAJOR_VERSION >= 2 && CV_MINOR_VERSION >=4
+
+#include "opencv2/contrib/contrib.hpp"
+
+class FaceRecognizerWrap: public node::ObjectWrap {
+  public:
+    cv::FaceRecognizer* rec;
+
+    static Persistent<FunctionTemplate> constructor;
+    static void Init(Handle<Object> target);
+    static Handle<Value> New(const Arguments &args);
+
+    FaceRecognizerWrap(cv::FaceRecognizer* f);
+
+    JSFUNC(Train)
+    JSFUNC(Update)
+    
+    JSFUNC(Predict)
+    static void EIO_Predict(eio_req *req);
+    static int EIO_AfterPredict(eio_req *req);
+
+};
+
+
+
+#endif
diff --git a/src/init.cc b/src/init.cc
index 063d984..1610800 100755
--- a/src/init.cc
+++ b/src/init.cc
@@ -6,6 +6,7 @@
 #include "Contours.h"
 #include "CamShift.h"
 #include "HighGUI.h"
+#include "FaceRecognizer.h"
 
 
 extern "C" void
@@ -18,8 +19,12 @@ init(Handle<Object> target) {
     VideoCaptureWrap::Init(target);
     Contour::Init(target);
 	  TrackedObject::Init(target);
-	  
     NamedWindow::Init(target);
+
+   #if CV_MAJOR_VERSION >= 2 && CV_MINOR_VERSION >=4
+     FaceRecognizerWrap::Init(target);
+   #endif
+
 };
 
 NODE_MODULE(opencv, init)