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