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(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(
rtol: typing.Any,
atol: typing.Any,
equal_nan: typing.Any,
v1: typing.Any,
v2: typing.Any,
) -> typing.Any:
function, *returns = _from_signature(
__name__,
dispatch_map,
(
_aztype(v1),
_aztype(v2),
),
)
with numpy.errstate(all="ignore"):
return _handler_of(v1, v2)._wrap_result(
_flavor_of(v1, v2),
function(
_lib_of(v1, v2),
rtol,
atol,
equal_nan,
*v1.azimuthal.elements,
*v2.azimuthal.elements,
),
returns,
2,
)
