coo_matrix.hpp

Go to the documentation of this file.

#ifndef PARGEMSLR_COO_MATRIX_H
#define PARGEMSLR_COO_MATRIX_H
 
#include <iostream>
 
#include "../utils/utils.hpp"
#include "../vectors/sequential_vector.hpp"
#include "matrix.hpp"
#include "csr_matrix.hpp"
 
namespace pargemslr
{
   
   template <typename T>
   class CooMatrixClass: public MatrixClass<T>
   {
   private:
      
      /* variables */
      
      IntVectorClass<int>        _i_vec;
      
      IntVectorClass<int>        _j_vec;
      
      SequentialVectorClass<T>   _a_vec;
      
      int                        _nrows;
      
      int                        _ncols;
      
      int                        _nnz;
      
   public:
      
      CooMatrixClass();
      
      CooMatrixClass(const CooMatrixClass<T> &mat);
      
      CooMatrixClass(CooMatrixClass<T> &&mat);
      
      CooMatrixClass<T>& operator= (const CooMatrixClass<T> &mat);
      
      CooMatrixClass<T>& operator= (CooMatrixClass<T> &&mat);
      
      virtual int       Clear();
      
      virtual ~CooMatrixClass();
      
      int               Setup(int nrows, int ncols);
      
      int               Setup(int nrows, int ncols, int reserve);
      
      int               Setup(int nrows, int ncols, int reserve, int location);
      
      virtual int       GetDataLocation() const;
      
      virtual int       GetNumRowsLocal() const;
      
      virtual int       GetNumColsLocal() const;
      
      virtual long int  GetNumNonzeros() const;
      
      int               SetNumCols(int cols);
      
      int               SetNumNonzeros();
      
      virtual int       Eye();
      
      virtual int       Fill(const T &v);
      
      virtual int       Scale(const T &alpha);
      
      virtual int       MoveData( const int &location);
      
      int               PushBack( int row, int col, T v);
      
      int*              GetI() const;
      
      int*              GetJ() const;
      
      T*                GetData() const;
      
      IntVectorClass<int>&          GetIVector();
      
      const IntVectorClass<int>&          GetIVector() const;
      
      IntVectorClass<int>&          GetJVector();
      
      const IntVectorClass<int>&          GetJVector() const;
      
      SequentialVectorClass<T>&     GetDataVector();
      
      const SequentialVectorClass<T>&     GetDataVector() const;
      
      int               ToCsr(int location, CsrMatrixClass<T> &csrmat_out);
      
      int               Laplacian(int nx, int ny, int nz, T alphax, T alphay, T alphaz, T shift, bool rand_perturb = false);
      
      int               Helmholtz(int n, T w);
      
      int               ReadFromMMFile(const char *matfile, int idxin);
      
      int            PlotPatternGnuPlot( const char *datafilename, int *rperm, int *cperm, int conditiona, int conditionb);
      
   };
   
   typedef CooMatrixClass<float>     matrix_coo_float;
   typedef CooMatrixClass<double>    matrix_coo_double;
   typedef CooMatrixClass<complexs>  matrix_coo_complexs;
   typedef CooMatrixClass<complexd>  matrix_coo_complexd;
   template<> struct PargemslrIsComplex<CooMatrixClass<complexs> > : public std::true_type {};
   template<> struct PargemslrIsComplex<CooMatrixClass<complexd> > : public std::true_type {};
   
}
 
#endif