NSE3DMGLevel2.h

// =======================================================================
// @(#)NSE3DMGLevel2.h        1.6 07/03/00
//
// Class:       TNSE3DMGLevel2
// 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 __NSE3DMGLEVEL2__
#define __NSE3DMGLEVEL2__

#include <NSE3DMGLevel.h>

class TNSE3DMGLevel2 : public TNSE3DMGLevel
{
  protected:
    TSquareMatrix3D *A;

    TSquareStructure3D *StructureA;

    int *ARowPtr;

    int *AKCol;

    double *AEntries;

    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:
    TNSE3DMGLevel2(int level, TSquareMatrix3D *A, 
                   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);

    ~TNSE3DMGLevel2();

    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