NSE_MGLevel3.h

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

#ifndef __NSE_MGLEVEL3__
#define __NSE_MGLEVEL3__

#include <NSE_MGLevel.h>

class TNSE_MGLevel3 : public TNSE_MGLevel
{
  protected:
#ifdef __2D__

    TSquareMatrix2D *A11;

    TSquareMatrix2D *A12;

    TSquareMatrix2D *A21;

    TSquareMatrix2D *A22;

    TSquareStructure2D *StructureA;

    TMatrix2D *B1;

    TMatrix2D *B2;

    TStructure2D *StructureB;

    TStructure2D *StructureBT;
#endif  

#ifdef __3D__

    TSquareMatrix3D *A11;

    TSquareMatrix3D *A12;

    TSquareMatrix3D *A13;

    TSquareMatrix3D *A21;

    TSquareMatrix3D *A22;

    TSquareMatrix3D *A23;

    TSquareMatrix3D *A31;

    TSquareMatrix3D *A32;

    TSquareMatrix3D *A33;

    TSquareStructure3D *StructureA;

    TMatrix3D *B1;

    TMatrix3D *B2;

    TMatrix3D *B3;

    TStructure3D *StructureB;
 
    TStructure3D *StructureBT;
#endif  

    int *ARowPtr;

    int *AKCol;

    double *A11Entries;

    double *A12Entries;

    double *A21Entries;

    double *A22Entries;

    double *B1Entries;

    double *B2Entries;

#ifdef __3D__

    double *A13Entries;

    double *A23Entries;

    double *A31Entries;

    double *A32Entries;

    double *A33Entries;

    double *B3Entries;
#endif  

    int *BRowPtr;

    int *BKCol;

    int *BTRowPtr;

    int *BTKCol;
    
  public:
#ifdef __2D__
    TNSE_MGLevel3(int level, 
                  TSquareMatrix2D *A11, TSquareMatrix2D *A12, 
                  TSquareMatrix2D *A21, TSquareMatrix2D *A22, 
                  TMatrix2D *B1, TMatrix2D *B2,
                  TStructure2D *structureBT, 
                  double *f1, double *u1,
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll,
                  int *dw);
#endif  
#ifdef __3D__
    TNSE_MGLevel3(int level, 
                  TSquareMatrix3D *A11, TSquareMatrix3D *A12, 
                  TSquareMatrix3D *A13, 
                  TSquareMatrix3D *A21, TSquareMatrix3D *A22, 
                  TSquareMatrix3D *A23, 
                  TSquareMatrix3D *A31, TSquareMatrix3D *A32, 
                  TSquareMatrix3D *A33, 
                  TMatrix3D *B1, TMatrix3D *B2, TMatrix3D *B3,  
                  TStructure3D *structureBT, 
                  double *f1, double *u1,
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll,
                  int *dw);
#endif  

    ~TNSE_MGLevel3();

    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