ClusterAssociationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_CLUSTER_ASSOCIATION_ALGORITHM_H
#define LAR_CLUSTER_ASSOCIATION_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include <unordered_map>

namespace lar_content
{

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

protected:
    virtual pandora::StatusCode Run();
    virtual pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    class ClusterAssociation
    {
    public:
        pandora::ClusterSet m_forwardAssociations;  
        pandora::ClusterSet m_backwardAssociations; 
    };

    typedef std::unordered_map<const pandora::Cluster *, ClusterAssociation> ClusterAssociationMap;

    virtual void GetListOfCleanClusters(const pandora::ClusterList *const pClusterList, pandora::ClusterVector &clusterVector) const = 0;

    virtual void PopulateClusterAssociationMap(const pandora::ClusterVector &clusterVector, ClusterAssociationMap &clusterAssociationMap) const = 0;

    virtual bool IsExtremalCluster(
        const bool isForward, const pandora::Cluster *const pCurrentCluster, const pandora::Cluster *const pTestCluster) const = 0;

private:
    void UnambiguousPropagation(const pandora::Cluster *const pCluster, const bool isForward, ClusterAssociationMap &clusterAssociationMap) const;

    void AmbiguousPropagation(const pandora::Cluster *const pCluster, const bool isForward, ClusterAssociationMap &clusterAssociationMap) const;

    void UpdateForUnambiguousMerge(const pandora::Cluster *const pClusterToEnlarge, const pandora::Cluster *const pClusterToDelete,
        const bool isForwardMerge, ClusterAssociationMap &clusterAssociationMap) const;

    void UpdateForAmbiguousMerge(const pandora::Cluster *const pCluster, ClusterAssociationMap &clusterAssociationMap) const;

    void NavigateAlongAssociations(const ClusterAssociationMap &clusterAssociationMap, const pandora::Cluster *const pCluster,
        const bool isForward, const pandora::Cluster *&pExtremalCluster, pandora::ClusterSet &clusterSet) const;

    mutable bool m_mergeMade;

    bool m_resolveAmbiguousAssociations; 
};

} // namespace lar_content

#endif // #ifndef LAR_CLUSTER_ASSOCIATION_ALGORITHM_H