SimpleNeutrinoId_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 SimpleNeutrinoId : SliceIdBaseTool {
  public:
    SimpleNeutrinoId(fhicl::ParameterSet const& pset);

    void ClassifySlices(SliceVector& slices, const art::Event& evt) override;
  };

  DEFINE_ART_CLASS_TOOL(SimpleNeutrinoId)

} // namespace lar_pandora

//------------------------------------------------------------------------------------------------------------------------------------------
// implementation follows

namespace lar_pandora {

  SimpleNeutrinoId::SimpleNeutrinoId(fhicl::ParameterSet const& /*pset*/) {}

  //------------------------------------------------------------------------------------------------------------------------------------------

  void SimpleNeutrinoId::ClassifySlices(SliceVector& slices, const art::Event& /*evt*/)
  {
    if (slices.empty()) return;

    // Find the most probable slice
    float highestNuScore(-std::numeric_limits<float>::max());
    unsigned int mostProbableSliceIndex(std::numeric_limits<unsigned int>::max());

    for (unsigned int sliceIndex = 0; sliceIndex < slices.size(); ++sliceIndex) {
      const float nuScore(slices.at(sliceIndex).GetTopologicalScore());
      if (nuScore > highestNuScore) {
        highestNuScore = nuScore;
        mostProbableSliceIndex = sliceIndex;
      }
    }

    // Tag the most probable slice as a neutrino
    slices.at(mostProbableSliceIndex).TagAsTarget();
  }

} // namespace lar_pandora