def make_unclustered_variant(themet, op, deltaX, deltaY):
variant = copy(themet)
variant['corrected_met_x'] = awkward.virtual(
lambda met: op(out['corrected_met_x'], out[f'{deltaX}']),
args=(out, ),
length=length,
form=form,
cache=lazy_cache
)
variant['corrected_met_y'] = awkward.virtual(
lambda met: op(out['corrected_met_y'], out[f'{deltaY}']),
args=(out, ),
length=length,
form=form,
cache=lazy_cache
)
variant[self.name_map['METpt']] = awkward.virtual(
lambda met: numpy.hypot(out['corrected_met_x'], out['corrected_met_y']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
variant[self.name_map['METphi']] = awkward.virtual(
lambda met: numpy.arctan2(out['corrected_met_y'], out['corrected_met_x']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
return variant
def make_variant(name, variation):
variant = copy(MET)
variant['corrected_met_x'] = awkward.virtual(
corrected_met_cartesian_unc,
args=(out, orig_jets, variation, self.name_map['METx'], self.name_map['JETx']),
length=length, form=form, cache=lazy_cache
)
variant['corrected_met_y'] = awkward.virtual(
corrected_met_cartesian_unc,
args=(out, orig_jets, variation, self.name_map['METy'], self.name_map['JETy']),
length=length, form=form, cache=lazy_cache
)
variant[self.name_map['METpt']] = awkward.virtual(
lambda met: numpy.hypot(met['corrected_met_x'], met['corrected_met_y']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
variant[self.name_map['METphi']] = awkward.virtual(
lambda met: numpy.arctan2(met['corrected_met_y'], met['corrected_met_x']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
return variant
def getUncertainty(self, **kwargs):
"""
Returns the set of uncertainties for all input jets for all the levels (== sources)
Use it like::
juncs = uncertainty.getUncertainty(JetProperty1=jet.property1,...)
#'juncs' will be formatted like [('SourceName', [[up_val down_val]_jet1 ... ]), ...]
#in a zip iterator
"""
cache = kwargs.pop('lazy_cache', None)
uncs = []
for i, func in enumerate(self._funcs):
sig = func.signature
args = tuple(kwargs[input] for input in sig)
if isinstance(args[0], awkward.highlevel.Array):
uncs.append(awkward.virtual(func, args=args, length=len(args[0]), cache=cache))
elif isinstance(args[0], numpy.ndarray):
uncs.append(func(*args)) # np is non-lazy
else:
raise Exception('Unknown array library for inputs.')
return zip(self._levels, uncs)
def build(self, MET, corrected_jets, lazy_cache):
if lazy_cache is None:
raise Exception('CorrectedMETFactory requires a awkward-array cache to function correctly.')
if (not isinstance(MET, awkward.highlevel.Array) or
not isinstance(corrected_jets, awkward.highlevel.Array)):
raise Exception("'MET' and 'corrected_jets' must be an awkward array of some kind!")
out = copy(MET)
form = out[self.name_map['METpt']].layout.form
length = len(out)
orig_jets = copy(corrected_jets)
orig_jets[self.name_map['JetPt']] = orig_jets[self.name_map['ptRaw']]
orig_jets[self.name_map['JetMass']] = orig_jets[self.name_map['massRaw']]
out['x_orig'] = getattr(out, self.name_map['METx'])
out['y_orig'] = getattr(out, self.name_map['METy'])
out[self.name_map['METpt'] + '_orig'] = out[self.name_map['METpt']]
out[self.name_map['METphi'] + '_orig'] = out[self.name_map['METphi']]
def corrected_met_cartesian(met, rawJets, corrJets, dim):
return met[f'{dim}_orig'] - awkward.sum(getattr(rawJets, dim) - getattr(corrJets, dim), axis=-1)
def corrected_met_cartesian_unc(met, rawJets, corrJets, dimMET, dimJets):
return getattr(met, dimMET) - awkward.sum(getattr(rawJets, dimJets) - getattr(corrJets, dimJets), axis=-1)
out['corrected_met_x'] = awkward.virtual(
corrected_met_cartesian,
args=(out, orig_jets, corrected_jets, self.name_map['JETx']),
length=length, form=form, cache=lazy_cache
)
out['corrected_met_y'] = awkward.virtual(
corrected_met_cartesian,
args=(out, orig_jets, corrected_jets, self.name_map['JETy']),
length=length, form=form, cache=lazy_cache
)
out[self.name_map['METpt']] = awkward.virtual(
lambda met: numpy.hypot(met['corrected_met_x'], met['corrected_met_y']),
args=(out, ),
length=length,
form=form,
cache=lazy_cache
)
out[self.name_map['METphi']] = awkward.virtual(
lambda met: numpy.arctan2(met['corrected_met_y'], met['corrected_met_x']),
args=(out, ),
length=length,
form=form,
cache=lazy_cache
)
def make_unclustered_variant(themet, op, deltaX, deltaY):
variant = copy(themet)
variant['corrected_met_x'] = awkward.virtual(
lambda met: op(out['corrected_met_x'], out[f'{deltaX}']),
args=(out, ),
length=length,
form=form,
cache=lazy_cache
)
variant['corrected_met_y'] = awkward.virtual(
lambda met: op(out['corrected_met_y'], out[f'{deltaY}']),
args=(out, ),
length=length,
form=form,
cache=lazy_cache
)
variant[self.name_map['METpt']] = awkward.virtual(
lambda met: numpy.hypot(out['corrected_met_x'], out['corrected_met_y']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
variant[self.name_map['METphi']] = awkward.virtual(
lambda met: numpy.arctan2(out['corrected_met_y'], out['corrected_met_x']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
return variant
unclus_up = make_unclustered_variant(MET, lambda x, y: x + y,
self.name_map['UnClusteredEnergyDeltaX'],
self.name_map['UnClusteredEnergyDeltaY'])
unclus_down = make_unclustered_variant(MET, lambda x, y: x - y,
self.name_map['UnClusteredEnergyDeltaX'],
self.name_map['UnClusteredEnergyDeltaY'])
out['MET_UnclusteredEnergy'] = awkward.zip({'up': unclus_up, 'down': unclus_down},
depth_limit=1,
with_name='METSystematic')
def make_variant(name, variation):
variant = copy(MET)
variant['corrected_met_x'] = awkward.virtual(
corrected_met_cartesian_unc,
args=(out, orig_jets, variation, self.name_map['METx'], self.name_map['JETx']),
length=length, form=form, cache=lazy_cache
)
variant['corrected_met_y'] = awkward.virtual(
corrected_met_cartesian_unc,
args=(out, orig_jets, variation, self.name_map['METy'], self.name_map['JETy']),
length=length, form=form, cache=lazy_cache
)
variant[self.name_map['METpt']] = awkward.virtual(
lambda met: numpy.hypot(met['corrected_met_x'], met['corrected_met_y']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
variant[self.name_map['METphi']] = awkward.virtual(
lambda met: numpy.arctan2(met['corrected_met_y'], met['corrected_met_x']),
args=(variant, ),
length=length,
form=form,
cache=lazy_cache
)
return variant
for unc in filter(lambda x: x.startswith(('JER', 'JES')), awkward.fields(corrected_jets)):
up = make_variant(unc, corrected_jets[unc].up)
down = make_variant(unc, corrected_jets[unc].down)
out[unc] = awkward.zip({'up': up, 'down': down},
depth_limit=1,
with_name='METSystematic')
return out
