SystemTNSE2D.h

#ifndef __SYSTEMTNSE2D__
#define __SYSTEMTNSE2D__

#include <SystemNSE2D.h>

class TSystemTNSE2D : public TSystemNSE2D
{
  protected:
    
    TSquareMatrix2D *SqmatrixM11, *SqmatrixM12, *SqmatrixM21, *SqmatrixM22;
  
    double *B;   
   
    double *defect;   
    
    double gamma;      

    TDiscreteForm2D *DiscreteFormRhs; 
    
    TAuxParam2D *NSE_Rhsaux;
    
    bool SystMatAssembled;
    
    double olderror_l_2_l_2u;
    
     
    
  public:
     TSystemTNSE2D(TFESpace2D *velocity_fespace, TFESpace2D *presssure_fespace, TFEVectFunct2D *Velocity, 
                      TFEFunction2D *p, double *sol, double *rhs, int disctype, int nsetype, int solver
#ifdef __PRIVATE__  
                     ,TFESpace2D *Projection_space, TFESpace2D *Stress_FeSpace,   TFESpace2D *Deformation_FeSpace
#endif    
                      );

    ~TSystemTNSE2D();

    void Init(CoeffFct2D *lincoeffs, BoundCondFunct2D *BoundCond, BoundValueFunct2D *U1BoundValue, BoundValueFunct2D *U2BoundValue,
              TAuxParam2D *aux, TAuxParam2D *nseaux_error);
    
    
    void Assemble(double *sol, double *rhs);
        
    void AssembleRhs(double *sol, double *rhs); 
    
    void AssembleSystMat(double scale, double *oldrhs, double *rhs, double *sol);

    void AssembleSystMatNonLinear();
     
    void RestoreMassMat();
        
    void AssembleANonLinear(double *sol, double *rhs);

    void  Solve(double *sol);
    
    void GetTNSEResidual(double *sol, double *res);   
 
    void MeasureTNSEErrors(DoubleFunct2D *ExactU1, DoubleFunct2D *ExactU2, DoubleFunct2D *ExactP, double *AllErrors);
};

#endif