160 lines
4.6 KiB
Python
160 lines
4.6 KiB
Python
import random
|
|
|
|
|
|
class Minesweeper:
|
|
board = []
|
|
|
|
def __init__(self, rows=10, cols=10, mines=5, board=None, board_progress=None):
|
|
self.rows = rows
|
|
self.cols = cols
|
|
self.mines = mines
|
|
|
|
if board is not None:
|
|
self.board = board
|
|
|
|
if board_progress is not None:
|
|
self.board_progress = board_progress
|
|
|
|
def create_board(self):
|
|
""" Creating the board cells with 0 as default value """
|
|
self.board = [[0 for col in range(self.cols)] for row in range(self.rows)]
|
|
self.board_progress = [
|
|
["-" for col in range(self.cols)] for row in range(self.rows)
|
|
]
|
|
|
|
def put_mine(self):
|
|
"""Put a single mine on the board.
|
|
The mine have a -1 value just for reference
|
|
"""
|
|
row = random.randrange(0, self.rows)
|
|
col = random.randrange(0, self.cols)
|
|
|
|
if self.is_mine(row, col):
|
|
self.put_mine()
|
|
|
|
self.board[row][col] = -1
|
|
return row, col
|
|
|
|
def put_mines(self):
|
|
""" Put the desired amount of mines on the board """
|
|
for mine in range(1, self.mines + 1):
|
|
row, col = self.put_mine()
|
|
|
|
self.create_mine_points(row, col)
|
|
|
|
def create_mine_points(self, row, col):
|
|
"""Populate the board with points that sorrounds the mine.
|
|
The reference used is the mine that was already placed"""
|
|
|
|
# North
|
|
self.increment_safe_point(row - 1, col)
|
|
|
|
# North-east
|
|
self.increment_safe_point(row - 1, col + 1)
|
|
|
|
# East
|
|
self.increment_safe_point(row, col + 1)
|
|
|
|
# South-east
|
|
self.increment_safe_point(row + 1, col + 1)
|
|
|
|
# South
|
|
self.increment_safe_point(row + 1, col)
|
|
|
|
# South-west
|
|
self.increment_safe_point(row + 1, col - 1)
|
|
|
|
# West
|
|
self.increment_safe_point(row, col - 1)
|
|
|
|
# North-west
|
|
self.increment_safe_point(row - 1, col - 1)
|
|
|
|
def is_mine(self, row, col):
|
|
""" Checks whether the given location have a mine """
|
|
if not self.is_point_in_board(row, col):
|
|
return False
|
|
return self.board[row][col] == -1
|
|
|
|
def is_empty(self, row, col):
|
|
""" Checks whether the given location is empty """
|
|
if not self.is_point_in_board(row, col):
|
|
return False
|
|
return self.board[row][col] == 0
|
|
|
|
def is_point(self, row, col):
|
|
""" Checks whether the given location have pontuation """
|
|
if not self.is_point_in_board(row, col):
|
|
return False
|
|
return self.board[row][col] > 0
|
|
|
|
def is_point_in_board(self, row, col):
|
|
""" Checks whether the location is inside board """
|
|
if row in range(0, self.rows) and col in range(0, self.cols):
|
|
return True
|
|
return False
|
|
|
|
def increment_safe_point(self, row, col):
|
|
""" Creates the mine's pontuation frame """
|
|
|
|
# Ignores if the point whether not in the board
|
|
if not self.is_point_in_board(row, col):
|
|
return
|
|
|
|
# Verify if the position have a mine on it
|
|
if self.board[row][col] == -1:
|
|
return
|
|
|
|
# Increment the value of the position becaus is close to some mine
|
|
self.board[row][col] += 1
|
|
|
|
def reveal_adjacents(self, row, col):
|
|
min_row = row - 1
|
|
if row <= 0:
|
|
min_row = 0
|
|
|
|
min_col = col - 1
|
|
if col <= 0:
|
|
min_col = 0
|
|
|
|
max_row = row + 2
|
|
if row >= len(self.board) - 1:
|
|
max_row = len(self.board)
|
|
|
|
max_col = col + 2
|
|
if col >= len(self.board[0]) - 1:
|
|
max_col = len(self.board[0])
|
|
|
|
r = min_row
|
|
while r < max_row:
|
|
c = min_col
|
|
while c < max_col:
|
|
if not self.board[r][c] == -1 and self.board_progress[r][c] == "-":
|
|
self.board_progress[r][c] = self.board[r][c]
|
|
|
|
if self.board[r][c] == 0:
|
|
self.reveal_adjacents(r, c)
|
|
c += 1
|
|
r += 1
|
|
|
|
def reveal(self, row, col):
|
|
""" Reveals the cell's content and yours adjacents cells """
|
|
self.board_progress[row][col] = self.board[row][col]
|
|
|
|
# We will show adjacents only if the cell clicked is zero
|
|
if self.board_progress[row][col] == 0:
|
|
self.reveal_adjacents(row, col)
|
|
|
|
def win(self):
|
|
""" Identify if the player won the game """
|
|
unrevealed = 0
|
|
for row in self.board_progress:
|
|
for cell in row:
|
|
if cell == -1:
|
|
return False
|
|
|
|
if cell == "-":
|
|
unrevealed += 1
|
|
if (unrevealed - self.mines) == 0:
|
|
return True
|