TConstrLHS.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 TCONSTRLHS_H
#define TCONSTRLHS_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 "TSampling.h"
#include "TAttribute.h"
#include "TStochAttribut.h"
#include "TDataServer.h"

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

    // Attributes
    bool                 _bAlreadyFilled;        

    TMatrixD             _currentLHS;            
    TMatrixD             _currentPerm;           
    
    vector<TMatrixD>     _currentConst;          
    vector<vector<int> > _rowSol;                
    vector<vector<int> > _colSol;                
    vector<vector<int> > _diag;                  
    vector<map<int, vector<int> > >  _modified;  
    vector<int>          _paramValues;           
    vector<int>          _nbdiag;                

    TSampling           *_sampling;              

    unsigned int        _ns;                     
    unsigned int        _nx;                     
    unsigned int        _nParam;                 
    
    unsigned int        _nConstr;                
    unsigned int        _iPerm;                  
    void (*_fConstrain)(double *, double*);      
    

    // Operations
    //---------------------------------------------

    TConstrLHS(URANIE::DataServer::TDataServer *tds, int sampleSize);

    virtual ~TConstrLHS();


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

    void generateSample(Option_t *option = "");

    void setSeed(Int_t ind = 0);

    void addConstraint( void (*fConstrain)(double *, double*), int nConstr, int nbparam, int *param );

    
private:
    
    //---------------------------------------------            
    void initElements();
    
    virtual void fillOtherAttributes();

    void createTuple();



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

    void getConstMatrix();


    void printMatrix(TMatrixD &dist);
    

    void findNextPermutation();
    

    void performPermutation(int Rcon, int Rrow, int Rcol, int Rvar);
    
    bool bpm(TMatrixD &lhs, int u, int seen[], int matchR[]);
    
    int maxBPM(TMatrixD &lhs);

    
    

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

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

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

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