added RemoveTemplates transform

Josh Klontz
1 parent a0513c84
Showing 3 changed files with 25 additions and 0 deletions
openbr/plugins/gallery.cpp
openbr/plugins/meta.cpp
openbr/plugins/misc.cpp
@@ -117,6 +117,8 @@ class galGallery : public Gallery
     void write(const Template &t)
     {
+        if (t.isEmpty() && t.file.isNull())
+            return;
         stream << t;
     }
 };
@@ -15,6 +15,7 @@
  * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 #include <QFutureSynchronizer>
+#include <QRegularExpression>
 #include <QtConcurrentRun>
 #include "openbr_internal.h"
 #include "openbr/core/common.h"
@@ -423,6 +423,28 @@ class RelabelTransform : public UntrainableMetaTransform
 BR_REGISTER(Transform, RelabelTransform)
+/*!
+ * \ingroup transforms
+ * \brief Remove templates with the specified file extension.
+ * \author Josh Klontz \cite jklontz
+ */
+class RemoveTemplatesTransform : public UntrainableMetaTransform
+{
+    Q_OBJECT
+    Q_PROPERTY(QString regexp READ get_regexp WRITE set_regexp RESET reset_regexp STORED false)
+    BR_PROPERTY(QString, regexp, "")
+
+    void project(const Template &src, Template &dst) const
+    {
+        const QRegularExpression re(regexp);
+        const QRegularExpressionMatch match = re.match(src.file.suffix());
+        if (match.hasMatch()) dst = Template();
+        else                  dst = src;
+    }
+};
+
+BR_REGISTER(Transform, RemoveTemplatesTransform)
+
 }
 #include "misc.moc"