compute colors only once

vpoulailleau · Jan 15, 2024 · 8fa626c · 8fa626c
1 parent 295d0af
commit 8fa626c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/space_collector/viewer/constants.py b/space_collector/viewer/constants.py
@@ -1,3 +1,5 @@
+import colorsys
+
 SCREEN_HEIGHT = 1000
 SCREEN_WIDTH = 1777
 SCREEN_TITLE = "Space collector"
@@ -9,6 +11,10 @@
     2: 65,
     3: 130,
 }
+TEAM_COLORS = {
+    team: tuple(int(c * 255) for c in colorsys.hsv_to_rgb(hue / 360, 1, 1))
+    for team, hue in TEAM_HUES.items()
+}
 
 MAP_MARGIN = 100
 MAP_MIN_X = SCORE_WIDTH + MAP_MARGIN

diff --git a/space_collector/viewer/planet.py b/space_collector/viewer/planet.py
@@ -1,11 +1,10 @@
 import logging
 import random
-import colorsys
 
 import arcade
 
 from space_collector.viewer.animation import AnimatedValue, Animation
-from space_collector.viewer.constants import TEAM_HUES
+from space_collector.viewer.constants import TEAM_HUES, TEAM_COLORS
 from space_collector.viewer.utils import (
     hue_changed_texture,
     map_coord_to_window_coord,
@@ -37,17 +36,11 @@ def animate(self) -> None:
 
     def draw(self) -> None:
         self.animate()
-        color_rgb = colorsys.hsv_to_rgb(TEAM_HUES[self.team] / 360, 1, 1)
         arcade.draw_circle_outline(
             self.sprite.position[0],
             self.sprite.position[1],
             self.size.value // 2 + 2,
-            (
-                int(color_rgb[0] * 255),
-                int(color_rgb[1] * 255),
-                int(color_rgb[2] * 255),
-                150,
-            ),
+            (*TEAM_COLORS[self.team], 150),
             4,
         )
         self.sprite.draw()