-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathVector3D.py
37 lines (26 loc) · 1.01 KB
/
Vector3D.py
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
import math
class Vector3D:
def __init__(self, x=0.0, y=0.0, z=0.0):
self.x = x;
self.y = y;
self.z = z;
def __str__(self):
return "({},{},{})".format(self.x, self.y, self.z)
def dot(self, target):
return self.x * target.x + self.y * target.y + self.z * target.z
def mag(self):
return math.sqrt(self.dot(self))
def norm(self):
return self/self.mag()
def __add__(self, target):
return Vector3D(self.x + target.x, self.y + target.y, self.z + target.z)
def __sub__(self, target):
return Vector3D(self.x - target.x, self.y - target.y, self.z - target.z)
def __mul__(self, num):
assert not isinstance(num, Vector3D)
return Vector3D(self.x * num, self.y * num, self.z * num)
def __rmul__(self, num):
return self.__mul__(num)
def __truediv__(self, num):
assert not isinstance(num, Vector3D)
return Vector3D(self.x / num, self.y / num, self.z / num)