EFTrackingSmearingConfig.py

Go to the documentation of this file.

 
#Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaCommon.Logging import logging
_log = logging.getLogger(__name__)
 
def EFTrackingSmearingCfg(flags, name = "EFTrackingSmearingAlg", **kwargs):
    # set common parameters
    histSvc = CompFactory.THistSvc(Output=[name +" DATAFILE='"+name+ ".root' OPT='RECREATE'"])
 
    result = ComponentAccumulator()    
    result.addService(histSvc) 
 
    alg = CompFactory.EFTrackingSmearingAlg(
        name = name,
        RootStreamName = name,
        RootDirName = "/EFTSmearing/",
        **kwargs
    )
 
    sf_str=str(kwargs['SmearingScaleFactor']).replace(".","p")
    if kwargs['SmearTruthParticle']:
        alg.OutputTruthParticleContainer = "TruthParticle_smeared_SF"+sf_str
    else:
        alg.OutputTrackParticleContainer = "InDetTrackParticles_smeared_SF"+sf_str
        
    result.addEventAlgo(alg)
    return result
 
 
  
 
if __name__ == "__main__":
    
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaCommon.Constants import INFO, DEBUG
    flags = initConfigFlags()    
    flags.Input.Files = defaultTestFiles.AOD_RUN3_MC
    flags.Input.isMC=True
    flags.Exec.MaxEvents = 5
    flags.lock()
 
    # create basic infrastructure
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    acc = MainServicesCfg(flags)        
    acc.getService("MessageSvc").debugLimit = 100000
 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    acc.merge(PoolReadCfg(flags))
    
    # needed to access the decorators for the truth particles
    from InDetPhysValMonitoring.InDetPhysValDecorationConfig import AddDecoratorCfg
    acc.merge(AddDecoratorCfg(flags))
    
    TestsmearFactor = 2
    TestEfficiencyFactor=1
    # example to smear the track particles 
    smearerTrack = EFTrackingSmearingCfg(
        flags,
        name = "testTrack",
        InputTrackParticleContainer = "InDetTrackParticles",
        OutputTracksPtCutGeV = 1,
        SmearingScaleFactor = TestsmearFactor,
        SmearedTrackEfficiency = TestEfficiencyFactor,
        ParameterizedTrackEfficiency = False,
        SmearTruthParticle = False,
        EnableMonitoring = True,
        OutputLevel = INFO)
    acc.merge(smearerTrack)
    
    
    # example to smear the truth particles 
    smearerTruth = EFTrackingSmearingCfg(
        flags,
        name = "testTruth",
        InputTruthParticleContainer = "TruthParticles",
        OutputTracksPtCutGeV = 1,
        SmearingScaleFactor = TestsmearFactor,
        SmearedTrackEfficiency = TestEfficiencyFactor,
        ParameterizedTrackEfficiency = False,
        SmearTruthParticle = True,
        ParameterizedTrackEfficiency_LRT = True,
        SmearedTrackEfficiency_d0low_LRT = .001,
        SmearedTrackEfficiency_d0high_LRT = 400.,
        EnableMonitoring = True,
        OutputLevel = DEBUG)
    acc.merge(smearerTruth)
    
    
    # validation of the smeared tracks and truth particles
    validationAlg = CompFactory.EFTrackingSmearMonAlg(
        name = "EFTrakingSmearMonAlg",
        OutputLevel = INFO,
        InputTrackParticleContainer = "InDetTrackParticles",
        InputTruthParticleContainer = "TruthParticles",
        SmearedTrackParticleContainer = "InDetTrackParticles_smeared_SF"+str(TestsmearFactor),
        SmearedTruthParticleContainer = "TruthParticle_smeared_SF"+str(TestsmearFactor),)
    acc.addEventAlgo(validationAlg)
    
    acc.wasMerged()
    
    # below is validation
    acc.printConfig(withDetails=True, summariseProps=True)
    
    
    # run the job
    status = acc.run()
 
 
    # report the execution status (0 ok, else error)
    import sys
    sys.exit(not status.isSuccess())