MultiDOF_Manager.h

Go to the documentation of this file.

#ifndef included_AMP_MultiDOF_Manager
#define included_AMP_MultiDOF_Manager
 
#include "AMP/discretization/DOF_Manager.h"
#include "AMP/discretization/MultiDOFHelper.h"
#include "AMP/mesh/Mesh.h"
#include "AMP/mesh/MeshElement.h"
#include <memory>
 
 
namespace AMP::Discretization {
 
 
class multiDOFManager : public DOFManager
{
public:
    using DOFManager::subset;
 
    multiDOFManager( const AMP_MPI &comm,
                     std::vector<std::shared_ptr<DOFManager>> managers,
                     std::shared_ptr<const AMP::Mesh::Mesh> mesh = {} );
 
    multiDOFManager( std::shared_ptr<DOFManager> manager );
 
    virtual ~multiDOFManager() override;
 
 
    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 AMP_MPI &comm ) override;
 
 
    std::shared_ptr<DOFManager> subset( const std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                                        bool useMeshComm = true ) override;
 
 
    std::shared_ptr<DOFManager> subset( const AMP::Mesh::MeshIterator &iterator,
                                        const AMP_MPI &comm ) override;
 
    void reset( std::vector<std::shared_ptr<DOFManager>> managers,
                std::shared_ptr<const AMP::Mesh::Mesh> mesh = {} );
 
    int getDOFsPerPoint() const override;
 
public:
    inline const auto &getDOFManagers() const { return d_managers; }
 
    inline const auto &getDOFManager( const size_t i ) const { return d_managers[i]; }
 
 
    std::vector<size_t> getGlobalDOF( const int DOFManager,
                                      const std::vector<size_t> &localDOF ) const;
 
 
    std::vector<size_t> getSubDOF( const int DOFManager,
                                   const std::vector<size_t> &globalDOF ) const;
 
 
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;
 
    std::vector<size_t> getLocalSizes() const override;
 
    // Get the map
    inline const multiDOFHelper &getMap() const { return d_dofMap; }
 
private:
    // Convert the local to global dof
    inline size_t subToGlobal( int manager, size_t dof ) const
    {
        return d_dofMap.subToGlobal( manager, dof );
    }
 
    // Convert the global to local dof
    inline std::pair<size_t, int> globalToSub( size_t dof ) const
    {
        return d_dofMap.globalToSub( dof );
    }
 
 
private:
    multiDOFManager() = delete;
 
    std::shared_ptr<const AMP::Mesh::Mesh> d_mesh;
    std::vector<std::shared_ptr<DOFManager>> d_managers;
    std::vector<size_t> d_globalSize;
    multiDOFHelper d_dofMap;
};
} // namespace AMP::Discretization
 
#endif