Convenience function to go between matrix and point list

Brendan Klare
1 parent 9bf09e4d
Showing 2 changed files with 15 additions and 1 deletions
openbr/core/eigenutils.cpp
openbr/core/eigenutils.h
@@ -95,3 +95,12 @@ MatrixXf removeRowCol(MatrixXf X, int row, int col) {
     }
     return Y;
 }
+
+MatrixXf pointsToMatrix(QList<QPointF> points) {
+    MatrixXf P(points.size(), 2);
+    for (int i = 0; i < points.size(); i++) {
+        P(i, 0) = points[i].x();
+        P(i, 1) = points[i].y();
+    }
+    return P;
+}
@@ -29,9 +29,13 @@ void printEigen(Eigen::MatrixXd X);
 void printEigen(Eigen::MatrixXf X);
 void printSize(Eigen::MatrixXf X);
-//Remove row and column from the matrix
+//Remove row and column from the matrix:
 Eigen::MatrixXf removeRowCol(Eigen::MatrixXf X, int row, int col);
+//Convert a point list into a matrix:
+Eigen::MatrixXf pointsToMatrix(QList<QPointF> points);
+
+
 template<typename _Scalar, int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>
 inline QDataStream &operator<<(QDataStream &stream, const Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > &mat)
 {
@@ -121,4 +125,5 @@ Eigen::MatrixBase&lt;T&gt; eigStd(const Eigen::MatrixBase&lt;T&gt;&amp; x,int dim)
     }
     qFatal("A matrix can only have two dimensions");
 }
+
 #endif // EIGENUTILS_H