ATLAS Offline Software
LArDigits2NtupleDumper.py
Go to the documentation of this file.
1 #!/usr/bin/env python
2 #
3 # Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
4 
5 from AthenaConfiguration.ComponentFactory import CompFactory
6 
7 if __name__=='__main__':
8 
9  import os,sys
10  import argparse
11  from AthenaCommon import Logging
12  log = Logging.logging.getLogger( 'LArDigits2Ntuple' )
13 
14  parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
15 
16  parser.add_argument('-i','--indir', dest='indir', default="/eos/atlas/atlastier0/rucio/data_test/calibration_pulseall/00414414/data_test.00414414.calibration_pulseall.daq.RAW/", help='input files dir', type=str)
17  parser.add_argument('-p','--inprefix', dest='inpref', default="data_test", help='Input filenames prefix', type=str)
18  parser.add_argument('-y','--inppatt', dest='inppatt', default="lb3512", help='Input filenames pattern', type=str)
19  parser.add_argument('-f','--infile', dest='infile', default="", help='Input filename (if given indir and inprefix are ignored', type=str)
20  parser.add_argument('-r','--run', dest='run', default=0, help='Run number (if not given trying to judge from input file name)', type=int)
21  parser.add_argument('-m','--maxev', dest='maxev', default=-1, help='Max number of events to dump', type=int)
22  parser.add_argument('-x','--outlevel', dest='olevel', default=3, help='OuputLevel for dumping algo', type=int)
23  parser.add_argument('-o','--outfile', dest='outfile', default="Digits.root", help='Output root filename', type=str)
24  parser.add_argument('-n','--nsamp', dest='nsamp', default=0, help='Number of samples to dump', type=int)
25  parser.add_argument('-a','--addHash', dest='ahash', default=False, help='Add hash number to output ntuple', action="store_true")
26  parser.add_argument('-j','--addOffline', dest='offline', default=False, help='Add offline Id to output ntuple', action="store_true")
27  parser.add_argument('-k','--addCalib', dest='calib', default=False, help='Add calib. info to output ntuple', action="store_true")
28  parser.add_argument('-t','--addGeom', dest='geom', default=False, help='Add real geom info to output ntuple', action="store_true")
29  parser.add_argument('-u','--addBC', dest='bc', default=False, help='Add Bad. chan info to output ntuple', action="store_true")
30  parser.add_argument('--addBCID', dest='bcid', default=False, help='Add BCID info to output ntuple', action="store_true")
31  parser.add_argument('-v','--addEvTree', dest='evtree', default=False, help='Add tree with per event info to output ntuple', action="store_true")
32  parser.add_argument('--EMF', dest='emf', default=False, help='Is it for EMF', action='store_true')
33  parser.add_argument('-d','--digikey', dest='dkey', default="FREE", help='Input digits key', type=str)
34  parser.add_argument('-e','--acckey', dest='acckey', default="", help='Input accumulated calib digits key', type=str)
35  parser.add_argument('--CALIB', dest='iscalib', default=False, help='Is it from calib run ?', action='store_true')
36  parser.add_argument('--FTs', dest='ft', default=[], nargs="+", type=int, help='list of FT which will be read out (space separated).')
37  parser.add_argument('--posneg', dest='posneg', default=[], nargs="+", help='side to read out (-1 means both), can give multiple arguments (space separated). Default %(default)s.', type=int,choices=range(-1,2))
38  parser.add_argument('--barrel_ec', dest='be', default=[], nargs="+", help='subdet to read out (-1 means both), can give multiple arguments (space separated) Default %(default)s.', type=int,choices=range(-1,2))
39 
40  args = parser.parse_args()
41  if help in args and args.help is not None and args.help:
42  parser.print_help()
43  sys.exit(0)
44 
45  for _, value in args._get_kwargs():
46  if value is not None:
47  log.debug(value)
48 
49  #Import the flag-container that is the arguemnt to the configuration methods
50  from AthenaConfiguration.AllConfigFlags import initConfigFlags
51  flags = initConfigFlags()
52 
53  # add SCDump flags, here re-used for digitsdump
54  from LArCafJobs.LArSCDumperFlags import addSCDumpFlags
55  addSCDumpFlags(flags)
56 
57 
58  if len(args.infile) > 0:
59  flags.Input.Files = [args.infile]
60  elif len(args.inppatt) > 0:
61  from LArCalibProcessing.GetInputFiles import GetInputFilesFromPattern
62  flags.Input.Files = GetInputFilesFromPattern(args.indir,args.inppatt)
63  else:
64  from LArCalibProcessing.GetInputFiles import GetInputFilesFromPrefix
65  flags.Input.Files = GetInputFilesFromPrefix(args.indir,args.inpref)
66 
67  if args.run != 0:
68  flags.Input.RunNumbers = [args.run]
69 
70  from LArCalibProcessing.LArCalibConfigFlags import addLArCalibFlags
71  addLArCalibFlags(flags)
72  if len(args.posneg) >= 0:
73  flags.LArCalib.Preselection.Side = args.posneg
74  if len(args.be) >=0:
75  flags.LArCalib.Preselection.BEC = args.be
76  if len(args.ft) > 0:
77  flags.LArCalib.Preselection.FT = args.ft
78  # first autoconfig
79  from LArConditionsCommon.LArRunFormat import getLArFormatForRun
80  try:
81  runinfo=getLArFormatForRun(flags.Input.RunNumbers[0], connstring="COOLONL_LAR/CONDBR2")
82  except Exception:
83  log.warning("Could not get run info, using defaults !")
84  if args.nsamp > 0:
85  flags.LArSCDump.nSamples=args.nsamp
86  else:
87  flags.LArSCDump.nSamples=4
88  else:
89  if runinfo is not None:
90  flags.LArSCDump.nSamples=runinfo.nSamples()
91 
92  flags.LArSCDump.digitsKey=args.dkey
93  if args.nsamp > 0 and args.nsamp < flags.LArSCDump.nSamples:
94  flags.LArSCDump.nSamples=args.nsamp
95 
96  log.info("Autoconfigured: ")
97  log.info("nSamples: %d digitsKey %s accKey %s",flags.LArSCDump.nSamples, flags.LArSCDump.digitsKey, args.acckey)
98 
99  #GEometry
100  from AthenaConfiguration.TestDefaults import defaultGeometryTags
101  flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN3
102 
103  # now construct the job
104  flags.LAr.doAlign=False
105 
106  if args.evtree: # should include trigger info
107  flags.Trigger.triggerConfig = 'DB'
108  flags.Trigger.L1.doCTP = True
109  flags.Trigger.L1.doMuon = False
110  flags.Trigger.L1.doCalo = False
111  flags.Trigger.L1.doTopo = False
112 
113  flags.Trigger.enableL1CaloLegacy = True
114  flags.Trigger.enableL1CaloPhase1 = True
115 
116  # additions for EMF
117  if args.emf:
118  flags.IOVDb.SqliteInput="/afs/cern.ch/user/p/pavol/public/EMF_otherCond.db"
119  flags.IOVDb.SqliteFolders = ("/LAR/BadChannelsOfl/BadChannels","/LAR/BadChannelsOfl/KnownBADFEBs","/LAR/BadChannelsOfl/KnownMNBFEBs","/LAR/BadChannelsOfl/MissingFEBs","/LAR/Identifier/OnOffIdMap",)
120 
121  flags.lock()
122  flags.dump()
123 
124  #Import the MainServices (boilerplate)
125  from AthenaConfiguration.MainServicesConfig import MainServicesCfg
126  from LArGeoAlgsNV.LArGMConfig import LArGMCfg
127 
128  acc = MainServicesCfg(flags)
129  acc.merge(LArGMCfg(flags))
130 
131  from LArCabling.LArCablingConfig import LArOnOffIdMappingCfg
132  acc.merge(LArOnOffIdMappingCfg(flags))
133 
134  if args.evtree: # should include trigger info
135  from LArCafJobs.LArSCDumperSkeleton import L1CaloMenuCfg
136  acc.merge(L1CaloMenuCfg(flags))
137  from TrigDecisionTool.TrigDecisionToolConfig import TrigDecisionToolCfg
138  tdt = acc.getPrimaryAndMerge(TrigDecisionToolCfg(flags))
139  else:
140  tdt = None
141 
142 
143  if args.bc:
144  from LArBadChannelTool.LArBadChannelConfig import LArBadFebCfg, LArBadChannelCfg
145  acc.merge(LArBadChannelCfg(flags))
146  acc.merge(LArBadFebCfg(flags))
147 
148  if args.geom:
149  log.warning("Adding real geometry is not working yet")
150  args.geom=False
151  # FIXME
152  #acc.addCondAlgo(CompFactory.CaloAlignCondAlg(LArAlignmentStore="",CaloCellPositionShiftFolder=""))
153  #acc.addCondAlgo(CompFactory.CaloSuperCellAlignCondAlg())
154  #AthReadAlg_ExtraInputs.append(('CaloSuperCellDetDescrManager', 'ConditionStore+CaloSuperCellDetDescrManager'))
155 
156  from LArCalibTools.LArDigits2NtupleConfig import LArDigits2NtupleCfg
157  acc.merge(LArDigits2NtupleCfg(flags, AddBadChannelInfo=args.bc, AddFEBTempInfo=False, isSC=False, isFlat=True,
158  OffId=args.offline, AddHash=args.ahash, AddCalib=args.calib, RealGeometry=args.geom, # from LArCond2NtupleBase
159  NSamples=flags.LArSCDump.nSamples, FTlist=flags.LArCalib.Preselection.FT, Slotlist=flags.LArCalib.Preselection.Slot, BElist=flags.LArCalib.Preselection.BEC, Sidelist=flags.LArCalib.Preselection.Side,
160  ContainerKey=flags.LArSCDump.digitsKey, AccContainerKey=args.acckey, isCalib=args.iscalib, # from LArDigits2Ntuple
161  FillLB=args.evtree, FillBCID=args.bcid,
162  OutputLevel=args.olevel
163  ))
164  # ROOT file writing
165  if os.path.exists(args.outfile):
166  os.remove(args.outfile)
167  acc.addService(CompFactory.NTupleSvc(Output = [ "FILE1 DATAFILE='"+args.outfile+"' OPT='NEW'" ]))
168  acc.setAppProperty("HistogramPersistency","ROOT")
169 
170  # some logging
171  log.info("Input files to be processed:")
172  for f in flags.Input.Files:
173  log.info(f)
174  log.info("Output file: ")
175  log.info(args.outfile)
176 
177 
178  # and run
179  acc.run(args.maxev)
python.LArBadChannelConfig.LArBadFebCfg
def LArBadFebCfg(configFlags, tag=None, dbname=None)
Definition: LArBadChannelConfig.py:43
LArSCDumperSkeleton.L1CaloMenuCfg
def L1CaloMenuCfg(flags)
Definition: LArSCDumperSkeleton.py:11
python.LArBadChannelConfig.LArBadChannelCfg
def LArBadChannelCfg(configFlags, tag=None, isSC=False)
Definition: LArBadChannelConfig.py:8
LArDigits2NtupleConfig.LArDigits2NtupleCfg
def LArDigits2NtupleCfg(flags, isEMF=False, isCalib=False, **kwargs)
Definition: LArDigits2NtupleConfig.py:6
LArRunFormat
LArSCDumperFlags.addSCDumpFlags
def addSCDumpFlags(flags)
Definition: LArSCDumperFlags.py:3
python.MainServicesConfig.MainServicesCfg
def MainServicesCfg(flags, LoopMgr='AthenaEventLoopMgr')
Definition: MainServicesConfig.py:310
plotBeamSpotVxVal.range
range
Definition: plotBeamSpotVxVal.py:194
LArCablingConfig.LArOnOffIdMappingCfg
def LArOnOffIdMappingCfg(configFlags)
Definition: LArCablingConfig.py:62
LArGMConfig.LArGMCfg
def LArGMCfg(flags)
Definition: LArGMConfig.py:8
python.GetInputFiles.GetInputFilesFromPrefix
def GetInputFilesFromPrefix(inputPath, prefix=".*")
Definition: GetInputFiles.py:37
python.TriggerInterface.TrigDecisionToolCfg
def TrigDecisionToolCfg(flags)
Definition: TriggerInterface.py:14
python.LArRunFormat.getLArFormatForRun
def getLArFormatForRun(run, quiet=False, connstring="COOLONL_LAR/CONDBR2")
Definition: LArRunFormat.py:60
python.AllConfigFlags.initConfigFlags
def initConfigFlags()
Definition: AllConfigFlags.py:19
python.GetInputFiles.GetInputFilesFromPattern
def GetInputFilesFromPattern(inputPath, prefix=".*", trigger=".*")
Definition: GetInputFiles.py:32
python.LArCalibConfigFlags.addLArCalibFlags
def addLArCalibFlags(flags, isSC=False)
Definition: LArCalibConfigFlags.py:3
Generated on Sun Oct 26 2025 21:13:14 for ATLAS Offline Software by   doxygen 1.8.18