uistratdispdata.h

Go to the documentation of this file.

#ifndef uistratdispdata_h
#define uistratdispdata_h

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bruno
 Date:          Mar 2010
 RCS:           $Id$
________________________________________________________________________

-*/

#include "uistratmod.h"
#include "uistratmod.h"
#include "callback.h"
#include "color.h"
#include "ranges.h"
#include "bufstring.h"
#include "bufstringset.h"


namespace Strat
{
    class UnitRef;
    class LeavedUnitRef;
    class NodeOnlyUnitRef;
    class NodeUnitRef;
    class RefTree;
}

class uiStratRefTree;
class uiTreeViewItem;

mExpClass(uiStrat) StratDispData
{
public:
                        StratDispData() {}
                        ~StratDispData() { eraseData(); }

    mStruct(uiStrat) Unit
    {
                        Unit(const char* nm, const char* fullcode=0,
                                    const Color& col = Color::White() )
                                : name_( nm )
                                , fullcode_(fullcode)
                                , color_(col)
                                , isdisplayed_(true)
                                {}

        const char*     name() const    { return name_.buf(); }
        const char*     fullCode() const { return fullcode_.buf(); }

        Color           color_;
        Interval<float> zrg_;
        bool            isdisplayed_;

        int             colidx_; //tree depth

    protected :

        BufferString    name_;
        BufferString    fullcode_;
    };


    mStruct(uiStrat) Level
    {
                        Level(const char* nm,const char* unitcode)
                                : unitcode_(unitcode)
                                , name_( nm )
                                {}

        const BufferString name_;
        const BufferString unitcode_;
        Color           color_;
        float           zpos_;
    };


    mStruct(uiStrat) Column
    {
                        Column( const char* nm )
                            : name_(nm)
                            , isdisplayed_(true)
                            {}

        const BufferString name_;
        bool            isdisplayed_;

        ObjectSet<Unit> units_;
        ObjectSet<Level> levels_;
    };



    void                eraseData()
                        {
                            for ( int idx=0; idx<cols_.size(); idx++ )
                            {
                                cols_[idx]->units_.erase();
                                cols_[idx]->levels_.erase();
                            }
                            cols_.erase();
                        }

    void                addCol( Column* col )
                            { cols_ += col; }

    int                 nrCols() const
                            { return cols_.size(); }
    int                 nrUnits( int colidx ) const
                            { return cols_[colidx]->units_.size(); }
    void                addUnit( int colidx, Unit* un )
                            { cols_[colidx]->units_ += un; un->colidx_=colidx; }

    const Column*       getCol( int idx ) const
                            { return cols_[idx]; }
    Column*             getCol( int idx )
                            { return cols_[idx]; }
    Unit*               getUnit( int colidx, int uidx )
                            { return gtUnit( colidx, uidx ); }
    const Unit*         getUnit( int colidx, int uidx ) const
                            { return gtUnit( colidx, uidx ); }

    int                 nrLevels( int colidx ) const
                            { return cols_[colidx]->levels_.size(); }
    const Level*        getLevel( int colidx, int lidx ) const
                            { return cols_[colidx]->levels_[lidx]; }

    int                 nrDisplayedCols() const
                        {
                            int nr = 0;
                            for ( int idx=0; idx<cols_.size(); idx++)
                                { if ( cols_[idx]->isdisplayed_ ) nr++; }
                            return nr;
                        }

protected :

    Unit*               gtUnit( int colidx, int uidx ) const
                            { return const_cast<Unit*>(
                                        cols_[colidx]->units_[uidx] ); }

    ObjectSet<Column> cols_;
};



mExpClass(uiStrat) uiStratTreeToDisp : public CallBacker
{
public:
                        uiStratTreeToDisp(StratDispData&,
                                        bool withaux=true,
                                        bool withlvls=true);
                        ~uiStratTreeToDisp();

    void                setTree();
    Notifier<uiStratTreeToDisp> newtreeRead;

    int                 levelColIdx() const     { return levelcolidx_; }

protected:

    StratDispData&      data_;
    Strat::RefTree*     tree_;

    bool                withauxs_; //lithologies & descriptions
    bool                withlevels_;
    int                 lithocolidx_;
    int                 desccolidx_;
    int                 levelcolidx_;

    void                readUnits();
    void                addUnit(const Strat::NodeUnitRef&);
    void                addDescs(const Strat::LeavedUnitRef&);
    void                addLithologies(const Strat::LeavedUnitRef&);
    void                addLevel(const Strat::LeavedUnitRef&);
    void                addAnnot(const char*,Interval<float>& posrg,bool);
    void                readFromTree();

    void                triggerDataChange(CallBacker*);
    void                treeDel(CallBacker*);
};


mExpClass(uiStrat) uiStratDispToTree : public CallBacker
{
public:
                        uiStratDispToTree(uiStratRefTree&);
                        ~uiStratDispToTree();

    uiTreeViewItem*     getItemFromTree(const char*);

    void                handleUnitMenu(const char*);
    void                handleUnitProperties(const char*);
    void                setUnitLvl(const char*);
    void                addUnit(const char*);

protected:

    uiStratRefTree&     uitree_;

public:
    uiTreeViewItem*     setCurrentTreeItem(const char*);
};

#endif