DeltaRayMatchingAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_DELTA_RAY_MATCHING_ALGORITHM_H
#define LAR_DELTA_RAY_MATCHING_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.h"

namespace lar_content
{

template<typename, unsigned int> class KDTreeLinkerAlgo;
template<typename, unsigned int> class KDTreeNodeInfoT;

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

class DeltaRayMatchingAlgorithm : public pandora::Algorithm
{
public:
    DeltaRayMatchingAlgorithm();

private:
    pandora::StatusCode Run();

    class Particle
    {
    public:
        Particle(const pandora::Cluster *const pCluster1, const pandora::Cluster *const pCluster2, const pandora::Cluster *const pCluster3,
            const pandora::ParticleFlowObject *const pPfo);

        const pandora::Cluster *GetClusterU() const;

        const pandora::Cluster *GetClusterV() const;

        const pandora::Cluster *GetClusterW() const;

        const pandora::ParticleFlowObject *GetParentPfo() const;

        unsigned int GetNViews() const;

        unsigned int GetNCaloHits() const;

    private:
        const pandora::Cluster             *m_pClusterU;    
        const pandora::Cluster             *m_pClusterV;    
        const pandora::Cluster             *m_pClusterW;    
        const pandora::ParticleFlowObject  *m_pParentPfo;   
    };

    typedef std::vector<Particle> ParticleList;

    typedef KDTreeLinkerAlgo<const pandora::CaloHit*, 2> HitKDTree2D;
    typedef KDTreeNodeInfoT<const pandora::CaloHit*, 2> HitKDNode2D;
    typedef std::vector<HitKDNode2D> HitKDNode2DList;

    typedef std::unordered_map<const pandora::Cluster*, pandora::ClusterList> ClusterToClustersMap;
    typedef std::unordered_map<const pandora::CaloHit*, const pandora::Cluster*> HitToClusterMap;

    void InitializeNearbyClusterMaps();

    void InitializeNearbyClusterMap(const std::string &clusterListName, ClusterToClustersMap &nearbyClusters);

    void ClearNearbyClusterMaps();

    void GetAllPfos(const std::string &inputPfoListName, pandora::PfoVector &pfoVector) const;

    void GetTrackPfos(const std::string &inputPfoListName, pandora::PfoVector &pfoVector) const;

    void GetClusters(const std::string &clusterListName, pandora::ClusterVector &clusterVector) const;

    typedef std::unordered_map<const pandora::Cluster*, float> ClusterLengthMap;
    typedef std::unordered_map<const pandora::ParticleFlowObject*, float> PfoLengthMap;

    void ThreeViewMatching(ClusterLengthMap &clusterLengthMap) const;

    void TwoViewMatching(ClusterLengthMap &clusterLengthMap) const;

    void OneViewMatching(ClusterLengthMap &clusterLengthMap) const;

    void ThreeViewMatching(const pandora::ClusterVector &clusters1, const pandora::ClusterVector &clusters2, const pandora::ClusterVector &clusters3,
        ClusterLengthMap &clusterLengthMap, PfoLengthMap &pfoLengthMap, ParticleList &particleList) const;

    void TwoViewMatching(const pandora::ClusterVector &clusters1, const pandora::ClusterVector &clusters2, ClusterLengthMap &clusterLengthMap,
        PfoLengthMap &pfoLengthMap, ParticleList &particleList) const;

    void OneViewMatching(const pandora::ClusterVector &clusters, ClusterLengthMap &clusterLengthMap, PfoLengthMap &pfoLengthMap, ParticleList &particleList) const;

    void SelectParticles(const ParticleList &inputParticles, ClusterLengthMap &clusterLengthMap, ParticleList &outputParticles) const;

    void CreateParticles(const ParticleList &particleList) const;

    void FindBestParentPfo(const pandora::Cluster *const pClusterU, const pandora::Cluster *const pClusterV, const pandora::Cluster *const pClusterW,
        ClusterLengthMap &clusterLengthMap, PfoLengthMap &pfoLengthMap, const pandora::ParticleFlowObject *&pBestPfo) const;

    float GetLengthFromCache(const pandora::Cluster *const pCluster, ClusterLengthMap &clusterLengthMap) const;

    float GetLengthFromCache(const pandora::ParticleFlowObject *const pPfo, PfoLengthMap &pfoLengthMap) const;

    float GetLength(const Particle &particle, ClusterLengthMap &clusterLengthMap) const;

    bool AreClustersMatched(const pandora::Cluster *const pCluster1, const pandora::Cluster *const pCluster2,
        const pandora::Cluster *const pCluster3) const;

    float GetDistanceSquaredToPfo(const pandora::Cluster *const pCluster, const pandora::ParticleFlowObject *const pPfo) const;

    void CreateDaughterPfo(const pandora::ClusterList &clusterList, const pandora::ParticleFlowObject *const pParentPfo) const;
  
    void AddToDaughterPfo(const pandora::ClusterList &clusterList, const pandora::ParticleFlowObject *const pParentPfo) const;

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

    std::string             m_parentPfoListName;          
    std::string             m_daughterPfoListName;        

    std::string             m_inputClusterListNameU;      
    std::string             m_inputClusterListNameV;      
    std::string             m_inputClusterListNameW;      

    unsigned int            m_minCaloHitsPerCluster;      
    float                   m_xOverlapWindow;             
    float                   m_distanceForMatching;        
    float                   m_pseudoChi2Cut;              

    float                   m_searchRegion1D;             
    ClusterToClustersMap    m_nearbyClustersU;            
    ClusterToClustersMap    m_nearbyClustersV;            
    ClusterToClustersMap    m_nearbyClustersW;            
};

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

inline const pandora::Cluster *DeltaRayMatchingAlgorithm::Particle::GetClusterU() const
{
    return m_pClusterU;
}

//------------------------------------------------------------------------------------------------------------------------------------------
inline const pandora::Cluster *DeltaRayMatchingAlgorithm::Particle::GetClusterV() const
{
    return m_pClusterV;
}

//------------------------------------------------------------------------------------------------------------------------------------------
inline const pandora::Cluster *DeltaRayMatchingAlgorithm::Particle::GetClusterW() const
{
    return m_pClusterW;
}

//------------------------------------------------------------------------------------------------------------------------------------------
inline const pandora::ParticleFlowObject *DeltaRayMatchingAlgorithm::Particle::GetParentPfo() const
{
    return m_pParentPfo;
}

} // namespace lar_content

#endif // #ifndef LAR_DELTA_RAY_MATCHING_ALGORITHM_H