Selector.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marcel Hohmann                                           *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include<analysis/dataobjects/Particle.h>
#include<vector>
 
namespace Belle2 {
  namespace CurlTagger {
 
    class Selector {
 
    public:
      Selector() {};
 
      virtual ~Selector() {};
 
      virtual float getResponse(Particle*, Particle*) = 0;
 
      virtual std::vector<float> getVariables(Particle*, Particle*) = 0;
 
      virtual void initialize() {};
 
      virtual void finalize() {};
 
      virtual void collectTrainingInfo(Particle*, Particle*)  {};
    }; //class
  } // CurlTagger namespace
} // Belle 2 namespace