simpleDOF_Manager.h

Go to the documentation of this file.

#ifndef included_AMP_simpleDOF_Manager
#define included_AMP_simpleDOF_Manager
 
#include "AMP/discretization/DOF_Manager.h"
#include "AMP/mesh/Mesh.h"
#include "AMP/mesh/MeshElement.h"
#include <memory>
 
 
namespace AMP::Discretization {
 
 
class simpleDOFManager : public DOFManager
{
public:
    using DOFManager::subset;
 
 
    static std::shared_ptr<DOFManager> create( std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                                               AMP::Mesh::GeomType type,
                                               int gcw,
                                               int DOFsPerElement,
                                               bool split = true );
 
    simpleDOFManager( std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                      const AMP::Mesh::MeshIterator &local,
                      const AMP::Mesh::MeshIterator &ghost,
                      AMP::Mesh::GeomType type,
                      int DOFsPerElement );
 
 
    static std::shared_ptr<DOFManager> create( std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                                               const AMP::Mesh::MeshIterator &ghost,
                                               const AMP::Mesh::MeshIterator &local,
                                               int DOFsPerElement );
 
 
    static std::shared_ptr<DOFManager> create( const AMP::Mesh::MeshIterator &it,
                                               int DOFsPerElement );
 
 
    virtual ~simpleDOFManager();
 
 
    std::string className() const override { return "simpleDOFManager"; }
 
 
    AMP::Mesh::MeshElementID getElementID( size_t dof ) const override;
 
 
    std::unique_ptr<AMP::Mesh::MeshElement> getElement( size_t dof ) const override;
 
 
    std::shared_ptr<const AMP::Mesh::Mesh> getMesh() const override;
 
 
    AMP::Mesh::MeshIterator getIterator() const override;
 
 
    std::vector<size_t> getRemoteDOFs() const override;
 
 
    std::shared_ptr<DOFManager> subset( const std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                                        bool useMeshComm = true ) override;
 
    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;
 
 
public: // Write/read restart data
    void registerChildObjects( AMP::IO::RestartManager *manager ) const override;
    void writeRestart( int64_t ) const override;
    simpleDOFManager( int64_t, AMP::IO::RestartManager * );
 
protected:
    // Private constructor
    simpleDOFManager() = delete;
 
    // Function to find the remote DOF given a set of mesh element IDs
    std::vector<size_t> getRemoteDOF( const std::vector<AMP::Mesh::MeshElementID> &ids ) const;
 
    // Function to initialize the data
    void initialize();
 
 
protected:                                                     // Data members
    bool d_isBaseMesh          = false;                        // Is the mesh a base mesh
    AMP::Mesh::GeomType d_type = AMP::Mesh::GeomType::Nullity; // entity type
    uint8_t d_DOFsPerElement   = 0;                            // # Of DOFs per type
    std::shared_ptr<const AMP::Mesh::Mesh> d_mesh;             // Mesh
    AMP::Mesh::MeshID d_meshID;                                // MeshID
    std::vector<AMP::Mesh::MeshID> d_baseMeshIDs;              // Must be global list
    AMP::Mesh::MeshIterator d_localIterator;                   // Local iterator
    AMP::Mesh::MeshIterator d_ghostIterator;                   // global iterator
    std::vector<AMP::Mesh::MeshElementID> d_local_id;          // List of local ids
    std::vector<AMP::Mesh::MeshElementID> d_remote_id;         // List of remote ids
    std::vector<size_t> d_remote_dof;                          // remote dofs
};
} // namespace AMP::Discretization
 
#endif