ATLAS Offline Software
Loading...
Searching...
No Matches
ActsPersistificationConfig Namespace Reference

Functions

ComponentAccumulator PersistifyClusters (flags, *, list[str] pixelClusterCollections=None, list[str] stripClusterCollections=None, list[str] hgtdClusterCollections=None)
ComponentAccumulator PersistifySpacePoints (flags, *, list[str] pixelSpacePointCollections=None, list[str] stripSpacePointCollections=None)
ComponentAccumulator PersistifyTracks (flags, *, list[str] extensions=None)
ComponentAccumulator PersistifyTrackParticles (flags, *, list[str] trackParticleCollections=None)

Function Documentation

◆ PersistifyClusters()

ComponentAccumulator PersistifyClusters ( flags,
* ,
list[str] pixelClusterCollections = None,
list[str] stripClusterCollections = None,
list[str] hgtdClusterCollections = None )

Definition at line 5 of file ActsPersistificationConfig.py.

9 hgtdClusterCollections: list[str] = None) -> ComponentAccumulator:
10
11 acc = ComponentAccumulator()
12 if not flags.Acts.EDM.PersistifyClusters:
13 return acc
14
15 toESD = []
16 toAOD = []
17 if pixelClusterCollections is not None:
18 pixel_cluster_shortlist = ['-pixelClusterLink',
19 '-validationMeasurementLink']
20 pixel_cluster_variables = '.'.join(pixel_cluster_shortlist)
21 for pixelClusterCollection in pixelClusterCollections:
22 toAOD += [f'xAOD::PixelClusterContainer#{pixelClusterCollection}',
23 f'xAOD::PixelClusterAuxContainer#{pixelClusterCollection}Aux.{pixel_cluster_variables}']
24
25 if stripClusterCollections is not None:
26 strip_cluster_shortlist = ['-sctClusterLink',
27 '-validationMeasurementLink']
28 strip_cluster_variables = '.'.join(strip_cluster_shortlist)
29 for stripClusterCollection in stripClusterCollections:
30 toAOD += [f"xAOD::StripClusterContainer#{stripClusterCollection}",
31 f"xAOD::StripClusterAuxContainer#{stripClusterCollection}Aux.{strip_cluster_variables}"]
32
33 if hgtdClusterCollections is not None:
34 hgtd_cluster_shortlist = ['-hgtdClusterLink']
35 hgtd_cluster_variables = '.'.join(hgtd_cluster_shortlist)
36 for hgtdClusterCollection in hgtdClusterCollections:
37 toAOD += [f"xAOD::HGTDClusterContainer#{hgtdClusterCollection}",
38 f"xAOD::HGTDClusterAuxContainer#{hgtdClusterCollection}Aux.{hgtd_cluster_variables}"]
39
40 from OutputStreamAthenaPool.OutputStreamConfig import addToESD, addToAOD
41 acc.merge(addToESD(flags, toESD + toAOD))
42 acc.merge(addToAOD(flags, toAOD))
43 return acc
44
45

◆ PersistifySpacePoints()

ComponentAccumulator PersistifySpacePoints ( flags,
* ,
list[str] pixelSpacePointCollections = None,
list[str] stripSpacePointCollections = None )

Definition at line 46 of file ActsPersistificationConfig.py.

49 stripSpacePointCollections: list[str] = None) -> ComponentAccumulator:
50
51 acc = ComponentAccumulator()
52 if not flags.Acts.EDM.PersistifySpacePoints:
53 return acc
54
55 toESD = []
56 toAOD = []
57 aux_container_type = "xAOD::SpacePointAuxContainer"
58 if flags.Acts.EDM.SlimContent:
59 aux_container_type = "xAOD::AuxContainerBase!"
60
61 if pixelSpacePointCollections is not None:
62 pixel_spacepoint_shortlist = ['-measurements',
63 '-pixelSpacePointLink']
64 if flags.Acts.EDM.SlimContent:
65 pixel_spacepoint_shortlist = ['globalPosition']
66 pixel_spacepoint_variables = '.'.join(pixel_spacepoint_shortlist)
67 for pixelSpacePointCollection in pixelSpacePointCollections:
68 toAOD += [f'xAOD::SpacePointContainer#{pixelSpacePointCollection}',
69 f"{aux_container_type}#{pixelSpacePointCollection}Aux.{pixel_spacepoint_variables}"]
70
71 if stripSpacePointCollections is not None:
72 strip_spacepoint_shortlist = ['topHalfStripLength',
73 'bottomHalfStripLength',
74 'topStripDirection',
75 'bottomStripDirection',
76 'stripCenterDistance',
77 'topStripCenter',
78 'measurementLink']
79 if flags.Acts.EDM.SlimContent:
80 strip_spacepoint_shortlist = ['globalPosition']
81 strip_spacepoint_variables = '.'.join(strip_spacepoint_shortlist)
82 for stripSpacePointCollection in stripSpacePointCollections:
83 toAOD += [f'xAOD::SpacePointContainer#{stripSpacePointCollection}',
84 f"{aux_container_type}#{stripSpacePointCollection}Aux.{strip_spacepoint_variables}"]
85
86 from OutputStreamAthenaPool.OutputStreamConfig import addToESD, addToAOD
87 acc.merge(addToESD(flags, toESD + toAOD))
88 acc.merge(addToAOD(flags, toAOD))
89 return acc
90
91

◆ PersistifyTrackParticles()

ComponentAccumulator PersistifyTrackParticles ( flags,
* ,
list[str] trackParticleCollections = None )

Definition at line 123 of file ActsPersistificationConfig.py.

125 trackParticleCollections: list[str] = None) -> ComponentAccumulator:
126 toESD = []
127 toAOD = []
128 if trackParticleCollections is not None:
129 trackparticles_shortlist = ['-clusterAssociation',
130 '-AssoClustersUFO',
131 '-trackParameterCovarianceMatrices',
132 '-caloExtensionLink',
133 '-parameterX', '-parameterY', '-parameterZ',
134 '-parameterPX', '-parameterPY', '-parameterPZ',
135 '-parameterPosition',
136 '-truthHitEfficiency', '-truthHitPurity']
137
138 # exclude TTVA decorations
139 trackparticles_shortlist += ['-TTVA_AMVFVertices',
140 '-TTVA_AMVFWeights',
141 '-TTVA_AMVFVertices_forReco',
142 '-TTVA_AMVFWeights_forReco']
143 # acts track link
144 if not flags.Acts.EDM.PersistifyTracks:
145 trackparticles_shortlist += ['-actsTrack',
146 '-hgtdTrackLink']
147
148 if not flags.Tracking.writeExtendedHGTDInfo:
149 trackparticles_shortlist += ['-HGTD_cluster_merged',
150 '-HGTD_cluster_raw_time',
151 '-HGTD_cluster_shadowed',
152 '-HGTD_cluster_time',
153 '-HGTD_cluster_truth_class',
154 '-HGTD_extrap_x',
155 '-HGTD_extrap_y',
156 '-HGTD_primary_expected',
157 '-HGTD_summaryinfo',
158 '-HGTD_times_of_compatible_hits']
159
160 trackparticles_variables = ".".join(trackparticles_shortlist)
161 # remove track decorations used internally by FTAG software
162 from InDetConfig.InDetTrackOutputConfig import FTAG_AUXDATA
163 trackparticles_variables += '.-'.join([''] + FTAG_AUXDATA)
164 # exclude IDTIDE decorations
165 from DerivationFrameworkInDet.IDTIDE import IDTIDE_AOD_EXCLUDED_AUXDATA
166 trackparticles_variables += '.-'.join([''] + IDTIDE_AOD_EXCLUDED_AUXDATA)
167 from DerivationFrameworkInDet.IDTRKVALID import IDTRKVALID_AOD_EXCLUDED_AUXDATA
168 trackparticles_variables += '.-'.join([''] + IDTRKVALID_AOD_EXCLUDED_AUXDATA)
169
170 for trackParticleCollection in trackParticleCollections:
171 toAOD += [f"xAOD::TrackParticleContainer#{trackParticleCollection}",
172 f"xAOD::TrackParticleAuxContainer#{trackParticleCollection}Aux." + trackparticles_variables]
173
174 acc = ComponentAccumulator()
175
176 from OutputStreamAthenaPool.OutputStreamConfig import addToESD, addToAOD
177 acc.merge(addToESD(flags, toESD + toAOD))
178 acc.merge(addToAOD(flags, toAOD))
179 return acc

◆ PersistifyTracks()

ComponentAccumulator PersistifyTracks ( flags,
* ,
list[str] extensions = None )

Definition at line 92 of file ActsPersistificationConfig.py.

94 extensions: list[str] = None) -> ComponentAccumulator:
95 toESD = []
96 toAOD = []
97 if extensions is not None:
98 for prefix in extensions:
99 toAOD += [f"xAOD::TrackSummaryContainer#{prefix}TrackSummary",
100 f"xAOD::TrackSummaryAuxContainer#{prefix}TrackSummaryAux.",
101 f"xAOD::TrackStateContainer#{prefix}TrackStates",
102 f"xAOD::TrackStateAuxContainer#{prefix}TrackStatesAux.-uncalibratedMeasurement",
103 f"xAOD::TrackParametersContainer#{prefix}TrackParameters",
104 f"xAOD::TrackParametersAuxContainer#{prefix}TrackParametersAux.",
105 f"xAOD::TrackJacobianContainer#{prefix}TrackJacobians",
106 f"xAOD::TrackJacobianAuxContainer#{prefix}TrackJacobiansAux.",
107 f"xAOD::TrackMeasurementContainer#{prefix}TrackMeasurements",
108 f"xAOD::TrackMeasurementAuxContainer#{prefix}TrackMeasurementsAux.",
109 f"xAOD::TrackSurfaceContainer#{prefix}TrackStateSurfaces",
110 f"xAOD::TrackSurfaceAuxContainer#{prefix}TrackStateSurfacesAux.",
111 f"xAOD::TrackSurfaceContainer#{prefix}TrackSurfaces",
112 f"xAOD::TrackSurfaceAuxContainer#{prefix}TrackSurfacesAux."]
113
114 acc = ComponentAccumulator()
115
116 from OutputStreamAthenaPool.OutputStreamConfig import addToESD, addToAOD
117 acc.merge(addToESD(flags, toESD + toAOD))
118 if flags.Acts.EDM.PersistifyTracks:
119 acc.merge(addToAOD(flags, toAOD))
120 return acc
121
122