def main(opts, args): if args[0] == 'peak': process_peak(opts, args[1:]) return parsed = yld.parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] for mode in modes: if opts.set and opts.set == 'interact': if 'generic' in datatype and tag == 'single': process_generic_single_mode(datatype, mode, label, opts.test) else: raise ValueError(tag) continue script, logfile, qjobname = create_script_logfile_jobname( datatype, tag, mode, label, opts.test) bash_file = yld.create_bash_file(script, 'get_backgrounds.sh') tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(opts, dt_type, stage, label, mode, sign): script = create_python_script(opts, dt_type, stage, label, mode, sign) bash_file = create_bash_file(opts, label, dt_type, script, 'fit-crossfeeds-%s.sh' %stage, subdir='csf') logfile = tools.set_file(extbase=attr.logpath, prefix='dir_'+label+'/crossfeeds', comname='stage_%s_%s_%s.txt' %(stage, mode, sign)) qjobname = '%s%s%s' % (stage, mode, sign) tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job_allinone(datatype, modes, label, test): script = create_python_script_allinone(datatype, modes, label, test) dt = datatype.replace('/', '-') bash_file_name = 'fit-backgrounds-%s-allinone.sh' % dt bash_file = create_bash_file(script, bash_file_name, 'setpyroot l') logname = '%s_allinone.txt' % datatype.replace('/', '_') logfile = tools.set_file(extbase=attr.logpath, prefix='dir_'+label+'/backgrounds', comname=logname) qn = datatype.split('/')[-1] qjobname = 'bkg%s' % qn tools.qsub_jobs(logfile, qjobname, bash_file, test)
def submit_batch_job_allinone(datatype, modes, label, test): script = create_python_script_allinone(datatype, modes, label, test) dt = datatype.replace('/', '-') bash_file_name = 'fit-backgrounds-%s-allinone.sh' % dt bash_file = create_bash_file(script, bash_file_name, 'setpyroot l') logname = '%s_allinone.txt' % datatype.replace('/', '_') logfile = tools.set_file(extbase=attr.logpath, prefix='dir_' + label + '/backgrounds', comname=logname) qn = datatype.split('/')[-1] qjobname = 'bkg%s' % qn tools.qsub_jobs(logfile, qjobname, bash_file, 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 = '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_nondiag(opts, dt_type, stage, label, x, y): script = create_python_script_nondiag(opts, dt_type, stage, label, x, y) bash_file = create_bash_file(opts, label, dt_type, script, 'fit-crossfeeds-%s.sh' %stage, subdir='csf') xstr = pair_to_str(x) ystr = pair_to_str(y) logfile = tools.set_file(extbase=attr.logpath, prefix='dir_'+label+'/crossfeeds', comname='stage_%s_%s_%s.txt' %(stage, xstr, ystr)) qjobname = 'x%s%s' % (xstr, ystr) 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, mode, label, lowmass, highmass, opts): script = create_python_script(opts, datatype, mode, label, lowmass, highmass) mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) bash_file_name = 'fit-%s.sh' % ms bash_file = create_bash_file(opts, label, datatype, script, bash_file_name, subdir='kkmass2/%s_%s' % (lowmass, highmass)) logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode) prefix='dir_%s/kkmass2/%s_%s' % (label, lowmass, highmass) logfile = tools.set_file(extbase=attr.logpath, prefix=prefix, comname=logname) qjobname = 'kkm%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 process_peak(opts, args): parsed = yld.parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] for mode in modes: if opts.set and opts.set == 'interact': process_peak_generic_single_mode(datatype, mode, label, opts.test) continue script, logfile, qjobname = create_peak_script_logfile_jobname( datatype, tag, mode, label, opts.test) bash_file = yld.create_bash_file(script, 'get_backgrounds_peak.sh') tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, mode, label, opts): script = create_python_script(opts, datatype, mode, label) dt = datatype.replace('/', '-') mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) bash_file_name = 'fit-sidebands-%s-%s.sh' % (dt, ms) #bash_file = create_bash_file(script, bash_file_name, 'setpyroot l') bash_file = create_bash_file(opts, label, datatype, script, bash_file_name) logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode) logfile = tools.set_file(extbase=attr.logpath, prefix='dir_'+label, comname=logname) qjobname = 'sdb%s' % ms tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def submit_batch_job(datatype, mode, label, opts): mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) script_name = 'kkmass-%s.py' % ms script = create_python_script(opts, script_name, datatype, mode, label) bash_file_name = 'kkmass-%s.sh' % ms bash_file = create_bash_file(opts, label, datatype, script, bash_file_name, subdir='sel') logname = '%s_%s.txt' %(datatype.replace('/', '_'), mode) logpath = os.path.join(attr.logpath, label, 'sel', 'kkmass') 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, opts): script = create_python_script(opts, datatype, mode, label) dt = datatype.replace('/', '-') mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) bash_file_name = 'fit-sidebands-%s-%s.sh' % (dt, ms) #bash_file = create_bash_file(script, bash_file_name, 'setpyroot l') bash_file = create_bash_file(opts, label, datatype, script, bash_file_name) logname = '%s_%s.txt' % (datatype.replace('/', '_'), mode) logfile = tools.set_file(extbase=attr.logpath, prefix='dir_' + label, comname=logname) qjobname = 'sdb%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(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, 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, mode, label, opts): mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) script_name = 'kkmass-%s.py' % ms script = create_python_script(opts, script_name, datatype, mode, label) bash_file_name = 'kkmass-%s.sh' % ms bash_file = create_bash_file(opts, label, datatype, script, bash_file_name, subdir='sel') logname = '%s_%s.txt' % (datatype.replace('/', '_'), mode) logpath = os.path.join(attr.logpath, label, 'sel', 'kkmass') logfile = tools.set_file(extbase=logpath, comname=logname) qjobname = 'sel%s' % ms tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def main(opts, args): parsed = parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] for mode in modes: if opts.set and opts.set == 'interact': if tag == 'single': single_tag_mode(datatype, mode, label, opts.test) else: raise ValueError(tag) continue script, logfile, qjobname = create_script_logfile_jobname( datatype, tag, mode, label, opts) bash_file = create_bash_file(opts, datatype, label, script, 'kssideband.sh', 'sel') tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def main(opts, args): parsed = parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] for mode in modes: if opts.set and opts.set == 'interact': if tag == 'single': single_tag_mode(datatype, mode, label, opts.test) else: raise ValueError(tag) continue script, logfile, qjobname = create_script_logfile_jobname( opts, datatype, tag, mode, label) bash_file = create_bash_file(opts, datatype, label, script, 'trkmtm.sh', 'sel') tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def ntuple_generic(opts, args): grade = args[0] skim = args[1] label = args[2] datasets = [] if 'data' in label: datasets.append(label) elif '281ipb' in label: numbers = ['31', '32', '33', '35', '36', '37'] for num in numbers: dataset = 'data' + num datasets.append(dataset) elif '537ipb' in label: numbers = ['43', '44', '45', '46'] for num in numbers: dataset = 'data' + num datasets.append(dataset) else: raise NameError(label) for dataset in datasets: if not opts.set or 'jobs' not in opts.set: update_dataselection_tcl('generic', dataset, grade, skim) logfile, qjobname, bash_file = create_bash_dtuple_generic( dataset, grade, skim, label) qsub_jobs(logfile, qjobname, bash_file, _test) continue njobs = parse_opts_set(opts.set, 'njobs') update_dataselection_tcl('generic', dataset, grade, skim, njobs) for job in range(njobs): subdataset = '%s_%s_%s' % (dataset, njobs, job + 1) logfile, qjobname, bash_file = create_bash_dtuple_generic( subdataset, grade, skim, label) qsub_jobs(logfile, qjobname, bash_file, _test)
def submit_batch_job(datatype, mode, label, lowmass, highmass, opts): script = create_python_script(opts, datatype, mode, label, lowmass, highmass) mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) bash_file_name = 'fit-%s.sh' % ms bash_file = create_bash_file(opts, label, datatype, script, bash_file_name, subdir='kkmass2/%s_%s' % (lowmass, highmass)) logname = '%s_%s.txt' % (datatype.replace('/', '_'), mode) prefix = 'dir_%s/kkmass2/%s_%s' % (label, lowmass, highmass) logfile = tools.set_file(extbase=attr.logpath, prefix=prefix, comname=logname) qjobname = 'kkm%s' % ms tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def entry(opts, args): parsed = yld.parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] for mode in modes: if opts.set and opts.set == 'interact': if datatype == 'signal' and tag == 'single': process_signal_single_mode(mode, label, opts.test) else: raise ValueError(tag) continue script, logfile, qjobname = create_script_logfile_jobname( opts, datatype, tag, mode, label) bash_file = yld.create_bash_file( opts, datatype, label, script, 'get_crossfeeds.sh') tools.qsub_jobs(logfile, qjobname, bash_file, opts.test)
def main(opts, args): var = args[0] parsed = yld.parse_args(args[1:]) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] for mode in modes: if opts.set and opts.set == 'interact': if tag == 'single': single_tag_mode(var, datatype, mode, label, opts.test) else: raise ValueError(tag) continue script, logfile, qjobname = create_script_logfile_jobname( var, datatype, tag, mode, label, opts.test) bash_file = yld.create_bash_file(script, 'get_var.sh') 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)