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