From 51f633f32529f38b6194a3c18579f0fd78320eb7 Mon Sep 17 00:00:00 2001
From: Austin Blanton <imaus10@gmail.com>
Date: Tue, 5 Nov 2013 15:38:26 -0500
Subject: [PATCH] Decode and enroll images from strings in C API

---
 openbr/core/core.cpp   | 15 ++++++++++++++-
 openbr/openbr.cpp      | 42 ++++++++++++++++++++++++++++++++++++++++++
 openbr/openbr.h        | 17 +++++++++++++++++
 openbr/openbr_plugin.h |  5 +++++
 4 files changed, 78 insertions(+), 1 deletion(-)

diff --git a/openbr/core/core.cpp b/openbr/core/core.cpp
index 06b7194..53b034c 100644
--- a/openbr/core/core.cpp
+++ b/openbr/core/core.cpp
@@ -160,7 +160,7 @@ struct AlgorithmCore
                                 data.removeAt(i);
                     const int numFiles = data.size();
 
-                    data >> *transform;
+                    enroll(data);
 
                     g->writeBlock(data);
                     const FileList newFiles = data.files();
@@ -184,6 +184,11 @@ struct AlgorithmCore
         return fileList;
     }
 
+    void enroll(TemplateList &data)
+    {
+        data >> *transform;
+    }
+
     void retrieveOrEnroll(const File &file, QScopedPointer<Gallery> &gallery, FileList &galleryFiles)
     {
         if (!file.getBool("enroll") && (QStringList() << "gal" << "mem" << "template").contains(file.suffix())) {
@@ -371,6 +376,14 @@ FileList br::Enroll(const File &input, const File &gallery)
     return AlgorithmManager::getAlgorithm(gallery.get<QString>("algorithm"))->enroll(input, gallery);
 }
 
+void br::Enroll(const Template &tmpl)
+{
+    QList<Template> tmpls;
+    tmpls.append(tmpl);
+    TemplateList tl(tmpls);
+    AlgorithmManager::getAlgorithm(tmpl.file.get<QString>("algorithm"))->enroll(tl);
+}
+
 void br::Compare(const File &targetGallery, const File &queryGallery, const File &output)
 {
     AlgorithmManager::getAlgorithm(output.get<QString>("algorithm"))->compare(targetGallery, queryGallery, output);
diff --git a/openbr/openbr.cpp b/openbr/openbr.cpp
index 0a0ccc7..b5c1d70 100644
--- a/openbr/openbr.cpp
+++ b/openbr/openbr.cpp
@@ -23,6 +23,7 @@
 #include "core/plot.h"
 #include "core/qtutils.h"
 #include "plugins/openbr_internal.h"
+#include <opencv2/highgui/highgui.hpp>
 
 using namespace br;
 
@@ -114,6 +115,14 @@ void br_initialize(int &argc, char *argv[], const char *sdk_path)
     Context::initialize(argc, argv, sdk_path);
 }
 
+void br_initialize_default()
+{
+    int argc = 1;
+    char app[] = "br";
+    char *argv[1] = {app};
+    Context::initialize(argc, argv, "");
+}
+
 bool br_is_classifier(const char *algorithm)
 {
     return IsClassifier(algorithm);
@@ -289,3 +298,36 @@ void br_slave_process(const char * baseName)
     worker->mainLoop();
     delete worker;
 }
+
+br_template br_load_img(const char *data)
+{
+    int size = strlen(data);
+    std::vector<char> buf(data, data+size);
+    cv::Mat img = cv::imdecode(cv::Mat(buf), CV_LOAD_IMAGE_ANYDEPTH);
+    Template *tmpl = new Template(img);
+    return (br_template)tmpl;
+}
+
+const unsigned char *br_unload_img(br_template tmpl)
+{
+    Template *t = (Template*)tmpl;
+    return t->m().data;
+}
+
+int br_img_rows(br_template tmpl)
+{
+    Template *t = (Template*)tmpl;
+    return t->m().rows;
+}
+
+int br_img_cols(br_template tmpl)
+{
+    Template *t = (Template*)tmpl;
+    return t->m().cols;
+}
+
+void br_enroll_template(br_template tmpl)
+{
+    Template *t = (Template*)tmpl;
+    Enroll(*t);
+}
diff --git a/openbr/openbr.h b/openbr/openbr.h
index 93d0d8f..175a384 100644
--- a/openbr/openbr.h
+++ b/openbr/openbr.h
@@ -206,6 +206,7 @@ BR_EXPORT void br_fuse(int num_input_simmats, const char *input_simmats[],
  * \see br_finalize
  */
 BR_EXPORT void br_initialize(int &argc, char *argv[], const char *sdk_path = "");
+BR_EXPORT void br_initialize_default();
 
 /*!
  * \brief Wraps br::IsClassifier()
@@ -414,6 +415,22 @@ BR_EXPORT const char *br_version();
   */
 BR_EXPORT void br_slave_process(const char * baseKey);
 
+// to avoid having to include unwanted headers
+// this will be this header's conception of a template
+// any functions that need a Template pointer
+// will take this typedef and cast it
+// (and then we'll cross our fingers)
+typedef void* br_template;
+/*!
+  * \brief Load an image from a string buffer.
+  *   Easy way to pass an image in memory from another programming language to openbr.
+  */
+BR_EXPORT br_template br_load_img(const char *data);
+BR_EXPORT const unsigned char* br_unload_img(br_template tmpl);
+BR_EXPORT int br_img_rows(br_template tmpl);
+BR_EXPORT int br_img_cols(br_template tmpl);
+BR_EXPORT void br_enroll_template(br_template tmpl);
+
 /*! @}*/
 
 #ifdef __cplusplus
diff --git a/openbr/openbr_plugin.h b/openbr/openbr_plugin.h
index eb563c9..48be572 100644
--- a/openbr/openbr_plugin.h
+++ b/openbr/openbr_plugin.h
@@ -1317,6 +1317,11 @@ BR_EXPORT void Train(const File &input, const File &model);
  * \see br_enroll
  */
 BR_EXPORT FileList Enroll(const File &input, const File &gallery = File());
+/*!
+ * \brief High-level function for enrolling templates.
+ * \see br_enroll
+ */
+BR_EXPORT void Enroll(const Template &tmpl);
 
 /*!
  * \brief High-level function for comparing galleries.
--
libgit2 0.21.4