preconditioner_multicolored.hpp

Go to the documentation of this file.

#ifndef PARALUTION_PRECONDITIONER_MULTICOLORED_HPP_
#define PARALUTION_PRECONDITIONER_MULTICOLORED_HPP_

#include "../solver.hpp"
#include "preconditioner.hpp"
#include "../../base/local_vector.hpp"

#include <vector>

namespace paralution {

template <class OperatorType, class VectorType, typename ValueType>
class MultiColored : public Preconditioner<OperatorType, VectorType, ValueType> {

public:

  MultiColored();
  virtual ~MultiColored();

  virtual void Clear(void);  

  virtual void Build(void);

  virtual void SetPrecondMatrixFormat(const unsigned int mat_format);

  virtual void SetDecomposition(const bool decomp);

  virtual void Solve(const VectorType &rhs,
                     VectorType *x);

protected:

  OperatorType *analyzer_op_;
  OperatorType *preconditioner_; 

  OperatorType ***preconditioner_block_; 

  VectorType **x_block_; 
  VectorType **diag_block_; 
  VectorType x_;
  VectorType diag_;

  Solver<OperatorType, VectorType, ValueType> **diag_solver_;

  int num_blocks_;
  int *block_sizes_;

  bool op_mat_format_; 
  unsigned int precond_mat_format_;

  bool decomp_;

  virtual void ExtractRHSinX_(const VectorType &rhs,
                              VectorType *x);

  virtual void SolveL_(void) = 0;
  virtual void SolveD_(void) = 0;
  virtual void SolveR_(void) = 0;

  virtual void Solve_(const VectorType &rhs,
                      VectorType *x) = 0;

  virtual void InsertSolution_(VectorType *x);

  
  virtual void Build_Analyser_(void); 
  virtual void Analyse_(void);
  virtual void Permute_(void);
  virtual void Factorize_(void);
  virtual void Decompose_(void);
  virtual void PostAnalyse_(void);

  virtual void MoveToHostLocalData_(void);
  virtual void MoveToAcceleratorLocalData_(void) ;

};


}

#endif // PARALUTION_PRECONDITIONER_MULTICOLORED_HPP_