FragmentRemovalHelper.h

Go to the documentation of this file.

#ifndef APRIL_FRAGMENT_REMOVAL_HELPER_H
#define APRIL_FRAGMENT_REMOVAL_HELPER_H 1

#include "Pandora/PandoraInternal.h"
#include "Pandora/StatusCodes.h"

namespace april_content
{

class ClusterContact
{
public:
    class Parameters
    {
    public:
        float           m_coneCosineHalfAngle1;         
        float           m_closeHitDistance1;            
        float           m_closeHitDistance2;            
        float           m_minCosOpeningAngle;           
        float           m_distanceThreshold;            
    };

    ClusterContact(const pandora::Pandora &pandora, const pandora::Cluster *const pDaughterCluster, const pandora::Cluster *const pParentCluster,
        const Parameters &parameters);

    const pandora::Cluster *GetDaughterCluster() const;

    const pandora::Cluster *GetParentCluster() const;

    unsigned int GetNContactLayers() const;

    float GetContactFraction() const;

    float GetDistanceToClosestHit() const;

    float GetCloseHitFraction1() const;

    float GetCloseHitFraction2() const;

    float GetConeFraction1() const;

protected:
    void HitDistanceComparison(const pandora::Cluster *const pDaughterCluster, const pandora::Cluster *const pParentCluster, const Parameters &parameters);

    const pandora::Cluster     *m_pDaughterCluster;         
    const pandora::Cluster     *m_pParentCluster;           

    unsigned int                m_nContactLayers;           
    float                       m_contactFraction;          
    float                       m_coneFraction1;            
    float                       m_closeHitFraction1;        
    float                       m_closeHitFraction2;        
    float                       m_distanceToClosestHit;     
};

typedef std::vector<ClusterContact> ClusterContactVector;
typedef std::map<const pandora::Cluster *, ClusterContactVector> ClusterContactMap;

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

class FragmentRemovalHelper
{
public:
    static float GetFractionOfCloseHits(const pandora::Cluster *const pClusterI, const pandora::Cluster *const pClusterJ, const float distanceThreshold);

    static float GetFractionOfHitsInCone(const pandora::Pandora &pandora, const pandora::Cluster *const pClusterI, const pandora::Cluster *const pClusterJ,
        const float coneCosineHalfAngle);

    static float GetFractionOfHitsInCone(const pandora::Cluster *const pCluster, const pandora::Track *const pTrack, const float coneCosineHalfAngle);

    static float GetFractionOfHitsInCone(const pandora::Cluster *const pCluster, const pandora::CartesianVector &coneApex,
        const pandora::CartesianVector &coneDirection, const float coneCosineHalfAngle);

    static unsigned int GetNLayersCrossed(const pandora::Pandora &pandora, const pandora::Helix &helix, const float zStart,
        const float zEnd, const unsigned int nSamplingPoints = 100);

    static pandora::StatusCode GetClusterHelixDistance(const pandora::Cluster *const pCluster, const pandora::Helix &helix,
        const unsigned int startLayer, const unsigned int endLayer, const unsigned int maxOccupiedLayers, float &closestDistanceToHit,
        float &meanDistanceToHits);

    static pandora::StatusCode GetClusterContactDetails(const pandora::Cluster *const pClusterI, const pandora::Cluster *const pClusterJ,
        const float distanceThreshold, unsigned int &nContactLayers, float &contactFraction);

    static float GetEMEnergyWeightedLayerSeparation(const pandora::Cluster *const pClusterI, const pandora::Cluster *const pClusterJ);

    static pandora::CartesianVector GetEMEnergyWeightedPosition(const pandora::Cluster *const pCluster);
};

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

inline const pandora::Cluster *ClusterContact::GetDaughterCluster() const
{
    return m_pDaughterCluster;
}

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

inline const pandora::Cluster *ClusterContact::GetParentCluster() const
{
    return m_pParentCluster;
}

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

inline unsigned int ClusterContact::GetNContactLayers() const
{
    return m_nContactLayers;
}

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

inline float ClusterContact::GetContactFraction() const
{
    return m_contactFraction;
}

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

inline float ClusterContact::GetDistanceToClosestHit() const
{
    return m_distanceToClosestHit;
}

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

inline float ClusterContact::GetCloseHitFraction1() const
{
    return m_closeHitFraction1;
}

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

inline float ClusterContact::GetCloseHitFraction2() const
{
    return m_closeHitFraction2;
}

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

inline float ClusterContact::GetConeFraction1() const
{
    return m_coneFraction1;
}

} 

#endif