horizon2dextender.h

Go to the documentation of this file.

#ifndef horizon2dextender_h
#define horizon2dextender_h

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        K. Tingdahl
 Date:          May 2006
 RCS:           $Id$
________________________________________________________________________

-*/

#include "mpeenginemod.h"
#include "sectionextender.h"

#include "coord.h"
#include "posinfo2dsurv.h"
#include "trckeyvalue.h"


namespace EM { class Horizon2D; }

namespace MPE
{

mExpClass(MPEEngine) Horizon2DExtender : public SectionExtender
{
public:
                                Horizon2DExtender(EM::Horizon2D&,
                                                  EM::SectionID);
    static SectionExtender*     create(EM::EMObject*,EM::SectionID);
    static void                 initClass();

    void                        setAngleThreshold(float radians);
    float                       getAngleThreshold() const;

    void                        setDirection(const TrcKeyValue&);
    const TrcKeyValue*          getDirection() const;
    void                        setGeomID(Pos::GeomID);
    Pos::GeomID                 geomID() const;

    int                         nextStep();

protected:

    void                        addNeighbor(bool upwards,
                                            const TrcKey& sourcesid);
    virtual float               getDepth(const TrcKey& src,
                                         const TrcKey& target) const;
    virtual void                prepareDataIfRequired()         {}

    float                       anglethreshold_;
    bool                        alldirs_;
    Coord                       xydirection_;
    TrcKeyValue                 direction_;
    EM::Horizon2D&              hor2d_;
    Pos::GeomID                 geomid_;
};

} // namespace MPE

#endif