VertexBasedPfoRecoveryAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_VERTEX_BASED_PFO_RECOVERY_ALGORITHM_H
#define LAR_VERTEX_BASED_PFO_RECOVERY_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.h"
#include "larpandoracontent/LArObjects/LArPointingCluster.h"

namespace lar_content
{

class VertexBasedPfoRecoveryAlgorithm : public pandora::Algorithm
{
public:
    VertexBasedPfoRecoveryAlgorithm();

private:
    pandora::StatusCode Run();

    class Particle
    {
    public:
        Particle(const pandora::Cluster *const pClusterU, const pandora::Cluster *const pClusterV, const pandora::Cluster *const pClusterW);

        const pandora::Cluster  *m_pClusterU;    
        const pandora::Cluster  *m_pClusterV;    
        const pandora::Cluster  *m_pClusterW;    
    };

    typedef std::vector<Particle> ParticleList;

    pandora::StatusCode GetAvailableClusters(const pandora::StringVector inputClusterListName, pandora::ClusterVector &clusterVector) const;

    void BuildSlidingFitResultMap(const pandora::ClusterVector &clusterVector, TwoDSlidingFitResultMap &slidingFitResultMap) const;

    void SelectVertexClusters(const pandora::Vertex *const pVertex, const TwoDSlidingFitResultMap &slidingFitResultMap,
        const pandora::ClusterVector &inputClusters, pandora::ClusterVector &outputClusters) const;

    void MatchThreeViews(const pandora::Vertex *const pVertex, const TwoDSlidingFitResultMap &slidingFitResultMap,
        const pandora::ClusterVector &selectedClusters, pandora::ClusterSet &vetoList, ParticleList &particleList) const;

    void MatchTwoViews(const pandora::Vertex *const pVertex, const TwoDSlidingFitResultMap &slidingFitResultMap,
        const pandora::ClusterVector &selectedClusters, pandora::ClusterSet &vetoList, ParticleList &particleList) const;

    void GetBestChi2(const pandora::Vertex *const pVertex, const TwoDSlidingFitResultMap &slidingFitResultMap,
        const pandora::ClusterVector &clusters1, const pandora::ClusterVector &clusters2, const pandora::ClusterVector &clusters3,
        const pandora::Cluster *&pBestCluster1, const pandora::Cluster *&pBestCluster2, const pandora::Cluster *&pBestCluster3,
        float &chi2) const;

    void GetBestChi2(const pandora::Vertex *const pVertex, const TwoDSlidingFitResultMap &slidingFitResultMap,
        const pandora::ClusterVector &clusters1, const pandora::ClusterVector &clusters2,
        const pandora::Cluster *&pBestCluster1, const pandora::Cluster *&pBestCluster2, float &chi2) const;

    float GetChi2(const pandora::Vertex *const pVertex, const LArPointingCluster &pointingCluster1,
        const LArPointingCluster &pointingCluster2) const;

    float GetChi2(const pandora::Vertex *const pVertex, const LArPointingCluster &pointingCluster1,
        const LArPointingCluster &pointingCluster2, const LArPointingCluster &pointingCluster3) const;

    void SelectAvailableClusters(const pandora::ClusterSet &vetoList, const pandora::ClusterVector &inputVector,
        pandora::ClusterVector &outputVector) const;

    void SelectClusters(const pandora::HitType hitType, const pandora::ClusterVector &inputVector,
        pandora::ClusterVector &outputVector) const;

    const LArPointingCluster::Vertex &GetInnerVertex(const pandora::CartesianVector &vertex, const LArPointingCluster &cluster) const;

    const LArPointingCluster::Vertex &GetOuterVertex(const pandora::CartesianVector &vertex, const LArPointingCluster &cluster) const;

    void BuildParticles(const ParticleList &particleList);

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

    pandora::StringVector m_inputClusterListNames;        
    std::string           m_outputPfoListName;            

    unsigned int          m_slidingFitHalfWindow;         
    float                 m_maxLongitudinalDisplacement;  
    float                 m_maxTransverseDisplacement;    
    float                 m_twoViewChi2Cut;               
    float                 m_threeViewChi2Cut;             
};

} // namespace lar_content

#endif // #ifndef LAR_VERTEX_BASED_PFO_RECOVERY_ALGORITHM_H