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