SimpleBeamParticleId_tool.cc

Go to the documentation of this file.

 
#include "art/Utilities/ToolMacros.h"
#include "fhiclcpp/ParameterSet.h"
 
#include "larpandora/LArPandoraEventBuilding/Slice.h"
#include "larpandora/LArPandoraEventBuilding/SliceIdBaseTool.h"
 
namespace lar_pandora {
 
  class SimpleBeamParticleId : SliceIdBaseTool {
  public:
    SimpleBeamParticleId(fhicl::ParameterSet const& pset);
 
    void ClassifySlices(SliceVector& slices, const art::Event& evt) override;
 
  private:
    float m_minBDTScore; 
  };
 
  DEFINE_ART_CLASS_TOOL(SimpleBeamParticleId)
 
} // namespace lar_pandora
 
//------------------------------------------------------------------------------------------------------------------------------------------
// implementation follows
 
namespace lar_pandora {
 
  SimpleBeamParticleId::SimpleBeamParticleId(fhicl::ParameterSet const& pset)
    : m_minBDTScore(pset.get<float>("MinBDTScore"))
  {}
 
  //------------------------------------------------------------------------------------------------------------------------------------------
 
  void SimpleBeamParticleId::ClassifySlices(SliceVector& slices, const art::Event& /*evt*/)
  {
    for (Slice& slice : slices) {
      if (slice.GetTopologicalScore() > m_minBDTScore) slice.TagAsTarget();
    }
  }
 
} // namespace lar_pandora