PndProjectedKNN.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) *
 *                                    *
 *  pid classifier                    *
 *                                    *
 * Modified:                          *
 *                                    *
 * ************************************/
#pragma once
#ifndef PND_PROJECTED_KNN_H
#define PND_PROJECTED_KNN_H

// Local Headers
#include "PndKnnClassify.h"

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

  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(unsigned int const val);

  void InitKNN();

 private:
  PndProjectedKNN(PndProjectedKNN const &other);
  PndProjectedKNN &operator=(PndProjectedKNN const &other);

  // Free allocated memory.
  void destroy();

  std::vector<std::vector<std::string>> m_varCombinations;

  // First = classifier, Second = parameter indices
  std::vector<std::pair<PndKnnClassify *, std::vector<int> *>> m_classifiers;

  float m_ScaleFact;  
  double m_weight;    
  unsigned int m_knn; 
};
// End, class interface

inline void PndProjectedKNN::SetEvtParam(float const scFact, double const weight)
{
  m_ScaleFact = scFact;
  m_weight = weight;
};

inline void PndProjectedKNN::SetKnn(unsigned int const val)
{
  m_knn = val;
};
#endif