created a knobs module to outsource the grouping of the knobs

2025-02-28 13:16:39 +01:00
parent 043c847801
commit 06006194a8
3 changed files with 94 additions and 87 deletions
--- a/knobs.py
+++ b/knobs.py
@ -0,0 +1,83 @@
+import pygame
+from colors import *
+from fonts import *
+from knob import *
+
+class Knobs:
+
+    def __init__(self, gui):
+
+        self.gui = gui
+
+        knob_names = [
+            'Delay Time',
+            'Delay Vol',
+            'Reverb Wet',
+            'Reverb Size',
+            'Filter Freq',
+            'Distortion',
+            'Detune',
+            'Gain'
+        ]
+
+        # Create and position the knobs
+        self.knobs_radius = self.gui.screenW / 25
+        knobs_spacing = self.knobs_radius * 2
+        knobs_x = (self.gui.screenW - (self.knobs_radius + knobs_spacing) * 2) / 2
+        x = knobs_x
+        knobs_y = (self.gui.screenH - (self.knobs_radius + knobs_spacing) * 3) / 2
+        y = knobs_y
+        self.knobs = []
+        for i in range(8):
+            self.knobs.append(Knob(self, knob_names[i], self.knobs_radius, (x, y)))
+            y += self.knobs_radius + knobs_spacing
+            if i == 3 or i == 6:
+                x += self.knobs_radius + knobs_spacing
+                y = knobs_y
+
+    def display_label(self, knob):
+        label_fx_name = font_helvetica.render(knob.name, False, color_primary_light)
+        label_fx_value = font_helvetica.render('[ ' + str(knob.value) + '% ]',
+                                        False, color_primary_light)
+
+        label_x = self.gui.screenW - label_fx_name.get_width() - 20
+        label_y = (self.gui.screenH - label_fx_name.get_height() - label_fx_value.get_height()) / 2
+        self.gui.screen.blit(label_fx_name, (label_x, label_y))
+        self.gui.screen.blit(label_fx_value, (label_x, self.gui.screenH / 2))
+        pygame.draw.line(self.gui.screen, color_primary_dark, (label_x - 10, label_y), (label_x - 10, self.gui.screenH - label_y), 2)
+        
+        line_points = []
+        line_points.append(knob.get_position())
+
+        if (knob == self.knobs[0] or knob == self.knobs[1] or
+            knob == self.knobs[4] or knob == self.knobs[5]):
+            x = knob.get_position()[0] + self.knobs_radius * 1.5
+            y = knob.get_position()[1] + self.knobs_radius * 1.5
+            line_points.append((x, y))
+        elif (knob == self.knobs[2] or knob == self.knobs[3] or knob == self.knobs[6]):
+            x = knob.get_position()[0] + self.knobs_radius * 1.5
+            y = knob.get_position()[1] - self.knobs_radius * 1.5
+            line_points.append((x, y))
+        elif (knob == self.knobs[7]):
+            x = knob.get_position()[0] - self.knobs_radius * 1.5
+            y = knob.get_position()[1] + self.knobs_radius * 1.5
+            line_points.append((x, y))
+
+        if (knob == self.knobs[0] or knob == self.knobs[3]):
+            line_points.append((self.knobs[0].get_position()[0] + self.knobs_radius * 1.5, self.gui.screenH/2))
+        elif (knob == self.knobs[4] or knob == self.knobs[6] or knob == self.knobs[7]):
+            line_points.append((self.knobs[4].get_position()[0] + self.knobs_radius * 1.5, self.gui.screenH/2))
+
+        line_points.append((label_x - 10, self.gui.screenH/2))
+
+        for i in range(len(line_points)-1):
+            pygame.draw.line(self.gui.screen, color_primary_dark, line_points[i], line_points[i+1], 2)
+
+    def deactivate_knobs_except(self, knob):
+        for k in self.knobs:
+            if k != knob: k.set_focused(False)
+
+    def display(self):
+        for knob in self.knobs:
+            if knob.focused: self.display_label(knob)
+            knob.display(self.gui.screen)