LArPandoraSliceIdHelper.h

Go to the documentation of this file.

#ifndef LAR_PANDORA_SLICE_ID_HELPER_H
#define LAR_PANDORA_SLICE_ID_HELPER_H

#include "larpandora/LArPandoraEventBuilding/Slice.h"
#include "larpandora/LArPandoraInterface/LArPandoraHelper.h"

namespace recob {
  class Hit;
}

namespace simb {
  class MCNeutrino;
  class MCTruth;
}

namespace art {
  class Event;
}

#include "canvas/Persistency/Common/Ptr.h"

#include <string>
#include <unordered_map>
#include <vector>

namespace lar_pandora {

  class LArPandoraSliceIdHelper {
  public:
    class SliceMetadata {
    public:
      SliceMetadata();

      float m_purity;        
      float m_completeness;  
      unsigned int m_nHits;  
      bool m_isMostComplete; 
    };

    typedef std::vector<SliceMetadata> SliceMetadataVector;

    static void GetSliceMetadata(const SliceVector& slices,
                                 const art::Event& evt,
                                 const std::string& truthLabel,
                                 const std::string& mcParticleLabel,
                                 const std::string& hitLabel,
                                 const std::string& backtrackLabel,
                                 const std::string& pandoraLabel,
                                 SliceMetadataVector& sliceMetadata,
                                 simb::MCNeutrino& mcNeutrino);

  private:
    typedef std::unordered_map<art::Ptr<recob::Hit>, bool> HitToBoolMap;

    static art::Ptr<simb::MCTruth> GetBeamNeutrinoMCTruth(const art::Event& evt,
                                                          const std::string& truthLabel);

    static void CollectNeutrinoMCParticles(const art::Event& evt,
                                           const std::string& truthLabel,
                                           const std::string& mcParticleLabel,
                                           const art::Ptr<simb::MCTruth>& beamNuMCTruth,
                                           MCParticleVector& mcParticles);

    static void GetHitOrigins(const art::Event& evt,
                              const std::string& hitLabel,
                              const std::string& backtrackLabel,
                              const MCParticleVector& mcParticles,
                              HitVector& hits,
                              HitToBoolMap& hitToIsNuInducedMap);

    static unsigned int CountNeutrinoHits(const HitVector& hits,
                                          const HitToBoolMap& hitToIsNuInducedMap);

    static void GetPFParticleToHitsMap(const art::Event& evt,
                                       const std::string& pandoraLabel,
                                       PFParticlesToHits& pfParticleToHitsMap);

    static void GetReconstructedHitsInSlice(const Slice& slice,
                                            const PFParticlesToHits& pfParticleToHitsMap,
                                            HitVector& hits);

    static void CollectHits(const PFParticleVector& pfParticles,
                            const PFParticlesToHits& pfParticleToHitsMap,
                            HitVector& hits);

    static void GetSliceMetadata(const SliceVector& slices,
                                 const PFParticlesToHits& pfParticleToHitsMap,
                                 const HitToBoolMap& hitToIsNuInducedMap,
                                 const unsigned int nNuHits,
                                 SliceMetadataVector& sliceMetadata);
  };

} // namespace lar_pandora

#endif // LAR_PANDORA_SLICE_ID_HELPER_H