Skip to content
Snippets Groups Projects

Add configuration to write extended PRD for TRT

Merged Thomas Strebler requested to merge tstreble/athena:WriteExtendedPRD_TRT into 23.0
2 files
+ 126
109
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -7,6 +7,7 @@
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.Enums import BeamType
# Track collection merger
@@ -49,9 +50,11 @@ def HardScatterVertexDecoratorCfg(flags, name = "DFCommonHSDecorator", **kwargs)
def TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs):
"""Configure the TSOS decorator"""
acc = ComponentAccumulator()
# To produce SCT_DetectorElementCollection
from SCT_GeoModel.SCT_GeoModelConfig import SCT_ReadoutGeometryCfg
acc = SCT_ReadoutGeometryCfg(flags)
acc.merge(SCT_ReadoutGeometryCfg(flags))
kwargs.setdefault("DecorationPrefix", "notSet")
@@ -73,6 +76,7 @@ def TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs):
kwargs.setdefault("StorePixel", flags.Tracking.writeExtendedSi_PRDInfo)
kwargs.setdefault("StoreSCT", flags.Tracking.writeExtendedSi_PRDInfo)
kwargs.setdefault("StoreTRT", flags.Tracking.writeExtendedTRT_PRDInfo)
kwargs.setdefault("AddExtraEventInfo", flags.Beam.Type is BeamType.Cosmics)
if kwargs["StoreTRT"] and "TRT_ToT_dEdx" not in kwargs:
from InDetConfig.TRT_ElectronPidToolsConfig import TRT_dEdxToolCfg
@@ -163,7 +167,7 @@ def SiSPTSOS_CommonKernelCfg(flags, name="SiSPTSOS_CommonKernel",
listOfAugmTools = []
for extension in listOfExtensions:
SiSPTrackStateOnSurfaceDecorator = result.getPrimaryAndMerge(
SiSPTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
SiSPTrackStateOnSurfaceDecoratorCfg(
flags, name = f"SiSP{extension}TrackStateOnSurfaceDecorator",
ContainerName = f"SiSPSeededTracks{extension}TrackParticles",
@@ -183,7 +187,7 @@ def GSFTrackStateOnSurfaceDecoratorCfg(
kwargs.setdefault("PixelMsosName", "GSF_Pixel_MSOSs")
kwargs.setdefault("SctMsosName", "GSF_SCT_MSOSs")
kwargs.setdefault("TrtMsosName", "GSF_TRT_MSOSs")
kwargs.setdefault("PRDtoTrackMap", "PRDtoTrackMapGSFTracks")
kwargs.setdefault("PRDtoTrackMap", "")
return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
def GSFTSOS_CommonKernelCfg(flags, name="GSFTSOS_CommonKernel"):
Loading