SystemCD3D.h

#ifndef __SYSTEMCD3D__
#define __SYSTEMCD3D__

#include <SquareMatrix3D.h>
#include <ItMethod.h>
#include <AssembleMat3D.h>

#ifdef _MPI
//#include "mpi.h"
#include <ParFEMapper3D.h>
#include <ParFECommunicator3D.h>

#include <ParDirectSolver.h>
#endif

#ifdef _OMPONLY
#include <ParDirectSolver.h>
#endif

class TSystemCD3D
{
  protected:
#ifdef _MPI
    TParFEMapper3D **ParMapper;
    TParFECommunicator3D **ParComm;
    MPI_Comm Comm;
    TParDirectSolver *DS;
#endif
    
#ifdef _OMPONLY
    TParDirectSolver *DS;
#endif
  
    int N_Levels;
    
    int Start_Level;    
    
    TFESpace3D **FeSpaces, *fesp[1], *ferhs[1];
    
    BoundCondFunct3D *BoundaryConditions[1]; 
    
    BoundValueFunct3D *BoundaryValues[1];    
    
    int N_DOF;
    
    int Disctype;
    
    int SOLVER;
       
    int N_Matrices;

    TAssembleMat3D **AMatRhsAssemble;
    
    TSquareStructure3D **sqstructure;

    TSquareMatrix3D **sqmatrixA, *SQMATRICES[1];
    TSquareMatrix **sqmatrices;
        
    TDiscreteForm3D *DiscreteFormARhs;
    
    double **SolArray, **RhsArray, *RHSs[1];
    
    double Parameters[2], N_aux, *Itmethod_sol, *Itmethod_rhs;
    TMultiGrid3D *MG;
    TMGLevel3D *MGLevel;
    TItMethod *Itmethod, *prec;
   
  public:
  TSystemCD3D(int N_levels, TFESpace3D **fespaces, double **sol, double **rhs, int disctype, int solver);
    
    ~TSystemCD3D();
    
    void Init(CoeffFct3D *BilinearCoeffs, BoundCondFunct3D *BoundCond, BoundValueFunct3D *BoundValue,
              TAuxParam3D *aux);
 
    void Assemble();

    void  Solve(double *sol, double *rhs);
    
#ifdef _MPI
    TParFECommunicator3D *Get_ParComm(int level){
      return ParComm[level];
    }
#endif
};

#endif