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 = false) {};

    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,
                                   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