import pygame   
pygame.init()
screen = pygame.display.set_mode((600, 600))
pygame.display.set_caption('五子棋-EduCoder')
space = 20
cell_size = 40  
cell_num = 15 

chess_arr = [] 

def get_one_dire_num(lx, ly, dx, dy, m):  
    tx = lx  
     
    ty = ly  
     
    s = 0  
    while True:  
        
        tx += dx  
        ty += dy  
          
        if tx < 0 or tx >= cell_num or ty < 0 or ty >= cell_num or m[ty][tx] == 0: return s  
          
        s+=1
def check_win(chess_arr, flag):  
    m = [[0] * cell_num for i in range(cell_num)] 
    for x, y, c in chess_arr:  
        if c == flag:  
            m[y][x] = 1  
    lx = chess_arr[-1][0]  
    ly = chess_arr[-1][1]  
    dire_arr = [[(-1, 0), (1, 0)], [(0, -1), (0, 1)], [(-1, -1), (1, 1)],  
                [(-1, 1), (1, -1)]]  
    for dire1, dire2 in dire_arr:  
       
        dx, dy = dire1  
        num1 = get_one_dire_num(lx, ly, dx, dy, m)  
         
        dx, dy = dire2  
        num2 = get_one_dire_num(lx, ly, dx, dy, m)
        
        if num1 + num2 + 1 >= 5: return True
    return False


flag=1
game_state = 1
while True: 
    for event in pygame.event.get(): 
        if event.type == pygame.QUIT: 
            pygame.quit()
            exit() 
        if event.type == pygame.MOUSEBUTTONUP: 
            x, y = pygame.mouse.get_pos() 
            xi = int(round((x - space) * 1.0 / cell_size))  
            xj = xi *cell_size + space 
            yi = int(round((y - space) * 1.0 / cell_size)) 
            yj = yi * cell_size + space
            print((xi,yi,flag))
            if xi >= 0 and xi < cell_num and yi >= 0 and yi < cell_num and (xj,yj,1) not in chess_arr and (xj,yj,2) not in chess_arr:
                chess_arr.append((xj, yj,flag))
                if check_win(chess_arr, flag):  
                    
                    game_state = 2 if flag == 1 else 3
                else:  
                    
                    flag = 2 if flag == 1 else 1  
    screen.fill((204, 153, 102)) 
    
    for x in range(0, cell_size * cell_num, cell_size):
        pygame.draw.line(screen, (200, 200, 200), (x + space, 0 + space),
                         (x + space, cell_size * (cell_num - 1) + space), 1)
    
    for y in range(0, cell_size * cell_num, cell_size):
        pygame.draw.line(screen, (200, 200, 200), (0 + space, y + space),
                         (cell_size * (cell_num - 1) + space, y + space), 1)
    
    for x, y,c in chess_arr:
        chess_color = (30, 30, 30) if c == 1 else (225, 225, 225)
        pygame.draw.circle(screen, chess_color, [x, y], 16, 16)
    
    if (4, 4,1) not in chess_arr and (4, 4,2) not in chess_arr:
        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 4 + space, cell_size * 4 + space), 2, 1)
    if (10, 10, 1) not in chess_arr and (10, 10, 2) not in chess_arr:
        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 10 + space, cell_size * 10 + space), 2, 1)
    if (10, 4, 1) not in chess_arr and (10, 4, 2) not in chess_arr:
        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 10 + space, cell_size * 4 + space), 2, 1)
    if (4, 10, 1) not in chess_arr and (4 ,10, 2) not in chess_arr:
        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 4 + space, cell_size * 10 + space), 2, 1)
    if (7, 7, 1) not in chess_arr and (7, 7, 2) not in chess_arr:
        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 7 + space, cell_size * 7 + space), 2, 1)
    pygame.display.update()