mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2024-12-27 22:01:37 +03:00
51 lines
2.2 KiB
Python
51 lines
2.2 KiB
Python
'''
|
|
This file handles the game logic for the Gomoku game.
|
|
'''
|
|
import pygame
|
|
class Board:
|
|
def __init__(self, rows, cols, player_1_icon, player_2_icon, width, height):
|
|
self.rows = rows
|
|
self.cols = cols
|
|
self.width = width
|
|
self.height = height
|
|
self.board = [[None for _ in range(cols)] for _ in range(rows)]
|
|
self.current_player = 1
|
|
self.player_1_icon =player_1_icon
|
|
self.player_2_icon =player_2_icon
|
|
self.game_state = 'ongoing'
|
|
self.winner = None
|
|
def place_stone(self, x, y):
|
|
if self.game_state == 'ended':
|
|
return
|
|
row, col = self.get_board_position(x, y)
|
|
if row >= 0 and row < self.rows and col >= 0 and col < self.cols:
|
|
if self.board[row][col] is None:
|
|
self.board[row][col] = self.current_player
|
|
if self.check_win(row, col):
|
|
self.game_state = 'ended'
|
|
self.winner = self.current_player
|
|
self.current_player = 1 if self.current_player == 2 else 2
|
|
def get_board_position(self, x, y):
|
|
row = y // (self.height // self.rows)
|
|
col = x // (self.width // self.cols)
|
|
return row, col
|
|
def check_win(self, row, col):
|
|
# Check horizontal, vertical and diagonal lines for a win
|
|
directions = [(0, 1), (1, 0), (1, 1), (1, -1)]
|
|
for dx, dy in directions:
|
|
if self.count_stones(row, col, dx, dy) + self.count_stones(row, col, -dx, -dy) - 1 >= 5:
|
|
return True
|
|
return False
|
|
def count_stones(self, row, col, dx, dy):
|
|
count = 0
|
|
while 0 <= row < self.rows and 0 <= col < self.cols and self.board[row][col] == self.current_player:
|
|
count += 1
|
|
row += dx
|
|
col += dy
|
|
return count
|
|
def draw(self, window):
|
|
for row in range(self.rows):
|
|
for col in range(self.cols):
|
|
if self.board[row][col] is not None:
|
|
icon = self.player_1_icon if self.board[row][col] == 1 else self.player_2_icon
|
|
window.blit(icon, (col * (self.width // self.cols), row * (self.height // self.rows))) |