7 if extensions_list
is None:
10 from OutputStreamAthenaPool.OutputStreamConfig
import addToESD, addToAOD
15 excludedAuxData = (
'-clusterAssociation.-TTVA_AMVFVertices_forReco.-AssoClustersUFO'
16 '.-TTVA_AMVFWeights_forReco')
18 excludedAuxData +=
'.-'.
join([
''] + FTAG_AUXDATA)
21 excludedAuxData +=
'.-TTVA_AMVFVertices.-TTVA_AMVFWeights'
24 from DerivationFrameworkInDet.IDTIDE
import IDTIDE_AOD_EXCLUDED_AUXDATA
25 excludedAuxData +=
'.-'.
join([
''] + IDTIDE_AOD_EXCLUDED_AUXDATA)
26 from DerivationFrameworkInDet.IDTRKVALID
import IDTRKVALID_AOD_EXCLUDED_AUXDATA
27 excludedAuxData +=
'.-'.
join([
''] + IDTRKVALID_AOD_EXCLUDED_AUXDATA)
29 if not flags.Tracking.writeExtendedSi_PRDInfo:
30 excludedAuxData +=
'.-msosLink'
34 "InDet::SCT_ClusterContainer#ITkStripClusters",
35 "InDet::PixelClusterContainer#ITkPixelClusters",
36 "InDet::PixelGangedClusterAmbiguities#ITkPixelClusterAmbiguitiesMap",
38 if flags.Tracking.doPixelClusterSplitting:
40 "InDet::PixelGangedClusterAmbiguities#ITkSplitClusterAmbiguityMap"]
42 from InDetConfig.ITkTrackRecoConfig
import ITkClusterSplitProbabilityContainerName
43 toESD += [
"Trk::ClusterSplitProbabilityContainer#" +
47 if flags.Tracking.doTruth:
49 "TrackTruthCollection#CombinedITkTracksTrackTruthCollection",
50 "DetailedTrackTruthCollection#CombinedITkTracksDetailedTrackTruth"]
52 if flags.Tracking.doStoreSiSPSeededTracks:
54 listOfExtensionsRequesting = [
55 e
for e
in extensions_list
56 if (e ==
'' or flags.Tracking.__getattr__(f
"ITk{e}Pass").storeSiSPSeededTracks) ]
58 for extension
in listOfExtensionsRequesting:
60 f
"xAOD::TrackParticleContainer#SiSPSeededTracks{extension}TrackParticles",
61 f
"xAOD::TrackParticleAuxContainer#SiSPSeededTracks{extension}TrackParticlesAux.{excludedAuxData}"]
63 if flags.Tracking.doStoreTrackSeeds:
64 listOfExtensionsRequesting = [
65 e
for e
in extensions_list
66 if (e ==
'' or flags.Tracking.__getattr__(f
"ITk{e}Pass").storeTrackSeeds) ]
68 for extension
in listOfExtensionsRequesting:
69 toESD += [
"TrackCollection#SiSPSeedSegments"+extension]
71 toESD += [
"TrackCollection#CombinedITkTracks"]
75 "xAOD::TrackParticleContainer#InDetTrackParticles",
76 f
"xAOD::TrackParticleAuxContainer#InDetTrackParticlesAux.{excludedAuxData}"
79 if flags.Tracking.writeExtendedSi_PRDInfo:
81 "xAOD::TrackMeasurementValidationContainer#ITkPixelClusters",
82 "xAOD::TrackMeasurementValidationAuxContainer#ITkPixelClustersAux.",
83 "xAOD::TrackMeasurementValidationContainer#ITkStripClusters",
84 "xAOD::TrackMeasurementValidationAuxContainer#ITkStripClustersAux.",
85 "xAOD::TrackStateValidationContainer#ITkPixelMSOSs",
86 "xAOD::TrackStateValidationAuxContainer#ITkPixelMSOSsAux.",
87 "xAOD::TrackStateValidationContainer#ITkStripMSOSs",
88 "xAOD::TrackStateValidationAuxContainer#ITkStripMSOSsAux."
91 if flags.Tracking.doStoreSiSPSeededTracks:
93 "xAOD::TrackStateValidationContainer#SiSP_ITkPixel_MSOSs",
94 "xAOD::TrackStateValidationAuxContainer#SiSP_ITkPixel_MSOSsAux.",
95 "xAOD::TrackStateValidationContainer#SiSP_ITkStrip_MSOSs",
96 "xAOD::TrackStateValidationAuxContainer#SiSP_ITkStrip_MSOSsAux."
99 if (flags.Tracking.doLargeD0
and
100 flags.Tracking.storeSeparateLargeD0Container):
102 "xAOD::TrackParticleContainer#InDetLargeD0TrackParticles",
103 f
"xAOD::TrackParticleAuxContainer#InDetLargeD0TrackParticlesAux.{excludedAuxData}"
108 result.merge(
addToESD(flags, toAOD+toESD))
109 result.merge(
addToAOD(flags, toAOD))