NSE3DMGLevel4.h

// =======================================================================
// @(#)NSE3DMGLevel4.h        1.6 07/03/00
//
// Class:       TNSE3DMGLevel4
// 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, B1T, B2T, B3T)
//
// Author:      Volker John 25.07.2000
//
// History:     25.07.2000 start of implementation
//
// =======================================================================

#ifndef __NSE3DMGLEVEL4__
#define __NSE3DMGLEVEL4__

#include <NSE3DMGLevel.h>

class TNSE3DMGLevel4 : public TNSE3DMGLevel
{
  protected:
    TSquareMatrix3D *A11;

    TSquareMatrix3D *A12;

    TSquareMatrix3D *A13;

    TSquareMatrix3D *A21;

    TSquareMatrix3D *A22;

    TSquareMatrix3D *A23;

    TSquareMatrix3D *A31;

    TSquareMatrix3D *A32;

    TSquareMatrix3D *A33;

    TSquareStructure3D *StructureA;

    int *ARowPtr;

    int *AKCol;

    double *A11Entries;

    double *A12Entries;

    double *A13Entries;

    double *A21Entries;

    double *A22Entries;

    double *A23Entries;

    double *A31Entries;

    double *A32Entries;

    double *A33Entries;

    TMatrix3D *B1;

    TMatrix3D *B2;

    TMatrix3D *B3;

    TStructure3D *StructureB;

    int *BRowPtr;

    int *BKCol;

    double *B1Entries;

    double *B2Entries;

    double *B3Entries;

    TMatrix3D *B1T;

    TMatrix3D *B2T;

    TMatrix3D *B3T;

    TStructure3D *StructureBT;

    int *BTRowPtr;

    int *BTKCol;

    double *B1TEntries;

    double *B2TEntries;
 
    double *B3TEntries;

  public:
    TNSE3DMGLevel4(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 *f2, double *f3, double *g,
                   double *u1, double *u2, double *u3, double *p,
                   int n_aux, double al, int VelocitySpace, 
                   int PressureSpace, TCollection *coll);

    ~TNSE3DMGLevel4();

    virtual void Defect(double *u1, double *u2, double *u3, double *p,
                        double *f1, double *f2, double *f3, double *g,
                        double *d1, double *d2, double *d3, double *d4,
                        double &res);

    virtual void CorrectNodes(double *u1, double *u2, double *u3, double *p);

    virtual void CellVanka(double *u1, double *u2,  double *u3, double *p,
        double *rhs1, double *rhs2, double *rhs3, double *rhs4,
        double *def1, double *def2, double *def3, double *def4,
        double *aux, double *Counters,
        int N_Parameters, double *Parameters, int smoother, int N_Levels);

    virtual void NodalVanka(double *u1, double *u2, double *u3, double *p,
        double *rhs1, double *rhs2, double *rhs3, double *rhs4,
        double *def1, double *def2, double *def3, double *def4,
        double *aux, double *Counters,
        int N_Parameters, double *Parameters, int smoother, int N_Levels);

    virtual void SolveExact(double *u1, double *u2, double *u3, double *p,
        double *rhs1, double *rhs2, double *rhs3, double *rhs4,
        double *aux, double *Counters,
        int N_Parameters, double *Parameters);

    virtual void BraessSarazin(double *u1, double *u2, double *u3, double *p,
        double *rhs1, double *rhs2, double *rhs3, double *rhs4,
        double *aux, double *Counters,
        int N_Parameters, double *Parameters,int N_Levels);

    virtual double StepLengthControl(double *u1, double *u2, double *u3, double *p,
           double *u1old, double *u2old, double *u3old, double *pold,
           double *def1, double *def2, double *def3, double *def4,
           int N_Parameters, double *Parameter);

    virtual void PrintAll();

};

#endif