seisloader.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:          July 2010
________________________________________________________________________

*/

#include "seiscommon.h"
#include "datachar.h"
#include "executor.h"
#include "paralleltask.h"
#include "survgeom.h"
#include "trckeyzsampling.h"
#include "uistring.h"

class BinDataDesc;
class BinIDValueSet;
class IOObj;
class RegularSeisDataPack;
class Scaler;

template <class T> class Array2D;
template <class T> class Array3D;
class GatherSetDataPack;

namespace Seis
{

class Provider;
class SelData;

mExpClass(Seis) Loader
{ mODTextTranslationClass(Loader)
public:

    virtual             ~Loader();

    void                setDataChar(DataCharacteristics::UserType);

    void                setComponents(const TypeSet<int>&);
    bool                setOutputComponents(const TypeSet<int>&);
    void                setComponentScaler(const Scaler&,int compidx);
    void                setScaler(const Scaler*);

    virtual RegularSeisDataPack*        getDataPack()   { return dp_; }

    virtual uiString    nrDoneText() const;
    virtual uiString    message() const                 { return msg_; }

    od_int64            totalNr() const
                        { return tkzs_.hsamp_.totalNr(); }

protected:
                        Loader(const IOObj&,const TrcKeyZSampling*,
                               const TypeSet<int>* components);

    void                adjustDPDescToScalers(const BinDataDesc& trcdesc);

    RefMan<RegularSeisDataPack> dp_;
    IOObj*              ioobj_;
    TrcKeyZSampling     tkzs_;

    DataCharacteristics dc_;
    TypeSet<int>        components_;
    ObjectSet<Scaler>   compscalers_; //Same size as components_
    TypeSet<int>*       outcomponents_;
    Scaler*             scaler_;

    uiString            msg_;
};


mExpClass(Seis) ParallelFSLoader3D : public Loader
                                   , public ParallelTask
{ mODTextTranslationClass(ParallelFSLoader3D)
public:
                        ParallelFSLoader3D(const IOObj&,const TrcKeyZSampling&);
                        ParallelFSLoader3D(const IOObj&,BinIDValueSet&,
                                           const TypeSet<int>& components);
    void                setDataPack(RegularSeisDataPack*);

    virtual uiString    nrDoneText() const;
    virtual uiString    message() const;

protected:
    od_int64            nrIterations() const            { return totalnr_; }

private:
    bool                doPrepare(int);
    bool                doWork(od_int64,od_int64,int);

    BinIDValueSet*      bidvals_;
    const od_int64      totalnr_;
};


mExpClass(Seis) ParallelFSLoader2D : public Loader
                                   , public ParallelTask
{ mODTextTranslationClass(ParallelFSLoader2D)
public:
                        ParallelFSLoader2D(const IOObj&,
                                           const TrcKeyZSampling&,
                                           const TypeSet<int>* comps=0);
    RegularSeisDataPack* getDataPack(); // The caller now owns the datapack

    virtual uiString    nrDoneText() const;
    virtual uiString    message() const;

protected:
    od_int64            nrIterations() const            { return totalnr_; }

private :
    bool                doPrepare(int);
    bool                doWork(od_int64,od_int64,int);

    bool                dpclaimed_;
    const od_int64      totalnr_;
};


mExpClass(Seis) SequentialFSLoader : public Loader
                                   , public Executor
{ mODTextTranslationClass(SequentialFSLoader)
public:
                        SequentialFSLoader(const IOObj&,
                                           const TrcKeyZSampling* =0,
                                           const TypeSet<int>* components=0);
                        ~SequentialFSLoader();

    bool                setDataPack(RegularSeisDataPack&,od_ostream* strm=0);
    od_int64            nrDone() const                  { return nrdone_; }
    virtual uiString    nrDoneText() const;
    virtual uiString    message() const;

protected:
    bool                init();
    int                 nextStep();

    Seis::Provider*     prov_;
    Seis::SelData*      sd_;
    Interval<int>       samprg_;

    int                 queueid_;

    od_int64            nrdone_;
    bool                initialized_;

};


/*
\brief Loads a Prestack Seismic Dataset into a GatherSetDataPack
*/

mExpClass(Seis) SequentialPSLoader : public Executor
{ mODTextTranslationClass(SequentialPSLoader)
public:
                        SequentialPSLoader(const IOObj&,
                                           const Interval<int>* linerg=0,
                                           Pos::GeomID geomid=mUdfGeomID);
                        ~SequentialPSLoader();

    GatherSetDataPack*  getPSDataPack()                 { return gatherdp_;}

    od_int64            totalNr() const;
    od_int64            nrDone() const                  { return nrdone_; }
    virtual uiString    nrDoneText() const;
    virtual uiString    message() const                 { return msg_; }

protected:
    bool                init();
    int                 nextStep();

    RefMan<GatherSetDataPack>   gatherdp_;

    Seis::Provider*     prov_;
    Seis::SelData*      sd_;
    Pos::GeomID         geomid_;
    IOObj*              ioobj_;

    od_int64            nrdone_;
    uiString            msg_;
};

} // namespace Seis