TPMCABC.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/>.
// $Id$
// $Author$
// $Date$
// $Revision 1.2 $
// $State$

#ifndef __PMC_ABC__
#define __PMC_ABC__

// Root
#include "TTree.h"
#include "TMatrixD.h"

// Uranie
#include "TABC.h"
#include "TDataServer.h"

namespace URANIE
{
namespace Calibration
{
    
class TPMCABC : public URANIE::Calibration::TABC
{

public :
    URANIE::DataServer::TDataServer *_tdsParStep;   
    vector<double> _frequencies;                    
    vector<double> _indices;                        
    vector<double> _vEpsilon;                       
    TMatrixD _Sigmat;                               
    int _kStep;                                     
    int _nSteps;                                    
    
    //---------------------------------------------

    TPMCABC(URANIE::DataServer::TDataServer *tds, URANIE::Relauncher::TRun *run, int ns = 1, Option_t *option = "");
    
    TPMCABC(URANIE::DataServer::TDataServer *tds, void (*fcn)(Double_t*,Double_t*),  const char *varexpinput, const char *varexpoutput, int ns = 100, Option_t *option = "");
    
    
    TPMCABC(URANIE::DataServer::TDataServer *tds, const char *fcn, const char *varexpinput, const char *varexpoutput, int ns = 100, Option_t *option = "");
    
    TPMCABC(URANIE::DataServer::TDataServer *tds, URANIE::Launcher::TCode *fcode, int ns = 1, Option_t *option = "");
    

    virtual ~TPMCABC();
    
    //---------------------------------------------
    virtual void setDistance(const char *distName, URANIE::DataServer::TDataServer *tdsRef, const char *input, const char *reference, const char *weight="");

    virtual void setDistance(URANIE::Calibration::TDistanceLikelihoodFunction *distFunc, URANIE::DataServer::TDataServer *tdsRef, const char *input, const char *reference, const char *weight="");
    

    void computeParameters(Option_t *option = "");
    
    void initPMCABC();
    void generateNewSample();
    double pdfGaussianMultivariate(TMatrixD X,TMatrixD MU,TMatrixD Sigma);
    
    int getNSteps()
    {
        return _nSteps;
    }
    
    
    URANIE::DataServer::TDataServer *getTdsStep()
    {
        return _tdsParStep;
    }
    
    void setThresholds(vector<double> vectTh)
    {
        _vEpsilon = vectTh;
        try
        {
            for(int i=0; i < (int)_vEpsilon.size(); i++)
            {
                if ((_vEpsilon[i]>1.0) or (_vEpsilon[i]<0.0))
                {
                    throw URANIE::Exceptions::UErrorExceptions(__FILE__, __LINE__,
                            Form( "TABC::setThresholds: the percentile must be set between 0 and 1: however it is fix at %4.2f at step %d ! ",_vEpsilon[i],i));
                }
            }
        }
        catch (URANIE::Exceptions::UErrorExceptions& ue)
        {
            ue.printMessage();
            throw ue;
        }
        _nSteps = (int)_vEpsilon.size();
    }
    void setNSteps(int steps)
    {
        try
        {
            if ((steps<1))
            {
                throw URANIE::Exceptions::UErrorExceptions(__FILE__, __LINE__,
                        Form( "TABC::setNSteps: the number of steps must be strictly upper than 0: however it is fix at %d ! ", steps));
            }
        }
        catch (URANIE::Exceptions::UErrorExceptions& ue)
        {
            ue.printMessage();
            throw ue;
        }
        _nSteps = steps;
    }
    
    
    ClassDef(URANIE::Calibration::TPMCABC, ID_CALIBRATION)
    
};
    
} // Fin du namespace ABC
} // Fin du namespace URANIE
#endif