6 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
10 @file TileD3PDConfig.py
11 @brief Python configuration of TileD3PD for the Run III
15 def _args(level, name, kwin, **kw):
18 for (k, v)
in kwin.items():
19 if k.startswith(name +
'_'):
20 kw[k[len(name)+1:]] = v
24 def TileD3PDCfg(flags, outputFile=None, saveCells=True, saveMBTS=True,
25 saveE4pr=False, saveClusters=False, saveMuId=False,
26 saveMuonFit=False, savePosition=True, saveEventInfo=False,
28 ''' Function to configure Tile D3PD.'''
32 from TileGeoModel.TileGMConfig
import TileGMCfg
35 from LArGeoAlgsNV.LArGMConfig
import LArGMCfg
38 from TileConditions.TileCablingSvcConfig
import TileCablingSvcCfg
41 from TileConditions.TileBadChannelsConfig
import TileBadChannelsCondAlgCfg
45 runNumber = flags.Input.RunNumbers[0]
46 outputFile = f
'tile_{runNumber}.aan.root'
48 from D3PDMakerCoreComps.MakerAlgConfig
import MakerAlgConfig
49 alg =
MakerAlgConfig(flags, acc,
'caloD3PD', outputFile, ExistDataHeader=
False)
52 from EventCommonD3PDMaker.EventInfoD3PDObject
import EventInfoD3PDObject
55 from CaloSysD3PDMaker.TileDetailsD3PDObject
import TileDetailsD3PDObject
56 from CaloSysD3PDMaker.CaloInfoD3PDObject
import CaloInfoD3PDObject
57 from CaloD3PDMaker.MBTSD3PDObject
import MBTSD3PDObject
59 isBiGainRun = flags.Tile.RunType.isBiGain()
64 prefix=
'tile_', Kinematics_WriteEtaPhi=
True,
65 TileDetails_SavePositionInfo=savePosition))
71 alg +=
MBTSD3PDObject(**
_args(1,
'MBTS', kwargs, prefix=
'mbtsLG_', sgkey=
'MBTSContainerLG'))
75 sgkey=
'E4prContainerHG', MBTS_SaveEtaPhiInfo=
False))
78 sgkey=
'E4prContainerLG', MBTS_SaveEtaPhiInfo=
False))
82 prefix=
'tile_', Kinematics_WriteEtaPhi=
True,
83 TileDetails_SavePositionInfo=savePosition))
92 sgkey=
'E4prContainer', MBTS_SaveEtaPhiInfo=
False))
95 from CaloD3PDMaker.xAODClusterD3PDObject
import xAODClusterD3PDObject
99 from CaloSysD3PDMaker.TileMuD3PDObject
import TileMuD3PDObject
100 alg += TileMuD3PDObject(**
_args(0,
'TileMus', kwargs, sgkey=
'TileMuObj', prefix=
'tilemuid_'))
103 from CaloSysD3PDMaker.TileCosmicMuonD3PDObject
import TileCosmicMuonD3PDObject
104 alg += TileCosmicMuonD3PDObject(**
_args(2,
'TileCosMusHT', kwargs, sgkey=
'TileCosmicMuonHT', prefix=
'TileCosmicsHT_'))
106 acc.addEventAlgo(alg.alg)
110 if __name__ ==
'__main__':
113 from D3PDMakerConfig.D3PDMakerFlags
import configFlags
as flags
114 parser = flags.getArgumentParser()
115 parser.add_argument(
'--cells', action=
'store_true', help=
'Save Tile cells in D3PD')
116 parser.add_argument(
'--mbts', action=
'store_true', help=
'Save Tile MBTS in D3PD')
117 parser.add_argument(
'--e4pr', action=
'store_true', help=
'Save Tile E4pr in D3PD')
118 parser.add_argument(
'--muid', action=
'store_true', help=
'Save Tile MuID in D3PD')
119 parser.add_argument(
'--muonfit', action=
'store_true', help=
'Save Tile Muon Fitter in D3PD')
120 parser.add_argument(
'--clusters', action=
'store_true', help=
'Save Tile clusters in D3PD')
121 parser.add_argument(
'--postExec', help=
'Code to execute after setup')
122 args, _ = parser.parse_known_args()
125 from AthenaCommon.Logging
import log
129 from AthenaConfiguration.TestDefaults
import defaultTestFiles
130 flags.Input.Files = defaultTestFiles.ESD
131 flags.Exec.MaxEvents = 3
132 flags.fillFromArgs(parser=parser)
134 log.info(
'FINAL CONFIG FLAGS SETTINGS FOLLOW')
140 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
143 from AthenaPoolCnvSvc.PoolReadConfig
import PoolReadCfg
147 saveCells=args.cells,
151 saveMuonFitter=args.muonfit,
152 saveClusters=args.clusters,
153 saveEventInfo=
True) )
155 cfg.setAppProperty(
'CreateSvc', [cfg.getService(
'TileCablingSvc').getFullJobOptName()])
159 log.info(
'Executing postExec: %s', args.postExec)
162 cfg.printConfig(withDetails=
True, summariseProps=
True)
164 cfg.store(
open(
'TileD3PDConfig.pkl',
'wb') )
170 sys.exit(0
if sc.isSuccess()
else 1)