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