specdecompattrib.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Nanne Hemstra
 Date:          Jan 2004
________________________________________________________________________
-*/

#include "attributesmod.h"
#include "attribprovider.h"
#include "arrayndalgo.h"
#include "wavelettrans.h"
#include "fourier.h"

namespace Attrib
{

class DataHolder;

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

    static const char*  attribName()            { return "SpecDecomp"; }
    static const char*  transformTypeStr()      { return "transformtype"; }
    static const char*  windowStr()             { return "window"; }
    static const char*  gateStr()               { return "gate"; }
    static const char*  deltafreqStr()          { return "deltafreq"; }
    static const char*  dwtwaveletStr()         { return "dwtwavelet"; }
    static const char*  cwtwaveletStr()         { return "cwtwavelet"; }
    static const char*  transTypeNamesStr(int);

    void                getCompNames(BufferStringSet&) const;
    bool                prepPriorToOutputSetup();

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

    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& relpos,
                                    int t0,int nrsamples,int threadid) const;
    bool                calcDFT(const DataHolder&,int t0,int nrsamples) const;
    bool                calcDWT(const DataHolder&,int t0,int nrsamples) const;
    bool                calcCWT(const DataHolder&,int t0,int nrsamples) const;

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

    bool                areAllOutputsEnabled() const;

    int                                 transformtype_;
    ArrayNDWindow::WindowType           windowtype_;
    Interval<float>                     gate_;
    float                               deltafreq_;
    WaveletTransform::WaveletType       dwtwavelet_;
    CWT::WaveletType                    cwtwavelet_;

    Interval<int>                       samplegate_;

    ArrayNDWindow*                      window_;
    Fourier::CC*                        fft_;
    CWT                                 cwt_;
    int                                 scalelen_;

    float                               df_;
    int                                 fftsz_;
    int                                 sz_;

    bool                                fftisinit_;
    Interval<int>                       desgate_;

    int                                 realidx_;
    int                                 imagidx_;

    const DataHolder*                   redata_;
    const DataHolder*                   imdata_;
};

}; // namespace Attrib