raytrace1d.h

Go to the documentation of this file.

#ifndef raytrace1d_h
#define raytrace1d_h
/*
________________________________________________________________________

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

*/

#include "algomod.h"
#include "ailayer.h"
#include "factory.h"
#include "odcomplex.h"
#include "reflectivitymodel.h"
#include "survinfo.h"
#include "paralleltask.h"
#include "velocitycalc.h"

template <class T> class Array2DImpl;
template <class T> class Array1D;
class TimeDepthModel;


mExpClass(Algo) RayTracer1D : public ParallelTask
{ mODTextTranslationClass(RayTracer1D);
public:
    mDefineFactoryInClass( RayTracer1D, factory );

    static RayTracer1D* createInstance(const IOPar&,uiString&);

                        ~RayTracer1D();

    mExpClass(Algo) Setup
    {
    public:
                        Setup()
                            : pdown_( true )
                            , pup_( true )
                            , doreflectivity_(true) {}
        virtual ~Setup()        {}

        mDefSetupMemb(bool,pdown);
        mDefSetupMemb(bool,pup);
        mDefSetupMemb(bool,doreflectivity);

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

    virtual RayTracer1D::Setup& setup()         = 0;
    virtual const RayTracer1D::Setup& setup() const     = 0;
    virtual bool        hasSameParams(const RayTracer1D&) const;

    bool                setModel(const ElasticModel&);
    const ElasticModel& getModel() const        { return model_; }
                        // model top depth must be TWT = 0ms
    void                setOffsets(const TypeSet<float>& offsets);
    void                getOffsets(TypeSet<float>& offsets) const;
    bool                isPSWithoutZeroOffset() const;
                        // If PreStack & Offset Range do not have zero offset

    uiString            errMsg() const { return errmsg_; }

                        //Available after execution
    float               getSinAngle(int layeridx,int offsetidx) const;
    bool                getReflectivity(int offset,ReflectivityModel&) const;
    bool                getTDModel(int offset,TimeDepthModel&) const;
    bool                getZeroOffsTDModel(TimeDepthModel&) const;

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

    static const char*  sKeyPWave()        { return "Wavetypes"; }
    static const char*  sKeyOffset()       { return "Offset Range"; }
    static const char*  sKeyReflectivity() { return "Compute reflectivity"; }
    static const char*  sKeyBlock()        { return "Block model"; }
    static const char*  sKeyBlockRatio()   { return "Blocking ratio threshold";}
    static float        cDefaultBlockRatio();

    static void         setIOParsToZeroOffset(IOPar& iop);

protected:
                        RayTracer1D();

    od_int64            nrIterations() const;
    virtual bool        doPrepare(int);
    virtual bool        compute(int,int,float);
    bool                getTDM(const Array1D<float>&,TimeDepthModel&) const;
    void                setZeroOffsetTWT();

                        //Setup variables
    ElasticModel        model_; // model top depth must be TWT = 0ms
    TypeSet<float>      offsets_;
    uiString            errmsg_;

                        //Runtime variables
    TypeSet<int>        offsetpermutation_;
    TypeSet<float>      velmax_;
    TypeSet<float>      depths_;

                                //Results
    Array2DImpl<float>*         sini_;
    Array2DImpl<float>*         twt_;
    Array1D<float>*             zerooffstwt_;
    Array2DImpl<float_complex>* reflectivity_;

public:

    float               getDepth(int layer) const;
    float               getTime(int layer,int offset) const;
};


mExpClass(Algo) VrmsRayTracer1D : public RayTracer1D
{ mODTextTranslationClass(VrmsRayTracer1D);
public:

    mDefaultFactoryInstantiation( RayTracer1D, VrmsRayTracer1D,
                                  "VrmsRayTracer",
                                  tr("Simple RayTracer") );

    RayTracer1D::Setup&         setup() { return setup_; }
    const RayTracer1D::Setup&   setup() const   { return setup_; }

protected:
    bool                        doWork(od_int64,od_int64,int);

    bool                        compute(int,int,float);

    RayTracer1D::Setup          setup_;
};


#endif