PetscLinearSolver.h

Go to the documentation of this file.

#ifndef PETSCLINEARSOLVER_H
#define PETSCLINEARSOLVER_H

#include "petscksp.h"
#include <iostream>
#include <string>
using namespace std;

#ifdef SINGLE
#define real float
#else
#define real double
#endif


class PetscLinearSolver {
  
 public:
  PetscLinearSolver();

  void LinearSolver(int matrixSize, 
                    real **valueAMatrix, 
                    int **columnAMatrix, 
                    int totalColumn, 
                    real *rhs ,
                    real *Sol,  
                    PCType preCond, 
                    KSPType KrylovM, 
                    real tol, 
                    int &zeroInitialGuess
                    );

  int finalize( );

  ~PetscLinearSolver();

 private: 
  int initialize(int matrixSize,PCType preCond, KSPType KrylovM, real tol, int &zeroInitialGuess);
  int setRHS(real *rhs);
  int setAMatrix(real **AMatrix);
  int setSparseAMatrix(real **AMatrix, int **columnAMatrix, int totalColumn, int &zeroInitialGuess);
  int solve(int &zeroInitialGuess);
  void vecPrint(Vec x, int n);
  void getSol(real *Sol);
  
  PetscInt       n, its;
  PCType         preConditioner;
  KSPType        KrylovMethod;

  Vec            x, b;      // approx solution, RHS
  Mat            A;         // linear system matrix

  KSP            ksp;       // linear solver context
  PC             pc;        // preconditioner context
  PetscErrorCode ierr;
  real           KSPTolerance;
};

#endif

---