PerfectFragmentRemovalAlgorithm.h

Go to the documentation of this file.

#ifndef PERFECTFRAGMENTREMOVALALGORITHM_H
#define PERFECTFRAGMENTREMOVALALGORITHM_H 1

#include "Pandora/Algorithm.h"

namespace april_content
{

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

    PerfectFragmentRemovalAlgorithm();

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


    void SimpleMCParticleCollection(const pandora::Cluster* const pCaloHit, MCParticleToClusterListMap &mcParticleToClusterListMap) const;

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

    void AddToClusterListMap(const pandora::Cluster *const pClusterToAdd, const pandora::MCParticle *const pMCParticle, MCParticleToClusterListMap &mcParticleToClusterListMap) const;

    pandora::StatusCode MergeClusters(const MCParticleToClusterListMap &mcParticleToClusterListMap) 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
};

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

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

}

#endif  // PERFECTFRAGMENTREMOVALALGORITHM_H