structuredMeshIterator.h

Go to the documentation of this file.

#ifndef included_AMP_structuredMeshIterators
#define included_AMP_structuredMeshIterators
 
#include "AMP/mesh/MeshIterator.h"
#include "AMP/mesh/structured/BoxMesh.h"
#include "AMP/mesh/structured/structuredMeshElement.h"
#include <memory>
 
#include <array>
 
 
namespace AMP::Mesh {
 
 
class structuredMeshIterator final : public MeshIteratorBase
{
public:
    structuredMeshIterator();
 
    structuredMeshIterator( const BoxMesh::MeshElementIndex &first,
                            const BoxMesh::MeshElementIndex &last,
                            const AMP::Mesh::BoxMesh *mesh,
                            size_t pos = 0 );
 
    structuredMeshIterator( const BoxMesh::MeshElementIndexIterator &it,
                            const AMP::Mesh::BoxMesh *mesh,
                            size_t pos = 0 );
 
 
    structuredMeshIterator( std::shared_ptr<const std::vector<BoxMesh::MeshElementIndex>> elements,
                            const AMP::Mesh::BoxMesh *mesh,
                            size_t pos = 0 );
 
    virtual ~structuredMeshIterator();
 
    structuredMeshIterator( structuredMeshIterator && ) = delete;
 
    structuredMeshIterator( const structuredMeshIterator & );
 
    structuredMeshIterator &operator=( structuredMeshIterator && ) = delete;
 
    structuredMeshIterator &operator=( const structuredMeshIterator & );
 
    std::string className() const override { return "structuredMeshIterator"; }
 
    void setPos( size_t ) override;
 
    MeshIteratorBase &operator++() override;
 
    MeshIteratorBase &operator--() override;
 
    // Arithmetic operator+=
    MeshIteratorBase &operator+=( int N ) override;
 
    bool operator==( const MeshIteratorBase &rhs ) const override;
 
    bool operator!=( const MeshIteratorBase &rhs ) const override;
 
    MeshIterator begin() const override;
 
    std::unique_ptr<MeshIteratorBase> clone() const override;
 
    using MeshIteratorBase::operator==;
    using MeshIteratorBase::operator!=;
 
 
public: // Write/read restart data
    void registerChildObjects( AMP::IO::RestartManager *manager ) const override;
    void writeRestart( int64_t fid ) const override;
    structuredMeshIterator( int64_t fid, AMP::IO::RestartManager *manager );
 
 
public: // Advanced interfaces
    // Get the elements in the iterator
    std::shared_ptr<const std::vector<BoxMesh::MeshElementIndex>> getElements() const;
 
    // Get the current index
    BoxMesh::MeshElementIndex getCurrentIndex() const;
 
 
private:
    // Data members
    BoxMesh::MeshElementIndexIterator d_it;
    std::shared_ptr<const std::vector<BoxMesh::MeshElementIndex>> d_elements;
    const AMP::Mesh::BoxMesh *d_mesh;
    mutable structuredMeshElement d_cur_element;
};
 
 
} // namespace AMP::Mesh
 
#endif