MLRange.h

/*************************************************************************
* BASF2 (Belle Analysis Framework 2)                                     *
* Copyright(C) 2015 - Belle II Collaboration                             *
*                                                                        *
* Author: The Belle II Collaboration                                     *
* Contributors: Thomas Madlener                                          *
*                                                                        *
* This software is provided "as is" without any warranty.                *
**************************************************************************/
 
#pragma once
 
#include <TTree.h>
#include <array>
 
namespace Belle2 {
  template<typename ClassifierType, size_t Ndims = 9, typename CutType = double>
  class MLRange {
    ClassifierType* m_classifier;
    CutType m_cut;
 
  public:
    MLRange(ClassifierType* classifier, CutType cut) : m_classifier(classifier), m_cut(cut) { ; }
 
    inline bool contains(std::array<double, Ndims> hits) const { return !(m_classifier->analyze(hits) < m_cut); }
 
    void persist(TTree* /*t*/, const std::string& /*branchname*/, const std::string& /*variablename*/)
    {
      // TODO
    }
  };
}