horizonmodifier.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        N. Hemstra
 Date:          April 2006
________________________________________________________________________

-*/

#include "earthmodelmod.h"
#include "emposid.h"
#include "dbkey.h"
#include "ranges.h"
#include "posinfo2dsurv.h"

namespace EM { class Horizon; }
class TrcKeySamplingIterator;

mExpClass(EarthModel) HorizonModifier
{
public:

                                HorizonModifier(bool is2d=false);
                                ~HorizonModifier();

    enum ModifyMode             { Shift, Remove };

    bool                        setHorizons(const DBKey&,const DBKey&);
    void                        setStaticHorizon(bool tophor);
    void                        setMode(ModifyMode);

    void                        doWork();

protected:

    bool                        getNextNode(BinID&);
    bool                        getNextNode3D(BinID&);
    bool                        getNextNode2D(BinID&);
    void                        getLines(const EM::Horizon*);
    float                       getDepth2D(const EM::Horizon*,const BinID&);
    void                        shiftNode(const BinID&);
    void                        removeNode(const BinID&);

    EM::Horizon*                tophor_;
    EM::Horizon*                bothor_;

    bool                        is2d_;
    TypeSet<Pos::GeomID>        geomids_;
    TypeSet<StepInterval<int> > trcrgs_;
    TrcKeySamplingIterator*     iter_;

    ModifyMode                  modifymode_;
    bool                        topisstatic_;
};