TrackClusterCreationAlgorithm.h

Go to the documentation of this file.

 
#ifndef LAR_TRACK_CLUSTER_CREATION_ALGORITHM_H
#define LAR_TRACK_CLUSTER_CREATION_ALGORITHM_H 1
 
#include "Pandora/Algorithm.h"
 
#include <unordered_map>
 
namespace lar_content
{
 
class TrackClusterCreationAlgorithm : public pandora::Algorithm
{
public:
    TrackClusterCreationAlgorithm();
 
private:
    class HitAssociation
    {
    public:
        HitAssociation(const pandora::CaloHit *const pPrimaryTarget, const float primaryDistanceSquared);
 
        void SetSecondaryTarget(const pandora::CaloHit *const pSecondaryTarget, const float secondaryDistanceSquared);
 
        const pandora::CaloHit *GetPrimaryTarget() const;
 
        const pandora::CaloHit *GetSecondaryTarget() const;
 
        float GetPrimaryDistanceSquared() const;
 
        float GetSecondaryDistanceSquared() const;
 
    private:
        const pandora::CaloHit *m_pPrimaryTarget;   
        const pandora::CaloHit *m_pSecondaryTarget; 
        float m_primaryDistanceSquared;             
        float m_secondaryDistanceSquared;           
    };
 
    typedef std::unordered_map<const pandora::CaloHit *, HitAssociation> HitAssociationMap;
    typedef std::unordered_map<const pandora::CaloHit *, const pandora::CaloHit *> HitJoinMap;
    typedef std::unordered_map<const pandora::CaloHit *, const pandora::Cluster *> HitToClusterMap;
 
    pandora::StatusCode Run();
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);
 
    pandora::StatusCode FilterCaloHits(const pandora::CaloHitList *const pCaloHitList, pandora::OrderedCaloHitList &selectedCaloHitList,
        pandora::OrderedCaloHitList &rejectedCaloHitList) const;
 
    pandora::StatusCode AddFilteredCaloHits(const pandora::OrderedCaloHitList &selectedCaloHitList,
        const pandora::OrderedCaloHitList &rejectedCaloHitList, HitToClusterMap &hitToClusterMap) const;
 
    void MakePrimaryAssociations(const pandora::OrderedCaloHitList &orderedCaloHitList, HitAssociationMap &forwardHitAssociationMap,
        HitAssociationMap &backwardHitAssociationMap) const;
 
    void MakeSecondaryAssociations(const pandora::OrderedCaloHitList &orderedCaloHitList, HitAssociationMap &forwardHitAssociationMap,
        HitAssociationMap &backwardHitAssociationMap) const;
 
    void IdentifyJoins(const pandora::OrderedCaloHitList &orderedCaloHitList, const HitAssociationMap &forwardHitAssociationMap,
        const HitAssociationMap &backwardHitAssociationMap, HitJoinMap &hitJoinMap) const;
 
    void CreateClusters(const pandora::OrderedCaloHitList &orderedCaloHitList, const HitJoinMap &hitJoinMap, HitToClusterMap &hitToClusterMap) const;
 
    void CreatePrimaryAssociation(const pandora::CaloHit *const pCaloHitI, const pandora::CaloHit *const pCaloHitJ,
        HitAssociationMap &forwardHitAssociationMap, HitAssociationMap &backwardHitAssociationMap) const;
 
    void CreateSecondaryAssociation(const pandora::CaloHit *const pCaloHitI, const pandora::CaloHit *const pCaloHitJ,
        HitAssociationMap &forwardHitAssociationMap, HitAssociationMap &backwardHitAssociationMap) const;
 
    const pandora::CaloHit *GetJoinHit(const pandora::CaloHit *const pCaloHit, const HitAssociationMap &hitAssociationMapI,
        const HitAssociationMap &hitAssociationMapJ) const;
 
    const pandora::CaloHit *TraceHitAssociation(const pandora::CaloHit *const pCaloHit, const HitAssociationMap &hitAssociationMapI,
        const HitAssociationMap &hitAssociationMapJ, unsigned int &nSteps) const;
 
    bool m_mergeBackFilteredHits;        
    unsigned int m_maxGapLayers;         
    float m_maxCaloHitSeparationSquared; 
    float m_minCaloHitSeparationSquared; 
    float m_closeSeparationSquared;      
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline TrackClusterCreationAlgorithm::HitAssociation::HitAssociation(const pandora::CaloHit *const pPrimaryTarget, const float primaryDistanceSquared) :
    m_pPrimaryTarget(pPrimaryTarget),
    m_pSecondaryTarget(NULL),
    m_primaryDistanceSquared(primaryDistanceSquared),
    m_secondaryDistanceSquared(std::numeric_limits<float>::max())
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline void TrackClusterCreationAlgorithm::HitAssociation::SetSecondaryTarget(const pandora::CaloHit *const pSecondaryTarget, const float secondaryDistanceSquared)
{
    m_pSecondaryTarget = pSecondaryTarget;
    m_secondaryDistanceSquared = secondaryDistanceSquared;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CaloHit *TrackClusterCreationAlgorithm::HitAssociation::GetPrimaryTarget() const
{
    return m_pPrimaryTarget;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CaloHit *TrackClusterCreationAlgorithm::HitAssociation::GetSecondaryTarget() const
{
    return m_pSecondaryTarget;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline float TrackClusterCreationAlgorithm::HitAssociation::GetPrimaryDistanceSquared() const
{
    return m_primaryDistanceSquared;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline float TrackClusterCreationAlgorithm::HitAssociation::GetSecondaryDistanceSquared() const
{
    return m_secondaryDistanceSquared;
}
 
} // namespace lar_content
 
#endif // #ifndef LAR_TRACK_CLUSTER_CREATION_ALGORITHM_H