Implemented LoadStoreDistance

Josh Klontz
1 parent 22e1be95
Showing 2 changed files with 81 additions and 3 deletions
openbr/openbr_plugin.cpp
openbr/plugins/core/loadstore.cpp
@@ -1686,15 +1686,17 @@ Distance *Distance::make(QString str, QObject *parent)
     if (Globals->abbreviations.contains(str))
         return make(Globals->abbreviations[str], parent);
  
+    // Check for use of '<...>' as shorthand for LoadStore(...)
+    if (str.startsWith('<') && str.endsWith('>'))
+        return make("LoadStore(" + str.mid(1, str.size()-2) + ")", parent);
+
     { // Check for use of '+' as shorthand for Pipe(...)
         QStringList words = parse(str, '+');
         if (words.size() > 1)
             return make("Pipe([" + words.join(",") + "])", parent);
     }
  
-    File f = "." + str;
-    Distance *distance = Factory<Distance>::make(f);
-
+    Distance *const distance = Factory<Distance>::make("." + str);
     distance->setParent(parent);
     return distance;
 }
@@ -163,6 +163,82 @@ private:
  
 BR_REGISTER(Transform, LoadStoreTransform)
  
+/*!
+ * \ingroup distances
+ * \brief Caches Distance training.
+ * \author Josh Klontz \cite jklontz
+ */
+class LoadStoreDistance : public Distance
+{
+    Q_OBJECT
+    Q_PROPERTY(QString distanceString READ get_distanceString WRITE set_distanceString RESET reset_distanceString STORED false)
+    Q_PROPERTY(QString fileName READ get_fileName WRITE set_fileName RESET reset_fileName STORED false)
+    BR_PROPERTY(QString, distanceString, QString())
+    BR_PROPERTY(QString, fileName, QString())
+
+    QSharedPointer<Distance> distance;
+
+private:
+    void init()
+    {
+        const QString resolvedFileName = getFileName();
+        if (resolvedFileName.isEmpty()) {
+            distance.reset(Distance::make(distanceString));
+            return;
+        }
+
+        qDebug("Loading %s", qPrintable(resolvedFileName));
+        QFile file(resolvedFileName);
+        if (!file.open(QFile::ReadOnly))
+            qFatal("Failed to open %s for reading.", qPrintable(resolvedFileName));
+
+        QDataStream stream(&file);
+        stream >> distanceString;
+
+        distance.reset(Distance::make(distanceString));
+        distance->load(stream);
+    }
+
+    QString getFileName() const
+    {
+        foreach (const QString &file, QStringList() << fileName
+                                                    << Globals->sdkPath + "/share/openbr/models/distances/" + fileName)
+            if (QFileInfo(file).exists())
+                return file;
+        return QString();
+    }
+
+    void train(const TemplateList &src)
+    {
+        if (QFileInfo(getFileName()).exists())
+            return;
+
+        qDebug("Training %s", qPrintable(fileName));
+        distance->train(src);
+
+        qDebug("Storing %s", qPrintable(fileName));
+        QFile file(fileName);
+        if (!file.open(QFile::WriteOnly))
+            qFatal("Failed to open %s for writing.", qPrintable(fileName));
+
+        QDataStream stream(&file);
+        stream << distanceString;
+        distance->store(stream);
+    }
+
+    float compare(const Template &a, const Template &b) const
+    {
+        return distance->compare(a, b);
+    }
+
+    float compare(const uchar *a, const uchar *b, size_t size) const
+    {
+        return distance->compare(a, b, size);
+    }
+};
+
+BR_REGISTER(Distance, LoadStoreDistance)
+
 } // namespace br
  
 #include "core/loadstore.moc"