TMetropHasting.h

Go to the documentation of this file.

// Copyright (C) 2013-2024 CEA/DES
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or any
// later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#ifndef TMetropHasting_H
#define TMetropHasting_H

// Uranie
#include "TCalibration.h"

namespace URANIE
{
namespace Calibration
{

class TMetropHasting: public URANIE::Calibration::TCalibration
{

public:
    
    // Attributes
    vector<double>                                       _values;         
    vector<double>                                       _vstd;           
    vector<double>                                       _dBaseSaved;     
    vector<int>                                          _acceptSaved;    
    vector<int>                                          _rejectSaved;    

    TRandom3                                             _randSaved;      

    int                                                  _burnin;         
    int                                                  _lag;            
    int                                                  _nbDump;         
    double                                               _lowAccRange;    
    double                                               _higAccRange;    
    bool                                                 _mbGDVersion;    
    bool                                                 _bcleaningAtt;   

    //---------------------------------------------

    TMetropHasting(URANIE::DataServer::TDataServer *tds, URANIE::Relauncher::TRun *run, int nS=100, Option_t *option = "lhs");

    TMetropHasting(URANIE::DataServer::TDataServer *tds, void (*fcn)(Double_t*,Double_t*), const char *varexpinput, const char *varexpoutput, int ns = 100, Option_t *option = "");

    TMetropHasting(URANIE::DataServer::TDataServer *tds, const char *fcn, const char *varexpinput, const char *varexpoutput, int ns = 100, Option_t *option = "");

    TMetropHasting(URANIE::DataServer::TDataServer *tds, URANIE::Launcher::TCode *fcode, int ns = 100, Option_t *option = "");
    
    virtual ~TMetropHasting();

    //---------------------------------------------
    
    void computeParameters(Option_t *option = "");

    void continueCalculation(int new_Ns);

    void exportMHData(const char *fileName);

    void readMHData(const char *fileName);

    void checktdsParContent();


    //---------------------------------------------

    void setBurnin( int burn ){ _burnin = burn; }
    void setLag( int lag ){ _lag = lag; }
    void setNbDump( int nbDump ){ _nbDump = nbDump; }

    void setAcceptationRatioRange( double lower, double higher);

    string getDefaultCut(){
        string cut = "";
        if( _burnin != -1 )
            cut += string(Form("%s > %d",_tdsPar->getIteratorName(), _burnin));
        if( _lag != -1 )
            cut += ((cut=="")? "": " && " ) + string(Form("%s %% %d",_tdsPar->getIteratorName(), _lag));

        if(cut!="")
            cout<<"TMetropHasting drawing method apply default cut ["<<cut<<"]"<<endl;
        return cut;
    }

    void clearDefaultCut(){ setBurnin(-1); setLag(-1); }

    void setInitialisation(int n, double *values, double *standDev);

    void setInitialisation(vector<double> values, vector<double> standDev);

    void setDistanceAndReference(const char *funcName, URANIE::DataServer::TDataServer *tdsRef, const char *input, const char *output, const char *weight="");

    void setDistanceAndReference(URANIE::Calibration::TDistanceFunction *distFunc, URANIE::DataServer::TDataServer *tdsRef, const char *input, const char *output, const char *weight="");
    
    
    //---------------------------------------------
     
    void drawTrace(TString sTitre, const char *variable="*", const char *select="1>0", Option_t *option="");

    void drawAcceptationRatio(TString sTitre, const char *variable="*", const char *select="1>0", Option_t *option="");

    void drawParameters(TString sTitre, const char *variable = "*", const char *select = "1>0", Option_t * option = "");

    void getAutoCorrelation(vector<int> l, vector<double> *out, int cut=0);
    
    //---------------------------------------------
    virtual void printLog(Option_t *option = "");
    
protected:
    //---------------------------------------------

    void logPriorPdf(double &ret);


    void parseOption(Option_t *option = "");


    ClassDef(URANIE::Calibration::TMetropHasting, ID_CALIBRATION)
    //Classe
};
}
}


#endif