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