SystemTNSE2D_ALE.h

#ifndef __SYSTEMTNSE2D_ALE__
#define __SYSTEMTNSE2D_ALE__

#include <SystemTNSE2D.h>

class TSystemTNSE2D_ALE : public TSystemTNSE2D
{
  protected:
    int N_GridDOFs, N_GridActive, *GridKCol, *GridRowPtr;
    
    double *MeshVelo, *gridpos, *gridpos_old, *gridpos_ref, *griddisp, *GridRhs, *Entries[4];   
    
    TFESpace2D *GridFESpace;    
    
    BoundValueFunct2D *GridBoundValue[1];
     
    TFEVectFunct2D *GridPos, *RefGridPos;
     
    TFEFunction2D *MeshVeloFct[2];    
    
    TDiscreteForm2D *DiscreteFormMARhs, *DiscreteFormGrid;  
    
//     /** M - mass/system mat for TNSE velocity component   */
//     TSquareMatrix2D *OldSqmatrixM11, *OldSqmatrixM12, *OldSqmatrixM21, *OldSqmatrixM22;

       
    TSquareMatrix2D *SqmatrixG11, *SqmatrixG12, *SqmatrixG21, *SqmatrixG22, *SQMATRICES_GRID[4];
 
    TSquareStructure2D *SquareStructureG;
    
    TAuxParam2D *Aux_ALE, *Meshaux;

    BoundCondFunct2D *GridBoundaryConditions[1];
    BoundValueFunct2D *GridBoundValues[1];
      
    ModifyMeshCoords *ModifyCoord;
    
    ModifyBoundCoords *ModifyBoudary;   
    
    bool SolveLinearElastic, CONSERVATIVEALE, NeedInterMassMat;
    
  public:
    
    TSystemTNSE2D_ALE(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   
                                           TFESpace2D *gridFESpace, TFEVectFunct2D *MeshVelocity, bool conservativeale);
// 
//     /** destrcutor */
//     ~TSystemTNSE2D_ALE();
// 
    void Init(CoeffFct2D *lincoeffs, BoundCondFunct2D *BoundCond, BoundValueFunct2D *U1BoundValue, BoundValueFunct2D *U2BoundValue,
              CoeffFct2D *GridBilinearCoeffs, BoundCondFunct2D *GridBoundCond, BoundValueFunct2D *gridBoundValue, TAuxParam2D *aux, TAuxParam2D *nseaux_error);
    
    
    void AddMeshModifyFunction(ModifyMeshCoords *modifyCoord)
    {  ModifyCoord = modifyCoord; SolveLinearElastic = FALSE; }
    
    void AddBoundModifyFunction(ModifyBoundCoords *modifyboudary)
    {  ModifyBoudary = modifyboudary; SolveLinearElastic = TRUE; }  
 
    void Assemble(double *sol, double *rhs);
  
//     /** store M mat for next time step */
//     void StoreMmat();
    
//     /** move mesh  */   
//     void MoveMesh(int N_MovVert, TVertex **MovBoundVert, TIsoBoundEdge **Free_Joint, 
//                                              double * Iso_refX, double Currtime);
//     
//     /** move mesh  */
//     void MoveMesh(double Currtime);
    
    void GetMeshVeloAndMove(int N_MovVert, TVertex **MovBoundVert, TIsoBoundEdge **Free_Joint, 
                     double * Iso_refX,  double Currtime, double tau);
    
    void GetMeshVeloAndMove(double Currtime, double tau);
    
    void AssembleMeshMat();
    
    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);   
 
//     /** measure the error in the NSE */
//     void MeasureTNSEErrors(DoubleFunct2D *ExactU1, DoubleFunct2D *ExactU2, DoubleFunct2D *ExactP, double *AllErrors);
//     
//     void GridMovingScheme(TFESpace2D *GridFESpace);
};

#endif