Fixed bubble rendering issue, added ConfigurableZone

CHANGELOG.md

@@ -15,6 +15,7 @@ Copy and pasting the git commit messages is __NOT__ enough.
 - Added ego's mission details into the `FormatObs` wrapper.
 - Added `SmartsLaneChangingModel` and `SmartsJunctionModel` to types available for use with the new smarts traffic engine within Scenario Studio.
 - Added option to `AgentInterface` to include traffic signals (lights) in `EgoVehicleObservation` objects.
+- Added `ConfigurableZone` for `Zone` object to types which enable users to build bubble by providing coordinates of the polygon.
 
 ### Deprecated
 - Deprecated a few things related to traffic in the `Scenario` class, including the `route` argument to the `Scenario` initializer, the `route`, `route_filepath` and `route_files_enabled` properties, and the `discover_routes()` static method.  In general, the notion of "route" (singular) here is being replaced with "`traffic_specs`" (plural) that allow for specifying traffic controlled by the SMARTS engine as well as Sumo.

envision/web/src/helpers/state_unpacker.js

@@ -33,9 +33,9 @@ const WorldState = Object.freeze({
   SCENARIO_ID: 1,
   SCENARIO_NAME: 2,
   TRAFFIC: 3,
-  BUBBLES: 4,
-  SCORES: 5,
-  EGO_AGENT_IDS: 6,
+  BUBBLES: 5,
+  SCORES: 6,
+  EGO_AGENT_IDS: 7,
 });
 
 const Traffic = Object.freeze({

smarts/sstudio/types.py

@@ -27,7 +27,7 @@
 from dataclasses import dataclass, field
 from enum import IntEnum
 from sys import maxsize
-from typing import Any, Callable, Dict, NewType, Optional, Sequence, Tuple, Union
+from typing import Any, Callable, Dict, NewType, Optional, Sequence, Tuple, Union, List
 
 import numpy as np
 from shapely.affinity import rotate as shapely_rotate
@@ -38,6 +38,7 @@
     MultiPolygon,
     Point,
     Polygon,
+    box,
 )
 from shapely.ops import split, unary_union
 
@@ -653,7 +654,7 @@ class GroupedLapMission:
 class Zone:
     """The base for a descriptor that defines a capture area."""
 
-    def to_geometry(self, road_map: RoadMap) -> Polygon:
+    def to_geometry(self, road_map: Optional[RoadMap] = None) -> Polygon:
         """Generates the geometry from this zone."""
         raise NotImplementedError
 
@@ -814,6 +815,51 @@ def to_geometry(self, road_map: Optional[RoadMap] = None) -> Polygon:
         return shapely_translate(poly, xoff=x, yoff=y)
 
 
+@dataclass(frozen=True)
+class ConfigurableZone(Zone):
+    """A descripter that defines a specific configurableZone defined by user"""
+
+    ext_coordinates: List[Tuple[float, float]]
+    """external coordinates of the polygon
+    < 2 points provided: error
+    = 2 points provided: generates a box using these two points as diagonal
+    > 2 points provided: generates a polygons according to the coordinates"""
+    rotation: Optional[float] = None
+    """The heading direction of the bubble(radians, clock-wise rotation)"""
+
+    def __post_init__(self):
+        if not self.ext_coordinates or not isinstance(self.ext_coordinates[0], tuple):
+            raise ValueError(
+                "Two points or more are needed to create a polygon. (less than two points are provided)"
+            )
+
+        x_set = set(point[0] for point in self.ext_coordinates)
+        y_set = set(point[1] for point in self.ext_coordinates)
+        if len(x_set) == 1 or len(y_set) == 1:
+            raise ValueError(
+                "Parallel line cannot form a polygon. (points provided form a parallel line)"
+            )
+
+    def to_geometry(self, road_map: Optional[RoadMap] = None) -> Polygon:
+        """Generate a polygon according to given coordinates"""
+        poly = None
+        if (
+            len(self.ext_coordinates) == 2
+        ):  # if user only specified two points, create a box
+            x_min = min(self.ext_coordinates[0][0], self.ext_coordinates[1][0])
+            x_max = max(self.ext_coordinates[0][0], self.ext_coordinates[1][0])
+            y_min = min(self.ext_coordinates[0][1], self.ext_coordinates[1][1])
+            y_max = max(self.ext_coordinates[0][1], self.ext_coordinates[1][1])
+            poly = box(x_min, y_min, x_max, y_max)
+
+        else:  # else create a polygon according to the coordinates
+            poly = Polygon(self.ext_coordinates)
+
+        if self.rotation is not None:
+            poly = shapely_rotate(poly, self.rotation, use_radians=True)
+        return
+
+
 @dataclass(frozen=True)
 class BubbleLimits:
     """Defines the capture limits of a bubble."""
@@ -890,6 +936,12 @@ def __post_init__(self):
                 "Only boids can have keep_alive enabled (for persistent boids)"
             )
 
+        poly = None
+        if not isinstance(self.zone, MapZone):
+            poly = self.zone.to_geometry(road_map=None)
+        if poly is not None and (poly.is_valid == False):
+            raise ValueError("The Zone Polygon is not a valid closed loop")
+
     @staticmethod
     def to_actor_id(actor, mission_group):
         """Mashes the actor id and mission group to create what needs to be a unique id."""