raytrace1d.h

Go to the documentation of this file.

#pragma once
/*
________________________________________________________________________
 
 (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 const char*  sKeyOffsetInFeet() { return "Offset in Feet";}
    static float        cDefaultBlockRatio();
 
    static StepInterval<float>  sDefOffsetRange();
 
    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_;
};