import pygame

import sys

import time

import random

pygame.init()

screen_size = (800,560)

WIDTH = 720

HEIGHT = 720

GRID_WIDTH = WIDTH // 20

WHITE = (255, 250, 255)

BLACK = (0, 0, 0)

GREEN = (0, 0xff, 0)

RED = (0xff, 0, 0)

color = [BLACK,WHITE]

class checker:

    def __init__(self,x,y):

        self.ox = x

        self.oy = y

        self.color = (0,0,0)

        self.state = 0   # 0 1 2

checks = []

colorful = []

aiful = []

for i in range(1,16):

    for j in range(1,16):

        achecker = checker(i,j)

        checks.append(achecker)

        colorful.append(0)

        aiful.append(0)

for i in range(1,16):

    colorful.append(0)

    aiful.append(0)

movements=[]

movements1 = []

movements2 = []

screen = pygame.display.set_mode(screen_size,0, 32)

pygame.display.set_caption('欢乐五子棋')

clock = pygame.time.Clock()

bg = pygame.image.load("image\\checker.jpg").convert()

weixiao = pygame.image.load("image\\weixiao.jpg").convert()

huaji =  pygame.image.load("image\\huaji.jpg").convert()

arrow = pygame.image.load("image\\arrow.jpg").convert()

bgcolor = (255,255,255)

turn = 0

def draw_check(surf):

    screen.fill(bgcolor)

    surf.blit(bg, (0, 0))

def draw_column(surf,where):

    surf.blit(huaji, (650, 100))

    surf.blit(weixiao, (650, 300))

    if where == 1:

        surf.blit(arrow, (580, 100))

    else:

        surf.blit(arrow, (580, 300))

def draw_checker(surf):

    for c in checks:

        if c.state == 1:

            pygame.draw.circle(surf, c.color, (c.ox * 35, c.oy * 35), 10, 0)

        if c.state == 2:

            pygame.draw.circle(surf, c.color, (c.ox * 35, c.oy * 35), 10, 0)

def tmax(a):

    max = a[0]

    for i in a:

        if max < i:

            max = i

    return max

def Ai_choose():

    for i in aiful:

        i = 0

    if len(movements) == 0:

        return (7,7)

    for i in movements1:

        aiful[i[0]+i[1]*15] = 0

        if i[0] > 0:

            aiful[i[0]-1+i[1]*15] +=1

        if i[0] < 15:

            aiful[i[0]+1 + i[1] * 15] += 1

        if i[1] > 0:

            aiful[i[0]+ (i[1]-1) * 15] += 1

        if i[1] < 15:

            aiful[i[0] + (i[1] + 1) * 15] += 1

    c = tmax(aiful)

    print(c)

def check_win(x,y):

    i = x

    j = y

    count = 0

    ck = colorful[x+y*15]

    if ck == 0:

        return False

    #横着

    while i>=0 and colorful[i+y*15] == ck:

        i -= 1

        count += 1

    i = x+1

    while i< 15 and colorful[i+y*15] == ck:

        i += 1

        count += 1

    if count >= 5:

        print("win")

        return True

    #竖着

    i = x

    count = 0

    while j>=0 and colorful[i+15*j] == ck:

        j-=1

        count +=1

    j = y+1

    while j < 15 and colorful[i+15*j] == ck:

        j  += 1

        count +=1

    if count >= 5:

        print("win")

        return True

    #斜着左上

    j = y

    count = 0

    while i >= 0 and j >=0 and colorful[i +15* j] == ck:

        j -= 1

        i -= 1

        count += 1

    i = x+1

    j = y+1

    while i< 15 and j <15 and colorful[i+15*j] == ck:

        i += 1

        j +=1

        count += 1

    if count >= 5:

        print("win")

        return True

    #斜着右上

    j = y

    i = x

    count = 0

    while i >= 0 and j <15 and colorful[i + 15 * j] == ck:

        i -= 1

        j += 1

        count += 1

    i = x + 1

    j = y - 1

    while i< 15 and j >0 and colorful[i+15*j] == ck:

        i += 1

        j -=1

        count += 1

    if count >= 5:

        print("win")

        return True

    return False

runing = 1

Ai_wait = 0

while runing:

    for event in pygame.event.get():

        if event.type == pygame.QUIT:

            sys.exit()

        if event.type == pygame.MOUSEBUTTONDOWN:

            if turn == 1:

                pos = event.pos

                grid = (int(round(pos[0] / 35)), int(round(pos[1] / 35)))

                for c in checks:

                    if c.state == 0:

                        if c.ox == grid[0] and c.oy == grid[1]:

                            movements.append(grid)

                            movements2.append(grid)

                            colorful[(c.ox) + (c.oy) * 15] = 2

                            c.color = WHITE

                            c.state = 2

                            turn = 0

                            if True == check_win(c.ox,c.oy):

                                runing = 0

                                print("bingo")

    if turn == 0:

        if Ai_wait == 30:

            Ai_wait= 0

            #pos = Ai_choose()

            grid = pos

            for i in movements:

                while i == pos:

                    pos = (random.randint(1, 19), random.randint(1, 19))

                    grid = pos

            for c in checks:

                if c.state == 0:

                    if c.ox == grid[0] and c.oy == grid[1]:

                        movements.append(grid)

                        movements1.append(grid)

                        colorful[(c.ox) + (c.oy) * 15] = 1

                        c.color = BLACK

                        c.state = 1

                        turn = 1

                        if True == check_win(c.ox, c.oy):

                            runing = 0

                            print("bingo")

        else:

            Ai_wait+=1

    draw_check(screen)

    draw_column(screen,turn)

    draw_checker(screen)

    pygame.display.update()

    clock.tick(50)

while True:

    for event in pygame.event.get():

        if event.type == pygame.QUIT:

            sys.exit()