1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
use na::RealField;
use crate::aliases::{TVec, TVec3};
use crate::traits::Number;
pub fn cross<T: Number>(x: &TVec3<T>, y: &TVec3<T>) -> TVec3<T> {
x.cross(y)
}
pub fn distance<T: RealField, const D: usize>(p0: &TVec<T, D>, p1: &TVec<T, D>) -> T {
(p1 - p0).norm()
}
pub fn dot<T: Number, const D: usize>(x: &TVec<T, D>, y: &TVec<T, D>) -> T {
x.dot(y)
}
pub fn faceforward<T: Number, const D: usize>(
n: &TVec<T, D>,
i: &TVec<T, D>,
nref: &TVec<T, D>,
) -> TVec<T, D> {
if nref.dot(i) < T::zero() {
n.clone()
} else {
-n.clone()
}
}
pub fn length<T: RealField, const D: usize>(x: &TVec<T, D>) -> T {
x.norm()
}
pub fn magnitude<T: RealField, const D: usize>(x: &TVec<T, D>) -> T {
x.norm()
}
pub fn normalize<T: RealField, const D: usize>(x: &TVec<T, D>) -> TVec<T, D> {
x.normalize()
}
pub fn reflect_vec<T: Number, const D: usize>(i: &TVec<T, D>, n: &TVec<T, D>) -> TVec<T, D> {
let _2 = T::one() + T::one();
i - n * (n.dot(i) * _2)
}
pub fn refract_vec<T: RealField, const D: usize>(
i: &TVec<T, D>,
n: &TVec<T, D>,
eta: T,
) -> TVec<T, D> {
let ni = n.dot(i);
let k = T::one() - eta * eta * (T::one() - ni * ni);
if k < T::zero() {
TVec::<_, D>::zeros()
} else {
i * eta - n * (eta * dot(n, i) + k.sqrt())
}
}