runTauOnly_MLTopo Namespace Reference

Functions
	tauSpecialContent (flags, cfg)
	_run ()

Variables
	statusCode = None

Function Documentation

_run()

runTauOnly_MLTopo._run ( )

protected

Definition at line 25 of file runTauOnly_MLTopo.py.

def _run():
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
    # input
    from AthenaConfiguration.TestDefaults import defaultTestFiles, defaultConditionsTags
    flags.Exec.MaxEvents = 20
    flags.Input.Files = defaultTestFiles.RDO_RUN3
    flags.IOVDb.GlobalTag = defaultConditionsTags.RUN3_MC
    from AthenaConfiguration.Enums import ProductionStep
    flags.Common.ProductionStep = ProductionStep.Reconstruction
 
    # output
    flags.Output.ESDFileName = "myESD.pool.root"
    flags.Output.AODFileName = "myAOD.pool.root"
 
    # uncomment given something like export ATHENA_CORE_NUMBER=2
    # flags.Concurrency.NumThreads = 2
 
    # Setup detector flags
    from AthenaConfiguration.DetectorConfigFlags import setupDetectorFlags
    setupDetectorFlags(flags, None, use_metadata=True,
                       toggle_geometry=True, keep_beampipe=True)
 
    # Schedule Tau Reco
    from tauRec.ConfigurationHelpers import StandaloneTauRecoFlags
    StandaloneTauRecoFlags(flags)
    flags.Tau.TauRec.SeedJetCollection = "AntiKt4MLTopoJets"
    flags.Tau.doTauEleRMRec = False
    flags.lock()
 
    from RecJobTransforms.RecoSteering import RecoSteering
    acc = RecoSteering(flags)
 
    # keep only tau containers
    tauSpecialContent(flags,acc)
 
    # Special message service configuration
    from DigitizationConfig.DigitizationSteering import DigitizationMessageSvcCfg
    acc.merge(DigitizationMessageSvcCfg(flags))
 
    from AthenaConfiguration.Utils import setupLoggingLevels
    setupLoggingLevels(flags, acc)
 
    # Print reco domain status
    from RecJobTransforms.RecoConfigFlags import printRecoFlags
    printRecoFlags(flags)
 
    # running
    statusCode = acc.run()
 
    return statusCode
 
 

tauSpecialContent()

runTauOnly_MLTopo.tauSpecialContent	(		flags,
			cfg )

Definition at line 15 of file runTauOnly_MLTopo.py.

def tauSpecialContent(flags,cfg):
    from OutputStreamAthenaPool.OutputStreamConfig import outputStreamName
    StreamAOD = cfg.getEventAlgo(outputStreamName("AOD"))
    newList = [x for x in StreamAOD.ItemList if "Tau" in x]
    StreamAOD.ItemList = newList
 
    StreamESD = cfg.getEventAlgo(outputStreamName("ESD"))
    newList = [x for x in StreamESD.ItemList if "Tau" in x]
    StreamESD.ItemList = newList
 

Variable Documentation

statusCode

runTauOnly_MLTopo.statusCode = None

Definition at line 79 of file runTauOnly_MLTopo.py.