LocalAssembling2D.h

#ifndef __LOCAL_ASSEMBLING_2D__
#define __LOCAL_ASSEMBLING_2D__

#include <Enumerations.h>
#include <Constants.h>
#include <string>
#include <vector>

enum LocalAssembling2D_type { CD2D_Galerkin,
                              CD2D_SUPG,
                              CD2D_GLS,
                              CD2D_Axiax3D_Galerkin,
                              TCD2D_Mass_Rhs_Galerkin, // mass matrix and rhs
                              TCD2D_Stiff_Rhs_Galerkin,// stiffness matrix + rhs
                              TCD2D_Mass_Rhs_SUPG, // mass matrix and rhs
                              TCD2D_Stiff_Rhs_SUPG, // stiffness matrix + rhs
                              NSE2D_Galerkin,
                              NSE2D_Galerkin_Nonlinear,
                              Darcy2D_Galerkin
};

class LocalAssembling2D
{
  protected:
    std::string name;

    int N_Terms;

    int N_Spaces;

    bool *Needs2ndDerivatives;

    std::vector<MultiIndex2D> Derivatives;

    std::vector<int> FESpaceNumber;

    std::vector<int> RowSpace;

    std::vector<int> ColumnSpace;

    std::vector<int> RhsSpace;

    CoeffFct2D *Coeffs;

    AssembleFctParam2D *AssembleParam;

    ManipulateFct2D *Manipulate;

    double ***AllOrigValues;

    double **OrigValues;
    
    int N_Matrices;
    int N_Rhs;
    
    
    int N_ParamFct;
    
    std::vector<ParamFct*> ParameterFct;
    
    std::vector<int> BeginParameter;
    
    // number of parameters
    int N_Parameters;
    
    TFEFunction2D **FEFunctions2D;
    
    int N_FEValues;
    
    std::vector<int> FEValue_FctIndex;
    
    std::vector<MultiIndex2D> FEValue_MultiIndex;

    void set_parameters_for_nse(LocalAssembling2D_type type);
  public:
    LocalAssembling2D(LocalAssembling2D_type type, TFEFunction2D **fefunctions2d,
                      CoeffFct2D *coeffs);

    ~LocalAssembling2D();
    
    

    void GetLocalForms(int N_Points, double *weights, double *AbsDetjk,
                       double *X, double *Y,
                       int *N_BaseFuncts, BaseFunct2D *BaseFuncts, 
                       double **Parameters, double **AuxArray,
                       TBaseCell *Cell, int N_Matrices, int N_Rhs,
                       double ***LocMatrix, double **LocRhs,
                       double factor = 1.);
    
    void GetParameters(int n_points, TCollection *Coll,
                       TBaseCell *cell, int cellnum,
                       double *x, double *y,
                       double **Parameters);

    void GetParameters(int N_Points, TCollection *Coll,
                       TBaseCell *cell, int cellnum,
                       double *s, int joint,
                       double **Parameters);
    
    

    const std::string& get_name() const
    { return name; }

    bool *GetNeeds2ndDerivatives() const
    { return Needs2ndDerivatives; }

    CoeffFct2D *GetCoeffFct() const
    { return Coeffs; }
    
    int rowSpaceOfMat(int i) const
    { return RowSpace[i]; }
    
    int colSpaceOfMat(int i) const
    { return ColumnSpace[i]; }
    
    int GetN_ParamFct() const
    { return N_ParamFct; }
    
    int GetN_Parameters() const
    { return N_Parameters; }
};


#endif // __LOCAL_ASSEMBLING_2D__