first draft of br-enroll reference implementation

Josh Klontz
1 parent 99e03750
Showing 3 changed files with 32 additions and 28 deletions
app/br-download/br-download.cpp
app/br-enroll/br-enroll.cpp
openbr/universal_template.cpp
@@ -60,9 +60,6 @@ static bool processReply(QNetworkReply* reply)
     if (!permissive && imdecode(Mat(1, data.size(), CV_8UC1, (void*)data.data()), IMREAD_ANYDEPTH | IMREAD_ANYCOLOR).empty())
         return false;
-    static QMutex lock;
-    QMutexLocker locker(&lock);
-
     const QByteArray hash = QCryptographicHash::hash(data, QCryptographicHash::Md5);
     br_append_utemplate_contents(stdout, reinterpret_cast<const int8_t*>(hash.data()), reinterpret_cast<const int8_t*>(hash.data()), 3, data.size(), reinterpret_cast<const int8_t*>(data.data()));
     return true;
@@ -75,7 +72,7 @@ int main(int argc, char *argv[])
     QNetworkAccessManager networkAccessManager;
     for (int i=1; i<argc; i++) {
-        if      (!strcmp(argv[i], "-help"      )) { help(); exit(0); }
+        if      (!strcmp(argv[i], "-help"      )) { help(); exit(EXIT_SUCCESS); }
         else if (!strcmp(argv[i], "-json"      )) json = true;
         else if (!strcmp(argv[i], "-permissive")) permissive = true;
         else                                      { url_provided = processReply(networkAccessManager.get(QNetworkRequest(QUrl(argv[i])))); }
@@ -46,17 +46,25 @@ static void enroll_utemplate(br_const_utemplate utemplate)
     if (utemplate->algorithmID != 3)
         qFatal("Expected an encoded image.");
-    TemplateList templateList;
-    templateList.append(Template(imdecode(Mat(1, utemplate->size, CV_8UC1, (void*) utemplate->data), IMREAD_UNCHANGED)));
-    templateList >> *algorithm;
+    TemplateList templates;
+    templates.append(Template(imdecode(Mat(1, utemplate->size, CV_8UC1, (void*) utemplate->data), IMREAD_UNCHANGED)));
+    templates >> *algorithm;
+
+    foreach (const Template &t, templates) {
+        const Mat &m = t.m();
+        const uint32_t size = m.rows * m.cols * m.elemSize();
+        const QByteArray templateID = QCryptographicHash::hash(QByteArray((const char*) m.data, size), QCryptographicHash::Md5);
+        br_append_utemplate_contents(stdout, utemplate->imageID, (const int8_t*) templateID.data(), -1, size, (const int8_t*) m.data);
+    }
 }
 int main(int argc, char *argv[])
 {
     for (int i=1; i<argc; i++)
-        if (!strcmp(argv[i], "-help")) { help(); exit(0); }
+        if (!strcmp(argv[i], "-help")) { help(); exit(EXIT_SUCCESS); }
     Context::initialize(argc, argv, "", false);
+    Globals->quiet = true;
     algorithm = Transform::fromAlgorithm("FaceRecognition");
     br_iterate_utemplates_file(stdin, enroll_utemplate);
     Context::finalize();
@@ -27,13 +27,14 @@ void br_append_utemplate(FILE *file, br_const_utemplate utemplate)
 void br_append_utemplate_contents(FILE *file, const int8_t *imageID, const int8_t *templateID, int32_t algorithmID, uint32_t size, const int8_t *data)
 {
-    QFile qFile;
-    qFile.open(file, QFile::WriteOnly | QFile::Append);
-    qFile.write((const char*) imageID, 16);
-    qFile.write((const char*) templateID, 16);
-    qFile.write((const char*) &algorithmID, 4);
-    qFile.write((const char*) &size, 4);
-    qFile.write((const char*) data, size);
+    static QMutex lock;
+    QMutexLocker locker(&lock);
+
+    fwrite(imageID, 16, 1, file);
+    fwrite(templateID, 16, 1, file);
+    fwrite(&algorithmID, 4, 1, file);
+    fwrite(&size, 4, 1, file);
+    fwrite(data, 1, size, file);
 }
 void br_iterate_utemplates(br_const_utemplate begin, br_const_utemplate end, br_utemplate_callback callback)
@@ -46,18 +47,16 @@ void br_iterate_utemplates(br_const_utemplate begin, br_const_utemplate end, br_
 void br_iterate_utemplates_file(FILE *file, br_utemplate_callback callback)
 {
-    QFile qFile;
-    qFile.open(file, QFile::ReadOnly);
-    while (!qFile.atEnd()) {
-        br_universal_template header;
-        if (qFile.peek((char*) &header, sizeof(br_universal_template)) != sizeof(br_universal_template))
-            qFatal("Unexpected EOF when peeking universal template header.");
-
-        const uint32_t size = sizeof(br_universal_template) + header.size;
-        QByteArray data = qFile.read(size);
-        if (uint32_t(data.size()) != size)
-            qFatal("Unexepected EOF when reading universal template.");
-
-        callback(reinterpret_cast<br_const_utemplate>(data.data()));
+    while (!feof(file)) {
+        br_utemplate t = (br_utemplate) malloc(sizeof(br_universal_template));
+
+        if (fread(t, sizeof(br_universal_template), 1, file) > 0) {
+            t = (br_utemplate) realloc(t, sizeof(br_universal_template) + t->size);
+            if (fread(t+1, 1, t->size, file) != t->size)
+                qFatal("Unexepected EOF when reading universal template data.");
+            callback(t);
+        }
+
+        free(t);
     }
 }