DLCheatHierarchyTool.h

Go to the documentation of this file.

#ifndef LAR_DL_CHEAT_HIERARCHY_TOOL_H
#define LAR_DL_CHEAT_HIERARCHY_TOOL_H 1

#include "Pandora/PandoraInternal.h"

#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

#include "larpandoradlcontent/LArThreeDReco/LArEventBuilding/LArHierarchyPfo.h"

using namespace lar_content;

namespace lar_dl_content
{

class DLCheatHierarchyTool : public pandora::AlgorithmTool
{
public:
    typedef std::map<const pandora::ParticleFlowObject *, const pandora::MCParticle *> PfoToMCParticleMap;
    typedef std::map<const pandora::ParticleFlowObject *, std::pair<const pandora::ParticleFlowObject *, int>> ChildToParentPfoMap;

    DLCheatHierarchyTool();

    pandora::StatusCode Run(const PfoToMCParticleMap &pfoToMCParticleMap, const ChildToParentPfoMap &childToParentPfoMap,
        const HierarchyPfo &parentPfo, const HierarchyPfo &childPfo, bool &isTrueLink, bool &trueParentOrientation, bool &trueChildOrientation);

    pandora::StatusCode Run(const PfoToMCParticleMap &pfoToMCParticleMap, const ChildToParentPfoMap &childToParentPfoMap,
        const pandora::ParticleFlowObject *const pNeutrinoPfo, const HierarchyPfo &childPfo, bool &isTrueLink, bool &trueChildOrientation);

    void FillHierarchyMap(
        const pandora::Algorithm *const pAlgorithm, PfoToMCParticleMap &pfoToMCParticleMap, ChildToParentPfoMap &childToParentPfoMap) const;

    pandora::StatusCode IsNeutronChild(
        const pandora::ParticleFlowObject *const pPfo, const PfoToMCParticleMap &pfoToMCParticleMap, bool &isNeutronChild) const;

private:
    typedef std::map<const pandora::MCParticle *, const pandora::MCParticle *> MCToMCMap;

    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    bool IsUpstreamTrueVertex(const PfoToMCParticleMap &pfoToMCParticleMap, const pandora::ParticleFlowObject *const pPfo,
        const pandora::CartesianVector &upstreamVertex, const pandora::CartesianVector &downstreamVertex);

    bool GetMCParticleList(const pandora::Algorithm *const pAlgorithm, const pandora::MCParticleList *&pMCParticleList) const;

    bool GetNeutrinoPfo(const pandora::Algorithm *const pAlgorithm, const pandora::ParticleFlowObject *&pNeutrinoPfo) const;

    void MatchPFParticles(const pandora::Algorithm *const pAlgorithm, PfoToMCParticleMap &pfoToMCParticleMap) const;

    float GetNSpacepoints(const pandora::ParticleFlowObject *const pPfo) const;

    bool IsEMParticle(const pandora::MCParticle *const pMCParticle) const;

    const pandora::MCParticle *GetLeadEMParticle(const pandora::MCParticle *const pMCParticle) const;

    float SumEnergy(const pandora::CaloHitList &caloHitList) const;

    void GetVisibleMCHierarchy(const PfoToMCParticleMap &pfoToMCParticleMap, MCToMCMap &childToParentMCMap) const;

    void GetVisiblePfoHierarchy(const pandora::ParticleFlowObject *const pNeutrinoPfo, const PfoToMCParticleMap &pfoToMCParticleMap,
        const MCToMCMap &childToParentMCMap, ChildToParentPfoMap &childToParentPfoMap) const;

    void BuildSplitHierarchy(const std::pair<const pandora::MCParticle *, pandora::PfoList> &splitPfo, ChildToParentPfoMap &childToParentPfoMap) const;

    float GetClosestDistance(const pandora::ParticleFlowObject *const pPfo1, const pandora::ParticleFlowObject *const pPfo2) const;

    const pandora::ParticleFlowObject *BestParentInSplitHierarchy(
        const pandora::ParticleFlowObject *const pChildPfo, const pandora::PfoList &splitParticle) const;

    void AssignGeneration(const pandora::ParticleFlowObject *const pParentPfo, const int generationToFind, ChildToParentPfoMap &childToParentPfoMap) const;

    std::string m_mcParticleListName;     
    std::string m_neutrinoPfoListName;    
    pandora::StringVector m_pfoListNames; 
};

} // namespace lar_dl_content

#endif // #ifndef LAR_DL_CHEAT_HIERARCHY_TOOL_H