Debugging

lib/objects/trajectory.py

@@ -2,8 +2,9 @@ from math import atan2
 from typing import List, Tuple, Union
 
 from matplotlib import pyplot as plt
-from lib.objects import variables as V
+from lib import variables as V
 
+import numpy as np
 
 class Trajectory(object):
 
@@ -31,6 +32,10 @@ class Trajectory(object):
     def as_paired_list(self):
         return list(zip(self.vertices[:-1], self.vertices[1:]))
 
+    @property
+    def np_vertices(self):
+        return [np.array(vertice) for vertice in self.vertices]
+
     def __init__(self, vertices: Union[List[Tuple[int]], None] = None):
         assert any((isinstance(vertices, list), vertices is None))
         if vertices is not None: