EventSlicingTool.h

Go to the documentation of this file.

#ifndef LAR_EVENT_SLICING_TOOL_H
#define LAR_EVENT_SLICING_TOOL_H 1

#include "larpandoracontent/LArControlFlow/SlicingAlgorithm.h"

#include "larpandoracontent/LArObjects/LArThreeDSlidingConeFitResult.h"

#include <unordered_map>

namespace lar_content
{

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

class SimpleCone;

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

class EventSlicingTool : public EventSlicingBaseTool
{
public:
    EventSlicingTool();

    void RunSlicing(const pandora::Algorithm *const pAlgorithm, const SlicingAlgorithm::HitTypeToNameMap &caloHitListNames,
        const SlicingAlgorithm::HitTypeToNameMap &clusterListNames, SlicingAlgorithm::SliceList &sliceList);

private:
    void CopyAllHitsToSingleSlice(const pandora::Algorithm *const pAlgorithm, const SlicingAlgorithm::HitTypeToNameMap &caloHitListNames,
        SlicingAlgorithm::SliceList &sliceList) const;

    typedef std::unordered_map<const pandora::Cluster*, const pandora::ParticleFlowObject*> ClusterToPfoMap;

    void GetThreeDClusters(const pandora::Algorithm *const pAlgorithm, const std::string &pfoListName, pandora::ClusterList &clusters3D,
        ClusterToPfoMap &clusterToPfoMap) const;

    typedef std::vector<pandora::ClusterVector> ClusterSliceList;

    void GetClusterSliceList(const pandora::ClusterList &trackClusters3D, const pandora::ClusterList &showerClusters3D,
        ClusterSliceList &clusterSliceList) const;

    void CollectAssociatedClusters(const pandora::Cluster *const pClusterInSlice, const pandora::ClusterVector &candidateClusters, const ThreeDSlidingFitResultMap &trackFitResults,
        const ThreeDSlidingConeFitResultMap &showerConeFitResults, pandora::ClusterVector &clusterSlice, pandora::ClusterSet &usedClusters) const;

    bool PassPointing(const pandora::Cluster *const pClusterInSlice, const pandora::Cluster *const pCandidateCluster, const ThreeDSlidingFitResultMap &trackFitResults) const;

    bool PassProximity(const pandora::Cluster *const pClusterInSlice, const pandora::Cluster *const pCandidateCluster) const;

    bool PassShowerCone(const pandora::Cluster *const pConeCluster, const pandora::Cluster *const pNearbyCluster, const ThreeDSlidingConeFitResultMap &showerConeFitResults) const;

    bool CheckClosestApproach(const LArPointingCluster &cluster1, const LArPointingCluster &cluster2) const;

    bool CheckClosestApproach(const LArPointingCluster::Vertex &vertex1, const LArPointingCluster::Vertex &vertex2) const;

    bool IsNode(const LArPointingCluster &cluster1, const LArPointingCluster &cluster2) const;

    bool IsEmission(const LArPointingCluster &cluster1, const LArPointingCluster &cluster2) const;

    typedef std::unordered_map<const pandora::Cluster*, unsigned int> ClusterToSliceIndexMap;

    void CreateSlices(const ClusterSliceList &clusterSliceList, SlicingAlgorithm::SliceList &sliceList, ClusterToSliceIndexMap &clusterToSliceIndexMap) const;

    void CopyPfoHitsToSlices(const ClusterToSliceIndexMap &clusterToSliceIndexMap, const ClusterToPfoMap &clusterToPfoMap, SlicingAlgorithm::SliceList &sliceList,
        pandora::ClusterSet &assignedClusters) const;

    void GetRemainingClusters(const pandora::Algorithm *const pAlgorithm, const SlicingAlgorithm::HitTypeToNameMap &clusterListNames,
        const pandora::ClusterSet &assignedClusters, pandora::ClusterList &remainingClusters) const;

    void GetRemainingClusters(const pandora::Algorithm *const pAlgorithm, const std::string &clusterListName,
        const pandora::ClusterSet &assignedClusters, pandora::ClusterList &remainingClusters) const;

    void AssignRemainingHitsToSlices(const pandora::ClusterList &remainingClusters, const ClusterToSliceIndexMap &clusterToSliceIndexMap,
        SlicingAlgorithm::SliceList &sliceList) const;

    typedef KDTreeLinkerAlgo<const pandora::CartesianVector*, 2> PointKDTree2D;
    typedef KDTreeNodeInfoT<const pandora::CartesianVector*, 2> PointKDNode2D;
    typedef std::vector<PointKDNode2D> PointKDNode2DList;

    typedef std::list<const pandora::CartesianVector*> PointList;
    typedef std::unordered_map<const pandora::CartesianVector*, unsigned int> PointToSliceIndexMap;

    void GetKDTreeEntries2D(const SlicingAlgorithm::SliceList &sliceList, PointList &pointsU, PointList &pointsV, PointList &pointsW,
        PointToSliceIndexMap &pointToSliceIndexMap) const;

    void GetKDTreeEntries3D(const ClusterToSliceIndexMap &clusterToSliceIndexMap, PointList &pointsU, PointList &pointsV,
        PointList &pointsW, PointToSliceIndexMap &pointToSliceIndexMap) const;

    const PointKDNode2D *MatchClusterToSlice(const pandora::Cluster *const pCluster2D, PointKDTree2D &kdTree) const;

    static bool SortPoints(const pandora::CartesianVector *const pLhs, const pandora::CartesianVector *const pRhs);

    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    std::string     m_trackPfoListName;                 
    std::string     m_showerPfoListName;                

    unsigned int    m_minHitsPer3DCluster;              
    unsigned int    m_min3DHitsToSeedNewSlice;          
    unsigned int    m_halfWindowLayers;                 

    bool            m_usePointingAssociation;           
    float           m_minVertexLongitudinalDistance;    
    float           m_maxVertexLongitudinalDistance;    
    float           m_maxVertexTransverseDistance;      
    float           m_vertexAngularAllowance;           
    float           m_maxClosestApproach;               
    float           m_maxInterceptDistance;             

    bool            m_useProximityAssociation;          
    float           m_maxHitSeparationSquared;          

    bool            m_useShowerConeAssociation;         
    unsigned int    m_nConeFitLayers;                   
    unsigned int    m_nConeFits;                        
    float           m_coneLengthMultiplier;             
    float           m_maxConeLength;                    
    float           m_coneTanHalfAngle1;                
    float           m_coneBoundedFraction1;             
    float           m_coneTanHalfAngle2;                
    float           m_coneBoundedFraction2;             

    bool            m_use3DProjectionsInHitPickUp;      
};

} // namespace lar_content

#endif // #ifndef LAR_EVENT_SLICING_TOOL_H