mirror of
https://github.com/kristoferssolo/2048.git
synced 2025-10-21 15:20:35 +00:00
docs(game): add docstrings
This commit is contained in:
parent
c9ef031e1a
commit
b95a46e263
@ -30,6 +30,13 @@ def play() -> None:
|
||||
|
||||
|
||||
class Game2048:
|
||||
"""
|
||||
Initialize the 2048 game board.
|
||||
|
||||
Parameters:
|
||||
- size (int): Size of the board (default is 4)
|
||||
"""
|
||||
|
||||
def __init__(self, size: int = 4):
|
||||
self.size = size
|
||||
self.board = np.zeros((size, size), dtype=int)
|
||||
@ -39,13 +46,25 @@ class Game2048:
|
||||
self.add_random_tile()
|
||||
|
||||
def add_random_tile(self) -> None:
|
||||
"""Add a random tile to the board."""
|
||||
"""
|
||||
Add a random tile (either 2 or 4) to a random empty cell on the board.
|
||||
The probability of getting a 2 or 4 is determined by the Config.tile.probability weights.
|
||||
"""
|
||||
tile_value = random.choices([2, 4], weights=Config.tile.probability)[0]
|
||||
tile_row_options, tile_col_options = np.nonzero(np.logical_not(self.board))
|
||||
tile_loc = np.random.randint(0, len(tile_row_options))
|
||||
self.board[tile_row_options[tile_loc], tile_col_options[tile_loc]] = tile_value
|
||||
|
||||
def move(self, direction: Direction) -> tuple[bool, int]:
|
||||
"""
|
||||
Move the tiles in the specified direction and update the score.
|
||||
|
||||
Parameters:
|
||||
- direction (Direction): The direction to move the tiles.
|
||||
|
||||
Returns:
|
||||
tuple[bool, int]: A tuple indicating whether the move was successful and the updated score.
|
||||
"""
|
||||
match direction:
|
||||
case Direction.LEFT:
|
||||
return self.move_left()
|
||||
@ -57,6 +76,12 @@ class Game2048:
|
||||
return self.move_down()
|
||||
|
||||
def move_right(self) -> tuple[bool, int]:
|
||||
"""
|
||||
Move tiles to the right and update the score.
|
||||
|
||||
Returns:
|
||||
tuple[bool, int]: A tuple indicating whether the move was successful and the updated score.
|
||||
"""
|
||||
self.board, moved = self._move_and_merge(self.board)
|
||||
|
||||
if moved:
|
||||
@ -64,6 +89,12 @@ class Game2048:
|
||||
return moved, self.score
|
||||
|
||||
def move_left(self) -> tuple[bool, int]:
|
||||
"""
|
||||
Move tiles to the left and update the score.
|
||||
|
||||
Returns:
|
||||
tuple[bool, int]: A tuple indicating whether the move was successful and the updated score.
|
||||
"""
|
||||
board = np.rot90(self.board, 2)
|
||||
board, moved = self._move_and_merge(board)
|
||||
self.board = np.rot90(board, 2)
|
||||
@ -73,6 +104,12 @@ class Game2048:
|
||||
return moved, self.score
|
||||
|
||||
def move_up(self) -> tuple[bool, int]:
|
||||
"""
|
||||
Move tiles upwards and update the score.
|
||||
|
||||
Returns:
|
||||
tuple[bool, int]: A tuple indicating whether the move was successful and the updated score.
|
||||
"""
|
||||
board = np.rot90(self.board, -1)
|
||||
board, moved = self._move_and_merge(board)
|
||||
self.board = np.rot90(board, 1)
|
||||
@ -82,6 +119,12 @@ class Game2048:
|
||||
return moved, self.score
|
||||
|
||||
def move_down(self) -> tuple[bool, int]:
|
||||
"""
|
||||
Move tiles downwards and update the score.
|
||||
|
||||
Returns:
|
||||
tuple[bool, int]: A tuple indicating whether the move was successful and the updated score.
|
||||
"""
|
||||
board = np.rot90(self.board, 1)
|
||||
board, moved = self._move_and_merge(board)
|
||||
self.board = np.rot90(board, -1)
|
||||
@ -91,12 +134,30 @@ class Game2048:
|
||||
return moved, self.score
|
||||
|
||||
def _move_and_merge(self, board: np.ndarray) -> tuple[np.ndarray, bool]:
|
||||
"""
|
||||
Move and merge tiles in the specified direction.
|
||||
|
||||
Parameters:
|
||||
- board (np.ndarray): The game board.
|
||||
|
||||
Returns:
|
||||
tuple[np.ndarray, bool]: A tuple containing the updated board and whether any tiles were moved or merged.
|
||||
"""
|
||||
board, has_pushed = self._push_board_right(board)
|
||||
board, has_merged = self._merge(board)
|
||||
board, _ = self._push_board_right(board)
|
||||
return board, has_pushed or has_merged
|
||||
|
||||
def _merge(self, board: np.ndarray) -> tuple[np.ndarray, bool]:
|
||||
"""
|
||||
Merge adjacent tiles with the same value.
|
||||
|
||||
Parameters:
|
||||
- board (np.ndarray): The game board.
|
||||
|
||||
Returns:
|
||||
tuple[np.ndarray, bool]: A tuple containing the updated board and whether any tiles were merged.
|
||||
"""
|
||||
done = False
|
||||
for row in range(self.size):
|
||||
for col in range(self.size - 1, 0, -1):
|
||||
@ -109,6 +170,9 @@ class Game2048:
|
||||
return board, done
|
||||
|
||||
def display(self) -> None:
|
||||
"""
|
||||
Display the current state of the game board and the score.
|
||||
"""
|
||||
print(f"Score: {self.score}")
|
||||
for row in self.board:
|
||||
for val in row:
|
||||
@ -116,6 +180,15 @@ class Game2048:
|
||||
print()
|
||||
|
||||
def _push_board_right(self, board: np.ndarray) -> tuple[np.ndarray, bool]:
|
||||
"""
|
||||
Push tiles to the right, removing empty spaces.
|
||||
|
||||
Parameters:
|
||||
- board (np.ndarray): The game board.
|
||||
|
||||
Returns:
|
||||
tuple[np.ndarray, bool]: A tuple containing the updated board and whether any tiles were pushed.
|
||||
"""
|
||||
new_board = np.zeros((self.size, self.size), dtype=int)
|
||||
done = False
|
||||
for row in range(self.size):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user