Connect4GUI.py

import numpy as np
import pygame
import sys
import math

BLUE = (22,111,190)
BLACK = (0,0,0)
RED = (255,0,0)
YELLOW = (255,255,0)
WHITE = (255,255,255)

ROW_COUNT = 6
COL_COUNT = 7
def create_board():
    board = np.zeros((ROW_COUNT,COL_COUNT))
    return board

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

def is_valid_location(board,col):
    return board[ROW_COUNT-1][col]== 0

def get_next_open_row(board,col):
    for r in range(ROW_COUNT):
        if board[r][col] == 0:
            return r

def print_board(board):
    print(np.flip(board, 0))

def winning_move(board, piece):
    #HorizontalCheck
    for c in range(COL_COUNT-3):
        for r in range(ROW_COUNT):
            if board[r][c] == piece and board[r][c+1] ==piece and board[r][c+2]==piece and board[r][c+3]==piece:
                return True
    #VerticalCheck
    for c in range(COL_COUNT):
        for r in range(ROW_COUNT-3):
            if board[r][c] == piece and board[r+1][c] ==piece and  board[r+2][c]==piece and board[r+3][c]==piece:
                return True
    #CheckPositiveDiagonal/
    for c in range(COL_COUNT-3):
        for r in range(ROW_COUNT-3):
            if board[r][c] == piece and board[r+1][c+1] ==piece and board[r+2][c+2]==piece and board[r+3][c+3]==piece:
                return True
    #CheckNegativeDiagonal\
    for c in range(COL_COUNT-3):
        for r in range(3,ROW_COUNT):
            if board[r][c] == piece and board[r-1][c+1] ==piece and board[r-2][c+2]==piece and board[r-3][c+3]==piece:
                return True

def draw_board(board):
    for c in range(COL_COUNT):
        for r in range(ROW_COUNT):
            pygame.draw.rect(screen, BLUE , (c*SQUARESIZE, r*SQUARESIZE+SQUARESIZE, SQUARESIZE, SQUARESIZE))
            pygame.draw.circle(screen, BLACK,(int(c*SQUARESIZE+SQUARESIZE/2), int(r*SQUARESIZE+SQUARESIZE+SQUARESIZE/2)),RADIUS)
    
    for c in range(COL_COUNT):
        for r in range(ROW_COUNT):
            if board[r][c]==1:
                pygame.draw.circle(screen, RED,(int(c*SQUARESIZE+SQUARESIZE/2), height - int(r*SQUARESIZE+SQUARESIZE/2)),RADIUS)
            elif board[r][c]==2:
                pygame.draw.circle(screen, YELLOW,(int(c*SQUARESIZE+SQUARESIZE/2), height - int(r*SQUARESIZE+SQUARESIZE/2)),RADIUS)
    pygame.display.update()

board = create_board()
print_board(board)
game_over = False
turn = 0

pygame.init()

SQUARESIZE = 100 #pixels
RADIUS = (SQUARESIZE/2 - 10)
width = COL_COUNT*SQUARESIZE
height = (ROW_COUNT*SQUARESIZE)+1
size = (width,height)

#SetScreen
screen = pygame.display.set_mode(size)
draw_board(board)
pygame.display.update()

myfont = pygame.font.SysFont("Consolas", 75)


while not game_over:

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            sys.exit()
        
        if event.type == pygame.MOUSEMOTION:
            pygame.draw.rect(screen, BLACK,(0,0,width,SQUARESIZE))
            posx = event.pos[0]
            if turn ==0:
                pygame.draw.circle(screen,RED,(posx,int(SQUARESIZE/2)),RADIUS)
            else:
                pygame.draw.circle(screen,YELLOW,(posx,int(SQUARESIZE/2)),RADIUS)
        pygame.display.update()
        
        if event.type == pygame.MOUSEBUTTONDOWN:
            print(" ")
            # Asking Player 1 for Input
            if turn == 0:
                posx= event.pos[0]
                col = int(math.floor(posx/SQUARESIZE))
                # col = int(input("Player 1 Make your Selection (0-6)"))

                if is_valid_location(board,col):
                    row = get_next_open_row(board,col)
                    drop_piece(board,row,col,1)
                
                if winning_move(board,1):
                    print("\n \t PLAYER 1 WINS!!!\t \n")
                    label = myfont.render("PLAYER 1 WINS!!", 1, WHITE)
                    screen.blit(label, (40,10))
                    game_over = True
            # Asking Player2 for Input
            else:
                posx= event.pos[0]
                col = int(math.floor(posx/SQUARESIZE))
                # col = int(input("Player 2 Make  your Selection (0-6)"))
                
                if is_valid_location(board,col):
                    row = get_next_open_row(board,col)
                    drop_piece(board,row,col,2)
                
                if winning_move(board,2):
                    print("\n \t PLAYER 2 WINS!!!\t \n")
                    label = myfont.render("PLAYER 2 WINS!!", 1, WHITE)
                    screen.blit(label, (40,10))
                    game_over = True
        
            print_board(board)
            draw_board(board)
            turn +=1
            turn %=2
            
        if game_over:
            pygame.time.wait(3000)