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/LArPointingCluster.h"
#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.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