Eigen helper functions

Brendan Klare
1 parent e0460ba3
Showing 2 changed files with 31 additions and 0 deletions
openbr/core/eigenutils.cpp
openbr/core/eigenutils.h
@@ -16,4 +16,32 @@ void writeEigen(MatrixXf X, QString filename) {
     format->write(br::Template(m));
 }
+void writeEigen(MatrixXd X, QString filename) {
+    Mat m(X.rows(),X.cols(),CV_32FC1);
+    for (int i = 0; i < X.rows(); i++) {
+        for (int j = 0; j < X.cols(); j++) {
+            m.at<float>(i,j) = (float)X(i,j);
+        }
+    }
+    QScopedPointer<br::Format> format(br::Factory<br::Format>::make(filename));
+    format->write(br::Template(m));
+}
+void writeEigen(VectorXd X, QString filename) {
+    Mat m(X.size(),1,CV_32FC1);
+    for (int i = 0; i < X.rows(); i++) {
+        m.at<float>(i,0) = (float)X(i);
+    }
+    QScopedPointer<br::Format> format(br::Factory<br::Format>::make(filename));
+    format->write(br::Template(m));
+}
+
+void printEigen(Eigen::MatrixXd X) {
+    for (int i = 0; i < X.rows(); i++) {
+        QString str;
+        for (int j = 0; j < X.cols(); j++) {
+            str.append(QString::number(X(i,j)) + " ");
+        }
+        qDebug() << str;
+    }
+}
@@ -22,6 +22,9 @@
 #include <assert.h>
 void writeEigen(Eigen::MatrixXf X, QString filename);
+void writeEigen(Eigen::MatrixXd X, QString filename);
+void writeEigen(Eigen::VectorXd X, QString filename);
+void printEigen(Eigen::MatrixXd X);
 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)