Add magnitude to vector

Author: hadiTab

lgsvl/geometry.py

@@ -3,6 +3,7 @@
 #
 # This software contains code licensed as described in LICENSE.
 #
+from math import sqrt
 
 class Vector:
   def __init__(self, x = 0.0, y = 0.0, z = 0.0):
@@ -47,6 +48,9 @@ def __mul__(self, v):
   def __rmul__(self, v):
     return self * v
 
+  def magnitude(self):
+    return sqrt(self.x**2 + self.y**2 + self.z**2)
+
 class BoundingBox:
   def __init__(self, min, max):
     self.min = min