PerfectClusteringAlgorithm.h

Go to the documentation of this file.

#ifndef PERFECTCLUSTERINGALGORITHM_H
#define PERFECTCLUSTERINGALGORITHM_H 1

#include "Pandora/Algorithm.h"

namespace april_content
{

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

    PerfectClusteringAlgorithm();

protected:
    virtual bool SelectMCParticlesForClustering(const pandora::MCParticle *const pMCParticle) const;

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

    typedef std::map<const pandora::MCParticle*, pandora::CaloHitList*> MCParticleToHitListMap;

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

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

    void AddToHitListMap(const pandora::CaloHit *const pCaloHitToAdd, const pandora::MCParticle *const pMCParticle, MCParticleToHitListMap &mcParticleToHitListMap) const;

    void CreateClusters(const MCParticleToHitListMap &mcParticleToHitListMap) const;

    pandora::IntVector  m_particleIdList;               
    bool                m_shouldUseOnlyECalHits;        
    bool                m_shouldUseIsolatedHits;        
    bool                m_simpleMCParticleCollection;   
    float               m_minWeightFraction;            
};

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

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

}

#endif  //  PERFECTCLUSTERINGALGORITHM_H