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,
        const 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;     

    unsigned int                    m_halfWindowLayers;         
    bool                            m_displayPfoInfoMap;        
};

//------------------------------------------------------------------------------------------------------------------------------------------

class PfoRelationTool : public pandora::AlgorithmTool
{
public:
    virtual void Run(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