TNeuralGas.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 TNeuralGas_H
#define TNeuralGas_H
 
// ROOT
#include "TMatrixD.h"
 
// Uranie
#include "TVQ.h"
 
namespace URANIE
{
namespace Sampler
{
class TNeuralGas: public TVQ
{
    // Attributes
public:
    TMatrixD _datamatrix;                         
 
    TMatrixD _dvw;                                
 
    TString _svardraw;
    Int_t _ntemperature;
    Double_t _dtmax;                           
    Double_t _dinitlambda;           
    Double_t _dlambda;               
    Double_t _dfinalambda;             
    Double_t _dinitalpha;              
    Double_t _dalpha;                  
    Double_t _dfinalalpha;              
 
public:
    //---------------------------------------------
 
    TNeuralGas(URANIE::DataServer::TDataServer *tds, Option_t *option,
            Int_t ns);
    virtual ~TNeuralGas();
 
    //---------------------------------------------
 
    void setLambda(Double_t dinitlambda, Double_t dfinallambda)
    {
        _dinitlambda = dinitlambda;
        _dfinalambda = dfinallambda;
    }
 
    void setAlpha(Double_t dinitalpha, Double_t dfinalalpha)
    {
        _dinitalpha = dinitalpha;
        _dfinalalpha = dfinalalpha;
    }
 
    void setTMax(Int_t dtmax)
    {
        _dtmax = 1.0 * dtmax;
    }
 
    Int_t getTMax()
    {
        return (Int_t)(_dtmax);
    }
 
    //---------------------------------------------
 
    void init();
    void learn(Int_t index);
 
    //---------------------------------------------
 
    void generateSample(Option_t *option = "");
 
    //---------------------------------------------
    virtual void printLog(Option_t *option = "");
 
    ClassDef(URANIE::Sampler::TNeuralGas, ID_SAMPLER)
    //Classe abstraite de g�n�ration de plan d'exp�rience num�rique
};
 
} // Fin du namespace Sampler
} // Fin du namespace URANIE
#endif
// fin du fichier $RCSfile$.