TSampler.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$
// $State$

#ifndef TSampler_H
#define TSampler_H

// Writes log
#include <iostream>

#include "Rtypes.h"
#include "TNamed.h"
#include "TString.h"

// Uranie
#include "TDataServer.h"
#include "TDSNtupleD.h"
#include "UExceptions.h"
#include "Sampler.h"

using std::cout;
using std::endl;

namespace URANIE
{
namespace Sampler
{
class TSampler: public TNamed
{

    // Attributes
public:
    Int_t _nS;                             
    Int_t _nX;                             
    URANIE::DataServer::TDSNtupleD* _ntsample;          
    TString _sMethod;                       
    Bool_t _blog;                              
    Bool_t _bupdateFile;                       
    URANIE::DataServer::TDataServer *_tds;      
    TList * _lstOfAttributesToSample; 

    // Operations
public:
    //---------------------------------------------
    TSampler(URANIE::DataServer::TDataServer *tds, Option_t *option,
            Int_t nCalcul);
    virtual ~TSampler();

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

    Int_t GetID()
    {
        return ID_SAMPLER;
    }
    void setMethodName(TString str)
    {
        _sMethod = str;
    }
    TString getMethodName()
    {
        return _sMethod;
    }

    virtual URANIE::DataServer::TDataServer *getTDS()
    {
        return _tds;
    }

    void parseOption(Option_t * option);

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

    //---------------------------------------------
    virtual void createListOfAttributes();
    virtual void createTuple();
    URANIE::DataServer::TDSNtupleD * getTuple()
    {
        return _ntsample;
    }
    virtual void fillOtherAttributes();

    //---------------------------------------------
    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::TSampler, ID_SAMPLER)
};

} // Fin du namespace Sampler
} // Fin du namespace URANIE
#endif
// fin du fichier $RCSfile$.