NeutrinoHierarchyAlgorithm.h

Go to the documentation of this file.

 
#ifndef LAR_NEUTRINO_HIERARCHY_ALGORITHM_H
#define LAR_NEUTRINO_HIERARCHY_ALGORITHM_H 1
 
#include "Pandora/Algorithm.h"
 
#include "larpandoracontent/LArObjects/LArThreeDSlidingFitResult.h"
 
#include <unordered_map>
 
namespace lar_content
{
 
class PfoRelationTool;
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
class NeutrinoHierarchyAlgorithm : public pandora::Algorithm
{
public:
    NeutrinoHierarchyAlgorithm();
 
    class PfoInfo
    {
    public:
        PfoInfo(const pandora::ParticleFlowObject *const pPfo, const unsigned int halfWindowLayers, const float layerPitch);
 
        PfoInfo(const PfoInfo &rhs);
 
        PfoInfo &operator=(const PfoInfo &rhs);
 
        ~PfoInfo();
 
        const pandora::ParticleFlowObject *GetThisPfo() const;
 
        const pandora::Cluster *GetCluster3D() const;
 
        const ThreeDSlidingFitResult *GetSlidingFitResult3D() const;
 
        bool IsNeutrinoVertexAssociated() const;
 
        bool IsInnerLayerAssociated() const;
 
        const pandora::ParticleFlowObject *GetParentPfo() const;
 
        const pandora::PfoList &GetDaughterPfoList() const;
 
        void SetNeutrinoVertexAssociation(const bool isNeutrinoVertexAssociated);
 
        void SetInnerLayerAssociation(const bool isInnerLayerAssociated);
 
        void SetParentPfo(const pandora::ParticleFlowObject *const pParentPfo);
 
        void RemoveParentPfo();
 
        void AddDaughterPfo(const pandora::ParticleFlowObject *const pDaughterPfo);
 
        void RemoveDaughterPfo(const pandora::ParticleFlowObject *const pDaughterPfo);
 
    private:
        const pandora::ParticleFlowObject *m_pThisPfo; 
        const pandora::Cluster *m_pCluster3D;          
        const pandora::Vertex *m_pVertex3D;            
        ThreeDSlidingFitResult *m_pSlidingFitResult3D; 
 
        bool m_isNeutrinoVertexAssociated; 
        bool m_isInnerLayerAssociated;     
        const pandora::ParticleFlowObject *m_pParentPfo; 
        pandora::PfoList m_daughterPfoList;              
    };
 
    typedef std::unordered_map<const pandora::ParticleFlowObject *, PfoInfo *> PfoInfoMap;
 
    void SeparatePfos(const NeutrinoHierarchyAlgorithm::PfoInfoMap &pfoInfoMap, pandora::PfoVector &assignedPfos, pandora::PfoVector &unassignedPfos) const;
 
private:
    pandora::StatusCode Run();
 
    void GetNeutrinoPfo(const pandora::ParticleFlowObject *&pNeutrinoPfo) const;
 
    void GetCandidateDaughterPfoList(pandora::PfoList &candidateDaughterPfoList) const;
 
    void GetInitialPfoInfoMap(const pandora::PfoList &pfoList, PfoInfoMap &pfoInfoMap) const;
 
    void ProcessPfoInfoMap(const pandora::ParticleFlowObject *const pNeutrinoPfo, const pandora::PfoList &candidateDaughterPfoList,
        PfoInfoMap &pfoInfoMap, const unsigned int callDepth = 0) const;
 
    void AdjustVertexAndPfoInfo(const pandora::ParticleFlowObject *const pNeutrinoPfo, const pandora::PfoList &candidateDaughterPfoList,
        PfoInfoMap &pfoInfoMap) const;
 
    void DisplayPfoInfoMap(const pandora::ParticleFlowObject *const pNeutrinoPfo, const PfoInfoMap &pfoInfoMap) const;
 
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);
 
    typedef std::vector<PfoRelationTool *> PfoRelationToolVector;
    PfoRelationToolVector m_algorithmToolVector; 
 
    std::string m_neutrinoPfoListName;            
    pandora::StringVector m_daughterPfoListNames; 
 
    std::string m_neutrinoVertexListName; 
 
    unsigned int m_halfWindowLayers; 
    bool m_displayPfoInfoMap;        
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
class PfoRelationTool : public pandora::AlgorithmTool
{
public:
    virtual void Run(const NeutrinoHierarchyAlgorithm *const pAlgorithm, const pandora::Vertex *const pNeutrinoVertex,
        NeutrinoHierarchyAlgorithm::PfoInfoMap &pfoInfoMap) = 0;
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::ParticleFlowObject *NeutrinoHierarchyAlgorithm::PfoInfo::GetThisPfo() const
{
    return m_pThisPfo;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::Cluster *NeutrinoHierarchyAlgorithm::PfoInfo::GetCluster3D() const
{
    return m_pCluster3D;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const ThreeDSlidingFitResult *NeutrinoHierarchyAlgorithm::PfoInfo::GetSlidingFitResult3D() const
{
    return m_pSlidingFitResult3D;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline bool NeutrinoHierarchyAlgorithm::PfoInfo::IsNeutrinoVertexAssociated() const
{
    return m_isNeutrinoVertexAssociated;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline bool NeutrinoHierarchyAlgorithm::PfoInfo::IsInnerLayerAssociated() const
{
    return m_isInnerLayerAssociated;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::ParticleFlowObject *NeutrinoHierarchyAlgorithm::PfoInfo::GetParentPfo() const
{
    return m_pParentPfo;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::PfoList &NeutrinoHierarchyAlgorithm::PfoInfo::GetDaughterPfoList() const
{
    return m_daughterPfoList;
}
 
} // namespace lar_content
 
#endif // #ifndef LAR_NEUTRINO_HIERARCHY_ALGORITHM_H