PndKnnClassify.h

Go to the documentation of this file.

/* ************************************
 *  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