emhorizon2d.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Kristofer Tingdahl
 Date:          4-11-2002
________________________________________________________________________


-*/

#include "earthmodelmod.h"
#include "emhorizon.h"
#include "bufstringset.h"
#include "horizon2dline.h"
#include "posinfo2dsurv.h"

class ZAxisTransform;
namespace Geometry { class Horizon2DLine; }
namespace Table { class FormatDesc; }
template <class T> class Array1D;

namespace EM
{
class EMManager;

mExpClass(EarthModel) Horizon2DGeometry : public HorizonGeometry
{
public:
                                Horizon2DGeometry(Surface&);
    Geometry::Horizon2DLine*    sectionGeometry(const SectionID&);
    const Geometry::Horizon2DLine* sectionGeometry(const SectionID&) const;

    int                         nrLines() const;

    int                         lineIndex(Pos::GeomID geomid) const;

    int                         lineIndex(const char* linenm) const;
    const char*                 lineName(int id) const;
    const char*                 lineSet(int id) const;
    Pos::GeomID                 geomID(int idx) const;

    PosID                       getPosID(const TrcKey&) const;
    TrcKey                      getTrcKey(const PosID&) const;

    bool                        includeLine(Pos::GeomID geomid,int step=1);

    bool                        addLine(Pos::GeomID geomid,int step=1);

    bool                        addLine(Pos::GeomID geomid,
                                        const StepInterval<int>& trcrg);

    void                        removeLine(Pos::GeomID geomid);

    bool                        isAtEdge(const PosID&) const;
    PosID                       getNeighbor(const PosID&,bool nextcol,
                                            bool retundef=false) const;
    static const char*          sKeyLineIDs()   { return "Line IDs"; }
    static const char*          sKeyLineNames() { return "Line names"; }
    static const char*          sKeyLineSets()  { return "Set ID of line "; }
    static const char*          sKeyID()        { return "ID"; }
    static const char*          sKeyTraceRange()
                                { return "Trace Range of line "; }
    static const char*          sKeyTrcRg()     { return "Trace range"; }
    static const char*          sKeyNrLines()   { return "Nr of Lines"; }

    int                         getConnectedPos(const PosID& posid,
                                                TypeSet<PosID>* res) const;
    StepInterval<int>           colRange(const SectionID&,
                                         Pos::GeomID geomid) const;

    StepInterval<int>           colRange(Pos::GeomID geomid) const;

protected:
    Geometry::Horizon2DLine*    createSectionGeometry() const;

    bool                        doAddLine(Pos::GeomID geomid,
                                          const StepInterval<int>& trcrg,
                                          bool mergewithdouble);

    void                        fillPar(IOPar&) const;
    bool                        usePar(const IOPar&);

    TypeSet<Pos::GeomID>        geomids_;
};


mExpClass(EarthModel) Horizon2D : public Horizon
{   mDefineEMObjFuncs( Horizon2D );
    mODTextTranslationClass( Horizon2D );
public:

    virtual float               getZ(const TrcKey&) const;
    virtual bool                setZ(const TrcKey&,float z,bool addtohist,
                                     NodeSourceType type=Auto);
    virtual bool                hasZ(const TrcKey&) const;
    virtual Coord3              getCoord(const TrcKey&) const;
    virtual void                setAttrib(const TrcKey&,int attr,bool yn,
                                          bool addtohist);
    virtual bool                isAttrib(const TrcKey&,int attr) const;

    virtual float               getZValue(const Coord&,bool allow_udf=true,
                                          int nr=0) const;

    bool                        unSetPos(const EM::PosID&,bool addtohist);
    bool                        unSetPos(const EM::SectionID&,const EM::SubID&,
                                         bool addtohist);
    Coord3                      getPos(const EM::PosID&) const;
    Coord3                      getPos(const EM::SectionID&,
                                       const EM::SubID&) const;
    TypeSet<Coord3>             getPositions(int lineidx,int trcnr) const;
    Coord3                      getPosition(EM::SectionID,int lidx,
                                            int trcnr) const;

    Coord3                      getPos(EM::SectionID,Pos::GeomID,int trc) const;

    virtual void                setNodeSourceType(const TrcKey&,
                                        NodeSourceType);
    virtual bool                isNodeSourceType(const PosID&,
                                        NodeSourceType)const;
    virtual bool                isNodeSourceType(const TrcKey&,
                                        NodeSourceType)const;

    bool                        setZPos(EM::SectionID,Pos::GeomID geomid,
                                       int trcnr,float z,bool addtohist,
                                       NodeSourceType type=Auto);

    Horizon2DGeometry&          geometry()              { return geometry_; }
    const Horizon2DGeometry&    geometry() const        { return geometry_; }

    virtual void                removeAll();
    void                        removeSelected(const Selector<Coord3>& selector,
                                               TaskRunner* tskr );

    bool                        setArray1D(const Array1D<float>&,
                                           const StepInterval<int>& trcrg,
                                           SectionID sid,Pos::GeomID geomid,
                                           bool onlyfillundefs);
    bool                        setArray1D(const Array1D<float>&,SectionID sid,
                                           Pos::GeomID geomid,
                                           bool onlyfillundefs );

    Array1D<float>*             createArray1D(SectionID,
                                              Pos::GeomID geomid,
                                              const ZAxisTransform* =0) const;

    TrcKey::SurvID              getSurveyID() const;
    uiString                    getUserTypeStr() const { return userTypeStr(); }
    static uiString             userTypeStr() { return tr("2D Horizon"); }
    void                        setSelectionColor(const Color&);
    const Color&                getSelectionColor() const;

protected:

    const IOObjContext&         getIOObjContext() const;
    void                        initNodeSourceArray(const TrcKey&);

    Horizon2DGeometry           geometry_;
    Color                       selectioncolor_;
    Array1D<char>*              nodesource_;
};


} // namespace EM