ATLAS Offline Software
CaloNoise2Ntuple.py
Go to the documentation of this file.
1 #!/bin/env python
2 # Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
3 
4 from AthenaConfiguration.ComponentFactory import CompFactory
5 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
6 
7 def CaloNoise2NtupleCfg(flags,outputFile):
8 
10  from LArGeoAlgsNV.LArGMConfig import LArGMCfg
11  cfg.merge(LArGMCfg(flags))
12  from TileGeoModel.TileGMConfig import TileGMCfg
13  cfg.merge(TileGMCfg(flags))
14 
15  from CaloTools.CaloNoiseCondAlgConfig import CaloNoiseCondAlgCfg
16  cfg.merge(CaloNoiseCondAlgCfg(flags,"totalNoise"))
17  cfg.merge(CaloNoiseCondAlgCfg(flags,"electronicNoise"))
18  cfg.merge(CaloNoiseCondAlgCfg(flags,"pileupNoise"))
19 
20  cfg.addEventAlgo(CompFactory.CaloNoise2Ntuple())
21  cfg.addService(CompFactory.THistSvc(Output = ["file1 DATAFILE='"+outputFile+"' OPT='RECREATE'"]))
22 
23  return cfg
24 
25 if __name__=="__main__":
26  import sys,argparse
27  parser= argparse.ArgumentParser()
28  parser.add_argument("--loglevel", default=None, help="logging level (ALL, VERBOSE, DEBUG,INFO, WARNING, ERROR, or FATAL")
29  parser.add_argument("-r","--runnumber",default=0x7fffffff, type=int, help="run number to query the DB")
30  parser.add_argument("-l","--lbnumber",default=1, type=int, help="LB number to query the DB")
31  parser.add_argument("-d","--database",default=None, help="Database name or sqlite file name")
32  parser.add_argument("-o","--output",default="caloNoise.root", help="output file name")
33  parser.add_argument("-f","--folder",default=None, help="database folder to read")
34  parser.add_argument("-t","--tag",default=None, help="folder-level tag to read")
35  parser.add_argument("-m","--mc", action='store_true', help="data or MC?")
36 
37 
38  (args,leftover)=parser.parse_known_args(sys.argv[1:])
39 
40  if len(leftover)>0:
41  print("ERROR, unhandled argument(s):",leftover)
42  sys.exit(-1)
43 
44  from AthenaConfiguration.AllConfigFlags import initConfigFlags
46  flags.Input.isMC = args.mc
47  flags.IOVDb.DatabaseInstance= "OFLP200" if args.mc else "CONDBR2"
48  from Campaigns.Utils import Campaign
49  flags.Input.MCCampaign=Campaign.Unknown
50  flags.LAr.doAlign=False
51  flags.LAr.doHVCorr=False
52  flags.Input.RunNumbers=[args.runnumber]
53  from AthenaConfiguration.TestDefaults import defaultConditionsTags, defaultGeometryTags
54  flags.IOVDb.GlobalTag=defaultConditionsTags.RUN3_MC if args.mc else defaultConditionsTags.RUN3_DATA
55  flags.GeoModel.AtlasVersion=defaultGeometryTags.RUN3
56 
57 
58  if args.loglevel:
59  from AthenaCommon import Constants
60  if hasattr(Constants,args.loglevel):
61  flags.Exec.OutputLevel=getattr(Constants,args.loglevel)
62  else:
63  raise ValueError("Unknown log-level, allowed values are ALL, VERBOSE, DEBUG,INFO, WARNING, ERROR, FATAL")
64 
65  flags.lock()
66 
67  from AthenaConfiguration.MainServicesConfig import MainServicesCfg
68  cfg=MainServicesCfg(flags)
69  #MC Event selector since we have no input data file
70  from McEventSelector.McEventSelectorConfig import McEventSelectorCfg
71  cfg.merge(McEventSelectorCfg(flags,
72  FirstLB=args.lbnumber,
73  EventsPerRun = 1,
74  FirstEvent = 1,
75  InitialTimeStamp = 0,
76  TimeStampInterval = 1))
77  cfg.addEventAlgo(CompFactory.xAODMaker.EventInfoCnvAlg(AODKey = 'McEventInfo'),sequenceName="AthAlgSeq")
78 
79  cfg.merge(CaloNoise2NtupleCfg(flags,args.output))
80 
81  dbstr=None
82  if args.database:
83  dbstr="sqlite://;schema="+args.database+";dbname=" + flags.IOVDb.DatabaseInstance
84 
85  if args.tag or args.database:
86  from IOVDbSvc.IOVDbSvcConfig import addOverride
87  cfg.merge(addOverride(flags,"/LAR/NoiseOfl/CellNoise",args.tag,dbstr))
88 
89 
90  sc=cfg.run(1)
91  if sc.isSuccess():
92  sys.exit(0)
93  else:
94  sys.exit(1)
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
CaloNoise2Ntuple.CaloNoise2NtupleCfg
def CaloNoise2NtupleCfg(flags, outputFile)
Definition: CaloNoise2Ntuple.py:7
python.CaloNoiseCondAlgConfig.CaloNoiseCondAlgCfg
def CaloNoiseCondAlgCfg(flags, noisetype="totalNoise")
Definition: CaloNoiseCondAlgConfig.py:11
python.MainServicesConfig.MainServicesCfg
def MainServicesCfg(flags, LoopMgr='AthenaEventLoopMgr')
Definition: MainServicesConfig.py:312
python.McEventSelectorConfig.McEventSelectorCfg
def McEventSelectorCfg(flags, **kwargs)
Definition: McEventSelectorConfig.py:5
LArGMConfig.LArGMCfg
def LArGMCfg(flags)
Definition: LArGMConfig.py:8
print
void print(char *figname, TCanvas *c1)
Definition: TRTCalib_StrawStatusPlots.cxx:25
python.IOVDbSvcConfig.addOverride
def addOverride(flags, folder, tag, tagType="tag", db=None)
Definition: IOVDbSvcConfig.py:238
python.AllConfigFlags.initConfigFlags
def initConfigFlags()
Definition: AllConfigFlags.py:19
TileGMConfig.TileGMCfg
def TileGMCfg(flags)
Definition: TileGMConfig.py:7