seiscubeprov.h

Go to the documentation of this file.

#ifndef seismscprov_h
#define seismscprov_h

/*
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        A.H. Bril
 Date:          Jan 2007
 RCS:           $Id$
________________________________________________________________________

*/


#include "seismod.h"
#include "arraynd.h"
#include "trckeyzsampling.h"
#include "rowcol.h"
#include "objectset.h"
#include "uistring.h"

template <class T> class Array2D;
class IOObj;
class LineKey;
class SeisTrc;
class SeisTrcBuf;
class SeisTrcReader;
namespace Seis          { class SelData; }
class TaskRunner;


mExpClass(Seis) SeisMSCProvider
{ mODTextTranslationClass(SeisMSCProvider);
public:

                        SeisMSCProvider(const MultiID&);
                        SeisMSCProvider(const IOObj&);
                        SeisMSCProvider(const char* fnm);
    virtual             ~SeisMSCProvider();

    bool                is2D() const;
    bool                prepareWork();

                        // use the following after prepareWork
                        // but before the first next()
    void                forceFloatData( bool yn )
                        { intofloats_ = yn; }
    void                setStepout(int,int,bool required);
    void                setStepout(Array2D<bool>* mask);
    void                setStepoutStep( int i, int c )
                        { stepoutstep_.row() = i; stepoutstep_.col() = c; }
    int                 inlStepout( bool req ) const
                        { return req ? reqstepout_.row() : desstepout_.row(); }
    int                 crlStepout( bool req ) const
                        { return req ? reqstepout_.col() : desstepout_.col(); }
    void                setSelData(Seis::SelData*);

    enum AdvanceState   { NewPosition, Buffering, EndReached, Error };
    AdvanceState        advance();
    uiString            errMsg() const          { return errmsg_; }

    BinID               getPos() const;
    int                 getTrcNr() const;
    SeisTrc*            get(int deltainl,int deltacrl);
    SeisTrc*            get(const BinID&);
    const SeisTrc*      get( int i, int c ) const
                        { return const_cast<SeisMSCProvider*>(this)->get(i,c); }
    const SeisTrc*      get( const BinID& bid ) const
                        { return const_cast<SeisMSCProvider*>(this)->get(bid); }

    int                 comparePos(const SeisMSCProvider&) const;
    int                 estimatedNrTraces() const; 

    SeisTrcReader&      reader()                { return rdr_; }
    const SeisTrcReader& reader() const         { return rdr_; }

protected:

    SeisTrcReader&      rdr_;
    ObjectSet<SeisTrcBuf> tbufs_;
    RowCol              reqstepout_;
    RowCol              desstepout_;
    RowCol              stepoutstep_;
    Array2D<bool>*      reqmask_;
    bool                intofloats_;
    bool                workstarted_;
    enum ReadState      { NeedStart, ReadOK, ReadAtEnd, ReadErr };
    ReadState           readstate_;

    uiString            errmsg_;
    mutable int         estnrtrcs_;

                        // Indexes of new pos ready, equals -1 while buffering.
    int                 bufidx_;
    int                 trcidx_;
                        // Indexes of next position to be examined.
    int                 pivotidx_;
    int                 pivotidy_;

    void                init();
    bool                startWork();
    int                 readTrace(SeisTrc&);
    bool                isReqBoxFilled() const;
    bool                doAdvance();
};


mExpClass(Seis) SeisFixedCubeProvider
{ mODTextTranslationClass(SeisFixedCubeProvider);
public:
                        SeisFixedCubeProvider(const MultiID&);
                        ~SeisFixedCubeProvider();

    void                clear();
    bool                isEmpty() const;
    bool                readData(const TrcKeyZSampling&,TaskRunner* tr=0);
    bool                readData(const TrcKeyZSampling&,
                                 const Pos::GeomID geomid, TaskRunner* tr = 0);

    const SeisTrc*      getTrace(const BinID&) const;
    const SeisTrc*      getTrace(int trcnr) const;
    float               getTrcDist() const              { return trcdist_; }
    uiString            errMsg() const;

protected:

    Array2D<SeisTrc*>*  data_;

    TrcKeyZSampling     tkzs_;
    IOObj*              ioobj_;
    uiString            errmsg_;
    float               trcdist_;

    bool                calcTrcDist(const Pos::GeomID);
};

#endif