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;
    typedef std::set<unsigned int> UIntSet;

    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