160def setupGeoR4TestCfg(args, flags = None):
161
162 if flags is None:
163 from AthenaConfiguration.AllConfigFlags import initConfigFlags
164 flags = initConfigFlags()
165 flags.Concurrency.NumThreads = args.threads
166 flags.Concurrency.NumConcurrentEvents = args.threads
167 flags.Exec.MaxEvents = args.nEvents
168 flags.Exec.SkipEvents = args.skipEvents
169 from os import path, system, listdir
170 flags.Input.Files = []
171
173 for fileArg in args.inputFile:
174 if path.isdir(fileArg):
175 flags.Input.Files += [ "{dir}/{file}".format(dir=fileArg, file=y) for y in listdir(fileArg) ]
176 else:
177 if fileArg[fileArg.rfind(".")+1 :]not in ["txt", "conf"]:
178 flags.Input.Files+=[fileArg]
179 else:
180 with open(fileArg) as inStream:
181 flags.Input.Files+=[ line.strip() for line in inStream if line[0]!='#']
182
183 flags.Exec.FPE= 500
184 flags.Exec.EventPrintoutInterval = 500
185
186 if args.defaultGeoFile == "RUN3":
187 flags.GeoModel.SQLiteDBFullPath = MuonPhaseIITestDefaults.GEODB_R3
188 elif args.defaultGeoFile == "RUN4":
189 flags.GeoModel.SQLiteDBFullPath = MuonPhaseIITestDefaults.GEODB_R4
190 elif args.defaultGeoFile == "RUN3MSOnly":
191 flags.GeoModel.SQLiteDBFullPath = MuonPhaseIITestDefaults.GEODB_R3MSOnly
192 elif args.defaultGeoFile == "RUN4MSOnly":
193 flags.GeoModel.SQLiteDBFullPath = MuonPhaseIITestDefaults.GEODB_R4MSOnly
194 elif args.defaultGeoFile == "ITkR3MS":
195 flags.GeoModel.SQLiteDBFullPath = MuonPhaseIITestDefaults.GEODB_ITk_R3MS
196 elif args.geoModelFile.startswith("root://"):
197 if not path.exists("Geometry/{geoTag}.db".format(geoTag=args.geoTag)):
198 print ("Copy geometry file from EOS {source}".format(source = args.geoModelFile))
199 system("mkdir Geometry/")
200 system("xrdcp {source} Geometry/{geoTag}.db".format(source = args.geoModelFile,
201 geoTag=args.geoTag))
202
203 args.geoModelFile = "Geometry/{geoTag}.db".format(geoTag=args.geoTag)
204 else:
205 flags.GeoModel.SQLiteDBFullPath = args.geoModelFile
206
207 flags.GeoModel.SQLiteDB = True
208 configureDefaultTagsCfg(flags)
209
210 flags.Detector.GeometryBpipe = False
211
212 flags.Detector.GeometryBCM = False
213 flags.Detector.GeometryPixel = False
214 flags.Detector.GeometrySCT = False
215 flags.Detector.GeometryTRT = False
216
217 flags.Detector.GeometryPLR = False
218 flags.Detector.GeometryBCMPrime = False
219 flags.Detector.GeometryITkPixel = False
220 flags.Detector.GeometryITkStrip = False
221
222 flags.Detector.GeometryHGTD = False
223
224 flags.Detector.GeometryLAr = False
225 flags.Detector.GeometryTile = False
226 flags.Detector.GeometryMBTS = False
227 flags.Detector.GeometryCalo = False
228
229 flags.Detector.GeometryCSC = False
230 if args.noSTGC:
231 flags.Detector.GeometrysTGC = False
232 if args.noMM:
233 flags.Detector.GeometryMM = False
234 if args.noTgc:
235 flags.Detector.GeometryTGC = False
236 if args.noRpc:
237 flags.Detector.GeometryRPC = False
238 if args.noMdt:
239 flags.Detector.GeometryMDT = False
240
241 flags.Acts.TrackingGeometry.UseBlueprint = True
242
243 flags.Scheduler.CheckDependencies = True
244 flags.Scheduler.ShowDataDeps = True
245 flags.Scheduler.ShowDataFlow = True
246 flags.Scheduler.ShowControlFlow = True
247 flags.Scheduler.EnableVerboseViews = True
248 flags.Scheduler.AutoLoadUnmetDependencies = True
249
250 flags.lock()
251 flags.dump(evaluate = True)
252 cfg = setupServicesCfg(flags)
253
254 from MuonConfig.MuonGeometryConfig import MuonGeoModelCfg
255 cfg.merge(MuonGeoModelCfg(flags))
256
257 if not flags.Muon.usePhaseIIGeoSetup:
258 print ("WARNING: New Muon plugin is not part of the Geometry file {geoDBFile}".format(geoDBFile=args.geoModelFile))
259 else:
260 from ActsAlignmentAlgs.AlignmentAlgsConfig import ActsGeometryContextAlgCfg
261 cfg.merge(ActsGeometryContextAlgCfg(flags))
262
263 cfg.getService("MessageSvc").verboseLimit = 10000000
264 cfg.getService("MessageSvc").debugLimit = 10000000
265
266 return flags, cfg
267