TkNN.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 TKNN_H
#define TKNN_H


// Uranie
#include "TModeler.h"


namespace URANIE {
  namespace Modeler {
    enum EDistance { kEuclidean, kL1, kManhattan, kChebyshev, kMahalanobis };  
    class TkNN : public TModeler {
    public:
      Double_t     _dr2a;                             
      Int_t        _nK;                               
      EDistance    _nDistance;                        
      TString _sDraw;                                 
      URANIE::DataServer::TDataServer *_tdsQuery;     
      
      // Operations
    public:
    //---------------------------------------------
      TkNN(URANIE::DataServer::TDataServer *tds, TString architecture, Option_t *option ="");
      /*
        \param tds (URANIE::DataServer::TDataServer *) the dataserver
        \param varexpinput(const char*)[""] The list of input attributes to pass to the function separated by the caracter ":"
        \param varexpoutput (const char *)[""]  The list of output attributes separated by the caracter ":"
        \param 
       */
      TkNN (URANIE::DataServer::TDataServer *tds, const char *varexpinput, const char *varexpoutput, Int_t nkernel, Option_t * option="");
      virtual ~TkNN();
      
      //---------------------------------------------

      void setK( Int_t k) { _nK = k;}
      Int_t getK() { return _nK;}

      void setQueryPoints(URANIE::DataServer::TDataServer *tds);
      
      
      //---------------------------------------------
      void draw(const char * vars, Option_t * option = "");
#ifdef WITH_GPUANN
      void estimate(Option_t * option = "", Bool_t useGPU = true);
#else
      void estimate(Option_t * option = "", Bool_t useGPU = false);
#endif


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

      void exportModelCplusplus(std::ofstream * sourcefile) const;

      void exportModelFortran(std::ofstream * sourcefile) const;

      void exportModelPMML(const char* file = "",
                           const char* name = "",Option_t *option = "") const
      {
          TString soption = option;
          soption.ToLower();
          UNUSED(file,name);
          std::cerr
                  << "   --- Method TkNN::exportModelPMML(ofstream * sourcefile) not yet implemented"
                  << endl;
          throw URANIE::Exceptions::UWarningExceptions(__FILE__, __LINE__,
              "TkNN::exportFunction\n The PMML language is not yet implemented.");
      };
      void exportModelPython(std::ofstream * sourcefile) const
      {
          UNUSED(sourcefile);
          std::cerr
                  << "   --- Method TkNN::exportModelPython(ofstream * sourcefile) not yet implemented"
                  << endl;
          throw URANIE::Exceptions::UWarningExceptions(__FILE__, __LINE__,
              "TkNN::exportFunction\n The Python language is not yet implemented.");
      };

      

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

    private:
      void estimate_cpu(Option_t * option = "");

      void LSort(int n, double* array, int* index, Int_t nbthreads);

      void beginLoop(Int_t iq, Int_t nsq);
      void endLoop(Int_t iq, int nsr, double* vecDistance, int* vecIndex, Float_t *vecReferenceOutput, Double_t *tabValue, TBranch *tby, Int_t nbthreads);

#ifdef WITH_GPUANN
    void estimate_GPU(Option_t * option = "");
#endif


      ClassDef(URANIE::Modeler::TkNN, ID_MODELER)       //Classe de
        };
    
  } // Fin du namespace Modeler
} // Fin du namespace URANIE
#endif