DeltaRayMatchingContainers.h

Go to the documentation of this file.

 
#ifndef LAR_DELTA_RAY_MATCHING_CONTAINERS_H
#define LAR_DELTA_RAY_MATCHING_CONTAINERS_H 1
 
#include "Pandora/PandoraInternal.h"
 
#include "larpandoracontent/LArUtility/KDTreeLinkerAlgoT.h"
 
namespace lar_content
{
 
class DeltaRayMatchingContainers
{
public:
    typedef std::map<const pandora::Cluster *, const pandora::ParticleFlowObject *> ClusterToPfoMap;
    typedef std::map<const pandora::Cluster *, pandora::ClusterList> ClusterProximityMap;
 
    DeltaRayMatchingContainers();
 
    const ClusterToPfoMap &GetClusterToPfoMap(const pandora::HitType hitType) const;
 
    const ClusterProximityMap &GetClusterProximityMap(const pandora::HitType hitType) const;
 
    void FillContainers(const pandora::PfoList &inputPfoList, const pandora::ClusterList &inputClusterList1,
        const pandora::ClusterList &inputClusterList2 = pandora::ClusterList(),
        const pandora::ClusterList &inputClusterList3 = pandora::ClusterList());
 
    void AddClustersToPfoMaps(const pandora::ParticleFlowObject *const pPfo);
 
    void AddClustersToContainers(const pandora::ClusterVector &newClusterVector, const pandora::PfoVector &pfoVector);
 
    void RemoveClusterFromContainers(const pandora::Cluster *const pDeletedCluster);
 
    void ClearContainers();
 
    float m_searchRegion1D; 
 
private:
    typedef std::map<const pandora::CaloHit *, const pandora::Cluster *> HitToClusterMap;
    typedef KDTreeLinkerAlgo<const pandora::CaloHit *, 2> HitKDTree2D;
    typedef KDTreeNodeInfoT<const pandora::CaloHit *, 2> HitKDNode2D;
    typedef std::vector<HitKDNode2D> HitKDNode2DList;
 
    void FillHitToClusterMap(const pandora::ClusterList &inputClusterList);
 
    void AddToClusterMap(const pandora::Cluster *const pCluster);
 
    void FillClusterToPfoMaps(const pandora::PfoList &pfoList);
 
    void FillClusterProximityMap(const pandora::ClusterList &inputClusterList);
 
    void BuildKDTree(const pandora::HitType hitType);
 
    void AddToClusterProximityMap(const pandora::Cluster *const pCluster);
 
    HitToClusterMap m_hitToClusterMapU;         
    HitToClusterMap m_hitToClusterMapV;         
    HitToClusterMap m_hitToClusterMapW;         
    HitKDTree2D m_kdTreeU;                      
    HitKDTree2D m_kdTreeV;                      
    HitKDTree2D m_kdTreeW;                      
    ClusterProximityMap m_clusterProximityMapU; 
    ClusterProximityMap m_clusterProximityMapV; 
    ClusterProximityMap m_clusterProximityMapW; 
    ClusterToPfoMap m_clusterToPfoMapU;         
    ClusterToPfoMap m_clusterToPfoMapV;         
    ClusterToPfoMap m_clusterToPfoMapW;         
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const DeltaRayMatchingContainers::ClusterProximityMap &DeltaRayMatchingContainers::GetClusterProximityMap(const pandora::HitType hitType) const
{
    return ((hitType == pandora::TPC_VIEW_U) ? m_clusterProximityMapU : (hitType == pandora::TPC_VIEW_V) ? m_clusterProximityMapV : m_clusterProximityMapW);
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const DeltaRayMatchingContainers::ClusterToPfoMap &DeltaRayMatchingContainers::GetClusterToPfoMap(const pandora::HitType hitType) const
{
    return ((hitType == pandora::TPC_VIEW_U) ? m_clusterToPfoMapU : (hitType == pandora::TPC_VIEW_V) ? m_clusterToPfoMapV : m_clusterToPfoMapW);
}
 
} // namespace lar_content
 
#endif // #ifndef LAR_DELTA_RAY_MATCHING_CONTAINERS_H