CheatingClusterCreationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_CHEATING_CLUSTER_CREATION_ALGORITHM_H
#define LAR_CHEATING_CLUSTER_CREATION_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

#include <unordered_map>

namespace lar_content
{

class CheatingClusterCreationAlgorithm : public pandora::Algorithm
{
public:
    CheatingClusterCreationAlgorithm();

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

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

    void GetMCParticleToHitListMap(MCParticleToHitListMap &mcParticleToHitListMap) const;

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

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

    void CreateClusters(const MCParticleToHitListMap &mcParticleToHitListMap) const;

    bool                m_collapseToPrimaryMCParticles; 
    std::string         m_mcParticleListName;           

    pandora::IntVector  m_particleIdList;               
};

} // namespace lar_content

#endif // #ifndef LAR_CHEATING_CLUSTER_CREATION_ALGORITHM_H