From 6d0573723e3299b409d1ba52c07adeefcc1a8649 Mon Sep 17 00:00:00 2001
From: fheinrich <fheinrich@techfak.uni-bielefeld.de>
Date: Mon, 29 Jan 2024 10:24:39 +0100
Subject: [PATCH] Readded colors for chefs
---
overcooked_simulator/gui_2d_vis/drawing.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/overcooked_simulator/gui_2d_vis/drawing.py b/overcooked_simulator/gui_2d_vis/drawing.py
index d0eaa7e2..c257730d 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
--
GitLab