Add destinations to spawn points

Author: hadiTab

lgsvl/geometry.py

@@ -105,3 +105,29 @@ def to_json(self):
 
     def __repr__(self):
         return "Transform(position={}, rotation={})".format(self.position, self.rotation)
+
+
+class Spawn:
+    def __init__(self, transform=None, destinations=None):
+        if transform is None: transform = Transform()
+        if destinations is None: destinations = []
+        self.position = transform.position
+        self.rotation = transform.rotation
+        self.destinations = destinations
+
+    @staticmethod
+    def from_json(j):
+        spawn_point = Transform.from_json(j)
+        destinations = []
+        for d in j["destinations"]:
+            destinations.append(Transform.from_json(d))
+
+        return Spawn(spawn_point, destinations)
+
+    def to_json(self):
+        return {"position": self.position.to_json(), "rotation": self.rotation.to_json()}
+
+    def __repr__(self):
+        return "Spawn(position={}, rotation={}, destinations={})".format(
+            self.transform.position, self.transform.rotation, self.destinations
+        )

lgsvl/simulator.py

@@ -7,7 +7,7 @@
 from .remote import Remote
 from .agent import Agent, AgentType, AgentState
 from .sensor import GpsData
-from .geometry import Vector, Transform
+from .geometry import Vector, Transform, Spawn
 from .utils import accepts, ObjectState
 from .controllable import Controllable
 
@@ -171,7 +171,7 @@ def set_time_of_day(self, time, fixed=True):
 
     def get_spawn(self):
         spawns = self.remote.command("map/spawn/get")
-        return [Transform.from_json(spawn) for spawn in spawns]
+        return [Spawn.from_json(spawn) for spawn in spawns]
 
     @accepts(Transform)
     def map_to_gps(self, transform):