mirror of
https://github.com/coding-horror/basic-computer-games.git
synced 2025-12-22 07:10:42 -08:00
168 lines
4.5 KiB
Python
168 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
|
from random import randrange
|
|
from typing import List, Tuple
|
|
|
|
PointType = Tuple[int, int]
|
|
VectorType = PointType
|
|
SeaType = Tuple[List[int], ...]
|
|
|
|
SEA_WIDTH = 6
|
|
DESTROYER_LENGTH = 2
|
|
CRUISER_LENGTH = 3
|
|
AIRCRAFT_CARRIER_LENGTH = 4
|
|
|
|
|
|
def random_vector() -> Tuple[int, int]:
|
|
while True:
|
|
vector = (randrange(-1, 2), randrange(-1, 2))
|
|
|
|
if vector == (0, 0):
|
|
# We can't have a zero vector, so try again
|
|
continue
|
|
|
|
return vector
|
|
|
|
|
|
def add_vector(point: PointType, vector: VectorType) -> PointType:
|
|
return (point[0] + vector[0], point[1] + vector[1])
|
|
|
|
|
|
def place_ship(sea: SeaType, size: int, code: int) -> None:
|
|
while True:
|
|
start = (randrange(1, SEA_WIDTH + 1), randrange(1, SEA_WIDTH + 1))
|
|
vector = random_vector()
|
|
|
|
# Get potential ship points
|
|
point = start
|
|
points = []
|
|
|
|
for _ in range(size):
|
|
point = add_vector(point, vector)
|
|
points.append(point)
|
|
|
|
if (not all([is_within_sea(point, sea) for point in points]) or
|
|
any([value_at(point, sea) for point in points])):
|
|
# ship out of bounds or crosses other ship, trying again
|
|
continue
|
|
|
|
# We found a valid spot, so actually place it now
|
|
for point in points:
|
|
set_value_at(code, point, sea)
|
|
|
|
break
|
|
|
|
|
|
def print_encoded_sea(sea: SeaType) -> None:
|
|
for x in range(len(sea)):
|
|
print(' '.join([str(sea[y][x]) for y in range(len(sea) - 1, -1, -1)]))
|
|
|
|
|
|
def is_within_sea(point: PointType, sea: SeaType) -> bool:
|
|
return (1 <= point[0] <= len(sea)) and (1 <= point[1] <= len(sea))
|
|
|
|
|
|
def has_ship(sea: SeaType, code: int) -> bool:
|
|
return any(code in row for row in sea)
|
|
|
|
|
|
def count_sunk(sea: SeaType, *codes: int) -> int:
|
|
return sum(not has_ship(sea, code) for code in codes)
|
|
|
|
|
|
def value_at(point: PointType, sea: SeaType) -> int:
|
|
return sea[point[1] - 1][point[0] -1]
|
|
|
|
|
|
def set_value_at(value: int, point: PointType, sea: SeaType) -> None:
|
|
sea[point[1] - 1][point[0] -1] = value
|
|
|
|
|
|
def get_next_target(sea: SeaType) -> PointType:
|
|
while True:
|
|
try:
|
|
guess = input('? ')
|
|
point = guess.split(',')
|
|
|
|
if len(point) != 2:
|
|
raise ValueError()
|
|
|
|
point = (int(point[0]), int(point[1]))
|
|
|
|
if not is_within_sea(point, sea):
|
|
raise ValueError()
|
|
|
|
return point
|
|
except ValueError:
|
|
print(f'INVALID. SPECIFY TWO NUMBERS FROM 1 TO {len(sea)}, SEPARATED BY A COMMA.')
|
|
|
|
|
|
def setup_ships(sea: SeaType):
|
|
place_ship(sea, DESTROYER_LENGTH, 1)
|
|
place_ship(sea, DESTROYER_LENGTH, 2)
|
|
place_ship(sea, CRUISER_LENGTH, 3)
|
|
place_ship(sea, CRUISER_LENGTH, 4)
|
|
place_ship(sea, AIRCRAFT_CARRIER_LENGTH, 5)
|
|
place_ship(sea, AIRCRAFT_CARRIER_LENGTH, 6)
|
|
|
|
|
|
def main() -> None:
|
|
sea = tuple(([0 for _ in range(SEA_WIDTH)] for _ in range(SEA_WIDTH)))
|
|
setup_ships(sea)
|
|
print(f'''
|
|
BATTLE
|
|
CREATIVE COMPUTING MORRISTOWN, NEW JERSEY
|
|
|
|
THE FOLLOWING CODE OF THE BAD GUYS' FLEET DISPOSITION
|
|
HAS BEEN CAPTURED BUT NOT DECODED:
|
|
|
|
''')
|
|
print_encoded_sea(sea)
|
|
print('''
|
|
|
|
DE-CODE IT AND USE IT IF YOU CAN
|
|
BUT KEEP THE DE-CODING METHOD A SECRET.
|
|
|
|
START GAME''')
|
|
splashes = 0
|
|
hits = 0
|
|
|
|
while True:
|
|
target = get_next_target(sea)
|
|
target_value = value_at(target, sea)
|
|
|
|
if target_value < 0:
|
|
print(f'YOU ALREADY PUT A HOLE IN SHIP NUMBER {abs(target_value)} AT THAT POINT.')
|
|
|
|
if target_value <= 0:
|
|
print('SPLASH! TRY AGAIN.')
|
|
splashes += 1
|
|
continue
|
|
|
|
print(f'A DIRECT HIT ON SHIP NUMBER {target_value}')
|
|
hits += 1
|
|
set_value_at(-target_value, target, sea)
|
|
|
|
if not has_ship(sea, target_value):
|
|
print('AND YOU SUNK IT. HURRAH FOR THE GOOD GUYS.')
|
|
print('SO FAR, THE BAD GUYS HAVE LOST')
|
|
print(f'{count_sunk(sea, 1, 2)} DESTROYER(S),',
|
|
f'{count_sunk(sea, 3, 4)} CRUISER(S),',
|
|
f'AND {count_sunk(sea, 5, 6)} AIRCRAFT CARRIER(S).')
|
|
|
|
if any(has_ship(sea, code) for code in range(1, 7)):
|
|
print(f'YOUR CURRENT SPLASH/HIT RATIO IS {splashes}/{hits}')
|
|
continue
|
|
|
|
print('YOU HAVE TOTALLY WIPED OUT THE BAD GUYS\' FLEET '
|
|
f'WITH A FINAL SPLASH/HIT RATIO OF {splashes}/{hits}')
|
|
|
|
if not splashes:
|
|
print('CONGRATULATIONS -- A DIRECT HIT EVERY TIME.')
|
|
|
|
print("\n****************************")
|
|
break
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|