MultiGrid2D.h

// =======================================================================
// @(#)MultiGrid2D.h        1.5 05/05/00
//
// Class:       TMultiGrid2D
// Purpose:     store all data for a multi grid method
//
// Author:      Gunar Matthies 02.11.1998
//
// History:     02.11.1998 start of implementation
//
// =======================================================================
#ifdef __2D__

#ifndef __MULTIGRID2D__
#define __MULTIGRID2D__

#include <MGLevel2D.h>

// #define MAXN_LEVELS 100

class TMultiGrid2D
{
  protected:
    int N_Levels;

    int N_Problems;

    int N_Parameters;

    double *Parameters;

    TMGLevel2D *MultiGridLevels[MAXN_LEVELS];

    TFESpace2D *FESpaces[MAXN_LEVELS];

    double **FunctionVectors[MAXN_LEVELS];

    double **RhsVectors[MAXN_LEVELS];

    double **AuxVectors[MAXN_LEVELS];

    int mg_recursions[MAXN_LEVELS];

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

    int GetN_Levels()
    { return N_Levels; }

    void AddLevel(TMGLevel2D *MGLevel);

    void ReplaceLevel(int i,TMGLevel2D *MGLevel);

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

    void RestrictToAllGrids();

    void SetDirichletNodes(int i);

    void Cycle(int i, double &res);

    void SetRecursion(int levels);
};

#endif

#endif // #ifdef __2D__