feat: add face_landmark_detection

config.m4

@@ -25,7 +25,8 @@ if test "$PHP_PDLIB" != "no"; then
   PHP_SUBST(PDLIB_SHARED_LIBADD)
 
   pdlib_src_files="pdlib.cc \
-  src/face_detection.cc"
+  src/face_detection.cc \
+  src/face_landmark_detection.cc"
 
   AC_MSG_CHECKING(for pkg-config)
   if test ! -f "$PKG_CONFIG"; then

pdlib.cc

@@ -29,6 +29,7 @@ extern "C" {
 }
 #include "php_pdlib.h"
 #include "src/face_detection.h"
+#include "src/face_landmark_detection.h"
 
 /* If you declare any globals in php_pdlib.h uncomment this:
 ZEND_DECLARE_MODULE_GLOBALS(pdlib)
@@ -151,6 +152,7 @@ PHP_MINFO_FUNCTION(pdlib)
 const zend_function_entry pdlib_functions[] = {
 	PHP_FE(confirm_pdlib_compiled,	NULL)
 	PHP_FE(dlib_face_detection, dlib_face_detection_arginfo)
+	PHP_FE(dlib_face_landmark_detection, dlib_face_landmark_detection_arginfo)
 	PHP_FE_END	/* Must be the last line in pdlib_functions[] */
 };
 /* }}} */

src/face_landmark_detection.cc

@@ -0,0 +1,63 @@
+
+#include "../php_pdlib.h"
+#include "face_landmark_detection.h"
+
+#include <dlib/image_processing/frontal_face_detector.h>
+#include <dlib/image_processing/render_face_detections.h>
+#include <dlib/image_processing.h>
+#include <dlib/image_io.h>
+#include <iostream>
+
+#define ARRAY_NAME_WITH_INDEX(name, index) name##index
+
+using namespace dlib;
+using namespace std;
+
+PHP_FUNCTION(dlib_face_landmark_detection)
+{
+    char  *shape_predictor_file_path;
+    char  *img_path;
+    size_t  shape_predictor_file_path_len, img_path_len;
+
+    if(zend_parse_parameters(ZEND_NUM_ARGS(), "ss", &shape_predictor_file_path, &shape_predictor_file_path_len,
+            &img_path, &img_path_len) == FAILURE){
+        RETURN_FALSE;
+    }
+    zval return_val;
+    array_init(&return_val);
+
+    try {
+        frontal_face_detector detector = get_frontal_face_detector();
+        shape_predictor sp;
+        deserialize(shape_predictor_file_path) >> sp;
+
+        array2d<rgb_pixel> img;
+        load_image(img, img_path);
+        pyramid_up(img);
+
+        std::vector<rectangle> dets = detector(img);
+
+        std::vector<full_object_detection> shapes;
+        for (unsigned long j = 0; j < dets.size(); ++j)
+        {
+            full_object_detection shape = sp(img, dets[j]);
+
+            zval ARRAY_NAME_WITH_INDEX(face, j);
+            array_init(&ARRAY_NAME_WITH_INDEX(face, j));
+            for (int k = 0; k < shape.num_parts(); k++) {
+                zval ARRAY_NAME_WITH_INDEX(part, k);
+                array_init(&ARRAY_NAME_WITH_INDEX(part, k));
+                dlib::point p = shape.part(k);
+                add_next_index_long(&ARRAY_NAME_WITH_INDEX(part, k),p.x());
+                add_next_index_long(&ARRAY_NAME_WITH_INDEX(part, k),p.y());
+                add_next_index_zval(&ARRAY_NAME_WITH_INDEX(face, j), &ARRAY_NAME_WITH_INDEX(part, k));
+            }
+            add_next_index_zval(&return_val, &ARRAY_NAME_WITH_INDEX(face, j));
+        }
+        RETURN_ZVAL(&return_val, 0, 0);
+    }
+    catch (exception& e)
+    {
+        RETURN_FALSE;
+    }
+}

src/face_landmark_detection.h

@@ -0,0 +1,14 @@
+//
+// Created by goodspb on 2018/5/23.
+//
+
+#ifndef PDLIB_FACE_LANDMARK_DETECTION_H
+#define PDLIB_FACE_LANDMARK_DETECTION_H
+
+ZEND_BEGIN_ARG_INFO_EX(dlib_face_landmark_detection_arginfo, 0, 0, 1)
+ZEND_ARG_INFO(0, shape_predictor_file_path)
+ZEND_ARG_INFO(0, img_path)
+ZEND_END_ARG_INFO()
+PHP_FUNCTION(dlib_face_landmark_detection);
+
+#endif //PDLIB_FACE_LANDMARK_DETECTION_H