FourLeptonVertexing.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
 
 
def FourLeptonVertexerCfg(flags, name="FourLeptonVertexAlg", **kwargs):
    result = ComponentAccumulator()
    
    from MuonSelectorTools.MuonSelectorToolsConfig import MuonSelectionToolCfg
    kwargs.setdefault("MuonSelectionTool", result.popToolsAndMerge(MuonSelectionToolCfg(flags, 
                                                                MaxEta=2.7,
                                                                DisablePtCuts=True,
                                                                MuQuality=2, 
                                                                )) )
 
    
    from ElectronPhotonSelectorTools.AsgElectronLikelihoodToolsConfig import AsgElectronLikelihoodToolCfg
    from ElectronPhotonSelectorTools.ElectronLikelihoodToolMapping import  electronLHmenu
    from ElectronPhotonSelectorTools.LikelihoodEnums import LikeEnum
    from AthenaConfiguration.Enums import LHCPeriod
    kwargs.setdefault("ElectronSelectionTool", result.popToolsAndMerge(AsgElectronLikelihoodToolCfg(flags,
                                                                           name= "ElectronSelTool",
                                                                           quality = LikeEnum.VeryLoose,
                                                                           menu=electronLHmenu.offlineMC21 if flags.GeoModel.Run >= LHCPeriod.Run3 else electronLHmenu.offlineMC20)))
    
    from TrkConfig.TrkVKalVrtFitterConfig import TrkVKalVrtFitterCfg
    kwargs.setdefault("VertexFitter", result.popToolsAndMerge(
        TrkVKalVrtFitterCfg(flags, FirstMeasuredPoint = False)))
    kwargs.setdefault("MinMuonPt", 4000)
    kwargs.setdefault("MinElecPt", 6000)    
    vtx_tool = CompFactory.DerivationFramework.FourLeptonVertexingAlgorithm(name , **kwargs)
    result.addEventAlgo(vtx_tool, primary = True)
    return result
 
if __name__ == "__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
    args = flags.fillFromArgs()
 
    flags.Input.Files = args.input    
    flags.Concurrency.NumThreads=args.threads
    flags.Concurrency.NumConcurrentEvents=args.threads
 
    flags.lock()
    flags.dump()
 
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    cfg = MainServicesCfg(flags)
    msgService = cfg.getService('MessageSvc')
    msgService.Format = "S:%s E:%e % F%128W%S%7W%R%T  %0W%M"
 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    cfg.merge(PoolReadCfg(flags))
    cfg.merge(FourLeptonVertexerCfg(flags))
 
    sc = cfg.run(flags.Exec.MaxEvents)
    if not sc.isSuccess():
        exit(1)