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$.