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

#ifndef TLINEARBAYESIAN_H
#define TLINEARBAYESIAN_H

// Uranie
#include "TCalibration.h"

namespace URANIE
{
namespace Calibration
{

class TLinearBayesian: public URANIE::Calibration::TCalibration {

protected:
    
    void (*_fTransfoParam)(double *, double*);        
    vector<double> _aPrioriMode;                      

    TMatrixD _mParCovariance;                         
    TMatrixD _mParValues;                             
    TMatrixD _mTransfoParValues;                      

    TMatrixD _mH;                                     
    vector<string> _vRegName;                         
    string _regname;                                  
public:
    
    //---------------------------------------------


    TLinearBayesian(URANIE::DataServer::TDataServer *tds, URANIE::Relauncher::TRun *run, Int_t ns=1, Option_t * option = "");

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

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

    TLinearBayesian(URANIE::DataServer::TDataServer *tds, URANIE::Launcher::TCode *fcode, int ns = 1, const char *option = "");
    
    virtual ~TLinearBayesian();

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

    void computePredictionVariance(URANIE::DataServer::TDataServer *tds_new, string outname);

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

    void setParameterTransformationFunction( void (*fTransfoParam)(double *, double*)){ _fTransfoParam = fTransfoParam; }

    void setRegressorName(const char *regName);
    
    void setLikelihood(const char *likelihoodName, URANIE::DataServer::TDataServer *tdsRef, const char *input, const char *output, const char *weight="");

    void setLikelihood(URANIE::Calibration::TDistanceLikelihoodFunction *likelihoodFunc, URANIE::DataServer::TDataServer *tdsRef, const char *input, const char *output, const char *weight="");  

    TMatrixD getParameterValueMatrix(){ return _mParValues;}

    TMatrixD getTransfParameterValueMatrix(){ return _mTransfoParValues;}
    
    TMatrixD getParameterCovarianceMatrix(){ return _mParCovariance;}
    

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

    void drawParameters(TString sTitre, const char *variable = "*", Option_t * option = "");
    
    void printLog(Option_t *option = "");
    

protected:
    void computeParameters(Option_t *option="");
    void checktdsParContent();
    
    ClassDef(URANIE::Calibration::TLinearBayesian, ID_CALIBRATION)

    
};

} // Fin du namespace Calibration
} // Fin du namespace URANIE

#endif