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
|
use std::fmt;
use std::ops;
#[derive(Debug, Clone, Copy)]
pub struct Vector {
pub x: f32,
pub y: f32,
pub z: f32,
}
impl fmt::Display for Vector {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "({:.2}, {:.2}, {:.2})", self.x, self.y, self.z)
}
}
impl ops::Add for Vector {
type Output = Vector;
fn add(self, rhs: Self) -> Self::Output {
Vector {
x: self.x + rhs.x,
y: self.y + rhs.y,
z: self.z + rhs.z,
}
}
}
impl ops::Sub for Vector {
type Output = Vector;
fn sub(self, rhs: Self) -> Self::Output {
Vector {
x: self.x - rhs.x,
y: self.y - rhs.y,
z: self.z - rhs.z,
}
}
}
impl ops::Mul<f32> for Vector {
type Output = Vector;
fn mul(self, rhs: f32) -> Self::Output {
Vector {
x: self.x * rhs,
y: self.y * rhs,
z: self.z * rhs,
}
}
}
impl ops::Div<f32> for Vector {
type Output = Vector;
fn div(self, rhs: f32) -> Self::Output {
Vector {
x: self.x / rhs,
y: self.y / rhs,
z: self.z / rhs,
}
}
}
impl Vector {
pub fn magnitude_squared(self) -> f32 {
self.x * self.x + self.y * self.y + self.z * self.z
}
pub fn magnitude(self) -> f32 {
self.magnitude_squared().sqrt()
}
pub fn normalized(self) -> Vector {
let mag = self.magnitude();
self / mag
}
pub fn rotate_z(self, angle: f32) -> Vector {
// TODO: multiply by a matirx
Vector {
x: self.x * angle.cos() + self.y * angle.sin(),
y: self.y * angle.cos() - self.x * angle.sin(),
z: self.z,
}
}
}
|
