def submit_batch_job(datatype, tag, mode, label, opts):
mode_sign = tools.get_modekey_sign(mode)
ms = tools.pair_to_str(mode_sign)
pyline = 'from sel import multcand; multcand.%s_tag_mode("%s", "%s", "%s", test=%s)'% (tag, datatype, mode, label, opts.test)
bash_file_name = 'multcand-%s.sh' % ms
bash_file = create_bash_file_pyline(opts, label, datatype, pyline,
bash_file_name, subdir='sel')
logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode)
logpath = os.path.join(attr.logpath, label, 'multcand')
logfile = tools.set_file(extbase=logpath, comname=logname)
qjobname = 'sel%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, mode, label, lowmass, highmass, opts):
mode_sign = tools.get_modekey_sign(mode)
ms = tools.pair_to_str(mode_sign)
pyline = 'from fit import kpimass; kpimass.fit_single_mode("%s", "%s", "%s", %s, %s, %s)' % (datatype, mode, label, lowmass, highmass, opts.test)
bash_file_name = 'kpimass-fit-%s.sh' % ms
bash_file = create_bash_file_pyline(opts, label, datatype, pyline,
bash_file_name)
logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode)
prefix='dir_%s/kpimass/%s_%s' % (label, lowmass, highmass)
logfile = tools.set_file(extbase=attr.logpath, prefix=prefix, comname=logname)
qjobname = 'kpi%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, tag, mode, label, opts):
mode_sign = tools.get_modekey_sign(mode)
ms = tools.pair_to_str(mode_sign)
pyline = 'import yld; yld.%s_tag_mode("%s", "%s", "%s", test=%s)'% (
tag, datatype, mode, label, opts.test)
bash_file_name = 'get-yld.sh'
bash_file = create_bash_file_pyline(opts, label, datatype, pyline,
bash_file_name)
logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode)
logpath = os.path.join(attr.logpath, label, 'yld')
logfile = tools.set_file(extbase=logpath, comname=logname)
qjobname = 'sel%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(reweight, datatype, tag, mode, label, opts):
mode_sign = tools.get_modekey_sign(mode)
ms = tools.pair_to_str(mode_sign)
pyline = 'from sel import multspec; multspec.%s_tag_mode("%s", "%s", "%s", "%s", test=%s)' % (
tag, reweight, datatype, mode, label, opts.test)
bash_file_name = 'multspec-%s.sh' % ms
bash_file = create_bash_file_pyline(opts,
label,
datatype,
pyline,
bash_file_name,
subdir='sel')
logname = '%s_%s.txt' % (datatype.replace('/', '_'), mode)
logpath = os.path.join(attr.logpath, label, 'multspec')
logfile = tools.set_file(extbase=logpath, comname=logname)
qjobname = 'sel%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, mode, label, lowmass, highmass, opts):
mode_sign = tools.get_modekey_sign(mode)
ms = tools.pair_to_str(mode_sign)
pyline = 'from fit import kpimass; kpimass.fit_single_mode("%s", "%s", "%s", %s, %s, %s)' % (
datatype, mode, label, lowmass, highmass, opts.test)
bash_file_name = 'kpimass-fit-%s.sh' % ms
bash_file = create_bash_file_pyline(opts, label, datatype, pyline,
bash_file_name)
logname = '%s_%s.txt' % (datatype.replace('/', '_'), mode)
prefix = 'dir_%s/kpimass/%s_%s' % (label, lowmass, highmass)
logfile = tools.set_file(extbase=attr.logpath,
prefix=prefix,
comname=logname)
qjobname = 'kpi%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, tag, mode, label, opts):
mode_sign = tools.get_modekey_sign(mode)
if mode_sign[1] == None:
ms = '%s_%s' %(mode_sign[0])
else:
ms = tools.pair_to_str(mode_sign)
pyline = 'import fit; fit.%s_tag_mode("%s", "%s", "%s", test=%s)'% (
tag, datatype, mode, label, opts.test)
bash_file_name = 'fit-mbc.sh'
bash_file = create_bash_file_pyline(opts, label, datatype, pyline,
bash_file_name)
#logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode)
#logpath = os.path.join(attr.logpath, label)#, 'yld')
#logfile = tools.set_file(extbase=logpath, comname=logname)
logfile = tools.set_file('txt', datatype, mode_sign[0], tag,
prefix='dir_'+label, extbase=attr.logpath)
qjobname = 'fit%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, tag, mode, label, opts):
mode_sign = tools.get_modekey_sign(mode)
if mode_sign[1] == None:
ms = '%s_%s' % (mode_sign[0])
else:
ms = tools.pair_to_str(mode_sign)
pyline = 'import fit; fit.%s_tag_mode("%s", "%s", "%s", test=%s)' % (
tag, datatype, mode, label, opts.test)
bash_file_name = 'fit-mbc.sh'
bash_file = create_bash_file_pyline(opts, label, datatype, pyline,
bash_file_name)
#logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode)
#logpath = os.path.join(attr.logpath, label)#, 'yld')
#logfile = tools.set_file(extbase=logpath, comname=logname)
logfile = tools.set_file('txt',
datatype,
mode_sign[0],
tag,
prefix='dir_' + label,
extbase=attr.logpath)
qjobname = 'fit%s' % ms
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def main(opts, args):
if args[0] == 'crossfeeds':
import crossfeeds
crossfeeds.main(opts, args[1:])
return
if args[0] == 'backgrounds':
import backgrounds
backgrounds.main(opts, args[1:])
return
if args[0] == 'sidebands':
import sidebands
sidebands.main(opts, args[1:])
return
if args[0] == 'kkmass':
import kkmass
kkmass.main(opts, args[1:])
return
if args[0] == 'kkmass2':
import kkmass2
kkmass2.main(opts, args[1:])
return
if args[0] == 'kpimass':
import kpimass
kpimass.main(opts, args[1:])
return
ROOT.gROOT.SetBatch(1)
ROOT.gROOT.SetStyle("Plain")
parsed = parse_args(args)
dt_type = parsed[0]
tag = parsed[1]
modes = parsed[2]
label = parsed[3]
if modes == ['double_all_d0s']:
mode = modes[0]
if opts.set and opts.set == 'interact':
double_all_d0s(dt_type, mode, label, test=opts.test)
else:
pyline = 'import fit; fit.%s("%s", "%s", "%s", test=%s)' % (
mode, dt_type, mode, label, opts.test)
bash_file_name = 'fit-mbc.sh'
bash_file = create_bash_file_pyline(opts, label, dt_type, pyline,
bash_file_name)
logfile = tools.set_file('txt',
dt_type,
mode,
tag,
prefix='dir_' + label,
extbase=attr.logpath)
qjobname = 'fitd0s'
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
return
if modes == ['double_all_dps']:
mode = modes[0]
if opts.set and opts.set == 'interact':
double_all_dps(dt_type, mode, label, test=opts.test)
else:
pyline = 'import fit; fit.%s("%s", "%s", "%s", test=%s)' % (
mode, dt_type, mode, label, opts.test)
bash_file_name = 'fit-mbc.sh'
bash_file = create_bash_file_pyline(opts, label, dt_type, pyline,
bash_file_name)
logfile = tools.set_file('txt',
dt_type,
mode,
tag,
prefix='dir_' + label,
extbase=attr.logpath)
qjobname = 'fitdps'
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
return
for mode in modes:
if opts.set and opts.set == 'interact':
if tag == 'single':
single_tag_mode(dt_type,
mode,
label,
interact=True,
test=opts.test)
elif tag == 'double':
double_tag_mode(dt_type,
mode,
label,
interact=True,
test=opts.test)
else:
raise ValueError(tag)
continue
else:
submit_batch_job(dt_type, tag, mode, label, opts)
def main(opts, args):
if args[0] == 'crossfeeds':
import crossfeeds
crossfeeds.main(opts, args[1:])
return
if args[0] == 'backgrounds':
import backgrounds
backgrounds.main(opts, args[1:])
return
if args[0] == 'sidebands':
import sidebands
sidebands.main(opts, args[1:])
return
if args[0] == 'kkmass':
import kkmass
kkmass.main(opts, args[1:])
return
if args[0] == 'kkmass2':
import kkmass2
kkmass2.main(opts, args[1:])
return
if args[0] == 'kpimass':
import kpimass
kpimass.main(opts, args[1:])
return
ROOT.gROOT.SetBatch(1)
ROOT.gROOT.SetStyle("Plain")
parsed = parse_args(args)
dt_type = parsed[0]
tag = parsed[1]
modes = parsed[2]
label = parsed[3]
if modes == ['double_all_d0s']:
mode = modes[0]
if opts.set and opts.set == 'interact':
double_all_d0s(dt_type, mode, label, test=opts.test)
else:
pyline = 'import fit; fit.%s("%s", "%s", "%s", test=%s)'% (
mode, dt_type, mode, label, opts.test)
bash_file_name = 'fit-mbc.sh'
bash_file = create_bash_file_pyline(opts, label, dt_type, pyline,
bash_file_name)
logfile = tools.set_file('txt', dt_type, mode, tag,
prefix='dir_'+label, extbase=attr.logpath)
qjobname = 'fitd0s'
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
return
if modes == ['double_all_dps']:
mode = modes[0]
if opts.set and opts.set == 'interact':
double_all_dps(dt_type, mode, label, test=opts.test)
else:
pyline = 'import fit; fit.%s("%s", "%s", "%s", test=%s)'% (
mode, dt_type, mode, label, opts.test)
bash_file_name = 'fit-mbc.sh'
bash_file = create_bash_file_pyline(opts, label, dt_type, pyline,
bash_file_name)
logfile = tools.set_file('txt', dt_type, mode, tag,
prefix='dir_'+label, extbase=attr.logpath)
qjobname = 'fitdps'
tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
return
for mode in modes:
if opts.set and opts.set == 'interact':
if tag == 'single':
single_tag_mode(dt_type, mode, label, interact=True,
test=opts.test)
elif tag == 'double':
double_tag_mode(dt_type, mode, label, interact=True,
test=opts.test)
else:
raise ValueError(tag)
continue
else:
submit_batch_job(dt_type, tag, mode, label, opts)