IsolatedClusterMopUpAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_ISOLATED_CLUSTER_MOP_UP_ALGORITHM_H
#define LAR_ISOLATED_CLUSTER_MOP_UP_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArTwoDReco/LArClusterMopUp/ClusterMopUpBaseAlgorithm.h"

#include <unordered_map>

namespace lar_content
{

template<typename, unsigned int> class KDTreeLinkerAlgo;
template<typename, unsigned int> class KDTreeNodeInfoT;

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

class IsolatedClusterMopUpAlgorithm : public ClusterMopUpBaseAlgorithm
{
public:
    IsolatedClusterMopUpAlgorithm();

private:
    void ClusterMopUp(const pandora::ClusterList &pfoClusters, const pandora::ClusterList &remnantClusters) const;

    void DissolveClustersToHits(const pandora::ClusterList &clusterList, pandora::CaloHitList &caloHitList) const;

    typedef std::unordered_map<const pandora::CaloHit*, const pandora::Cluster*> CaloHitToClusterMap;

    void GetCaloHitToClusterMap(const pandora::CaloHitList &caloHitList, const pandora::ClusterList &clusterList, CaloHitToClusterMap &caloHitToClusterMap) const;

    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    typedef KDTreeLinkerAlgo<const pandora::CaloHit*, 2> HitKDTree2D;
    typedef KDTreeNodeInfoT<const pandora::CaloHit*, 2> HitKDNode2D;
    typedef std::vector<HitKDNode2D> HitKDNode2DList;

    unsigned int    m_maxCaloHitsInCluster;     
    float           m_maxHitClusterDistance;    
    bool            m_addHitsAsIsolated;        
};

} // namespace lar_content

#endif // #ifndef LAR_ISOLATED_CLUSTER_MOP_UP_ALGORITHM_H