class EfficiencyCorrector(object):
def __init__(self,cfg):
self.fileMu = ROOT.TFile(getFullPath(cfg.muonFile))
self.histMu = self.fileMu.Get(cfg.muonHisto)
self.fileEle = ROOT.TFile(getFullPath(cfg.eleFile))
self.histEle = self.fileEle.Get(cfg.eleHisto)
self.name = cfg.name
self.weightMu = WeightCalculator(self.histMu)
self.weightEle = WeightCalculator(self.histEle)
def attachToObject(self,object):
if abs(object.pdgId()) ==11:
w = self.weightEle.getWeight(object.pt(),abs(object.eta()))
wU =w+ self.weightEle.getWeightErr(object.pt(),abs(object.eta()))
wD =w- self.weightEle.getWeightErr(object.pt(),abs(object.eta()))
setattr(object,self.name,w)
setattr(object,self.name+"Up",wU)
setattr(object,self.name+"Dwn",wD)
if abs(object.pdgId()) ==13:
w = self.weightMu.getWeight(object.pt(),(object.eta()))
wU =w+ self.weightMu.getWeightErr(object.pt(),(object.eta()))
wD =w- self.weightMu.getWeightErr(object.pt(),(object.eta()))
setattr(object,self.name,w)
setattr(object,self.name+"Up",wU)
setattr(object,self.name+"Dwn",wD)
class EfficiencyCorrector(object):
def __init__(self, cfg):
self.fileMu = ROOT.TFile(getFullPath(cfg.muonFile))
self.histMu = self.fileMu.Get(cfg.muonHisto)
self.fileEle = ROOT.TFile(getFullPath(cfg.eleFile))
self.histEle = self.fileEle.Get(cfg.eleHisto)
self.name = cfg.name
self.weightMu = WeightCalculator(self.histMu)
self.weightEle = WeightCalculator(self.histEle)
def attachToObject(self, object):
if abs(object.pdgId()) == 11:
w = self.weightEle.getWeight(object.pt(), abs(object.eta()))
wU = w + self.weightEle.getWeightErr(object.pt(), abs(object.eta()))
wD = w - self.weightEle.getWeightErr(object.pt(), abs(object.eta()))
setattr(object, self.name, w)
setattr(object, self.name + "Up", wU)
setattr(object, self.name + "Dwn", wD)
if abs(object.pdgId()) == 13:
w = self.weightMu.getWeight(object.pt(), (object.eta()))
wU = w + self.weightMu.getWeightErr(object.pt(), (object.eta()))
wD = w - self.weightMu.getWeightErr(object.pt(), (object.eta()))
setattr(object, self.name, w)
setattr(object, self.name + "Up", wU)
setattr(object, self.name + "Dwn", wD)
def __init__(self,cfg):
self.fileMu = ROOT.TFile(getFullPath(cfg.muonFile))
self.histMu = self.fileMu.Get(cfg.muonHisto)
self.fileEle = ROOT.TFile(getFullPath(cfg.eleFile))
self.histEle = self.fileEle.Get(cfg.eleHisto)
self.name = cfg.name
self.weightMu = WeightCalculator(self.histMu)
self.weightEle = WeightCalculator(self.histEle)
def __init__(self, cfg):
self.fileMu = ROOT.TFile(getFullPath(cfg.muonFile))
self.histMu = self.fileMu.Get(cfg.muonHisto)
self.fileEle = ROOT.TFile(getFullPath(cfg.eleFile))
self.histEle = self.fileEle.Get(cfg.eleHisto)
self.name = cfg.name
self.weightMu = WeightCalculator(self.histMu)
self.weightEle = WeightCalculator(self.histEle)
