TransverseAssociationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_TRANSVERSE_ASSOCIATION_ALGORITHM_H
#define LAR_TRANSVERSE_ASSOCIATION_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArTwoDReco/LArClusterAssociation/ClusterAssociationAlgorithm.h"

namespace lar_content
{

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

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

class TransverseAssociationAlgorithm : public ClusterAssociationAlgorithm
{
public:
    TransverseAssociationAlgorithm();

private:
    class LArTransverseCluster
    {
    public:
        LArTransverseCluster(const pandora::Cluster *const pSeedCluster, const pandora::ClusterVector &associatedClusters);

        const pandora::Cluster *GetSeedCluster() const;

        const pandora::ClusterVector &GetAssociatedClusters() const;

        const pandora::CartesianVector &GetInnerVertex() const;

        const pandora::CartesianVector &GetOuterVertex() const;

        const pandora::CartesianVector &GetDirection() const;

    private:
        const pandora::Cluster     *m_pSeedCluster;
        pandora::ClusterVector      m_associatedClusters;
        pandora::CartesianVector    m_innerVertex;
        pandora::CartesianVector    m_outerVertex;
        pandora::CartesianVector    m_direction;
    };

    typedef std::vector<LArTransverseCluster*> TransverseClusterList;

    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::ClusterSet> ClusterToClustersMap;
    typedef std::unordered_map<const pandora::CaloHit*, const pandora::Cluster*> HitToClusterMap;

    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);
    void GetListOfCleanClusters(const pandora::ClusterList *const pClusterList, pandora::ClusterVector &clusterVector) const;
    void PopulateClusterAssociationMap(const pandora::ClusterVector &clusterVector, ClusterAssociationMap &clusterAssociationMap) const;
    bool IsExtremalCluster(const bool isForward, const pandora::Cluster *const pCurrentCluster, const pandora::Cluster *const pTestCluster) const;

    void GetNearbyClusterMap(const pandora::ClusterVector &allClusters, ClusterToClustersMap &nearbyClusters) const;

    void SortInputClusters(const pandora::ClusterVector &inputClusters, pandora::ClusterVector &shortClusters,
        pandora::ClusterVector &transverseMediumClusters, pandora::ClusterVector &longitudinalMediumClusters,
        pandora::ClusterVector &longClusters) const;

    void FillReducedAssociationMap(const ClusterToClustersMap &nearbyClusters, const pandora::ClusterVector &firstVector,
        const pandora::ClusterVector &secondVector, ClusterAssociationMap &clusterAssociationMap) const;

    void FillAssociationMap(const ClusterToClustersMap &nearbyClusters, const pandora::ClusterVector &firstVector,
        const pandora::ClusterVector &secondVector, ClusterAssociationMap &firstAssociationMap,
        ClusterAssociationMap &secondAssociationMap) const;

    void FillTransverseClusterList(const ClusterToClustersMap &nearbyClusters, const  pandora::ClusterVector &inputClusters,
        const ClusterAssociationMap &inputAssociationMap, TransverseClusterList &transverseClusterList) const;

    void FillTransverseAssociationMap(const ClusterToClustersMap &nearbyClusters, const TransverseClusterList &transverseClusterList,
        const ClusterAssociationMap &transverseAssociationMap, ClusterAssociationMap &clusterAssociationMap) const;

    void GetAssociatedClusters(const ClusterToClustersMap &nearbyClusters, const pandora::Cluster *const pCluster,
        const ClusterAssociationMap &inputAssociationMap, pandora::ClusterVector &associatedClusters) const;

    bool IsAssociated(const bool isForward, const pandora::Cluster *const pCluster1, const pandora::Cluster *const pCluster2,
        const ClusterToClustersMap &nearbyClusters) const;

    bool IsTransverseAssociated(const pandora::Cluster *const pCluster1, const pandora::Cluster *const pCluster2,
        const ClusterToClustersMap &nearbyClusters) const;

    bool IsTransverseAssociated(const LArTransverseCluster *const pTransverseCluster1, const LArTransverseCluster *const pTransverseCluster2,
        const ClusterToClustersMap &nearbyClusters) const;

    bool IsTransverseAssociated(const LArTransverseCluster *const pTransverseCluster, const pandora::CartesianVector &testPosition) const;

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

    float GetTransverseSpan(const pandora::Cluster *const pCluster) const;

    float GetLongitudinalSpan(const pandora::Cluster *const pCluster) const;

    float GetTransverseSpan(const pandora::Cluster *const pCluster, const pandora::ClusterVector &associatedClusters) const;

    void GetExtremalCoordinatesX(const pandora::Cluster *const pCluster, float &minX, float &maxX) const;

    void GetExtremalCoordinatesZ(const pandora::Cluster *const pCluster, float &minZ, float &maxZ) const;

    void GetExtremalCoordinatesXZ(const pandora::Cluster *const pCluster, const bool useX, float &minXZ, float &maxXZ) const;

    void GetExtremalCoordinatesX(const pandora::Cluster *const pCluster, pandora::CartesianVector &innerCoordinate,
        pandora::CartesianVector &outerCoordinate) const;

    void FillReducedAssociationMap(const ClusterAssociationMap &inputAssociationMap, ClusterAssociationMap &outputAssociationMap) const;

    void FillReducedAssociationMap(const ClusterAssociationMap &firstAssociationMap, const ClusterAssociationMap &secondAssociationMap,
        const ClusterAssociationMap &secondAssociationMapSwapped, ClusterAssociationMap &clusterAssociationMap) const;

    void FillSymmetricAssociationMap(const ClusterAssociationMap &inputAssociationMap, ClusterAssociationMap &outputAssociationMap) const;

    void FinalizeClusterAssociationMap(const ClusterAssociationMap &inputAssociationMap, ClusterAssociationMap &outputAssociationMap) const;

    float        m_firstLengthCut;                   
    float        m_secondLengthCut;                  

    float        m_clusterWindow;                    
    float        m_clusterAngle;                     
    float        m_clusterCosAngle;                  
    float        m_clusterTanAngle;                  

    float        m_maxTransverseOverlap;             
    float        m_maxProjectedOverlap;              
    float        m_maxLongitudinalOverlap;           

    float        m_transverseClusterMinCosTheta;     
    float        m_transverseClusterMinLength;       
    float        m_transverseClusterMaxDisplacement; 

    float        m_searchRegionX;                    
    float        m_searchRegionZ;                    
};

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

inline const pandora::Cluster *TransverseAssociationAlgorithm::LArTransverseCluster::GetSeedCluster() const
{
    return m_pSeedCluster;
}

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

inline const pandora::ClusterVector &TransverseAssociationAlgorithm::LArTransverseCluster::GetAssociatedClusters() const
{
    return m_associatedClusters;
}

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

inline const pandora::CartesianVector &TransverseAssociationAlgorithm::LArTransverseCluster::GetInnerVertex() const
{
    return m_innerVertex;
}

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

inline const pandora::CartesianVector &TransverseAssociationAlgorithm::LArTransverseCluster::GetOuterVertex() const
{
    return m_outerVertex;
}

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

inline const pandora::CartesianVector &TransverseAssociationAlgorithm::LArTransverseCluster::GetDirection() const
{
    return m_direction;
}

} // namespace lar_content

#endif // #ifndef LAR_TRANSVERSE_ASSOCIATION_ALGORITHM_H