def get_lumi(filename):
sample_name = extract_sample(filename)
lumi = read_lumi(
os.path.join(
'inputs',
jobid,
sample_name+".lumicalc.sum"
)
)
return lumi
def get_lumi(filename):
sample_name = extract_sample(filename)
lumi = read_lumi(
os.path.join(
'inputs',
jobid,
sample_name+".lumicalc.sum"
)
)
return lumi
def get_lumis(pattern):
lumi_files = glob.glob( os.path.join('inputs', os.environ['jobid'], '*.lumicalc.sum') )
ret = 0.
#print pattern
for lumi_file in lumi_files:
sample_name = os.path.basename(lumi_file).split('.')[0]
#print lumi_file, sample_name
if fnmatch(sample_name, pattern):
#print 'Matches!'
ret += read_lumi(lumi_file)
return ret
def get_lumis(pattern):
lumi_files = glob.glob(
os.path.join('inputs', os.environ['jobid'], '*.lumicalc.sum'))
ret = 0.
#print pattern
for lumi_file in lumi_files:
sample_name = os.path.basename(lumi_file).split('.')[0]
#print lumi_file, sample_name
if fnmatch(sample_name, pattern):
#print 'Matches!'
ret += read_lumi(lumi_file)
return ret
import re
#from pudb import set_trace; set_trace()
paths = {
#os.environ['current'],
'current' : 'inputs/%s/' % os.environ['jobid'],
'previous' : 'inputs/2013-Jun-30-8TeV/',
}
for dataset in ['DoubleMu', 'DoubleElectron', 'MuEG']:
lumi_dict = {}
for jobid, path in paths.iteritems():
lumi_dict[jobid] = {}
for lumifile in glob.glob(path+'/data_'+dataset+'*.lumicalc.sum'):
lumi_dict[jobid][extract_sample(lumifile)] = read_lumi(lumifile)
print dataset
print '%60s%20s%20s' % ('', 'current', 'previous')
keys = []
for i in lumi_dict.values():
keys += i.keys()
keys = list(set(keys))
total_lumis = dict([(i, {'current' : 0., 'previous' : 0.}) for i in ['2012A', '2012B', '2012C', '2012D', 'TOTAL',]])
for sample in keys:
curr_l = lumi_dict['current'][sample] if sample in lumi_dict['current'] else 0.
total_lumis['TOTAL']['current'] += curr_l
previous_l = lumi_dict['previous'][sample] if sample in lumi_dict['previous'] else 0.
total_lumis['TOTAL']['previous'] += previous_l
import re
#from pudb import set_trace; set_trace()
paths = {
#os.environ['current'],
'current': 'inputs/%s/' % os.environ['jobid'],
'previous': 'inputs/2013-Jun-30-8TeV/',
}
for dataset in ['DoubleMu', 'DoubleElectron', 'MuEG']:
lumi_dict = {}
for jobid, path in paths.iteritems():
lumi_dict[jobid] = {}
for lumifile in glob.glob(path + '/data_' + dataset +
'*.lumicalc.sum'):
lumi_dict[jobid][extract_sample(lumifile)] = read_lumi(lumifile)
print dataset
print '%60s%20s%20s' % ('', 'current', 'previous')
keys = []
for i in lumi_dict.values():
keys += i.keys()
keys = list(set(keys))
total_lumis = dict([(i, {
'current': 0.,
'previous': 0.
}) for i in [
'2012A',
'2012B',
'2012C',
