demo.py

import os
import random

import pygame
from pygame import Color

from src.PygameUIKit import Group
from src.PygameUIKit import text_input, button, slider, combobox
from src.PygameUIKit.barchart import BarChart

#
#
# This is a file to showcase every EasyObject
#
#

RED = (255, 0, 0)
BTN_GREEN = (0, 169, 0)
BTN_BLUE = (83, 131, 232)

cwd = os.path.dirname(__file__)
img_play = pygame.image.load(os.path.join(cwd, "assets/play.png"))
img_stop = pygame.image.load(os.path.join(cwd, "assets/stop.png"))


class Demo:
    def __init__(self):
        self.done = False
        self.screen = pygame.display.set_mode((500, 500))
        self.clock = pygame.time.Clock()

        self.easy_objects = Group()
        self.text_input = text_input.TextInput(placeholder="This is a placeholder text", fixed_width=200, border_radius=2, ui_group=self.easy_objects,
                                               font=def_font)
        self.btn_pause = button.ButtonTwoStates(img_play, img_stop, do_nothing, ui_group=self.easy_objects)
        self.btn_png = button.ButtonPngIcon(img_play, self.change_values, inflate=10, ui_group=self.easy_objects)

        self.btn_pause.connect(self.toggle_dance)

        self.slider = slider.Slider(0, 100, 1, show_value=True, ui_group=self.easy_objects, font=def_font)
        self.slider.connect(self.change_values)

        self.dropdown = combobox.ComboBox(["Hello", "World", "And", "You"], ui_group=self.easy_objects, font=def_font)

        data = [10, 20, 30, 40, 50]
        self.chart = BarChart(data, ui_group=self.easy_objects, max_value=100, font=def_font)

        for i in range(4):
            self.dropdown.add_action(i, lambda i=i: self.text_input.set_text(self.dropdown.elements[i]))

        self.dancing = False

    def run(self):
        while not self.done:
            dt = self.clock.tick(60) / 1000
            self.events()
            self.update(dt)
            self.draw(self.screen)

    def events(self):
        events = pygame.event.get()
        for event in events:
            self.easy_objects.handle_event(event)
            if event.type == pygame.QUIT:
                self.done = True
            if event.type == pygame.MOUSEBUTTONDOWN:
                print(f"position: {event.pos}")

        self.text_input.handle_events(events)

    def update(self, dt):
        self.easy_objects.update(dt)
        if self.dancing:
            self.change_values()

    def draw(self, win):
        W, H = self.screen.get_size()
        win.fill(Color(224, 224, 224))
        self.text_input.draw(win, W // 2 - self.text_input.rect.w // 2, H - 200)
        self.btn_pause.draw(win, 100, 200)
        self.btn_png.draw(win, 100, 300)
        self.slider.draw(win, 100, 400, 300, 5)
        self.chart.draw(win, 159, 164, 300, 100)
        self.dropdown.draw(win, 100, 100)
        pygame.display.flip()

    def change_values(self):
        i = random.randint(0, len(self.chart.values) - 1)
        new_value = random.randint(0, 100)
        self.chart.change_value(i, new_value)

    def toggle_dance(self):
        self.dancing = not self.dancing


def do_nothing():
    pass


if __name__ == '__main__':
    pygame.init()
    def_font = pygame.font.SysFont("Arial", 20)

    Demo().run()
    pygame.quit()