FourLeptonVertexingAlgorithm.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
#ifndef DERIVATIONFRAMEWORK_FourLeptonVertexingAlgorithm_H
#define DERIVATIONFRAMEWORK_FourLeptonVertexingAlgorithm_H
 
// Gaudi & Athena basics
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "CxxUtils/checker_macros.h"
#include "EgammaAnalysisInterfaces/IAsgElectronLikelihoodTool.h"
#include "MuonAnalysisInterfaces/IMuonSelectionTool.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "TrkVertexFitterInterfaces/IVertexFitter.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEventInfo/EventInfo.h"
#include "xAODMuon/MuonContainer.h"
#include "xAODTracking/VertexContainer.h"
namespace DerivationFramework {
    class FourLeptonVertexingAlgorithm : public AthReentrantAlgorithm {
    public:
        FourLeptonVertexingAlgorithm(const std::string& n, ISvcLocator* p);
        StatusCode initialize() override;
        StatusCode execute(const EventContext& ctx) const override;
        StatusCode finalize() override;
 
        using LeptonQuadruplet = std::array<const xAOD::IParticle*, 4>;
 
    private:
        bool passSelection(const xAOD::Electron* elec) const;
        bool passSelection(const xAOD::Muon* muon) const;
        bool passSelection(const LeptonQuadruplet& quad) const;
 
        std::vector<LeptonQuadruplet> buildAllQuadruplets(const EventContext& ctx) const;
 
        std::unique_ptr<xAOD::Vertex> fitQuadruplet(const EventContext& ctx, const LeptonQuadruplet& quad) const;
 
        const xAOD::TrackParticle* trackParticle(const xAOD::IParticle* part) const;
        int charge(const xAOD::IParticle* part) const;

        SG::ReadHandleKey<xAOD::MuonContainer> m_muonKey{this, "MuonContainer", "Muons",
                                                         "Location of the input muon container in the StoreGate"};
        SG::ReadHandleKey<xAOD::ElectronContainer> m_elecKey{this, "ElectronContainer", "Electrons",
                                                             "Location of the electron container in the store gate"};
        SG::ReadHandleKey<xAOD::EventInfo> m_evtKey{this, "EventInfoKey", "EventInfo"};
 
        ToolHandle<CP::IMuonSelectionTool> m_muonSelTool{this, "MuonSelectionTool", "",
                                                         "Applied preselection on the muons. Will be ignored if empty"};
        ToolHandle<IAsgElectronLikelihoodTool> m_elecSelTool{this, "ElectronSelectionTool", "",
                                                             "Applied preselection on the electrons. Will be ignored if empty"};
 
        ToolHandle<Trk::IVertexFitter> m_fitter{this, "VertexFitter", "", "Vertex fitter tool"};
 
        Gaudi::Property<float> m_minMuonPt{this, "MinMuonPt", 3. * Gaudi::Units::GeV,
                                           "Minimum pt cut applied on the muon -- Unfortunately not handled by the MST"};
        Gaudi::Property<float> m_minElecPt{this, "MinElecPt", 4.5 * Gaudi::Units::GeV, " Minimum pt cut applied on the electron"};
 
        using MuonTrk = xAOD::Muon::TrackParticleType;
        Gaudi::Property<int> m_muonTrkProp{this, "PickMuonTrk", MuonTrk::InnerDetectorTrackParticle,
                                           "Pick the proper track particle from the muon"};
        MuonTrk m_muonTrk{MuonTrk::Primary};
 
        Gaudi::Property<bool> m_elecUseGSF{this, "PickElecGsfTrk", true,
                                           "If this property is set to true it will pick up the electron GSF track"};
 
        Gaudi::Property<float> m_z0Cut{this, "DeltaZ0Cut", 7.5 * Gaudi::Units::mm,
                                       "All leptons in the quadruplet have to be seperated from each other by maximum"};
 
        
        Gaudi::Property<float> m_lowSFOS_Cut{this, "LowSFOSMass", 3.5 * Gaudi::Units::GeV, "Minimal mass for the lower SFOS pair"};
        Gaudi::Property<float> m_highPair_Cut{this, "HighMassPair", 40.*Gaudi::Units::GeV, "Minimum mass for the largest lepton pair"};
 
        Gaudi::Property<float> m_LeadPtCut{this, "LeadingLeptonPt", 15. *Gaudi::Units::GeV,"Minimal momentum of the leading lepton in the quad "};
        Gaudi::Property<float> m_SubLeadPtCut{this, "SubLeadingLeptonPt", 10. *Gaudi::Units::GeV,"Minimal momentum of the leading lepton in the quad "};
     
        Gaudi::Property<float> m_VtxChi2Cut{this, "VertexChi2Cut", 20, "Maximal chii n.D.o.F cut on the reconstructed vertex" };
        
        
        SG::WriteHandleKey<xAOD::VertexContainer> m_vtxKey{this, "OutVertexContainer", "FourLeptonVertices", "Output vertex container"};
        Gaudi::Property<bool> m_pruneCov{this, "PruneCovariance", true, "Clears the covariance vector"};
        Gaudi::Property<bool> m_pruneWeight{this, "PruneTrackWeights", true, "Clears the track weight vector"};
        
        mutable std::array<std::atomic<unsigned int>, 8>  m_num_lep ATLAS_THREAD_SAFE {};
        mutable std::array<std::atomic<unsigned int>, 8>  m_num_ele ATLAS_THREAD_SAFE {};
        mutable std::array<std::atomic<unsigned int>, 8>  m_num_muo ATLAS_THREAD_SAFE {};
     
        mutable std::atomic<unsigned int>  m_tried_fits{0};
        mutable std::atomic<unsigned int>  m_good_fits{0};
        
 
    };
 
}  // namespace DerivationFramework
#endif