Added Retain Transform, which retains on listed key/value pairs in a File's local metadata

Scott Klum
1 parent 0978a74d
Showing 3 changed files with 32 additions and 3 deletions
openbr/openbr_plugin.cpp
openbr/plugins/template.cpp
openbr/plugins/validate.cpp
@@ -410,7 +410,6 @@ TemplateList TemplateList::fromGallery(const br::File &amp;gallery)
                 } else if (newTemplates[i].file.getBool("allPartitions")) {
                     // The allPartitions flag is used to add an extended set
                     // of target images to every partition
-
                     newTemplates[i].file.set("Partition", -1);
                 } else {
                     const QByteArray md5 = QCryptographicHash::hash(newTemplates[i].file.get<QString>("Subject").toLatin1(), QCryptographicHash::Md5);
+#include "openbr_internal.h"
+
+namespace br
+{
+
+/*!
+ * \ingroup transforms
+ * \brief Retains only the values for the keys listed, to reduce template size
+ * \author Scott Klum \cite sklum
+ */
+class RetainTransform : public UntrainableTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(QStringList keys READ get_keys WRITE set_keys RESET reset_keys STORED false)
+    BR_PROPERTY(QStringList, keys, QStringList())
+
+    void project(const Template &src, Template &dst) const
+    {
+        dst = src;
+        foreach(const QString& localKey, dst.file.localKeys()) {
+            if (!keys.contains(localKey)) dst.file.remove(localKey);
+        }
+    }
+};
+
+BR_REGISTER(Transform, RetainTransform)
+
+} // namespace br
+
+#include "template.moc"
@@ -157,11 +157,11 @@ class MetadataDistance : public Distance
             if (aValue[0] == '(') /* Range */ {
                 QStringList values = aValue.split(',');
-                int age = values[0].mid(1).toInt();
+                int value = values[0].mid(1).toInt();
                 values[1].chop(1);
                 int upperBound = values[1].toInt();
-                while (age <= upperBound) {
+                while (value <= upperBound) {
                     if (aValue == bValue) {
                         keep = true;
                         break;