161 lines
6.9 KiB
Python
161 lines
6.9 KiB
Python
import pygame
|
|
from .colors import *
|
|
from .fonts import *
|
|
from .knob import *
|
|
import random
|
|
|
|
class Knobs:
|
|
|
|
def __init__(self, gui, knob_names, is_left_side=True):
|
|
|
|
self.gui = gui
|
|
self.knobs_surface = pygame.Surface((gui.screenW, gui.screenH), pygame.SRCALPHA)
|
|
|
|
self.opacity = 0
|
|
self.fade_speed = 5
|
|
self.fade_in = False
|
|
self.fade_out = False
|
|
self.knob_names = knob_names
|
|
|
|
self.is_left_side = is_left_side
|
|
|
|
# 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(len(knob_names)):
|
|
self.knobs.append(Knob(self, knob_names[i], self.knobs_radius, (x, y), 0))
|
|
y += self.knobs_radius + knobs_spacing
|
|
if i == 3 or (self.is_left_side and i == 6) or (not self.is_left_side and i == 0):
|
|
x += self.knobs_radius + knobs_spacing
|
|
y = knobs_y
|
|
self.label_vanish_parameter = 0
|
|
self.label_border_height = 0
|
|
|
|
def display_label(self, knob):
|
|
if not knob.vanish_label_path: self.label_vanish_parameter = 0
|
|
if len(knob.label_path) == 2:
|
|
if self.label_vanish_parameter > 0:
|
|
self.label_vanish_parameter -= 1
|
|
elif knob.vanish_label_path: self.label_vanish_parameter = 5
|
|
label_fx_name = font_helvetica.render(knob.name, False, color_primary_light)
|
|
label_fx_value = font_helvetica.render('[ ' + str(int(round(knob.value * 100, 0))) + '% ]',
|
|
False, color_primary_light)
|
|
min_x = self.knobs[5].get_position()[0] + self.knobs_radius * 1.5
|
|
label_x = min_x + (self.gui.screenW - min_x - label_fx_name.get_width()) / 2
|
|
label_y = (self.gui.screenH - label_fx_name.get_height() - label_fx_value.get_height()) / 2
|
|
|
|
if self.label_vanish_parameter == 0:
|
|
self.gui.label_surface.blit(label_fx_name, (label_x, label_y))
|
|
self.gui.label_surface.blit(label_fx_value, (label_x, self.gui.screenH / 2))
|
|
|
|
if knob.vanish_label_path:
|
|
if self.label_border_height < self.gui.screenH / 2:
|
|
self.label_border_height += 2
|
|
else: self.label_border_height = self.gui.screenH / 2
|
|
else:
|
|
self.label_border_height = label_y
|
|
|
|
pygame.draw.line(
|
|
self.gui.label_surface,
|
|
color_primary_dark,
|
|
(label_x - 10, self.label_border_height),
|
|
(label_x - 10, self.gui.screenH - self.label_border_height),
|
|
2
|
|
)
|
|
|
|
self.calculate_label_path(knob, label_x)
|
|
|
|
for i in range(len(knob.label_path)-1):
|
|
pygame.draw.line(
|
|
self.gui.label_surface,
|
|
color_primary_dark,
|
|
knob.label_path[i],
|
|
knob.label_path[i+1],
|
|
2
|
|
)
|
|
|
|
def calculate_label_path(self, knob, label_x):
|
|
if len(knob.label_path) == 0:
|
|
knob.label_path.append(knob.get_position())
|
|
knob_index = knob.get_index(self.knobs)
|
|
if self.is_left_side:
|
|
if knob_index in [0, 1, 4, 5] :
|
|
x = knob.get_position()[0] + self.knobs_radius * 1.5
|
|
y = knob.get_position()[1] + self.knobs_radius * 1.5
|
|
elif knob_index in [2, 3, 6]:
|
|
x = knob.get_position()[0] + self.knobs_radius * 1.5
|
|
y = knob.get_position()[1] - self.knobs_radius * 1.5
|
|
else: # knob 7
|
|
x = knob.get_position()[0] - self.knobs_radius * 1.5
|
|
y = knob.get_position()[1] + self.knobs_radius * 1.5
|
|
knob.label_path.append((x, y))
|
|
|
|
if (knob == self.knobs[0] or knob == self.knobs[3]):
|
|
knob.label_path.append((knob.get_position()[0] + self.knobs_radius * 1.5, self.gui.screenH/2))
|
|
elif (knob == self.knobs[4] or knob == self.knobs[7]):
|
|
knob.label_path.append((self.knobs[4].get_position()[0] + self.knobs_radius * 1.5, self.gui.screenH/2))
|
|
else:
|
|
if knob_index in [1, 2, 4, 5]:
|
|
x = knob.get_position()[0] + self.knobs_radius * 1.5
|
|
y = knob.get_position()[1] + self.knobs_radius * 1.5
|
|
elif knob_index in [3, 6, 7]:
|
|
x = knob.get_position()[0] + self.knobs_radius * 1.5
|
|
y = knob.get_position()[1] - self.knobs_radius * 1.5
|
|
else: # knob 0
|
|
x = knob.get_position()[0] + self.knobs_radius * 1.5
|
|
y = knob.get_position()[1] + self.knobs_radius * 1.5
|
|
knob.label_path.append((x, y))
|
|
|
|
if (knob == self.knobs[0] or knob == self.knobs[1]):
|
|
knob.label_path.append((knob.get_position()[0] + self.knobs_radius * 1.5, self.gui.screenH/2))
|
|
elif (knob == self.knobs[4] or knob == self.knobs[7]):
|
|
knob.label_path.append((knob.get_position()[0] + self.knobs_radius * 1.5, self.gui.screenH/2))
|
|
|
|
knob.label_path.append((label_x - 10, self.gui.screenH/2))
|
|
|
|
elif knob.vanish_label_path:
|
|
if len(knob.label_path) > 1:
|
|
p1 = list(knob.label_path[0])
|
|
p2 = list(knob.label_path[1])
|
|
if abs(p1[0] - p2[0]) <= 10 and abs(p1[1] - p2[1]) <= 10:
|
|
knob.label_path.remove(tuple(p1))
|
|
else:
|
|
if p1[0] > p2[0]:
|
|
p1[0] -= 10
|
|
elif p1[0] < p2[0]:
|
|
p1[0] += 10
|
|
if p1[1] > p2[1]:
|
|
p1[1] -= 10
|
|
elif p1[1] < p2[1]:
|
|
p1[1] += 10
|
|
knob.label_path[0] = tuple(p1)
|
|
else:
|
|
knob.vanish_label_path = False
|
|
knob.label_path = []
|
|
|
|
def set_focused(self, knob):
|
|
for k in self.knobs:
|
|
k.set_focused(k == knob)
|
|
if knob is None:
|
|
k.vanish_label_path = False
|
|
|
|
def set_value(self, index, value):
|
|
self.set_focused(self.knobs[index])
|
|
self.knobs[index].set_value(value)
|
|
|
|
def display(self):
|
|
if self.gui.show_knobs:
|
|
if self.fade_in and self.opacity < 255: self.opacity += self.fade_speed
|
|
if self.fade_out and self.opacity > 0: self.opacity -= self.fade_speed
|
|
self.knobs_surface = pygame.Surface((self.gui.screenW, self.gui.screenH), pygame.SRCALPHA)
|
|
for knob in self.knobs:
|
|
if knob.focused or knob.vanish_label_path: self.display_label(knob)
|
|
knob.display(self.knobs_surface)
|
|
|
|
self.gui.screen.blit(self.knobs_surface, (0,0))
|