PerfectChargedClusterSeparationAlgorithm.h

Go to the documentation of this file.

#ifndef PERFECTCHARGEDCLUSTERSEPARATIONALGORITHM_H
#define PERFECTCHARGEDCLUSTERSEPARATIONALGORITHM_H 1

#include "Pandora/Algorithm.h"

namespace april_content
{

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

    PerfectChargedClusterSeparationAlgorithm();

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

    //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 ClusterSeparation(pandora::ClusterList& localClusterList, MCParticleToCaloHitListMap& mcParticleToCaloHitListMap);

    void AddToClusterMap(const pandora::CaloHit* pCaloHit,  const pandora::MCParticle* pMCParticle, 
                        MCParticleToCaloHitListMap& mcParticleToCaloHitListMap);
  
    pandora::StatusCode CreateClusters(pandora::ClusterList& clusterList, MCParticleToCaloHitListMap& mcParticleToCaloHitListMap) const;

};

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

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

}

#endif  // PERFECTCHARGEDCLUSTERSEPARATIONALGORITHM_H