Add ApplyMaskTransform to apply a mask in the template metadata

Austin Van Blanton
1 parent f16ec251
Showing 1 changed file with 20 additions and 0 deletions
openbr/plugins/mask.cpp
@@ -180,6 +180,26 @@ class LargestConvexAreaTransform : public UntrainableTransform
  
 BR_REGISTER(Transform, LargestConvexAreaTransform)
  
+/*!
+ * \ingroup transforms
+ * \brief Applies a mask from the metadata.
+ * \author Austin Blanton \cite imaus10
+ */
+class ApplyMaskTransform : public UntrainableTransform
+{
+    Q_OBJECT
+
+    void project(const Template &src, Template &dst) const
+    {
+        if (src.file.contains("Mask"))
+            src.m().copyTo(dst, src.file.get<Mat>("Mask"));
+        else
+            dst = src;
+    }
+};
+
+BR_REGISTER(Transform, ApplyMaskTransform)
+
 } // namespace br
  
 #include "mask.moc"