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)