Fix lines

metadrive/component/lane/extension_lane.py

@@ -20,6 +20,15 @@ def width_at(self, longitudinal: float) -> float:
         else:
             return self.width - (longitudinal / self.length) * self.width
 
+    def get_polyline(self, interval=2, lateral=0):
+        ret = []
+        for i in np.arange(0, self.length, interval):
+            max_lateral = self.width_at(i) - self.width / 2
+            ret.append(self.position(i, min(lateral, max_lateral)))
+        last_lateral = self.width_at(self.length) - self.width / 2
+        ret.append(self.position(self.length, min(lateral, last_lateral)))
+        return np.array(ret)
+
     @property
     def polygon(self):
         if self._polygon is not None:

metadrive/component/map/pg_map.py

@@ -143,7 +143,7 @@ def get_boundary_line_vector(self, interval):
                             continue
                         color = l.line_colors[side]
                         line_type = self.get_line_type(type, color)
-                        lateral = l.width_at(0) / 2
+                        lateral = l.width / 2
                         if side == 0:
                             lateral *= -1
                         ret["{}_{}".format(l.index, side)] = {

metadrive/component/pgblock/create_pg_block_utils.py

@@ -28,7 +28,7 @@ def create_extension(
 ):
     extension_end = start_position + [extension_lane_length, 0.0]
     extension = ExtendingLane(
-        direction, start_position, extension_end, width, [PGLineType.NONE, PGLineType.NONE], forbidden, speed_limit,
+        direction, start_position, extension_end, width, [PGLineType.NONE, PGLineType.CONTINUOUS], forbidden, speed_limit,
         priority
     )
     return extension