class TauEnergyScale: def __init__(self, year, algorithm): self.testool = TauESTool(YEARLIB[year], ALGOLIB[algorithm]) self.tfestool = TauESTool(YEARLIB[year]) #tlv is four vector to correct def applyES(self, tlv, dm, genmatch): if genmatch == 1 or genmatch == 3: if dm == 0 or dm == 1: tlv *= self.tfestool.getFES(tlv.Eta(), dm, genmatch) else: pass else: tlv *= self.tfestool.getTES(tlv.Pt(), dm, genmatch) return tlv #tlv is four vector to correct def readES(self, tlv, dm, genmatch): if genmatch == 1 or genmatch == 3: if dm == 0 or dm == 1: return self.tfestool.getFES(tlv.Eta(), dm, genmatch) else: return 1. else: return self.tfestool.getTES(tlv.Pt(), dm, genmatch)
class TauEnergyScale: def __init__(self, year, algorithm): self.testool = TauESTool(YEARLIB[year], algorithm) self.tfestool = TauESTool(YEARLIB[year]) def applyES(tlv, dm, genmatch): if genmatch == 1 or genmatch == 3: tlv *= self.tfestool.getFES(tlv.Eta(), dm, genmatch) else: tlv *= self.tfestool.getTES(tlv.Pt(), dm, genmatch) return tlv