CheatingPfoCreationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_CHEATING_PFO_CREATION_ALGORITHM_H
#define LAR_CHEATING_PFO_CREATION_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

#include <unordered_map>

namespace lar_content
{

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

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

    typedef std::unordered_map<const pandora::MCParticle *, pandora::ClusterList> MCParticleToClusterListMap;

    void GetMCParticleToClusterListMap(const pandora::ClusterList *const pClusterList,
        const LArMCParticleHelper::MCRelationMap &mcPrimaryMap, MCParticleToClusterListMap &mcParticleToClusterListMap) const;

    void CreatePfos(const MCParticleToClusterListMap &mcParticleToClusterListMap) const;

    unsigned int GetNHitTypesAboveThreshold(const pandora::ClusterList &clusterList, const unsigned int nHitsThreshold) const;

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

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

    typedef std::map<pandora::HitType, unsigned int> HitTypeMap;
    typedef std::set<int> ParticleIdList;

    pandora::StringVector m_inputClusterListNames; 
    std::string m_outputPfoListName;               
    std::string m_outputVertexListName;            

    bool m_collapseToPrimaryMCParticles; 
    std::string m_mcParticleListName;    

    bool m_useOnlyAvailableClusters;    
    bool m_addVertices;                 
    bool m_replaceCurrentVertexList;    
    unsigned int m_minGoodHitTypes;     
    unsigned int m_nHitsForGoodHitType; 
    ParticleIdList m_particleIdList;    
};

} // namespace lar_content

#endif // #ifndef LAR_CHEATING_PFO_CREATION_ALGORITHM_H