now supporting threaded janus_enroll

Josh Klontz
1 parent 48097472
Showing 3 changed files with 39 additions and 2 deletions
openbr/CMakeLists.txt
openbr/janus
openbr/janus_io.cpp
@@ -5,7 +5,7 @@ include_directories(${CMAKE_CURRENT_BINARY_DIR})
 # Janus API
 include_directories(janus/include)
 install(DIRECTORY janus/include DESTINATION .)
-set(BR_JANUS janus/src/janus_io.cpp janus/src/opencv_io/opencv_io.cpp)
+set(BR_JANUS janus/src/opencv_io/opencv_io.cpp)
 add_definitions(-DJANUS_LIBRARY)
 # Collect source files
-Subproject commit 2ee03fde082140e3a4210775cbbc2898e888e90a
+Subproject commit 670601106366ca802289ca1916a47b6f2ed4976e
+#include <QtConcurrent>
+
+#define JANUS_CUSTOM_ADD_SAMPLE
+#define JANUS_CUSTOM_CREATE_GALLERY
+#include "janus/src/janus_io.cpp"
+
+static void _janus_add_sample(vector<double> &samples, double sample)
+{
+    static QMutex sampleLock;
+    QMutexLocker sampleLocker(&sampleLock);
+    samples.push_back(sample);
+}
+
+static void _janus_create_template(const char *data_path, TemplateData templateData, janus_gallery gallery)
+{
+    janus_template template_;
+    janus_template_id templateID;
+    JANUS_ASSERT(TemplateIterator::create(data_path, templateData, &template_, &templateID))
+
+    static QMutex enrollLock;
+    QMutexLocker enrollLocker(&enrollLock);
+
+    JANUS_ASSERT(janus_enroll(template_, templateID, gallery))
+}
+
+janus_error janus_create_gallery(const char *data_path, janus_metadata metadata, janus_gallery gallery)
+{
+    TemplateIterator ti(metadata, true);
+    TemplateData templateData = ti.next();
+    QFutureSynchronizer<void> futures;
+    while (!templateData.templateIDs.empty()) {
+        futures.addFuture(QtConcurrent::run(_janus_create_template, data_path, templateData, gallery));
+        templateData = ti.next();
+    }
+    futures.waitForFinished();
+    return JANUS_SUCCESS;
+}