ClusterExtensionAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_CLUSTER_EXTENSION_ALGORITHM_H
#define LAR_CLUSTER_EXTENSION_ALGORITHM_H 1

#include "larpandoracontent/LArTwoDReco/LArClusterAssociation/ClusterMergingAlgorithm.h"

#include <unordered_map>

namespace lar_content
{

class ClusterExtensionAlgorithm : public ClusterMergingAlgorithm
{
protected:
    void PopulateClusterMergeMap(const pandora::ClusterVector &clusterVector, ClusterMergeMap &clusterMergeMatrix) const;

    class ClusterAssociation
    {
    public:
        enum VertexType
        {
            UNDEFINED = 0,
            INNER     = 1,
            OUTER     = 2
        };

        enum AssociationType
        {
            NONE   = 0,
            WEAK   = 1,
            STRONG = 2
        };

        ClusterAssociation(const VertexType parent, const VertexType daughter, const AssociationType association, const float fom);

        VertexType GetParent() const;

        VertexType GetDaughter() const;

        AssociationType GetAssociation() const;

        float GetFigureOfMerit() const;

    private:
        VertexType      m_parent;           
        VertexType      m_daughter;         
        AssociationType m_association;      
        float           m_fom;              
    };

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

    virtual void FillClusterAssociationMatrix(const pandora::ClusterVector &clusterVector, ClusterAssociationMatrix &clusterAssociationMatrix) const = 0;

    virtual void FillClusterMergeMap(const ClusterAssociationMatrix &clusterAssociationMatrix, ClusterMergeMap &clusterMergeMap) const = 0;
};

//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------

inline ClusterExtensionAlgorithm::ClusterAssociation::ClusterAssociation(const VertexType parent, const VertexType daughter,
        const AssociationType association, const float fom) :
    m_parent(parent),
    m_daughter(daughter),
    m_association(association),
    m_fom(fom)
{
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline ClusterExtensionAlgorithm::ClusterAssociation::VertexType ClusterExtensionAlgorithm::ClusterAssociation::GetParent() const
{
    return m_parent;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline ClusterExtensionAlgorithm::ClusterAssociation::VertexType ClusterExtensionAlgorithm::ClusterAssociation::GetDaughter() const
{
    return m_daughter;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline ClusterExtensionAlgorithm::ClusterAssociation::AssociationType ClusterExtensionAlgorithm::ClusterAssociation::GetAssociation() const
{
    return m_association;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline float ClusterExtensionAlgorithm::ClusterAssociation::GetFigureOfMerit() const
{
    return m_fom;
}

} // namespace lar_content

#endif // #ifndef LAR_CLUSTER_EXTENSION_ALGORITHM_H