gsFlowAssemblerBase.h

Go to the documentation of this file.

 
#pragma once
 
#include <gsIncompressibleFlow/src/gsFlowSolverParams.h>
#include <gsIncompressibleFlow/src/gsFlowVisitors.h>
 
namespace gismo
{
 
template<class T, int MatOrder>
class gsFlowAssemblerBase
{
 
protected: // *** Class members ***
 
    gsFlowSolverParams<T> m_params;
    index_t m_dofs;
    short_t m_tarDim;
    bool m_isInitialized;
    bool m_isBaseReady;
    bool m_isSystemReady;
    std::vector<gsDofMapper> m_dofMappers;
    std::vector<gsMatrix<T> > m_ddof;
    gsMatrix<T> m_solution;
 
 
public: // *** Constructor/destructor ***
 
    gsFlowAssemblerBase(const gsFlowSolverParams<T>& params):
    m_params(params)
    { }
 
    virtual ~gsFlowAssemblerBase()
    { }
 
 
protected: // *** Member functions ***
 
    void initMembers();
 
    virtual void updateSizes()
    { GISMO_NO_IMPLEMENTATION }
 
    virtual void updateDofMappers()
    { GISMO_NO_IMPLEMENTATION }
 
    void computeDirichletDofs(const index_t unk, const index_t basisID, gsMatrix<T>& ddofVector);
 
    void computeDirichletDofsIntpl(const index_t unk, const gsDofMapper & mapper, const gsMultiBasis<T> & mbasis, gsMatrix<T>& ddofVector);
 
    void computeDirichletDofsL2Proj(const index_t unk, const gsDofMapper & mapper, const gsMultiBasis<T> & mbasis, gsMatrix<T>& ddofVector);
 
    void assembleBlock(gsFlowVisitor<T, MatOrder>& visitor, index_t testBasisID, gsSparseMatrix<T, MatOrder>& block, gsMatrix<T>& blockRhs);
 
    void assembleRhs(gsFlowVisitor<T, MatOrder>& visitor, index_t testBasisID, gsMatrix<T>& rhs);
 
    virtual void assembleLinearPart()
    {GISMO_NO_IMPLEMENTATION}
 
    virtual void assembleNonlinearPart()
    {GISMO_NO_IMPLEMENTATION}
 
    virtual void updateAssembly();
 
 
    virtual void updateCurrentSolField(const gsMatrix<T> & solVector, bool updateSol)
    {GISMO_NO_IMPLEMENTATION}
 
 
public: // *** Member functions ***
 
    virtual void initialize();
 
    virtual void update(const gsMatrix<T> & solVector, bool updateSol = true);
 
    virtual void markDofsAsEliminatedZeros(const std::vector< gsMatrix< index_t > > & boundaryDofs, const index_t unk)
    {GISMO_NO_IMPLEMENTATION}
 
    virtual gsField<T> constructSolution(const gsMatrix<T>& solVector, index_t unk) const
    {GISMO_NO_IMPLEMENTATION}
 
 
public: // *** Getters/setters ***
 
    index_t numDofs() const 
    { 
        GISMO_ASSERT(m_dofs > 0, "Something went wrong, number of DOFs is zero!");
        return m_dofs; 
    }
 
    short_t getTarDim() const { return m_tarDim; }
 
    bool isInitialized() { return m_isInitialized; }
 
    const std::vector<gsDofMapper>& getMappers() const { return m_dofMappers; }
 
    const std::vector<gsMatrix<T> >& getDirichletDofs() const { return m_ddof; }
 
    const gsMatrix<T>& getSolution() const { return m_solution; }
 
    const gsMultiPatch<T>& getPatches() const { return m_params.getPde().patches(); }
 
    std::vector< gsMultiBasis<T> >& getBases() { return m_params.getBases(); }
    const std::vector< gsMultiBasis<T> >& getBases() const { return m_params.getBases(); }
 
    const gsBoundaryConditions<T>& getBCs() const { return m_params.getBCs(); }
 
    const gsFunction<T>* getRhsFcn() const { return m_params.getPde().rhs(); }
 
    gsAssemblerOptions getAssemblerOptions() const { return m_params.assemblerOptions(); }
    
    gsOptionList options() const { return m_params.options(); }
 
    virtual const gsSparseMatrix<T, MatOrder>& matrix() const
    {GISMO_NO_IMPLEMENTATION}
 
    virtual const gsSparseMatrix<T, MatOrder>& matrix(index_t unk) const
    {GISMO_NO_IMPLEMENTATION}
 
    virtual gsSparseMatrix<T, MatOrder>& getMassMatrix(index_t unkID)
    {GISMO_NO_IMPLEMENTATION}
    
    virtual const gsSparseMatrix<T, MatOrder>& getMassMatrix(index_t unkID) const
    {GISMO_NO_IMPLEMENTATION}
 
    virtual const gsMatrix<T>& rhs() const
    {GISMO_NO_IMPLEMENTATION}
 
    virtual const gsMatrix<T>& rhs(index_t unk) const
    {GISMO_NO_IMPLEMENTATION}
 
};
 
} // namespace gismo
 
#ifndef GISMO_BUILD_LIB
#include GISMO_HPP_HEADER(gsFlowAssemblerBase.hpp)
#endif