PerfectIsoHitRemovalAlgorithm.h

Go to the documentation of this file.

#ifndef PERFECTISOHITREMOVALALGORITHM_H
#define PERFECTISOHITREMOVALALGORITHM_H 1

#include "Pandora/Algorithm.h"

namespace april_content
{

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

    PerfectIsoHitRemovalAlgorithm();


protected:

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;

    pandora::StatusCode MergeCaloHits(const MCParticleToCaloHitListMap &mcParticleToCaloHitListMap,
                                      const MCParticleToClusterMap &mcParticleToClusterMap) const;

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

    //pandora::IntVector  m_particleIdList;               ///< list of particle ids of MCPFOs to be selected
    //bool                m_shouldUseOnlyECalHits;        ///< Whether to only use ecal hits in the clustering algorithm
    //bool                m_shouldUseIsolatedHits;        ///< Whether to use isolated hits in the clustering algorithm
    //bool                m_simpleMCParticleCollection;   ///< Whether to use simple mc particle collection mechanism, or full mechanism
    //float               m_minWeightFraction;            ///< The minimum mc particle calo hit weight for clustering consideration
    int                   m_CaloHitMCGetterFailures;
};

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

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

}

#endif  // PERFECTFRAGMENTREMOVALALGORITHM_H