Converted printEigen to a qDebug operator

Scott Klum
1 parent 80f91a7c
Showing 2 changed files with 27 additions and 21 deletions
openbr/core/eigenutils.cpp
openbr/core/eigenutils.h
@@ -45,25 +45,6 @@ void EigenUtils::writeEigen(VectorXf X, QString filename) {
     format->write(br::Template(m));
 }
-void EigenUtils::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;
-    }
-}
-void EigenUtils::printEigen(Eigen::MatrixXf 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;
-    }
-}
-
 void EigenUtils::printSize(Eigen::MatrixXf X) {
     qDebug() << "Rows=" << X.rows() << "\tCols=" << X.cols();
 }
@@ -18,6 +18,7 @@
 #define EIGENUTILS_H
 #include <QDataStream>
+#include <QDebug>
 #include <Eigen/Core>
 #include <assert.h>
@@ -25,12 +26,29 @@
 namespace EigenUtils
 {
+    template<typename _Scalar, int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>
+    QString matrixToString(const Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > &mat)
+    {
+        QString result;
+        if (mat.rows() > 1) result += "{ ";
+        for (int r=0; r<mat.rows(); r++) {
+            if ((mat.rows() > 1) && (r > 0)) result += "  ";
+            if (mat.cols() > 1) result += "[";
+            for (int c=0; c<mat.cols(); c++) {
+                result += QString::number(mat(r, c));
+                if (c < mat.cols() - 1) result += ", ";
+            }
+            if (mat.cols() > 1) result += "]";
+            if (r < mat.rows()-1) result += "\n";
+        }
+        if (mat.rows() > 1) result += " }";
+        return result;
+    }
+
     void writeEigen(Eigen::MatrixXf X, QString filename);
     void writeEigen(Eigen::MatrixXd X, QString filename);
     void writeEigen(Eigen::VectorXd X, QString filename);
     void writeEigen(Eigen::VectorXf X, QString filename);
-    void printEigen(Eigen::MatrixXd X);
-    void printEigen(Eigen::MatrixXf X);
     void printSize(Eigen::MatrixXf X);
     // Converts x y points in a single vector to two column matrix
@@ -52,6 +70,13 @@ namespace EigenUtils
 }
 template<typename _Scalar, int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>
+inline QDebug operator<<(QDebug dbg, const Eigen::Matrix< _Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols > &mat)
+{
+    dbg.nospace() << EigenUtils::matrixToString(mat);
+    return dbg.space();
+}
+
+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)
 {
     int r = mat.rows();