player.py

# ============================================================
# player.py - Logique du jeu (grille, coups, victoire)
# ============================================================

from settings import ROWS, COLS, EMPTY, PLAYER1, PLAYER2

def create_board():
    board = []
    for r in range(ROWS):
        row = []
        for c in range(COLS):
            row.append(EMPTY)
        board.append(row)
    return board


def is_valid_column(board, col):
    return board[0][col] == EMPTY


def get_valid_columns(board):
    valid = []
    for c in range(COLS):
        if is_valid_column(board, c):
            valid.append(c)
    return valid


def get_next_row(board, col):
    for r in range(ROWS - 1, -1, -1):
        if board[r][col] == EMPTY:
            return r
    return -1


def drop_piece(board, row, col, player):
    board[row][col] = player


def remove_piece(board, row, col):
    board[row][col] = EMPTY


def check_win(board, player):
    # Horizontal
    for r in range(ROWS):
        for c in range(COLS - 3):
            if (board[r][c]   == player and
                board[r][c+1] == player and
                board[r][c+2] == player and
                board[r][c+3] == player):
                return True
    # Vertical
    for r in range(ROWS - 3):
        for c in range(COLS):
            if (board[r][c]   == player and
                board[r+1][c] == player and
                board[r+2][c] == player and
                board[r+3][c] == player):
                return True
    # Diagonale /
    for r in range(3, ROWS):
        for c in range(COLS - 3):
            if (board[r][c]     == player and
                board[r-1][c+1] == player and
                board[r-2][c+2] == player and
                board[r-3][c+3] == player):
                return True
    # Diagonale \
    for r in range(ROWS - 3):
        for c in range(COLS - 3):
            if (board[r][c]     == player and
                board[r+1][c+1] == player and
                board[r+2][c+2] == player and
                board[r+3][c+3] == player):
                return True
    return False


def get_winning_cells(board, player):
    # Horizontal
    for r in range(ROWS):
        for c in range(COLS - 3):
            cells = [(r,c),(r,c+1),(r,c+2),(r,c+3)]
            if all(board[rr][cc] == player for rr, cc in cells):
                return cells
    # Vertical
    for r in range(ROWS - 3):
        for c in range(COLS):
            cells = [(r,c),(r+1,c),(r+2,c),(r+3,c)]
            if all(board[rr][cc] == player for rr, cc in cells):
                return cells
    # Diagonale /
    for r in range(3, ROWS):
        for c in range(COLS - 3):
            cells = [(r,c),(r-1,c+1),(r-2,c+2),(r-3,c+3)]
            if all(board[rr][cc] == player for rr, cc in cells):
                return cells
    # Diagonale \
    for r in range(ROWS - 3):
        for c in range(COLS - 3):
            cells = [(r,c),(r+1,c+1),(r+2,c+2),(r+3,c+3)]
            if all(board[rr][cc] == player for rr, cc in cells):
                return cells
    return []


def is_board_full(board):
    return len(get_valid_columns(board)) == 0


def is_terminal(board):
    return (check_win(board, PLAYER1) or
            check_win(board, PLAYER2) or
            is_board_full(board))