NSE_MGLevel4.h

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

#ifndef __NSE_MGLEVEL4__
#define __NSE_MGLEVEL4__

#include <NSE_MGLevel.h>

class TNSE_MGLevel4 : public TNSE_MGLevel
{
  protected:
#ifdef __2D__

    TSquareMatrix2D *A11;

    TSquareMatrix2D *A12;

    TSquareMatrix2D *A21;

    TSquareMatrix2D *A22;

    TSquareStructure2D *StructureA;

    TMatrix2D *B1;

    TMatrix2D *B2;

    TMatrix2D *B1T;

    TMatrix2D *B2T;

    TStructure2D *StructureB;

    TStructure2D *StructureBT;

    TStructure2D *StructureC;

    TMatrix2D *C;
#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;

    TMatrix3D *B1T;

    TMatrix3D *B2T;

    TMatrix3D *B3T;

    TStructure3D *StructureB;
 
    TStructure3D *StructureBT;

    TStructure3D *StructureC;

    TMatrix3D *C;
#endif  

    int *ARowPtr;

    int *AKCol;

    double *A11Entries;

    double *A12Entries;

    double *A21Entries;

    double *A22Entries;

    double *B1Entries;

    double *B2Entries;

    double *B1TEntries;

    double *B2TEntries;
#ifdef __3D__

    double *A13Entries;

    double *A23Entries;

    double *A31Entries;

    double *A32Entries;

    double *A33Entries;

    double *B3Entries;

    double *B3TEntries;

#endif  

    int *BRowPtr;

    int *BKCol;

    int *BTRowPtr;

    int *BTKCol;

    int *CRowPtr;

    int *CKCol;

    double *CEntries;
    
  public:
#ifdef __2D__
    TNSE_MGLevel4(int level, 
                  TSquareMatrix2D *A11, TSquareMatrix2D *A12, 
                  TSquareMatrix2D *A21, TSquareMatrix2D *A22, 
                  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_MGLevel4(int level,
                  TSquareMatrix2D *a11, TSquareMatrix2D *a12,
                  TSquareMatrix2D *a21, TSquareMatrix2D *a22,
                  TMatrix2D *b1, TMatrix2D *b2,
                  TMatrix2D *b1t, TMatrix2D *b2t,
                  TMatrix2D *c,
                  double *f1, double *u1,
                  int n_aux, double *al, int velocity_space,
                  int pressure_space, TCollection *Coll,
                  int *dw);
#endif  
#ifdef __3D__
    TNSE_MGLevel4(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,  
                  TMatrix3D *B1T, TMatrix3D *B2T, TMatrix3D *B3T,
                  double *f1, double *u1,
                  int n_aux, double *al, int VelocitySpace, 
                  int PressureSpace, TCollection *coll, int *dw);

TNSE_MGLevel4(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,
                               TMatrix3D *b1t, TMatrix3D *b2t, TMatrix3D *b3t,
                               TMatrix3D *c,
                               double *f1, double *u1,
                               int n_aux, double *al, int velocity_space,
                               int pressure_space, TCollection *Coll,
          int *dw);

#endif  

    ~TNSE_MGLevel4();

    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