RPhiFeatureTool.h

Go to the documentation of this file.

#ifndef LAR_RPHI_FEATURE_TOOL_H
#define LAR_RPHI_FEATURE_TOOL_H 1

#include "larpandoracontent/LArVertex/VertexSelectionBaseAlgorithm.h"

#include "larpandoracontent/LArUtility/KDTreeLinkerAlgoT.h"

namespace lar_content
{

class RPhiFeatureTool : public VertexSelectionBaseAlgorithm::VertexFeatureTool
{
public:
    RPhiFeatureTool();

    void Run(LArMvaHelper::MvaFeatureVector &featureVector, const VertexSelectionBaseAlgorithm * const pAlgorithm, const pandora::Vertex * const pVertex,
        const VertexSelectionBaseAlgorithm::SlidingFitDataListMap &, const VertexSelectionBaseAlgorithm::ClusterListMap &,
        const VertexSelectionBaseAlgorithm::KDTreeMap &kdTreeMap, const VertexSelectionBaseAlgorithm::ShowerClusterListMap &,
        const float beamDeweightingScore, float &bestFastScore);

private:
    class KernelEstimate
    {
    public:
        KernelEstimate(const float sigma);

        float Sample(const float x) const;

        typedef std::multimap<float, float> ContributionList;   

        const ContributionList &GetContributionList() const;

        float GetSigma() const;

        void AddContribution(const float x, const float weight);

    private:
        ContributionList            m_contributionList;         
        const float                 m_sigma;                    
    };

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

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

    float GetFastScore(const KernelEstimate &kernelEstimateU, const KernelEstimate &kernelEstimateV, const KernelEstimate &kernelEstimateW) const;

    float GetMidwayScore(const KernelEstimate &kernelEstimateU, const KernelEstimate &kernelEstimateV, const KernelEstimate &kernelEstimateW) const;

    float GetFullScore(const KernelEstimate &kernelEstimateU, const KernelEstimate &kernelEstimateV, const KernelEstimate &kernelEstimateW) const;

    void FillKernelEstimate(const pandora::Vertex *const pVertex, const pandora::HitType hitType, VertexSelectionBaseAlgorithm::HitKDTree2D &kdTree, KernelEstimate &kernelEstimate) const;

    bool AcceptVertexLocation(const pandora::Vertex *const pVertex, const pandora::VertexList &selectedVertexList) const;

    float atan2Fast(const float y, const float x) const;

    bool            m_fastScoreCheck;               
    bool            m_fastScoreOnly;                
    bool            m_fullScore;                    

    float           m_kernelEstimateSigma;          
    float           m_kappa;                        
    float           m_maxHitVertexDisplacement1D;   

    float           m_minFastScoreFraction;         
    unsigned int    m_fastHistogramNPhiBins;        
    float           m_fastHistogramPhiMin;          
    float           m_fastHistogramPhiMax;          

    bool            m_enableFolding;                
};

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

inline RPhiFeatureTool::KernelEstimate::KernelEstimate(const float sigma) :
    m_sigma(sigma)
{
    if (m_sigma < std::numeric_limits<float>::epsilon())
        throw pandora::StatusCodeException(pandora::STATUS_CODE_INVALID_PARAMETER);
}

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

inline const RPhiFeatureTool::KernelEstimate::ContributionList &RPhiFeatureTool::KernelEstimate::GetContributionList() const
{
    return m_contributionList;
}

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

inline float RPhiFeatureTool::KernelEstimate::GetSigma() const
{
    return m_sigma;
}

} // namespace lar_content

#endif // #ifndef LAR_RPHI_FEATURE_TOOL_H