PerfectNeutralHitRecoveryAlgorithm.h

Go to the documentation of this file.

#ifndef PERFECTNEUTRALHITRECOVERYALGORITHM_H
#define PERFECTNEUTRALHITRECOVERYALGORITHM_H 1

#include "Pandora/Algorithm.h"

namespace april_content
{

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

    PerfectNeutralHitRecoveryAlgorithm();

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;
    typedef std::map<const pandora::MCParticle*, pandora::ClusterList*> MCParticleToClusterListMap;
    typedef std::map<const pandora::CaloHit*, const pandora::Cluster*> CaloHitToClusterMap;

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

    //void SimpleMCParticleClusterCollection(const pandora::Cluster* const pCluster, MCParticleToClusterMap &mcParticleToClusterMap) const;
  
    void SimpleMCParticleClusterListCollection(const pandora::Cluster *const pCluster, 
            MCParticleToClusterListMap &mcParticleToClusterListMap) const;

    //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 AddToClusterListMap(const pandora::Cluster* const pClusterToAdd, 
         const pandora::MCParticle *const pMCParticle, MCParticleToClusterListMap &mcParticleToClusterListMap) const;
  
    void ClearCaloHitListMap(MCParticleToCaloHitListMap &mcParticleToCaloHitListMap) const;
  
    void CreateNeutralClusters(MCParticleToCaloHitListMap& mcParticleToCaloHitListMap) const;
  
    void RemoveClusterCaloHitAssociations(CaloHitToClusterMap& caloHitToClusterRemoveMap) const;

    void AddClusterCaloHitAssociations(CaloHitToClusterMap& caloHitToClusterAddMap) const;

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

};

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

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

}

#endif  // PERFECTNEUTRALHITRECOVERYALGORITHM_H