6 if __name__==
'__main__':
10 from AthenaCommon
import Logging
11 log = Logging.logging.getLogger(
'LArRamp' )
14 parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
16 parser.add_argument(
'-r',
'--run', dest=
'run', default=
'00408920', help=
'Run number string as in input filename', type=str)
17 parser.add_argument(
'-g',
'--gain', dest=
'gain', default=
"MEDIUM", help=
'Gain string', type=str)
18 parser.add_argument(
'-p',
'--partition', dest=
'partition', default=
"Em", help=
'Data taking partition string', type=str)
19 parser.add_argument(
'-f',
'--fileprefix', dest=
'fprefix', default=
"data25_calib", help=
'File prefix string', type=str)
20 parser.add_argument(
'-i',
'--indirprefix', dest=
'dprefix', default=
"/eos/atlas/atlastier0/rucio/", help=
'Input directory prefix string', type=str)
21 parser.add_argument(
'-d',
'--indir', dest=
'indir', default=
"", help=
'Full input dir string', type=str)
22 parser.add_argument(
'-t',
'--trigger', dest=
'trig', default=
'calibration_', help=
'Trigger string in filename', type=str)
23 parser.add_argument(
'-o',
'--outrprefix', dest=
'outrprefix', default=
"LArRamp", help=
'Prefix of output ramp root filename', type=str)
24 parser.add_argument(
'-j',
'--outpprefix', dest=
'outpprefix', default=
"LArRamp", help=
'Prefix of output ramp pool filename', type=str)
25 parser.add_argument(
'-e',
'--outrdir', dest=
'outrdir', default=
"/eos/atlas/atlascerngroupdisk/det-larg/Temp/Weekly/ntuples", help=
'Output root file directory', type=str)
26 parser.add_argument(
'-k',
'--outpdir', dest=
'outpdir', default=
"/eos/atlas/atlascerngroupdisk/det-larg/Temp/Weekly/poolFiles", help=
'Output pool file directory', type=str)
27 parser.add_argument(
'-u',
'--inpsqlite', dest=
'inpsql', default=
"mysql.db", help=
'Input sqlite file with pedestals, in pool output dir.', type=str)
28 parser.add_argument(
'-l',
'--inofcsqlite', dest=
'inofcsql', default=
"mysql_delay.db", help=
'Input sqlite file with ofcs, in pool output dir', type=str)
29 parser.add_argument(
'-n',
'--outsqlite', dest=
'outsql', default=
"mysql_ramp.db", help=
'Output sqlite file, in pool output dir.', type=str)
30 parser.add_argument(
'-m',
'--subdet', dest=
'subdet', default=
"EMB", help=
'Subdetector, EMB, EMEC, HEC or FCAL', type=str)
31 parser.add_argument(
'-s',
'--side', dest=
'side', default=
"C", help=
'Detector side empty (means both), C or A', type=str)
32 parser.add_argument(
'-c',
'--isSC', dest=
'supercells', default=
False, action=
"store_true", help=
'is SC data ?')
33 parser.add_argument(
'-a',
'--isRawdata', dest=
'rawdata', default=
False, action=
"store_true", help=
'is raw data ?')
34 parser.add_argument(
'-b',
'--badchansqlite', dest=
'badsql', default=
"SnapshotBadChannel.db", help=
'Output sqlite file, in pool output dir.', type=str)
35 parser.add_argument(
'-x',
'--ignoreBarrel', dest=
'ignoreB', default=
False, action=
"store_true", help=
'ignore Barrel channels ?')
36 parser.add_argument(
'-v',
'--ignoreEndcap', dest=
'ignoreE', default=
False, action=
"store_true", help=
'ignore Endcap channels ?')
37 parser.add_argument(
'-w',
'--doValid', dest=
'doValid', default=
False, action=
"store_true", help=
'run vcalidation ?')
38 parser.add_argument(
'--FW6', dest=
'fw6', default=
False, help=
'Is it for fw v. 6', action=
'store_true')
39 parser.add_argument(
'--EMF', dest=
'emf', default=
False, help=
'Is it for EMF', action=
'store_true')
42 args = parser.parse_args()
43 if help
in args
and args.help
is not None and args.help:
47 for _, value
in args._get_kwargs():
52 args.run = args.run.zfill(8)
58 gain=args.gain.lower().capitalize()
60 if not args.supercells:
61 partstr = args.partition
63 partstr = args.partition+
"-DT"
68 InputDir = args.dprefix+args.fprefix+
"/calibration_LArElec-Ramp-32s-"+gain+
"-"+partstr+
"/"+args.run+
"/"+args.fprefix+
"."+args.run+
".calibration_LArElec-Ramp-32s-"+gain+
"-"+partstr+
".daq.RAW/"
70 InputDir = args.dprefix+args.fprefix+
"/calibration_LArElec-Ramp-7s-"+gain+
"-"+partstr+
"/"+args.run+
"/"+args.fprefix+
"."+args.run+
".calibration_LArElec-Ramp-7s-"+gain+
"-"+partstr+
".daq.RAW/"
74 from LArCalibProcessing.LArCalib_RampConfig
import LArRampCfg
77 from AthenaConfiguration.MainServicesConfig
import MainServicesCfg
80 from AthenaConfiguration.AllConfigFlags
import initConfigFlags
81 from LArCalibProcessing.LArCalibConfigFlags
import addLArCalibFlags
87 flags.LArCalib.Input.Dir = InputDir
88 flags.LArCalib.Input.Type = args.trig
89 flags.LArCalib.Input.RunNumbers = [
int(args.run),]
90 flags.LArCalib.Input.Database = args.outpdir +
"/" +args.inpsql
91 flags.LArCalib.Input.isRawData = args.rawdata
92 if 'db' in args.inofcsql:
93 flags.LArCalib.Input.Database2 = args.outpdir +
"/" +args.inofcsql
95 flags.LArCalib.Input.Database2 = args.inofcsql
96 gainNumMap={
"HIGH":0,
"MEDIUM":1,
"LOW":2}
97 flags.LArCalib.Gain=gainNumMap[args.gain.upper()]
100 flags.Input.Files=flags.LArCalib.Input.Files
102 print (
"Input files to be processed:")
103 for f
in flags.Input.Files:
106 if len(flags.Input.Files) == 0 :
107 print(
"Unable to find any input files. Please check the input directory:",InputDir)
112 if not flags.LArCalib.isSC:
113 if args.subdet ==
'EMB' or args.subdet ==
'EMEC':
114 flags.LArCalib.Input.SubDet=
"EM"
116 flags.LArCalib.Input.SubDet=args.subdet
119 flags.LArCalib.Preselection.Side = [0,1]
120 elif args.side ==
"C":
121 flags.LArCalib.Preselection.Side = [0]
122 elif args.side ==
"A":
123 flags.LArCalib.Preselection.Side = [1]
125 print(
"unknown side ",args.side)
128 if args.subdet !=
"EM":
129 if args.subdet ==
'EMB':
130 flags.LArCalib.Preselection.BEC = [0]
132 flags.LArCalib.Preselection.BEC = [1]
134 if args.subdet ==
'FCAL':
135 flags.LArCalib.Preselection.FT = [6]
136 elif args.subdet ==
'HEC':
137 flags.LArCalib.Preselection.FT = [3,10,16,22]
138 elif args.subdet ==
'HECFCAL':
139 flags.LArCalib.Preselection.FT = [3,6,10,16,22]
141 flags.LArCalib.Input.SubDet=args.subdet
145 flags.LArCalib.BadChannelDB = args.outpdir +
"/" + args.badsql
148 OutputRampRootFileName = args.outrprefix +
"_" + args.run
149 OutputRampPoolFileName = args.outpprefix +
"_" + args.run
151 if args.subdet !=
"" and not flags.LArCalib.isSC:
152 OutputRampRootFileName +=
"_"+args.subdet
153 OutputRampPoolFileName +=
"_"+args.subdet
154 if flags.LArCalib.Input.SubDet==
"EM":
155 OutputRampRootFileName += args.side
156 OutputRampPoolFileName += args.side
158 OutputRampRootFileName +=
".root"
159 OutputRampPoolFileName +=
".pool.root"
161 flags.LArCalib.Output.ROOTFile = args.outrdir +
"/" + OutputRampRootFileName
162 flags.LArCalib.Output.POOLFile = args.outpdir +
"/" + OutputRampPoolFileName
163 flags.IOVDb.DBConnection=
"sqlite://;schema="+args.outpdir +
"/" + args.outsql +
";dbname=CONDBR2"
166 flags.IOVDb.GlobalTag =
"LARCALIB-RUN2-00"
168 from AthenaConfiguration.TestDefaults
import defaultGeometryTags
169 flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN3
172 if flags.LArCalib.isSC:
173 flags.LArCalib.doValidation= args.doValid
175 flags.LArCalib.doValidation=
True
179 pdir=
'/afs/cern.ch/user/l/lardaq/public/detlar/athena/P1CalibrationProcessing/run/Patterns/P1/'
181 pdir +=
'LatomeRuns/'
182 if 'Emec' in args.partition:
183 pfile = pdir +
'emec-std/SC_HighRamp/parameters.dat'
185 pfile = pdir +
'barrel/Ramp_' + args.partition[:-4] +
'/parameters.dat'
187 flags.LArCalib.Input.paramsFile = pfile
196 flags.Exec.OutputLevel = INFO
198 from AthenaConfiguration.Enums
import LHCPeriod
199 flags.GeoModel.Run = LHCPeriod.Run3
203 flags.IOVDb.SqliteInput=
"/afs/cern.ch/user/p/pavol/public/EMF_otherCond.db"
204 flags.IOVDb.SqliteFolders = (
"/LAR/BadChannelsOfl/BadChannelsSC",
"/LAR/BadChannels/BadChannelsSC",
"/LAR/Identifier/OnOffIdMap",)
205 fldrs=cfg.getService(
"IOVDbSvc").Folders
206 for i
in range(0, len(fldrs)):
207 if 'LatomeMapping' in fldrs[i]: fldrs[i] +=
'<tag>LARIdentifierLatomeMapping-EMF</tag>'
231 if flags.LArCalib.isSC:
237 log.info(
"Got DT run info !")
239 log.warning(
"Could not get DT run info, using defaults !")
241 fwversion=runinfo.FWversion()
242 if args.fw6
or fwversion==6:
246 fldrs=cfg.getService(
"IOVDbSvc").Folders
247 for i
in range(0, len(fldrs)):
248 if 'LatomeMapping' in fldrs[i]: fldrs[i] +=
'<tag>LARIdentifierLatomeMapping-fw6</tag>'
254 cfg.getEventAlgo(
"LArRawSCDataReadingAlg").LATOMEDecoder.IgnoreBarrelChannels=args.ignoreB
256 cfg.getEventAlgo(
"LArRawSCCalibDataReadingAlg").LATOMEDecoder.IgnoreBarrelChannels=args.ignoreB
259 cfg.getEventAlgo(
"LArRawSCDataReadingAlg").LATOMEDecoder.IgnoreEndcapChannels=args.ignoreE
261 cfg.getEventAlgo(
"LArRawSCCalibDataReadingAlg").LATOMEDecoder.IgnoreEndcapChannels=args.ignoreE
263 cfg.getService(
"IOVDbSvc").DBInstance=
""
270 cmdline = ([
'/afs/cern.ch/user/l/larcalib/LArDBTools/python/BuildTagHierarchy.py',args.outpdir +
"/" + args.outsql , flags.IOVDb.GlobalTag])
273 subprocess.run(cmdline, check=
True)
274 except Exception
as e:
275 print(
'Could not create tag hierarchy in output sqlite file !!!!')