initial interface for writing model files to likely

Josh Klontz
1 parent c11ed94c
Showing 9 changed files with 133 additions and 1 deletions
app/br/br.cpp
openbr/core/likely.cpp
openbr/core/likely.h
openbr/openbr.cpp
openbr/openbr.h
openbr/openbr_plugin.h
openbr/plugins/core/independent.cpp
openbr/plugins/core/pipe.cpp
openbr/plugins/imgproc/quantize.cpp
@@ -189,6 +189,9 @@ public:
             } else if (!strcmp(fun, "deduplicate")) {
                 check(parc == 3, "Incorrect parameter count for 'deduplicate'.");
                 br_deduplicate(parv[0], parv[1], parv[2]);
+            } else if (!strcmp(fun, "likely")) {
+                check(parc == 3, "Incorrect parameter count for 'likely'.");
+                br_likely(parv[0], parv[1], parv[2]);
             }
             // Miscellaneous
@@ -280,7 +283,8 @@ private:
                "-plotLandmarking <file> ... <file> {destination}\n"
                "-plotMetadata <file> ... <file> <columns>\n"
                "-project <input_gallery> {output_gallery}\n"
-               "-deduplicate <input_gallery> <output_gallery> <threshold>"
+               "-deduplicate <input_gallery> <output_gallery> <threshold>\n"
+               "-likely <input_type> <output_type> <output_likely_source>\n"
                "-getHeader <matrix>\n"
                "-setHeader {<matrix>} <target_gallery> <query_gallery>\n"
                "-<key> <value>\n"
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * Copyright 2015 Rank One Computing Corporation                             *
+ *                                                                           *
+ * Licensed under the Apache License, Version 2.0 (the "License");           *
+ * you may not use this file except in compliance with the License.          *
+ * You may obtain a copy of the License at                                   *
+ *                                                                           *
+ *     http://www.apache.org/licenses/LICENSE-2.0                            *
+ *                                                                           *
+ * Unless required by applicable law or agreed to in writing, software       *
+ * distributed under the License is distributed on an "AS IS" BASIS,         *
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  *
+ * See the License for the specific language governing permissions and       *
+ * limitations under the License.                                            *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#include "likely.h"
+
+using namespace br;
+
+void br::Likely(const QString &inputType, const QString &outputType, const File &outputSourceFile)
+{
+    const QSharedPointer<Transform> t(Transform::fromAlgorithm(Globals->algorithm));
+
+    QFile file(outputSourceFile.name);
+    if (!file.open(QFile::WriteOnly))
+        qFatal("Failed to open Likely output source file for writing!");
+
+    file.write("; Automatically generated source code from:\n");
+    file.write("; $ br -algorithm ");
+    file.write(Globals->algorithm.toLatin1());
+    file.write(" -likely ");
+    file.write(inputType.toLatin1());
+    file.write(" ");
+    file.write(outputType.toLatin1());
+    file.write(" ");
+    file.write(outputSourceFile.flat().toLatin1());
+    file.write("\n\n");
+
+    file.write("f :=\n");
+    file.write("  src :->\n");
+
+    file.write(t->likely(""));
+    file.write("\n");
+
+    file.write("\n(extern ");
+    file.write(outputType.toLatin1());
+    file.write(" ");
+    file.write(Globals->algorithm.toLower().toLatin1());
+    file.write(" ");
+    file.write(inputType.toLatin1());
+    file.write(" f)\n");
+}
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * Copyright 2015 Rank One Computing Corporation                             *
+ *                                                                           *
+ * Licensed under the Apache License, Version 2.0 (the "License");           *
+ * you may not use this file except in compliance with the License.          *
+ * You may obtain a copy of the License at                                   *
+ *                                                                           *
+ *     http://www.apache.org/licenses/LICENSE-2.0                            *
+ *                                                                           *
+ * Unless required by applicable law or agreed to in writing, software       *
+ * distributed under the License is distributed on an "AS IS" BASIS,         *
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  *
+ * See the License for the specific language governing permissions and       *
+ * limitations under the License.                                            *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#ifndef BR_LIKELY_H
+#define BR_LIKELY_H
+
+#include "openbr/openbr_plugin.h"
+
+namespace br
+{
+    void Likely(const QString &inputType, const QString &outputType, const File &outputSourceFile);
+}
+
+#endif // BR_LIKELY_H
@@ -20,6 +20,7 @@
 #include "core/cluster.h"
 #include "core/eval.h"
 #include "core/fuse.h"
+#include "core/likely.h"
 #include "core/plot.h"
 #include "core/qtutils.h"
 #include "plugins/openbr_internal.h"
@@ -481,3 +482,8 @@ void br_deduplicate(const char *input_gallery, const char *output_gallery, const
 {
     br::Deduplicate(input_gallery, output_gallery, threshold);
 }
+
+void br_likely(const char *input_type, const char *output_type, const char *output_source_file)
+{
+    br::Likely(input_type, output_type, output_source_file);
+}
@@ -115,6 +115,8 @@ BR_EXPORT const char *br_version();
 BR_EXPORT void br_slave_process(const char * baseKey);
+BR_EXPORT void br_likely(const char *input_type, const char *output_type, const char *output_source_file);
+
 // to avoid having to include unwanted headers
 // this will be this header's conception of a Template
 // any functions that need a Template pointer
@@ -813,6 +813,7 @@ public:
     }
     virtual Transform * simplify(bool &newTransform) { newTransform = false; return this; }
+    virtual QByteArray likely(const QByteArray &indentation) const { (void) indentation; return "src"; }
 protected:
     Transform(bool independent = true, bool trainable = true);
@@ -222,6 +222,13 @@ class IndependentTransform : public MetaTransform
         for (int i=0; i<size; i++)
             transforms[i]->load(stream);
     }
+
+    QByteArray likely(const QByteArray &indentation) const
+    {
+        if (transforms.size() != 1)
+            return "src"; // TODO: implement
+        return transforms.first()->likely(indentation);
+    }
 };
 BR_REGISTER(Transform, IndependentTransform)
@@ -179,6 +179,23 @@ class PipeTransform : public CompositeTransform
         CompositeTransform::init();
     }
+    QByteArray likely(const QByteArray &indentation) const
+    {
+        QByteArray result;
+        result.append("{\n");
+        foreach (Transform *t, transforms) {
+            const QByteArray dst = t->likely(indentation + "  ");
+            if (dst == "src")
+                continue; // Not implemented
+            result.append(indentation + "  src := ");
+            result.append(dst);
+            result.append("\n");
+        }
+
+        result.append(indentation + "  src\n}");
+        return result;
+    }
+
 protected:
     // Template list project -- process templates in parallel through Transform::project
     // or if parallelism is disabled, handle them sequentially
@@ -48,6 +48,21 @@ class QuantizeTransform : public Transform
     {
         src.m().convertTo(dst, CV_8U, a, b);
     }
+
+    QByteArray likely(const QByteArray &indentation) const
+    {
+        QByteArray result;
+        result.append("\n" + indentation + "{ ; Quantize\n");
+        result.append(indentation + "  dst := (imitate-size src (imitate-dimensions u8 src.type))\n");
+        result.append(indentation + "  (dst src) :=>\n");
+        result.append(indentation + "    dst :<- src.(* ");
+        result.append(QByteArray::number(a, 'g', 9));
+        result.append(").(+ ");
+        result.append(QByteArray::number(b, 'g', 9));
+        result.append(")\n");
+        result.append(indentation + "}");
+        return result;
+    }
 };
 BR_REGISTER(Transform, QuantizeTransform)