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 pClusterToEnlarge, const pandora::Cluster *const pClusterToDelete, const bool isForwardMerge,
        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