Skip to content

Commit fb2d40f

Browse files
committed
Introduce GRP fetching/setup task
1 parent 46edf50 commit fb2d40f

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

MC/bin/o2dpg_sim_workflow.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ def getDPL_global_options(bigshm=False):
330330
if args.iniBkg!= '':
331331
INIBKG=' --configFile ' + args.iniBkg
332332

333-
BKGtask=createTask(name='bkgsim', lab=["GEANT"], needs=[BKG_CONFIG_task['name']], cpu=NWORKERS )
333+
BKGtask=createTask(name='bkgsim', lab=["GEANT"], needs=[BKG_CONFIG_task['name'], GRP_TASK['name']], cpu=NWORKERS )
334334
BKGtask['cmd']='${O2_ROOT}/bin/o2-sim -e ' + SIMENGINE + ' -j ' + str(NWORKERS) + ' -n ' + str(NBKGEVENTS) \
335335
+ ' -g ' + str(GENBKG) + ' ' + str(MODULES) + ' -o bkg ' + str(INIBKG) \
336336
+ ' --field ' + str(BFIELD) + ' ' + str(CONFKEYBKG) \
@@ -394,6 +394,11 @@ def getDPL_global_options(bigshm=False):
394394
MATBUD_DOWNLOADER_TASK['cmd'] = '[ -f matbud.root ] || ${O2_ROOT}/bin/o2-ccdb-downloadccdbfile --host http://alice-ccdb.cern.ch/ -p GLO/Param/MatLUT -o matbud.root --no-preserve-path --timestamp ' + str(args.timestamp)
395395
workflow['stages'].append(MATBUD_DOWNLOADER_TASK)
396396

397+
orbitsPerTF=256
398+
GRP_TASK = createTask(name='grpcreate', cpu='0')
399+
GRP_TASK['cmd'] = 'o2-grp-simgrp-tool createGRPs --run ' + str(args.run) + ' --publishto ${ALICEO2_CCDB_LOCALCACHE:-.ccdb} -o grp --hbfpertf ' + str(orbitsPerTF) + ' --field ' + args.field
400+
workflow['stages'].append(GRP_TASK)
401+
397402
# loop over timeframes
398403
for tf in range(1, NTIMEFRAMES + 1):
399404
TFSEED = SIMSEED + tf
@@ -501,7 +506,7 @@ def getDPL_global_options(bigshm=False):
501506
# transport signals
502507
# -----------------
503508
signalprefix='sgn_' + str(tf)
504-
signalneeds=[ SGN_CONFIG_task['name'] ]
509+
signalneeds=[ SGN_CONFIG_task['name'], GRP_TASK['name'] ]
505510

506511
# add embedIntoFile only if embeddPattern does contain a '@'
507512
embeddinto= "--embedIntoFile ../bkg_MCHeader.root" if (doembedding & ("@" in args.embeddPattern)) else ""
@@ -570,7 +575,6 @@ def getDPL_global_options(bigshm=False):
570575
simsoption=' --sims ' + ('bkg,'+signalprefix if doembedding else signalprefix)
571576

572577
# each timeframe should be done for a different bunch crossing range, depending on the timeframe id
573-
orbitsPerTF = 256
574578
startOrbit = (tf-1 + int(args.production_offset)*NTIMEFRAMES)*orbitsPerTF
575579
globalTFConfigValues = { "HBFUtils.orbitFirstSampled" : args.first_orbit + startOrbit,
576580
"HBFUtils.nHBFPerTF" : orbitsPerTF,

0 commit comments

Comments
 (0)