119 lines
4.0 KiB
Python
119 lines
4.0 KiB
Python
import pygame
|
|
|
|
class HorizontalSlider:
|
|
def __init__(self, screen, x, y, width, height, min_value, max_value):
|
|
self.screen = screen
|
|
self.x = x
|
|
self.y = y
|
|
self.width = width
|
|
self.height = height
|
|
self.min_value = min_value
|
|
self.max_value = max_value
|
|
self.value = min_value
|
|
|
|
self.background_colour = (255, 255, 255)
|
|
self.knob_colour = (255, 0, 0)
|
|
self.rect_radius = 5
|
|
|
|
self.knob_radius = 10
|
|
self.knob_x = self.x + self.knob_radius
|
|
self.dragging = False
|
|
|
|
self.fill = False
|
|
self.fill_colour = (128, 128, 0)
|
|
|
|
def _knob_to_value(self, knob_x):
|
|
return self.min_value + ((self.max_value - self.min_value) * ((knob_x - (self.x + self.knob_radius)) / (self.width-self.knob_radius)))
|
|
|
|
def _value_to_knob(self, value):
|
|
return self.x + ((self.width-self.knob_radius) * ((value - self.min_value) / (self.max_value - self.min_value))) + self.knob_radius
|
|
|
|
def handle_event(self, event):
|
|
"""Handle mouse events for dragging"""
|
|
if event.type == pygame.MOUSEBUTTONDOWN:
|
|
if (
|
|
event.pos[0] >= self.x
|
|
and event.pos[0] <= self.x + self.width
|
|
and event.pos[1] >= self.y
|
|
and event.pos[1] <= self.y + self.height
|
|
):
|
|
if event.pos[0] < self.x + self.knob_radius:
|
|
self.knob_x = self.x + self.knob_radius
|
|
elif event.pos[0] > self.x + self.width - self.knob_radius:
|
|
self.knob_x = self.x + self.width - self.knob_radius
|
|
else:
|
|
self.knob_x = event.pos[0]
|
|
self.dragging = True
|
|
elif event.type == pygame.MOUSEBUTTONUP:
|
|
self.dragging = False
|
|
elif event.type == pygame.MOUSEMOTION and self.dragging:
|
|
if (
|
|
event.pos[0] >= self.x + self.knob_radius
|
|
and event.pos[0] <= self.x + self.width - self.knob_radius
|
|
):
|
|
self.knob_x = event.pos[0]
|
|
self.value = self._knob_to_value(self.knob_x)
|
|
|
|
def _draw_slider(self):
|
|
if self.fill:
|
|
pygame.draw.rect(self.screen, self.fill_colour, (self.x, self.y, self.knob_x, self.height), border_radius=self.rect_radius)
|
|
pygame.draw.rect(self.screen, self.background_colour, (self.knob_x, self.y, self.width - (self.knob_x - self.x), self.height), border_radius=self.rect_radius)
|
|
else:
|
|
pygame.draw.rect(self.screen, self.background_colour, (self.x, self.y, self.width, self.height), border_radius=self.rect_radius)
|
|
|
|
def _draw_knob(self):
|
|
pygame.draw.circle(self.screen, self.knob_colour, (int(self.knob_x), self.y + self.height // 2), self.knob_radius)
|
|
|
|
def draw(self):
|
|
"""
|
|
Draw the slider on the screen
|
|
"""
|
|
self._draw_slider()
|
|
self._draw_knob()
|
|
|
|
|
|
def set_value(self, value):
|
|
"""
|
|
Set the value of the slider
|
|
Caps the value to the min and max values
|
|
"""
|
|
self.value = int(value)
|
|
if value < self.min_value:
|
|
self.value = self.min_value
|
|
elif value > self.max_value:
|
|
self.value = self.max_value
|
|
else:
|
|
self.value = value
|
|
|
|
self.knob_x = self._value_to_knob(value)
|
|
|
|
def set_radius(self, radius):
|
|
self.knob_radius = radius
|
|
|
|
def set_background_colour(self, colour):
|
|
self.background_colour = colour
|
|
|
|
def set_knob_colour(self, colour):
|
|
self.knob_colour = colour
|
|
|
|
def set_fill_colour(self, colour):
|
|
self.fill_colour = colour
|
|
|
|
if __name__ == "__main__":
|
|
pygame.init()
|
|
screen = pygame.display.set_mode((800, 600))
|
|
slider = HorizontalSlider(screen, 50, 50, 700, 20, 0, 1000)
|
|
running = True
|
|
|
|
while running:
|
|
screen.fill((0, 0, 0))
|
|
slider.draw()
|
|
pygame.display.flip()
|
|
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
running = False
|
|
slider.handle_event(event)
|
|
|
|
|
|
|