DOF_Manager.h

Go to the documentation of this file.

#ifndef included_AMP_DOF_Manager
#define included_AMP_DOF_Manager
 
#include "AMP/discretization/DOF_ManagerParameters.h"
#include "AMP/mesh/Mesh.h"
#include "AMP/mesh/MeshElement.h"
#include "AMP/mesh/MeshID.h"
#include "AMP/utils/AMP_MPI.h"
 
 
namespace AMP::IO {
class RestartManager;
}
 
 
namespace AMP::Discretization {
 
 
class DOFManager : public AMP::enable_shared_from_this<AMP::Discretization::DOFManager>
{
public:
    DOFManager( size_t N_local, const AMP_MPI &comm, std::vector<size_t> remoteDOFs = {} );
 
 
    virtual ~DOFManager();
 
 
    virtual std::string className() const;
 
 
    virtual bool operator==( const DOFManager &rhs ) const;
 
 
    bool operator!=( const DOFManager &rhs ) const;
 
 
    virtual AMP::Mesh::MeshElementID getElementID( size_t dof ) const;
 
 
    virtual std::unique_ptr<AMP::Mesh::MeshElement> getElement( size_t dof ) const;
 
 
    virtual int getDOFsPerPoint() const;
 
 
    void getDOFs( const AMP::Mesh::MeshElementID &id, std::vector<size_t> &dofs ) const;
 
 
    void getDOFs( const std::vector<AMP::Mesh::MeshElementID> &ids,
                  std::vector<size_t> &dofs ) const;
 
 
    void getDOFs( int N_ids, const AMP::Mesh::MeshElementID *ids, std::vector<size_t> &dofs ) const;
 
 
    virtual std::shared_ptr<const AMP::Mesh::Mesh> getMesh() const;
 
 
    virtual AMP::Mesh::MeshIterator getIterator() const;
 
 
    virtual size_t beginDOF() const;
 
 
    virtual size_t endDOF() const;
 
 
    virtual size_t numLocalDOF() const;
 
 
    virtual size_t numGlobalDOF() const;
 
 
    virtual std::vector<size_t> getLocalSizes() const;
 
 
    inline const AMP_MPI &getComm() const { return d_comm; }
 
 
    virtual std::vector<size_t> getRemoteDOFs() const;
 
    virtual void replaceRemoteDOFs( std::vector<size_t> &newRemote ) { d_remoteDOFs = newRemote; }
 
 
    std::vector<size_t> getRowDOFs( const AMP::Mesh::MeshElementID &id ) const;
 
 
    virtual std::shared_ptr<DOFManager> subset( const AMP_MPI &comm );
 
 
    virtual std::shared_ptr<DOFManager> subset( const std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                                                bool useMeshComm = true );
 
 
    virtual std::shared_ptr<DOFManager> subset( const AMP::Mesh::MeshIterator &iterator,
                                                const AMP_MPI &comm );
 
 
    uint64_t getID() const;
 
 
public: // Advanced interfaces
    virtual size_t getRowDOFs( const AMP::Mesh::MeshElementID &id,
                               size_t *dofs,
                               size_t N_alloc,
                               bool sort = true ) const;
 
    // Append DOFs to the list
    virtual size_t appendDOFs( const AMP::Mesh::MeshElementID &id,
                               size_t *dofs,
                               size_t index,
                               size_t capacity ) const;
 
 
public: // Write/read restart data
    virtual void registerChildObjects( AMP::IO::RestartManager *manager ) const;
 
    virtual void writeRestart( int64_t fid ) const;
 
    DOFManager( int64_t fid, AMP::IO::RestartManager *manager );
 
 
protected:
    DOFManager(){};
 
    std::shared_ptr<DOFManagerParameters> params;
 
    size_t d_begin = 0, d_end = 0, d_global = 0;
    mutable std::vector<size_t> d_localSize;
 
    std::vector<size_t> d_remoteDOFs;
 
    AMP_MPI d_comm;
};
 
 
} // namespace AMP::Discretization
 
#endif