posinfo2d.h

Go to the documentation of this file.

#ifndef posinfo2d_h
#define posinfo2d_h

/*+
________________________________________________________________________

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

-*/

#include "basicmod.h"
#include "typeset.h"
#include "coord.h"
#include "bufstring.h"
#include "od_iosfwd.h"


namespace PosInfo
{

mExpClass(Basic) Line2DPos
{
public:

                Line2DPos( int n=0 ) : nr_(n)           {}
    bool        operator ==( const Line2DPos& p ) const { return nr_ == p.nr_; }
    bool        operator !=( const Line2DPos& p ) const { return nr_ != p.nr_; }
    bool        operator >( const Line2DPos& p ) const  { return nr_ > p.nr_; }
    bool        operator <( const Line2DPos& p ) const  { return nr_ < p.nr_; }
    bool        operator >=( const Line2DPos& p ) const { return nr_>=p.nr_; }
    bool        operator <=( const Line2DPos& p ) const { return nr_<=p.nr_; }

    int         nr_;
    Coord       coord_;

};


mExpClass(Basic) Line2DPos3D : public Line2DPos
{
public:
                Line2DPos3D( int n=0, float z=mUdf(float) )
                    : Line2DPos(n), z_(z)               {}

    float       z_;
};


mExpClass(Basic) Line2DData
{
public:
                        Line2DData(const char* lnm=0);

                        Line2DData(const Line2DData& l2d)
                        : zrg_(l2d.zRange())
                        , lnm_(l2d.lineName())
                        , posns_(l2d.positions()){}

    const StepInterval<float>& zRange() const           { return zrg_; }
    const OD::String&   lineName() const                { return lnm_; }
    void                setZRange( const StepInterval<float>& zrg )
                                                        { zrg_ = zrg; }
    void                setLineName( const char* lnm )  { lnm_ = lnm; }
    int                 size() const                    { return posns_.size();}
    const TypeSet<Line2DPos>&  positions() const        { return posns_; }
    bool                isEmpty() const         { return posns_.isEmpty(); }

    void                add(const Line2DPos&);
    void                remove(int trcnr);
    void                setEmpty()              { posns_.erase(); }
    void                limitTo(Interval<int> trcrg);

    int                 indexOf(int trcnr) const;
    int                 nearestIdx( const Coord& crd ) const
                                                { return gtIndex(crd); }
    int                 nearestIdx(const Coord&,
                                   const Interval<int>& trcnrrg) const;

    bool                getPos(const Coord& crd,Line2DPos& l2p,
                               float* dist=0) const;
    bool                getPos(const Coord& crd,Line2DPos& l2p,
                               float threshold_distance) const;
    bool                getPos(int trcnr,Line2DPos&) const;

    void                dump(od_ostream&,bool pretty=true) const;
    bool                read(od_istream&,bool asc);
    bool                write(od_ostream&,bool asc,bool newlns=false) const;

    const TypeSet<int>& getBendPoints() const;
    void                getBendPositions(TypeSet<Line2DPos>&) const;
    void                setBendPoints(const TypeSet<int>&);

    StepInterval<Pos::TraceID>  trcNrRange() const;
    Coord               getNormal(int trcnr) const;
    void                compDistBetwTrcsStats(float& max,float& median) const;
    float               distBetween(int startnr,int stopnr) const;

    bool                coincidesWith(const Line2DData&) const;
protected:

    StepInterval<float> zrg_;
    BufferString        lnm_;
    TypeSet<Line2DPos>  posns_;
    TypeSet<int>        bendpoints_;

    int                 gtIndex(int,bool&) const;
    int                 gtIndex(const Coord&,double* sqdist=0) const;
    int                 getSegmentIndexClosestToPoint(const Coord&) const;

    friend class        Line2DDataIterator;

};


mExpClass(Basic) Line2DDataIterator
{
public:

                        Line2DDataIterator( const Line2DData& ld )
                            : ld_(ld), idx_(-1) {}

    inline bool         next()
                        {
                            idx_++;
                            return idx_ < ld_.posns_.size();
                        }

    inline void         reset()         { idx_ = -1; }
    inline const Line2DPos& line2DPos() const { return ld_.posns_[idx_]; }
    inline int          trcNr() const
                        { return idx_>=0 ? ld_.posns_[idx_].nr_ : mUdf(int); }
    inline void         setTrcNr( int trcnr )
                        { idx_ = ld_.indexOf( trcnr ); }

    const Line2DData&   ld_;
    int                 idx_;

};


} // namespace PosInfo

#endif