MultiGrid3D.h

// =======================================================================
// %W% %G%
//
// Class:       TMultiGrid3D
// Purpose:     store all data for a multi grid method
//
// Author:      Gunar Matthies 26.06.2000
//
// History:     26.06.2000 start of implementation
//
// =======================================================================

#ifndef __MULTIGRID3D__
#define __MULTIGRID3D__

#include <MGLevel3D.h>
#ifdef _MPI   
   #ifdef __3D__
    #include <ParFECommunicator3D.h>
   #else
    #include <ParFECommunicator2D.h>
   #endif
#endif 
// #define MAXN_LEVELS 25

class TMultiGrid3D
{
  protected:
    int N_Levels;

    int N_Problems;

    int N_Parameters;

    double *Parameters;

    TMGLevel3D *MultiGridLevels[MAXN_LEVELS];

    TFESpace3D *FESpaces[MAXN_LEVELS];

    double **FunctionVectors[MAXN_LEVELS];

    double **RhsVectors[MAXN_LEVELS];

    double **AuxVectors[MAXN_LEVELS];

    int mg_recursions[MAXN_LEVELS];

  public:
    TMultiGrid3D(int n_problems, int n_parameters, double *parameters);

    int GetN_Levels()
    { return N_Levels; }

    void AddLevel(TMGLevel3D *MGLevel);

    void ReplaceLevel(int i,TMGLevel3D *MGLevel);

    TMGLevel3D *GetLevel(int i)
    { return MultiGridLevels[i]; }

    void RestrictToAllGrids();

    void SetDirichletNodes(int i);

    void Cycle(int i, double &res);

    void SetRecursion(int levels);
    
    void Smooth(int smoother_type, TMGLevel3D *Level, 
#ifdef _MPI
        TParFECommunicator3D *ParComm, 
#endif
        double &oldres);
};

#endif