gsBaseAssembler.h

Go to the documentation of this file.

 
#pragma once
 
#include <gsAssembler/gsAssembler.h>
 
namespace gismo
{
 
template <class T>
class gsBaseAssembler : public gsAssembler<T>
{
public:
    typedef memory::shared_ptr<gsBaseAssembler> Ptr;
    typedef memory::unique_ptr<gsBaseAssembler> uPtr;
 
    virtual bool assemble(const gsMatrix<T> & solutionVector,
                          const std::vector<gsMatrix<T> > & fixedDDoFs) = 0;
 
    virtual void assemble(bool /* saveEliminationMatrix */) {};
    virtual void assemble() { assemble(false); };
    virtual void assemble(const gsMultiPatch<T> & /* curSolution */)
    {GISMO_NO_IMPLEMENTATION}
 
    virtual int numDofs() const { return gsAssembler<T>::numDofs(); }
 
    virtual void constructSolution(const gsMatrix<T> & solVector,
                                   const std::vector<gsMatrix<T> > & fixedDDofs,
                                   gsMultiPatch<T> & result,
                                   const gsVector<index_t> & unknowns) const;
 
    virtual void constructSolution(const gsMatrix<T>& /* solVector */,
                                   gsMultiPatch<T>& /* result */, short_t /* unk */ = 0) const
    {GISMO_NO_IMPLEMENTATION}
    virtual void constructSolution(const gsMatrix<T>& /* solVector */,
                                   gsMultiPatch<T>& /* result */,
                                   const gsVector<index_t>  & /* unknowns */) const
    {GISMO_NO_IMPLEMENTATION}
 
    virtual void constructSolution(const gsMatrix<T> & /* solVector */,
                                   const std::vector<gsMatrix<T> > & /* fixedDDofs */,
                                   gsMultiPatch<T> & /* result */) const {};
 
    //--------------------- DIRICHLET BC SHENANIGANS ----------------------------------//
 
    virtual void setFixedDofs(index_t patch, boxSide side, const gsMatrix<T> & ddofs, bool oneUnk = false);
 
    virtual void setFixedDofs(const std::vector<gsMatrix<T> > & ddofs);
 
    virtual void getFixedDofs(index_t patch, boxSide side, gsMatrix<T> & ddofs) const;
 
    virtual index_t numFixedDofs() const;
 
    virtual void eliminateFixedDofs();
 
    //virtual void modifyDirichletDofs(size_t patch, boxSide side, const gsMatrix<T> & ddofs);
 
    //--------------------- OTHER ----------------------------------//
 
    virtual void setRHS(const gsMatrix<T> & rhs) {m_system.rhs() = rhs;}
 
    virtual void setMatrix(const gsSparseMatrix<T> & matrix) {m_system.matrix() = matrix;}
 
protected:
    using gsAssembler<T>::m_pde_ptr;
    using gsAssembler<T>::m_bases;
    using gsAssembler<T>::m_system;
    using gsAssembler<T>::m_ddof;
 
    gsSparseMatrix<T> eliminationMatrix;
    gsMatrix<T> rhsWithZeroDDofs;
};
 
} // namespace ends
 
#ifndef GISMO_BUILD_LIB
#include GISMO_HPP_HEADER(gsBaseAssembler.hpp)
#endif