TImportanceSampling.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 TImportanceSampling_H
#define TImportanceSampling_H

//ROOT
#include "Rtypes.h"
#include "TLeaf.h"
#include "TMath.h"

// uranie
#include "TDataServer.h"
#include "UExceptions.h"
#include "TStochAttribut.h"
#include "TSamplerStochastic.h"
#include "TSampling.h"

namespace URANIE
{
namespace Sampler
{
class TImportanceSampling: public TSamplerStochastic
{

public:
    enum EQuantileMethod
    {
        kQuantile1, kQuantile2, kQuantile3, kQuantile4, kQuantileUnknown
    };
    URANIE::DataServer::TDataServer * _tds_IS;
    URANIE::DataServer::TStochasticAttribute * _x;
    URANIE::DataServer::TStochasticAttribute * _is_x;
    Double_t _mean;
    Double_t _std;
    Double_t _proba;

protected:
    Bool_t _blog;
    TSampling *_sampling;

public:
    TImportanceSampling(URANIE::DataServer::TDataServer *tds, TString xname,
            URANIE::DataServer::TStochasticAttribute *att, Int_t ns);
    virtual ~TImportanceSampling();

    //---------------------------------------------
    URANIE::DataServer::TDataServer *getTDS()
    {
        return _tds_IS;
    }

    //---------------------------------------------
    Double_t computeMean(TString u);
    Double_t computeStd(TString u);
    Double_t computeThreshold(TString u, Double_t dThreshold);
    Double_t computeQuantile(TString sVar, Double_t dAlpha,
            EQuantileMethod nMethod = kQuantile2);


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

    //---------------------------------------------
    void createListOfAttributes();

    //---------------------------------------------
    void setLog()
    {
        _blog = kTRUE;
    }
    void unsetLog()
    {
        _blog = kFALSE;
    }
    void changeLog()
    {
        _blog = _blog ? kFALSE : kTRUE;
    }
    Bool_t getLog()
    {
        return _blog;
    }
    virtual void printLog(Option_t *option = "");

    ClassDef(URANIE::Sampler::TImportanceSampling, ID_DATASERVER)
    //Definition of an variable

};

} // Fin du namespace DataServer
} // Fin du names

#endif