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