crosstyan
3ce5b564bf
- Add beartype dependency to pyproject.toml and uv.lock - Replace typeguard with beartype in type checking - Create camera module with type-safe camera parameter definitions - Migrate utility function to use beartype and JAX numpy
22 lines
557 B
Python
22 lines
557 B
Python
from typing import Any
|
|
|
|
from jaxtyping import Num, jaxtyped
|
|
from beartype import beartype
|
|
from jax import numpy as jnp, Array
|
|
|
|
|
|
@jaxtyped(typechecker=beartype)
|
|
def calculate_perpendicular_distance(
|
|
point: Num[Array, "2"],
|
|
line: Num[Array, "2 2"],
|
|
) -> jnp.floating[Any]:
|
|
"""
|
|
Calculate the perpendicular distance between a point and a line.
|
|
"""
|
|
line_start, line_end = line
|
|
distance = jnp.linalg.norm(
|
|
jnp.cross(line_end - line_start, line_start - point)
|
|
) / jnp.linalg.norm(line_end - line_start)
|
|
|
|
return distance
|