MGLevel2D.h

// =======================================================================
// @(#)MGLevel2D.h        1.3 08/24/99
//
// Class:       TMGLevel2D
// Purpose:     store all data for one level in a multi grid method
//
// Author:      Gunar Matthies 02.11.1998
//
// History:     02.11.1998 start of implementation
//
// =======================================================================

#ifndef __MGLEVEL2D__
#define __MGLEVEL2D__

#include <SquareMatrix2D.h>

class TMGLevel2D
{
  protected:
    int Level;

    TFESpace2D *FESpace;

    int *Permutation;

    int N_Active;

    int HangingNodeBound;

    int N_Dirichlet;

    int N_DOF;

    TSquareMatrix2D *A;

    TSquareStructure2D *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;   
    
     TParFECommunicator2D *ParComm;

     TFEFunction2D *C;     
     
     TFESpace2D *OwnScalarSpace;

     TFEFunction2D *OwnC;
 
     double *OwnSolArray;
#endif    
    
    
  public:
    TMGLevel2D(int level, TSquareMatrix2D *A,
               double *rhs, double *sol, int n_aux,
               int *permutation);

#ifdef _MPI   
     TMGLevel2D(int level, TSquareMatrix2D *a, double *rhs, double *sol, 
                TFEFunction2D *c, TParFECommunicator2D *parComm,TFESpace2D *ownScalarSpace, int n_aux,
                int *permutation);  
#endif    
    
    ~TMGLevel2D();

    double *GetAuxVector(int i);

    double *GetSolution()
    { return X; }

    double *GetRhs()
    { return Rhs; }

    double **GetAuxVectors()
    { return Aux; }

    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);

    TFESpace2D *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       
    TFESpace2D *GetOwnFESpace()
       { return OwnScalarSpace; }

    TParFECommunicator2D *GetParComm()
      { return ParComm; }

    double *GetOwnSolution()
      { return OwnSolArray; }  

     TFEFunction2D *GetFEFunction()
      { return C; }

     TFEFunction2D *GetOwnFEFunction()
      { return OwnC; }

    
#endif

};

#endif