Added setOrientation functionality to driving domain

Author: Eric-Vin

src/scenic/domains/driving/actions.py

@@ -18,6 +18,7 @@
 import numpy as np
 
 from scenic.core.simulators import Action
+from scenic.core.type_support import toOrientation
 from scenic.core.vectors import Vector
 
 ## Mixin classes indicating support for various types of actions.
@@ -108,6 +109,16 @@ def applyTo(self, obj, sim):
         obj.setVelocity(vel)
 
 
+class SetOrientationAction(Action):
+    """Set the orientation of an agent."""
+
+    def __init__(self, orientation):
+        self.orientation = toOrientation(orientation)
+
+    def applyTo(self, obj, sim):
+        obj.setOrientation(self.orientation)
+
+
 ## Actions available to vehicles which can steer
 
 

src/scenic/domains/driving/model.scenic

@@ -270,6 +270,9 @@ class DrivingObject:
     def setVelocity(self, vel):
         raise NotImplementedError
 
+    def setOrientation(self, orientation):
+        raise NotImplementedError
+
 class Vehicle(DrivingObject):
     """Vehicles which drive, such as cars.