stratunitrefiter.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bert
 Date:          Dec 2003 / Sep 2010
________________________________________________________________________


-*/

#include "stratmod.h"
#include "stratunitref.h"


namespace Strat
{

mExpClass(Strat) UnitRefIter
{
public:

    enum Pol            { All, Leaves, AllNodes, LeavedNodes, NodesOnly };
    static Pol          polOf(UnitRef::Type);
    static Pol          polOf(const UnitRef*);

                        UnitRefIter(const NodeUnitRef&,Pol p=All);
                        UnitRefIter( const UnitRefIter& uri )   { *this = uri; }
    UnitRefIter&        operator =(const UnitRefIter&);

    void                reset();
    bool                next();
    UnitRef*            unit()          { return gtUnit(); }
    const UnitRef*      unit() const    { return gtUnit(); }
    Pol                 pol() const     { return pol_; }
    void                setPol( Pol p ) { pol_ = p; reset(); }
    Interval<int>       levelRange() const;

    static bool         isValid(const UnitRef&,Pol);

protected:

    Pol                 pol_;
    NodeUnitRef*        itnode_;
    NodeUnitRef*        curnode_;
    int                 curidx_;

    UnitRef*            gtUnit() const;
    bool                toNext();

};

} // namespace