subsetCommSelfDOFManager.h

Go to the documentation of this file.

#ifndef included_AMP_subsetCommSelfDOFManager
#define included_AMP_subsetCommSelfDOFManager
 
#include "AMP/discretization/DOF_Manager.h"
#include "AMP/mesh/MeshElement.h"
#include <memory>
 
 
namespace AMP::Discretization {
 
 
class subsetCommSelfDOFManager : public DOFManager
{
public:
    using DOFManager::subset;
 
    subsetCommSelfDOFManager();
 
    subsetCommSelfDOFManager( std::shared_ptr<const DOFManager> parentDOFManager );
 
 
    AMP::Mesh::MeshElementID getElementID( size_t dof ) const override;
 
 
    std::unique_ptr<AMP::Mesh::MeshElement> getElement( size_t dof ) const override;
 
 
    virtual ~subsetCommSelfDOFManager() = default;
 
 
    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;
 
 
    std::vector<size_t> getLocalSizes() const override { return { d_end - d_begin }; }
 
    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:
    // Convert dof indicies
    size_t getSubsetDOF( size_t N, size_t *dofs ) const;
 
private:
    std::shared_ptr<const DOFManager> d_parentDOFManager;
 
    size_t d_parentBegin, d_parentEnd;
};
} // namespace AMP::Discretization
 
#endif