3 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory
import CompFactory
5 from AthenaConfiguration.Enums
import Format
10 MdtCache =
"MdtPrdCache"
11 CscCache =
"CscPrdCache"
12 CscStripCache =
"CscStripPrdCache"
13 RpcCache =
"RpcPrdCache"
14 TgcCache =
"TgcPrdCache"
15 sTgcCache =
"sTgcPrdCache"
16 MmCache =
"MmPrdCache"
17 RpcCoinCache =
"RpcCoinCache"
18 TgcCoinCache =
"TgcCoinCache"
30 MuonPRDCacheCreator=CompFactory.MuonPRDCacheCreator
31 cacheCreator =
MuonPRDCacheCreator(CscStripCacheKey = (MuonPrdCacheNames.CscStripCache
if flags.Detector.GeometryCSC
else ""),
32 MdtCacheKey = MuonPrdCacheNames.MdtCache,
33 CscCacheKey = (MuonPrdCacheNames.CscCache
if flags.Detector.GeometryCSC
else ""),
34 RpcCacheKey = MuonPrdCacheNames.RpcCache,
35 TgcCacheStr = MuonPrdCacheNames.TgcCache,
36 sTgcCacheKey = (MuonPrdCacheNames.sTgcCache
if flags.Detector.GeometrysTGC
else ""),
37 MmCacheKey = (MuonPrdCacheNames.MmCache
if flags.Detector.GeometryMM
else ""),
38 TgcCoinCacheStr = MuonPrdCacheNames.TgcCoinCache,
39 RpcCoinCacheKey = MuonPrdCacheNames.RpcCoinCache,
42 acc.addEventAlgo( cacheCreator, primary=
True )
48 from MuonConfig.MuonGeometryConfig
import MuonGeoModelCfg
50 if flags.Muon.usePhaseIIGeoSetup:
51 from ActsAlignmentAlgs.AlignmentAlgsConfig
import ActsGeometryContextAlgCfg
54 kwargs.setdefault(
"DoSeededDecoding", flags.Trigger.doHLT )
55 the_alg = CompFactory.MuonRdoToPrepDataAlg(name, **kwargs)
56 result.addEventAlgo(the_alg, primary =
True)
65 if flags.Input.isMC
and flags.Muon.usePhaseIIGeoSetup
and \
66 len([x
for x
in flags.Input.TypedCollections
if x.find(
"RpcPadContainer#") != -1]):
68 from MuonConfig.MuonByteStreamCnvTestConfig
import RpcRdoToRpcDigitCfg, NrpcDigitToNrpcRDOCfg
71 if RDOContainer: cnv_args.setdefault(
"RpcRdoContainer", RDOContainer)
74 name=
"RpcRdoDigitPatchAlg" + suffix,
75 RpcDigitContainer=
"CnvRpcDigits",
79 name=
"RpcBackConverterAlg" + suffix,
80 RpcDigitContainer=
"CnvRpcDigits",
81 NrpcRdoKey=
"CnvRpcRDOs"))
82 kwargs.setdefault(
"RpcRdoContainer",
"CnvRpcRDOs")
86 kwargs.setdefault(
"RpcRdoContainer", RDOContainer)
88 if flags.Input.isMC
and flags.Muon.usePhaseIIGeoSetup:
89 from MuonConfig.MuonCablingConfig
import NRPCCablingConfigCfg
91 from AthenaConfiguration.Enums
import LHCPeriod
92 kwargs.setdefault(
"decode2DStrips", flags.GeoModel.Run >= LHCPeriod.Run4)
93 if not kwargs[
"decode2DStrips"]:
94 kwargs.setdefault(
"OutputContainer",
"xRpcMeasurements")
95 the_tool = CompFactory.MuonR4.RpcRdoToRpcPrepDataTool(name=
"RpcPrepDataProviderTool", **kwargs)
96 result.setPrivateTools(the_tool)
99 from MuonConfig.MuonCablingConfig
import RPCLegacyCablingConfigCfg
102 if not flags.Input.isMC:
103 kwargs[
"reduceCablingOverlap"] =
True
104 kwargs[
"produceRpcCoinDatafromTriggerWords"] =
True
105 kwargs[
"overlap_timeTolerance"] = 1000
106 kwargs[
"solvePhiAmbiguities"] =
True
107 kwargs[
"etaphi_coincidenceTime"] = 1000
108 if not flags.Trigger.doHLT:
109 kwargs[
"RpcPrdContainerCacheKey"] =
""
110 kwargs[
"RpcCoinDataContainerCacheKey"] =
""
112 from MuonConfig.MuonCondAlgConfig
import RpcCondDbAlgCfg
115 kwargs[
"RPCInfoFromDb"] =
False
117 if not flags.Muon.enableNRPC:
118 kwargs[
"NrpcInputCollection"] =
""
120 kwargs[
"xAODKey"] =
"xRpcMeasurements" if flags.Muon.writexAODPRD
or \
121 flags.Muon.usePhaseIIGeoSetup
else ""
128 if not flags.Input.isMC
and (flags.Muon.MuonTrigger
or flags.Input.RunNumbers[0] >= 454434):
129 kwargs[
"timeShift"] = 37.5
131 if flags.Muon.MuonTrigger:
132 kwargs[
"RdoDecoderTool"] = CompFactory.Muon.RpcRDO_Decoder(
"RpcRDO_Decoder", BCZERO=flags.Trigger.L1MuonSim.RPCNBCZ)
134 kwargs.setdefault(
"isMC", flags.Input.isMC)
135 the_tool = CompFactory.Muon.RpcRdoToPrepDataToolMT(name=
"RpcPrepDataProviderTool",**kwargs)
136 result.setPrivateTools(the_tool)
140 def RpcRDODecodeCfg(flags, name="RpcRdoToRpcPrepData", RDOContainer = None, **kwargs):
143 suffix = name[name.find(
"_") :]
if name.find(
"_") != -1
else ""
146 kwargs.setdefault(
"DecodingTool", acc.popToolsAndMerge(
RpcRdoToPrepDataToolCfg(flags, suffix=suffix, RDOContainer=RDOContainer)))
148 from RegionSelector.RegSelToolConfig
import regSelTool_RPC_Cfg
150 kwargs.setdefault(
"useROBs",
False)
156 if flags.Muon.usePhaseIIGeoSetup
and flags.Input.isMC:
157 from xAODMuonTrkPrepDataCnv.MuonPrepDataCnvCfg
import xRpcToRpcPrepDataCnvAlgCfg
159 from AthenaConfiguration.Enums
import LHCPeriod
160 if flags.GeoModel.Run >= LHCPeriod.Run4:
161 from xAODMuonViewAlgs.ViewAlgsConfig
import RpcMeasViewAlgCfg
166 def TgcRDODecodeCfg(flags, name="TgcRdoToTgcPrepData", RDOContainer = None, **kwargs):
170 from MuonConfig.MuonCablingConfig
import TGCCablingConfigCfg
175 if not flags.Trigger.doHLT:
176 tool_args.setdefault(
"PrdCacheString",
"")
177 tool_args.setdefault(
"CoinCacheString",
"")
178 tool_args.setdefault(
"xAODKey",
"xTgcStrips" if flags.Muon.writexAODPRD
or flags.Muon.usePhaseIIGeoSetup
else "")
180 if RDOContainer: tool_args.setdefault(
"RDOContainer", RDOContainer)
181 kwargs.setdefault(
"DecodingTool", CompFactory.Muon.TgcRdoToPrepDataToolMT(name=
"TgcPrepDataProviderTool", **tool_args))
184 from RegionSelector.RegSelToolConfig
import regSelTool_TGC_Cfg
186 kwargs.setdefault(
"useROBs",
False)
195 the_tool = CompFactory.Muon.TgcPrepDataReplicationToolAllBCto3BC(name, **kwargs)
196 acc.setPrivateTools(the_tool)
202 acc.addEventAlgo(CompFactory.Muon.TgcPrepDataReplicationAlg(name, **kwargs))
208 kwargs.setdefault(
"PrdCacheKey" , MuonPrdCacheNames.sTgcCache
if flags.Muon.MuonTrigger
else "")
209 kwargs.setdefault(
"UseR4DetMgr", flags.Muon.usePhaseIIGeoSetup)
210 if flags.Muon.writexAODPRD
or flags.Muon.usePhaseIIGeoSetup:
211 kwargs.setdefault(
"xAODStripKey",
"xAODsTgcStrips")
212 kwargs.setdefault(
"xAODWireKey",
"xAODsTgcWires")
213 kwargs.setdefault(
"xAODPadKey",
"xAODsTgcPads")
215 from MuonConfig.MuonRecToolsConfig
import SimpleSTgcClusterBuilderToolCfg
217 from MuonConfig.MuonCalibrationConfig
import NSWCalibToolCfg
218 kwargs.setdefault(
"NSWCalibTool", result.popToolsAndMerge(
NSWCalibToolCfg(flags)))
219 the_tool = CompFactory.Muon.sTgcRdoToPrepDataToolMT(name, **kwargs)
220 result.setPrivateTools(the_tool)
229 from RegionSelector.RegSelToolConfig
import regSelTool_STGC_Cfg
231 kwargs.setdefault(
"useROBs",
False)
236 if flags.Muon.writexAODPRD
or flags.Muon.usePhaseIIGeoSetup:
237 suffix = name[name.find(
"_") :]
if name.find(
"_") != -1
else ""
238 from xAODMuonViewAlgs.ViewAlgsConfig
import sTgcMeasViewAlgCfg
246 kwargs.setdefault(
"PrdCacheKey" , MuonPrdCacheNames.MmCache
if flags.Trigger.doHLT
else "")
248 from MuonConfig.MuonRecToolsConfig
import SimpleMMClusterBuilderToolCfg
250 from MuonConfig.MuonCalibrationConfig
import NSWCalibToolCfg
251 kwargs.setdefault(
"NSWCalibTool", result.popToolsAndMerge(
NSWCalibToolCfg(flags)))
252 kwargs[
"xAODKey"] =
"xAODMMClusters" if flags.Muon.writexAODPRD
or flags.Muon.usePhaseIIGeoSetup
else ""
254 the_tool = CompFactory.Muon.MmRdoToPrepDataToolMT(name, **kwargs)
255 result.setPrivateTools(the_tool)
264 from RegionSelector.RegSelToolConfig
import regSelTool_MM_Cfg
266 kwargs.setdefault(
"useROBs",
False)
272 def MdtRDODecodeCfg(flags, name="MdtRdoToMdtPrepData", RDOContainer = None, **kwargs):
274 from MuonConfig.MuonCalibrationConfig
import MdtCalibrationToolCfg
279 tool_kwargs.setdefault(
'UseTwin',
True)
280 if flags.IOVDb.DatabaseInstance ==
'COMP200':
281 tool_kwargs[
'UseTwin'] =
False
284 from MuonConfig.MuonCablingConfig
import MDTCablingConfigCfg, MdtTwinTubeMapCondAlgCfg
286 if tool_kwargs[
'UseTwin']:
289 writexAOD = flags.Muon.writexAODPRD
or flags.Muon.usePhaseIIGeoSetup
290 tool_kwargs[
"xAODKey"] =
"xMdtDriftCircles" if writexAOD
else ""
291 tool_kwargs[
"xAODTwinKey"] =
"xMdtTwinDriftCircles" if writexAOD
else ""
294 tool_kwargs[
"UseR4DetMgr"] = flags.Muon.usePhaseIIGeoSetup
295 tool_kwargs[
"CalibrationTool"] = acc.popToolsAndMerge(
MdtCalibrationToolCfg(flags, TimeWindowSetting = 2,
296 DoPropagationCorrection =
False))
297 if RDOContainer: tool_kwargs[
"RDOContainer"] = RDOContainer
299 kwargs.setdefault(
"DecodingTool", CompFactory.Muon.MdtRdoToPrepDataToolMT(name=
"MdtPrepDataProviderTool", **tool_kwargs))
302 from RegionSelector.RegSelToolConfig
import regSelTool_MDT_Cfg
308 suffix = name[name.find(
"_") :]
if name.find(
"_") != -1
else ""
309 from xAODMuonViewAlgs.ViewAlgsConfig
import MdtMeasViewAlgCfg
314 def CscRDODecodeCfg(flags, name="CscRdoToCscPrepData", RDOContainer = None, **kwargs):
318 from MuonConfig.MuonCablingConfig
import CSCCablingConfigCfg
321 from MuonConfig.MuonCondAlgConfig
import CscCondDbAlgCfg
326 kwargs.setdefault(
"useROBs",
False)
327 kwargs.setdefault(
"DecodingTool", CompFactory.Muon.CscRdoToCscPrepDataToolMT(name=
"CscPrepDataProviderTool"))
329 kwargs[
"DecodingTool"].RDOContainer = RDOContainer
331 from RegionSelector.RegSelToolConfig
import regSelTool_CSC_Cfg
341 from MuonConfig.MuonGeometryConfig
import MuonIdHelperSvcCfg
342 from MuonConfig.MuonSegmentFindingConfig
import CalibCscStripFitterCfg, QratCscClusterFitterCfg, CscAlignmentTool
343 from MuonConfig.MuonCalibrationConfig
import CscCalibToolCfg
350 SimpleCscClusterFitter = CompFactory.SimpleCscClusterFitter(CscAlignmentTool =
CscAlignmentTool(flags) )
351 CscSplitClusterFitter = CompFactory.CscSplitClusterFitter( precision_fitter = QratCscClusterFitter,
352 default_fitter = SimpleCscClusterFitter )
354 CscThresholdClusterBuilderTool=CompFactory.CscThresholdClusterBuilderTool
356 MuonIdHelperSvc = MuonIdHelperSvc,
357 strip_fitter = CalibCscStripFitter,
358 precision_fitter = QratCscClusterFitter,
359 default_fitter = SimpleCscClusterFitter,
360 split_fitter = CscSplitClusterFitter,
361 cscCalibTool = CscCalibTool)
364 CscThresholdClusterBuilder=CompFactory.CscThresholdClusterBuilder
366 cluster_builder = CscClusterBuilderTool,
367 MuonIdHelperSvc = MuonIdHelperSvc
369 acc.addEventAlgo(CscClusterBuilder)
375 from MuonConfig.MuonGeometryConfig
import MuonIdHelperSvcCfg
379 if not flags.Detector.GeometryMDT: kwargs.setdefault(
"MdtPrdKey",
"")
380 if not flags.Detector.GeometryRPC: kwargs.setdefault(
"RpcPrdKey",
"")
381 if not flags.Detector.GeometryTGC: kwargs.setdefault(
"TgcPrdKey",
"")
382 if not flags.Detector.GeometryCSC: kwargs.setdefault(
"CscPrdKey",
"")
384 if not flags.Detector.GeometrysTGC: kwargs.setdefault(
"sTgcPrdKey",
"")
385 if not flags.Detector.GeometryMM: kwargs.setdefault(
"MmPrdKey",
"")
387 kwargs.setdefault(
"TgcPrdKey",
'TGC_MeasurementsAllBCs' if not flags.Muon.useTGCPriorNextBC
else 'TGC_Measurements')
388 result.addEventAlgo(CompFactory.MuonPRD_MultiTruthMaker(name, **kwargs), primary =
True)
396 if flags.Detector.GeometryMDT:
399 if flags.Detector.GeometryRPC:
402 if flags.Detector.GeometryTGC:
405 if flags.Detector.GeometrysTGC:
408 if flags.Detector.GeometryMM:
412 if flags.Detector.GeometryCSC:
416 if flags.Input.isMC
and not flags.Muon.usePhaseIIGeoSetup:
423 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
424 from AthenaConfiguration.TestDefaults
import defaultConditionsTags, defaultGeometryTags, defaultTestFiles
426 flags.Input.Files = defaultTestFiles.RAW_RUN2
427 flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN2
428 flags.IOVDb.GlobalTag = defaultConditionsTags.RUN2_DATA
433 from AthenaCommon.Logging
import log
436 log.info(
'About to setup Raw data decoding')
438 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
444 from MuonConfig.MuonBytestreamDecodeConfig
import MuonCacheCfg
447 if flags.Input.Format
is Format.BS:
448 from MuonConfig.MuonBytestreamDecodeConfig
import MuonByteStreamDecodersCfg
453 log.info(
'Print Config')
454 cfg.printConfig(withDetails=
True)
457 pklName =
'MuonRdoDecode_Cache.pkl'
459 pklName =
'MuonRdoDecode.pkl'
462 log.info(
'Save Config')
463 with open(pklName,
'wb')
as f:
471 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
473 flags.Input.Files = [
"/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/TriggerTest/valid1.110401.PowhegPythia_P2012_ttbar_nonallhad.recon.RDO.e3099_s2578_r7572_tid07644622_00/RDO.07644622._000001.pool.root.1"]
478 from AthenaCommon.Logging
import log
481 log.info(
'About to setup Rpc RDO data decoding')
486 from AthenaPoolCnvSvc.PoolReadConfig
import PoolReadCfg
492 log.info(
'Print Config')
493 cfg.printConfig(withDetails=
True)
496 log.info(
'Save Config')
497 with open(
'MuonRdoDecode.pkl',
'wb')
as f:
502 if __name__==
"__main__":