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