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
 RCS:           $Id$
________________________________________________________________________
 
 
-*/
 
#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