import pygame
# 导入pygame模块
print(pygame.ver)
# 检查pygame的版本，检查pygame有没有导入成功

EMPTY = 0
BLACK = 1
WHITE = 2
# 定义三个常量函数，用来表示白棋，黑棋，以及 空

black_color = [0, 0, 0]
# 定义黑色（黑棋用，画棋盘）
white_color = [255, 255, 255]
# 定义白色（白棋用）

# 定义棋盘这个类
class RenjuBoard(object):

 def __init__(self):
 # self._board = board = [[EMPTY] * 15 for _ in range(15)]
 # 将棋盘每一个交叉点都看作列表的一个元素位，一共有15*15共225个元素
 self._board = [[]] * 15
 self.reset()
 #重置棋盘
 def reset(self):
 for row in range(len(self._board)):
  self._board[row] = [EMPTY] * 15
 #定义棋盘上的下棋函数，row表示行，col表示列，is_black表示判断当前点位该下黑棋，还是白棋
 def move(self, row, col, is_black):
 if self._board[row][col] == EMPTY:
  self._board[row][col] = BLACK if is_black else WHITE
  return True
 return False
 # 给棋盘定义一个函数将自己在screen上面画出来，使用pygame.draw()函数。并且顺便将下了的棋子也画出来
 def draw(self, screen):
 for h in range(1, 16):
  pygame.draw.line(screen, black_color,
    [40, h * 40], [600, h * 40], 1)
  pygame.draw.line(screen, black_color,
 # 给棋盘加一个外框，使美观
 pygame.draw.rect(screen, black_color, [36, 36, 568, 568], 3)
 # 在棋盘上标出，天元以及另外4个特殊点位
 pygame.draw.circle(screen, black_color, [320, 320], 5, 0)
 pygame.draw.circle(screen, black_color, [160, 160], 3, 0)
 pygame.draw.circle(screen, black_color, [160, 480], 3, 0)
 pygame.draw.circle(screen, black_color, [480, 160], 3, 0)
 pygame.draw.circle(screen, black_color, [480, 480], 3, 0)
 #做2次for循环取得棋盘上所有交叉点的坐标
 for row in range(len(self._board)):
  for col in range(len(self._board[row])):
  # 将下在棋盘上的棋子画出来
  if self._board[row][col] != EMPTY:
   ccolor = black_color \
   if self._board[row][col] == BLACK else white_color
   # 取得这个交叉点下的棋子的颜色，并将棋子画出来
   pos = [40 * (col + 1), 40 * (row + 1)]
   # 画出棋子
   pygame.draw.circle(screen, ccolor, pos, 18, 0)

# 定义函数，传入当前棋盘上的棋子列表，输出结果，不管黑棋白棋胜，都是传回False，未出结果则为True
def is_win(board):
 for n in range(15):
 # 判断垂直方向胜利
 flag = 0
 # flag是一个标签，表示是否有连续以上五个相同颜色的棋子
 for b in board._board:
  if b[n] == 1:
  flag += 1
  if flag == 5:
   print('黑棋胜')
   return False
  else:
  # else表示此时没有连续相同的棋子，标签flag重置为0
  flag = 0

 flag = 0
 for b in board._board:
  if b[n] == 2:
  flag += 1
  if flag == 5:
   print('白棋胜')
   return False
  else:
  flag = 0

 # 判断水平方向胜利
 flag = 0
 for b in board._board[n]:
  if b == 1:
  flag += 1
  if flag == 5:
   print('黑棋胜')
   return False
  else:
  flag = 0

 flag = 0
 for b in board._board[n]:
  if b == 2:
  flag += 1
  if flag == 5:
   print('白棋胜')
   return False
  else:
  flag = 0

 # 判断正斜方向胜利

 for x in range(4, 25):
  flag = 0
  for i,b in enumerate(board._board):
  if 14 >= x - i >= 0 and b[x - i] == 1:
   flag += 1
   if flag == 5:
   print('黑棋胜')
   return False
  else:
   flag = 0

 for x in range(4, 25):
  flag = 0
  for i,b in enumerate(board._board):
  if 14 >= x - i >= 0 and b[x - i] == 2:
   flag += 1
   if flag == 5:
   print('白棋胜')
   return False
  else:
   flag = 0

 #判断反斜方向胜利
 for x in range(11, -11, -1):
  flag = 0
  for i,b in enumerate(board._board):
  if 0 <= x + i <= 14 and b[x + i] == 1:
   flag += 1
   if flag == 5:
   print('黑棋胜')
   return False
  else:
   flag = 0

 for x in range(11, -11, -1):
  flag = 0
  for i,b in enumerate(board._board):
  if 0 <= x + i <= 14 and b[x + i] == 2:
   flag += 1
   if flag == 5:
   print('白棋胜')
   return False
  else:
   flag = 0

 return True


def main():
 # 创建棋盘对象
 board = RenjuBoard()
 # 用于判断是下黑棋还是白棋
 is_black = True
 # pygame初始化函数，固定写法
 pygame.init()
 pygame.display.set_caption('五子棋') # 改标题
 # pygame.display.set_mode()表示建立个窗口，左上角为坐标原点，往右为x正向，往下为y轴正向
 screen = pygame.display.set_mode((640,640))
 # 给窗口填充颜色，颜色用三原色数字列表表示
 screen.fill([125,95,24])
 board.draw(screen) # 给棋盘类发命令，调用draw()函数将棋盘画出来
 pygame.display.flip() # 刷新窗口显示

 running = True
 # while 主循环的标签，以便跳出循环
 while running:
 # 遍历建立窗口后发生的所有事件，固定写法
 for event in pygame.event.get():
  # 根据事件的类型，进行判断
  if event.type == pygame.QUIT:
  running = False

  elif event.type == pygame.KEYUP:
  pass
  # pygame.MOUSEBUTTONDOWN表示鼠标的键被按下
  elif event.type == pygame.MOUSEBUTTONDOWN and \
   event.button == 1:# button表示鼠标左键
  x, y = event.pos # 拿到鼠标当前在窗口上的位置坐标
  # 将鼠标的(x, y)窗口坐标，转化换为棋盘上的坐标
  row = round((y - 40) / 40) 
  col = round((x - 40) / 40)
  if board.move(row, col, is_black):
   is_black = not is_black
   screen.fill([125, 95, 24])
   board.draw(screen)
   pygame.display.flip()
   # 调用判断胜负函数
   if not is_win(board):
   #break
   running = False
   # 这里我有个bug没找到解决办法，就是判断出胜负后，使用break跳出事件遍历的for循环，但是老是不能跳出来，导致胜负分出来了还可以继续下，这里我采用判断胜负后就将running标签赋值为False，跳出主循环，但是这样棋盘的窗口也没了。明天再找找bug在哪

 pygame.quit()

if __name__ == '__main__':
 main()