Train Active

lib/objects/trajectory.py

@@ -8,43 +8,51 @@ import numpy as np
 
 class Trajectory(object):
 
+    @property
+    def vertices(self):
+        return self._vertices
+
+    @property
+    def xy_vertices(self):
+        return [(x,y) for _, x,y in self._vertices]
+
     @property
     def endpoints(self):
         return self.start, self.dest
 
     @property
     def start(self):
-        return self.vertices[0]
+        return self._vertices[0]
 
     @property
     def dest(self):
-        return self.vertices[-1]
+        return self._vertices[-1]
 
     @property
     def xs(self):
-        return [x[1] for x in self.vertices]
+        return [x[1] for x in self._vertices]
 
     @property
     def ys(self):
-        return [x[0] for x in self.vertices]
+        return [x[0] for x in self._vertices]
 
     @property
     def as_paired_list(self):
-        return list(zip(self.vertices[:-1], self.vertices[1:]))
+        return list(zip(self._vertices[:-1], self._vertices[1:]))
 
     @property
     def np_vertices(self):
-        return [np.array(vertice) for vertice in self.vertices]
+        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:
-            self.vertices = vertices
+            self._vertices = vertices
         pass
 
     def is_equal_to(self, other_trajectory):
         # ToDo: do further equality Checks here
-        return self.vertices == other_trajectory.vertices
+        return self._vertices == other_trajectory.vertices
 
     def draw(self, highlights=True, label=None, **kwargs):
         if label is not None: