PndMvaClassifier.h

Go to the documentation of this file.

/* **********************************************
 * MVA classifiers interface.                   *
 * Author: M.Babai@rug.nl                       *
 * LICENSE:                                     *
 * Version:                                     *
 * License:                                     *
 * **********************************************
 */
//#pragma once
#ifndef PND_MVA_CLASSIFIER_H
#define PND_MVA_CLASSIFIER_H

// C++ includes
#include <iostream>
#include <vector>
#include <map>
#include <utility>
#include <algorithm>
#include <cmath>
#include <cassert>

// PND PID includes.
#include "PndMvaDataSet.h"
#include "PndMvaUtil.h"

class PndMvaClassifier
{
 public:

  explicit PndMvaClassifier(std::string const& InPut,
                            std::vector<std::string> const& ClassNames,
                            std::vector<std::string> const& VarNames);
  virtual ~PndMvaClassifier();
  
  virtual void GetMvaValues( std::vector<float> EvtData,
                             std::map<std::string, float>& result ) = 0;
  virtual std::string *Classify( std::vector<float> EvtData ) = 0;

  //virtual std::string *Classify( std::vector<float> EvtData,
  //                     std::map<std::string, float>& result ) = 0;
  
  virtual void Initialize();

  inline std::vector<PndMvaClass> const& GetClasses() const;
  
  inline std::vector<PndMvaVariable> const& GetVariables() const;

 protected:
  virtual void NormalizeEvent(std::vector<float>& EvtVector) const;

  inline void SetAppType(AppType t);
  
  PndMvaDataSet m_dataSets;
  
 private:
  // To avoid mistakes.
  PndMvaClassifier (PndMvaClassifier const& other);
  PndMvaClassifier& operator=(PndMvaClassifier const& other);
};

inline void PndMvaClassifier::SetAppType(AppType t)
{
  m_dataSets.SetAppType(t);
};
//End of class interface
//___________________ Inline implement ___________
inline std::vector<PndMvaClass> const& PndMvaClassifier::GetClasses() const
{
  return m_dataSets.GetClasses();
};

inline std::vector<PndMvaVariable> const& PndMvaClassifier::GetVariables() const
{
  return m_dataSets.GetVars();
};
#endif