liblaf.jarp.warp.types ¶

def __getattr__(name: str) -> type:
    """Resolve dynamic shorthand names such as `floating`, `vec3`, or `mat33`."""
    if name in {"float", "float_"}:
        warnings.warn(
            f"{__name__}.{name} is deprecated, use {__name__}.floating instead",
            DeprecationWarning,
            stacklevel=2,
        )
        return _floating()
    if name == "floating":
        return _floating()
    if (result := re.fullmatch(r"vec(?P<length>[1-9])", name)) is not None:
        length = int(result.group("length"))
        return wp.types.vector(length, _floating())
    if (result := re.fullmatch(r"mat(?P<rows>[1-9])(?P<cols>[1-9])", name)) is not None:
        rows = int(result.group("rows"))
        cols = int(result.group("cols"))
        return wp.types.matrix((rows, cols), _floating())
    msg: str = f"module '{__name__}' has no attribute '{name}'"
    raise AttributeError(msg, name=name, obj=sys.modules[__name__])

def matrix(shape: tuple[int, int]) -> type:
    """Build a Warp matrix dtype using the default floating scalar type."""
    return wp.types.matrix(shape, _floating())

def vector(length: int) -> type:
    """Build a Warp vector dtype using the default floating scalar type."""
    return wp.types.vector(length, _floating())