Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[SimWF,WIP] Enable MaterialManager input from CCDB #1650

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

[SimWF,WIP] Enable MaterialManager input from CCDB #1650

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 21 additions & 2 deletions MC/bin/o2dpg_sim_workflow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@
parser.add_argument('-ini',help='generator init parameters file (full paths required), for example: ${O2DPG_ROOT}/MC/config/PWGHF/ini/GeneratorHF.ini', default='')
parser.add_argument('-confKey',help='generator or trigger configuration key values, for example: "GeneratorPythia8.config=pythia8.cfg;A.x=y"', default='')
parser.add_argument('--readoutDets',help='comma separated string of detectors readout (does not modify material budget - only hit creation)', default='all')
parser.add_argument('--material-params-ccdb', dest='material_params_ccdb', action='store_true', help='to fetch the material parameters used during transport from CCDB')

parser.add_argument('-interactionRate',help='Interaction rate, used in digitization', default=-1)
parser.add_argument('-bcPatternFile',help='Bunch crossing pattern file, used in digitization (a file name or "ccdb")', default='')
Expand Down Expand Up @@ -362,6 +363,24 @@ def getDPL_global_options(bigshm=False, ccdbbackend=True):
if (includeLocalQC or includeFullQC) and not isdir(qcdir):
mkdir(qcdir)

# prepare fetching and configuration for MaterialManagerParam
transport_needs = []
if args.material_params_ccdb:
# the target file name
mat_mgr_file_name = 'material_manager_params.json'
# the path on CCDB
ccdb_path = 'SIM_TEST/ALIBI/SIM_CUTS'
# this is the full local path of where the parameter file will be copied to
mat_mgr_file_path = f'${{ALICEO2_CCDB_LOCALCACHE}}/{ccdb_path}/{mat_mgr_file_name}'
# simply append, the last one takes precedence
args.confKey += f';MaterialManagerParam.inputFile={mat_mgr_file_path}'
args.confKeyBkg += f';MaterialManagerParam.inputFile={mat_mgr_file_path}'

MAT_PARAMS_DOWNLOADER_TASK = createTask(name='download_mat_mgr_params', cpu=0)
MAT_PARAMS_DOWNLOADER_TASK['cmd'] = f'${{O2_ROOT}}/bin/o2-ccdb-downloadccdbfile --host http://ccdb-test.cern.ch:8080 -p {ccdb_path} --timestamp -1 -d ${{ALICEO2_CCDB_LOCALCACHE}} -o {mat_mgr_file_name}'
workflow['stages'].append(MAT_PARAMS_DOWNLOADER_TASK)
transport_needs.append(MAT_PARAMS_DOWNLOADER_TASK['name'])

# create/publish the GRPs and other GLO objects for consistent use further down the pipeline
orbitsPerTF=int(args.orbitsPerTF)
GRP_TASK = createTask(name='grpcreate', cpu='0')
Expand Down Expand Up @@ -455,7 +474,7 @@ def getDPL_global_options(bigshm=False, ccdbbackend=True):
# determine final configKey values for background transport
CONFKEYBKG = constructConfigKeyArg(create_geant_config(args, args.confKeyBkg))

BKGtask=createTask(name='bkgsim', lab=["GEANT"], needs=[BKG_CONFIG_task['name'], GRP_TASK['name']], cpu=NWORKERS )
BKGtask=createTask(name='bkgsim', lab=["GEANT"], needs=[BKG_CONFIG_task['name'], GRP_TASK['name']] + transport_needs, cpu=NWORKERS )
BKGtask['cmd']='${O2_ROOT}/bin/o2-sim -e ' + SIMENGINE + ' -j ' + str(NWORKERS) + ' -n ' + str(NBKGEVENTS) \
+ ' -g ' + str(GENBKG) + ' ' + str(MODULES) + ' -o bkg ' + str(INIBKG) \
+ ' --field ccdb ' + str(CONFKEYBKG) \
Expand Down Expand Up @@ -700,7 +719,7 @@ def getDPL_global_options(bigshm=False, ccdbbackend=True):
# -----------------
# transport signals
# -----------------
signalneeds=[ SGN_CONFIG_task['name'], GRP_TASK['name'] ]
signalneeds=[ SGN_CONFIG_task['name'], GRP_TASK['name'] ] + transport_needs
if (args.pregenCollContext == True):
signalneeds.append(PreCollContextTask['name'])

Expand Down