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