CosmicRayBaseMatchingAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_COSMIC_RAY_BASE_MATCHING_ALGORITHM_H
#define LAR_COSMIC_RAY_BASE_MATCHING_ALGORITHM_H 1

#include "Pandora/Algorithm.h"
#include "Pandora/AlgorithmHeaders.h"

#include <unordered_map>

namespace lar_content
{

class CosmicRayBaseMatchingAlgorithm : public pandora::Algorithm
{
protected:
    pandora::StatusCode Run();
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    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;
    typedef std::unordered_map<const pandora::Cluster*, pandora::ClusterList> ClusterAssociationMap;

    virtual void SelectCleanClusters(const pandora::ClusterVector &inputVector, pandora::ClusterVector &outputVector) const = 0;

    virtual bool MatchClusters(const pandora::Cluster *const pCluster1, const pandora::Cluster *const pCluster2) const = 0;

    virtual bool CheckMatchedClusters3D(const pandora::Cluster *const pCluster1, const pandora::Cluster *const pCluster2,
        const pandora::Cluster *const pCluster3) const = 0;

    virtual void SetPfoParameters(const CosmicRayBaseMatchingAlgorithm::Particle &protoParticle, 
        PandoraContentApi::ParticleFlowObject::Parameters &pfoParameters) const = 0;

private:
    pandora::StatusCode GetAvailableClusters(const std::string inputClusterListName, pandora::ClusterVector &clusterVector) const;
    void MatchClusters(const pandora::ClusterVector &clusterVector1, const pandora::ClusterVector &clusterVector2,
        CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters12) const;

    void MatchThreeViews(const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters12,
        const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters23,
        const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters31,
        ParticleList &particleList) const;

    void MatchTwoViews(const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters12,
        const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters23,
        const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters31,
        ParticleList &particleList) const;

    void MatchTwoViews(const CosmicRayBaseMatchingAlgorithm::ClusterAssociationMap &matchedClusters12,
        ParticleList &particleList) const;

    void ResolveAmbiguities(const ParticleList &inputList, ParticleList &outputList) const;

    void BuildParticles(const ParticleList &particleList);

    std::string    m_inputClusterListNameU;        
    std::string    m_inputClusterListNameV;        
    std::string    m_inputClusterListNameW;        
    std::string    m_outputPfoListName;            
};

} // namespace lar_content

#endif // #ifndef LAR_COSMIC_RAY_BASE_MATCHING_ALGORITHM_H