NSE_MGLevel5.h

// =======================================================================
// @(#)NSE_MGLevel5.h        1.1 07/03/03
//
// Class:       TNSE_MGLevel5
// Purpose:     store all data for one level in a multi grid method
//              for solving a Stokes-/ Navier-Stokes system of
//              type 5 (A)
//
// Author:      Gunar Matthies (28.10.2003)
//
// History:     28.10.2003 start of implementation
//
// =======================================================================

#ifndef __NSE_MGLEVEL5__
#define __NSE_MGLEVEL5__

#include <NSE_MGLevel.h>

#ifdef __2D__
#include <SquareMatrixNSE2D.h>
#include <StructureNSE2D.h>
#endif

#ifdef __3D__
#include <SquareMatrixNSE3D.h>
#include <StructureNSE3D.h>
#endif

class TNSE_MGLevel5 : public TNSE_MGLevel
{
  protected:
#ifdef __2D__

    TSquareMatrixNSE2D *A;

    TStructureNSE2D *StructureA;
#endif  

#ifdef __3D__

    TSquareMatrixNSE3D *A;

    TStructureNSE3D *StructureA;
#endif  

    int *BeginJb;
    int *jb;
    int N_DOFperJoint;
    double *Alpha;

    int N_Cells;

    int *ARowPtr;

    int *AKCol;

    double *AEntries;

  public:
#ifdef __2D__
    TNSE_MGLevel5(int level, TSquareMatrixNSE2D *A, 
                  double *f1, double *u1,
                  int n_aux, double *al,
                  int VelocitySpace, int PressureSpace,
                  TFESpace2D *PSpace, TCollection *coll);
#endif  

#ifdef __3D__
    TNSE_MGLevel5(int level, TSquareMatrixNSE3D *A, 
                  double *f1, double *u1,
                  int n_aux, double *al,
                  int VelocitySpace, int PressureSpace,
                  TFESpace3D *PSpace, TCollection *coll);
#endif  

    ~TNSE_MGLevel5();

    virtual void Defect(double *u1, double *f1, double *d1, double &res);

    virtual void CorrectNodes(double *u1);

    virtual void  CellVanka(double *u1, double *rhs1, double *aux,
                            int N_Parameters, double *Parameters, 
                            int smoother, int N_Levels);

    virtual void NodalVanka(double *u1, double *rhs1, double *aux,
        int N_Parameters, double *Parameters, int smoother, int N_Levels);

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

    virtual void SolveExactUMFPACK(double *u1, double *rhs1, int &umfpack_flag);

    virtual void BraessSarazin(double *u1, double *rhs1, double *aux,
        int N_Parameters, double *Parameters,int N_Levels);

    virtual double  StepLengthControl(double *u1, double *u1old, double *def1, 
           int N_Parameters, double *Parameter);

    virtual void PrintAll();

    void GetUstarFromU(double *u, double *ustar);

    void GetUFromUstar(double *ustar, double *u);

    void GetDstarFromD(double *d, double *dstar);

    void GetDFromDstar(double *dstar, double *d);
};

#endif