TkPCA.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 TkPCA_H
#define TkPCA_H
//ROOT
#include "Rtypes.h"
#include "TNamed.h"
#include "TMatrixTBase.h"
#include "TVectorT.h"

//URANIE
#include "Modeler.h"
#include "TDataServer.h"
#include "TDSNtupleD.h"
#include "TKernel.h"

namespace URANIE
{
namespace Modeler
{
class TkPCA: public TNamed
{
public:
    Bool_t _blog;                             
    Bool_t _CentData;                         
    Bool_t _CentGram;                         
    Bool_t _bCompletedFile;                   
    URANIE::DataServer::TDataServer *_tds;     
    URANIE::DataServer::TDSNtupleD * _ntdResult;          
    TKernel *_ker;                             
    // TNtupleD * _ntdLoading;                    ///< The NTupleD of Loading
    // TNtupleD * _ntdLoadingCircle;     ///< The NTupleD of Loading for the circle
    TList *_listOfInputAttributes;             
    
    TMatrixD _matK;                         
    TMatrixD _matKCent;                     
    int _nS;                                
    TVectorD _vecEigenValues;                 
    TVectorD _vecPctEigenValues; 
    // TMatrixD _matEigenVectors;                ///< The matrix of Eigenvectors

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

    TkPCA(URANIE::DataServer::TDataServer *tds, const char *varexp = "*",
            Option_t * option = "");

    virtual ~TkPCA();

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

    URANIE::DataServer::TDSNtupleD* getResultNtupleD(){
        return _ntdResult;
    }


    void setKernel(TKernel::EKernelType nKernel){
        _ker->setKernel(nKernel);
    }    
    void setKernel(TKernel::EKernelType nKernel, Double_t dparam1){
        _ker->setKernel(nKernel, dparam1);
    }

    void setKernel(TKernel::EKernelType nKernel, Double_t dparam1, Double_t dparam2){
        _ker->setKernel(nKernel, dparam1, dparam2);
    }

    void setKernel(TKernel::EKernelType nKernel, Double_t dparam1, Double_t dparam2, Double_t dparam3){
        _ker->setKernel(nKernel, dparam1, dparam2, dparam3);
    }
    

    // TDSNtupleD* getVariableResultNtupleD()
    // {
    //     return _ntdVarResult;
    // }

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

    void compute(Option_t* option = "");

    void generateKCenteredMatrix(Option_t * option);


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

    void drawkPCA(Int_t indX, Int_t indY, TString scut="", Option_t* option = "");

    //---------------------------------------------
    void setLog()
    {
        _blog = kTRUE;
    }
    void unsetLog()
    {
        _blog = kFALSE;
    }
    void changeLog()
    {
        _blog = _blog ? kFALSE : kTRUE;
    }
    Bool_t getLog()
    {
        return _blog;
    }
    virtual void printLog(Option_t *option = "RAS");

    ClassDef(URANIE::Modeler::TkPCA, ID_MODELER)
    //Definition of an variable

};

} // Fin du namespace Modeler
} // Fin du namespace URANIE

#endif