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/EventSlicingBaseTool.h"

#include "larpandoracontent/LArObjects/LArThreeDSlidingConeFitResult.h"

#include <unordered_map>

namespace lar_content
{

typedef std::map<pandora::HitType, std::string> HitTypeToNameMap;

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 HitTypeToNameMap &caloHitListNames,
        const HitTypeToNameMap &clusterListNames, SliceList &sliceList);

private:
    void CopyAllHitsToSingleSlice(const pandora::Algorithm *const pAlgorithm, const HitTypeToNameMap &caloHitListNames, 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, SliceList &sliceList, ClusterToSliceIndexMap &clusterToSliceIndexMap) const;

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

    void GetRemainingClusters(const pandora::Algorithm *const pAlgorithm, const 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, 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 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