156 lines
5.0 KiB
Python
156 lines
5.0 KiB
Python
import pygame
|
|
import sys
|
|
from constants import *
|
|
from player import *
|
|
import copy
|
|
|
|
def print_board(board):
|
|
for i in range(BOARD_SIZE):
|
|
for j in range(BOARD_SIZE):
|
|
print(f"{board[i][j]} ", end="")
|
|
print()
|
|
print()
|
|
|
|
|
|
def main():
|
|
print("Starting BattleShip!")
|
|
|
|
pygame.init()
|
|
|
|
if pygame.get_init() == False:
|
|
pygame.quit()
|
|
sys.exit(1)
|
|
|
|
print("Please input your list of positions, from smallest ship to the largest, for your ships.")
|
|
print("Format: a[A]1-a[A]5 b[B]2-b[B]3. [A] means it can either be upper or lower case. Please input the positions from lowest to highest.")
|
|
print("You must have 5 ships. The ships can be of any size between 1 and 5.")
|
|
print("Have fun.")
|
|
|
|
good_inp = False
|
|
player_board = []
|
|
enemy_board = []
|
|
|
|
while good_inp == False:
|
|
inp = input()
|
|
good_inp, player_board, ship_pos = check_valid_input(inp)
|
|
enemy_board = copy.deepcopy(player_board)
|
|
|
|
player = Player(player_board, ship_pos.copy(), "Player")
|
|
enemy = Player(enemy_board, ship_pos.copy(), "Enemy")
|
|
|
|
#print_board(player_board)
|
|
|
|
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
|
|
|
|
turn_counter = 0
|
|
first_iteration = True
|
|
is_enemy_turn = False
|
|
|
|
while True:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
print("Quitting Battleships")
|
|
return
|
|
|
|
pygame.Surface.fill(screen, (255,255,255))
|
|
|
|
if first_iteration == False:
|
|
if is_enemy_turn == False:
|
|
player.update(enemy)
|
|
is_enemy_turn = True
|
|
else:
|
|
enemy.random_shoot(player)
|
|
is_enemy_turn = False
|
|
else:
|
|
first_iteration = False
|
|
|
|
if check_win(player, enemy) == True:
|
|
print(f"{player.name} won!")
|
|
sys.exit(0)
|
|
|
|
if check_win(enemy, player) == True:
|
|
print(f"{player.name} won!")
|
|
sys.exit(0)
|
|
|
|
player.draw(screen, enemy)
|
|
|
|
pygame.display.flip() #refresh screen
|
|
|
|
turn_counter += 1
|
|
print("turn: ", turn_counter)
|
|
|
|
|
|
def check_win(player, enemy):
|
|
print(f"checking win for: {player.name}")
|
|
for i in range(0, BOARD_SIZE):
|
|
for j in range(0, BOARD_SIZE):
|
|
if enemy.board[i][j] == 0b01:
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def check_valid_input(inp):
|
|
ship_pos = []
|
|
s = inp.split(' ')
|
|
#print("s", s)
|
|
board = []
|
|
for i in range(BOARD_SIZE):
|
|
board.append([])
|
|
for _ in range(BOARD_SIZE):
|
|
board[i].append(0)
|
|
|
|
if len(s) != 5:
|
|
print("Not valid input. Please input again")
|
|
return (False, board, ship_pos)
|
|
else:
|
|
count = 0
|
|
for position in s:
|
|
count += 1
|
|
#print("c", count)
|
|
position = position.lower()
|
|
t = position.split("-")
|
|
#print("t", t)
|
|
ship_pos.append((t[0], t[1]))
|
|
|
|
if t[0][0] == t[1][0] and ord(t[0][0]) >= 97 and ord(t[0][0]) <= 106 and ord(t[1][0]) >= 97 and ord(t[1][0]) <= 106: #check if the fist chars match
|
|
if int(t[0][1]) >= 0 and int(t[0][1]) <= 9 and int(t[1][1]) >= 0 and int(t[1][1]) <= 9 and int(t[0][1]) != int(t[1][1]) and abs(int(t[0][1]) - int(t[1][1]))+1 < 6: #check if number is in correct range
|
|
if int(t[0][1]) > int(t[1][1]):
|
|
step = -1
|
|
else:
|
|
step = 1
|
|
|
|
for column in range(int(t[0][1]), int(t[1][1])+1, step):
|
|
if board[ord(t[0][0])-97][column] == 0b01:
|
|
print("this position is already taken!")
|
|
return (False, board, ship_pos)
|
|
|
|
board[ord(t[0][0])-97][column] = 0b01 #first means if it is hit, second means placement
|
|
#print_board(board)
|
|
else:
|
|
#print("e1")
|
|
return (False, board, ship_pos)
|
|
elif t[0][1] == t[1][1] and int(t[0][1]) <= 9 and int(t[1][1]) >= 0 and int(t[1][1]) <= 9 and abs(int(t[0][1]) - int(t[1][1]))+1 < 6:
|
|
if ord(t[0][0]) >= 97 and ord(t[0][0]) <= 106 and ord(t[1][0]) >= 97 and ord(t[1][0]) <= 106 and t[0][0] != t[1][0]:
|
|
if ord(t[0][0]) > ord(t[1][0]):
|
|
step = -1
|
|
else:
|
|
step = 1
|
|
|
|
for row in range(ord(t[0][0])-97, ord(t[1][0])-97+1, step):
|
|
if board[row][int(t[0][1])] == 0b01:
|
|
print("this position is already taken!")
|
|
return (False, board, ship_pos)
|
|
|
|
board[row][int(t[0][1])] = 0b01 #first means if it is hit, second means placement
|
|
#print_board(board)
|
|
else:
|
|
#print("e2")
|
|
return (False, board, ship_pos)
|
|
|
|
#print("e3")
|
|
return (True, board, ship_pos)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|