Server: use JSON file for reference faces

Author: fuzun

server/config.ini

@@ -13,7 +13,7 @@ Enable=1
 Location=log.txt
 
 [DLIB]
-ReferenceFile=face.txt
+ReferenceFile=faces.json
 FaceLandmarkModelFile=shape_predictor_5_face_landmarks.dat
 FaceRecognitionModelFile=dlib_face_recognition_resnet_model_v1.dat
 ImageNetClassifierFile=resnet34_1000_imagenet_classifier.dnn

server/source/DLIBWorker/DLIBWorker.cpp

@@ -23,6 +23,9 @@
 #include <QPixmap>
 #include <QSettings>
 #include <QFile>
+#include <QJsonDocument>
+#include <QJsonArray>
+#include <QJsonObject>
 #include <vector>
 
 #include <Client/Client.h>
@@ -63,22 +66,32 @@ DLIBWorker::DLIBWorker(class QSettings* config, const Settings *settings)
     }
     else
     {
-        while(!refFile.atEnd())
+        QByteArray data = refFile.readAll();
+
+        QJsonDocument doc(QJsonDocument::fromJson(data));
+
+        for (auto&& i : doc.array())
         {
-            QString line = refFile.readLine();
-            QStringList lineSplit = line.split(":");
-            if (!lineSplit[1].contains(","))
-            {
-                m_refPhotoFileList.push_back(qMakePair(lineSplit[0].trimmed(), lineSplit[1].trimmed()));
-            }
-            else
+            if (!i.isObject())
+                continue;
+
+            QJsonObject obj = i.toObject();
+
+            QString tag = obj.value("tag").toString();
+
+            auto&& k = obj.value("photos");
+
+            if (k.isArray())
             {
-                QStringList lineSplit2 = lineSplit[1].split(",");
-                for (const auto& it : lineSplit2)
+                for (auto&& j : k.toArray())
                 {
-                    m_refPhotoFileList.push_back(qMakePair(lineSplit[0].trimmed(), it.trimmed()));
+                    m_refPhotoFileList.push_back(qMakePair(tag, j.toString()));
                 }
             }
+            else if (k.isString())
+            {
+                m_refPhotoFileList.push_back(qMakePair(tag, k.toString()));
+            }
         }
     }
 }

server/source/config.h

@@ -56,7 +56,7 @@
 #define CONFIG_DLIB_FACE_PADDING "FacePadding"
 #define CONFIG_DLIB_DEFAULT_FACE_PADDING 0.25
 #define CONFIG_DLIB_REFERENCEFACEFILE "ReferenceFile"
-#define CONFIG_DLIB_DEFAULT_REFERENCEFACEFILE "face.txt"
+#define CONFIG_DLIB_DEFAULT_REFERENCEFACEFILE "faces.json"
 #define CONFIG_DLIB_IMAGENETCLASSIFIERFILE "ImageNetClassifierFile"
 #define CONFIG_DLIB_DEFAULT_IMAGENETCLASSIFIERFILE "resnet34_1000_imagenet_classifier.dnn"
 #define CONFIG_DLIB_NUMCROPS "NumberOfCrops"