2 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
3 from AthenaConfiguration.ComponentFactory
import CompFactory
6 """Configure the EVNT->xAOD converter alg"""
11 xAODTruthCnvAlg = CompFactory.xAODMaker.xAODTruthCnvAlg
12 acc.addEventAlgo(xAODTruthCnvAlg(name,**kwargs))
18 writeInTimePileUpTruth=
False
22 if flags.Digitization.PileUp
and flags.Digitization.DigiSteeringConf
in [
'StandardPileUpToolsAlg',
'StandardInTimeOnlyTruthPileUpToolsAlg',
'StandardInTimeOnlyGeantinoTruthPileUpToolsAlg']:
23 writeInTimePileUpTruth =
True
25 kwargs.setdefault(
'WriteInTimePileUpTruth', writeInTimePileUpTruth)
26 kwargs.setdefault(
'AODContainerName',
'TruthEvent')
28 algo = CompFactory.xAODMaker.xAODTruthCnvAlg(name, **kwargs)
29 acc.addEventAlgo(algo, primary =
True)
31 from OutputStreamAthenaPool.OutputStreamConfig
import addToESD,addToAOD
32 toAOD = [
"xAOD::TruthEventContainer#TruthEvents",
"xAOD::TruthEventAuxContainer#TruthEventsAux.",
33 "xAOD::TruthVertexContainer#TruthVertices",
"xAOD::TruthVertexAuxContainer#TruthVerticesAux.",
34 "xAOD::TruthParticleContainer#TruthParticles",
"xAOD::TruthParticleAuxContainer#TruthParticlesAux.",
37 if "xAOD::TruthParticleContainer#TruthPileupParticles" in flags.Input.TypedCollections:
38 toAOD.append(
"xAOD::TruthParticleContainer#TruthPileupParticles")
39 toAOD.append(
"xAOD::TruthParticleAuxContainer#TruthPileupParticlesAux.")
41 if writeInTimePileUpTruth:
42 toAOD.append(
"xAOD::TruthPileupEventContainer#TruthPileupEvents")
43 toAOD.append(
"xAOD::TruthPileupEventAuxContainer#TruthPileupEventsAux.")
47 if flags.Output.doWriteESD:
48 acc.merge(
addToESD(flags, toAOD+toESD))
49 if flags.Output.doWriteAOD:
56 HepMCTruthReader = CompFactory.HepMCTruthReader(
58 HepMCContainerKey = flags.HepMCContainerKey,
59 Do4momPtEtaPhi = flags.Do4momPtEtaPhi
61 acc.addEventAlgo(HepMCTruthReader)
66 xAODTruthReader = CompFactory.xAODReader.xAODTruthReader(
68 xAODTruthEventContainerKey = flags.xAODTruthEventContainerKey,
70 DoPUEventPrintout = flags.DoPUEventPrintout,
71 Do4momPtEtaPhi = flags.Do4momPtEtaPhi
73 acc.addEventAlgo(xAODTruthReader)