NSE_MGLevel2.h

// =======================================================================
// @(#)NSE_MGLevel2.h        1.6 07/03/00
//
// Class:       TNSE_MGLevel2
// Purpose:     store all data for one level in a multi grid method
//              for solving a Stokes-/ Navier-Stokes system of
//              type 2 (A, B1, B2, B3, B1T, B2T, B3T)
//
// Author:      Volker John 25.07.2000
//
// History:     25.07.2000 start of implementation
//
// =======================================================================

#ifndef __NSE_MGLEVEL2__
#define __NSE_MGLEVEL2__

#include <NSE_MGLevel.h>

class TNSE_MGLevel2 : public TNSE_MGLevel
{
  protected:
#ifdef __2D__

    TSquareMatrix2D *A;

    TSquareStructure2D *StructureA;

    TStructure2D *StructureB;

    TMatrix2D *B1;

    TMatrix2D *B2;

    TStructure2D *StructureBT;

    TMatrix2D *B1T;

    TMatrix2D *B2T;

    TStructure2D *StructureC;

    TMatrix2D *C;
#endif  

#ifdef __3D__

    TSquareMatrix3D *A;

    TSquareStructure3D *StructureA;

    TStructure3D *StructureB;

    TMatrix3D *B1;

    TMatrix3D *B2;

    TMatrix3D *B3;

    TStructure3D *StructureBT;

    TMatrix3D *B1T;

    TMatrix3D *B2T;

    TMatrix3D *B3T;

    TStructure3D *StructureC;

    TMatrix3D *C;
#endif  

    int *ARowPtr;

    int *AKCol;

    double *AEntries;

    int *BRowPtr;

    int *BKCol;

    double *B1Entries;

    double *B2Entries;

#ifdef __3D__

    double *B3Entries;
#endif  

    int *BTRowPtr;

    int *BTKCol;

    double *B1TEntries;

    double *B2TEntries;

#ifdef __3D__

    double *B3TEntries;
#endif  

    int *CRowPtr;

    int *CKCol;

    double *CEntries;

  public:
#ifdef __2D__
    TNSE_MGLevel2(int level, TSquareMatrix2D *A, 
                  TMatrix2D *B1, TMatrix2D *B2,
                  TMatrix2D *B1T, TMatrix2D *B2T,
                  double *f1, double *u1, 
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll,
                  int *dw);

    TNSE_MGLevel2(int level, TSquareMatrix2D *A, 
                  TMatrix2D *B1, TMatrix2D *B2,
                  TMatrix2D *B1T, TMatrix2D *B2T,
                  TMatrix2D *C,
                  double *f1, double *u1, 
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll,
                  int *dw);
#endif  

#ifdef __3D__
    TNSE_MGLevel2(int level, TSquareMatrix3D *A, 
                  TMatrix3D *B1, TMatrix3D *B2, TMatrix3D *B3,
                  TMatrix3D *B1T, TMatrix3D *B2T, TMatrix3D *B3T,
                  double *f1, double *u1, 
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll,
                  int *dw);

    TNSE_MGLevel2(int level, TSquareMatrix3D *A, 
                  TMatrix3D *B1, TMatrix3D *B2, TMatrix3D *B3,
                  TMatrix3D *B1T, TMatrix3D *B2T, TMatrix3D *B3T,
                  TMatrix3D *C,
                  double *f1, double *u1, 
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll,
                  int *dw);
#endif  

    ~TNSE_MGLevel2();

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

};

#endif