depth_limited_search.py

import curses, time, copy, UI_elements as UI
from checks_counts import *
from math import inf
from algo import *

def get_move(turn, n, arr, depth_limit):

    alpha = -inf
    beta  =  inf
    p = NextWithPruning(turn, n, arr, alpha, beta, 0, depth_limit)

    return p


n = int(input("Enter n for size of board : "))
player = int(input("Enter 1 if you want to play as player 1 otherwise enter 2 : "))
[sc, win, h, w ] = UI.initialise_screen(n)
flag = -1
played = 0

if(player == 2):
    played = 1

arr = [[-1]*n for i in range(n)]


# for i in range(n*n):
#     start_time = time.time()
#     p = get_move(i%2, n, arr, 3)
#     print("--- %s seconds ---" % (time.time() - start_time))
#     print(i%2, p)
#     arr[p[1]][p[2]] = i%2

start_time = time.time()
p = NextWithPruning(0, n, arr, -inf, inf, 0, 6)
print("--- %s seconds ---" % (time.time() - start_time))
print(p)

while True:

    win.border(0)
    win.timeout(100)
    curses.noecho()
    next_key = win.getch()

    if(next_key == curses.KEY_MOUSE and played == 0):
        _, mx, my, _, _ = curses.getmouse()
        arr_x = int((n)*my/h)
        arr_y = int((n)*mx/w)
        arr[arr_x][arr_y] = 0
        win.addstr(int(((2*arr_x + 1)*h)/(2*n)),int(((2*arr_y + 1)*w)/(2*n)),'X')
        played = 1

    if(next_key == 55):
        break

    c = check(arr,n)
    if(c == 0):
        break

    if(played == 1):
        p = NextWithPruning(1, n, arr, -inf, inf,0, 2*n)
        arr[p[1]][p[2]] = 1
        win.addstr(int(((2*p[1] + 1)*h)/(2*n)),int(((2*p[2] + 1)*w)/(2*n)),'0')
        played = 0
        c = check(arr,n)
        if(c == 1):
            flag = 0
            break


## Checking conditions for winning or losing or draw
if(flag==0):
    s = "You Lost!"
elif(flag==1):
    s = "You Won!"
elif(flag==-1):
    s = "Draw!"

## Outputting Result
sc.addstr(int(h/2),int(w/2),s)
sc.refresh()
time.sleep(2)
curses.endwin()