PndKnnClassify.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

/* ************************************
 *  Author: M. Babai (M.Babai@rug.nl) *
 *                                    *
 *  KNN based pid classifier          *
 *                                    *
 * Modified:                          *
 *                                    *
 * ************************************/
//#pragma once
#ifndef PND_KNN_CLASSIFY_H
#define PND_KNN_CLASSIFY_H

// LOCAL includes
#include "PndMvaClassifier.h"

// TMVA
#include "TMVA/NodekNN.h"
#include "TMVA/ModulekNN.h"

//____________________________________________
// typedef std::list < std::pair<const TMVA::kNN::Node<TMVA::kNN::Event>*, float> > ResList;
//____________________________________________

class PndKnnClassify : public PndMvaClassifier {
 public:
  explicit PndKnnClassify(std::string const &inputFile, std::vector<std::string> const &classNames, std::vector<std::string> const &varNames);
  virtual ~PndKnnClassify();

  void GetMvaValues(std::vector<float> eventData, std::map<std::string, float> &result);
  std::string *Classify(std::vector<float> EvtData);

  inline void SetEvtParam(float const scFact, double const weight);

  inline void SetKnn(size_t const N);

  inline size_t GetKnn();

  virtual void Initialize();

  void print() { m_module->Print(); }

  // ==================  Private ===============
 private:
  // To avoid mistakes. ;)
  PndKnnClassify(PndKnnClassify const &other);
  PndKnnClassify &operator=(PndKnnClassify const &other);

  void InitKNN();

  typedef std::list<std::pair<const TMVA::kNN::Node<TMVA::kNN::Event> *, float>> ResList;

  size_t m_knn;                   
  float m_ScaleFact;              
  double m_weight;                
  TMVA::kNN::ModulekNN *m_module; // TMVA Knn module.

  std::map<std::string, size_t> m_classIndices;
}; // End of classifier interface definition

//___________________ Inline implementation. __________________________________
inline void PndKnnClassify::SetEvtParam(float const scFact, double const weight)
{
  m_ScaleFact = scFact;
  m_weight = weight;
};

inline void PndKnnClassify::SetKnn(size_t const N)
{
  m_knn = N;
};

inline size_t PndKnnClassify::GetKnn()
{
  return m_knn;
};
#endif