dense_matrix.hpp

Go to the documentation of this file.

#ifndef PARGEMSLR_DENSE_MATRIX_H
#define PARGEMSLR_DENSE_MATRIX_H
 
#include <iostream>
 
#include "../utils/utils.hpp"
#include "../vectors/sequential_vector.hpp"
#include "matrix.hpp"
 
namespace pargemslr
{
   
   template <typename T>
   class DenseMatrixClass: public MatrixClass<T>
   {
   private:
      
      /* variables */
      
      SequentialVectorClass<T>   _data_vec;
      
      int                        _nrows;
      
      int                        _ncols;
      
      int                        _ldim;
      
      
   public:
      
      DenseMatrixClass();
      
      DenseMatrixClass(const DenseMatrixClass<T> &mat);
      
      DenseMatrixClass(DenseMatrixClass<T> &&mat);
      
      DenseMatrixClass<T>& operator= (const DenseMatrixClass<T> &mat);
      
      DenseMatrixClass<T>& operator= (DenseMatrixClass<T> &&mat);
      
      virtual ~DenseMatrixClass();
      
      int            Setup(int nrows, int ncols);
      
      int            Setup(int nrows, int ncols, bool setzero);
      
      int            Setup(int nrows, int ncols, int location, bool setzero);
      
      int            SetupPtr( T* data, int nrows, int ncols, int ldim, int location);
      
      int            SetupPtr(const DenseMatrixClass<T> &mat_in, int row_start, int col_start, int num_rows, int num_cols);
      
      int            SubMatrix( int row_start, int col_start, int num_rows, int num_cols, int location, DenseMatrixClass<T> &mat_out);
      
      T&             operator()( int row, int col);
      
      virtual int    Clear();
      
      virtual int    GetDataLocation() const;
      
      virtual int    GetNumRowsLocal() const;
      
      virtual int    GetNumColsLocal() const;
      
      virtual long int  GetNumNonzeros() const;
      
      T*             GetData() const;
      
      SequentialVectorClass<T>& GetDataVector();
      
      const SequentialVectorClass<T>& GetDataVector() const;
      
      int            GetLeadingDimension() const;
      
      bool           IsHoldingData() const;
      
      virtual int    Eye();
      
      int            Rand();
      
      virtual int    Fill(const T &v);
      
      virtual int    MoveData( const int &location);
      
      virtual int    Scale(const T &alpha);
      
      virtual int    MatVec( char trans, const T &alpha, const VectorClass<T> &x, const T &beta, VectorClass<T> &y);
      
      virtual int    MatVec( char trans, const T &alpha, const VectorClass<T> &x, const T &beta, const VectorClass<T> &y, VectorClass<T> &z);
      
      int            MatMat( const T &alpha, const DenseMatrixClass<T> &A, char transa, const DenseMatrixClass<T> &B, char transb, const T &beta);
      
      int            Invert();
      
      int            InvertUpperTriangular();
      
      int            Hess( DenseMatrixClass<T> &Q);
      
      int            Hess( DenseMatrixClass<T> &Q, int start, int end);
      
      int            HessSchur( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi);
      
      int            HessSchur( DenseMatrixClass<T> &Q, int start, int end, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi);
      
      int            HessSchur( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &w);
      
      int            HessSchur( DenseMatrixClass<T> &Q, int start, int end, SequentialVectorClass<T> &w);
      
      int            Schur( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi);
      
      int            Schur( DenseMatrixClass<T> &Q, int start, int end, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi);
      
      int            Schur( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &w);
      
      int            Schur( DenseMatrixClass<T> &Q, int start, int end, SequentialVectorClass<T> &w);
      
      int            HessEig( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi);
      
      int            HessEig( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &w);
      
      int            Eig( DenseMatrixClass<T> &QS, DenseMatrixClass<T> &QE, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi);
      
      int            Eig( DenseMatrixClass<T> &QS, DenseMatrixClass<T> &QE, SequentialVectorClass<T> &w);
      
      int            OrdSchur( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi, vector_int &select);
      
      int            OrdSchur( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &w, vector_int &select);
      
      int            OrdSchurClusters( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &wr, SequentialVectorClass<T> &wi, vector_int &clusters);
      
      int            OrdSchurClusters( DenseMatrixClass<T> &Q, SequentialVectorClass<T> &w, vector_int &clusters);
      
      int            Plot( int conditiona, int conditionb, int width);
      
   };
   
   typedef DenseMatrixClass<float>     matrix_dense_float;
   typedef DenseMatrixClass<double>    matrix_dense_double;
   typedef DenseMatrixClass<complexs>  matrix_dense_complexs;
   typedef DenseMatrixClass<complexd>  matrix_dense_complexd;
   template<> struct PargemslrIsComplex<DenseMatrixClass<complexs> > : public std::true_type {};
   template<> struct PargemslrIsComplex<DenseMatrixClass<complexd> > : public std::true_type {};
   
}
 
#endif