Decode and enroll images from strings in C API

Austin Blanton
1 parent 8ea461f0
Showing 4 changed files with 78 additions and 1 deletions
openbr/core/core.cpp
openbr/openbr.cpp
openbr/openbr.h
openbr/openbr_plugin.h
@@ -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 &amp;input, const File &amp;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);
@@ -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 &amp;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);
+}
@@ -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
@@ -1317,6 +1317,11 @@ BR_EXPORT void Train(const File &amp;input, const File &amp;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.