81 from AthenaCommon.Logging
import logging
82 log = logging.getLogger(
'FilterHit_tf')
83 log.info(
'****************** STARTING HIT FILTERING *****************')
85 log.info(
'**** Transformation run arguments')
86 log.info(
str(runArgs))
88 log.info(
'**** Setting-up configuration flags')
89 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
94 if hasattr(runArgs,
'inputHITSFile'):
95 flags.Input.Files = runArgs.inputHITSFile
97 raise RuntimeError(
'No input HITS file defined')
100 from SimuJobTransforms.SimulationHelpers
import getDetectorsFromRunArgs
102 from AthenaConfiguration.DetectorConfigFlags
import setupDetectorFlags
103 setupDetectorFlags(flags, detectors, use_metadata=
True, toggle_geometry=
True, keep_beampipe=
True)
105 if hasattr(runArgs,
'outputHITS_FILTFile'):
106 if runArgs.outputHITS_FILTFile ==
'None':
107 flags.Output.HITSFileName =
''
110 flags.Output.HITSFileName = runArgs.outputHITS_FILTFile
112 raise RuntimeError(
'No outputHITS_FILTFile defined')
115 flags.Output.TreeAutoFlush = {
'HITS': 1}
129 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
132 from AthenaPoolCnvSvc.PoolReadConfig
import PoolReadCfg
136 from IOVDbSvc.IOVDbSvcConfig
import IOVDbSvcCfg
140 from DetDescrCnvSvc.DetDescrCnvSvcConfig
import DetDescrCnvSvcCfg
144 if flags.Detector.EnableLAr:
145 from LArHitAlgs.LArHitAlgsConfig
import LArHitFilterCfg
148 if hasattr(runArgs,
'TruthReductionScheme'):
149 if runArgs.TruthReductionScheme !=
'SingleGenParticle':
150 log.warning(f
'Unknown TruthReductionScheme ({runArgs.TruthReductionScheme}). Currently just a dummy value, but please check.')
153 from McEventCollectionFilter.McEventCollectionFilterConfig
import McEventCollectionFilterCfg
156 for entry
in flags.Input.Collections:
157 if 'AntiKt4TruthJets' == entry:
158 from McEventCollectionFilter.McEventCollectionFilterConfig
import DecoratePileupAntiKt4TruthJetsCfg
160 if 'AntiKt6TruthJets' == entry:
161 from McEventCollectionFilter.McEventCollectionFilterConfig
import DecoratePileupAntiKt6TruthJetsCfg
163 if 'TruthPileupParticles' == entry:
164 from McEventCollectionFilter.McEventCollectionFilterConfig
import DecorateTruthPileupParticlesCfg
168 if flags.Detector.EnableBCM:
169 from McEventCollectionFilter.McEventCollectionFilterConfig
import BCM_HitsTruthRelinkCfg
171 if flags.Detector.EnablePixel:
172 from McEventCollectionFilter.McEventCollectionFilterConfig
import PixelHitsTruthRelinkCfg
174 if flags.Detector.EnableSCT:
175 from McEventCollectionFilter.McEventCollectionFilterConfig
import SCT_HitsTruthRelinkCfg
177 if flags.Detector.EnableTRT:
178 from McEventCollectionFilter.McEventCollectionFilterConfig
import TRT_HitsTruthRelinkCfg
181 if flags.Detector.EnableITkPixel:
182 from McEventCollectionFilter.McEventCollectionFilterConfig
import ITkPixelHitsTruthRelinkCfg
184 if flags.Detector.EnableITkPixel:
185 from McEventCollectionFilter.McEventCollectionFilterConfig
import ITkPixelHitsTruthRelinkCfg
187 if flags.Detector.EnablePLR:
188 from McEventCollectionFilter.McEventCollectionFilterConfig
import PLR_HitsTruthRelinkCfg
191 if flags.Detector.EnableHGTD:
192 from McEventCollectionFilter.McEventCollectionFilterConfig
import HGTD_HitsTruthRelinkCfg
195 if flags.Detector.EnableCSC:
196 from McEventCollectionFilter.McEventCollectionFilterConfig
import CSC_HitsTruthRelinkCfg
198 if flags.Detector.EnableMDT:
199 from McEventCollectionFilter.McEventCollectionFilterConfig
import MDT_HitsTruthRelinkCfg
201 if flags.Detector.EnableMM:
202 from McEventCollectionFilter.McEventCollectionFilterConfig
import MM_HitsTruthRelinkCfg
204 if flags.Detector.EnableRPC:
205 from McEventCollectionFilter.McEventCollectionFilterConfig
import RPC_HitsTruthRelinkCfg
207 if flags.Detector.EnableTGC:
208 from McEventCollectionFilter.McEventCollectionFilterConfig
import TGC_HitsTruthRelinkCfg
210 if flags.Detector.EnablesTGC:
211 from McEventCollectionFilter.McEventCollectionFilterConfig
import sTGC_HitsTruthRelinkCfg
214 from OutputStreamAthenaPool.OutputStreamConfig
import OutputStreamCfg
218 from xAODMetaDataCnv.InfileMetaDataConfig
import SetupMetaDataForStreamCfg
228 from PyUtils.AMITagHelperConfig
import AMITagCfg
235 log.info(
"Ran FilterHit_tf in " +
str(time.time()-tic) +
" seconds")
237 sys.exit(
not sc.isSuccess())