block_jacobi.hpp

Go to the documentation of this file.

#ifndef PARGEMSLR_BLOCK_JACOBI_H
#define PARGEMSLR_BLOCK_JACOBI_H
 
#include <iostream>
 
#include "../utils/utils.hpp"
#include "../vectors/int_vector.hpp"
#include "../vectors/sequential_vector.hpp"
#include "../vectors/parallel_vector.hpp"
#include "../matrices/csr_matrix.hpp"
#include "../matrices/parallel_csr_matrix.hpp"
#include "../solvers/solver.hpp"
 
#ifdef PARGEMSLR_CUDA
#include "cusparse.h"
#endif
 
namespace pargemslr
{
   template <class ParallelMatrixType, class ParallelVectorType, class LocalMatrixType, class LocalVectorType, typename DataType>
   class BlockJacobiClass: public SolverClass<ParallelMatrixType, ParallelVectorType, DataType>
   {
   private:
      
      SolverClass<LocalMatrixType, LocalVectorType, DataType>                 *_local_precond;
      
      int                                                                     _own_local_preconditioner;
      
      int                                                                     _location;
      
    public:
      BlockJacobiClass();
     
      virtual int Clear();
     
      virtual ~BlockJacobiClass();
      
      BlockJacobiClass(const BlockJacobiClass<ParallelMatrixType, ParallelVectorType, LocalMatrixType, LocalVectorType, DataType> &solver);
      
      BlockJacobiClass(BlockJacobiClass<ParallelMatrixType, ParallelVectorType, LocalMatrixType, LocalVectorType, DataType> &&solver);
      
      BlockJacobiClass<ParallelMatrixType, ParallelVectorType, LocalMatrixType, LocalVectorType, DataType>& operator= (const BlockJacobiClass<ParallelMatrixType, ParallelVectorType, LocalMatrixType, LocalVectorType, DataType> &solver);
      
      BlockJacobiClass<ParallelMatrixType, ParallelVectorType, LocalMatrixType, LocalVectorType, DataType>& operator= (BlockJacobiClass<ParallelMatrixType, ParallelVectorType, LocalMatrixType, LocalVectorType, DataType> &&solver);
      
      virtual int Setup( ParallelVectorType &x, ParallelVectorType &rhs);
      
      virtual int Solve( ParallelVectorType &x, ParallelVectorType &rhs);
      
      virtual long int  GetNumNonzeros();
      
      int         SetLocalPreconditioner(SolverClass<LocalMatrixType, LocalVectorType, DataType> &local_precond);
      
      int         SetLocalPreconditionerP(SolverClass<LocalMatrixType, LocalVectorType, DataType>* local_precond);
      
      SolverClass<LocalMatrixType, LocalVectorType, DataType>* GetLocalPreconditionerP();
      
      int         SetOwnLocalPreconditioner(bool own_local_preconditioner);
      
      virtual int SetSolveLocation( const int &location);
      
      virtual int MoveData( const int &location);
      
    };
   
   typedef BlockJacobiClass<matrix_csr_par_float, vector_par_float, matrix_csr_float, vector_seq_float, float>                precond_bj_csr_par_float;
   typedef BlockJacobiClass<matrix_csr_par_double, vector_par_double, matrix_csr_double, vector_seq_double, double>           precond_bj_csr_par_double;
   typedef BlockJacobiClass<matrix_csr_par_complexs, vector_par_complexs, matrix_csr_complexs, vector_seq_complexs, complexs> precond_bj_csr_par_complexs;
   typedef BlockJacobiClass<matrix_csr_par_complexd, vector_par_complexd, matrix_csr_complexd, vector_seq_complexd, complexd> precond_bj_csr_par_complexd;
   
}
 
#endif