78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
import pygame
|
|
import enums
|
|
from random import choice
|
|
|
|
BRICKS = [
|
|
" X " + "\n" + "XXX" + "\n" + " X ",
|
|
" X" + "\n" + "XXX",
|
|
"X " + "\n" + "XXX",
|
|
"X",
|
|
"XXXX",
|
|
"XX" + "\n" + "XX",
|
|
" XX" + "\n" + "XX ",
|
|
"XX " + "\n" + " XX",
|
|
"X X" + "\n" + "XXX",
|
|
"XXX" + "\n" + " X "
|
|
]
|
|
|
|
TILE_SIZE = 48
|
|
|
|
class Brick:
|
|
def __init__(self, brick, state = enums.BrickState.Next):
|
|
self.layout = []
|
|
self.color = choice(list(enums.BrickColor))
|
|
self.set_state(state)
|
|
self.angle = 0
|
|
self.direction = None
|
|
self.load_brick(brick)
|
|
self.get_image()
|
|
self.block_image = pygame.transform.scale(self.img, (48, 48)) if self.img else None
|
|
self.draw_brick()
|
|
|
|
def get_image(self):
|
|
match self.color:
|
|
case enums.BrickColor.Magenta:
|
|
self.img = pygame.image.load("assets/magenta_block.png").convert_alpha()
|
|
case enums.BrickColor.Cyan:
|
|
self.img = pygame.image.load("assets/cyan_block.png").convert_alpha()
|
|
case enums.BrickColor.Yellow:
|
|
self.img = pygame.image.load("assets/yellow_block.png").convert_alpha()
|
|
case _:
|
|
self.img = pygame.image.load("assets/magenta_block.png").convert_alpha()
|
|
|
|
def load_brick(self, brick):
|
|
self.layout = [l for l in BRICKS[brick].splitlines()]
|
|
|
|
self.rows = len(self.layout)
|
|
self.cols = len(self.layout[0])
|
|
self.width = self.cols * TILE_SIZE
|
|
self.height = self.rows * TILE_SIZE
|
|
|
|
def draw_brick(self):
|
|
self.brick = pygame.Surface((self.width, self.height))
|
|
for y, row in enumerate(self.layout):
|
|
for x, char in enumerate(row):
|
|
if char == "X":
|
|
self.brick.blit(self.block_image, (1 + x * TILE_SIZE, 1 + y * TILE_SIZE))
|
|
|
|
def is_current(self):
|
|
return True if self.state == enums.BrickState.Current else False
|
|
|
|
def update(self):
|
|
pass
|
|
|
|
def rotate(self):
|
|
print("Rotating")
|
|
|
|
def set_state(self, state):
|
|
self.state = state
|
|
print(f"State set to {self.state}")
|
|
|
|
def move_right(self):
|
|
print("Moving right")
|
|
|
|
def move_left(self):
|
|
print("Moving left")
|
|
|
|
def drop(self):
|
|
print("Dropping") |