frequencyattrib.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Kristofer Tingdahl
 Date:          07-10-1999
________________________________________________________________________

-*/

#include "attributesmod.h"
#include "attribprovider.h"
#include "bufstringset.h"
#include "fourier.h"
#include "mathfunc.h"
#include "valseries.h"
#include "valseriesinterpol.h"

#include <complex>


class ArrayNDWindow;
template<class T> class Array1DImpl;
class FFTFilter;

namespace Attrib
{

class DataHolder;

mExpClass(Attributes) Frequency : public Provider
{
public:
    static void                 initClass();
                                Frequency(Desc&);

    static const char*          attribName()            { return "Frequency"; }
    static const char*          gateStr()               { return "gate"; }
    static const char*          normalizeStr()          { return "normalize"; }
    static const char*          windowStr()             { return "window"; }
    static const char*          paramvalStr()           { return "paramval"; }
    static const char*          dumptofileStr()         { return "dumptofile"; }
    static const char*          smoothspectrumStr()     { return "smoothspect";}

    void                        prepPriorToBoundsCalc();

protected:
                                ~Frequency();
    static Provider*            createInstance(Desc&);
    static void                 updateDesc(Desc&);
    static void                 updateDefaults(Desc&);

    bool                        checkInpAndParsAtStart();
    bool                        allowParallelComputation() const
                                { return false; }
    bool                        getInputOutput(int input,
                                               TypeSet<int>& res) const;
    bool                        getInputData(const BinID&,int idx);
    bool                        computeData(const DataHolder&,const BinID& rel,
                                            int z0,int nrsamples,
                                            int threadid) const;

    const Interval<float>*      reqZMargin(int input,int output) const;

    Interval<float>             gate_;
    Interval<int>               samplegate_;
    bool                        dumptofile_;
    BufferString                windowtype_;
    bool                        smoothspectrum_;

    //smoothspectrum_==true
    FFTFilter*                  fftfilter_;
    //else
    ArrayNDWindow*              window_;
    bool                        fftisinit_;
    int                         fftsz_;
    Fourier::CC*                fft_;
    Array1DImpl<float_complex>* timedomain_;
    Array1DImpl<float_complex>* freqdomain_;

    bool                        normalize_;
    float                       variable_;

    const DataHolder*           redata_;
    const DataHolder*           imdata_;
    int                         realidx_;
    int                         imagidx_;

    BufferStringSet             dumpset_;

    Array1DImpl<float_complex>* signal_;

    mExpClass(Attributes) FreqFunc : public FloatMathFunction
    {
    public:
                        FreqFunc(const ValueSeries<float>& func, int sz)
                            : func_( func )
                            , sz_(sz)                   {}

    virtual float       getValue( float x ) const
                        {
                                ValueSeriesInterpolator<float> interp(sz_);
                                return interp.value(func_,x);
                        }
    virtual float       getValue( const float* p ) const
                        { return getValue(*p); }

    protected:

        const ValueSeries<float>& func_;
        int sz_;

    };
};

}; // namespace Attrib