SlidingConePfoMopUpAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_SLIDING_CONE_PFO_MOP_UP_ALGORITHM_H
#define LAR_SLIDING_CONE_PFO_MOP_UP_ALGORITHM_H 1

#include "larpandoracontent/LArUtility/PfoMopUpBaseAlgorithm.h"

#include <unordered_map>

namespace lar_content
{

class SlidingConePfoMopUpAlgorithm : public PfoMopUpBaseAlgorithm
{
public:
    SlidingConePfoMopUpAlgorithm();

private:
    class ClusterMerge
    {
    public:
        ClusterMerge(const pandora::Cluster *const pParentCluster, const float boundedFraction1, const float boundedFraction2);

        const pandora::Cluster *GetParentCluster() const;

        float GetBoundedFraction1() const;

        float GetBoundedFraction2() const;

        bool operator<(const ClusterMerge &rhs) const;

    private:
        const pandora::Cluster *m_pParentCluster;       
        float                   m_boundedFraction1;     
        float                   m_boundedFraction2;     
    };

    typedef std::vector<ClusterMerge> ClusterMergeList;

    pandora::StatusCode Run();

    void GetInteractionVertex(const pandora::Vertex *&pVertex) const;

    typedef std::unordered_map<const pandora::Cluster*, const pandora::ParticleFlowObject*> ClusterToPfoMap;

    void GetThreeDClusters(pandora::ClusterVector &clusters3D, ClusterToPfoMap &clusterToPfoMap) const;

    typedef std::unordered_map<const pandora::Cluster*, ClusterMergeList> ClusterMergeMap;

    void GetClusterMergeMap(const pandora::Vertex *const pVertex, const pandora::ClusterVector &clusters3D, const ClusterToPfoMap &clusterToPfoMap,
        ClusterMergeMap &clusterMergeMap) const;

    typedef std::unordered_map<const pandora::Cluster*, bool> VertexAssociationMap;

    bool IsVertexAssociated(const pandora::Cluster *const pCluster, const pandora::Vertex *const pVertex,
        VertexAssociationMap &vertexAssociationMap, const ThreeDSlidingFitResult *const pSlidingFitResult = nullptr) const;

    bool IsVertexAssociated(const pandora::Cluster *const pCluster, const pandora::CartesianVector &vertexPosition,
        const ThreeDSlidingFitResult *const pSlidingFitResult = nullptr) const;

    bool MakePfoMerges(const ClusterToPfoMap &clusterToPfoMap, const ClusterMergeMap &clusterMergeMap) const;

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

    typedef std::unordered_map<const pandora::Cluster*, const pandora::Cluster*> ClusterReplacementMap;

    pandora::StringVector   m_inputPfoListNames;                
    bool                    m_useVertex;                        
    unsigned int            m_maxIterations;                    
    unsigned int            m_maxHitsToConsider3DTrack;         
    unsigned int            m_minHitsToConsider3DShower;        
    unsigned int            m_halfWindowLayers;                 
    unsigned int            m_nConeFitLayers;                   
    unsigned int            m_nConeFits;                        
    float                   m_coneLengthMultiplier;             
    float                   m_maxConeLength;                    
    float                   m_coneTanHalfAngle1;                
    float                   m_coneBoundedFraction1;             
    float                   m_coneTanHalfAngle2;                
    float                   m_coneBoundedFraction2;             
    float                   m_minVertexLongitudinalDistance;    
    float                   m_maxVertexTransverseDistance;      
};

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

inline SlidingConePfoMopUpAlgorithm::ClusterMerge::ClusterMerge(const pandora::Cluster *const pParentCluster, const float boundedFraction1, const float boundedFraction2) :
    m_pParentCluster(pParentCluster),
    m_boundedFraction1(boundedFraction1),
    m_boundedFraction2(boundedFraction2)
{
}

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

inline const pandora::Cluster *SlidingConePfoMopUpAlgorithm::ClusterMerge::GetParentCluster() const
{
    return m_pParentCluster;
}

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

inline float SlidingConePfoMopUpAlgorithm::ClusterMerge::GetBoundedFraction1() const
{
    return m_boundedFraction1;
}

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

inline float SlidingConePfoMopUpAlgorithm::ClusterMerge::GetBoundedFraction2() const
{
    return m_boundedFraction2;
}

} // namespace lar_content

#endif // #ifndef LAR_SLIDING_CONE_PFO_MOP_UP_ALGORITHM_H