SystemTCD2D_ALE.h

#ifndef __SYSTEMCD2D_ALE__
#define __SYSTEMCD2D_ALE__

#include <SquareMatrix2D.h>
#include <SystemTCD2D.h>

class TSystemTCD2D_ALE : public TSystemTCD2D
{
  protected:
    int N_GridDOFs, N_GridActive, *GridKCol, *GridRowPtr;
    
    double *rhs_Qp1; // dG(1)  
    
    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;  
    
    TSquareMatrix2D *sqmatrixM_old;  
    
    TSquareMatrix2D *sqmatrixM_Qp1, *sqmatrixA_Qp1;   // dG(1)   
    
    TSquareMatrix2D *SqmatrixG11, *SqmatrixG12, *SqmatrixG21, *SqmatrixG22, *SQMATRICES_GRID[4];
    
    TSquareStructure2D *SquareStructureG;
    
    TAuxParam2D *Aux_ALE, *Meshaux;

    BoundCondFunct2D *GridBoundaryConditions[1];
    BoundValueFunct2D *GridBoundValues[1];
      
    ModifyMeshCoords *ModifyCoord;
    
    ModifyBoundCoords *ModifyBoudary;   
    
    int N_MovVert;
    TVertex **MovBoundVert;
    TIsoBoundEdge **Free_Joint;
    double *Iso_refX;
    
    bool SolveLinearElastic, CONSERVATIVEALE, NeedInterMassMat;
    
  public:
    TSystemTCD2D_ALE(TFESpace2D *fespace, int disctype, int solver, TFESpace2D *gridFESpace, TFEVectFunct2D *MeshVelocity, 
                                                       bool conservativeale);

//     /** destrcutor */
//     ~TSystemMatTimeScalar2D();

    void Init(CoeffFct2D *BilinearCoeffs, BoundCondFunct2D *BoundCond, BoundValueFunct2D *BoundValue, 
              CoeffFct2D *GridBilinearCoeffs, BoundCondFunct2D *GridBoundCond, BoundValueFunct2D *gridBoundValue,
              TAuxParam2D *aux);

    void AddMeshModifyFunction(ModifyMeshCoords *modifyCoord)
    {  ModifyCoord = modifyCoord; SolveLinearElastic = FALSE; }
    
    void AddBoundModifyFunction(ModifyBoundCoords *modifyboudary, int n_MovVert, TVertex **movBoundVert, 
                TIsoBoundEdge **free_Joint, double * iso_refX);  
    
    
//     /** return the stiffness matric */
//     TSquareMatrix2D *GetAMatrix()
//     { return sqmatrixA; }

    void StoreMmat();
    
    void MoveMesh(double Currtime);
    
    void GetMeshVelo(double Currtime, double tau, bool MoveMesh);
    
    void AssembleMeshMat();
    
    void AssembleMRhs(double *sol, double *rhs);   
    
    void AssembleMARhs(double *sol, double *rhs); 

    void AssembleALEMat(double *sol, double *rhs, double tau);
    
    void AssembleSystMat(double *oldrhs, double *oldsol, double *rhs, double *sol);
 
    void Solve(double *sol, double *rhs);  
    
//     /** return the residual of the system for the given sol*/
//     double GetResidual(double *sol);
//     
};

#endif