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;

    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