formatting and comments

This commit is contained in:
Mateusz Dokowicz 2023-03-16 20:33:22 +01:00
parent 9ac833550c
commit d080b04763
3 changed files with 37 additions and 28 deletions

View File

@ -1,33 +1,33 @@
import sys import sys
import pygame import pygame
class Colors():
class Colors:
BLACK = 0, 0, 0 BLACK = 0, 0, 0
WHITE = 255, 255, 255 WHITE = 255, 255, 255
RED = 255, 0, 0 RED = 255, 0, 0
GREEN = 0, 255, 0 GREEN = 0, 255, 0
DEFAULT_COLOR = Colors.WHITE DEFAULT_COLOR = Colors.WHITE
def default_color(func):
def default_color(func):
def wrap(*args, **kwargs): def wrap(*args, **kwargs):
if 'color' not in kwargs: if "color" not in kwargs:
kwargs['color'] = DEFAULT_COLOR kwargs["color"] = DEFAULT_COLOR
result = func(*args, **kwargs) result = func(*args, **kwargs)
return result return result
return wrap return wrap
class GridDraw():
def __init__( class GridDraw:
self, def __init__(self, width=None, height=None, background=None):
width = None,
height = None,
background = None
):
self.width = width if width != None else 100 self.width = width if width != None else 100
self.height = height if height != None else 100 self.height = height if height != None else 100
self.background = background if background != None else Colors.BLACK self.background = background if background != None else Colors.BLACK
pygame.init() pygame.init()
self.screen = pygame.display.set_mode((self.width, self.height)) self.screen = pygame.display.set_mode((self.width, self.height))
@ -38,11 +38,11 @@ class GridDraw():
pygame.display.flip() pygame.display.flip()
@default_color @default_color
def line(self, x_1, y_1, x_2, y_2, color = None): def line(self, x_1, y_1, x_2, y_2, color=None):
pygame.draw.line(self.screen, color, (x_1, y_1), (x_2, y_2)) pygame.draw.line(self.screen, color, (x_1, y_1), (x_2, y_2))
@default_color @default_color
def board(self, tiles_x, tiles_y, color = None): def board(self, tiles_x, tiles_y, color=None):
tiles_width = self.width / tiles_x tiles_width = self.width / tiles_x
tiles_height = self.height / tiles_y tiles_height = self.height / tiles_y
@ -53,5 +53,5 @@ class GridDraw():
self.line(0, tiles_height * i, self.width, tiles_height * i, color=color) self.line(0, tiles_height * i, self.width, tiles_height * i, color=color)
@default_color @default_color
def circle(self, x, y, radius, color = None): def circle(self, x, y, radius, color=None):
pygame.draw.circle(self.screen, color, (x, y), radius) pygame.draw.circle(self.screen, color, (x, y), radius)

View File

@ -1,17 +1,25 @@
import pygame import pygame
import sys import sys
def moving_cleaner(grid_width, grid_height,tile_width,tile_height, x, y):
def moving_cleaner(grid_width, grid_height, tile_width, tile_height, x, y):
for event in pygame.event.get(): for event in pygame.event.get():
if event.type == pygame.QUIT: if event.type == pygame.QUIT:
sys.exit() sys.exit()
if event.type == pygame.KEYDOWN: if event.type == pygame.KEYDOWN:
if event.key == pygame.K_LEFT and x>(tile_width / 2): # go left
if event.key == pygame.K_LEFT and x > (tile_width / 2):
x -= tile_width x -= tile_width
if event.key == pygame.K_RIGHT and x<(grid_width-(tile_width / 2)):
# go right
if event.key == pygame.K_RIGHT and x < (grid_width - (tile_width / 2)):
x += tile_width x += tile_width
if event.key == pygame.K_UP and y>(tile_height / 2):
# go up
if event.key == pygame.K_UP and y > (tile_height / 2):
y -= tile_height y -= tile_height
if event.key == pygame.K_DOWN and y<(grid_height-(tile_height / 2)):
# go down
if event.key == pygame.K_DOWN and y < (grid_height - (tile_height / 2)):
y += tile_height y += tile_height
return(x,y) return (x, y)

View File

@ -3,7 +3,7 @@ import sys
import pygame import pygame
from Interface.movement import moving_cleaner from Interface.movement import moving_cleaner
# dummy function
def initial_draw(grid_width, grid_height): def initial_draw(grid_width, grid_height):
grid = GridDraw(grid_width, grid_height) grid = GridDraw(grid_width, grid_height)
tile_width = grid_width / 10 tile_width = grid_width / 10
@ -11,11 +11,12 @@ def initial_draw(grid_width, grid_height):
x = tile_width / 2 x = tile_width / 2
y = tile_height / 2 y = tile_height / 2
radius = 15 radius = 15
# rendering loop
while True: while True:
grid.start_draw() grid.start_draw()
grid.board(10, 10) grid.board(10, 10)
(x,y) = moving_cleaner(grid_width,grid_height,tile_width,tile_height, x, y) (x, y) = moving_cleaner(grid_width, grid_height, tile_width, tile_height, x, y)
grid.circle(x, y, 20, color=Colors.RED) grid.circle(x, y, 20, color=Colors.RED)
grid.end_draw() grid.end_draw()
pygame.time.delay(10) pygame.time.delay(10)