def test_add_button(self, _init_pygame, default_ui_manager):
panel = UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=default_ui_manager,
margins={
'left': 10,
'right': 10,
'top': 5,
'bottom': 5
})
assert panel.layer_thickness == 1
button = UIButton(
relative_rect=pygame.Rect(100, 100, 150, 30),
text="Test Button",
tool_tip_text=
"This is a test of the button's tool tip functionality.",
manager=default_ui_manager,
container=panel)
assert button.layer_thickness == 1
assert panel.get_container(
).layer_thickness == 2 # happens 'cause elements added to container hover 1 layer up
panel.update(0.05)
assert panel.layer_thickness == 2
def test_process_event(self, _init_pygame, default_ui_manager):
panel = UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=default_ui_manager,
margins={
'left': 10,
'right': 10,
'top': 5,
'bottom': 5
})
consumed_event_left = panel.process_event(
pygame.event.Event(pygame.MOUSEBUTTONDOWN, {
'button': pygame.BUTTON_LEFT,
'pos': panel.rect.center
}))
consumed_event_right = panel.process_event(
pygame.event.Event(pygame.MOUSEBUTTONDOWN, {
'button': pygame.BUTTON_RIGHT,
'pos': panel.rect.center
}))
consumed_event_middle = panel.process_event(
pygame.event.Event(pygame.MOUSEBUTTONDOWN, {
'button': pygame.BUTTON_MIDDLE,
'pos': panel.rect.center
}))
assert consumed_event_left and consumed_event_right and consumed_event_middle
def test_kill(self, _init_pygame, default_ui_manager: IUIManagerInterface):
panel = UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=default_ui_manager,
margins={
'left': 10,
'right': 10,
'top': 5,
'bottom': 5
})
button = UIButton(
relative_rect=pygame.Rect(100, 100, 150, 30),
text="Test Button",
tool_tip_text=
"This is a test of the button's tool tip functionality.",
manager=default_ui_manager,
container=panel)
assert len(default_ui_manager.get_root_container().elements) == 2
assert len(default_ui_manager.get_sprite_group().sprites()) == 4
assert default_ui_manager.get_sprite_group().sprites() == [
default_ui_manager.get_root_container(), panel,
panel.get_container(), button
]
panel.kill()
assert len(default_ui_manager.get_root_container().elements) == 0
assert len(default_ui_manager.get_sprite_group().sprites()) == 1
assert default_ui_manager.get_sprite_group().sprites() == [
default_ui_manager.get_root_container()
]
def test_visible_panel_children_behaviour_on_hide(
self, _init_pygame, default_ui_manager: IUIManagerInterface,
_display_surface_return_none):
panel = UIPanel(pygame.Rect(100, 100, 200, 200),
manager=default_ui_manager,
visible=1,
starting_layer_height=5)
button = UIButton(relative_rect=pygame.Rect(0, 0, 50, 50),
text="",
manager=default_ui_manager,
container=panel)
assert panel.visible == 1
assert button.visible == 1
panel.hide()
assert panel.visible == 0
assert button.visible == 0
def test_creation_with_margins(self, _init_pygame, default_ui_manager):
UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=default_ui_manager,
margins={
'left': 10,
'right': 10,
'top': 5,
'bottom': 5
})
def test_disable(self, _init_pygame: None, default_ui_manager: UIManager,
_display_surface_return_none: None):
panel = UIPanel(relative_rect=pygame.Rect(0, 0, 150, 400),
starting_layer_height=5,
manager=default_ui_manager)
button_1 = UIButton(
relative_rect=pygame.Rect(10, 10, 150, 30),
text="Test Button",
tool_tip_text=
"This is a test of the button's tool tip functionality.",
manager=default_ui_manager,
container=panel)
button_2 = UIButton(relative_rect=pygame.Rect(10, 50, 150, 30),
text="Test Button 2",
manager=default_ui_manager,
container=panel)
panel.disable()
assert panel.is_enabled is False
assert button_1.is_enabled is False
assert button_2.is_enabled is False
# process a mouse button down event
button_1.process_event(
pygame.event.Event(pygame.MOUSEBUTTONDOWN, {
'button': 1,
'pos': button_1.rect.center
}))
# process a mouse button up event
button_1.process_event(
pygame.event.Event(pygame.MOUSEBUTTONUP, {
'button': 1,
'pos': button_1.rect.center
}))
button_1.update(0.01)
assert button_1.check_pressed() is False
def test_show_hide_rendering(self, _init_pygame, default_ui_manager,
_display_surface_return_none):
resolution = (400, 400)
empty_surface = pygame.Surface(resolution)
empty_surface.fill(pygame.Color(0, 0, 0))
surface = empty_surface.copy()
manager = UIManager(resolution)
panel = UIPanel(pygame.Rect(25, 25, 375, 150),
manager=manager,
visible=0,
starting_layer_height=1)
button = UIButton(relative_rect=pygame.Rect(0, 0, 50, 50),
text="",
manager=manager,
container=panel)
manager.update(0.01)
manager.draw_ui(surface)
assert compare_surfaces(empty_surface, surface)
surface.fill(pygame.Color(0, 0, 0))
panel.show()
manager.update(0.01)
manager.draw_ui(surface)
assert not compare_surfaces(empty_surface, surface)
surface.fill(pygame.Color(0, 0, 0))
panel.hide()
manager.update(0.01)
manager.draw_ui(surface)
assert compare_surfaces(empty_surface, surface)
def test_rebuild_from_changed_theme_data_bad_values(
self, _init_pygame, _display_surface_return_none):
manager = UIManager((800, 600),
os.path.join("tests", "data", "themes",
"ui_panel_bad_values.json"))
panel = UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=manager,
margins={
'left': 10,
'right': 10,
'top': 5,
'bottom': 5
})
assert panel.image is not None
def test_update(self, _init_pygame, default_ui_manager):
panel = UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=default_ui_manager,
margins={
'left': 10,
'right': 10,
'top': 5,
'bottom': 5
})
assert panel.layer_thickness == 1
assert panel.get_container().layer_thickness == 0
panel.get_container().layer_thickness = 4
assert panel.layer_thickness == 1
panel.update(0.05)
assert panel.layer_thickness == 4
def test_set_dimensions(self, _init_pygame, default_ui_manager):
test_container = UIContainer(relative_rect=pygame.Rect(
10, 10, 300, 300),
manager=default_ui_manager)
element_1 = UIPanel(relative_rect=pygame.Rect(30, 30, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'left',
'right': 'left',
'top': 'top',
'bottom': 'top'
})
assert element_1.relative_right_margin is None
assert element_1.relative_bottom_margin is None
element_1.set_dimensions((20, 20))
assert element_1.relative_rect.topleft == (30, 30)
assert element_1.relative_rect.size == (20, 20)
assert element_1.relative_rect.bottomright == (50, 50)
assert element_1.rect.topleft == (40, 40)
assert element_1.rect.size == (20, 20)
assert element_1.rect.bottomright == (60, 60)
assert element_1.relative_right_margin is None
assert element_1.relative_bottom_margin is None
element_2 = UIPanel(relative_rect=pygame.Rect(-60, 10, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'right',
'right': 'right',
'top': 'top',
'bottom': 'top'
})
assert element_2.relative_right_margin == 10
assert element_2.relative_bottom_margin is None
element_2.set_dimensions((60, 60))
assert element_2.relative_rect.topleft == (-60, 10)
assert element_2.relative_rect.size == (60, 60)
assert element_2.relative_rect.bottomright == (0, 70)
assert element_2.rect.topleft == (250, 20)
assert element_2.rect.size == (60, 60)
assert element_2.rect.bottomright == (310, 80)
assert element_2.relative_right_margin == 0
assert element_2.relative_bottom_margin is None
element_3 = UIPanel(relative_rect=pygame.Rect(-70, -70, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'right',
'right': 'right',
'top': 'bottom',
'bottom': 'bottom'
})
assert element_3.relative_right_margin == 20
assert element_3.relative_bottom_margin == 20
element_3.set_dimensions((30, 30))
assert element_3.relative_rect.topleft == (-70, -70)
assert element_3.relative_rect.size == (30, 30)
assert element_3.relative_rect.bottomright == (-40, -40)
assert element_3.rect.topleft == (240, 240)
assert element_3.rect.size == (30, 30)
assert element_3.rect.bottomright == (270, 270)
assert element_3.relative_right_margin == 40
assert element_3.relative_bottom_margin == 40
element_4 = UIPanel(relative_rect=pygame.Rect(50, -50, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'left',
'right': 'left',
'top': 'bottom',
'bottom': 'bottom'
})
assert element_4.relative_right_margin is None
assert element_4.relative_bottom_margin == 0
element_4.set_dimensions((100, 100))
assert element_4.relative_rect.topleft == (50, -50)
assert element_4.relative_rect.size == (100, 100)
assert element_4.relative_rect.bottomright == (150, 50)
assert element_4.rect.topleft == (60, 260)
assert element_4.rect.size == (100, 100)
assert element_4.rect.bottomright == (160, 360)
assert element_4.relative_right_margin is None
assert element_4.relative_bottom_margin == -50
element_5 = UIPanel(relative_rect=pygame.Rect(10, 10, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'left',
'right': 'right',
'top': 'top',
'bottom': 'bottom'
})
assert element_5.relative_right_margin == 240
assert element_5.relative_bottom_margin == 240
element_5.set_dimensions((90, 90))
assert element_5.relative_rect.topleft == (10, 10)
assert element_5.relative_rect.size == (90, 90)
assert element_5.relative_rect.bottomright == (100, 100)
assert element_5.rect.topleft == (20, 20)
assert element_5.rect.size == (90, 90)
assert element_5.rect.bottomright == (110, 110)
assert element_5.relative_right_margin == 200
assert element_5.relative_bottom_margin == 200
def test_creation(self, _init_pygame, default_ui_manager):
UIPanel(relative_rect=pygame.Rect(50, 50, 150, 400),
starting_layer_height=5,
manager=default_ui_manager)
def test_set_position(self, _init_pygame, default_ui_manager):
test_container = UIContainer(relative_rect=pygame.Rect(
10, 10, 300, 300),
manager=default_ui_manager)
element = UIPanel(relative_rect=pygame.Rect(100, 100, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5)
element.set_position((150, 30))
assert element.relative_rect.topleft == (140, 20)
element_1 = UIPanel(relative_rect=pygame.Rect(0, 0, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'left',
'right': 'left',
'top': 'top',
'bottom': 'top'
})
element_1.set_position((20, 20))
assert element_1.relative_rect.topleft == (10, 10)
assert element_1.relative_rect.size == (50, 50)
assert element_1.relative_rect.bottomright == (60, 60)
element_2 = UIPanel(relative_rect=pygame.Rect(0, 0, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'right',
'right': 'right',
'top': 'top',
'bottom': 'top'
})
element_2.set_position((280, 120))
assert element_2.relative_rect.topleft == (-30, 110)
assert element_2.relative_rect.size == (50, 50)
assert element_2.relative_rect.bottomright == (20, 160)
element_3 = UIPanel(relative_rect=pygame.Rect(0, 0, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'right',
'right': 'right',
'top': 'bottom',
'bottom': 'bottom'
})
element_3.set_position((230, 230))
assert element_3.relative_rect.topleft == (-80, -80)
assert element_3.relative_rect.size == (50, 50)
assert element_3.relative_rect.bottomright == (-30, -30)
element_4 = UIPanel(relative_rect=pygame.Rect(0, 0, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'left',
'right': 'left',
'top': 'bottom',
'bottom': 'bottom'
})
element_4.set_position((130, 230))
assert element_4.relative_rect.topleft == (120, -80)
assert element_4.relative_rect.size == (50, 50)
assert element_4.relative_rect.bottomright == (170, -30)
element_5 = UIPanel(relative_rect=pygame.Rect(0, 0, 50, 50),
manager=default_ui_manager,
container=test_container,
starting_layer_height=5,
anchors={
'left': 'left',
'right': 'right',
'top': 'top',
'bottom': 'bottom'
})
assert element_5.relative_right_margin == 250
assert element_5.relative_bottom_margin == 250
element_5.set_position((20, 20))
assert element_5.relative_rect.topleft == (10, 10)
assert element_5.relative_rect.size == (50, 50)
assert element_5.relative_rect.bottomright == (60, 60)
assert element_5.relative_right_margin == 240
assert element_5.relative_bottom_margin == 240
def init_gui(self):
bg_panel = UIPanel(pygame.Rect(0, 0, SCREEN_WIDTH, SCREEN_HEIGHT),
starting_layer_height=4,
manager=ui_manager)
UILabel(pygame.Rect(SCREEN_WIDTH / 2 - 300, 100, 600, 100),
"Tower Defender",
ui_manager,
container=bg_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
},
object_id="#title_label")
self.start_btn = UIButton(pygame.Rect(SCREEN_WIDTH / 2 - 100, 260, 200,
40),
"Start",
ui_manager,
container=bg_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.start_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: switch_mode(MODE_SELECT_LEVEL))
self.editor_btn = UIButton(pygame.Rect(SCREEN_WIDTH / 2 - 100, 320,
200, 40),
"Editor",
ui_manager,
container=bg_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.editor_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: switch_mode(MODE_EDITOR))
pygame_gui.elements.ui_image.UIImage(
relative_rect=pygame.Rect(SCREEN_WIDTH / 2 - 800,
SCREEN_HEIGHT / 2 - 240, 720, 480),
image_surface=resource_cache.get_resource(
ENEMIES_PATH + "1/Golem_01_Walking_000.png",
resource_cache.SurfaceType,
alpha=True),
manager=ui_manager,
container=bg_panel)
pygame_gui.elements.ui_image.UIImage(
relative_rect=pygame.Rect(SCREEN_WIDTH / 2 + 100,
SCREEN_HEIGHT / 2 - 240, 720, 480),
image_surface=resource_cache.get_resource(
ENEMIES_PATH + "2/reversed/Golem_02_Walking_000.png",
resource_cache.SurfaceType,
alpha=True),
manager=ui_manager,
container=bg_panel)
def init_gui(self):
# top panel
UIPanel(pygame.Rect(0, 0, SCREEN_WIDTH, 30),
starting_layer_height=4,
manager=ui_manager)
# right panel
right_panel_w = SCREEN_WIDTH - TILE_SIZE * MAP_W
right_panel_h = SCREEN_HEIGHT - 30
right_panel = UIPanel(pygame.Rect(TILE_SIZE * MAP_W, 30, right_panel_w,
right_panel_h),
starting_layer_height=4,
manager=ui_manager)
# load button
self.load_btn = UIButton(pygame.Rect(20, 120, right_panel_w * 0.8, 40),
"Load",
ui_manager,
container=right_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.load_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: self.open_file_dialog(False))
# save button
self.save_btn = UIButton(pygame.Rect(20, 60, right_panel_w * 0.8, 40),
"Save",
ui_manager,
container=right_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.save_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: self.open_file_dialog())
# settings button
self.settings_btn = UIButton(pygame.Rect(20, 180, right_panel_w * 0.8,
40),
"Settings",
ui_manager,
container=right_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.settings_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: MapSettingsWindow())
# clear button
self.clear_btn = UIButton(pygame.Rect(20, 260, right_panel_w * 0.8,
40),
"Clear",
ui_manager,
container=right_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.clear_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: self.on_clear_btn_click())
# back button
self.back_btn = UIButton(pygame.Rect(20, 320, right_panel_w * 0.8, 40),
"Back To Menu",
ui_manager,
container=right_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
register_ui_callback(self.back_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: self.on_back_to_menu_btn_click())
def init_gui(self):
bg_panel = UIPanel(pygame.Rect(0, 0, SCREEN_WIDTH, SCREEN_HEIGHT),
starting_layer_height=4,
manager=ui_manager)
UILabel(pygame.Rect(SCREEN_WIDTH / 2 - 300, 10, 600, 100),
"Select Level",
ui_manager,
container=bg_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
},
object_id="#title_label")
LEVEL_BUTTON_SIZE = 125
MARGIN = 15
for y in range(0, 4):
for x in range(0, 6):
nr = 1 + x + y * 6
level_btn = UIButton(pygame.Rect(
SCREEN_WIDTH / 2 -
(3 * LEVEL_BUTTON_SIZE + 2 * MARGIN + MARGIN / 2) + x *
(LEVEL_BUTTON_SIZE + MARGIN), SCREEN_HEIGHT / 2 -
(2 * LEVEL_BUTTON_SIZE + MARGIN + MARGIN / 2) + y *
(LEVEL_BUTTON_SIZE + MARGIN), LEVEL_BUTTON_SIZE,
LEVEL_BUTTON_SIZE),
str(nr),
ui_manager,
container=bg_panel,
anchors={
"left": "left",
"right": "right",
"top": "top",
"bottom": "bottom"
})
if nr > 8:
level_btn.disable()
else:
register_ui_callback(
level_btn,
pygame_gui.UI_BUTTON_PRESSED,
lambda e, n=nr: switch_mode(MODE_GAME,
file_name=f"{n}.tdmap"))
back_btn = UIButton(pygame.Rect(40, -80, 200, 60),
"Back",
ui_manager,
container=bg_panel,
anchors={
"left": "left",
"right": "left",
"top": "bottom",
"bottom": "bottom"
})
register_ui_callback(back_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: switch_mode(MODE_MENU))
select_custom_btn = UIButton(pygame.Rect(-240, -80, 200, 60),
"Select Custom",
ui_manager,
container=bg_panel,
anchors={
"left": "right",
"right": "right",
"top": "bottom",
"bottom": "bottom"
})
register_ui_callback(
select_custom_btn, pygame_gui.UI_BUTTON_PRESSED,
lambda e: SaveLoadWindow(
"maps", "Select Map", lambda f: switch_mode(
MODE_GAME, file_name=f), False))
