ClusterGrowingAlgorithm.h

Go to the documentation of this file.

 
#ifndef LAR_CLUSTER_GROWING_ALGORITHM_H
#define LAR_CLUSTER_GROWING_ALGORITHM_H 1
 
#include "Pandora/Algorithm.h"
 
#include <unordered_map>
 
namespace lar_content
{
 
class ClusterGrowingAlgorithm : public pandora::Algorithm
{
public:
    ClusterGrowingAlgorithm();
 
protected:
    virtual pandora::StatusCode Run();
    virtual pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);
 
    typedef std::unordered_map<const pandora::Cluster *, pandora::ClusterList> ClusterMergeMap;
 
    virtual void GetListOfCleanClusters(const pandora::ClusterList *const pClusterList, pandora::ClusterVector &cleanClusters) const = 0;
 
    virtual void GetListOfSeedClusters(const pandora::ClusterVector &cleanClusters, pandora::ClusterVector &seedClusters) const = 0;
 
private:
    void GetListOfNonSeedClusters(const pandora::ClusterVector &inputClusters, const pandora::ClusterVector &seedClusters,
        pandora::ClusterVector &nonSeedClusters) const;
 
    void PopulateClusterMergeMap(
        const pandora::ClusterVector &seedClusters, const pandora::ClusterVector &nonSeedClusters, ClusterMergeMap &clusterMergeMap) const;
 
    void MergeClusters(const ClusterMergeMap &clusterMergeMap) const;
 
    std::string m_inputClusterListName; 
 
    float m_maxClusterSeparation; 
};
 
} // namespace lar_content
 
#endif // #ifndef LAR_CLUSTER_GROWING_ALGORITHM_H