43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import random
|
|
|
|
from entity.Piece import Piece
|
|
from util.ConfigurationManager import ConfigurationManager
|
|
|
|
"""
|
|
TODO Add link that goes through random piece generation
|
|
"""
|
|
class PieceGenerator:
|
|
|
|
__pieces = []
|
|
|
|
@classmethod
|
|
def get_piece(cls, position):
|
|
|
|
if len(cls.__pieces) == 0:
|
|
for _ in range(7):
|
|
cls.__pieces.append(Piece(cls.__get_piece_shape(), position, cls.__get_piece_color()))
|
|
|
|
return cls.__pieces.pop()
|
|
|
|
def __get_piece_shape():
|
|
random_number = random.randint(0, 6)
|
|
|
|
if random_number == 0:
|
|
return Piece.I_SHAPE
|
|
if random_number == 1:
|
|
return Piece.J_SHAPE
|
|
if random_number == 2:
|
|
return Piece.L_SHAPE
|
|
if random_number == 3:
|
|
return Piece.O_SHAPE
|
|
if random_number == 4:
|
|
return Piece.S_SHAPE
|
|
if random_number == 5:
|
|
return Piece.T_SHAPE
|
|
if random_number == 6:
|
|
return Piece.Z_SHAPE
|
|
|
|
def __get_piece_color():
|
|
random_number = random.randint(1, 3)
|
|
return ConfigurationManager.configuration["color"]["base-" + str(random_number)]
|