def dispatch(v):
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v),
_ltype(v),
_ttype(v),
),
)
with numpy.errstate(all="ignore"):
return v._wrap_result(
function(v.lib, *v.azimuthal.elements, *v.longitudinal.elements,
*v.temporal.elements),
returns,
)
def dispatch(u, i, j, k, vec):
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(vec),
_ltype(vec),
),
)
with numpy.errstate(all="ignore"):
return vec._wrap_result(
function(
vec.lib, u, i, j, k, *vec.azimuthal.elements, *vec.longitudinal.elements
),
returns,
)
def dispatch(v1, v2):
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v1),
_aztype(v2),
),
)
with numpy.errstate(all="ignore"):
return _handler((v1, v2))._wrap_result(
function(
_lib_of((v1, v2)),
*v1.azimuthal.elements,
*v2.azimuthal.elements,
),
returns,
)
def dispatch(obj, v):
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v),
_ltype(v),
_ttype(v),
),
)
with numpy.errstate(all="ignore"):
return v._wrap_result(
function(v.lib, obj["xx"], obj["xy"], obj["xz"], obj["xt"],
obj["yx"], obj["yy"], obj["yz"], obj["yt"], obj["zx"],
obj["zy"], obj["zz"], obj["zt"], obj["tx"], obj["ty"],
obj["tz"], obj["tt"], *v.azimuthal.elements,
*v.longitudinal.elements, *v.temporal.elements),
returns,
)
def dispatch(factor, v):
function, *returns = _from_signature(__name__, dispatch_map, (_aztype(v),))
with numpy.errstate(all="ignore"):
return v._wrap_result(function(v.lib, factor, *v.azimuthal.elements), returns)
