Lock around order drawing

overcooked_simulator/gui_2d_vis/overcooked_gui.py

@@ -576,44 +576,44 @@ class PyGameGUI:
         pygame.draw.rect(order_screen, bg_color, order_screen.get_rect())
 
         order_rects_start = (orders_height // 2) - (self.grid_size // 2)
+        with self.simulator.env.lock:
+            for idx, order in enumerate(state["orders"]):
+                order: Order
+                order_upper_left = [
+                    order_rects_start + idx * self.grid_size * 1.5,
+                    order_rects_start,
+                ]
+                pygame.draw.rect(
+                    order_screen,
+                    colors["red"],
+                    pygame.Rect(
+                        order_upper_left[0],
+                        order_upper_left[1],
+                        self.grid_size,
+                        self.grid_size,
+                    ),
+                    width=2,
+                )
+                center = np.array(order_upper_left) + np.array(
+                    [self.grid_size / 2, self.grid_size / 2]
+                )
+                self.draw_thing(
+                    center,
+                    self.visualization_config["Plate"]["parts"],
+                    screen=order_screen,
+                )
+                self.draw_item(
+                    center,
+                    order.meal,
+                    plate=True,
+                    screen=order_screen,
+                )
+                order_done_seconds = (
+                    (order.start_time + order.max_duration) - state["env_time"]
+                ).total_seconds()
 
-        for idx, order in enumerate(state["orders"]):
-            order: Order
-            order_upper_left = [
-                order_rects_start + idx * self.grid_size * 1.5,
-                order_rects_start,
-            ]
-            pygame.draw.rect(
-                order_screen,
-                colors["red"],
-                pygame.Rect(
-                    order_upper_left[0],
-                    order_upper_left[1],
-                    self.grid_size,
-                    self.grid_size,
-                ),
-                width=2,
-            )
-            center = np.array(order_upper_left) + np.array(
-                [self.grid_size / 2, self.grid_size / 2]
-            )
-            self.draw_thing(
-                center,
-                self.visualization_config["Plate"]["parts"],
-                screen=order_screen,
-            )
-            self.draw_item(
-                center,
-                order.meal,
-                plate=True,
-                screen=order_screen,
-            )
-            order_done_seconds = (
-                (order.start_time + order.max_duration) - state["env_time"]
-            ).total_seconds()
-
-            percentage = order_done_seconds / order.max_duration.total_seconds()
-            self.draw_progress_bar(center, percentage, screen=order_screen)
+                percentage = order_done_seconds / order.max_duration.total_seconds()
+                self.draw_progress_bar(center, percentage, screen=order_screen)
 
         orders_rect = order_screen.get_rect()
         orders_rect.center = [