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