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
139 trackparticles_shortlist += ['-TTVA_AMVFVertices',
140 '-TTVA_AMVFWeights',
141 '-TTVA_AMVFVertices_forReco',
142 '-TTVA_AMVFWeights_forReco']
143
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
162 from InDetConfig.InDetTrackOutputConfig import FTAG_AUXDATA
163 trackparticles_variables += '.-'.join([''] + FTAG_AUXDATA)
164
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