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