From e859e021de9400a8dfe338592e11412a5c8ca174 Mon Sep 17 00:00:00 2001
From: Correl Roush <correl@gmail.com>
Date: Sat, 26 Sep 2020 01:29:40 -0400
Subject: [PATCH] Make colors configurable

---
 turntable.sample.json |  8 +++++++-
 turntable/gui.py      | 26 ++++++++++++++++++--------
 2 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/turntable.sample.json b/turntable.sample.json
index a40c1ce..33b8186 100644
--- a/turntable.sample.json
+++ b/turntable.sample.json
@@ -35,6 +35,12 @@
     },
     "gui": {
         "width": 1280,
-        "height": 720
+        "height": 720,
+        "bars": [
+            [ 0, [  0, 255,   0]],
+            [50, [255, 255,   0]],
+            [75, [255,   0,   0]]
+        ],
+        "lines": [100, 100, 100]
     }
 }
diff --git a/turntable/gui.py b/turntable/gui.py
index 871d31a..697b66f 100644
--- a/turntable/gui.py
+++ b/turntable/gui.py
@@ -25,6 +25,10 @@ class Plot:
         width: int,
         height: int,
         bars: int = 20,
+        bar_colors: List[Tuple[int, Tuple[int, int, int]]] = [
+            (0, (255, 255, 255)),
+        ],
+        lines_color: Tuple[int, int, int] = (128, 128, 128),
     ) -> None:
         self.screen = screen
         self.x = x
@@ -32,6 +36,8 @@ class Plot:
         self.width = width
         self.height = height
         self.bars = bars
+        self.bar_colors = bar_colors
+        self.lines_color = lines_color
         self.audio = models.PCM(44100, 2)
 
     def spectrum(self) -> np.array:
@@ -60,7 +66,7 @@ class Plot:
         for i, line in enumerate(lines):
             pygame.draw.line(
                 self.screen,
-                (128, 128, 128),
+                self.lines_color,
                 (self.x + i * 4, self.height),
                 (self.x + i * 4, self.height - int(line)),
             )
@@ -81,13 +87,8 @@ class Plot:
         light_steps = self.height // light_height // 2
         lights = fft * light_steps / 100
 
-        colors = [
-            (0, (0, 255, 0)),
-            (50, (255, 255, 0)),
-            (75, (255, 0, 0)),
-        ]
-        color_keys = [k for k, v in colors]
-        color_values = [v for k, v in colors]
+        color_keys = [k for k, v in self.bar_colors]
+        color_values = [v for k, v in self.bar_colors]
         for i, steps in enumerate(lights):
             for step in range(int(steps)):
                 color = color_values[bisect(color_keys, step / light_steps * 100) - 1]
@@ -153,6 +154,15 @@ def main():
         width=screen.get_width(),
         height=screen.get_height() - 50,
         bars=15,
+        bar_colors=config.get(
+            "bars",
+            [
+                (0, (0, 255, 0)),
+                (50, (255, 255, 0)),
+                (75, (255, 0, 0)),
+            ],
+        ),
+        lines_color=config.get("lines", (128, 128, 128)),
     )
 
     try: