diff --git a/overcooked_simulator/gui_2d_vis/drawing.py b/overcooked_simulator/gui_2d_vis/drawing.py index d0eaa7e2d79e7ba8edf49e6befe8b4252e94c41d..c257730d3be73f30bbacbcbc090a045da3b34eb1 100644 --- a/overcooked_simulator/gui_2d_vis/drawing.py +++ b/overcooked_simulator/gui_2d_vis/drawing.py @@ -144,6 +144,13 @@ class Visualizer: facing = np.array(player_dict["facing_direction"]) if USE_PLAYER_COOK_SPRITES: + pygame.draw.circle( + screen, + self.player_colors[p_idx], + pos - facing * grid_size * 0.25, + grid_size * 0.2, + ) + img_path = self.config["Cook"]["parts"][0]["path"] rel_x, rel_y = facing angle = -np.rad2deg(math.atan2(rel_y, rel_x)) + 90