NSE_MultiGrid.h

// =======================================================================
// @(#)NSE_MultiGrid.h        1.4 02/08/00
//
// Class:       TNSE_MultiGrid
// Purpose:     store all data for a multi grid method for 
//              Stokes or Navier-Stokes problems
//
// Author:      Volker John 25.07.2000
//
// History:     25.07.2000 start of implementation
//
// =======================================================================

#ifndef __NSE_MULTIGRID__
#define __NSE_MULTIGRID__

#include <NSE_MGLevel.h>

class TNSE_MultiGrid
{
  protected:
    int N_Levels;

    int N_Problems;

    int N_Parameters;

    double *Parameters;

    TNSE_MGLevel *MultiGridLevels[MAXN_LEVELS];

#ifdef __2D__

    TFESpace2D *USpaces[MAXN_LEVELS];

    TFESpace2D *PSpaces[MAXN_LEVELS];
#endif  
#ifdef __3D__

    TFESpace3D *USpaces[MAXN_LEVELS];

    TFESpace3D *PSpaces[MAXN_LEVELS];
#endif  

    double **U1Vectors[MAXN_LEVELS];

    double **U2Vectors[MAXN_LEVELS];

#ifdef __3D__

    double **U3Vectors[MAXN_LEVELS];
#endif  

    double **PVectors[MAXN_LEVELS];

    double **Rhs1Vectors[MAXN_LEVELS];

    double **Rhs2Vectors[MAXN_LEVELS];

#ifdef __3D__

    double **Rhs3Vectors[MAXN_LEVELS];
#endif  

    double **RhsPVectors[MAXN_LEVELS];

    double **AuxVectors[MAXN_LEVELS];

    int mg_recursions[MAXN_LEVELS];

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

    int GetN_Levels()
    { return N_Levels; }

    void AddLevel(TNSE_MGLevel *MGLevel);

    TNSE_MGLevel *ReplaceLevel(int i, TNSE_MGLevel *MGLevel);

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

    void RestrictToAllGrids();

    void SetDirichletNodes(int i);

    double GetResidual(int i);

    void Cycle(int i, double &res);

    void SetRecursion(int levels);

    double GetParam(int i);

    void SetParam(int i, double a);

};

#endif