seisbayesclass.h

Go to the documentation of this file.

#ifndef seisbayesclass_h
#define seisbayesclass_h

/*
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bert
 Date:          Feb 2010
 RCS:           $Id$
________________________________________________________________________

*/

#include "seismod.h"
#include "executor.h"
#include "bufstringset.h"
class SeisTrc;
class SeisTrcBuf;
class ProbDenFunc;
class SeisTrcReader;
class SeisTrcWriter;


mExpClass(Seis) SeisBayesClass : public Executor
{ mODTextTranslationClass(SeisBayesClass)
public:

                                SeisBayesClass(const IOPar&);
                                ~SeisBayesClass();

    static const char*          sKeyPDFID();
    static const char*          sKeyAPProbID();
    static const char*          sKeySeisInpID();
    static const char*          sKeySeisOutID();
    static const char*          sKeyPreNorm();
    static const char*          sKeyPostNorm();
    static const char*          sKeyPreScale();

    int                         nextStep();
    uiString                    uiMessage() const;
    uiString                    uiNrDoneText() const;
    od_int64                    nrDone() const;
    od_int64                    totalNr() const;

protected:

    bool                        is2d_;
    ObjectSet<ProbDenFunc>      inppdfs_;
    ObjectSet<ProbDenFunc>      pdfs_;
    ObjectSet<SeisTrcReader>    rdrs_;
    ObjectSet<SeisTrcReader>    aprdrs_;
    ObjectSet<SeisTrcWriter>    wrrs_;
    SeisTrcBuf&                 inptrcs_;
    SeisTrcBuf&                 aptrcs_;
    SeisTrcBuf&                 outtrcs_;
    const IOPar&                pars_;
    ObjectSet< TypeSet<int> >   pdfxtbls_;
    bool                        doprenorm_;
    bool                        dopostnorm_;
    TypeSet<float>              prescales_;

    const int                   nrdims_;
    const int                   nrpdfs_;
    const bool                  needclass_;
    od_int64                    nrdone_;
    od_int64                    totalnr_;
    uiString                    msg_;
    BufferStringSet             pdfnames_;
    int                         initstep_;
    mutable TypeSet<float>      pdfinpvals_;

    bool                        getPDFs();
    void                        preScalePDFs();
    void                        postScaleProbs();
    bool                        getReaders();
    bool                        getWriters();

    SeisTrcReader*              getReader(const char*,bool,int);
    int                         readInpTrcs();
    int                         createOutput();
    int                         closeDown();

    float                       getPDFValue(int ipdf,int isamp,int icomp,
                                            bool inp=false) const;
    float                       getAPTrcVal(int ipdf,int isamp,int icomp);
    void                        calcPerBinProbs();
    void                        calcProbs(int);
    void                        calcClass();
    void                        calcDet();
    void                        cleanUp();
    void                        prepOutTrc(SeisTrc&,bool) const;
    void                        getClass(const TypeSet<float>&,int&,
                                         float&) const;

};

#define mGetSeisBayesKey(ky,nr) \
    IOPar::compKey(SeisBayesClass::sKey##ky(),nr)
#define mGetSeisBayesIDKey(ky,nr) mGetSeisBayesKey(ky##ID,nr)

#define mGetSeisBayesPreScaleKey(nr) mGetSeisBayesKey(PreScale,nr)
#define mGetSeisBayesPDFIDKey(nr) mGetSeisBayesIDKey(PDF,nr)
#define mGetSeisBayesAPProbIDKey(nr) mGetSeisBayesIDKey(APProb,nr)
#define mGetSeisBayesSeisInpIDKey(nr) mGetSeisBayesIDKey(SeisInp,nr)
#define mGetSeisBayesSeisOutIDKey(nr) mGetSeisBayesIDKey(SeisOut,nr)


#endif