From 6f130e93e17f698322c362962de45bb2d19b8b5b Mon Sep 17 00:00:00 2001
From: bhklein <benhunter1990@gmail.com>
Date: Mon, 17 Nov 2014 12:52:59 -0500
Subject: [PATCH] Make janus_create_templates multi-threaded

---
 openbr/janus_io.cpp | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+), 0 deletions(-)

diff --git a/openbr/janus_io.cpp b/openbr/janus_io.cpp
index 4f704c8..56ae726 100644
--- a/openbr/janus_io.cpp
+++ b/openbr/janus_io.cpp
@@ -2,6 +2,7 @@
 
 #define JANUS_CUSTOM_ADD_SAMPLE
 #define JANUS_CUSTOM_CREATE_GALLERY
+#define JANUS_CUSTOM_CREATE_TEMPLATES
 #include "janus/src/janus_io.cpp"
 
 static void _janus_add_sample(vector<double> &samples, double sample)
@@ -35,3 +36,29 @@ janus_error janus_create_gallery(const char *data_path, janus_metadata metadata,
     futures.waitForFinished();
     return JANUS_SUCCESS;
 }
+
+janus_error janus_create_templates(const char *data_path, janus_metadata metadata, const char *gallery_file, int verbose)
+{
+    TemplateIterator ti(metadata, true);
+    TemplateData templateData = ti.next();
+    unsigned int num_templates = 1;
+
+    janus_gallery gallery;
+    JANUS_ASSERT(janus_allocate_gallery(&gallery))
+    QFutureSynchronizer<void> futures;
+    while (!templateData.templateIDs.empty()) {
+        futures.addFuture(QtConcurrent::run(_janus_create_template, data_path, templateData, gallery, verbose));
+        templateData = ti.next();
+        num_templates++;
+    }
+    futures.waitForFinished();
+    janus_flat_gallery flat_gallery = new janus_data[num_templates*janus_max_template_size()];
+    size_t bytes;
+    JANUS_ASSERT(janus_flatten_gallery(gallery, flat_gallery, &bytes))
+
+    std::ofstream file;
+    file.open(gallery_file, std::ios::out | std::ios::binary | std::ios::ate);
+    file.write((char*)flat_gallery, bytes);
+    file.close();
+    return JANUS_SUCCESS;
+}
--
libgit2 0.21.4