CosmicRayTaggingMonitoringTool.h

Go to the documentation of this file.

#ifndef LAR_COSMIC_RAY_TAGGING_MONITORING_TOOL_H
#define LAR_COSMIC_RAY_TAGGING_MONITORING_TOOL_H 1

#include "larpandoracontent/LArControlFlow/MasterAlgorithm.h"

#include "larpandoracontent/LArHelpers/LArFormattingHelper.h"
#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

#include <unordered_map>

namespace lar_content
{

class CosmicRayTaggingMonitoringTool : public CosmicRayTaggingBaseTool
{
public:
    CosmicRayTaggingMonitoringTool();

    void FindAmbiguousPfos(const pandora::PfoList &parentCosmicRayPfos, pandora::PfoList &ambiguousPfos, const MasterAlgorithm *const pAlgorithm);

private:
    enum Classification
    {
        CR_MUON,
        CR_OTHER,
        TARGET,
        FRAGMENTED,
        ABSORBED,
        MIXED,
        SPARSE,
        UNCLASSIFIED
    };

    typedef std::map<const pandora::ParticleFlowObject *, float> PfoToFloatMap;
    typedef std::map<const pandora::ParticleFlowObject *, Classification> PfoClassificationMap;

    void CalculatePfoMetrics(const LArMCParticleHelper::PfoToMCParticleHitSharingMap &hitSharingMap,
        const LArMCParticleHelper::PfoContributionMap &pfoToCaloHitListMap, const LArMCParticleHelper::MCContributionMapVector &targetsToGoodHitsMaps,
        PfoToFloatMap &pfoSignificanceMap, PfoToFloatMap &pfoPurityMap, PfoClassificationMap &pfoClassificationMap) const;

    bool IsMainMCParticleMuon(const pandora::ParticleFlowObject *const pPfo) const;

    Classification ClassifyPfo(const unsigned int &nHits, const float &significance, const float &purity, const bool isMuon) const;

    LArFormattingHelper::Color GetClassificationColor(const Classification &classification) const;

    std::string GetClassificationName(const Classification &classification) const;

    void PrintPfoTable(const pandora::PfoVector &orderedPfoVector,
        const LArMCParticleHelper::PfoContributionMap &pfoToReconstructable2DHitsMap, const PfoToFloatMap &pfoPurityMap,
        const PfoToFloatMap &pfoSignificanceMap, const PfoClassificationMap &pfoClassificationMap, const pandora::PfoList &ambiguousPfos) const;

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

    LArMCParticleHelper::PrimaryParameters m_parameters; 
    unsigned int m_minHitsToConsiderTagging;             
    float m_minPurity;                                   
    float m_minImpurity;                                 
    float m_minSignificance;                             
    std::string m_caloHitList2D;                         
};

} // namespace lar_content

#endif // #ifndef LAR_COSMIC_RAY_TAGGING_MONITORING_TOOL_H