MeshVariable.h

Go to the documentation of this file.

#ifndef included_AMP_MeshVariable_H
#define included_AMP_MeshVariable_H
 
#include "AMP/mesh/Mesh.h"
#include "AMP/vectors/SubsetVariable.h"
 
 
namespace AMP::LinearAlgebra {
 
 
class MeshVariable : public SubsetVariable
{
public:
    MeshVariable( const std::string &name,
                  std::shared_ptr<const AMP::Mesh::Mesh> mesh,
                  bool useMeshComm = true );
 
    std::shared_ptr<AMP::Discretization::DOFManager>
        getSubsetDOF( std::shared_ptr<AMP::Discretization::DOFManager> ) const override;
 
    AMP::AMP_MPI getComm( const AMP::AMP_MPI &comm ) const override;
 
public: // Functions inherited from Variable
    std::string className() const override { return "MeshVariable"; }
    uint64_t getID() const override;
    std::shared_ptr<VectorSelector> createVectorSelector() const override;
    void writeRestart( int64_t ) const override;
    MeshVariable( int64_t );
 
private:
    MeshVariable();
    bool d_useMeshComm = false;
    std::shared_ptr<const AMP::Mesh::Mesh> d_mesh;
};
 
 
class MeshIteratorVariable : public SubsetVariable
{
public:
    MeshIteratorVariable( const std::string &name,
                          const AMP::Mesh::MeshIterator &iterator,
                          const AMP_MPI &comm );
 
    std::shared_ptr<AMP::Discretization::DOFManager>
        getSubsetDOF( std::shared_ptr<AMP::Discretization::DOFManager> ) const override;
 
    AMP::AMP_MPI getComm( const AMP::AMP_MPI &comm ) const override;
 
public: // Functions inherited from Variable
    std::string className() const override { return "MeshIteratorVariable"; }
    uint64_t getID() const override;
    std::shared_ptr<VectorSelector> createVectorSelector() const override;
    void writeRestart( int64_t ) const override;
    MeshIteratorVariable( int64_t );
 
private:
    MeshIteratorVariable();
    const AMP::AMP_MPI d_comm;
    const AMP::Mesh::MeshIterator d_iterator;
};
 
} // namespace AMP::LinearAlgebra
 
#endif