EventClusterValidationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_EVENT_CLUSTER_VALIDATION_ALGORITHM_H
#define LAR_EVENT_CLUSTER_VALIDATION_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

namespace lar_content
{

class EventClusterValidationAlgorithm : public pandora::Algorithm
{
private:
    struct ClusterMetrics
    {
        ClusterMetrics();

        std::vector<int> m_nRecoHits;
        std::vector<float> m_purities;
        std::vector<float> m_completenesses;
        int m_nHits;
        int m_nClusters;
        int m_nMainMCs;
    };

    enum ValidationType
    {
        ALL,
        SHOWER,
        TRACK
    };

public:
    struct CaloHitParents
    {
        CaloHitParents();

        const pandora::MCParticle *m_pMainMC;
        const pandora::Cluster *m_pCluster;
    };

    EventClusterValidationAlgorithm();

    ~EventClusterValidationAlgorithm();

private:
    typedef std::map<const pandora::Cluster *const, std::map<const pandora::MCParticle *const, int>> ContingencyTable;

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

    void GetMetrics(const std::map<const pandora::CaloHit *const, CaloHitParents> &hitParents, ClusterMetrics &metrics) const;

    float CalcRandIndex(std::map<const pandora::CaloHit *const, CaloHitParents> &hitParents) const;

    void GetHitParents(const pandora::CaloHitList &caloHits, const pandora::ClusterList &clusters,
        std::map<const pandora::CaloHit *const, CaloHitParents> &hitParents) const;

    void ApplyMCParticleMinSumHits(std::map<const pandora::CaloHit *const, CaloHitParents> &hitParents) const;

    std::map<const pandora::CaloHit *const, CaloHitParents> ApplyPDGCut(
        std::map<const pandora::CaloHit *const, CaloHitParents> &hitParents, const ValidationType &valType) const;

    void SetBranches(ClusterMetrics &metrics, float randIdx, std::string branchPrefix) const;

    int m_eventNumber;                           
    std::string m_fileName;                      
    std::string m_treeName;                      
    std::string m_caloHitListName;               
    std::vector<std::string> m_clusterListNames; 
    int m_minMCHitsPerView; 
};

} // namespace lar_content

#endif // LAR_EVENT_CLUSTER_VALIDATION_ALGORITHM_H