6 if __name__==
'__main__':
12 parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
14 parser.add_argument(
'-r',
'--run', dest=
'run', default=
'00408918', help=
'Run number string as in input filename', type=str)
15 parser.add_argument(
'-g',
'--gain', dest=
'gain', default=
"MEDIUM", help=
'Gain string', type=str)
16 parser.add_argument(
'-p',
'--partition', dest=
'partition', default=
"Em", help=
'Data taking partition string', type=str)
17 parser.add_argument(
'-f',
'--fileprefix', dest=
'fprefix', default=
"data24_calib", help=
'File prefix string', type=str)
18 parser.add_argument(
'-i',
'--indirprefix', dest=
'dprefix', default=
"/eos/atlas/atlastier0/rucio/", help=
'Input directory prefix string', type=str)
19 parser.add_argument(
'-d',
'--indir', dest=
'indir', default=
"", help=
'Full input dir string', type=str)
20 parser.add_argument(
'-t',
'--trigger', dest=
'trig', default=
'calibration_', help=
'Trigger string in filename', type=str)
21 parser.add_argument(
'-o',
'--outrwaveprefix', dest=
'outrwaveprefix', default=
"LArCaliWave", help=
'Prefix of CaliWave output root filename', type=str)
22 parser.add_argument(
'-l',
'--outpprefix', dest=
'outpprefix', default=
"LArCaliWave_OFC_Cali", help=
'Prefix of output pool filename', type=str)
23 parser.add_argument(
'-j',
'--outrofcprefix', dest=
'outrofcprefix', default=
"LArOFCCali", help=
'Prefix of output Cali OFC root filename', type=str)
24 parser.add_argument(
'-e',
'--outrdir', dest=
'outrdir', default=
"/eos/atlas/atlascerngroupdisk/det-larg/Temp/Weekly/ntuples", help=
'Output root file directory', type=str)
25 parser.add_argument(
'-k',
'--outpdir', dest=
'outpdir', default=
"/eos/atlas/atlascerngroupdisk/det-larg/Temp/Weekly/poolFiles", help=
'Output pool file directory', type=str)
26 parser.add_argument(
'-u',
'--insqlite', dest=
'insql', default=
"mysql.db", help=
'Input sqlite file with pedestals, in pool output dir.', type=str)
27 parser.add_argument(
'-n',
'--outsqlite', dest=
'outsql', default=
"mysql_delay.db", help=
'Output sqlite file, in pool output dir.', type=str)
28 parser.add_argument(
'-m',
'--subdet', dest=
'subdet', default=
"EMB", help=
'Subdetector, EMB, EMEC, HEC, FCAL or HECFCAL', type=str)
29 parser.add_argument(
'-s',
'--side', dest=
'side', default=
"C", help=
'Detector side empty (means both), C or A', type=str)
30 parser.add_argument(
'-c',
'--isSC', dest=
'supercells', default=
False, action=
"store_true", help=
'is SC data ?')
31 parser.add_argument(
'-a',
'--isRawdata', dest=
'rawdata', default=
False, action=
"store_true", help=
'is raw data ?')
32 parser.add_argument(
'-x',
'--ignoreBarrel', dest=
'ignoreB', default=
False, action=
"store_true", help=
'ignore Barrel channels ?')
33 parser.add_argument(
'-v',
'--ignoreEndcap', dest=
'ignoreE', default=
False, action=
"store_true", help=
'ignore Endcap channels ?')
34 parser.add_argument(
'-b',
'--badchansqlite', dest=
'badsql', default=
"SnapshotBadChannel.db", help=
'Output sqlite file, in pool output dir.', type=str)
36 args = parser.parse_args()
37 if help
in args
and args.help
is not None and args.help:
41 for _, value
in args._get_kwargs():
46 args.run = args.run.zfill(8)
52 gain=args.gain.lower().capitalize()
53 if not args.supercells:
54 partstr = args.partition
56 partstr = args.partition+
"-DT"
60 InputDir = args.dprefix+args.fprefix+
"/calibration_LArElec-Delay-32s-"+gain+
"-"+partstr+
"/"+args.run+
"/"+args.fprefix+
"."+args.run+
".calibration_LArElec-Delay-32s-"+gain+
"-"+partstr+
".daq.RAW/"
63 from LArCalibProcessing.LArCalib_Delay_OFCCaliConfig
import LArDelay_OFCCaliCfg
66 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
69 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
70 from LArCalibProcessing.LArCalibConfigFlags
import addLArCalibFlags
76 flags.LArCalib.Input.Dir = InputDir
77 flags.LArCalib.Input.Type = args.trig
78 flags.LArCalib.Input.RunNumbers = [
int(args.run),]
79 flags.LArCalib.Input.Database = args.outpdir +
"/" +args.insql
80 gainNumMap={
"HIGH":0,
"MEDIUM":1,
"LOW":2}
81 flags.LArCalib.Gain=gainNumMap[args.gain.upper()]
85 flags.Input.Files=flags.LArCalib.Input.Files
87 print (
"Input files to be processed:")
88 for f
in flags.Input.Files:
91 if len(flags.Input.Files) == 0 :
92 print(
"Unable to find any input files. Please check the input directory:",InputDir)
97 if not flags.LArCalib.isSC:
98 if args.subdet ==
'EMB' or args.subdet ==
'EMEC':
99 flags.LArCalib.Input.SubDet=
"EM"
101 flags.LArCalib.Input.SubDet=args.subdet
104 flags.LArCalib.Preselection.Side = [0,1]
105 elif args.side ==
"C":
106 flags.LArCalib.Preselection.Side = [0]
107 elif args.side ==
"A":
108 flags.LArCalib.Preselection.Side = [1]
110 print(
"unknown side ",args.side)
113 if args.subdet !=
"EM":
114 if args.subdet ==
'EMB':
115 flags.LArCalib.Preselection.BEC = [0]
117 flags.LArCalib.Preselection.BEC = [1]
119 if args.subdet ==
'FCAL':
120 flags.LArCalib.Preselection.FT = [6]
121 elif args.subdet ==
'HEC':
122 flags.LArCalib.Preselection.FT = [3,10,16,22]
123 elif args.subdet ==
'HECFCAL':
124 flags.LArCalib.Preselection.FT = [3,6,10,16,22]
128 flags.LArCalib.BadChannelDB = args.outpdir +
"/" + args.badsql
131 OutputCaliWaveRootFileName = args.outrwaveprefix +
"_" + args.run
132 OutputPoolFileName = args.outpprefix +
"_" + args.run
133 OutputOFCCaliRootFileName = args.outrofcprefix +
"_" + args.run
135 if args.subdet !=
"" and not flags.LArCalib.isSC:
136 OutputCaliWaveRootFileName +=
"_"+args.subdet
137 OutputPoolFileName +=
"_"+args.subdet
138 OutputOFCCaliRootFileName +=
"_"+args.subdet
140 if flags.LArCalib.Input.SubDet==
"EM":
141 OutputCaliWaveRootFileName += args.side
142 OutputPoolFileName += args.side
143 OutputOFCCaliRootFileName += args.side
145 OutputCaliWaveRootFileName +=
".root"
146 OutputPoolFileName +=
".pool.root"
147 OutputOFCCaliRootFileName +=
".root"
149 flags.LArCalib.Output.ROOTFile = args.outrdir +
"/" + OutputCaliWaveRootFileName
150 flags.LArCalib.Output.POOLFile = args.outpdir +
"/" + OutputPoolFileName
151 flags.LArCalib.Output.ROOTFile2 = args.outrdir +
"/" + OutputOFCCaliRootFileName
152 if args.outsql.startswith(
"/"):
153 flags.IOVDb.DBConnection=
"sqlite://;schema=" + args.outsql +
";dbname=CONDBR2"
155 flags.IOVDb.DBConnection=
"sqlite://;schema="+args.outpdir +
"/" + args.outsql +
";dbname=CONDBR2"
158 flags.IOVDb.GlobalTag =
"LARCALIB-RUN2-00"
162 flags.LArCalib.doValidation=
False
168 flags.Exec.OutputLevel = INFO
170 from AthenaConfiguration.Enums
import LHCPeriod
171 flags.GeoModel.Run = LHCPeriod.Run3
173 from AthenaConfiguration.TestDefaults
import defaultGeometryTags
174 flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN3
194 cfg.getEventAlgo(
"LArRawSCCalibDataReadingAlg").LATOMEDecoder.IgnoreBarrelChannels=args.ignoreB
196 cfg.getEventAlgo(
"LArRawSCCalibDataReadingAlg").LATOMEDecoder.IgnoreEndcapChannels=args.ignoreE
198 cfg.getService(
"MessageSvc").defaultLimit=20000
205 cmdline = ([
'/afs/cern.ch/user/l/larcalib/LArDBTools/python/BuildTagHierarchy.py',args.outpdir +
"/" + args.outsql , flags.IOVDb.GlobalTag])
208 subprocess.run(cmdline, check=
True)
209 except Exception
as e:
210 print(
'Could not create tag hierarchy in output sqlite file !!!!')