runs = Runs (['./nomadspin.root'])
file_out = open('fail.txt','w')
geninfo,geninfoLabel = Handle("GenEventInfoProduct"), "generator"
lheinfo,lheinfoLabel = Handle("LHEEventProduct"), "externalLHEProducer"
lheruninfo,lheruninfoLabel = Handle("LHERunInfoProduct"), "externalLHEProducer"
handlePruned = Handle ("std::vector<reco::GenParticle>")
labelPruned = ("prunedGenParticles")
# loop over events
count= 0
countweighted = 0
count = count+1
runs.getByLabel(lheruninfoLabel,lheruninfo)
lherun = lheruninfo.product()
iiter = lherun.headers_begin()
print len(iiter.lines())
for ii in range(0,lherun.headers_size()-1):
if ii<57: continue
# if 'initrwgt' not in iiter.tag(): continue
print 'tag:', iiter.tag()
for iline in range(0,iiter.lines().size()):
print iiter.lines()[iline]
iiter = iiter+1
th1f_ele1_pt = ROOT.TH1F("ele1 pt","",11,25,300)
th1f_ele11_pt = ROOT.TH1F("ele11 pt","",11,25,300)
th1f_ele12_pt = ROOT.TH1F("ele12 pt","",11,25,300)
th1f_ele13_pt = ROOT.TH1F("ele13 pt","",11,25,300)
class LHEWeightAnalyzer(Analyzer):
"""Read the WeightsInfo objects of the LHE branch and store them
in event.LHE_weights list.
If the WeightsInfo.id is a string, replace it with an integer.
So far the only allowed string format is "mg_reweight_X",
which gets stored as str(10000+int(X))
If w.id is an unknown string or anything but a string or int,
a RuntimeError is raised.
"""
def __init__(self, cfg_ana, cfg_comp, looperName):
super(LHEWeightAnalyzer, self).__init__(cfg_ana, cfg_comp, looperName)
self.index = []
def declareHandles(self):
super(LHEWeightAnalyzer, self).declareHandles()
self.mchandles['LHEweights'] = AutoHandle('externalLHEProducer',
'LHEEventProduct',
mayFail=True,
fallbackLabel='source',
lazy=False)
def beginLoop(self, setup):
super(LHEWeightAnalyzer, self).beginLoop(setup)
self.run = Runs(self.cfg_comp.files[0])
runsHandle = Handle('LHERunInfoProduct')
self.run.getByLabel('externalLHEProducer', runsHandle)
self.scale = TH1F('scale', 'scale', 10, 0, 10)
self.pdf = TH1F('pdf', 'pdf', 101, 0, 101)
self.alpha_s = TH1F('alpha_s', 'alpha_s', 2, 0, 2)
iterator = runsHandle.product().headers_begin()
# scan iterator so that it includes weight information
for it in range(100):
flag = False
for itt in iterator.base().lines():
print itt
if itt.find('weight id') != -1: flag = True
if flag:
break
else:
iterator.next()
else:
print 'Could not find any weight information !!'
self.index = []
self.pdfindex = []
self.alphaindex = []
idx = 0
for a in iterator.base().lines():
if a.find('weight id') == -1: continue
print a
# For Higgs signal
# if (a.find('muR')!=-1 and a.find('muF')!=-1 and a.find('muR=2 muF=0.5')==-1 and a.find('muR=0.5 muF=2')==-1):
# For LO Z->tautau
if (a.find('mur') != -1 and a.find('muf') != -1
and a.find('mur=2 muf=0.5') == -1
and a.find('mur=0.5 muf=2') == -1):
self.index.append(idx)
# For Higgs signal
# if a.find('PDF')!=-1:
# flag = False
# for weight_id in range(260000, 260101):
# if a.find(str(weight_id))!=-1: flag = True
#
# if flag:
# self.pdfindex.append(idx)
#
# if a.find('PDF')!=-1 and a.find('265000')!=-1 or a.find('266000')!=-1:
# self.alphaindex.append(idx)
# For LO Z->tautau
if a.find('Member') != -1:
flag = False
for weight_id in range(10, 111):
if a.find('id=\"' + str(weight_id) + '\"') != -1:
flag = True
if flag:
self.pdfindex.append(idx)
# if a.find('pdf')!=-1 and a.find('292301')!=-1 or a.find('292302')!=-1:
idx += 1
self.alphaindex.append(0)
self.alphaindex.append(1)
print 'scale variation index: ', self.index, 'len', len(self.index)
print 'pdf variation index: ', self.pdfindex, 'len', len(self.pdfindex)
print 'alpha variation index: ', self.alphaindex, 'len', len(
self.alphaindex)
def process(self, event):
self.readCollections(event.input)
# import pdb; pdb.set_trace()
# if not MC, nothing to do
if not self.cfg_comp.isMC:
return True
# Add LHE weight info
event.LHE_originalWeight = 1.0
if self.mchandles['LHEweights'].isValid() == False:
print 'LHE information not available !'
return True
event.LHE_originalWeight = self.mchandles['LHEweights'].product(
).originalXWGTUP()
event.scale_variation = []
event.pdf_variation = []
event.alpha_variation = []
for ii, windex in enumerate(self.index):
LHE_weight = self.mchandles['LHEweights'].product().weights(
)[windex].wgt
event.scale_variation.append(LHE_weight / event.LHE_originalWeight)
self.scale.Fill(ii, LHE_weight / event.LHE_originalWeight)
for ii, windex in enumerate(self.pdfindex):
LHE_weight = self.mchandles['LHEweights'].product().weights(
)[windex].wgt
event.pdf_variation.append(LHE_weight / event.LHE_originalWeight)
self.pdf.Fill(ii, LHE_weight / event.LHE_originalWeight)
for ii, windex in enumerate(self.alphaindex):
LHE_weight = self.mchandles['LHEweights'].product().weights(
)[windex].wgt
event.alpha_variation.append(LHE_weight / event.LHE_originalWeight)
self.alpha_s.Fill(ii, LHE_weight / event.LHE_originalWeight)
return True
def write(self, setup):
# import pdb; pdb.set_trace()
super(LHEWeightAnalyzer, self).write(setup)
self.rootfile = TFile('/'.join([self.dirName, 'weight.root']),
'recreate')
self.alpha_s.Write()
self.pdf.Write()
self.scale.Write()
self.rootfile.Write()
self.rootfile.Close()