#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sat May 24 19:54:17 2025

@author: zhangzekai
"""

import pygame
from enum import Enum

# 初始化pygame
pygame.init()

# 常量定义
SCREEN_SIZE = (600, 600)
CELL_SIZE = 40
CELL_NUM = 15
SPACE = 20
STAR_POINTS = [(4, 4), (10, 10), (10, 4), (4, 10), (7, 7)]

# 颜色常量
BACKGROUND_COLOR = (204, 153, 102)
LINE_COLOR = (200, 200, 200)
BLACK_CHESS_COLOR = (30, 30, 30)
WHITE_CHESS_COLOR = (225, 225, 225)
STAR_POINT_COLOR = (0, 0, 0)
TEXT_COLOR = (210, 210, 0)

# 游戏状态枚举
class GameState(Enum):
    ONGOING = 1
    BLACK_WIN = 2
    WHITE_WIN = 3

# 初始化窗口
screen = pygame.display.set_mode(SCREEN_SIZE)
pygame.display.set_caption('五子棋-EduCoder')

# 初始化游戏状态
board = [[0 for _ in range(CELL_NUM)] for _ in range(CELL_NUM)]  # 棋盘状态 0:空 1:黑 2:白
current_player = 1  # 当前玩家 1:黑 2:白
game_state = GameState.ONGOING

def get_line_num(tx, ty, dx, dy, board, player):
    """计算指定方向连续棋子数量"""
    count = 0
    while True:
        tx += dx
        ty += dy
        if not (0 <= tx < CELL_NUM and 0 <= ty < CELL_NUM):
            break
        if board[ty][tx] != player:
            break
        count += 1
    return count

def check_win(board, x, y, player):
    """检查是否获胜"""
    directions = [
        [(-1, 0), (1, 0)],   # 水平
        [(0, -1), (0, 1)],    # 垂直
        [(-1, -1), (1, 1)],   # 主对角线
        [(-1, 1), (1, -1)]    # 副对角线
    ]
    for d in directions:
        dx1, dy1 = d[0]
        dx2, dy2 = d[1]
        num1 = get_line_num(x, y, dx1, dy1, board, player)
        num2 = get_line_num(x, y, dx2, dy2, board, player)
        if num1 + num2 + 1 >= 5:
            return True
    return False

def draw_board():
    """绘制棋盘"""
    screen.fill(BACKGROUND_COLOR)
    # 绘制网格线
    for i in range(CELL_NUM):
        pygame.draw.line(screen, LINE_COLOR,
                         (SPACE, SPACE + i * CELL_SIZE),
                         (SPACE + (CELL_NUM-1)*CELL_SIZE, SPACE + i * CELL_SIZE))
        pygame.draw.line(screen, LINE_COLOR,
                         (SPACE + i * CELL_SIZE, SPACE),
                         (SPACE + i * CELL_SIZE, SPACE + (CELL_NUM-1)*CELL_SIZE))
    # 绘制星位
    for (x, y) in STAR_POINTS:
        if board[y][x] == 0:
            center = (x * CELL_SIZE + SPACE, y * CELL_SIZE + SPACE)
            pygame.draw.circle(screen, STAR_POINT_COLOR, center, 3)
    # 绘制棋子
    for y in range(CELL_NUM):
        for x in range(CELL_NUM):
            if board[y][x] == 1:
                color = BLACK_CHESS_COLOR
            elif board[y][x] == 2:
                color = WHITE_CHESS_COLOR
            else:
                continue
            center = (x * CELL_SIZE + SPACE, y * CELL_SIZE + SPACE)
            pygame.draw.circle(screen, color, center, 16)
    # 绘制胜利信息
    if game_state != GameState.ONGOING:
        font = pygame.font.Font(None, 60)
        text = "Black Wins!" if game_state == GameState.BLACK_WIN else "White Wins!"
        text_surface = font.render(text, True, TEXT_COLOR)
        text_rect = text_surface.get_rect(center=(SCREEN_SIZE[0]//2, SCREEN_SIZE[1]//2))
        screen.blit(text_surface, text_rect)

def handle_click(pos):
    """处理鼠标点击事件"""
    global current_player, game_state
    x, y = pos
    # 转换为棋盘坐标
    grid_x = (x - SPACE) / CELL_SIZE
    grid_y = (y - SPACE) / CELL_SIZE
    # 检查是否在棋盘范围内
    if not (0 <= grid_x < CELL_NUM and 0 <= grid_y < CELL_NUM):
        return
    # 计算最近交叉点坐标
    xi = round(grid_x)
    yi = round(grid_y)
    # 检查落点有效性
    if board[yi][xi] != 0:
        return
    # 检查点击精度（距离交叉点不超过1/4格子）
    distance = ((grid_x - xi)**2 + (grid_y - yi)**2)**0.5
    if distance > 0.25:
        return
    # 落子
    board[yi][xi] = current_player
    # 检查胜利
    if check_win(board, xi, yi, current_player):
        game_state = GameState.BLACK_WIN if current_player == 1 else GameState.WHITE_WIN
    else:
        current_player = 2 if current_player == 1 else 1

# 游戏主循环
running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        elif event.type == pygame.MOUSEBUTTONUP and game_state == GameState.ONGOING:
            handle_click(pygame.mouse.get_pos())
    
    draw_board()
    pygame.display.update()

pygame.quit()
exit()