pygame_gui_example.py

import pygame
from pygame.transform import rotate
from pygame.sprite import Sprite, Group
from pygame.math import Vector2
import pygame_gui
from pygame_gui.elements import UIButton

# Here we need to call init to load font text for the buttons
pygame.init()

SCREEN_SIZE = (300, 400)
MAP_SIZE = (300, 320)
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
FPS = 60


class Ant(Sprite):

    def __init__(self, size=(25, 25), speed=(2, 2)):
        # Call the parent class (Sprite) constructor
        super(Ant, self).__init__()

        # Load an image of the ant
        self.image = pygame.image.load("ant_worker.png")
        self.rect = self.image.get_rect()
        self.rect.inflate_ip(*size)
        self.rect.update(0, 0, *self.rect.size)
        self.vector = Vector2(speed)

    @property
    def direction(self):
        return self.vector.normalize()

    def reverse(self):
        self.vector = self.vector.reflect(self.vector)

    def speed_up(self):
        self.vector *= 1.5

    def slow_down(self):
        self.vector *= 0.5

    def update(self):
        self.rect.move_ip(self.vector)

        if self.rect.left < 0 or self.rect.right > MAP_SIZE[0]:
            self.vector = self.vector.reflect((1, 0))

        if self.rect.top < 0 or self.rect.bottom > MAP_SIZE[1]:
            self.vector = self.vector.reflect((0, 1))

    def draw(self, surface):
        angle = self.vector.angle_to(Vector2(0, -1))
        surface.blit(rotate(self.image, angle), self.rect)


if __name__ == '__main__':

    # Creates a graphical window, based on system hardware
    # settings
    screen = pygame.display.set_mode(SCREEN_SIZE)

    # Create Pygame GUI UIManager to tae control of processing
    # events
    manager = pygame_gui.UIManager(SCREEN_SIZE)

    flip_button = UIButton(
        relative_rect=pygame.Rect((0, 320), (100, 80)),
        text='Flip it!',
        manager=manager
    )
    speed_button = UIButton(
        relative_rect=pygame.Rect((100, 320), (100, 80)),
        text='Speed up!',
        manager=manager
    )
    slow_button = UIButton(
        relative_rect=pygame.Rect((200, 320), (100, 80)),
        text='Slow down!',
        manager=manager
    )

    ant = Ant()

    colony = Group(ant)

    clock = pygame.time.Clock()
    running = True

    while running:

        time_delta = clock.tick(FPS) / 1000.0

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False

            if event.type == pygame.USEREVENT:
                if event.user_type == pygame_gui.UI_BUTTON_PRESSED:
                    if event.ui_element == flip_button:
                        ant.reverse()
                    elif event.ui_element == speed_button:
                        ant.speed_up()
                    elif event.ui_element == slow_button:
                        ant.slow_down()

            manager.process_events(event)

        manager.update(time_delta)

        colony.update()

        screen.fill(WHITE)

        for ant in colony.sprites():
            ant.draw(screen)

        manager.draw_ui(screen)
        pygame.display.flip()