DiMuonTaggingAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_DIMUONTAGGINGALG_H
#define DERIVATIONFRAMEWORK_DIMUONTAGGINGALG_H
 
 
// Gaudi & Athena basics
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "TriggerMatchingTool/IMatchingTool.h"
#include "MuonAnalysisInterfaces/IMuonSelectionTool.h"
 
// xAOD header files
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandle.h"
#include "xAODMuon/MuonContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
 
 
namespace DerivationFramework {
 
    class DiMuonTaggingAlg : public AthReentrantAlgorithm {
    public:
        DiMuonTaggingAlg(const std::string& name, ISvcLocator* pSvcLocator);
 
        virtual ~DiMuonTaggingAlg() = default;
        // Athena algtool's Hooks
        StatusCode initialize() override;
 
        virtual StatusCode execute(const EventContext& ctx) const override;
 
    private:
        using TrackPassDecor = SG::WriteDecorHandle<xAOD::TrackParticleContainer, bool>;
        bool passKinematicCuts(const xAOD::IParticle* mu, const float ptMin, const float absEtaMax) const;
 
        bool passMuonCuts(const xAOD::Muon* muon, const float ptMin, const float absEtaMax, const bool applyQuality) const;
 
        bool passTrigger(const xAOD::Muon* muon, const std::vector<std::string>& trigList) const;
 
        template <class probe_type> bool muonPairCheck(const xAOD::Muon* mu1, const probe_type* mu2) const;
 
        void maskNearbyIDtracks(const xAOD::IParticle* mu, TrackPassDecor& decor) const;
        
        
        Gaudi::Property<bool> m_applyTrig{this, "applyTrigger", true,
                                         "Flag deciding whether the trigger selection should be applied"};
        
        Gaudi::Property<std::vector<std::string>> m_orTrigs{this, "OrTrigs", {}};
        Gaudi::Property<std::vector<std::string>> m_andTrigs{this, "AndTrigs", {}};
 
        
        ToolHandle<Trig::IMatchingTool> m_matchingTool{this, "TrigMatchingTool", "",
                                                       "Tool to access the trigger decision"};
 
 
        Gaudi::Property<float> m_triggerMatchDeltaR{this, "TriggerMatchDeltaR", 0.1};
 
 
        SG::ReadHandleKey<xAOD::MuonContainer> m_muonSGKey{this, "MuonContainerKey", "Muons"};
 
        SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackSGKey{this, "TrackContainerKey", "InDetTrackParticles"};
 
        SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthSGKey{this, "TruthKey", ""};
 
        SG::WriteDecorHandleKey<xAOD::MuonContainer> m_muonKeepKey{this, "MuonKeepKey", m_muonSGKey, "", 
                                "Key to whitelist the muon for writeout. Will be overwritten by BranchPrefix property"};
        SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_trkKeepKey{this, "TrackKeepKey", m_trackSGKey, "",  
                                "Key to whitelist the tracks for writeout. Will be overwritten by BranchPreFix property"};
        SG::WriteHandleKey<int> m_skimmingKey{this, "SkimmingKey", "", "Set via BranchPreFixProperty + DIMU_pass"};
 
        ToolHandle<CP::IMuonSelectionTool> m_muonSelTool{this, "SelectionTool", ""};
        Gaudi::Property<float> m_mu1PtMin{this, "Mu1PtMin", -1};
        Gaudi::Property<float> m_mu1AbsEtaMax{this, "Mu1AbsEtaMax", 10};
        Gaudi::Property<bool>  m_applyQualityMu1{this, "Mu1RequireQual", false};
 
        Gaudi::Property<float> m_mu2PtMin{this, "Mu2PtMin", -1};
        Gaudi::Property<float> m_mu2AbsEtaMax{this, "Mu2AbsEtaMax", 10};
        Gaudi::Property<bool>  m_applyQualityMu2{this, "Mu2RequireQual", false};
        
        Gaudi::Property<bool> m_isMC{this, "isMC", false};
        Gaudi::Property<bool> m_useTrackProbe{this, "UseTrackProbe", false};
        Gaudi::Property<std::string> m_br_prefix{this, "BranchPrefix", ""};
 
        Gaudi::Property<bool> m_requireOS{this, "OppositeCharge", true};
        Gaudi::Property<float> m_dPhiMin{this, "PairDPhiMin", -1};
        Gaudi::Property<float> m_invariantMassLow{this, "InvariantMassLow", 2.0 * Gaudi::Units::GeV};
        Gaudi::Property<float> m_invariantMassHigh{this, "InvariantMassHigh",-1.*Gaudi::Units::GeV};
 
        Gaudi::Property<float> m_thinningConeSize{this, "IDTrackThinningConeSize", 0.4};
 
        float m_invariantMassLow2{0};
        float m_invariantMassHigh2{0};
        float m_thinningConeSize2{0};
    };
}  // namespace DerivationFramework
#endif  // DERIVATIONFRAMEWORK_SKIMMINGTOOLEXAMPLE_H