MatrixUtils.hpp

Go to the documentation of this file.

 
#pragma once
#include "Headers/mytypes.hpp"
#include "Headers/mydebug.hpp"
#include <Eigen/Core>
#include <Eigen/Dense>
#include <Eigen/Sparse>
 
namespace MatrixUtils {
    template<class DenseMatrix>
    class EigenSolver;
 
    template<typename _Scalar, int _Rows, int _Cols, int _Options, int _MaxRows, int _MaxCols>
    class EigenSolver< Eigen::Matrix<_Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols> > {
        private:
            using Matrix_t = Eigen::Matrix<_Scalar, _Rows, _Cols, _Options, _MaxRows, _MaxCols>;
 
            Matrix_t                                m_inMatrix;
            Eigen::SelfAdjointEigenSolver<Matrix_t> m_solver;
            Eigen::ComputationInfo                  m_info   = Eigen::InvalidInput;
            Eigen::DecompositionOptions             m_option = Eigen::ComputeEigenvectors;
 
        public:
            EigenSolver()                   = delete;
            EigenSolver(EigenSolver const&) = delete;
            EigenSolver(EigenSolver&&)      = delete;
            EigenSolver(int dim) : m_inMatrix(dim, dim) {}
 
            EigenSolver(Matrix_t const& mat) = delete;
            EigenSolver(Matrix_t&& mat) : m_inMatrix{std::move(mat)} { this->compute(); }
 
            EigenSolver& compute() {
                m_solver.compute(m_inMatrix, m_option);
                m_info = m_solver.info();
                return *this;
            }
 
            EigenSolver& setOpt(Eigen::DecompositionOptions option) {
                m_option = option;
                return *this;
            }
 
            auto eigenvalues() -> decltype(m_solver.eigenvalues()) const& {
                return m_solver.eigenvalues();
            }
 
            auto eigenvectors() -> decltype(m_solver.eigenvectors()) const& {
                return m_solver.eigenvectors();
            }
 
            Matrix_t& inMatrix() { return m_inMatrix; }
    };
}  // namespace MatrixUtils
 
#ifdef GPU
    #include "MatrixUtils.cuh"
#endif