ChoosableFromVarSetFilter.dcl.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 <tracking/trackFindingCDC/filters/base/FilterOnVarSet.dcl.h>
 
#include <tracking/trackFindingCDC/varsets/BaseVarSet.h>
 
#include <RtypesCore.h>
 
#include <string>
#include <memory>
 
namespace Belle2 {
  class ModuleParamList;
  namespace TrackFindingCDC {

    template <class AFilter>
    class ChoosableFromVarSet : public OnVarSet<AFilter> {
 
    private:
      using Super = OnVarSet<AFilter>;
 
    public:
      using Object = typename AFilter::Object;
 
    private:
      using AVarSet = BaseVarSet<Object>;
 
    public:
      ChoosableFromVarSet(std::unique_ptr<AVarSet> varSet, std::string varName = "");

      ~ChoosableFromVarSet();

      void exposeParameters(ModuleParamList* parameterList, const std::string& prefix) override;

      void initialize() override;

      Weight operator()(const Object& object) override;
 
    private:
      std::string m_param_varName;

      Float_t* m_variable = nullptr;
    };

    template <class ATruthVarSet>
    class ChoosableFromVarSetFilter
      : public ChoosableFromVarSet<Filter<typename ATruthVarSet::Object>> {
 
    private:
      using Super = ChoosableFromVarSet<Filter<typename ATruthVarSet::Object>>;
 
    public:
      using Object = typename ATruthVarSet::Object;
 
    public:
      ChoosableFromVarSetFilter(const std::string& varName = "");

      ~ChoosableFromVarSetFilter();
    };
  }
}