PndPDEFoamClassify.h

Go to the documentation of this file.

/* ***************************************
 * PDEFoam  Classifier                   *
 * Author: M.Babai@rug.nl                *
 * LICENSE:                              *
 * Version: 0.1 beta1.                   *
 * License:                              *
 * ***************************************
 */
#pragma once
#ifndef PND_PDE_FOAMCLASSIFY_H
#define PND_PDE_FOAMCLASSIFY_H

// Local includes
#include "PndGpidClassifier.h"

// ROOT AND PANDA
#include "TMVA/PDEFoam.h"
#include "TMVA/Event.h"

class PndPDEFoamClassify //: public PndGpidClassifier
{
 public:
  PndPDEFoamClassify(const std::string &inputFile, const std::vector<std::string> &classNames, const std::vector<std::string> &varNames);
  virtual ~PndPDEFoamClassify();

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

  const std::string &Classify(std::vector<float> EvtData) const;

 protected:
  void ReadInputFile();

 private:
  // Functions
  PndPDEFoamClassify(const PndPDEFoamClassify &other);
  PndPDEFoamClassify &operator=(const PndPDEFoamClassify &other);

  // Variables.
  // Comment:
  /*
   * Mabe wee need to inherit from classifiers in order to avoid
   * double implementation. For the time being, lets use this.
   */
  std::string m_inputFile;
  std::vector<TMVA::PDEFoam *> m_foams;
  std::vector<PndMvaVariable> m_vars;
  std::vector<PndMvaClass> m_classes;
};
#endif // end of interface definition