MuonPerformanceAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONPERFORMANCEALG_H
#define MUONPERFORMANCEALG_H
 
#include <fstream>
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODEventInfo/EventInfo.h"
#include "xAODMuon/Muon.h"
#include "xAODMuon/MuonContainer.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "StoreGate/ReadDecorHandleKey.h"
 
class MuonPerformanceAlg : public AthAlgorithm {
public:
    // Algorithm Constructor
    MuonPerformanceAlg(const std::string& name, ISvcLocator* pSvcLocator);
    virtual ~MuonPerformanceAlg() = default;
    // Gaudi algorithm hooks
    virtual StatusCode initialize() override;
    virtual StatusCode execute() override;
    virtual StatusCode finalize() override;
 
    virtual unsigned int cardinality() const override final { return 1;}
 
    // statistics to be counted
    std::vector<std::string> m_muonLocationList;
 
private:
    bool passID(const xAOD::TrackParticle*, bool debug) const;

    bool m_writeToFile;
    std::string m_fileName;

    std::ofstream m_fileOutput;
 
    // Containers
    SG::ReadHandleKey<xAOD::MuonContainer> m_muonsNameKey{this, "MuonContainerName", "Muons", "muon container"};
    SG::ReadDecorHandleKey<xAOD::MuonContainer> m_muonTruthParticleKey {this,"muonTruthParticleLinkKey", m_muonsNameKey, "truthParticleLink"};
    unsigned int m_nevents;
    std::vector<std::string> m_hitCutString;
    std::vector<int> m_ntruth;
    std::vector<int> m_ntruth5;
    std::vector<int> m_ntruth10;
    std::vector<int> m_nfound;
    std::vector<int> m_nfound5;
    std::vector<int> m_nfound10;
    std::vector<int> m_nfoundr;
    std::vector<int> m_nfoundr5;
    std::vector<int> m_nfoundr10;
    std::vector<int> m_nreco;
    std::vector<int> m_nreco5;
    std::vector<int> m_nreco10;
 
    SG::ReadHandleKey<xAOD::EventInfo> m_eventInfo{this, "EventInfo", "EventInfo", "event info"};
    SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthMuons{this, "TruthMuons", "MuonTruthParticles", "truth muons"};
    SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_truthMuonRecoMuonLinkKey {this,"truthMuonRecoMuonLinkKey",m_truthMuons, "recoMuonLink"};
    SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_truthMuonTruthType {this,"truthMuonTruthType",m_truthMuons, "truthType"};
    SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_truthMuonTruthOrigin {this,"truthMuonTruthOrigin",m_truthMuons, "truthOrigin"};
 
    int m_runNumber;
    int m_eventNumber;
 
    void print(const std::string& txt, const xAOD::TruthParticle* muon);
    void print(const std::string& txt, const xAOD::Muon* muon);
 
    IntegerArrayProperty m_pdgsToBeConsidered;
    std::set<int> m_selectedPdgs;  // set storing particle PDG's considered for matching
    bool selectPdg(int pdg) const { return m_selectedPdgs.count(pdg); }
};
 
#endif  // MUONPERFORMANCEALG