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