def create_python_script(opts, dt_type, mode, label): content = '''#!/usr/bin/env python from fit import sidebands sidebands.fit_sidebands_single_mode("%s", "%s", "%s", %s) '''% (dt_type, mode, label, opts.test) mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) filename = 'sidebands-%s-%s-%s.py' % (dt_type, ms, label) filename = filename.replace('/', '-') file_ = os.path.join(attr.datpath, dt_type, label, 'src', filename) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename
def create_python_script(opts, dt_type, tag, mode, label): if tag not in ['single', 'double'] : raise NameError(tag) content = '''#!/usr/bin/env python import fit fit.%s_tag_mode("%s", %s, "%s") '''% (tag, dt_type, mode, label) if tag == 'single': filename = '%s-%s-m%s-%s.py' % (dt_type, tag, mode, label) else: filename = '%s-%s-m%s-%s-%s.py' % ( dt_type, tag, mode[0], mode[1], label) file_ = os.path.join(attr.datpath, dt_type, label, 'src', filename) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename
def create_python_script(opts, datatype, mode, label, lowmass, highmass): content = '''#!/usr/bin/env python from fit import kkmass2 kkmass2.fit_single_mode("%s", "%s", "%s", %s, %s, %s) '''% (datatype, mode, label, lowmass, highmass, opts.test) mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) filename = 'fit-%s.py' % ms filename = filename.replace('/', '-') file_ = os.path.join(attr.datpath, datatype, label, 'src', 'kkmass2/%s_%s' % (lowmass, highmass) , filename) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename
def create_script_logfile_jobname(var, datatype, tag, mode, label, test): logname = '%s_%s.log' %(datatype, mode) logpath = os.path.join(attr.yldlogpath, label, 'var', var) logfile = tools.set_file(extbase=logpath, comname=logname) content = '''#!/usr/bin/env python from yld import var var.single_tag_mode("%s", "%s", "%s", "%s", test=%s) '''% (var, datatype, mode, label, test) mode, sign = tools.get_modekey_sign(mode) if tag == 'single': if sign == 1: sign = 'p' else: sign = 'm' filename = 'var-%s-%s-m%s-%s-%s.py' % (datatype, tag, mode, sign, label) qjobname = 'var%s,%s' % (mode, sign) else: filename = 'var-%s-%s-m%s-%s-%s.py' % (datatype, tag, mode[0], mode[1], label) qjobname = 'var%s,%s' % (mode[0], mode[1]) file_ = os.path.join(attr.base, 'src', attr.src, 'yld', filename) f = UserFile() f.data.append(content) f.output(file_) os.chmod(file_, 0755) return filename, logfile, qjobname
def get_data_statonly_crosssectionsdef(prefix): filename = 'data_statonly_crosssectionsdef' file_ = tools.set_file(extbase=attr.brfpath, prefix=prefix, comname=filename) if '281ipb' in prefix: lumi = '281.50' elif '537ipb' in prefix: lumi = '536.57' elif '818ipb' in prefix or 'v13' in prefix: lumi = '818.10' else: raise NameError(prefix) content = '''0 4 %s 0. 0 n 0 n n n ''' % lumi f = UserFile() f.append(content) f.output(file_, verbose=1)
def main(opts, args): parsed = parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] figname = 'kkmass_'+'_'.join(args[:-1]).replace('/', '_') org = UserFile() org.append(attr.fig_web_header) binbase = parse_opts_set(opts.set, 'binbase') binwidth = parse_opts_set(opts.set, 'binwidth') numbins = parse_opts_set(opts.set, 'numbins') for i in xrange(numbins): lowmass = binbase+i*binwidth highmass = binbase+(i+1)*binwidth for mode in modes: msg = create_fig_mode(opts, tag, datatype, mode, lowmass, highmass, label) org.append(msg) org.append(attr.fig_web_footer) orgname = '%s.org' % figname orgfile = os.path.join(attr.figpath, label, orgname) org.output(orgfile, verbose=1) orglink = '[[./fig/%s/%s][figure]]' %(label, orgname) sys.stdout.write('\n%s\n\n' % orglink) tools.org_export_as_html(orgfile)
def cleog_mode(opts, mode, label, tasks=[1]): sys.stdout.write('dhad.gen: Mode %s \n' %mode ) sys.stdout.write('dhad.gen: Label %s \n' %label) sys.stdout.write('dhad.gen: Task %s \n' %tasks) check_decfile(opts, mode, label) check_and_copy_file(opts, label, 'genmc.tcl') check_and_copy_file(opts, label, 'runlist') check_and_copy_file(opts, label, 'cleog-generic-array.sh') jobtype = 'cleog' seeds = tools.parse_opts_set(opts.set, 'seeds') edit_fix_single_mode_list(opts, mode, label) create_tag_numbers_file(opts, mode, label) for task_id in tasks: if opts.set and 'interact' in opts.set: bash_file, logfile = create_bash_interact( jobtype, mode, task_id, label=label, seeds=seeds) if opts.test: sys.stdout.write(bash_file + '\n') else: subprocess.Popen(bash_file) continue if opts.set and 'bg' in opts.set: bash_file, logfile = create_bash_interact( jobtype, mode, task_id, label=label, seeds=seeds) if opts.test: sys.stdout.write(bash_file + '\n') sys.stdout.write(logfile + '\n') else: sys.stdout.write('Save log as %s.\n' %logfile) process = subprocess.Popen(bash_file, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) f = UserFile() f.append(process.communicate()[0]) f.output(logfile) continue bash_file = create_bash_fix_mc_job(jobtype, label, task_id, seeds=seeds) if opts.test: sys.stdout.write(bash_file + '\n') continue output = commands.getoutput(bash_file) sys.stdout.write(output+ '\n')
def create_tag_numbers_file(opts, mode, label): tag_numbers = tools.parse_opts_set(opts.set, 'tag_numbers') if tag_numbers == None: tag_numbers = attr.get_tag_numbers(mode) tag_numbers_file = os.path.join(attr.datpath, 'signal', label, 'src', 'tag_numbers', mode) f = UserFile() f.data.append(str(tag_numbers)) verbose = opts.verbose if opts.test: verbose = 1 f.output(tag_numbers_file, verbose=verbose)
def create_python_script_allinone(dt_type, modes, label, test): content = '''#!/usr/bin/env python from fit import backgrounds backgrounds.fit_backgrounds_single("%s", %s, "%s", %s) '''% (dt_type, modes, label, test) filename = 'backgrounds-%s-allinone-%s.py' % (dt_type, label) filename = filename.replace('/', '-') file_ = os.path.join(attr.srcfitpath, filename) f = UserFile() f.append(content) f.output(file_) os.chmod(file_, 0755) return filename
def get_data_crosssectionsdef(prefix): filename = 'data_crosssectionsdef' file_ = tools.set_file(extbase=attr.brfpath, prefix=prefix, comname=filename) if '281ipb' in prefix: lumi = '281.50' lumi_err = '2.8150' elif '537ipb' in prefix: lumi = '536.57' lumi_err = '5.3657' elif '818ipb' in prefix or 'v13' in prefix: lumi = '818.10' lumi_err = '8.1810' else: raise NameError(prefix) bffilename = 'bf_stat' bffile = tools.set_file(extbase=attr.brfpath, prefix=prefix, comname=bffilename) brf = BrfFile(bffile) ne_xsec_err = brf.parsed['sigma(D0D0bar)'].split('+-')[-1].strip() ch_xsec_err = brf.parsed['sigma(D+D-)'].split('+-')[-1].strip() xsec_err = brf.parsed['sigma(DDbar)'].split('+-')[-1].strip() xsec_ratio_err = brf.parsed['chg/neu'].split('+-')[-1].strip() content = '''0 4 %s %s 0 y %s 0 y %s y %s y %s ''' % (lumi, lumi_err, ne_xsec_err, ch_xsec_err, xsec_err, xsec_ratio_err) f = UserFile() f.append(content) f.output(file_, verbose=1)
def create_script_logfile_jobname(datatype, tag, mode, label, opts): logname = '%s_%s.log' %(datatype, mode) logpath = os.path.join(attr.sellogpath, label, 'kssideband') logfile = tools.set_file(extbase=logpath, comname=logname) content = '''#!/usr/bin/env python from sel import kssideband kssideband.single_tag_mode("%s", "%s", "%s", test=%s) '''% (datatype, mode, label, opts.test) mode, sign = tools.get_modekey_sign(mode) if tag == 'single': if sign == [-1, 1]: filename = 'sel-%s-%s-m%s-%s.py' % (datatype, tag, mode, label) qjobname = 'sel%s' % mode else: if sign == 1: sign = 'p' else: sign = 'm' filename = 'sel-%s-%s-m%s-%s-%s.py' % ( datatype, tag, mode, sign, label) qjobname = 'sel%s,%s' % (mode, sign) else: filename = 'sel-%s-%s-m%s-%s-%s.py' % ( datatype, tag, mode[0], mode[1], label) qjobname = 'sel%s,%s' % (mode[0], mode[1]) file_ = os.path.join(attr.datpath, datatype, label, 'src', 'sel', filename) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.data.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename, logfile, qjobname
def pass2_mode(opts, mode, label, tasks=[1]): sys.stdout.write('dhad.gen.pass2: Mode %s \n' %mode ) sys.stdout.write('dhad.gen.pass2: Label %s \n' %label) sys.stdout.write('dhad.gen.pass2: Task %s \n' %tasks) jobtype = 'pass2' cleog.edit_fix_single_mode_list(opts, mode, label) cleog.check_and_copy_file(opts, label, 'mcp2.tcl') cleog.check_and_copy_file(opts, label, 'pass2-generic-array.sh') for task_id in tasks: if opts.set == 'interact': bash_file, logfile = cleog.create_bash_interact( jobtype, mode, task_id, label=label) if opts.test: sys.stdout.write(bash_file + '\n') else: subprocess.Popen(bash_file) return if opts.set == 'bg': bash_file, logfile = cleog.create_bash_interact( jobtype, mode, task_id, label=label) if opts.test: sys.stdout.write(bash_file + '\n') sys.stdout.write(logfile + '\n') else: sys.stdout.write('Save log as %s.\n' %logfile) process = subprocess.Popen(bash_file, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) f = UserFile() f.append(process.communicate()[0]) f.output(logfile) return bash_file = cleog.create_bash_fix_mc_job(jobtype, label, task_id) if opts.test: sys.stdout.write(bash_file + '\n') continue output = commands.getoutput(bash_file) sys.stdout.write(output+ '\n')
def create_python_script(opts, filename, datatype, mode, label): content = '''#!/usr/bin/env python from sel import kkmass kkmass.single_tag_mode("%s", "%s", "%s", test=%s) '''% (datatype, mode, label, opts.test) file_ = os.path.join(attr.datpath, datatype, label, 'src', 'sel', filename) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.data.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename
def create_peak_script_logfile_jobname(datatype, tag, mode, label, test): prefix = 'dir_'+label logname = 'peak_bkg_%s_%s.log' %(datatype, mode) logname = logname.replace('/', '_') logfile = tools.set_file( extbase=attr.yldlogpath, prefix=prefix, comname=logname) content = '''#!/usr/bin/env python from yld import backgrounds backgrounds.process_peak_generic_single_mode("%s", "%s", "%s", test=%s) '''% (datatype, mode, label, test) mode, sign = tools.get_modekey_sign(mode) if tag == 'single': if sign == 1: sign = 'p' else: sign = 'm' filename = 'peak-bkg-%s-%s-m%s-%s-%s.py' % ( datatype, tag, mode, sign, label) qjobname = 'bkg%s%s' % (mode, sign) else: filename = 'peak-bkg-%s-%s-m%s-%s-%s.py' % ( datatype, tag, mode[0], mode[1], label) qjobname = 'bkg%s,%s' % (mode[0], mode[1]) file_ = os.path.join(attr.base, 'src', attr.src, 'yld', filename) f = UserFile() f.data.append(content) f.output(file_) os.chmod(file_, 0755) return filename, logfile, qjobname
def create_script_logfile_jobname(opts, datatype, tag, mode, label): prefix = 'dir_'+label logname = 'bkg_%s_%s.log' %(datatype, mode) logfile = tools.set_file( extbase=attr.yldlogpath, prefix=prefix, comname=logname) content = '''#!/usr/bin/env python from yld import crossfeeds crossfeeds.process_signal_single_mode("%s", "%s", test=%s) '''% (mode, label, opts.test) mode, sign = tools.get_modekey_sign(mode) if tag == 'single': if sign == 1: sign = 'p' else: sign = 'm' filename = 'csf-%s-%s-m%s-%s.py' % (datatype, tag, mode, sign) qjobname = 'csf%s,%s' % (mode, sign) else: filename = 'csf-%s-%s-m%s-%s.py' % (datatype, tag, mode[0], mode[1]) qjobname = 'csf%s,%s' % (mode[0], mode[1]) file_ = os.path.join(attr.datpath, datatype, label, 'src', 'yld', filename) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.data.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename, logfile, qjobname
def create_bash_file(opts, datatype, label, script, bashname='get-yields.sh', subdir='yld', optcommand=''): script_dir = os.path.join( attr.datpath, datatype, label, 'src', subdir) bash_content = '''#!/bin/sh #$ -S /usr/local/bin/bash #$ -j y #$ -m a #$ -M [email protected] date hostname . ~/.bashrc setdhad %s %s cd %s ./%s date ''' % (attr.version, optcommand, script_dir, script) bash_file = os.path.join(script_dir, bashname) verbose = opts.verbose if opts.test: verbose = 1 f = UserFile() f.append(bash_content) f.output(bash_file, verbose=verbose) os.chmod(bash_file, 0755) return bash_file
def create_sh(filename, randomseeds=''): code_dir = os.path.join(attr.base, 'src', attr.src, 'gen') sh_file = tools.set_file(extbase=code_dir, comname=filename) sh_content = '''#!/bin/sh # request Bourne shell as shell for job #$ -S /usr/local/bin/bash #$ -j y #$ -m a #$ -M [email protected] date hostname . /nfs/cleo3/Offline/scripts/cleo3logins . /nfs/cleo3/Offline/scripts/cleo3defs c3rel $CGRELEASE echo $SHELL cd $SCRIPTDIR %s export BATCH=$SGE_TASK_ID export NUMEVT=`cat $SCRIPTDIR/tag_numbers/$1` export RUNNUMBER=`nl -s " " $SCRIPTDIR/runlist | grep " $SGE_TASK_ID " | awk '{ print $2 }'` export DECAYTITLE=$1 export UDECAY=$SCRIPTDIR/tag_decfiles/$1.dec export OUTDIR=$TOPDIR/cleog$CGSUFFIX suez -f genmc.tcl date ''' % (randomseeds) f = UserFile() f.append(sh_content) f.output(sh_file)
def create_python_script(dt_type, mode, label, test): content = '''#!/usr/bin/env python from fit import backgrounds backgrounds.fit_backgrounds_single_mode("%s", "%s", "%s", %s) '''% (dt_type, mode, label, test) mode_sign = tools.get_modekey_sign(mode) ms = tools.pair_to_str(mode_sign) filename = 'backgrounds-%s-%s-%s.py' % (dt_type, ms, label) filename = filename.replace('/', '-') file_ = os.path.join(attr.srcfitpath, filename) f = UserFile() f.append(content) f.output(file_) os.chmod(file_, 0755) return filename
def create_script_logfile_jobname(opts, datatype, tag, mode, label, test): logfile = tools.set_file('log', datatype, mode, tag, prefix='dir_'+label, extbase=attr.yldlogpath) content = '''#!/usr/bin/env python import yld yld.%s_tag_mode("%s", "%s", "%s", test=%s) '''% (tag, datatype, mode, label, test) mode, sign = tools.get_modekey_sign(mode) if tag == 'single': if sign == 1: sign = 'p' else: sign = 'm' filename = '%s-%s-m%s-%s.py' % (datatype, tag, mode, sign) qjobname = 'yld%s,%s' % (mode, sign) else: filename = '%s-%s-m%s-%s.py' % (datatype, tag, mode[0], mode[1]) qjobname = 'yld%s,%s' % (mode[0], mode[1]) file_ = os.path.join(attr.datpath, datatype, label, 'src', 'yld', filename) verbose = opts.verbose if opts.test: verbose = 1 sys.stdout.write('logfile: %s\n' %logfile) f = UserFile() f.data.append(content) f.output(file_, verbose=verbose) os.chmod(file_, 0755) return filename, logfile, qjobname
def check_decfile_single(opts, label, toppath, decfile, headername, modename): headerfile = os.path.join(toppath, 'tag_fragments', headername) fragfile = os.path.join(toppath, 'tag_fragments', modename) try: frag = open(fragfile, 'r') except IOError: tag_fragments(opts, [label]) frag = open(fragfile, 'r') f = UserFile() fin = open(headerfile, 'r') f.extend(fin.readlines()) f.extend(frag.readlines()) f.append('End\n') verbose = opts.verbose if opts.test: verbose = 1 f.output(decfile, verbose=verbose) fin.close() frag.close()
def main(opts, args): dt_type = args[0] stage = args[1] label = args[2]+'/crossfeeds' single_modes = get_single_modes(opts.set) org = UserFile() org.append(attr.fig_web_header) if stage == 'diag': for mode, sign in single_modes: msg = create_fig_diag_mode(dt_type, stage, label, mode, sign) org.append(msg) elif stage == 'nondiag': for x, y in [(x, y) for x in single_modes for y in single_modes if x!=y]: sys.stdout.write('Creating mode %s, %s ... ' % (x, y)) sys.stdout.flush() msg = create_fig_nondiag_mode(dt_type, stage, label, x, y) org.append(msg) sys.stdout.write('OK.\n') else: raise NameError(stage) figname = 'crossfeeds_' + '_'.join(args).replace('/', '_') org.append(attr.fig_web_footer) figlabel = label.split('/')[0] figname = figname.replace(figlabel, '') figname = figname.replace('__', '_') orgname = figname+'.org' orgname = orgname.replace('_.org', '.org') orgpath = os.path.join(attr.figpath, figlabel) orgfile = tools.check_and_join(orgpath, orgname) verbose = opts.verbose if opts.test: verbose = 1 org.output(orgfile, verbose=verbose) orglink = '[[./fig/%s/%s][figure]]' %(figlabel, orgname) sys.stdout.write('\n%s\n\n' % orglink) if opts.test: return tools.org_export_as_html(orgfile)
def create_bash_fix_mc_job(jobtype, label, task_id='1', seeds=None, verbose=0): CGRELEASE= attr.cleog P2RELEASE= attr.pass2 DSKIMRELEASE = attr.dskim CGSUFFIX = '_'+ CGRELEASE SUFFIX = CGSUFFIX + '_' + P2RELEASE TOPDIR = os.path.join(attr.base, 'dat/signal', label) SCRIPTDIR = os.path.join(TOPDIR, 'src') DODECAYTREE='0' DOCLEOG='0' DOPASS2='0' DODSKIM='0' DONTUPLE='0' if jobtype == 'decaytree': DODECAYTREE='1' elif jobtype == 'cleog': DOCLEOG='1' if seeds != None: randomseeds = 'export RANDOMSEEDS=%s' % seeds create_sh('cleog-generic-array.sh', randomseeds) elif jobtype == 'pass2': DOPASS2='1' elif jobtype == 'dskim': DODSKIM='1' elif jobtype == 'ntuple': DONTUPLE='1' else: sys.stdout.write('\nUnknow jobtype: %s\n' % jobtype) sys.exit() bash_content = '''#!/usr/local/bin/bash export CGRELEASE=%s export P2RELEASE=%s export DSKIMRELEASE=%s export CGSUFFIX=%s export SUFFIX=%s export TOPDIR=%s export SCRIPTDIR=%s LOGDIR=$TOPDIR/log$SUFFIX DODECAYTREE=%s DOCLEOG=%s DOPASS2=%s DODSKIM=%s DONTUPLE=%s FIXTASKID=%s FILELISTINGS="fix_single_mode_list" if [ ! -e $TOPDIR ] ; then mkdir -p $TOPDIR ; fi if [ ! -e $LOGDIR ] ; then mkdir -p $LOGDIR ; fi cd $SCRIPTDIR for STRING in `cat $FILELISTINGS` ; do if [[ $DODECAYTREE == 1 ]] ; then if [ ! -e $TOPDIR/decaytree$CGSUFFIX ] ; then mkdir $TOPDIR/decaytree$CGSUFFIX ; fi qsub -l arch=lx24-x86 -o $LOGDIR/decaytree-$STRING-\$TASK_ID.txt \ -N decaytree-$STRING -t $FIXTASKID \ -v CGSUFFIX,TOPDIR,SCRIPTDIR,CGRELEASE \ decaytree-generic-array.sh $STRING fi if [[ $DOCLEOG == 1 ]] ; then if [ ! -e $TOPDIR/cleog$CGSUFFIX ] ; then mkdir $TOPDIR/cleog$CGSUFFIX ; fi qsub -l arch=lx24-x86 -o $LOGDIR/cleog-$STRING-\$TASK_ID.txt \ -N cleog-$STRING -t $FIXTASKID \ -v CGSUFFIX,TOPDIR,SCRIPTDIR,CGRELEASE \ cleog-generic-array.sh $STRING fi if [[ $DOPASS2 == 1 ]] ; then if [ ! -e $TOPDIR/pass2$SUFFIX ] ; then mkdir $TOPDIR/pass2$SUFFIX ; fi qsub -l arch=lx24-x86 -o $LOGDIR/pass2-$STRING-\$TASK_ID.txt \ -N pass2-$STRING -hold_jid cleog-$STRING \ -t $FIXTASKID \ -v SUFFIX,CGSUFFIX,TOPDIR,SCRIPTDIR,P2RELEASE \ pass2-generic-array.sh $STRING fi if [[ $DODSKIM == 1 ]] ; then if [ ! -e $TOPDIR/dskim$SUFFIX ] ; then mkdir $TOPDIR/dskim$SUFFIX ; fi qsub -l arch=lx24-x86 -o $LOGDIR/dskim-$STRING-\$TASK_ID.txt \ -N dskim-$STRING -hold_jid pass2-$STRING \ -t $FIXTASKID \ -v SUFFIX,CGSUFFIX,TOPDIR,SCRIPTDIR,DSKIMRELEASE \ dskim-generic-array.sh $STRING fi if [[ $DONTUPLE == 1 ]] ; then if [ ! -e $SCRIPTDIR/tmp ] ; then mkdir -p $SCRIPTDIR/tmp; fi if [ ! -e $TOPDIR/dtuple$SUFFIX ] ; then mkdir $TOPDIR/dtuple$SUFFIX ; fi qsub -l arch=lx24-x86 -o $LOGDIR/dtuple-$STRING.txt \ -N dtuple-$STRING -hold_jid dskim-$STRING -v SUFFIX,TOPDIR,SCRIPTDIR \ dtuple-generic-array.sh $STRING fi done ''' % (CGRELEASE, P2RELEASE, DSKIMRELEASE, CGSUFFIX, SUFFIX, TOPDIR , SCRIPTDIR, DODECAYTREE, DOCLEOG, DOPASS2 , DODSKIM, DONTUPLE, task_id) bash_file = os.path.join(SCRIPTDIR, 'fix_mc_job.sh') f = UserFile() f.data.append(bash_content) f.output(bash_file, verbose) os.chmod(bash_file, 0755) return bash_file
def create_bash_interact(jobtype, mode, task_id=1, label='', seeds=None, verbose=0): bashname = '%s_task_%s.sh' %(jobtype, task_id) CGRELEASE= attr.cleog P2RELEASE= attr.pass2 DSKIMRELEASE = attr.dskim CGSUFFIX = '_'+ CGRELEASE SUFFIX = CGSUFFIX + '_' + P2RELEASE TOPDIR = os.path.join(attr.base, 'dat/signal', label) SCRIPTDIR = os.path.join(TOPDIR, 'src') logname = '%s-%s-%s.txt' %(jobtype, mode, task_id) logpath = os.path.join(TOPDIR, 'log%s' %SUFFIX) logfile = tools.set_file(extbase=logpath, comname=logname) core_content = attr.gen_core_content[jobtype] if jobtype == 'cleog' and seeds != None: randomseeds = 'export RANDOMSEEDS=%s' % seeds core_content = randomseeds + core_content bash_content = '''#!/usr/local/bin/bash date hostname . /nfs/cleo3/Offline/scripts/cleo3logins . /nfs/cleo3/Offline/scripts/cleo3defs export CGRELEASE=%s export P2RELEASE=%s export DSKIMRELEASE=%s export CGSUFFIX=%s export SUFFIX=%s export TOPDIR=%s export SCRIPTDIR=%s LOGDIR=$TOPDIR/log$SUFFIX STRING="%s" SGE_TASK_ID=%s if [ ! -e $TOPDIR ] ; then mkdir -p $TOPDIR ; fi if [ ! -e $LOGDIR ] ; then mkdir -p $LOGDIR ; fi cd $SCRIPTDIR export BATCH=$SGE_TASK_ID export DECAYTITLE=$STRING export UDECAY=$SCRIPTDIR/tag_decfiles/$STRING.dec %s # print date and time again date ''' % (CGRELEASE, P2RELEASE, DSKIMRELEASE, CGSUFFIX, SUFFIX, TOPDIR, SCRIPTDIR, mode, task_id, core_content) bash_file = os.path.join(SCRIPTDIR, bashname) f = UserFile() f.data.append(bash_content) f.output(bash_file, verbose) os.chmod(bash_file, 0755) return bash_file, logfile
def create_brsfile_three(args): comname = '_'.join(args) comname = comname.replace('.', '_') brsfile = set_file(extbase=attr.srcmnfpath, comname=comname, ext='mnf') brs = UserFile() header = r''' set def del 0 dep nmodes = 9 ''' brs.append(header) label_0, content_0 = get_brs(args[0]) label_1, content_1 = get_brs(args[2]) label_2, content_2 = get_brs(args[1]) brs.extend(content_0) brs.extend(content_1) hmin = 0.4 hmax = 1.6 xleft = 0.45 textsize = 0.45 if args[0] == 'pdg2010': hmin = 0.85 hmax = 1.10 xleft = 0.86 trunk = r''' exec colors dep yoffset = -0.15 exec pdgCompareMany %s %s white mustard burgundy "%s" 4 "%s" 8 ! Add another set of data points rename 1 11 rename 2 12 ''' % (hmin, hmax, label_0, label_1) brs.append(trunk) brs.extend(content_0) brs.extend(content_2) footer = r''' dep yoffset = 0.15 exec pdgCompareMany %s %s white mustard burgundy "%s" 3 "%s" 6.5 over 11 -60/lavender over 12 -60/lavender key 1 new 1000 %s 13 17.5 0.4 0 left cm -2000 lavender 3 0.01 black draw line 1 black 1 plot %s, -3.5 %s, -3.5 ! xleft is the x position of the mode label in plot units dep xleft = %s dep textsize = %s exec pdgCompareComment 1 xleft textsize "'K^-![p]^+'" exec pdgCompareComment 2 xleft textsize "'K^-![p]^+![p]^0!'" exec pdgCompareComment 3 xleft textsize "'K^-![p]^+![p]^-![p]^+!'" exec pdgCompareComment 4 xleft textsize "'K^-![p]^+![p]^+!'" exec pdgCompareComment 5 xleft textsize "'K^-![p]^+![p]^+![p]^0!'" exec pdgCompareComment 6 xleft textsize "'K^0?S![p]^+!'" exec pdgCompareComment 7 xleft textsize "'K^0?S![p]^+![p]^0!'" exec pdgCompareComment 8 xleft textsize "'K^0?S![p]^+![p]^-![p]^+!'" exec pdgCompareComment 9 xleft textsize "'K^-!K^+![p]^+!'" hard epost ''' % (hmin, hmax, label_0, label_2, label_1, hmin, hmax, xleft, textsize) brs.append(footer) brs.output(brsfile) return brsfile
def output_headerfiles(label): headernames = ['Double_D0D0B_header', 'Double_DpDm_header', 'Single_D0_header', 'Single_D0B_header', 'Single_Dp_header', 'Single_Dm_header'] outpath = os.path.join(attr.datpath, 'signal', label, 'src', 'tag_fragments') for hname in headernames: headerfile = tools.check_and_join(outpath, hname) f = UserFile() if 'D0D0B' in hname: alias_line = '#\nAlias myD0 D0\n#\n#\nAlias myanti-D0 anti-D0\n#\n' psi_decay_line = '1.000 myD0 myanti-D0 VSS;\n' if 'DpDm' in hname: alias_line = '#\nAlias myD+ D+\n#\n#\nAlias myD- D-\n#\n' psi_decay_line = '1.000 myD+ myD- VSS;\n' elif 'Single_D0_' in hname: alias_line = '#\nAlias myD0 D0\n#\n' psi_decay_line = '1.000 myD0 anti-D0 VSS;\n' elif 'Single_D0B_' in hname: alias_line = '#\nAlias myanti-D0 anti-D0\n#\n' psi_decay_line = '1.000 D0 myanti-D0 VSS;\n' elif 'Single_Dp_' in hname: alias_line = '#\nAlias myD+ D+\n#\n' psi_decay_line = '1.000 myD+ D- VSS;\n' elif 'Single_Dm_' in hname: alias_line = '#\nAlias myD- D-\n#\n' psi_decay_line = '1.000 D+ myD- VSS;\n' f.append(alias_line) f.append('Decay vpho\n1.000 psi(3770) gamma VPHOTOVISR ;\nEnddecay\n#\nDecay psi(3770)\n') f.append(psi_decay_line) f.append('Enddecay\n#') f.output(headerfile, verbose=1)
def output_fakemodes(fakemodes, modefile): f = UserFile() f.data.append('Mode || Number \n') for k, v in fakemodes.items(): f.data.append('%s | %s \n' %(k, v)) f.output(modefile)
def yields_and_efficiencies(opts, args): label = args[0] prefix = 'dir_'+label yldfile = tools.set_file(extbase=attr.brfpath, prefix=prefix, comname='yields_and_efficiencies') f = UserFile() single_modes = [(pair, sign) for pair in modes for sign in [-1, 1]] input_label = label if '.' in label: input_label = label.split('.')[0] prefix='dir_'+input_label+'/crossfeeds' dt_type = 'signal' for x in single_modes: if x[1] == 1: uname = modes[x[0]]['uname'] fname = modes[x[0]]['fname'] else: uname = modes[x[0]]['unamebar'] fname = modes[x[0]]['fnamebar'] txtname = '%s_Single_%s_fakes_Single_%s.txt' % (dt_type, fname, fname) txtfile = tools.set_file(extbase=attr.fitpath, prefix=prefix, comname=txtname) tab = DHadTable(txtfile) entries = tab.cell_get('entries', 'Value') entries = tab.cell_trim(entries, rnd='1') yld = tab.cell_get('yield', 'Value') yld = tab.cell_trim(yld, rnd='.000001') ratio = tab.cell_get('ratio', 'Value') ratio = tab.cell_trim(ratio, rnd='.000001') f.data.append('%s: %s %s %s\n' % (fname, entries, yld, ratio)) for x, y in [(x, y) for x in single_modes for y in single_modes if x!=y]: if x[1] == 1: uname = modes[x[0]]['uname'] fname = modes[x[0]]['fname'] else: uname = modes[x[0]]['unamebar'] fname = modes[x[0]]['fnamebar'] if y[1] == 1: unameb = modes[y[0]]['uname'] fnameb = modes[y[0]]['fname'] else: unameb = modes[y[0]]['unamebar'] fnameb = modes[y[0]]['fnamebar'] txtname = '%s_Single_%s_fakes_Single_%s.txt' % (dt_type, fname, fnameb) txtfile = tools.set_file(extbase=attr.fitpath, prefix=prefix, comname=txtname) tab = DHadTable(txtfile) entries = tab.cell_get('entries', 'Value') entries = tab.cell_trim(entries, rnd='1') yld = tab.cell_get('yld', 'Value') ylderr = tab.cell_get('yld', 'Error') try: float(yld)/float(ylderr) except: sys.stdout.write('The original ylderr is %s, set to 1.0e10.\n' % ylderr) ylderr = 1.0e10 yldsigma = str(float(yld)/float(ylderr)) yldsigma = tab.cell_trim(yldsigma, rnd='.000001') yld = tab.cell_trim(yld, rnd='.000001') ratio = str(float(yld)/float(entries)) ratio = tab.cell_trim(ratio, rnd='.000001') ratioerr = str(float(ylderr)/float(entries)) ratioerr = tab.cell_trim(ratioerr, rnd='.000001') f.data.append('%s fakes %s: %s %s %s %s %s\n' % ( fname, fnameb, yldsigma, entries, yld, ratio, ratioerr)) f.output(yldfile, verbose=1) return yldfile
def main(opts, args): if args[0] == 'backgrounds': import backgrounds backgrounds.main(opts, args[1:]) return if args[0] == 'brs': import brs brs.main(opts, args[1:]) return if args[0] == 'compare': import compare compare.main(opts, args[1:]) return if args[0] == 'cbx': import cbx cbx.main(opts, args[1:]) return if args[0] == 'crossfeeds': import crossfeeds crossfeeds.main(opts, args[1:]) return if args[0] == 'evt': import evt evt.main(opts, args[1:]) return if args[0] == 'fun': import fun fun.main(opts, args[1:]) return if args[0] == 'kkmass': import kkmass kkmass.main(opts, args[1:]) return if args[0] == 'kpimass': import kpimass kpimass.main(opts, args[1:]) return if args[0] == 'trim': import trim trim.main(opts, args[1:]) return if args[0] == 'trkmtm': import trkmtm trkmtm.main(opts, args[1:]) return if args[0] == 'trkmtm1': import trkmtm1 trkmtm1.main(opts, args[1:]) return if args[0] == 'trkmtm2': import trkmtm2 trkmtm2.main(opts, args[1:]) return if args[0] == 'var': import var var.main(opts, args[1:]) return figname = '_'.join(args).replace('/', '_') parsed = parse_args(args) dt_type = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] sqrt = False if opts.set and opts.set == 'sqrt': sqrt = True linr = False if opts.set and opts.set == 'linr': linr = True org = UserFile() org.append(attr.fig_web_header) for mode in modes: if mode == 'double_all_d0s' or mode == 'double_all_dps' : modekey = mode else: modekey = get_modekey(mode) if sqrt: sqrt_fig_mode(tag, dt_type, modekey, label) sys.stdout.write('Creating %s ...' % mode) sys.stdout.flush() msg = create_fig_mode(tag, dt_type, modekey, label) sys.stdout.write(' OK.\n') org.append(msg) org.append(attr.fig_web_footer) figlabel = label.split('/')[0] figname = figname.replace(figlabel, '') figname = figname.replace('__', '_') orgname = figname+'.org' orgname = orgname.replace('_.org', '.org') orgpath = os.path.join(attr.figpath, figlabel) orgfile = tools.check_and_join(orgpath, orgname) verbose = opts.verbose if opts.test: verbose = 1 org.output(orgfile, verbose=verbose) orglink = '[[./fig/%s/%s][figure]]' %(figlabel, orgname) sys.stdout.write('\n%s\n\n' % orglink) if opts.test: return tools.org_export_as_html(orgfile)
def output_headerfiles(label): headernames = [ 'Double_D0D0B_header', 'Double_DpDm_header', 'Single_D0_header', 'Single_D0B_header', 'Single_Dp_header', 'Single_Dm_header' ] outpath = os.path.join(attr.datpath, 'signal', label, 'src', 'tag_fragments') for hname in headernames: headerfile = tools.check_and_join(outpath, hname) f = UserFile() if 'D0D0B' in hname: alias_line = '#\nAlias myD0 D0\n#\n#\nAlias myanti-D0 anti-D0\n#\n' psi_decay_line = '1.000 myD0 myanti-D0 VSS;\n' if 'DpDm' in hname: alias_line = '#\nAlias myD+ D+\n#\n#\nAlias myD- D-\n#\n' psi_decay_line = '1.000 myD+ myD- VSS;\n' elif 'Single_D0_' in hname: alias_line = '#\nAlias myD0 D0\n#\n' psi_decay_line = '1.000 myD0 anti-D0 VSS;\n' elif 'Single_D0B_' in hname: alias_line = '#\nAlias myanti-D0 anti-D0\n#\n' psi_decay_line = '1.000 D0 myanti-D0 VSS;\n' elif 'Single_Dp_' in hname: alias_line = '#\nAlias myD+ D+\n#\n' psi_decay_line = '1.000 myD+ D- VSS;\n' elif 'Single_Dm_' in hname: alias_line = '#\nAlias myD- D-\n#\n' psi_decay_line = '1.000 D+ myD- VSS;\n' f.append(alias_line) f.append( 'Decay vpho\n1.000 psi(3770) gamma VPHOTOVISR ;\nEnddecay\n#\nDecay psi(3770)\n' ) f.append(psi_decay_line) f.append('Enddecay\n#') f.output(headerfile, verbose=1)
def main(opts, args): parsed = parse_args(args) datatype = parsed[0] tag = parsed[1] modes = parsed[2] label = parsed[3] figname = '_'.join(args).replace('/', '_') org = UserFile() org.append(attr.fig_web_header) if 'allinone' in opts.set: msg = create_fig(datatype, label) org.append(msg) else: for mode in modes: msg = create_fig_mode(tag, datatype, mode, label) org.append(msg) org.append(attr.fig_web_footer) orgname = '%s.org' % figname orgfile = os.path.join(attr.figpath, orgname) org.output(orgfile) tools.org_export_as_html(orgfile) orglink = '[[./%s/fig/%s][figure]]' %(attr.analysis, orgname) sys.stdout.write('\n%s\n\n' % orglink)