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