Add File::setList to automatically convert QLists to QVariantLists

Austin Blanton
1 parent c3fff4cf
Showing 3 changed files with 12 additions and 3 deletions
openbr/openbr_plugin.h
openbr/plugins/landmarks.cpp
openbr/plugins/slidingwindow.cpp
@@ -41,6 +41,7 @@
 #include <QVector>
 #include <opencv2/core/core.hpp>
 #include <openbr/openbr.h>
+#include <openbr/core/qtutils.h>
 /*!
  * \defgroup cpp_plugin_sdk C++ Plugin SDK
@@ -215,6 +216,14 @@ struct BR_EXPORT File
     static QVariant parse(const QString &value); /*!< \brief Try to convert the QString to a QPointF or QRectF if possible. */
     inline void set(const QString &key, const QVariant &value) { m_metadata.insert(key, value); } /*!< \brief Insert or overwrite the metadata key with the specified value. */
     void set(const QString &key, const QString &value); /*!< \brief Insert or overwrite the metadata key with the specified value. */
+
+    /*!< \brief Specialization for list type. Insert or overwrite the metadata key with the specified value. */
+    template <typename T>
+    void setList(const QString &key, const QList<T> &value)
+    {
+        set(key, QtUtils::toVariantList(value));
+    }
+
     inline void remove(const QString &key) { m_metadata.remove(key); } /*!< \brief Remove the metadata key. */
     /*!< \brief Returns a value for the key, throwing an error if the key does not exist. */
@@ -113,7 +113,7 @@ class ProcrustesTransform : public Transform
         // R(0,0), R(1,0), R(1,1), R(0,1), mean_x, mean_y, norm
         QList<float> procrustesStats;
         procrustesStats << R(0,0) << R(1,0) << R(1,1) << R(0,1) << mean[0] << mean[1] << norm;
-        dst.file.set("ProcrustesStats",QtUtils::toVariantList(procrustesStats));
+        dst.file.setList<float>("ProcrustesStats",procrustesStats);
         if (warp) {
             Eigen::MatrixXf dstMat = srcMat*R;
@@ -273,7 +273,7 @@ class DelaunayTransform : public UntrainableTransform
             dst.file.setRects(QList<QRectF>() << OpenCVUtils::fromRect(boundingBox));
         } else dst = src;
-        dst.file.set("DelaunayTriangles", QtUtils::toVariantList(validTriangles));
+        dst.file.setList<QPointF>("DelaunayTriangles", validTriangles);
     }
 };
@@ -120,7 +120,7 @@ private:
                         if (dst.file.contains("Confidences"))
                             confidences = dst.file.getList<float>("Confidences");
                         confidences.append(confidence);
-                        dst.file.set("Confidences", QtUtils::toVariantList(confidences));
+                        dst.file.setList<float>("Confidences", confidences);
                         if (takeLargestScale) return;
                     }
                 }