stratreftree.h

Go to the documentation of this file.

#ifndef stratreftree_h
#define stratreftree_h

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bert Bril
 Date:          Dec 2003
 RCS:           $Id$
________________________________________________________________________

-*/

#include "stratmod.h"
#include "stratunitref.h"
#include "stratlith.h"
#include "stratcontent.h"
#include "repos.h"
#include "od_iosfwd.h"

namespace Strat
{
class LevelSet;


mExpClass(Strat) RefTree : public NodeOnlyUnitRef
{
public:
                        RefTree();
                        ~RefTree();

    Repos::Source       source() const                  { return src_; }

    LithologySet&       lithologies()                   { return liths_; }
    const LithologySet& lithologies() const             { return liths_; }
    ContentSet&         contents()                      { return contents_; }
    const ContentSet&   contents() const                { return contents_; }

    static const char*  sKeyNoCode()                    { return "<no_code>"; }

    Notifier<RefTree>   unitAdded;
    Notifier<RefTree>   unitChanged;
    Notifier<RefTree>   unitToBeDeleted;
    const UnitRef*      notifUnit() const               { return notifun_; }

    const LeafUnitRef&  undefLeaf() const               { return udfleaf_; }
    virtual int         level() const                   { return 0; }

    static void         getStdNames(BufferStringSet&);
    static RefTree*     createStd(const char*);

    void                createFromLevelSet(const LevelSet&);
    const LeavedUnitRef* getLevelSetUnit(const char* lvlnm) const;

    void                addLevelUnit(const Level&);
    void                removeLevelUnit(const Level&);
                        // adds & removes units from levels on RefTree created
                        // using createFromLevelSet()

    Notifier<RefTree>   deleteNotif;

protected:

    void                initTree();
    void                setToActualTypes();

    LithologySet        liths_;
    ContentSet          contents_;
    const UnitRef*      notifun_;
    LeafUnitRef&        udfleaf_;

    void                levelToBeRemoved(CallBacker*);

public:

    Repos::Source       src_;
    BufferString        name_;

                                // for printing, export or something.
                                // otherwise, use RepositoryAccess::write()
    bool                read(od_istream&);
    bool                write(od_ostream&) const;

    friend class        RefTreeMgr;

    void                reportChange(const UnitRef*,bool isrem=false);
    void                reportAdd(const UnitRef*);
    bool                addLeavedUnit(const char*,const char*);
    Strat::LeavedUnitRef* getByLevel(int lvlid) const;//first match

};

mGlobal(Strat) const RefTree& RT();
inline RefTree& eRT()   { return const_cast<RefTree&>( RT() ); }

// Needless to say that if you push, make sure you pop (so afterwards the real
// default RefTree is restored
mGlobal(Strat) void pushRefTree(RefTree*);
mGlobal(Strat) void popRefTree();

mGlobal(Strat) void setRT(RefTree*);


} // namespace Strat


#endif