TBasicSampling.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 TBASICSAMPLING_H
#define TBASICSAMPLING_H

#ifndef ROOT_TMatrixD
#include "TMatrixD.h"
#endif
#ifndef ROOT_TVectorD
#include "TVectorD.h"
#endif
#ifndef ROOT_TNtuple
#include "TNtuple.h"
#endif
#ifndef ROOT_TString
#include "TString.h"
#endif
#ifndef ROOT_TNtupleD
#include "TNtupleD.h"
#endif
#ifndef ROOT_TNamed
#include "TNamed.h"
#endif

// Uranie
#include "TSamplerStochastic.h"
#include "TAttribute.h"
#include "TStochAttribut.h"
#include "TDataServer.h"
#include "LHSInput.H"

namespace URANIE
{
namespace Sampler
{
class TBasicSampling: public TSamplerStochastic
{
    // Associations

    // Attributes
public:
    enum EType
    {
        kLHS, kSRS, kUnknown
    };
    LHSInput* _fLhsInput;
    EType _ntype;          
    TMatrixD userCorrMat;     
    TMatrixD _corrMat;        
    TList* _listOfSimilar; 
    TList* _listOfFormulas; 
    Double_t _simTolerance; 
    Bool_t bApplyUserCorrelation; 

    // Operations
public:
    //---------------------------------------------

    TBasicSampling(URANIE::DataServer::TDataServer *tds, Option_t *option =
            "lhs", Int_t sampleSize = 1000);

    virtual ~TBasicSampling();

    //---------------------------------------------
    void setTolerance(Double_t tolerance)
    {
        _simTolerance = tolerance;
    }

    void setCorrelationMatrix(TMatrixD corrMat);

    void setUserCorrelation(TString attName1, TString attName2,
            Double_t corrValue);

    void findSimilar();

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

    void generateCorrSample(Option_t *option="");


    void generateSample(Option_t *option = "");

    virtual void fillOtherAttributes();

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

    virtual void printLog(Option_t *option = "");

    ClassDef(URANIE::Sampler::TBasicSampling, ID_SAMPLER)
    //Definition d'un plan d'exp�rience
};

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