def create_fig_mode(opts, tag, datatype, mode, lowmass, highmass, label):
modekey = tools.get_modekey(mode)
comname = '%s_%s' %(datatype.replace('/', '_'), mode)
prefix = 'dir_%s/kkmass/%s_%s' % (label, lowmass, highmass)
epsfile = tools.set_file('eps', datatype, modekey, tag,
prefix=prefix, extbase=attr.figpath)
txtfile = tools.set_file('txt', datatype, modekey, tag,
prefix=prefix, extbase=attr.fitpath)
head, figname = os.path.split(epsfile)
figname = figname.replace('.eps', '')
subdir = 'kkmass/%s_%s' % (lowmass, highmass)
pdflink = './%s/%s.pdf' %(subdir, figname)
pnglink = './%s/%s.png' %(subdir, figname)
loglink = '[[../../log/%s/%s/%s.txt][log]]' %(label, subdir, comname)
figlink = '[[%s][%s]]' %(pdflink, pnglink)
modename = tools.get_orgname_from_fname(mode)
secname = 'KK mass: %s - %s' % (lowmass, highmass)
tablink = ''
if os.access(txtfile, os.F_OK):
orgtabfile = txtfile.replace('.txt', '.org')
tab = DHadTable(txtfile)
tab.output_org(orgtabfile)
tablink = '#+INCLUDE: "%s"\n' % orgtabfile
msg = '\n* %s \n %s \n\n%s\n %s\n' % (
secname, figlink, tablink, loglink)
return msg
def ratios_signal(args):
if args[0] == '537ipb/281ipb' and args[1] == '537ipbv2/281ipb':
tabfilename1 = 'compare_yields_signal_divide_537ipb_9.03_regular12.txt'
title1 = args[0]
headname1 = title1
tabfilename2 = 'compare_yields_signal_divide_537ipbv2_9.03_regular12.txt'
title2 = args[1]
headname2 = title2
else:
raise NameError(args)
tabfile1 = os.path.join(attr.tabpath, tabfilename1)
tabfile2 = os.path.join(attr.tabpath, tabfilename2)
if opts.set and opts.set == 'info':
print 'here'
tab = DHadTable()
tab.column_append_from_tab_file('Mode', tabfile1, 'Mode')
tab.column_append_from_tab_file(title1, tabfile1, headname1)
tab.column_append_from_tab_file(title2, tabfile2, headname2)
tab.column_append_by_diff_sigma_pct('diff(%)', title2, title1)
tab.output(_tabname)
def parsefile(f, lowmass, highmass, indict):
x0 = lowmass
x1 = highmass
xc = (x0 + x1)/2.
tab = DHadTable(f)
N1 = float(tab.cell_get('N1', 'Value'))
N2 = float(tab.cell_get('N2', 'Value'))
if N1 > 1000 or N2 > 1000 :
return
indict['x'].append(xc)
indict['exl'].append(xc-x0)
indict['exh'].append(x1-xc)
N = N1 + N2
err_low_N1 = float(tab.cell_get('N1', 'Low'))
err_low_N2 = float(tab.cell_get('N2', 'Low'))
err_low = (err_low_N1**2 + err_low_N2**2)**.5
err_high_N1 = float(tab.cell_get('N1', 'High'))
err_high_N2 = float(tab.cell_get('N2', 'High'))
err_high = (err_high_N1**2 + err_high_N2**2)**.5
yc = float(N)
yl = float(err_low)
yh = float(err_high)
indict['y'].append(yc)
indict['eyl'].append(yl)
indict['eyh'].append(yh)
def md(opts, tabname, args):
para = 'md'
namestyle = 'fname'
rnd = '0.0000001'
tag = 's'
if opts.set:
for li in opts.set.split(';'):
name = li.split('=')[0]
value = li.split('=')[1]
sys.stdout.write('Set %s = %s \n' % (name, value))
if name == 'label':
label = value
if args[0] == '281ipb':
tab_A_name = args[1]
tab_B_name = args[2]
tab_C_name = args[3]
if label:
labels = label.split(',')
tab_A_name = labels[0]
tab_B_name = labels[1]
tab_C_name = labels[2]
tab = DHadTable()
tab.column_append_from_dict('Mode', namestyle)
tab.column_append_from_fit_files(tab_A_name, args[1], para, tag, rnd)
tab.column_append_from_fit_files(tab_B_name, args[2], para, tag, rnd)
tab.column_append_from_fit_files(tab_C_name, args[3], para, tag, rnd)
tab.output(_tabname)
else:
raise ValueError(args)
def get_generated_numbers_single(i, prefix=''):
if '281ipb' in prefix:
mult_fact = 10
elif '537ipb' in prefix:
mult_fact = 20
elif '818ipb' in prefix or 'v13' in prefix:
mult_fact = 30
else:
raise NameError(prefix)
mode = i[0]
sign = i[1]
tag_number = modes[mode]['tag_num_s']
count = tag_number * mult_fact
if mode == 205:
if sign == 1:
count = 29972 # 30000
if sign == -1:
count = 35969 # 36000
if 'noFSR' in prefix:
mode = i[0]
sign = i[1]
logfile = tools.set_file('log',
's',
mode,
's',
sign,
prefix,
extbase=evtlogbase)
tab = DHadTable(logfile)
FSR = tab.cell_get('FSR', 'Value')
count = count - int(FSR)
return count
def para_md(args):
para = 'md'
namestyle = 'fname'
rnd = '0.0000001'
tag = 's'
tab_A = args[0]
tab_B = args[1]
latel_A = tab_A
latel_B = tab_B
label = tools.parseopts_set(opts.set, 'label')
if label:
labels = label.split(',')
label_A = labels[0]
label_B = labels[1]
tab = DHadTable()
tab.column_append_from_dict('Mode', namestyle)
tab.column_append_from_fit_files(label_A, tab_A, para, tag, rnd)
tab.column_append_from_fit_files(label_B, tab_B, para, tag, rnd)
tab.column_append_by_diff('diff[MeV]', label_B,label_A, rnd=rnd, factor=1000)
tab.output(_tabname)
def parsefile(f, lowmass, highmass, indict):
x0 = lowmass
x1 = highmass
xc = (x0 + x1) / 2.
tab = DHadTable(f)
N1 = float(tab.cell_get('N1', 'Value'))
N2 = float(tab.cell_get('N2', 'Value'))
if N1 > 1000 or N2 > 1000:
return
indict['x'].append(xc)
indict['exl'].append(xc - x0)
indict['exh'].append(x1 - xc)
N = N1 + N2
err_low_N1 = float(tab.cell_get('N1', 'Low'))
err_low_N2 = float(tab.cell_get('N2', 'Low'))
err_low = (err_low_N1**2 + err_low_N2**2)**.5
err_high_N1 = float(tab.cell_get('N1', 'High'))
err_high_N2 = float(tab.cell_get('N2', 'High'))
err_high = (err_high_N1**2 + err_high_N2**2)**.5
yc = float(N)
yl = float(err_low)
yh = float(err_high)
indict['y'].append(yc)
indict['eyl'].append(yl)
indict['eyh'].append(yh)
def get_generated_numbers_single(i, prefix=''):
if '281ipb' in prefix:
mult_fact = 10
elif '537ipb' in prefix:
mult_fact = 20
elif '818ipb' in prefix or 'v13' in prefix:
mult_fact = 30
else:
raise NameError(prefix)
mode = i[0]
sign = i[1]
tag_number = modes[mode]['tag_num_s']
count = tag_number*mult_fact
if mode == 205:
if sign == 1:
count = 29972 # 30000
if sign == -1:
count = 35969 # 36000
if 'noFSR' in prefix:
mode = i[0]
sign = i[1]
logfile = tools.set_file('log', 's', mode, 's', sign, prefix,
extbase=evtlogbase)
tab = DHadTable(logfile)
FSR = tab.cell_get('FSR', 'Value')
count = count - int(FSR)
return count
def diff(args):
label= None
if opts.set:
for li in opts.set.split(':'):
name = li.split('=')[0]
value = li.split('=')[1]
sys.stdout.write('Set %s = %s \n' % (name, value))
if name == 'label':
label = value
if label:
labels = label.split(',')
label_A = labels[0]
label_B = labels[1]
tab_A = 'compare_%s.txt' % args[0]
tab_B = 'compare_%s.txt' % args[1]
tabfile_A = os.path.join(attr.tabpath, tab_A)
tabfile_B = os.path.join(attr.tabpath, tab_B)
tab = DHadTable()
tab.column_append_from_tab_file('Mode', tabfile_A, 'Mode')
tab.column_append_from_tab_file(label_A, tabfile_A, 'diff(%)')
tab.column_append_from_tab_file(label_B, tabfile_B, 'diff(%)')
tab.column_append_by_diff('diff', label_B,label_A)
tab.output(_tabname)
def entries(args):
dt_type = args[0]
tag = args[1]
label_A = args[2]
label_B = args[3]
evtpath = attr.evtpath
rnd='.01'
err_type=None
label = None
diff = 'pct'
tab = DHadTable()
tab.row_append(['Mode', label_A, label_B])
tab.head = tab.data[0]
for mode in attr.modes:
for sign in [1, -1]:
evtfile_A = tools.set_file('evt', dt_type, mode, tag, sign=sign,
prefix='dir_'+label_A, extbase=evtpath)
evtfile_B = tools.set_file('evt', dt_type, mode, tag, sign=sign,
prefix='dir_'+label_B, extbase=evtpath)
entries_A = tools.count_lines(evtfile_A)
entries_B = tools.count_lines(evtfile_B)
if sign == 1:
modename = attr.modes[mode]['fname']
else:
modename = attr.modes[mode]['fnamebar']
row = [modename, entries_A, entries_B]
tab.row_append(map(str, row))
tab.column_append_by_diff_pct('diff(%)', label_B,label_A, rnd=rnd, err_type=err_type)
tab.output(_tabname)
def get_generated_numbers_double(pair, prefix=''):
if '281ipb' in prefix:
mult_fact = 10
elif '537ipb' in prefix:
mult_fact = 20
elif '818ipb' in prefix or 'v13' in prefix:
mult_fact = 30
else:
raise NameError(prefix)
i = pair[0]
j = pair[1]
#genpath = attr.genpath
#print genpath
#sys.exit()
#if attr.analysis == '10.1' and '281ipbv0' in prefix :
# genpath = '/nfs/cor/user/ponyisi/cleog/scripts-summerconf/'
tag_number = 2000
count = tag_number * mult_fact
if 'v13' in prefix:
fname = modes[i]['fname']
fnamebar = modes[j]['fnamebar']
modename = 'Double_%s__%s' % (fname, fnamebar)
if modename == 'Double_Dp_to_KKpi__Dm_to_KKpi':
count = 47966
if modename == 'Double_Dp_to_KKpi__Dm_to_Kpipipi0':
count = 58403
if modename == 'Double_Dp_to_KKpi__Dm_to_Kpipi':
count = 57954
if modename == 'Double_Dp_to_KKpi__Dm_to_Kspipi0':
count = 57969
if modename == 'Double_Dp_to_KKpi__Dm_to_Kspipipi':
count = 24693
if modename == 'Double_Dp_to_KKpi__Dm_to_Kspi':
count = 57953
if modename == 'Double_Dp_to_Kpipi__Dm_to_KKpi':
count = 59952
if modename == 'Double_Dp_to_Kpipipi0__Dm_to_KKpi':
count = 55955
if modename == 'Double_Dp_to_Kspi__Dm_to_KKpi':
count = 57954
if modename == 'Double_Dp_to_Kspipi0__Dm_to_KKpi':
count = 57956
if modename == 'Double_Dp_to_Kspipipi__Dm_to_KKpi':
count = 57954
if 'noFSR' in prefix:
logfile = tools.set_file('log',
's',
pair,
'd',
prefix=prefix,
extbase=evtlogbase)
tab = DHadTable(logfile)
FSR = tab.cell_get('FSR', 'Value')
count = count - int(FSR)
return count
def create_fig_mode(tag, dt_type, mode, label):
if tag == 'single':
modename = attr.modes[mode]['orgname']
else:
if mode == 'double_all_d0s':
modename = 'Double all D0s'
elif mode == 'double_all_dps':
modename = 'Double all Dps'
else:
mode1 = mode[0]
mode2 = mode[1]
modename = '%s, %s' % (attr.modes[mode1]['orgname'],
attr.modes[mode2]['orgnamebar'])
prefix = 'dir_' + label
epsfile = tools.set_file('eps',
dt_type,
mode,
tag,
prefix=prefix,
extbase=attr.figpath)
head, tail = os.path.split(epsfile)
comname = tail.replace('.eps', '')
figlabel = label.split('/')[0]
relpath = label.replace(figlabel, '')
pdflink = '.%s/%s.pdf' % (relpath, comname)
pnglink = '.%s/%s.png' % (relpath, comname)
loglink = '[[../../log/%s/%s.txt][log]]' % (label, comname)
figlink = '[[%s][%s]]' % (pdflink, pnglink)
fitpath = attr.fitpath
tabfile = tools.set_file('txt',
dt_type,
mode,
tag,
prefix=prefix,
extbase=fitpath)
if os.access(tabfile, os.F_OK):
orgtabfile = tools.set_file('org',
dt_type,
mode,
tag,
prefix=prefix,
extbase=fitpath)
tab = DHadTable(tabfile)
tab.output_org(orgtabfile)
abspath = os.path.join(attr.base) #, attr.analysis)
orgtabfile = orgtabfile.replace(abspath, '../..')
tablink = '#+INCLUDE: "%s"\n' % orgtabfile
msg = '\n* %s \n %s \n\n%s\n %s\n' % (modename, figlink, tablink,
loglink)
else:
sys.stdout.write('File does not exist: %s \n' % tabfile)
sys.stdout.write('Skipping mode %s ... \n' % str(mode))
msg = '\n* %s \n Skipped.\n' % modename
return msg
def get_generated_numbers_double(pair, prefix=''):
if '281ipb' in prefix:
mult_fact = 10
elif '537ipb' in prefix:
mult_fact = 20
elif '818ipb' in prefix or 'v13' in prefix:
mult_fact = 30
else:
raise NameError(prefix)
i = pair[0]
j = pair[1]
#genpath = attr.genpath
#print genpath
#sys.exit()
#if attr.analysis == '10.1' and '281ipbv0' in prefix :
# genpath = '/nfs/cor/user/ponyisi/cleog/scripts-summerconf/'
tag_number = 2000
count = tag_number * mult_fact
if 'v13' in prefix:
fname = modes[i]['fname']
fnamebar = modes[j]['fnamebar']
modename = 'Double_%s__%s' %(fname, fnamebar)
if modename == 'Double_Dp_to_KKpi__Dm_to_KKpi':
count = 47966
if modename == 'Double_Dp_to_KKpi__Dm_to_Kpipipi0':
count = 58403
if modename == 'Double_Dp_to_KKpi__Dm_to_Kpipi':
count = 57954
if modename == 'Double_Dp_to_KKpi__Dm_to_Kspipi0':
count = 57969
if modename == 'Double_Dp_to_KKpi__Dm_to_Kspipipi':
count = 24693
if modename == 'Double_Dp_to_KKpi__Dm_to_Kspi':
count = 57953
if modename == 'Double_Dp_to_Kpipi__Dm_to_KKpi':
count = 59952
if modename == 'Double_Dp_to_Kpipipi0__Dm_to_KKpi':
count = 55955
if modename == 'Double_Dp_to_Kspi__Dm_to_KKpi':
count = 57954
if modename == 'Double_Dp_to_Kspipi0__Dm_to_KKpi':
count = 57956
if modename == 'Double_Dp_to_Kspipipi__Dm_to_KKpi':
count = 57954
if 'noFSR' in prefix:
logfile = tools.set_file('log', 's', pair, 'd', prefix=prefix,
extbase=evtlogbase)
tab =DHadTable(logfile)
FSR = tab.cell_get('FSR', 'Value')
count = count - int(FSR)
return count
def get_fsr_sys(label):
if label == '281ipbv12.0':
tabfile = os.path.join(attr.tabpath, '281ipbv0', 'FSR_syst.txt')
tab = DHadTable(tabfile)
print 'Loading %s ...' % tabfile
fsr_sys = tab.column_get('diff X 30%')
else:
tab = get_mode_dependent_syst_tab(label)
fsr_sys = tab.row_get('FSR')
return fsr_sys
def events(args):
parsed = yld.parse_args(args)
datatype = parsed[0]
tag = parsed[1]
modes = parsed[2]
label_A = parsed[3]
label_B = args[3]
datpath = attr.datpath
tab = DHadTable()
tab.row_append(['Mode', label_A, label_B, 'Common',
'Unique(%s)' %label_A, 'Unique(%s)' %label_B ])
tab.head = tab.data[0]
for mode in modes:
modename = mode.replace('Single_', '')
evtname = datatype + '_' + mode + '.evt'
evtpath_A = os.path.join(datpath, 'evt', label_A, 'events')
evtfile_A = os.path.join(evtpath_A, evtname)
f_A = UserFile(evtfile_A)
events_A = Set(f_A.data)
evtpath_B = os.path.join(datpath, 'evt', label_B, 'events')
evtfile_B = os.path.join(evtpath_B, evtname)
f_B = UserFile(evtfile_B)
events_B = Set(f_B.data)
events_inter = events_A & events_B
entries_A = len(events_A)
entries_B = len(events_B)
common = len(events_inter)
unique_A = len(events_A - events_inter)
unique_B = len(events_B - events_inter)
row = [modename, entries_A, entries_B, common, unique_A, unique_B]
tab.row_append(map(str, row))
tab.output(_tabname)
def signal_double_efficiencies(prefix, prefix_signal, verbose=0):
effmatrix = []
errmatrix = []
dt_type = 's'
tag = 'd'
mode_pair_list = attr.PossibleDoubleTags
for i in mode_pair_list:
txtfile = tools.set_file('txt',
dt_type,
i,
tag,
prefix=prefix_signal,
extbase=attr.fitpath)
N = float(DHadTable(txtfile).cell_get(1, 'Value'))
N_err = float(DHadTable(txtfile).cell_get(1, 'Error'))
tot_gen = get_generated_numbers_double(i, prefix_signal)
effline = []
errline = []
for j in mode_pair_list:
if i == j:
if '281ipbv0.1' in prefix and i == (1, 1):
tot_gen = tot_gen * (0.257402179684 / 0.140119760479)
eff = N / tot_gen
err = N_err / tot_gen * math.sqrt(1 - eff)
else:
eff = 0
err = 0
effline.append(eff)
errline.append(err)
effmatrix.append(effline)
errmatrix.append(errline)
eff_file_name = 'signal_double_efficiencies_for_werner'
err_file_name = 'signal_double_efficiencies_errors_for_werner'
eff_file = tools.set_file(extbase=attr.brfpath,
prefix=prefix,
comname=eff_file_name)
err_file = tools.set_file(extbase=attr.brfpath,
prefix=prefix,
comname=err_file_name)
output_matrix(effmatrix, eff_file)
output_matrix(errmatrix, err_file)
def get_signal_shape_sys(label):
if label == '281ipbv12.0':
tabfile = os.path.join(attr.tabpath, '281ipbv0',
'signal_line_shape_syst.txt')
print 'Loading %s ...' % tabfile
tab = DHadTable(tabfile)
line_shape_sys = tab.column_get('Total(%)')
else:
tab = get_mode_dependent_syst_tab(label)
line_shape_sys = tab.row_get('Signal shape')
return line_shape_sys
def bkg_cont_single_mode(datatype, mode, label, debug):
prefix = 'dir_'+label
bkgname = '%s_%s.txt' %(datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(
extbase=attr.bkgpath(), prefix=prefix, comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' %bkgfile)
modename = mode.replace('Single_', '')
result = [modename]
tab = DHadTable(bkgfile)
result.extend(tab.column_get('Number')[1:])
return result
def mctruth_ddbar_single_mode(datatype, mode, label, ranges, test):
var = 'mctruth'
mctruthfile = get_evtfile(datatype, mode, label, var)
if test:
sys.stdout.write('\ntest: mctruthfile %s\n' % mctruthfile)
modename = mode.replace('Single_', '')
mctruthtable = get_mctruth_tab_link(datatype, mode, label)
result = [modename, mctruthtable]
tab = DHadTable(mctruthfile)
result.extend(tab.column_analyze('Number', ranges))
return result
def files(args):
if args[0].startswith('/'):
dira = args[0]
else:
dira = os.path.join(attr.base, args[0])
if args[1].startswith('/'):
dirb = args[1]
else:
dirb = os.path.join(attr.base, args[1])
d = filecmp.dircmp(dira, dirb)
sys.stdout.write('There are %s different files. \n' % len(d.diff_files))
tab = DHadTable()
tab.row_append(['FileName', 'A', 'B'])
for f in d.diff_files:
filea = os.path.join(dira, f)
fileb = os.path.join(dirb, f)
tab.row_append_from_files(f, [filea, fileb])
tab.output_org()
def events(args):
parsed = yld.parse_args(args)
datatype = parsed[0]
tag = parsed[1]
modes = parsed[2]
label = parsed[3]
unique_evt_label = label
if _set and 'events' in _set:
unique_evt_label = _set.split('=')[1]
tab = DHadTable()
row = ['Mode']
row.extend(attr.datasets_281)
tab.row_append(row)
for mode in modes:
modename = mode.replace('Single_', '')
evtfile = get_unique_evtfile(datatype,
mode,
unique_evt_label,
debug=_debug)
f = EventsFile(evtfile)
row = map(str, f.datasets_sorted)
row.insert(0, modename)
tab.row_append(row)
tab.output(_tabname)
def bkg_cont_single_mode(datatype, mode, label, debug):
prefix = 'dir_' + label
bkgname = '%s_%s.txt' % (datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(extbase=attr.bkgpath(),
prefix=prefix,
comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' % bkgfile)
modename = mode.replace('Single_', '')
result = [modename]
tab = DHadTable(bkgfile)
result.extend(tab.column_get('Number')[1:])
return result
def mctruth_names(opts, args):
debug_ = opts.debug
test_ = opts.test
set_ = opts.set
prefix = None
if args[0] == 'peak':
prefix = 'peak'
args = args[1:]
parsed = yld.parse_args(args)
datatype = parsed[0]
tag = parsed[1]
modes = parsed[2]
label = parsed[3]
limit = None
if set_ != None and 'min' in set_:
min_ = set_.split('=')[1]
limit = '>%s' % min_
mctruthmodes = 50
if set_ != None and 'mctruthmodes' in set_:
mctruthmodes = int(set_.split('=')[1])
for mode in modes:
tabname = 'mctruth_names_%s_%s_%s' % (datatype, mode, label)
if prefix != None:
tabname = 'mctruth_names_%s_%s_%s_%s' % (prefix, datatype, mode,
label)
tabname = tabname.replace('/', '_')
tab = DHadTable()
modename = get_orgname_from_fname(mode)
mctruth_names, mctruth_numbers = mctruth_names_single_mode(
datatype, mode, label, limit, mctruthmodes, debug_, prefix)
tab.column_append(mctruth_names, title='MC Truth for %s' % modename)
tab.column_append(mctruth_numbers, title='Number of Events')
tab.sort_by_column('Number of Events', reverse=True)
orgfooter = ': '+sys.argv[0].split('/')[-1]+\
' tab mctruth names %s %s %s ' %(
datatype, mode, label)
if prefix != None:
orgfooter = ': '+sys.argv[0].split('/')[-1]+\
' tab mctruth names %s %s %s %s ' %(
prefix, datatype, mode, label)
tab.output(tabname, orgfooter=orgfooter, test=test_)
def get_argus_paras_single(label, dt_type, tag, mode):
if 'argus' in label:
prefix = 'dir_' + label.replace('argus', 'desideband')
argus_para_base = attr.fitpath
tabfile = tools.set_file('txt', dt_type, mode, tag,
prefix=prefix,
extbase=argus_para_base)
sys.stdout.write('Loading ARGUS parameters from %s ...\n' %tabfile)
tab = DHadTable(tabfile)
xi = float(tab.cell_get('xi', 'Value'))
p = float(tab.cell_get('p', 'Value'))
return xi, p
def bkg_ddbar_single_mode(datatype, mode, label, ranges, debug):
prefix = 'dir_'+label
bkgname = '%s_%s.txt' %(datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(
extbase=attr.bkgpath(), prefix=prefix, comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' %bkgfile)
modename = mode.replace('Single_', '')
bkgtable = get_bkg_tab_link(datatype, mode, label)
result = [modename, bkgtable]
tab = DHadTable(bkgfile)
result.extend(tab.column_analyze('Number', ranges))
return result
def bkg_ddbar_single_mode(datatype, mode, label, ranges, debug):
prefix = 'dir_' + label
bkgname = '%s_%s.txt' % (datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(extbase=attr.bkgpath(),
prefix=prefix,
comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' % bkgfile)
modename = mode.replace('Single_', '')
bkgtable = get_bkg_tab_link(datatype, mode, label)
result = [modename, bkgtable]
tab = DHadTable(bkgfile)
result.extend(tab.column_analyze('Number', ranges))
return result
def create_fig_mode(tag, dt_type, mode, label):
if tag == 'single':
modename = attr.modes[mode]['orgname']
else:
if mode == 'double_all_d0s':
modename = 'Double all D0s'
elif mode == 'double_all_dps':
modename = 'Double all Dps'
else:
mode1 = mode[0]
mode2 = mode[1]
modename = '%s, %s' % (attr.modes[mode1]['orgname'],
attr.modes[mode2]['orgnamebar'])
prefix = 'dir_' + label
epsfile = tools.set_file('eps', dt_type, mode, tag,
prefix=prefix, extbase=attr.figpath)
head, tail = os.path.split(epsfile)
comname = tail.replace('.eps', '')
figlabel = label.split('/')[0]
relpath = label.replace(figlabel, '')
pdflink = '.%s/%s.pdf' %(relpath, comname)
pnglink = '.%s/%s.png' %(relpath, comname)
loglink = '[[../../log/%s/%s.txt][log]]' %(label, comname)
figlink = '[[%s][%s]]' %(pdflink, pnglink)
fitpath = attr.fitpath
tabfile = tools.set_file('txt', dt_type, mode, tag,
prefix=prefix, extbase=fitpath)
if os.access(tabfile, os.F_OK):
orgtabfile = tools.set_file('org', dt_type, mode, tag,
prefix=prefix, extbase=fitpath)
tab = DHadTable(tabfile)
tab.output_org(orgtabfile)
abspath = os.path.join(attr.base)#, attr.analysis)
orgtabfile = orgtabfile.replace(abspath, '../..')
tablink = '#+INCLUDE: "%s"\n' % orgtabfile
msg = '\n* %s \n %s \n\n%s\n %s\n' % (
modename, figlink, tablink, loglink)
else:
sys.stdout.write('File does not exist: %s \n' %tabfile)
sys.stdout.write('Skipping mode %s ... \n' % str(mode))
msg = '\n* %s \n Skipped.\n' % modename
return msg
def make_overview_table(tabname, datatype, modes, label, test=False):
tab = DHadTable()
tab.row_append(['Mode', 'Backgrounds'])
for mode in modes:
modename = mode.replace('Single_', '')
bkgtable = get_bkg_tab_link(datatype, mode, label)
line = [modename, bkgtable]
tab.row_append(line)
tab.output(tabname, test=test)
def bkg_generic_cont(tabname, datatype, tag, modes, label, test, debug):
tab = DHadTable()
row = ['Mode', 'Continuum Backgrounds']
tab.row_append(row)
for mode in modes:
new_row = bkg_cont_single_mode(datatype, mode, label, debug)
tab.row_append(new_row)
tab.output(tabname, test=test)
def get_dcsd_sys(label, i, j):
if label in ['818ipbv7']:
sys_err = attr.sys_err(label)
dcsd_sys = float(sys_err['Double DCSD interference(Neutral DT)'])
dcsd_cor = None
elif label in ['818ipbv8', '818ipbv9', '818ipbv10']:
tabdir = '818ipbv7'
tabfile = os.path.join(attr.cbxtabpath, tabdir, 'dcsd_correction.txt')
tab = DHadTable(tabfile)
modename = '%s %s' % (modes[i]['fname'], modes[j]['fnamebar'])
dcsd_val_err = tab.cell_get(modename, 'Yield correction factor')
dcsd_cor = float(dcsd_val_err.split('+/-')[0])
dcsd_sys = float(dcsd_val_err.split('+/-')[1])
else:
raise NameError(lable)
return dcsd_sys, dcsd_cor
def evt(opts, args):
tabname = 'entries_evt_' + '_'.join(args).replace('/', '_')
if args[0] == 'log':
return evt_log(args[1:])
if args[0] not in ['data']:
raise NameError(args)
dt_type = args[0]
tag = args[1]
label = args[2]
evtpath = attr.evtpath()
if '/' in label:
ver = label.split('/')[0]
label = label.split('/')[1]
evtpath = attr.evtpath(ver)
prefix = 'dir_' + label
tab = DHadTable()
tab.row_append(['Mode', 'fname', 'fnamebar', 'combined', 'sum(theory)'])
for mode in attr.modes:
evtfile1 = tools.set_file('evt',
dt_type,
mode,
tag,
sign=-1,
prefix=prefix,
extbase=evtpath)
evtfile2 = tools.set_file('evt',
dt_type,
mode,
tag,
sign=1,
prefix=prefix,
extbase=evtpath)
evtfile_combined = tools.set_file('evt',
dt_type,
mode,
tag,
prefix=prefix,
extbase=evtpath)
entries1 = tools.count_lines(evtfile1)
entries2 = tools.count_lines(evtfile2)
sumup = entries1 + entries2
entries_combined = tools.count_lines(evtfile_combined)
row = [mode, entries1, entries2, entries_combined, sumup]
tab.row_append(map(str, row))
tab.output(tabname)
def data_yields_double(fo, prefix, opt='', label='', verbose=0):
dt_type = 'd'
tag = 'd'
mode_pair_list = attr.PossibleDoubleTags
n = 0
for mode_pair in mode_pair_list:
i = mode_pair[0]
j = mode_pair[1]
txtfile = tools.set_file('txt',
dt_type,
mode_pair,
tag,
prefix=prefix,
extbase=attr.fitpath)
tab = DHadTable(txtfile)
N = tab.cell_get(1, 'Value')
N_err = tab.cell_get(1, 'Error')
if 'DCSD_0' in opt:
dcsd_sys = 0
if opt == 'statonly':
syerrmul = 0
else:
syerrmul = 0
if i in (0, 1, 3):
dcsd_sys, dcsd_cor = get_dcsd_sys(label, i, j)
syerrmul += dcsd_sys**2
if dcsd_cor != None:
sys.stdout.write(
'Correcting mode (%s %s) with %s +/- %s ...\n' %
(i, j, dcsd_cor, dcsd_sys))
N = str(float(N) * dcsd_cor)
syerrmul = syerrmul**.5
N_err_sys = str(float(N) * syerrmul)
fo.write('%s\n%s\n%s\n' % (N, N_err, N_err_sys))
n += 1
if verbose > 1:
print '%s +/- %s +/- %s' % (N, N_err, N_err_sys)
if verbose > 0:
print 'Total double tag data yield lines ... %s' % n
def mctruth_names_single_mode(datatype,
mode,
label,
limit,
mctruthmodes,
debug,
prefix=None):
var = 'mctruth'
mctruthfile = get_evtfile(datatype, mode, label, var)
if debug:
sys.stdout.write('\ndebug: mctruthfile %s\n' % mctruthfile)
tab = DHadTable(mctruthfile, evalcell=True)
tab.sort_by_column('Number', reverse=True)
if limit != None:
mcddbars = tab.column_get('Mode', 'Number' + limit)[1:]
if mctruthmodes != None:
mcddbars = tab.column_get('Mode')[1:mctruthmodes + 1]
numbers = tab.column_get('Number')[1:len(mcddbars) + 1]
mcddbarmodenames = mcddbars_to_modenames(mcddbars)
return mcddbarmodenames, numbers
def bkg_names_single_mode(datatype,
mode,
label,
limit,
bkgmodes,
debug,
prefix=None):
bkgname = '%s_%s.txt' % (datatype, mode)
if prefix != None:
bkgname = '%s_%s_%s.txt' % (prefix, datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(extbase=attr.bkgpath(),
prefix='dir_' + label,
comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' % bkgfile)
tab = DHadTable(bkgfile, evalcell=True)
tab.sort_by_column('Number', reverse=True)
if limit != None:
mcddbars = tab.column_get('Mode', 'Number' + limit)[1:]
if bkgmodes != None:
mcddbars = tab.column_get('Mode')[1:bkgmodes + 1]
numbers = tab.column_get('Number')[1:len(mcddbars) + 1]
mcddbarmodenames = mcddbars_to_modenames(mcddbars)
return mcddbarmodenames, numbers
def bkg_names(opts, args):
debug_ = opts.debug
test_ = opts.test
set_ = opts.set
prefix = None
if args[0] == 'peak':
prefix = 'peak'
args = args[1:]
parsed = yld.parse_args(args)
datatype = parsed[0]
tag = parsed[1]
modes = parsed[2]
label = parsed[3]
limit = None
if set_ != None and 'min' in set_:
min_ = set_.split('=')[1]
limit = '>%s' %min_
bkgmodes = 20
if set_ != None and 'bkgmodes' in set_:
bkgmodes = int(set_.split('=')[1])
for mode in modes:
tabname = 'bkg_names_%s_%s_%s' %(
datatype, mode, label)
if prefix != None:
tabname = 'bkg_names_%s_%s_%s_%s' %(
prefix, datatype, mode, label)
tabname = tabname.replace('/', '_')
tab = DHadTable()
modename = get_orgname_from_fname(mode)
bkg_names, bkg_numbers = bkg_names_single_mode(
datatype, mode, label, limit, bkgmodes, debug_, prefix)
tab.column_append(bkg_names, title='Backgrounds for %s' %modename)
tab.column_append(bkg_numbers, title='Number of Events')
tab.sort_by_column('Number of Events', reverse=True)
orgfooter = ': '+sys.argv[0].split('/')[-1]+\
' tab backgrounds names %s %s %s ' %(
datatype, mode, label)
if prefix != None:
orgfooter = ': '+sys.argv[0].split('/')[-1]+\
' tab backgrounds names %s %s %s %s ' %(
prefix, datatype, mode, label)
tab.output(tabname, orgfooter=orgfooter, test=test_)
def init_paras_single(label, dt_type, tag, mode):
if '281ipb' in label:
init_para_prefix = 'dir_281ipbv12'
elif '537ipb' in label:
init_para_prefix = 'dir_537ipbv12'
elif '818ipb' in label or 'v13' in label:
init_para_prefix = 'dir_818ipbv12'
else:
raise ValueError(label)
init_para_base = attr.fitpath
if dt_type == 'generic':
if label in ['281ipbv7', '537ipbv7', '818ipbv7']:
dt_type = 'signal'
tabfile = tools.set_file('txt', dt_type, mode, tag,
prefix=init_para_prefix,
extbase=init_para_base)
sys.stdout.write('Loading init paras from %s ... \n' %tabfile)
tab = DHadTable(tabfile)
N1 = float(tab.cell_get('N1', 'Value'))
N2 = float(tab.cell_get('N2', 'Value'))
Nbkgd1 = float(tab.cell_get('Nbkgd1', 'Value'))
Nbkgd2 = float(tab.cell_get('Nbkgd2', 'Value'))
md = float(tab.cell_get('md', 'Value'))
p = float(tab.cell_get('p', 'Value'))
sigmap1 = float(tab.cell_get('sigmap1', 'Value'))
xi = float(tab.cell_get('xi', 'Value'))
return N1, N2, Nbkgd1, Nbkgd2, md, p, sigmap1, xi
def events(args):
parsed = yld.parse_args(args)
datatype = parsed[0]
tag = parsed[1]
modes = parsed[2]
label = parsed[3]
unique_evt_label = label
if _set and 'events' in _set:
unique_evt_label = _set.split('=')[1]
tab = DHadTable()
row = ['Mode']
row.extend(attr.datasets_281)
tab.row_append(row)
for mode in modes:
modename = mode.replace('Single_', '')
evtfile = get_unique_evtfile(
datatype, mode, unique_evt_label, debug=_debug)
f = EventsFile(evtfile)
row = map(str, f.datasets_sorted)
row.insert(0, modename)
tab.row_append(row)
tab.output(_tabname)
def get_argus_paras_single(label, dt_type, tag, mode):
if 'argus' in label:
prefix = 'dir_' + label.replace('argus', 'desideband')
argus_para_base = attr.fitpath
tabfile = tools.set_file('txt',
dt_type,
mode,
tag,
prefix=prefix,
extbase=argus_para_base)
sys.stdout.write('Loading ARGUS parameters from %s ...\n' % tabfile)
tab = DHadTable(tabfile)
xi = float(tab.cell_get('xi', 'Value'))
p = float(tab.cell_get('p', 'Value'))
return xi, p
def get_dcsd_sys(label, i, j):
if label in ['818ipbv7']:
sys_err = attr.sys_err(label)
dcsd_sys = float(sys_err['Double DCSD interference(Neutral DT)'])
dcsd_cor = None
elif label in ['818ipbv8', '818ipbv9', '818ipbv10']:
tabdir = '818ipbv7'
tabfile = os.path.join(attr.cbxtabpath, tabdir,
'dcsd_correction.txt')
tab = DHadTable(tabfile)
modename = '%s %s' %(modes[i]['fname'], modes[j]['fnamebar'])
dcsd_val_err = tab.cell_get(modename, 'Yield correction factor')
dcsd_cor = float(dcsd_val_err.split('+/-')[0])
dcsd_sys = float(dcsd_val_err.split('+/-')[1])
else:
raise NameError(lable)
return dcsd_sys, dcsd_cor
def get_brs_from_file(label):
tab = DHadTable()
bffilename = 'bf_stat_sys'
bffile = os.path.join(attr.brfpath, label, bffilename)
tab.column_append(parse_result(bffile, 'paras'), 'Parameters')
tab.column_append(parse_result(bffile, 'value'), 'value')
tab.column_append(parse_result(bffile, 'stat'), 'stat')
tab.column_append(parse_result(bffile, 'syst'), 'syst')
tab.rows_delete(['Parameters', 'ND0D0Bar', 'ND+D-'])
content = []
n = 0
for row in tab.data:
n += 1
line = 'exec addData %s %s %s %s \n' %(
n, row[1], row[2], row[3])
content.append(line)
return content
def get_bkg_total(datatype, mode, label, debug):
prefix = 'dir_'+label
bkgname = '%s_%s.txt' %(datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(
extbase=attr.bkgpath(), prefix=prefix, comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' %bkgfile)
tab = DHadTable(bkgfile)
col = tab.column_get('Number')
total = sum(col[1:])
if 'ddbar' in datatype:
factor = 20
if 'cont' in datatype:
factor = 5
total = total/float(factor)
return total
def generic_yields_double(fo, prefix, opt='', label='', verbose=0):
dt_type = 'g'
tag = 'd'
mode_pair_list = attr.PossibleDoubleTags
n = 0
for mode_pair in mode_pair_list:
i = mode_pair[0]
j = mode_pair[1]
txtfile = tools.set_file('txt', dt_type, mode_pair, tag,
prefix=prefix, extbase=attr.fitpath)
tab = DHadTable(txtfile)
N = tab.cell_get(1, 'Value')
N_err = tab.cell_get(1, 'Error')
if 'DCSD_0' in opt :
dcsd_sys = 0
if opt == 'statonly':
syerrmul = 0
else:
syerrmul = 0
if i in (0, 1, 3):
dcsd_sys, dcsd_cor = get_dcsd_sys(label, i, j)
syerrmul += dcsd_sys**2
if dcsd_cor != None:
sys.stdout.write(
'Correcting mode (%s %s) with %s +/- %s ...\n' %(
i,j,dcsd_cor, dcsd_sys))
N = str(float(N)*dcsd_cor)
syerrmul = syerrmul**.5
N_err_sys = str(float(N)*syerrmul)
fo.write('%s\n%s\n%s\n' % (N, N_err, N_err_sys))
n += 1
if verbose > 1:
print '%s +/- %s +/- %s' % (N, N_err, N_err_sys)
if verbose >0:
print 'Total double tag generic yield lines ... %s' %n
def get_resolution_paras(mode, label):
if 'resolution' in label:
Sigma = 0.005
Fa = 0.05
Fb = 0.05
Sa = 2.0
Sb = 2.0
else:
tab_name = 'para_momentum_resolution'
short_label = label.split('/')[0]
if short_label in [
'281ipbv0', '281ipbv12', '537ipbv12', '818ipbv12', '818ipbv13',
'v13'
]:
tab_name = '%s/%s' % (short_label, tab_name)
else:
raise NameError(label)
tab_prefix = ''
tab_opt_sigma = 'value'
tab_opt_fa = 'value'
tab_opt_fb = 'value'
tab_opt_sa = 'value'
tab_opt_sb = 'value'
tabfile = tools.set_file(extbase=attr.tabpath,
prefix=tab_prefix,
comname=tab_name,
ext='txt')
sys.stdout.write('Loading lineshape parameters from %s... \n' %
tabfile)
tab = DHadTable(tabfile)
line_no = attr.mode_line_dict[mode]
Sigma = float(tab.cell_get(line_no, 'sigma (MeV)',
tab_opt_sigma)) / 1000
Fa = float(tab.cell_get(line_no, 'fa', tab_opt_fa))
Fb = float(tab.cell_get(line_no, 'fb', tab_opt_fb))
Sa = float(tab.cell_get(line_no, 'sa', tab_opt_sa))
Sb = float(tab.cell_get(line_no, 'sb', tab_opt_sb))
return Sigma, Fa, Fb, Sa, Sb
def make_overview_table(tabname, datatype, modes, label, test=False):
tab = DHadTable()
tab.row_append(['Mode', 'Backgrounds'])
for mode in modes:
modename = mode.replace('Single_', '')
bkgtable = get_bkg_tab_link(datatype, mode, label)
line = [modename, bkgtable]
tab.row_append(line)
tab.output(tabname, test=test)
def get_bkg_total(datatype, mode, label, debug):
prefix = 'dir_' + label
bkgname = '%s_%s.txt' % (datatype, mode)
bkgname = bkgname.replace('/', '_')
bkgfile = tools.set_file(extbase=attr.bkgpath(),
prefix=prefix,
comname=bkgname)
if debug:
sys.stdout.write('\ndebug: bkgfile %s\n' % bkgfile)
tab = DHadTable(bkgfile)
col = tab.column_get('Number')
total = sum(col[1:])
if 'ddbar' in datatype:
factor = 20
if 'cont' in datatype:
factor = 5
total = total / float(factor)
return total
def para_momentum_resolution():
if tools.valid_version('10.1.5'):
tabnameA = 'line_shape_paras'
tabnameB = 'para_momentum_resolution'
else:
raise NameError(attr.src)
tabfileA = tools.set_file(extbase=attr.tabpath,
comname=tabnameA, ext='txt')
tabfileB = tools.set_file(extbase=attr.tabpath,
comname=tabnameB, ext='txt')
tabA = DHadTable(tabfileA)
tabB = DHadTable(tabfileB)
tab = tabA.diff_sigma_pct(tabB)
tab.output(_tabname)
def bkg_generic_cont(tabname, datatype, tag, modes,
label, test, debug):
tab = DHadTable()
row = ['Mode', 'Continuum Backgrounds']
tab.row_append(row)
for mode in modes:
new_row = bkg_cont_single_mode(datatype, mode, label, debug)
tab.row_append(new_row)
tab.output(tabname, test=test)
def evt_log(args):
print args
if args[0] not in ['data'] :
raise NameError(args)
dt_type = args[0]
tag = args[1]
label = args[2]
evtpath = attr.evtpath()
if '/' in label:
ver = label.split('/')[0]
label = label.split('/')[1]
evtpath = attr.evtpath(ver)
prefix = 'dir_' + label
evttab = 'entries_evt' + '_'.join(args).replace('/', '_') + '.tab'
tab = DHadTable()
tab.column_append_from_tab_file('Mode', evttab)
tab.output()
sys.exit()
def create_fig_mode(tag, datatype, mode, label):
if tag != 'single':
raise NameError(tag)
comname = '%s_%s' %(datatype.replace('/', '_'), mode)
epspath = os.path.join(attr.figpath, label)
epsname = '%s.eps' % comname
epsfile = os.path.join(epspath, epsname)
txtpath = os.path.join(attr.fitpath(), label)
txtname = '%s.txt' % comname
txtfile = os.path.join(txtpath, txtname)
pdflink = './%s/%s.pdf' %(label, comname)
pnglink = './%s/%s.png' %(label, comname)
loglink = '[[../log/%s/backgrounds/%s.txt][log]]' %(label, comname)
figlink = '[[%s][%s]]' %(pdflink, pnglink)
modename = tools.get_orgname_from_fname(mode)
tablink = ''
if os.access(txtfile, os.F_OK):
orgtabfile = txtfile.replace('.txt', '.org')
tab = DHadTable(txtfile)
tab.output_org(orgtabfile)
abspath = os.path.join(attr.base, attr.analysis)
orgtabfile = orgtabfile.replace(abspath, '..')
tablink = '#+INCLUDE: "%s"\n' % orgtabfile
msg = '\n* %s \n %s \n\n%s\n %s\n' % (
modename, figlink, tablink, loglink)
return msg
def get_resolution_paras(mode, label):
if 'resolution' in label:
Sigma = 0.005
Fa = 0.05
Fb = 0.05
Sa = 2.0
Sb = 2.0
else:
tab_name='para_momentum_resolution'
short_label = label.split('/')[0]
if short_label in ['281ipbv0', '281ipbv12', '537ipbv12',
'818ipbv12', '818ipbv13', 'v13']:
tab_name = '%s/%s' % (short_label, tab_name)
else:
raise NameError(label)
tab_prefix = ''
tab_opt_sigma = 'value'
tab_opt_fa = 'value'
tab_opt_fb = 'value'
tab_opt_sa = 'value'
tab_opt_sb = 'value'
tabfile = tools.set_file(
extbase=attr.tabpath, prefix=tab_prefix,
comname=tab_name, ext='txt')
sys.stdout.write('Loading lineshape parameters from %s... \n' %tabfile)
tab = DHadTable(tabfile)
line_no = attr.mode_line_dict[mode]
Sigma = float(tab.cell_get(line_no, 'sigma (MeV)',tab_opt_sigma))/1000
Fa = float(tab.cell_get(line_no, 'fa', tab_opt_fa))
Fb = float(tab.cell_get(line_no, 'fb', tab_opt_fb))
Sa = float(tab.cell_get(line_no, 'sa', tab_opt_sa))
Sb = float(tab.cell_get(line_no, 'sb', tab_opt_sb))
return Sigma, Fa, Fb, Sa, Sb
def generic_yields_single(fo, prefix, opt='', label='', verbose=0):
#sys_err = attr.sys_err(label)
dt_type = 'g'
tag = 's'
#sys_err_by_mode = attr.sys_err_by_mode(label)
for mode in modes:
if opt == 'statonly':
syerrmul = 0
else:
syerrmul = sys_err_by_mode[mode]['ST Background modeling']
if mode ==0:
lepton_veto = sys_err_by_mode[mode]['Lepton veto']
syerrmul = math.sqrt(syerrmul**2+ lepton_veto**2)
input_file = tools.set_file('txt', dt_type, mode, tag,
prefix=prefix, extbase=attr.fitpath)
tab = DHadTable(input_file)
N1 = tab.cell_get(1, 'Value')
N1_err = tab.cell_get(1, 'Error')
N1_err_sys = str(float(N1)*syerrmul)
N2 = tab.cell_get(2, 'Value')
N2_err = tab.cell_get(2, 'Error')
N2_err_sys = str(float(N2)*syerrmul)
fo.write('%s\n%s\n%s\n' % (N1, N1_err, N1_err_sys))
fo.write('%s\n%s\n%s\n' % (N2, N2_err, N2_err_sys))
if verbose > 1:
print '%s +/- %s +/- %s' % (N1, N1_err, N1_err_sys)
print '%s +/- %s +/- %s' % (N2, N2_err, N2_err_sys)
if verbose >0:
print 'single tag generic yield finished.'
