-
Notifications
You must be signed in to change notification settings - Fork 0
/
gameHelper.py
95 lines (78 loc) · 2.48 KB
/
gameHelper.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/python3
import random
def placePiece(grid, identifier, length, x, y, flipped):
#Return false if another ship is in the way
if not flipped:
for i in range(x, x + length):
if grid[y][i] != 0:
return False
else:
for i in range(y, y + length):
if grid[i][x] != 0:
return False
#Place the ship on the grid
if not flipped:
for i in range(x, x + length):
grid[y][i] = identifier
else:
for i in range(y, y + length):
grid[i][x] = identifier
return True
def placeShips(grid, boardSize, pieceInfo):
#Fill the grids with ships
for piece in pieceInfo.keys():
shipLength = pieceInfo[piece]
#Keep trying to place until it succeeds
placing = True
while placing:
#Generate position
flipped = bool(random.randint(0, 1))
reducedLength = boardSize - (shipLength - 1) - 1
if flipped:
x, y = random.randint(0, boardSize - 1), random.randint(0, reducedLength)
else:
x, y = random.randint(0, reducedLength), random.randint(0, boardSize - 1)
#Attempt to place the piece
if placePiece(grid, piece, pieceInfo[piece], x, y, flipped):
placing = False
def generateBoard(size):
pieceInfo = {"c": 5, "b": 4, "d": 3, "s": 3, "p": 2}
grid = [[0 for i in range(size)] for j in range(size)]
placeShips(grid, size, pieceInfo)
return grid
def searchGrid(grid, identifier):
for row in grid:
if identifier in row:
return True
return False
#Print the passed grid
def drawGrid(grid, hideShips, pieceIdentifiers):
#Print a line of text with the given colour
def getColoured(text, colour):
return f"{colour}{text}\033[0m"
def getRowRef(rowNum):
#Prefix a zero if single digit
if rowNum < 10:
return f"0{rowNum}"
return str(rowNum)
#Get a string of alphabetical grid references (A -> upper bound)
colRef = f" {''.join([chr(i + 65) for i in range(len(grid[0]))])}"
print(colRef)
#Iterate over grid coords
for rowNum, row in enumerate(grid):
rowRef = getRowRef(rowNum + 1)
rowBuffer = f"{rowRef} "
#Show each grid point
for col in row:
#Print hit markers in red
if col == "X":
rowBuffer += getColoured(col, "\033[31m")
elif col in pieceIdentifiers and not hideShips:
rowBuffer += getColoured(col, "\033[94m")
elif col == " ":
rowBuffer += " "
else:
rowBuffer += "0"
rowBuffer += f" {rowRef}"
print(rowBuffer)
print(colRef)