SystemTCD2D.h

#ifndef __SYSTEMTCD2D__
#define __SYSTEMTCD2D__

#include <SquareMatrix2D.h>
#include <SystemCD2D.h>
#include <CDSystemTimeDG.h>
#include <CDSystemTimeDG_1.h>

class TSystemTCD2D : public TSystemCD2D
{
  protected:
        
    TSquareMatrix2D *sqmatrixM;
    
    double *B;   
   
    double *defect;   
    
    double gamma;   
    
    TSquareMatrix2D *sqmatrixK;    
    
    TSquareMatrix2D *sqmatrixS;
    
    TDiscreteForm2D *DiscreteFormMRhs; 
    
    bool SystMatAssembled;
    
   TCDSystemTimeDG *TimeDG;
//    TCDSystemTimeDG_1 *TimeDG_1;
  
  public:
     TSystemTCD2D(TFESpace2D *fespace, int disctype, int solver);

    ~TSystemTCD2D();

    void Init(CoeffFct2D *BilinearCoeffs, BoundCondFunct2D *BoundCond, BoundValueFunct2D *BoundValue);
    
    TSquareMatrix2D *GetAMatrix()
    { return sqmatrixA; }
    
    void AssembleMRhs(TAuxParam2D *aux, double *sol, double *rhs); 
    
    void AssembleARhs(TAuxParam2D *aux, double *sol, double *rhs);   
    
    void AssembleSystMat(double *oldrhs, double *oldsol, double *rhs, double *sol);
    
    void RestoreMassMat();
    
    void Solve(double *sol, double *rhs);  
    
    double GetResidual(double *sol);
    
};

#endif