Merge branch 'master' into a_star

# Conflicts:
#	main.py
This commit is contained in:
s452645 2021-05-09 01:17:09 +02:00
commit 7d16dcc616
15 changed files with 1085 additions and 474 deletions

3
.gitignore vendored
View File

@ -1,5 +1,6 @@
# .idea files
.idea
.idea*
# Byte-compiled / optimized / DLL files
__pycache__/

View File

@ -1,21 +1,19 @@
import pygame
import project_constants as const
import json_generator as js
import json
# Class of our agent, initialization of it
# movement functions (those defiend by the 'go_' prefix are not meant to actually move our agent, they just return some values
# that are later used by another function called 'is_valid_move' (which is defined in Minefield));
# movement functions (those defined by the 'go_' prefix are not meant to actually move our agent, they just return some
# values that are later used by another function called 'is_valid_move' (which is defined in Minefield));
class Agent:
def __init__(self, json_path):
with open(json_path) as json_data:
data = json.load(json_data)
self.row, self.column = data['agent_starting_position'].split(",")
self.row, self.column = data["agents_initial_state"]["position"].split(",")
self.position = [int(self.row), int(self.column)]
# self.direction = const.Direction()
self.direction = const.Direction.UP
self.direction = const.Direction(data["agents_initial_state"]["direction"])
def rotate_left(self):
self.direction = self.direction.previous()

View File

@ -2,26 +2,25 @@ import pygame
import project_constants as const
from mine_models.standard_mine import StandardMine
from mine_models.chained_mine import ChainedMine
from mine_models.time_mine import TimeMine
# ================================= #
# === SO THE OLD BLITTING WORKS === #
# ================================= #
tile_asset_options = {
"BLUE": const.ASSET_TILE_BLUE,
"GREEN": const.ASSET_TILE_GREEN,
"ORANGE": const.ASSET_TILE_ORANGE,
"PURPLE": const.ASSET_TILE_PURPLE,
"RED": const.ASSET_TILE_RED,
"WHITE": const.ASSET_TILE_WHITE,
"YELLOW": const.ASSET_TILE_YELLOW
"MUD": const.ASSET_MUD,
"GRASS": const.ASSET_GRASS,
"CONCRETE": const.ASSET_CONCRETE
}
mine_asset_options = {
'A': const.ASSET_MINE_A,
'B': const.ASSET_MINE_B,
'F': const.ASSET_MINE_F,
'K': const.ASSET_MINE_K
"MINE": const.ASSET_MINE,
"CHAINS": const.ASSET_CHAINS,
"TIME_MINE": const.ASSET_TIME_MINE,
}
@ -30,21 +29,6 @@ mine_asset_options = {
# ====================== #
def test_blits():
display_concrete((2, 3))
display_mud((5, 6))
display_mud((5, 7))
display_grass((2, 1))
display_mine((2, 7))
display_mine((5, 7))
display_concrete((0, 0))
display_chained_mine((1, 2), (2, 3))
display_chained_mine((2, 3))
display_chained_mine((4, 8), (11, 15))
display_time_mine((1, 8), 12)
display_time_mine((2, 8), 34)
def blit_graphics(minefield):
# background grid (fills frame with white, blits grid)
const.SCREEN.fill((255, 255, 255))
@ -64,7 +48,7 @@ def blit_graphics(minefield):
# draw a tile
const.SCREEN.blit(
tile_asset_options.get(tile.color),
tile_asset_options.get(tile.terrain_type),
tile_screen_coords
)
@ -74,11 +58,16 @@ def blit_graphics(minefield):
# draw a mine on top if there is one
if tile.mine is not None:
# current icons don't represent actual types, thus every mine has the same icon (temporary solution)
const.SCREEN.blit(mine_asset_options['A'], tile_screen_coords)
if isinstance(tile.mine, StandardMine):
const.SCREEN.blit(mine_asset_options["MINE"], tile_screen_coords)
# all the tests in one place
test_blits()
if isinstance(tile.mine, ChainedMine):
predecessor = tile.mine.predecessor
predecessor_position = predecessor.position if predecessor is not None else None
display_chained_mine(tile.position, predecessor_position)
elif isinstance(tile.mine, TimeMine):
display_time_mine(tile.position, "0" + str(tile.mine.timer))
# sapper
display_sapper(
@ -264,9 +253,9 @@ def display_time_mine(coords, time):
mine_coords = calculate_screen_position(coords)
number_coords = mine_coords
if which_digit == const.Digit.ONES:
number_coords = (mine_coords[0] + 36, mine_coords[1] + 22)
elif which_digit == const.Digit.TENS:
number_coords = (mine_coords[0] + 44, mine_coords[1] + 22)
elif which_digit == const.Digit.TENS:
number_coords = (mine_coords[0] + 36, mine_coords[1] + 22)
const.SCREEN.blit(
number_asset,

View File

@ -4,12 +4,12 @@ import random
import project_constants as const
# import tile class
import tile as tl
from tile import Tile
# import mine models
import mine_models.standard_mine as sm
import mine_models.time_mine as tm
import mine_models.chained_mine as cm
from mine_models.standard_mine import StandardMine
from mine_models.time_mine import TimeMine
from mine_models.chained_mine import ChainedMine
class JsonGenerator:
@ -18,7 +18,7 @@ class JsonGenerator:
# constructor that can be used to set agent's initial state
def __init__(self, agents_initial_position=(0, 0), agents_initial_direction=const.Direction.UP.value):
# saving agent's initial state (position & direction)
self.agents_initial_position = agents_initial_position
self.agents_initial_position = format_position_to_tuple(agents_initial_position)
self.agents_initial_direction = agents_initial_direction
# saving data to the grid dictionary
@ -30,7 +30,7 @@ class JsonGenerator:
# sets agent's initial state
def set_agents_initial_state(self, position=(0, 0), direction=const.Direction.UP.value):
# setting fields in the instance
self.agents_initial_position = position
self.agents_initial_position = format_position_to_tuple(position)
self.agents_initial_direction = direction
# setting new agent's initial state
@ -44,6 +44,9 @@ class JsonGenerator:
# clearing grid field
self.clear_grid()
# formatting dimensions to tuple
dimensions = format_position_to_tuple(dimensions)
# getting grid dimensions
num_of_rows, num_of_columns = dimensions
@ -53,7 +56,7 @@ class JsonGenerator:
for j in range(num_of_columns):
# picking random values for tiles
random_tile_color = random.choice(const.STRUCT_TILE_COLORS)
random_tile_color = random.choice(const.STRUCT_TILE_TERRAINS)
# adding added tile's indexes to a pool
tile_pool.append((i, j))
@ -129,7 +132,7 @@ class JsonGenerator:
self.grid[format_position_to_str(position)]["mine"] = {}
for key in mine_values.keys():
self.grid[format_position_to_str()]["mine"][key] = mine_values[key]
self.grid[format_position_to_str(position)]["mine"][key] = mine_values[key]
# deletes a mine with a given position from the grid field
def delete_a_tile(self, position):
@ -227,16 +230,16 @@ def create_a_mine(mine_dict, position):
# if mine's type is "standard" - creating standard mine
if mine_dict["mine_type"] == "standard":
mine = sm.StandardMine(position)
mine = StandardMine(position)
# if mine's type is "time" - creating time mine
elif mine_dict["mine_type"] == "time":
timer_value = mine_dict["timer"]
mine = tm.TimeMine(position, timer_value)
mine = TimeMine(position, timer_value)
# if mine's type is "chained" - creating chained mine (no successors assigned yet)
elif mine_dict["mine_type"] == "chained":
mine = cm.ChainedMine(position)
mine = ChainedMine(position)
return mine
@ -248,10 +251,10 @@ def create_a_tile(tile_dict, position):
# getting tile's parameters
color = tile_dict["color"]
mine = create_a_mine(position, tile_dict["mine"])
mine = create_a_mine(tile_dict["mine"], position)
# creating and returning a tile with the parameters set above
return tl.Tile(position, color, mine)
return Tile(position, color, mine)
# returns a list of tuples containing chained mine's position and it's predecessors position
@ -268,7 +271,8 @@ def get_chained_mine_and_its_predecessor_pairs(minefield_dictionary):
# getting the chained mines and it's predecessors positions
this_mines_position = tuple(int(i) for i in key.split(','))
its_predecessors_position = tuple(int(i) for i in minefield_dictionary[key]["mine"]["predecessor"].split(','))
its_predecessors_position = \
tuple(int(i) for i in minefield_dictionary[key]["mine"]["predecessor"].split(','))
# adding the positions to the list as a tuple
predecessors.append((this_mines_position, its_predecessors_position))

98
main.py
View File

@ -1,20 +1,24 @@
# libraries
import pygame
import time
from pyglet.gl import * # for blocky textures
import random
# other files of this project
import project_constants as const
import minefield as mf
import searching_algorithms.a_star as a_star
import searching_algorithms.bfs as bfs
from display_assets import blit_graphics
from ui.input_box import *
from ui.button import *
from ui.text_box import *
import json_generator as jg
def main():
pygame.init()
pygame.display.set_caption(const.V_NAME_OF_WINDOW)
jes = jg.JsonGenerator()
print(jes.get_grid())
# for blocky textures
glEnable(GL_TEXTURE_2D)
@ -26,16 +30,78 @@ def main():
# create an instance of Minefield, pass necessary data
minefield = mf.Minefield(const.MAP_RANDOM_10x10)
# get sequence of actions found by BFS algorithm
action_sequence = a_star.graphsearch(
initial_state=a_star.State(
# setting flags for program
running = True
in_menu = True
# counting coordinates for ui components
ib_width, ib_height = 100, 65
bt_width, bt_height = 210, 55
ib_y = 200
ib1_x = const.SCREEN.get_width() / 2 - ib_width - 5
ib2_x = const.SCREEN.get_width() / 2 + 5
bt_x = const.SCREEN.get_width() / 2 - bt_width / 2
bt1_y = ib_y + ib_height + 10
bt2_y = bt1_y + bt_height + 10
# creating ui components used in menu
input1 = InputBox(
position=(ib1_x, ib_y),
dimensions=(ib_width, ib_height),
text="row",
input_centered=True,
clear_input_on_click=True
)
input2 = InputBox(
position=(ib2_x, ib_y),
dimensions=(ib_width, ib_height),
text="column",
input_centered=True,
clear_input_on_click=True
)
random_button = Button(position=(bt_x, bt1_y), dimensions=(bt_width, bt_height), text="random")
ok_button = Button(position=(bt_x, bt2_y), dimensions=(bt_width, bt_height), text="ok")
# initializing goal position
to_x, to_y = 0, 0
while running:
while running and in_menu:
const.SCREEN.fill((255, 255, 255))
events = pygame.event.get()
for event in events:
if event.type == pygame.QUIT:
running = False
input1.run(const.SCREEN, pygame.mouse.get_pos(), events)
input2.run(const.SCREEN, pygame.mouse.get_pos(), events)
ok_button.draw(const.SCREEN, pygame.mouse.get_pos())
random_button.draw(const.SCREEN, pygame.mouse.get_pos())
if ok_button.is_clicked(pygame.mouse.get_pos(), events):
in_menu = False
if random_button.is_clicked(pygame.mouse.get_pos(), events):
input1.set_texts(user_input=str(random.randint(0, 9)))
input2.set_texts(user_input=str(random.randint(0, 9)))
pygame.display.flip()
clock.tick(const.V_FPS)
if running:
to_x = int(input1.get_input())
to_y = int(input2.get_input())
action_sequence = bfs.graphsearch(
initial_state=bfs.State(
row=minefield.agent.position[0],
column=minefield.agent.position[1],
direction=const.Direction.UP),
minefield=minefield)
direction=minefield.agent.direction),
minefield=minefield, tox=to_x, toy=to_y)
running = True
while running:
while running and not in_menu:
# FPS control
clock.tick(const.V_FPS)
@ -58,11 +124,11 @@ def main():
time.sleep(const.ACTION_INTERVAL)
else:
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
in_menu = True
time.sleep(const.ACTION_INTERVAL)
# 'for event in pygame.event.get():
# if event.type == pygame.QUIT:
# running = False
if __name__ == "__main__":

View File

@ -1,10 +1,10 @@
import json
import agent as ag
import project_constants as const
import tile as tl
from mine_models import standard_mine as sm
from mine_models import time_mine as tm
from mine_models import chained_mine as cm
import json_generator as jg
class Minefield:
@ -14,29 +14,28 @@ class Minefield:
self.agent = ag.Agent(const.MAP_RANDOM_10x10)
# open JSON with minefield info
with open(json_path) as json_data:
data = json.load(json_data)
json_gen = jg.JsonGenerator()
json_gen.load_from_a_file(json_path)
# create matrix of a desired size, fill it with default tile objects
self.matrix = [
[
tl.Tile((row, column)) for column in range(const.V_GRID_HOR_TILES)
tl.Tile(
(row, column),
terrain_type=json_gen.get_tile((row, column))["terrain"],
mine=jg.create_a_mine(json_gen.get_mine((row, column)), (row, column))
) for column in range(const.V_GRID_HOR_TILES)
] for row in range(const.V_GRID_VER_TILES)
]
# iterate through tiles, set their colors and add mines
for row in range(const.V_GRID_VER_TILES):
for column in range(const.V_GRID_HOR_TILES):
# iterate through chained mines, set mine predecessors
for pair in jg.get_chained_mine_and_its_predecessor_pairs(json_gen.get_grid()):
successor_position, predecessor_position = pair
successor_row, successor_column = successor_position
predecessor_row, predecessor_column = predecessor_position
# load tile's data from json
tile_data = data[f"{row},{column}"]
# if there is a mine, create & assign new Mine object (type recognition included)
if tile_data["mine"] is not None:
mine = self._create_mine(tile_data["mine"], row, column)
self.matrix[row][column].mine = mine
self.matrix[row][column].color = tile_data["color"].upper()
predecessor = self.matrix[predecessor_row][predecessor_column]
self.matrix[successor_row][successor_column].mine.predecessor = predecessor
# ================ #
# === MOVEMENT === #

View File

@ -70,6 +70,12 @@ class Coords(Enum):
Y = 1
class Terrain(Enum):
CONCRETE = 1
GRASS = 2
MUD = 4
# =============== #
# === STRUCTS === #
# =============== #
@ -79,7 +85,7 @@ class Coords(Enum):
# # USED BY JSON GENERATOR
# used to generate random tile colors
STRUCT_TILE_COLORS = ["BLUE", "GREEN", "ORANGE", "PURPLE", "RED", "WHITE", "YELLOW"]
STRUCT_TILE_TERRAINS = ["CONCRETE", "GRASS", "MUD"]
# used to generate random mines and create not random mines
STRUCT_MINE_TYPES = ["standard", "chained", "time"]
@ -113,13 +119,11 @@ STRUCT_MINE_ATTRIBUTE_TYPES = {
"time": [int]
}
# ============== #
# ==== MAPS ==== #
# ============== #
MAP_RANDOM_10x10 = os.path.join("resources", "minefields", "secondmap.json")
MAP_RANDOM_10x10 = os.path.join("resources", "minefields", "fourthmap.json")
# ============== #
# === ASSETS === #
@ -170,7 +174,6 @@ ASSET_TIME_MINE = pygame.transform.scale(
(V_TILE_SIZE, V_TILE_SIZE)
)
# ==================== #
# === TIME NUMBERS === #
# ==================== #
@ -225,7 +228,6 @@ ASSET_NUMBER_TIME_9 = pygame.transform.scale(
(6, 16)
)
# ====================== #
# === CHAINS NUMBERS === #
# ====================== #
@ -310,7 +312,6 @@ ASSET_NUMBER_CHAINS_15 = pygame.transform.scale(
(6, 12)
)
# ================== #
# === OLD ASSETS === #
# ================== #

View File

@ -0,0 +1,476 @@
{
"0,0": {
"terrain": "CONCRETE",
"mine": null
},
"0,1": {
"terrain": "CONCRETE",
"mine": null
},
"0,2": {
"terrain": "GRASS",
"mine": null
},
"0,3": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"0,4": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"0,5": {
"terrain": "CONCRETE",
"mine": null
},
"0,6": {
"terrain": "MUD",
"mine": null
},
"0,7": {
"terrain": "GRASS",
"mine": null
},
"0,8": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": "0,3"
}
},
"0,9": {
"terrain": "GRASS",
"mine": {
"mine_type": "time",
"timer": 20
}
},
"1,0": {
"terrain": "GRASS",
"mine": null
},
"1,1": {
"terrain": "GRASS",
"mine": {
"mine_type": "standard"
}
},
"1,2": {
"terrain": "GRASS",
"mine": null
},
"1,3": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"1,4": {
"terrain": "MUD",
"mine": null
},
"1,5": {
"terrain": "GRASS",
"mine": null
},
"1,6": {
"terrain": "GRASS",
"mine": null
},
"1,7": {
"terrain": "GRASS",
"mine": null
},
"1,8": {
"terrain": "MUD",
"mine": null
},
"1,9": {
"terrain": "MUD",
"mine": null
},
"2,0": {
"terrain": "MUD",
"mine": null
},
"2,1": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "standard"
}
},
"2,2": {
"terrain": "MUD",
"mine": null
},
"2,3": {
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"2,4": {
"terrain": "GRASS",
"mine": null
},
"2,5": {
"terrain": "MUD",
"mine": {
"mine_type": "standard"
}
},
"2,6": {
"terrain": "CONCRETE",
"mine": null
},
"2,7": {
"terrain": "CONCRETE",
"mine": null
},
"2,8": {
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"2,9": {
"terrain": "CONCRETE",
"mine": null
},
"3,0": {
"terrain": "CONCRETE",
"mine": null
},
"3,1": {
"terrain": "MUD",
"mine": {
"mine_type": "time",
"timer": 19
}
},
"3,2": {
"terrain": "GRASS",
"mine": null
},
"3,3": {
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": "1,3"
}
},
"3,4": {
"terrain": "MUD",
"mine": null
},
"3,5": {
"terrain": "CONCRETE",
"mine": null
},
"3,6": {
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"3,7": {
"terrain": "MUD",
"mine": null
},
"3,8": {
"terrain": "GRASS",
"mine": null
},
"3,9": {
"terrain": "GRASS",
"mine": null
},
"4,0": {
"terrain": "CONCRETE",
"mine": null
},
"4,1": {
"terrain": "MUD",
"mine": null
},
"4,2": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "standard"
}
},
"4,3": {
"terrain": "CONCRETE",
"mine": null
},
"4,4": {
"terrain": "MUD",
"mine": null
},
"4,5": {
"terrain": "MUD",
"mine": null
},
"4,6": {
"terrain": "CONCRETE",
"mine": null
},
"4,7": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"4,8": {
"terrain": "CONCRETE",
"mine": null
},
"4,9": {
"terrain": "MUD",
"mine": null
},
"5,0": {
"terrain": "CONCRETE",
"mine": null
},
"5,1": {
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": "2,3"
}
},
"5,2": {
"terrain": "MUD",
"mine": null
},
"5,3": {
"terrain": "GRASS",
"mine": null
},
"5,4": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": "2,8"
}
},
"5,5": {
"terrain": "CONCRETE",
"mine": null
},
"5,6": {
"terrain": "MUD",
"mine": null
},
"5,7": {
"terrain": "MUD",
"mine": null
},
"5,8": {
"terrain": "GRASS",
"mine": null
},
"5,9": {
"terrain": "MUD",
"mine": null
},
"6,0": {
"terrain": "GRASS",
"mine": null
},
"6,1": {
"terrain": "CONCRETE",
"mine": null
},
"6,2": {
"terrain": "MUD",
"mine": null
},
"6,3": {
"terrain": "CONCRETE",
"mine": null
},
"6,4": {
"terrain": "CONCRETE",
"mine": null
},
"6,5": {
"terrain": "GRASS",
"mine": {
"mine_type": "standard"
}
},
"6,6": {
"terrain": "GRASS",
"mine": null
},
"6,7": {
"terrain": "CONCRETE",
"mine": null
},
"6,8": {
"terrain": "CONCRETE",
"mine": null
},
"6,9": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "standard"
}
},
"7,0": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": "4,7"
}
},
"7,1": {
"terrain": "CONCRETE",
"mine": null
},
"7,2": {
"terrain": "GRASS",
"mine": null
},
"7,3": {
"terrain": "MUD",
"mine": null
},
"7,4": {
"terrain": "GRASS",
"mine": null
},
"7,5": {
"terrain": "GRASS",
"mine": {
"mine_type": "standard"
}
},
"7,6": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "time",
"timer": 39
}
},
"7,7": {
"terrain": "MUD",
"mine": null
},
"7,8": {
"terrain": "GRASS",
"mine": null
},
"7,9": {
"terrain": "GRASS",
"mine": null
},
"8,0": {
"terrain": "GRASS",
"mine": null
},
"8,1": {
"terrain": "GRASS",
"mine": {
"mine_type": "time",
"timer": 24
}
},
"8,2": {
"terrain": "GRASS",
"mine": null
},
"8,3": {
"terrain": "CONCRETE",
"mine": {
"mine_type": "standard"
}
},
"8,4": {
"terrain": "CONCRETE",
"mine": null
},
"8,5": {
"terrain": "GRASS",
"mine": null
},
"8,6": {
"terrain": "GRASS",
"mine": null
},
"8,7": {
"terrain": "CONCRETE",
"mine": null
},
"8,8": {
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "3,6"
}
},
"8,9": {
"terrain": "GRASS",
"mine": null
},
"9,0": {
"terrain": "MUD",
"mine": null
},
"9,1": {
"terrain": "GRASS",
"mine": null
},
"9,2": {
"terrain": "CONCRETE",
"mine": null
},
"9,3": {
"terrain": "CONCRETE",
"mine": null
},
"9,4": {
"terrain": "CONCRETE",
"mine": null
},
"9,5": {
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "0,4"
}
},
"9,6": {
"terrain": "CONCRETE",
"mine": null
},
"9,7": {
"terrain": "MUD",
"mine": null
},
"9,8": {
"terrain": "MUD",
"mine": null
},
"9,9": {
"terrain": "MUD",
"mine": null
},
"agents_initial_state": {
"direction": 1,
"position": "0,0"
}
}

View File

@ -1,456 +1,456 @@
{
"0,0": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"0,1": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"0,2": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"0,3": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"0,4": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"0,5": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"0,6": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"0,7": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"0,8": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"0,9": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "time",
"timer": 28
}
},
"1,0": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"1,1": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"1,2": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"1,3": {
"color": "WHITE",
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "8,9"
}
},
"1,4": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"1,5": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"1,6": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"1,7": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"1,8": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"1,9": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"2,0": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"2,1": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "time",
"timer": 27
}
},
"2,2": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"2,3": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"2,4": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"2,5": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"2,6": {
"color": "BLUE",
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "7,2"
}
},
"2,7": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"2,8": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"2,9": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"3,0": {
"color": "WHITE",
"terrain": "GRASS",
"mine": {
"mine_type": "time",
"timer": 27
}
},
"3,1": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"3,2": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"3,3": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"3,4": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": "8,8"
}
},
"3,5": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"3,6": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"3,7": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"3,8": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"3,9": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"4,0": {
"color": "GREEN",
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"4,1": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "standard"
}
},
"4,2": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"4,3": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"4,4": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"4,5": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"4,6": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"4,7": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"4,8": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"4,9": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"5,0": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"5,1": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"5,2": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"5,3": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": {
"mine_type": "time",
"timer": 25
}
},
"5,4": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"5,5": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"5,6": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"5,7": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"5,8": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"5,9": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"6,0": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": "6,1"
}
},
"6,1": {
"color": "BLUE",
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "3,4"
}
},
"6,2": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"6,3": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"6,4": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"6,5": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"6,6": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"6,7": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"6,8": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"6,9": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"7,0": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": {
"mine_type": "time",
"timer": 30
}
},
"7,1": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"7,2": {
"color": "WHITE",
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "7,3"
}
},
"7,3": {
"color": "GREEN",
"terrain": "MUD",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"7,4": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"7,5": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "standard"
}
},
"7,6": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"7,7": {
"color": "BLUE",
"terrain": "GRASS",
"mine": null
},
"7,8": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"7,9": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"8,0": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"8,1": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"8,2": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"8,3": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"8,4": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"8,5": {
"color": "YELLOW",
"terrain": "GRASS",
"mine": null
},
"8,6": {
"color": "WHITE",
"terrain": "GRASS",
"mine": {
"mine_type": "standard"
}
},
"8,7": {
"color": "GREEN",
"terrain": "MUD",
"mine": null
},
"8,8": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"8,9": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": {
"mine_type": "chained",
"predecessor": null
}
},
"9,0": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"9,1": {
"color": "BLUE",
"terrain": "GRASS",
"mine": {
"mine_type": "chained",
"predecessor": "4,0"
}
},
"9,2": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"9,3": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"9,4": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"9,5": {
"color": "PURPLE",
"terrain": "CONCRETE",
"mine": null
},
"9,6": {
"color": "WHITE",
"terrain": "GRASS",
"mine": null
},
"9,7": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"9,8": {
"color": "ORANGE",
"terrain": "CONCRETE",
"mine": null
},
"9,9": {
"color": "RED",
"terrain": "MUD",
"mine": null
},
"agents_initial_state": {

View File

@ -47,13 +47,15 @@ def get_successors(state: State, minefield: Minefield):
return successors
def graphsearch(initial_state: State, minefield: Minefield, fringe: List[Node] = None, explored: List[Node] = None):
def graphsearch(initial_state: State, minefield: Minefield, fringe: List[Node] = None, explored: List[Node] = None, tox: int = None,toy: int = None):
# fringe and explored initialization
global GOAL
if fringe is None:
fringe = list()
if explored is None:
explored = list()
if tox is not None and toy is not None:
GOAL = (tox, toy)
explored_states = set()
fringe_states = set()

25
tile.py
View File

@ -1,7 +1,24 @@
class Tile:
def __init__(self, position, color=None, mine=None):
self.position = position
self.color = color
from project_constants import Terrain
# Assume_cost function assumes colour as an argument (colour that is already given to a tile) and depending
# on what it is returns value of Terrein Enum
# It is used in Tile.cost (giving the value to the tile)
def assume_cost(terrain_type):
if terrain_type == "CONCRETE":
return Terrain.CONCRETE
elif terrain_type == "GRASS":
return Terrain.GRASS
elif terrain_type == "MUD":
return Terrain.MUD
class Tile:
def __init__(self, position, terrain_type=None, mine=None):
self.position = position
self.terrain_type = terrain_type
self.cost = assume_cost(terrain_type)
# mine is an instance of Mine class
self.mine = mine

View File

@ -0,0 +1,6 @@
def _return_itself(fun):
def wrapper(*args, **kwargs):
fun(*args, **kwargs)
return wrapper

View File

@ -1,8 +1,14 @@
import pygame
from math import floor
# importing an auxiliary functions
from ui.auxiliary_decorator import _return_itself
from ui.text_box import get_fixed_rgb
# importing parent class
from ui.text_box import TextBox
class Button:
class Button(TextBox):
# constructor that can be used to set parameters of Button instance
def __init__(
self,
@ -17,90 +23,82 @@ class Button:
is_active=True,
fit_text=True,
outline=True,
irregular_outline=False,
outline_additional_pixel=False,
outline_hover_inclusive=True,
outline_thickness=None,
outline_color=(50, 50, 50),
box_transform_hover=1.16,
box_transform_inactive=0.9,
font_transform_hover=1.24,
font_transform_inactive=0.85
font_transform_inactive=0.85,
outline_transform_hover=1,
outline_transform_inactive=0.9
):
# extracting and setting values
smaller_dimension = min(dimensions)
width, height = dimensions
# counting default outline thickness
default_outline_thickness = (smaller_dimension + (width + height) / 4) / 50
# calling base class constructor
super().__init__(
position=position,
dimensions=dimensions,
text=text,
box_color=box_color,
font=font,
font_color=font_color,
font_size=font_size,
is_visible=is_visible,
fit_text=fit_text,
outline=outline,
irregular_outline=irregular_outline,
outline_additional_pixel=outline_additional_pixel,
outline_hover_inclusive=outline_hover_inclusive,
outline_thickness=outline_thickness,
outline_color=outline_color
)
# setting attributes
self.position = position
self.dimensions = dimensions
self.text = text
self.box_color = _get_proper_rgb(box_color)
self.font_color = _get_proper_rgb(font_color)
self.font_size = _return_value_or_default(font_size, floor(smaller_dimension / 2))
self.font = pygame.font.SysFont(font, self.font_size)
self.is_visible = is_visible
self.is_active = is_active
self.fit_text = fit_text
self.outline = outline
self.outline_thickness = _return_value_or_default(outline_thickness, default_outline_thickness)
self.outline_color = _get_proper_rgb(outline_color)
self.box_transform_hover = box_transform_hover
self.box_transform_inactive = box_transform_inactive
self.font_transform_hover = font_transform_hover
self.font_transform_inactive = font_transform_inactive
# rendering text to get it's width and height
rendered_text = self.font.render(text, True, (0, 0, 0))
# if text is out of bounds and fit_text=True - resizing text to fit the box
if self.fit_text and rendered_text.get_width() > self.dimensions[0]:
self.font_size = floor(self.font_size / (rendered_text.get_width() / width + 0.1))
self.font = pygame.font.SysFont(font, self.font_size)
self.outline_transform_hover = outline_transform_hover
self.outline_transform_inactive = outline_transform_inactive
# counting colors on: hover, inactive
self.box_hover_color = _get_proper_rgb(tuple(_transform(self.box_color, self.box_transform_hover)))
self.box_inactive_color = _get_proper_rgb(tuple(_transform(self.box_color, self.box_transform_inactive)))
self.font_hover_color = _get_proper_rgb(tuple(_transform(self.font_color, self.font_transform_hover)))
self.font_inactive_color = _get_proper_rgb(tuple(_transform(self.font_color, self.font_transform_inactive)))
self.box_hover_color = get_fixed_rgb(_transform_rgb(self.box_color, self.box_transform_hover))
self.box_inactive_color = get_fixed_rgb(_transform_rgb(self.box_color, self.box_transform_inactive))
self.font_hover_color = get_fixed_rgb(_transform_rgb(self.font_color, self.font_transform_hover))
self.font_inactive_color = get_fixed_rgb(_transform_rgb(self.font_color, self.font_transform_inactive))
self.outline_hover_color = get_fixed_rgb(_transform_rgb(self.outline_color, self.outline_transform_hover))
self.outline_inactive_color = get_fixed_rgb(_transform_rgb(self.outline_color, self.outline_transform_inactive))
# draws the Button instance
def draw(self, window, mouse_position):
# if is_visible=True - drawing Button
if self.is_visible:
# extracting and setting values from attributes
box_x, box_y = self.position
width, height = self.dimensions
# if outline=True - drawing an outline
if self.outline:
padding = self.outline_thickness
outline_coordinates = (box_x - padding, box_y - padding, width + 2 * padding, height + 2 * padding)
pygame.draw.rect(window, self.outline_color, outline_coordinates)
# setting values accordingly to InputBox'es state
# is active and mouse is not over
if not self.is_over(mouse_position) and self.is_active:
pygame.draw.rect(window, self.box_color, (box_x, box_y, width, height))
text_color = self.font_color
@_return_itself
def draw(self, window, mouse_position, *args, **kwargs):
# saving attribute values
box_color_attribute_value = self.box_color
text_color_attribute_value = self.font_color
outline_attribute_value = self.outline_color
# temporarily changing box and font color if necessary
# is active and mouse is over
elif self.is_active:
pygame.draw.rect(window, self.box_hover_color, (box_x, box_y, width, height))
text_color = self.font_hover_color
if self.is_active and self.is_over(mouse_position):
self.box_color = self.box_hover_color
self.font_color = self.font_hover_color
self.outline_color = self.outline_hover_color
# is not active
else:
pygame.draw.rect(window, self.box_inactive_color, (box_x, box_y, width, height))
text_color = self.font_inactive_color
self.box_color = self.box_inactive_color
self.font_color = self.font_inactive_color
self.outline_color = self.outline_inactive_color
# rendering text and counting its coordinates
rendered_text = self.font.render(self.text, True, text_color)
rendered_text_x = box_x + width / 2 - rendered_text.get_width() / 2
rendered_text_y = box_y + height / 2 - rendered_text.get_height() / 2
# calling the draw function of base class
super().draw(window)
# drawing the text on the coordinates
window.blit(rendered_text, (rendered_text_x, rendered_text_y))
# resetting colors to original values
self.box_color = box_color_attribute_value
self.font_color = text_color_attribute_value
self.outline_color = outline_attribute_value
# returns True if the Button is clicked, or False otherwise
def is_clicked(self, mouse_position, events):
@ -114,49 +112,6 @@ class Button:
return False
# checks if a position is over the InputBox and returns True or False
def is_over(self, position):
mouse_x, mouse_y = position
button_x, button_y = self.position
width, height = self.dimensions
return button_x <= mouse_x <= (button_x + width) and button_y <= mouse_y <= (button_y + height)
# returns text value
def get_text(self):
return self.text
# sets chosen coordinates
def set_coordinates(self, position=None, dimensions=None, outline_thickness=None):
if position is not None:
self.position = position
if dimensions is not None:
self.dimensions = dimensions
if outline_thickness is not None:
self.outline_thickness = outline_thickness
# sets text attribute
def set_text(self, text):
self.text = text
# sets chosen font attributes
def set_font(self, font=None, font_size=None):
if font_size is not None:
self.font_size = font_size
# rendering text to get it's width and height
rendered_text = self.font.render(self.text, True, (0, 0, 0))
# if text is out of bounds and fit_text=True - resizing text to fit the box
if self.fit_text and rendered_text.get_width() > self.dimensions[0]:
self.font_size = floor(self.font_size / (rendered_text.get_width() / self.dimensions[0] + 0.1))
self.font = pygame.font.SysFont(font, self.font_size)
if font is not None:
self.font = pygame.font.SysFont(font, self.font_size)
# sets chosen color attributes
def set_colors(
self,
@ -168,14 +123,7 @@ class Button:
font_transform_hover=None,
font_transform_inactive=None
):
if box_color is not None:
self.box_color = _get_proper_rgb(box_color)
if font_color is not None:
self.font_color = _get_proper_rgb(font_color)
if outline_color is not None:
self.outline_color = _get_proper_rgb(outline_color)
super().set_colors(box_color=box_color, font_color=font_color, outline_color=outline_color)
if box_transform_hover is not None:
self.box_transform_hover = box_transform_hover
@ -189,33 +137,44 @@ class Button:
if font_transform_inactive is not None:
self.font_transform_inactive = font_transform_inactive
self.box_hover_color = _get_proper_rgb(tuple(_transform(self.box_color, self.box_transform_hover)))
self.box_inactive_color = _get_proper_rgb(tuple(_transform(self.box_color, self.box_transform_inactive)))
self.font_hover_color = _get_proper_rgb(tuple(_transform(self.font_color, self.font_transform_hover)))
self.font_inactive_color = _get_proper_rgb(tuple(_transform(self.font_color, self.font_transform_inactive)))
self.box_hover_color = get_fixed_rgb(tuple(_transform_rgb(self.box_color, self.box_transform_hover)))
self.box_inactive_color = get_fixed_rgb(tuple(_transform_rgb(self.box_color, self.box_transform_inactive)))
self.font_hover_color = get_fixed_rgb(tuple(_transform_rgb(self.font_color, self.font_transform_hover)))
self.font_inactive_color = get_fixed_rgb(tuple(_transform_rgb(self.font_color, self.font_transform_inactive)))
# sets chosen flags
def set_flags(self, is_visible=None, is_active=None, outline=None):
if is_visible is not None:
self.is_visible = is_visible
def set_flags(
self,
is_visible=None,
is_active=None,
outline=None,
irregular_outline=None,
outline_additional_pixel=None,
outline_hover_inclusive=None
):
super().set_flags(
is_visible=is_visible,
outline=outline,
irregular_outline=irregular_outline,
outline_additional_pixel=outline_additional_pixel,
outline_hover_inclusive=outline_hover_inclusive
)
if is_active is not None:
self.is_active = is_active
if outline is not None:
self.outline = outline
def _transform_rgb(iterable, transform_value):
if isinstance(transform_value, int) or isinstance(transform_value, float):
return tuple([x * transform_value for x in iterable])
def _get_proper_rgb(x):
r, g, b = x
r, g, b = max(0, min(255, floor(r))), max(0, min(255, floor(g))), max(0, min(255, floor(b)))
elif isinstance(transform_value, tuple) or isinstance(transform_value, list):
if len(transform_value) == 3 and all([isinstance(x, int) or isinstance(x, float) for x in transform_value]):
rgb_list = [x for x in iterable]
return tuple((r, g, b))
for i in range(3):
rgb_list[i] += transform_value[i]
return tuple(rgb_list)
def _transform(iterable, transform_value):
return [x * transform_value for x in iterable]
def _return_value_or_default(value, default):
return default if value is None else value
return [0, 0, 0]

View File

@ -1,24 +1,33 @@
import pygame
from math import floor
from ui.auxiliary_decorator import _return_itself
class InputBox:
from ui.button import Button
from ui.button import get_fixed_rgb
from ui.button import _transform_rgb
from ui.text_box import _return_value_or_default
class InputBox(Button):
backspace_tick = 0
backspace_deleted_streak = 0
backspace_breakpoint = 6
deleting_speeds = [12, 4]
was_enter_hit = False
# constructor that can be used to set parameters of InputBox instance
def __init__(
self,
position,
dimensions,
text="",
input_box_position=None,
input_box_dimensions=None,
label="",
user_input="",
box_color=(195, 195, 195),
input_box_color=(225, 245, 245),
writing_highlight_color=(150, 255, 255),
inner_box_color=(205, 205, 205),
bottom_strip_color=(180, 180, 180),
font=pygame.font.get_default_font(),
@ -28,13 +37,46 @@ class InputBox:
is_active=True,
input_centered=False,
fit_text=True,
clear_input_on_click=False,
outline=True,
irregular_outline=False,
outline_additional_pixel=False,
outline_hover_inclusive=True,
outline_thickness=None,
outline_color=(50, 50, 50),
box_transform_hover=1,
box_transform_inactive=1,
font_transform_hover=1,
font_transform_inactive=1,
outline_transform_hover=1,
outline_transform_inactive=0.9,
input_box_transform_hover=1.07,
input_box_transform_selected=2,
input_box_transform_inactive=0.9
):
super().__init__(
position=position,
dimensions=dimensions,
text=text,
box_color=box_color,
font_color=font_color,
is_visible=is_visible,
is_active=is_active,
fit_text=fit_text,
outline=outline,
irregular_outline=irregular_outline,
outline_additional_pixel=outline_additional_pixel,
outline_hover_inclusive=outline_hover_inclusive,
outline_thickness=outline_thickness,
outline_color=outline_color,
box_transform_hover=box_transform_hover,
box_transform_inactive=box_transform_inactive,
font_transform_hover=font_transform_hover,
font_transform_inactive=font_transform_inactive,
outline_transform_hover=outline_transform_hover,
outline_transform_inactive=outline_transform_inactive
)
# extracting and setting values
smaller_dimension = min(dimensions)
box_x, box_y = position
@ -49,59 +91,59 @@ class InputBox:
default_input_box_position = (box_x + width / 12, box_y + height / 2)
default_input_box_dimensions = (width - width / 6, 3 * height / 10)
# counting default outline thickness
default_outline_thickness = (smaller_dimension + (width + height) / 4) / 50
# setting attributes
self.position = position
self.dimensions = dimensions
self.input_box_position = _return_value_or_default(input_box_position, default_input_box_position)
self.input_box_dimensions = _return_value_or_default(input_box_dimensions, default_input_box_dimensions)
self.label = label
self.user_input = user_input
self.box_color = _get_proper_rgb(box_color)
self.ib_color = _get_proper_rgb(input_box_color)
self.inner_box_color = _get_proper_rgb(inner_box_color)
self.bottom_strip_color = _get_proper_rgb(bottom_strip_color)
self.font_color = _get_proper_rgb(font_color)
self.ib_color = get_fixed_rgb(input_box_color)
self.typing_highlight_color = get_fixed_rgb(writing_highlight_color)
self.inner_box_color = get_fixed_rgb(inner_box_color)
self.bottom_strip_color = get_fixed_rgb(bottom_strip_color)
self.clear_input_on_click = clear_input_on_click
self.font_size = _return_value_or_default(font_size, floor(smaller_dimension / 2.8))
self.font = pygame.font.SysFont(font, self.font_size)
self.input_font_size = max(15, floor(self.input_box_dimensions[1] - 2))
self.input_font = pygame.font.SysFont(font, self.input_font_size)
self.is_visible = is_visible
self.is_active = is_active
self.is_selected = False
self.is_active = is_active
self.input_centered = input_centered
self.fit_text = fit_text
self.outline = outline
self.outline_thickness = _return_value_or_default(outline_thickness, default_outline_thickness)
self.outline_color = outline_color
self.ib_transform_hover = input_box_transform_hover
self.ib_transform_selected = input_box_transform_selected
self.ib_transform_inactive = input_box_transform_inactive
self.backspace_pressed_down = False
# rendering label to get it's width and height
rendered_label = self.font.render(label, True, (0, 0, 0))
# rendering text to get it's width and height
rendered_text = self.font.render(text, True, (0, 0, 0))
# if text is out of bounds and fit_text=True - resizing text to fit the box
if self.fit_text and rendered_label.get_width() > self.input_box_dimensions[0]:
self.font_size = floor(self.font_size / (rendered_label.get_width() / width + 0.1))
if self.fit_text and rendered_text.get_width() > self.input_box_dimensions[0]:
self.font_size = floor(self.font_size / (rendered_text.get_width() / width + 0.1))
self.font = pygame.font.SysFont(font, self.font_size)
# counting colors on: hover, inactive, selected
self.ib_hover_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_hover)))
self.ib_inactive_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_inactive)))
self.ib_selected_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_selected)))
self.ib_hover_color = get_fixed_rgb(tuple(_transform_rgb(self.ib_color, self.ib_transform_hover)))
self.ib_inactive_color = get_fixed_rgb(tuple(_transform_rgb(self.ib_color, self.ib_transform_inactive)))
self.ib_selected_color = get_fixed_rgb(tuple(_transform_rgb(self.ib_color, self.ib_transform_selected)))
# draws, updates and selects on mouse click the InputBox instance
def run(self, window, mouse_position, events):
self.select_on_click(mouse_position, events)
self.clear_user_input_on_click(mouse_position, events)
self.draw(window, mouse_position)
self.update(events)
@_return_itself
# draws an InputBox instance (doesn't run interactions [for everything use run(self, ...) method])
def draw(self, window, mouse_position):
def draw(self, window, mouse_position, *args, **kwargs):
text_attribute_value = self.text
self.text = ""
super().draw(window, mouse_position, *args, **kwargs)
# resetting original attribute values
self.text = text_attribute_value
# if is_visible=True drawing the InputBox
if self.is_visible:
# extracting and setting values from attributes
@ -110,12 +152,6 @@ class InputBox:
input_box_x, input_box_y = self.input_box_position
input_width, input_height = self.input_box_dimensions
# if outline=True - drawing an outline
if self.outline:
padding = self.outline_thickness
outline_coordinates = (box_x - padding, box_y - padding, width + 2 * padding, height + 2 * padding)
pygame.draw.rect(window, self.outline_color, outline_coordinates)
# setting "transparent" background color
text_background_color = self.ib_color
@ -123,7 +159,7 @@ class InputBox:
# is selected
if self.is_selected:
input_box_color = self.ib_selected_color
text_background_color = (150, 255, 255)
text_background_color = self.typing_highlight_color
# is active and mouse is not over
elif not self.is_over(mouse_position) and self.is_active:
@ -137,9 +173,6 @@ class InputBox:
else:
input_box_color = self.ib_inactive_color
# drawing outer box
pygame.draw.rect(window, self.box_color, (box_x, box_y, width, height))
# drawing inner (upper) box
pygame.draw.rect(window,
self.inner_box_color,
@ -152,13 +185,13 @@ class InputBox:
# drawing bottom strip
pygame.draw.rect(window, self.bottom_strip_color, (box_x, box_y + 9 * height / 10, width, 1 * height / 10))
# rendering label and counting its coordinates
rendered_label = self.font.render(self.label, True, self.font_color)
rendered_label_x = box_x + width / 2 - rendered_label.get_width() / 2
rendered_label_y = box_y + height / 4 - rendered_label.get_height() / 2
# rendering text and counting its coordinates
rendered_text = self.font.render(self.text, True, self.font_color)
rendered_text_x = box_x + width / 2 - rendered_text.get_width() / 2
rendered_text_y = box_y + height / 4 - rendered_text.get_height() / 2
# drawing the label on the coordinates
window.blit(rendered_label, (rendered_label_x, rendered_label_y))
# drawing the text on the coordinates
window.blit(rendered_text, (rendered_text_x, rendered_text_y))
# rendering input
rendered_input_text = self.input_font.render(self.user_input, True, self.font_color, text_background_color)
@ -177,6 +210,9 @@ class InputBox:
# updates InputBox'es attributes if user types something in it
def update(self, events):
# resetting attribute value
self.was_enter_hit = False
# if the InputBox is selected - updating the input text
if self.is_selected:
self._delete_characters_if_backspace_pressed()
@ -194,6 +230,7 @@ class InputBox:
elif event.key == pygame.K_RETURN:
self.is_selected = False
self.was_enter_hit = True
# if text isn't too long - adding a character
elif rendered_input.get_width() + 10 < self.input_box_dimensions[0]:
@ -210,28 +247,42 @@ class InputBox:
# if is active set is_selected attribute accordingly
if self.is_active:
for event in events:
# if mouse is clicked set as selected if mouse is over, or set as not selected otherwise
if event.type == pygame.MOUSEBUTTONUP:
self.set_is_selected(self.is_over(mouse_position))
# checks if a position is over the InputBox and returns True or False
def is_over(self, position):
mouse_x, mouse_y = position
button_x, button_y = self.position
width, height = self.dimensions
# clears user's input on click
def clear_user_input_on_click(self, mouse_position, events):
if self.is_active and self.clear_input_on_click:
for event in events:
return button_x <= mouse_x <= (button_x + width) and button_y <= mouse_y <= (button_y + height)
if event.type == pygame.MOUSEBUTTONUP and self.is_over(mouse_position):
self.clear_input()
# returns InputBox'es label
def get_label(self):
return self.label
# returns True if user input is empty, or False otherwise
def is_input_empty(self):
return not any(self.user_input)
# clears user's input
def clear_input(self):
self.user_input = ""
# returns True if enter key is hit, or False otherwise
def is_enter_hit(self, events):
if self.is_active and self.is_selected:
for event in events:
if event.type == pygame.KEYDOWN and event.key == pygame.K_RETURN:
return True
return self.was_enter_hit
# returns InputBox'es user's input
def get_input(self):
return self.user_input
# sets chosen coordinates
def set_space(
def set_coordinates(
self,
position=None,
dimensions=None,
@ -239,11 +290,7 @@ class InputBox:
input_box_dimensions=None,
outline_thickness=None
):
if position is not None:
self.position = position
if dimensions is not None:
self.dimensions = dimensions
super().set_coordinates(position=position, dimensions=dimensions, outline_thickness=outline_thickness)
if input_box_position is not None:
self.input_box_position = input_box_position
@ -251,13 +298,9 @@ class InputBox:
if input_box_dimensions is not None:
self.input_box_dimensions = input_box_dimensions
if outline_thickness is not None:
self.outline_thickness = outline_thickness
# sets chosen text attributes
def set_texts(self, label=None, user_input=None):
if label is not None:
self.label = label
def set_texts(self, text=None, user_input=None):
super().set_text(text=text)
if user_input is not None:
self.user_input = user_input
@ -268,15 +311,14 @@ class InputBox:
self.font_size = font_size
# rendering text to get it's width and height
rendered_text = self.font.render(self.user_input, True, (0, 0, 0))
rendered_text = self.font.render(self.text, True, (0, 0, 0))
# if text is out of bounds and fit_text=True - resizing text to fit the box
if self.fit_text and rendered_text.get_width() > self.dimensions[0]:
if self.fit_text and rendered_text.get_width() > self.input_box_dimensions[0]:
self.font_size = floor(self.font_size / (rendered_text.get_width() / self.dimensions[0] + 0.1))
self.font = pygame.font.SysFont(font, self.font_size)
if font is not None:
self.font = pygame.font.SysFont(font, self.font_size)
super().set_font(font=font)
# sets chosen color attributes
def set_colors(
@ -284,30 +326,39 @@ class InputBox:
box_color=None,
input_box_color=None,
inner_box_color=None,
writing_highlight_color=None,
bottom_strip_color=None,
font_color=None,
outline_color=None,
box_transform_hover=None,
box_transform_inactive=None,
font_transform_hover=None,
font_transform_inactive=None,
input_box_transform_hover=None,
input_box_transform_selected=None,
input_box_transform_inactive=None
input_box_transform_inactive=None,
):
if box_color is not None:
self.box_color = _get_proper_rgb(box_color)
super().set_colors(
box_color=box_color,
font_color=font_color,
outline_color=outline_color,
box_transform_hover=box_transform_hover,
box_transform_inactive=box_transform_inactive,
font_transform_hover=font_transform_hover,
font_transform_inactive=font_transform_inactive
)
if input_box_color is not None:
self.ib_color = _get_proper_rgb(input_box_color)
self.ib_color = get_fixed_rgb(input_box_color)
if writing_highlight_color is not None:
self.typing_highlight_color = writing_highlight_color
if inner_box_color is not None:
self.inner_box_color = _get_proper_rgb(inner_box_color)
self.inner_box_color = get_fixed_rgb(inner_box_color)
if bottom_strip_color is not None:
self.bottom_strip_color = _get_proper_rgb(bottom_strip_color)
if font_color is not None:
self.font_color = _get_proper_rgb(font_color)
if outline_color is not None:
self.outline_color = _get_proper_rgb(outline_color)
self.bottom_strip_color = get_fixed_rgb(bottom_strip_color)
if input_box_transform_hover is not None:
self.ib_transform_hover = input_box_transform_hover
@ -318,14 +369,29 @@ class InputBox:
if input_box_transform_inactive is not None:
self.ib_transform_inactive = input_box_transform_inactive
self.ib_hover_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_hover)))
self.ib_inactive_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_inactive)))
self.ib_selected_color = _get_proper_rgb(tuple(_transform(self.ib_color, self.ib_transform_selected)))
self.ib_hover_color = get_fixed_rgb(tuple(_transform_rgb(self.ib_color, self.ib_transform_hover)))
self.ib_inactive_color = get_fixed_rgb(tuple(_transform_rgb(self.ib_color, self.ib_transform_inactive)))
self.ib_selected_color = get_fixed_rgb(tuple(_transform_rgb(self.ib_color, self.ib_transform_selected)))
# sets chosen flag attributes
def set_flags(self, is_visible=None, is_active=None, input_centered=None, fit_text=None, outline=None):
if is_visible is not None:
self.is_visible = is_visible
def set_flags(
self,
is_visible=None,
is_active=None,
input_centered=None,
fit_text=None,
outline=None,
irregular_outline=None,
outline_additional_pixel=None,
outline_hover_inclusive=None
):
super().set_flags(
is_visible=is_visible,
outline=outline,
irregular_outline=None,
outline_additional_pixel=None,
outline_hover_inclusive=None
)
if is_active is not None:
self.is_active = is_active
@ -336,9 +402,6 @@ class InputBox:
if fit_text is not None:
self.fit_text = fit_text
if outline is not None:
self.outline = outline
# sets is_selected with a given value
def set_is_selected(self, is_selected):
self.is_selected = is_selected
@ -360,18 +423,3 @@ class InputBox:
elif self.backspace_pressed_down:
self.backspace_tick += 1
def _get_proper_rgb(x):
r, g, b = x
r, g, b = max(0, min(255, floor(r))), max(0, min(255, floor(g))), max(0, min(255, floor(b)))
return tuple((r, g, b))
def _transform(iterable, transform_value):
return [x * transform_value for x in iterable]
def _return_value_or_default(value, default):
return default if value is None else value

View File

@ -16,6 +16,9 @@ class TextBox:
is_visible=True,
fit_text=True,
outline=True,
irregular_outline=False,
outline_additional_pixel=False,
outline_hover_inclusive=True,
outline_thickness=None,
outline_color=(50, 50, 50)
):
@ -24,21 +27,24 @@ class TextBox:
width, height = dimensions
# counting default outline thickness
default_outline_thickness = (smaller_dimension + (width + height) / 4) / 50
default_outline_thickness = max(1, floor((2 * smaller_dimension + width + height) / 200))
# setting attributes
self.position = position
self.dimensions = dimensions
self.text = text
self.box_color = _get_proper_rgb(box_color)
self.font_color = _get_proper_rgb(font_color)
self.box_color = get_fixed_rgb(box_color)
self.font_color = get_fixed_rgb(font_color)
self.font_size = _return_value_or_default(font_size, floor(smaller_dimension / 2))
self.font = pygame.font.SysFont(font, self.font_size)
self.is_visible = is_visible
self.fit_text = fit_text
self.outline = outline
self.irregular_outline = irregular_outline
self.outline_additional_pixel = outline_additional_pixel
self.outline_hover_inclusive = outline_hover_inclusive
self.outline_thickness = _return_value_or_default(outline_thickness, default_outline_thickness)
self.outline_color = _get_proper_rgb(outline_color)
self.outline_color = get_fixed_rgb(outline_color)
# rendering text to get it's width and height
rendered_text = self.font.render(text, True, (0, 0, 0))
@ -49,7 +55,7 @@ class TextBox:
self.font = pygame.font.SysFont(font, self.font_size)
# draws the TextBox
def draw(self, window):
def draw(self, window, *args, **kwargs):
# if is_visible=True drawing the TextBox
if self.is_visible:
# extracting and setting values from attributes
@ -58,8 +64,9 @@ class TextBox:
# if outline=True - drawing an outline
if self.outline:
padding = self.outline_thickness
padding = self.outline_thickness - self.irregular_outline / 2 + self.outline_additional_pixel
outline_coordinates = (box_x - padding, box_y - padding, width + 2 * padding, height + 2 * padding)
pygame.draw.rect(window, self.outline_color, outline_coordinates)
# drawing the box
@ -73,6 +80,27 @@ class TextBox:
# drawing the box on the coordinates
window.blit(rendered_text, (rendered_text_x, rendered_text_y))
# checks if a position is over the InputBox and returns True or False
def is_over(self, position):
mouse_x, mouse_y = position
if not self.outline_hover_inclusive:
button_x, button_y = self.position
width, height = self.dimensions
else:
padding = self.outline_thickness
button_x, button_y = self.position
width, height = self.dimensions
button_x, button_y = button_x - padding, button_y - padding
width, height = width + 2 * padding, height + 2 * padding
return button_x <= mouse_x <= (button_x + width) and button_y <= mouse_y <= (button_y + height)
# returns text value
def get_text(self):
return self.text
# sets chosen coordinates
def set_coordinates(self, position=None, dimensions=None, outline_thickness=None):
if position is not None:
@ -86,6 +114,7 @@ class TextBox:
# sets the TextBox's text
def set_text(self, text):
if text is not None:
self.text = text
# sets chosen font attributes
@ -107,25 +136,41 @@ class TextBox:
# sets chosen color attributes
def set_colors(self, box_color=None, font_color=None, outline_color=None):
if box_color is not None:
self.box_color = _get_proper_rgb(box_color)
self.box_color = get_fixed_rgb(box_color)
if font_color is not None:
self.font_color = _get_proper_rgb(font_color)
self.font_color = get_fixed_rgb(font_color)
if outline_color is not None:
self.outline_color = outline_color
# sets chosen flags
def set_flags(self, is_visible=None, outline=None):
def set_flags(
self,
is_visible=None,
outline=None,
irregular_outline=None,
outline_additional_pixel=None,
outline_hover_inclusive=None
):
if is_visible is not None:
self.is_visible = is_visible
if outline is not None:
self.outline = outline
if irregular_outline is not None:
self.irregular_outline = irregular_outline
if outline_additional_pixel is not None:
self.outline_additional_pixel = outline_additional_pixel
if outline_hover_inclusive is not None:
self.outline_hover_inclusive = outline_hover_inclusive
# returns values that are not out of range
def _get_proper_rgb(x):
def get_fixed_rgb(x):
r, g, b = x
r, g, b = max(0, min(255, floor(r))), max(0, min(255, floor(g))), max(0, min(255, floor(b)))