TriangleMeshIterator.h

Go to the documentation of this file.

#ifndef included_AMP_TriangleMeshIterators
#define included_AMP_TriangleMeshIterators
 
 
#include "AMP/mesh/MeshIterator.h"
#include "AMP/mesh/triangle/TriangleMeshElement.h"
 
 
namespace AMP::Mesh {
 
 
template<uint8_t NG>
class TriangleMesh;
 
 
template<uint8_t NG>
class TriangleMeshIterator final : public MeshIteratorBase
{
public:
    TriangleMeshIterator();
 
    explicit TriangleMeshIterator( const AMP::Mesh::TriangleMesh<NG> *mesh,
                                   std::shared_ptr<const std::vector<ElementID>> list,
                                   size_t pos = 0 );
 
    virtual ~TriangleMeshIterator() = default;
 
    TriangleMeshIterator( const TriangleMeshIterator & );
 
    TriangleMeshIterator &operator=( const TriangleMeshIterator & );
 
    std::string className() const override;
 
    void setPos( size_t ) override;
 
    // Increment
    MeshIteratorBase &operator++() override;
 
    // Decrement
    MeshIteratorBase &operator--() override;
 
    // Arithmetic operator+=
    MeshIteratorBase &operator+=( int N ) override;
 
    // Check if two iterators are equal
    bool operator==( const MeshIteratorBase &rhs ) const override;
 
    // Check if two iterators are not equal
    bool operator!=( const MeshIteratorBase &rhs ) const override;
 
    // Return an iterator to the begining
    MeshIterator begin() const override;
 
    auto getList() const { return d_list; }
 
    std::unique_ptr<MeshIteratorBase> clone() const override;
 
    using MeshIteratorBase::operator==;
    using MeshIteratorBase::operator!=;
 
 
public: // Write/read restart data
    void registerChildObjects( AMP::IO::RestartManager * ) const override;
    void writeRestart( int64_t ) const override;
    TriangleMeshIterator( int64_t, AMP::IO::RestartManager * );
 
 
protected:
    // Data members
    const AMP::Mesh::TriangleMesh<NG> *d_mesh;
    std::shared_ptr<const std::vector<ElementID>> d_list;
    TriangleMeshElement<NG> d_cur_element;
};
 
 
} // namespace AMP::Mesh
 
#endif