subsetDOFManager.h

Go to the documentation of this file.

#ifndef included_AMP_subsetDOF_Manager
#define included_AMP_subsetDOF_Manager
 
#include "AMP/discretization/DOF_Manager.h"
#include "AMP/mesh/MeshElement.h"
#include <memory>
 
 
namespace AMP::Discretization {
 
 
class subsetDOFManager : public DOFManager
{
public:
    using DOFManager::subset;
 
    static std::shared_ptr<DOFManager> create( std::shared_ptr<const DOFManager> parentDOFManager,
                                               const std::vector<size_t> &dofs,
                                               const AMP::Mesh::MeshIterator &iterator,
                                               const AMP_MPI &comm );
 
 
    AMP::Mesh::MeshElementID getElementID( size_t dof ) const override;
 
 
    std::unique_ptr<AMP::Mesh::MeshElement> getElement( size_t dof ) const override;
 
 
    virtual ~subsetDOFManager();
 
 
    std::shared_ptr<const AMP::Mesh::Mesh> getMesh() const override;
 
 
    AMP::Mesh::MeshIterator getIterator() const override;
 
 
    std::vector<size_t> getRemoteDOFs() const override;
 
 
    virtual std::vector<size_t> getLocalParentDOFs() const;
 
 
    virtual std::vector<size_t> getParentDOF( const std::vector<size_t> & ) const;
 
 
    virtual std::vector<size_t> getSubsetDOF( const std::vector<size_t> & ) const;
 
 
    virtual std::shared_ptr<const DOFManager> getDOFManager() const;
 
 
    int getDOFsPerPoint() const override;
 
 
public: // Advanced interfaces
    size_t getRowDOFs( const AMP::Mesh::MeshElementID &id,
                       size_t *dofs,
                       size_t N_alloc,
                       bool sort = true ) const override;
    using DOFManager::getRowDOFs;
 
    // Append DOFs to the list
    size_t appendDOFs( const AMP::Mesh::MeshElementID &id,
                       size_t *dofs,
                       size_t index,
                       size_t capacity ) const override;
 
 
protected:
    // The constructor is protected
    subsetDOFManager() : d_parentBegin( 0 ), d_parentEnd( 0 ), d_parentGlobal( 0 ) {}
 
private:
    std::shared_ptr<const DOFManager> d_parentDOFManager;
 
    size_t d_parentBegin, d_parentEnd, d_parentGlobal;
 
    std::vector<size_t> d_localDOFs;
 
    std::vector<size_t> d_remoteParentDOFs;
    std::vector<size_t> d_remoteSubsetDOFs;
 
    AMP::Mesh::MeshIterator d_iterator;
};
} // namespace AMP::Discretization
 
#endif