From 478d45e558d5354a6696f24feb6713870438f557 Mon Sep 17 00:00:00 2001
From: fheinrich <fheinrich@techfak.uni-bielefeld.de>
Date: Fri, 15 Dec 2023 14:31:47 +0100
Subject: [PATCH] Flag for drawing interaction range
---
overcooked_simulator/pygame_gui/pygame_gui.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/overcooked_simulator/pygame_gui/pygame_gui.py b/overcooked_simulator/pygame_gui/pygame_gui.py
index 07056812..794c2773 100644
--- a/overcooked_simulator/pygame_gui/pygame_gui.py
+++ b/overcooked_simulator/pygame_gui/pygame_gui.py
@@ -20,6 +20,7 @@ from overcooked_simulator.pygame_gui.game_colors import colors, Color
from overcooked_simulator.simulation_runner import Simulator
PLAYER_DEBUG_VIZ = True
+SHOW_INTERACTION_RANGE = False
class PlayerKeySet:
@@ -87,7 +88,7 @@ class PyGameGUI:
def create_player_colors(self) -> list[Color]:
number_player = len(self.simulator.env.players)
- hue_values = np.linspace(0, 1, number_player+1)
+ hue_values = np.linspace(0, 1, number_player + 1)
colors_vec = np.array([col for col in colors.values()])
@@ -180,9 +181,10 @@ class PyGameGUI:
pygame.draw.circle(self.screen, color2, pos, size)
pygame.draw.circle(self.screen, BLUE, pos, size, width=1)
- pygame.draw.circle(
- self.screen, BLUE, pos, player.interaction_range, width=1
- )
+ if SHOW_INTERACTION_RANGE:
+ pygame.draw.circle(
+ self.screen, BLUE, pos, player.interaction_range, width=1
+ )
pygame.draw.circle(self.screen, colors[color1], pos, size // 2)
pos = player.pos
--
GitLab