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