LArMCParticleHelper.h

Go to the documentation of this file.

 
#ifndef LAR_MC_PARTICLE_HELPER_H
#define LAR_MC_PARTICLE_HELPER_H 1
 
#include "Pandora/PandoraInternal.h"
 
#include "larpandoracontent/LArObjects/LArMCParticle.h"
 
#include <functional>
#include <unordered_map>
 
namespace lar_content
{
 
class LArMCParticleHelper
{
public:
    typedef std::unordered_map<const pandora::MCParticle *, const pandora::MCParticle *> MCRelationMap;
    typedef std::unordered_map<const pandora::MCParticle *, int> MCParticleIntMap;
 
    typedef std::unordered_map<const pandora::MCParticle *, const pandora::ParticleFlowObject *> MCToPfoMap;
 
    typedef std::unordered_map<const pandora::CaloHit *, const pandora::MCParticle *> CaloHitToMCMap;
    typedef std::unordered_map<const pandora::CaloHit *, const pandora::ParticleFlowObject *> CaloHitToPfoMap;
 
    typedef std::unordered_map<const pandora::MCParticle *, pandora::CaloHitList> MCContributionMap;
    typedef std::vector<MCContributionMap> MCContributionMapVector;
 
    typedef std::unordered_map<const pandora::Cluster *, pandora::CaloHitList> ClusterContributionMap;
    typedef std::unordered_map<const pandora::ParticleFlowObject *, pandora::CaloHitList> PfoContributionMap;
    typedef std::unordered_map<const pandora::MCParticle *, PfoContributionMap> MCToPfoMatchingMap;
 
    typedef std::pair<const pandora::MCParticle *, pandora::CaloHitList> MCParticleCaloHitListPair;
    typedef std::pair<const pandora::ParticleFlowObject *, pandora::CaloHitList> PfoCaloHitListPair;
    typedef std::pair<const pandora::Cluster *, pandora::CaloHitList> ClusterCaloHitListPair;
 
    typedef std::vector<MCParticleCaloHitListPair> MCParticleToSharedHitsVector;
    typedef std::vector<PfoCaloHitListPair> PfoToSharedHitsVector;
 
    typedef std::map<const pandora::ParticleFlowObject *, MCParticleToSharedHitsVector> PfoToMCParticleHitSharingMap;
    typedef std::map<const pandora::MCParticle *, PfoToSharedHitsVector> MCParticleToPfoHitSharingMap;
 
    class PrimaryParameters
    {
    public:
        PrimaryParameters();
 
        unsigned int m_minPrimaryGoodHits;  
        unsigned int m_minHitsForGoodView;  
        unsigned int m_minPrimaryGoodViews; 
        bool m_selectInputHits;             
        float m_maxPhotonPropagation;       
        float m_minHitSharingFraction;      
        bool m_foldBackHierarchy; 
    };
 
    static bool DoesPrimaryMeetCriteria(const pandora::MCParticle *const pMCParticle, std::function<bool(const pandora::MCParticle *const)> fCriteria);
 
    static bool DoesLeadingMeetCriteria(const pandora::MCParticle *const pMCParticle, std::function<bool(const pandora::MCParticle *const)> fCriteria);
 
    static bool IsBeamNeutrinoFinalState(const pandora::MCParticle *const pMCParticle);
 
    static bool IsTriggeredBeamParticle(const pandora::MCParticle *const pMCParticle);
 
    static bool IsBeamParticle(const pandora::MCParticle *const pMCParticle);
 
    static bool IsLeadingBeamParticle(const pandora::MCParticle *const pMCParticle);
 
    static bool IsCosmicRay(const pandora::MCParticle *const pMCParticle);
 
    static unsigned int GetNuanceCode(const pandora::MCParticle *const pMCParticle);
 
    static bool IsNeutrino(const pandora::MCParticle *const pMCParticle);
 
    static bool IsPrimary(const pandora::MCParticle *const pMCParticle);
 
    static bool IsLeading(const pandora::MCParticle *const pMCParticle);
 
    static int GetHierarchyTier(const pandora::MCParticle *const pMCParticle);
 
    static bool IsVisible(const pandora::MCParticle *const pMCParticle);
 
    static void GetTrueNeutrinos(const pandora::MCParticleList *const pMCParticleList, pandora::MCParticleVector &trueNeutrinos);
 
    static void GetTrueTestBeamParticles(const pandora::MCParticleList *const pMCParticleList, pandora::MCParticleVector &trueTestBeamParticles);
 
    static void GetFirstVisibleMCParticles(const pandora::MCParticle *const pRoot, pandora::MCParticleList &visibleParticleList);
 
    static const pandora::MCParticle *GetPrimaryMCParticle(const pandora::MCParticle *const pMCParticle);
 
    static const pandora::MCParticle *GetLeadingMCParticle(const pandora::MCParticle *const pMCParticle, const int hierarchyTierLimit = 1);
 
    static void GetPrimaryMCParticleList(const pandora::MCParticleList *const pMCParticleList, pandora::MCParticleVector &mcPrimaryVector);
 
    static void GetLeadingMCParticleList(const pandora::MCParticleList *const pMCParticleList, pandora::MCParticleVector &mcLeadingVector);
 
    static const pandora::MCParticle *GetParentMCParticle(const pandora::MCParticle *const pMCParticle);
 
    static void GetAllDescendentMCParticles(const pandora::MCParticle *const pMCParticle, pandora::MCParticleList &descendentMCParticleList);
 
    static void GetAllDescendentMCParticles(const pandora::MCParticle *const pMCParticle, pandora::MCParticleList &descendentTrackParticles,
        pandora::MCParticleList &leadingShowerParticles, pandora::MCParticleList &leadingNeutrons);
 
    static void GetAllAncestorMCParticles(const pandora::MCParticle *const pMCParticle, pandora::MCParticleList &ancestorMCParticleList);
 
    static void GetMCPrimaryMap(const pandora::MCParticleList *const pMCParticleList, MCRelationMap &mcPrimaryMap);
 
    static void GetMCLeadingMap(const pandora::MCParticleList *const pMCParticleList, MCRelationMap &mcLeadingMap);
 
    static void GetMCToSelfMap(const pandora::MCParticleList *const pMCParticleList, MCRelationMap &mcToSelfMap);
 
    static const pandora::MCParticle *GetMainMCParticle(const pandora::ParticleFlowObject *const pPfo);
 
    static bool SortByMomentum(const pandora::MCParticle *const pLhs, const pandora::MCParticle *const pRhs);
 
    static void GetMCParticleToCaloHitMatches(const pandora::CaloHitList *const pCaloHitList, const MCRelationMap &mcToTargetMCMap,
        CaloHitToMCMap &hitToMCMap, MCContributionMap &mcToTrueHitListMap);
 
    static void SelectReconstructableMCParticles(const pandora::MCParticleList *pMCParticleList, const pandora::CaloHitList *pCaloHitList,
        const PrimaryParameters &parameters, std::function<bool(const pandora::MCParticle *const)> fCriteria,
        MCContributionMap &selectedMCParticlesToHitsMap);
 
    static void SelectReconstructableTestBeamHierarchyMCParticles(const pandora::MCParticleList *pMCParticleList,
        const pandora::CaloHitList *pCaloHitList, const PrimaryParameters &parameters,
        std::function<bool(const pandora::MCParticle *const)> fCriteria, MCContributionMap &selectedMCParticlesToHitsMap);
 
    static void GetPfoToReconstructable2DHitsMap(const pandora::PfoList &pfoList, const MCContributionMap &selectedMCParticleToHitsMap,
        PfoContributionMap &pfoToReconstructable2DHitsMap, const bool foldBackHierarchy);
 
    static void GetTestBeamHierarchyPfoToReconstructable2DHitsMap(const pandora::PfoList &pfoList,
        const MCContributionMap &selectedMCParticleToHitsMap, PfoContributionMap &pfoToReconstructable2DHitsMap, const bool foldBackHierarchy);
 
    static void GetPfoToReconstructable2DHitsMap(const pandora::PfoList &pfoList, const MCContributionMapVector &selectedMCParticleToHitsMaps,
        PfoContributionMap &pfoToReconstructable2DHitsMap, const bool foldBackHierarchy);
 
    static void GetTestBeamHierarchyPfoToReconstructable2DHitsMap(const pandora::PfoList &pfoList,
        const MCContributionMapVector &selectedMCParticleToHitsMaps, PfoContributionMap &pfoToReconstructable2DHitsMap, const bool foldBackHierarchy);
 
    static void GetClusterToReconstructable2DHitsMap(const pandora::ClusterList &clusterList, const MCContributionMap &selectedMCToHitsMap,
        ClusterContributionMap &clusterToReconstructable2DHitsMap);
 
    static void GetClusterToReconstructable2DHitsMap(const pandora::ClusterList &clusterList,
        const MCContributionMapVector &selectedMCToHitsMaps, ClusterContributionMap &clusterToReconstructable2DHitsMap);
 
    static void GetPfoMCParticleHitSharingMaps(const PfoContributionMap &pfoToReconstructable2DHitsMap,
        const MCContributionMapVector &selectedMCParticleToHitsMaps, PfoToMCParticleHitSharingMap &pfoToMCParticleHitSharingMap,
        MCParticleToPfoHitSharingMap &mcParticleToPfoHitSharingMap);
 
    static void SelectCaloHits(const pandora::CaloHitList *const pCaloHitList, const MCRelationMap &mcToTargetMCMap,
        pandora::CaloHitList &selectedCaloHitList, const bool selectInputHits, const float maxPhotonPropagation);
 
    static bool IsDescendentOf(const pandora::MCParticle *const pMCParticle, const int pdg, const bool isChargeSensitive = false);
 
    static void GetBreadthFirstHierarchyRepresentation(const pandora::MCParticle *const pMCParticle, pandora::MCParticleList &mcParticleList);
 
    static void SelectParticlesByHitCount(const pandora::MCParticleVector &candidateTargets, const MCContributionMap &mcToTrueHitListMap,
        const MCRelationMap &mcToTargetMCMap, const PrimaryParameters &parameters, MCContributionMap &selectedMCParticlesToHitsMap);
 
    static pandora::CaloHitList GetSharedHits(const pandora::CaloHitList &hitListA, const pandora::CaloHitList &hitListB);
 
    /*
     *  @brief  Check whether or not an MC particle comes from a Bremsstrahlung process
     *
     *  @param  pMCParticle The MC particle to consider
     *
     *  @return  Whether or not the MC particle came from Bremsstrahlung
     */
    static bool IsBremsstrahlung(const pandora::MCParticle *const pMCParticle);
 
    static bool IsCapture(const pandora::MCParticle *const pMCParticle);
 
    static bool IsDecay(const pandora::MCParticle *const pMCParticle);
 
    static bool IsElasticScatter(const pandora::MCParticle *const pMCParticle);
 
    static bool IsInelasticScatter(const pandora::MCParticle *const pMCParticle);
 
    static bool IsIonisation(const pandora::MCParticle *const pMCParticle);
 
    static bool IsNuclear(const pandora::MCParticle *const pMCParticle);
 
    static bool IsPairProduction(const pandora::MCParticle *const pMCParticle);
 
    static bool AreTopologicallyContinuous(
        const pandora::MCParticle *const pMCParent, const pandora::MCParticle *const pMCChild, const float cosAngleTolerance);
 
private:
    static void CollectReconstructable2DHits(const pandora::ParticleFlowObject *const pPfo, const MCContributionMapVector &selectedMCParticleToHitsMaps,
        pandora::CaloHitList &reconstructableCaloHitList2D, const bool foldBackHierarchy);
 
    static void CollectReconstructableTestBeamHierarchy2DHits(const pandora::ParticleFlowObject *const pPfo,
        const MCContributionMapVector &selectedMCParticleToHitsMaps, pandora::CaloHitList &reconstructableCaloHitList2D, const bool foldBackHierarchy);
 
    static void CollectReconstructable2DHits(const pandora::PfoList &pfoList, const MCContributionMapVector &selectedMCParticleToHitsMaps,
        pandora::CaloHitList &reconstructableCaloHitList2D);
 
    static void CollectReconstructable2DHits(const pandora::Cluster *const pCluster,
        const MCContributionMapVector &selectedMCParticleToHitsMaps, pandora::CaloHitList &reconstructableCaloHitList2D);
 
    static void SelectGoodCaloHits(const pandora::CaloHitList *const pSelectedCaloHitList, const MCRelationMap &mcToTargetMCMap,
        pandora::CaloHitList &selectedGoodCaloHitList, const bool selectInputHits, const float minHitSharingFraction);
 
    static void SelectParticlesMatchingCriteria(const pandora::MCParticleVector &inputMCParticles,
        std::function<bool(const pandora::MCParticle *const)> fCriteria, pandora::MCParticleVector &selectedParticles,
        const PrimaryParameters &parameters, const bool isTestBeam);
 
    static bool PassMCParticleChecks(const pandora::MCParticle *const pOriginalPrimary, const pandora::MCParticle *const pThisMCParticle,
        const pandora::MCParticle *const pHitMCParticle, const float maxPhotonPropagation);
};
 
} // namespace lar_content
 
#endif // #ifndef LAR_MC_PARTICLE_HELPER_H