use num_traits::AsPrimitive;
pub fn squared_norm_<T>(p: &[T]) -> T
where T: std::ops::Mul<Output=T> + std::ops::Add<Output=T> + Copy
{
assert_eq!(p.len(), 3);
p[0] * p[0] + p[1] * p[1] + p[2] * p[2]
}
pub fn norm_<T>(
v: &[T]) -> T
where T: num_traits::Float
{
(v[0] * v[0] + v[1] * v[1] + v[2] * v[2]).sqrt()
}
pub fn normalize_<T>(
v: &mut [T]) -> T
where T: num_traits::Float + std::ops::MulAssign
{
let l = norm_(v);
let linv = T::one() / l;
v[0] *= linv;
v[1] *= linv;
v[2] *= linv;
l
}
pub fn cross_mut_<T>(
vo: &mut [T],
v1: &[T],
v2: &[T])
where T: std::ops::Mul<Output=T> + std::ops::Sub<Output=T> + Copy
{
vo[0] = v1[1] * v2[2] - v2[1] * v1[2];
vo[1] = v1[2] * v2[0] - v2[2] * v1[0];
vo[2] = v1[0] * v2[1] - v2[0] * v1[1];
}
pub fn cross_<T>(
v1: &[T],
v2: &[T]) -> [T; 3]
where T: std::ops::Mul<Output=T> + std::ops::Sub<Output=T> + Copy
{
[
v1[1] * v2[2] - v2[1] * v1[2],
v1[2] * v2[0] - v2[2] * v1[0],
v1[0] * v2[1] - v2[0] * v1[1]]
}
pub fn dot_<T>(
a: &[T],
b: &[T]) -> T
where T: std::ops::Mul<Output=T> + std::ops::Add<Output=T> + Copy
{
a[0] * b[0] + a[1] * b[1] + a[2] * b[2]
}
pub fn sub_<T>(
a: &[T],
b: &[T]) -> [T; 3]
where T: std::ops::Sub<Output=T> + Copy
{
[a[0] - b[0], a[1] - b[1], a[2] - b[2]]
}
pub fn scale_<T>(
a: &[T],
s: T) -> [T; 3]
where T: Copy + std::ops::Mul<Output=T> {
[s * a[0], s * a[1], s * a[2]]
}
pub fn distance_<T>(
p0: &[T],
p1: &[T]) -> T
where T: num_traits::Float
{
let v0 = p1[0] - p0[0];
let v1 = p1[1] - p0[1];
let v2 = p1[2] - p0[2];
(v0 * v0 + v1 * v1 + v2 * v2).sqrt()
}
pub fn frame_from_z_vector<T>(
vec_n: nalgebra::Vector3::<T>) -> (nalgebra::Vector3::<T>, nalgebra::Vector3::<T>)
where T: nalgebra::RealField + 'static + Copy,
f64: num_traits::AsPrimitive<T>
{
let vec_s = nalgebra::Vector3::<T>::new(T::zero(), T::one(), T::zero() );
let mut vec_x = vec_s.cross(&vec_n);
let len = vec_x.norm();
if len < 1.0e-10_f64.as_() {
let vec_t = nalgebra::Vector3::<T>::new( T::one(), T::zero(), T::zero() );
let vec_x = vec_t.cross(&vec_n);
let vec_y = vec_n.cross(&vec_x);
(vec_x, vec_y)
} else {
let invlen = T::one() / len;
vec_x *= invlen;
let vec_y = vec_n.cross(&vec_x);
(vec_x, vec_y)
}
}