Skip to content

Commit c70bf6e

Browse files
committed
new pT hard spectra weight and remove tf index to cfg file
1 parent b74cca3 commit c70bf6e

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

MC/bin/o2dpg_sim_workflow.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
parser.add_argument('-ptHatBin',help='pT hard bin number', default=-1)
3131
parser.add_argument('-ptHatMin',help='pT hard minimum when no bin requested', default=0)
3232
parser.add_argument('-ptHatMax',help='pT hard maximum when no bin requested', default=-1)
33+
parser.add_argument('-weightPow',help='Flatten pT hard spectrum with power', default=-1)
34+
3335
parser.add_argument('-ptTrigMin',help='generated pT trigger minimum', default=0)
3436
parser.add_argument('-ptTrigMax',help='generated pT trigger maximum', default=-1)
3537

@@ -122,6 +124,9 @@ def getDPL_global_options(bigshm=False,nosmallrate=False):
122124
PTTRIGMIN=float(args.ptTrigMin)
123125
PTTRIGMAX=float(args.ptTrigMax)
124126

127+
## Pt Hat productions
128+
WEIGHTPOW=int(args.weightPow)
129+
125130
# Recover PTHATMIN and PTHATMAX from pre-defined array depending bin number PTHATBIN
126131
# or just the ones passed
127132
PTHATBIN=int(args.ptHatBin)
@@ -185,14 +190,16 @@ def getDPL_global_options(bigshm=False,nosmallrate=False):
185190
SGN_CONFIG_task=createTask(name='gensgnconf_'+str(tf), tf=tf, cwd=timeframeworkdir)
186191
if GENERATOR == 'pythia8':
187192
SGN_CONFIG_task['cmd'] = '${O2DPG_ROOT}/MC/config/common/pythia8/utils/mkpy8cfg.py \
188-
--output=pythia8_'+ str(tf) +'.cfg \
193+
--output=pythia8.cfg \
189194
--seed='+str(RNDSEED)+' \
190195
--idA='+str(PDGA)+' \
191196
--idB='+str(PDGB)+' \
192197
--eCM='+str(ECMS)+' \
193198
--process='+str(PROCESS)+' \
194199
--ptHatMin=' + str(PTHATMIN) + ' \
195200
--ptHatMax=' + str(PTHATMAX)
201+
if WEIGHTPOW > -1:
202+
SGN_CONFIG_task['cmd'] = SGN_CONFIG_task['cmd'] + ' --weightPow=' + str(WEIGHTPOW)
196203
workflow['stages'].append(SGN_CONFIG_task)
197204
# elif GENERATOR == 'extgen': what do we do if generator is not pythia8?
198205

0 commit comments

Comments
 (0)