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