IsoHitRemovalAlgorithm.h

Go to the documentation of this file.

#ifndef ISOHITREMOVALALGORITHM_H
#define ISOHITREMOVALALGORITHM_H 1

#include "Pandora/Algorithm.h"
#include "APRILUtility/KDTreeLinkerAlgoT.h"

typedef april_content::KDTreeLinkerAlgo<const pandora::CaloHit*, 4> HitKDTree;
typedef april_content::KDTreeNodeInfoT<const pandora::CaloHit*, 4> HitKDNode;
typedef map<const pandora::CaloHit*, const pandora::Cluster*> HitsToClustersMap;

namespace april_content
{

class IsoHitRemovalAlgorithm : public pandora::Algorithm
{
public:
    class Factory : public pandora::AlgorithmFactory
    {
    public:
        pandora::Algorithm *CreateAlgorithm() const;
    };

    IsoHitRemovalAlgorithm();


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

    typedef std::map<const pandora::MCParticle*, pandora::CaloHitList*> MCParticleToCaloHitListMap;
    typedef std::map<const pandora::MCParticle*, const pandora::Cluster*> MCParticleToClusterMap;

    void SimpleMCParticleCaloHitListCollection(const pandora::CaloHit* const pCaloHit, 
                                               MCParticleToCaloHitListMap &mcParticleToCaloHitListMap);

    void SimpleMCParticleClusterCollection(const pandora::Cluster* const pCluster, MCParticleToClusterMap &mcParticleToClusterMap);

    //void FullMCParticleCollection(const pandora::CaloHit *const pCaloHit, MCParticleToHitListMap &mcParticleToHitListMap) const;

    void AddToCaloHitListMap(const pandora::CaloHit* const pCaloHitToAdd, 
         const pandora::MCParticle *const pMCParticle, MCParticleToCaloHitListMap &mcParticleToCaloHitListMap) const;

    void AddToClusterMap(const pandora::Cluster* const pClusterToAdd, 
         const pandora::MCParticle *const pMCParticle, MCParticleToClusterMap &mcParticleToClusterMap) const;

    void CreateCluster(const pandora::CaloHitList *const caloHitList) const;

    int GetMCParticle(const pandora::Cluster* pCluster) const;

    pandora::StatusCode Test();
    pandora::StatusCode GetIsoHits(pandora::CaloHitList& caloHitList, MCParticleToCaloHitListMap& mcParticleToCaloHitListMap);
    pandora::StatusCode GetClusters(pandora::ClusterList& clusterList, MCParticleToClusterMap& mcParticleToClusterMap);
  
    pandora::StatusCode MCMergeCaloHits(const MCParticleToCaloHitListMap &mcParticleToCaloHitListMap,
                                      const MCParticleToClusterMap &mcParticleToClusterMap) const;

    pandora::StatusCode MergeCaloHits(const pandora::CaloHitList& isoHitList, const pandora::ClusterList& clusterList);
  
    pandora::StatusCode BuildKDTree(HitKDTree& hits_kdtree);
  
    pandora::StatusCode SearchNearbyCaloHits(const pandora::CaloHit* pCaloHit, 
            std::vector<const pandora::CaloHit*>& nearbyHits, float wideX, float wideY, float wideZ, int layers);

    pandora::StatusCode MeanShiftClustering(pandora::CaloHitList& isoHitList, std::vector<pandora::CaloHitList>& clusterHitsCollection);

private:
    float                 m_kernelBandwidth;
    float                 m_clusterEpsilon;
};

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

inline pandora::Algorithm *IsoHitRemovalAlgorithm::Factory::CreateAlgorithm() const
{
    return new IsoHitRemovalAlgorithm();
}

}

#endif  // ISOHITREMOVALALGORITHM_H