MGLevel3D.h

// =======================================================================
// %W% %G%
//
// Class:       TMGLevel3D
// Purpose:     store all data for one level in a multi grid method in 3d
//
// Author:      Gunar Matthies 26.06.2000
//
// History:     26.06.2000 start of implementation
//
// =======================================================================

#ifndef __MGLEVEL3D__
#define __MGLEVEL3D__

#include <SquareMatrix3D.h>
#ifdef _MPI   
   #ifdef __3D__
    #include <ParFECommunicator3D.h>
   #else
    #include <ParFECommunicator2D.h>
   #endif
#endif 

// #include <omp.h>
class TMGLevel3D
{
  protected:
    int Level;

    TFESpace3D *FESpace;

    int *Permutation;

    int N_Active;

    int HangingNodeBound;

    int N_Dirichlet;

    int N_DOF;

    TSquareMatrix3D *A;

    TSquareStructure3D *MatrixStructure;

    int *RowPtr;

    int *KCol;

    double *Entries;

    double *Rhs;

    double *X;

    int N_Aux;

    double **Aux;

    double *Additional;

    void ILUDecomposition();
    
#ifdef _MPI

     int OwnN_DOF;  
     
     double *Temp_arr;
    
     TParFECommunicator3D *ParComm; 
     TParFEMapper3D *ParMapper;
     
     int *Reorder,N_Master,N_Int,N_Dept;
     
     int *Reorder_M,*Reorder_D1,*Reorder_D2,*Reorder_I;
     
     int N_InterfaceM, N_Dept1, N_Dept2;// N_Dept3;
     
#endif
     
#ifdef _HYBRID     

     int N_CMaster, N_CDept1, N_CDept2, N_CInt, *ptrCMaster, *ptrCDept1, *ptrCDept2, *ptrCInt;
#endif

  public:

    
    TMGLevel3D(int level, TSquareMatrix3D *A,
             double *rhs, double *sol, int n_aux,
             int *permutation);
    
#ifdef _MPI

    TMGLevel3D(int level, TSquareMatrix3D *a, double *rhs, double *sol, 
                       TParFECommunicator3D *parComm, TParFEMapper3D *parMapper, int n_aux,
                       int *permutation);
    double *GetTemp_arr()
    { return Temp_arr; }
#endif  

    ~TMGLevel3D();

    double *GetAuxVector(int i);

    double *GetSolution()
    { return X; }

    double *GetRhs()
    { return Rhs; }
    
    

    double **GetAuxVectors()
    { return Aux; }

    TSquareMatrix3D *GetMatrix()
    { return A; }

    int GetN_DOF()
    { return N_DOF; }

    int GetHangingNodeBound()
    { return HangingNodeBound; }

    int GetN_Dirichlet()
    { return N_Dirichlet; }

    void Defect(double *sol, double *f, double *d, double &res);

    void Update(double *sol, double *upd);

    void CorrectNodes(double *vect);

    void CorrectDefect(double *vect);

    void Reset(double *vect);

    TFESpace3D *GetFESpace()
    { return FESpace; }

    void ILU(double *sol, double *f, double *aux,
        int N_Parameters, double *Parameters);

    void SOR(double *sol, double *f, double *aux,
        int N_Parameters, double *Parameters);

    void SSOR(double *sol, double *f, double *aux,
        int N_Parameters, double *Parameters);

    void Jacobi(double *sol, double *f, double *aux,
        int N_Parameters, double *Parameters);
     

    
    void Block2x2(double *sol, double *f, double *aux,
        int N_Parameters, double *Parameters);

    void SolveExact(double *u1, double *rhs1);

    double StepLengthControl(double *u, 
                         double *uold, 
                         double *def,
                         int N_Parameters, 
                         double *Parameters);
    
    
        
#ifdef _MPI       

    TParFECommunicator3D *GetParComm()
      { return ParComm; }
      
       void SOR_Re(double *sol, double *f, double *aux,
        int N_Parameters, double *Parameters);   
    
#endif

#ifdef _HYBRID
        void SOR_Re_Color(double *sol, double *f, double *aux,
              int N_Parameters, double *Parameters, bool firstTime, bool lastTime);
    void SOR_Re_Color(double *sol, double *f, double *aux, int N_Parameters, double *Parameters,int smooth);
    
    void SOR_Re_Color_Coarse(double *sol, double *f, double *aux,
                 int N_Parameters, double *Parameters);
#endif
       
};

#endif