def dispatch(tolerance: typing.Any, v1: typing.Any,
v2: typing.Any) -> typing.Any:
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v1),
_ltype(v1),
_aztype(v2),
_ltype(v2),
),
)
with numpy.errstate(all="ignore"):
return _handler_of(v1, v2)._wrap_result(
_flavor_of(v1, v2),
function(
_lib_of(v1, v2),
tolerance,
*v1.azimuthal.elements,
*v1.longitudinal.elements,
*v2.azimuthal.elements,
*v2.longitudinal.elements,
),
returns,
2,
)
def dispatch(angle: typing.Any, v1: typing.Any, v2: typing.Any) -> typing.Any:
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v1), # v1 is the axis about which we're rotating
_ltype(v1),
_aztype(v2), # v2 is the primary vector, the one being rotated
_ltype(v2),
),
)
with numpy.errstate(all="ignore"):
return _handler_of(v2)._wrap_result( # note: _handler_of(v2)
_flavor_of(v2), # note: _flavor_of(v2)
function(
_lib_of(v1, v2),
angle,
*v1.azimuthal.elements,
*v1.longitudinal.elements,
*v2.azimuthal.elements,
*v2.longitudinal.elements,
),
returns,
1,
)
def dispatch(v: typing.Any) -> typing.Any:
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v),
_ltype(v),
),
)
with numpy.errstate(all="ignore"):
return v._wrap_result(
_flavor_of(v),
function(v.lib, *v.azimuthal.elements, *v.longitudinal.elements),
returns,
1,
)
def dispatch(obj: typing.Any, v: typing.Any) -> typing.Any:
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v),
_ltype(v),
_ttype(v),
),
)
with numpy.errstate(all="ignore"):
return v._wrap_result(
_flavor_of(v),
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,
1,
)
