MLRange.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#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
    }
  };
}