# ----------------------------------------------------------------------
# 定義棋子類型，輸贏情況
# ----------------------------------------------------------------------
EMPTY = 0
BLACK = 1
WHITE = 2


# ----------------------------------------------------------------------
# 定義棋盤類，繪制棋盤的形狀，切換先后手，判斷輸贏等
# ----------------------------------------------------------------------
class ChessBoard(object):
 def __init__(self):
 self.__board = [[EMPTY for n in range(15)] for m in range(15)]
 self.__dir = [[(-1, 0), (1, 0)], [(0, -1), (0, 1)], [(-1, 1), (1, -1)], [(-1, -1), (1, 1)]]
 #  (左 右) (上 下) (左下 右上) (左上 右下)

 def board(self): # 返回?cái)?shù)組對(duì)象
 return self.__board

 def draw_xy(self, x, y, state): # 獲取落子點(diǎn)坐標(biāo)的狀態(tài)
 self.__board[x][y] = state

 def get_xy_on_logic_state(self, x, y): # 獲取指定點(diǎn)坐標(biāo)的狀態(tài)
 return self.__board[x][y]

 def get_next_xy(self, point, direction): # 獲取指定點(diǎn)的指定方向的坐標(biāo)
 x = point[0] + direction[0]
 y = point[1] + direction[1]
 if x < 0 or x >= 15 or y < 0 or y >= 15:
  return False
 else:
  return x, y

 def get_xy_on_direction_state(self, point, direction): # 獲取指定點(diǎn)的指定方向的狀態(tài)
 if point is not False:
  xy = self.get_next_xy(point, direction)
  if xy is not False:
  x, y = xy
  return self.__board[x][y]
 return False

 def anyone_win(self, x, y):
 state = self.get_xy_on_logic_state(x, y) # 當(dāng)前落下的棋是黑棋還是白棋，它的狀態(tài)存儲(chǔ)在state中
 for directions in self.__dir: # 對(duì)米字的4個(gè)方向分別檢測(cè)是否有5子相連的棋
  count = 1 # 初始記錄為1，因?yàn)閯偮湎碌钠逡菜?
  for direction in directions: # 對(duì)落下的棋子的同一條線的兩側(cè)都要檢測(cè)，結(jié)果累積
  point = (x, y) # 每次循環(huán)前都要刷新
  while True:
   if self.get_xy_on_direction_state(point, direction) == state:
   count += 1
   point = self.get_next_xy(point, direction)
   else:
   break
  if count >= 5:
  return state
 return EMPTY

 def reset(self): # 重置
 self.__board = [[EMPTY for n in range(15)] for m in range(15)]

from chessboard import ChessBoard
from ai import searcher

WIDTH = 540
HEIGHT = 540
MARGIN = 22
GRID = (WIDTH - 2 * MARGIN) / (15 - 1)
PIECE = 34
EMPTY = 0
BLACK = 1
WHITE = 2


import sys
from PyQt5 import QtCore, QtGui
from PyQt5.QtWidgets import QApplication, QWidget, QLabel, QMessageBox
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QPixmap, QIcon, QPalette, QPainter
from PyQt5.QtMultimedia import QSound


# ----------------------------------------------------------------------
# 定義線程類執(zhí)行AI的算法
# ----------------------------------------------------------------------
class AI(QtCore.QThread):
 finishSignal = QtCore.pyqtSignal(int, int)

 # 構(gòu)造函數(shù)里增加形參
 def __init__(self, board, parent=None):
 super(AI, self).__init__(parent)
 self.board = board

 # 重寫 run() 函數(shù)
 def run(self):
 self.ai = searcher()
 self.ai.board = self.board
 score, x, y = self.ai.search(2, 2)
 self.finishSignal.emit(x, y)


# ----------------------------------------------------------------------
# 重新定義Label類
# ----------------------------------------------------------------------
class LaBel(QLabel):
 def __init__(self, parent):
 super().__init__(parent)
 self.setMouseTracking(True)

 def enterEvent(self, e):
 e.ignore()


class GoBang(QWidget):
 def __init__(self):
 super().__init__()
 self.initUI()

 def initUI(self):

 self.chessboard = ChessBoard() # 棋盤類

 palette1 = QPalette() # 設(shè)置棋盤背景
 palette1.setBrush(self.backgroundRole(), QtGui.QBrush(QtGui.QPixmap('img/chessboard.jpg')))
 self.setPalette(palette1)
 # self.setStyleSheet("board-image:url(img/chessboard.jpg)") # 不知道這為什么不行
 self.setCursor(Qt.PointingHandCursor) # 鼠標(biāo)變成手指形狀
 self.sound_piece = QSound("sound/luozi.wav") # 加載落子音效
 self.sound_win = QSound("sound/win.wav") # 加載勝利音效
 self.sound_defeated = QSound("sound/defeated.wav") # 加載失敗音效

 self.resize(WIDTH, HEIGHT) # 固定大小 540*540
 self.setMinimumSize(QtCore.QSize(WIDTH, HEIGHT))
 self.setMaximumSize(QtCore.QSize(WIDTH, HEIGHT))

 self.setWindowTitle("GoBang") # 窗口名稱
 self.setWindowIcon(QIcon('img/black.png')) # 窗口圖標(biāo)

 # self.lb1 = QLabel('  ', self)
 # self.lb1.move(20, 10)

 self.black = QPixmap('img/black.png')
 self.white = QPixmap('img/white.png')

 self.piece_now = BLACK # 黑棋先行
 self.my_turn = True # 玩家先行
 self.step = 0 # 步數(shù)
 self.x, self.y = 1000, 1000

 self.mouse_point = LaBel(self) # 將鼠標(biāo)圖片改為棋子
 self.mouse_point.setScaledContents(True)
 self.mouse_point.setPixmap(self.black) #加載黑棋
 self.mouse_point.setGeometry(270, 270, PIECE, PIECE)
 self.pieces = [LaBel(self) for i in range(225)] # 新建棋子標(biāo)簽，準(zhǔn)備在棋盤上繪制棋子
 for piece in self.pieces:
  piece.setVisible(True) # 圖片可視
  piece.setScaledContents(True) #圖片大小根據(jù)標(biāo)簽大小可變

 self.mouse_point.raise_() # 鼠標(biāo)始終在最上層
 self.ai_down = True # AI已下棋，主要是為了加鎖，當(dāng)值是False的時(shí)候說明AI正在思考，這時(shí)候玩家鼠標(biāo)點(diǎn)擊失效，要忽略掉 mousePressEvent

 self.setMouseTracking(True)
 self.show()

 def paintEvent(self, event): # 畫出指示箭頭
 qp = QPainter()
 qp.begin(self)
 self.drawLines(qp)
 qp.end()

 def mouseMoveEvent(self, e): # 黑色棋子隨鼠標(biāo)移動(dòng)
 # self.lb1.setText(str(e.x()) + ' ' + str(e.y()))
 self.mouse_point.move(e.x() - 16, e.y() - 16)

 def mousePressEvent(self, e): # 玩家下棋
 if e.button() == Qt.LeftButton and self.ai_down == True:
  x, y = e.x(), e.y() # 鼠標(biāo)坐標(biāo)
  i, j = self.coordinate_transform_pixel2map(x, y) # 對(duì)應(yīng)棋盤坐標(biāo)
  if not i is None and not j is None: # 棋子落在棋盤上，排除邊緣
  if self.chessboard.get_xy_on_logic_state(i, j) == EMPTY: # 棋子落在空白處
   self.draw(i, j)
   self.ai_down = False
   board = self.chessboard.board()
   self.AI = AI(board) # 新建線程對(duì)象，傳入棋盤參數(shù)
   self.AI.finishSignal.connect(self.AI_draw) # 結(jié)束線程，傳出參數(shù)
   self.AI.start() # run

 def AI_draw(self, i, j):
 if self.step != 0:
  self.draw(i, j) # AI
  self.x, self.y = self.coordinate_transform_map2pixel(i, j)
 self.ai_down = True
 self.update()

 def draw(self, i, j):
 x, y = self.coordinate_transform_map2pixel(i, j)

 if self.piece_now == BLACK:
  self.pieces[self.step].setPixmap(self.black) # 放置黑色棋子
  self.piece_now = WHITE
  self.chessboard.draw_xy(i, j, BLACK)
 else:
  self.pieces[self.step].setPixmap(self.white) # 放置白色棋子
  self.piece_now = BLACK
  self.chessboard.draw_xy(i, j, WHITE)

 self.pieces[self.step].setGeometry(x, y, PIECE, PIECE) # 畫出棋子
 self.sound_piece.play() # 落子音效
 self.step += 1 # 步數(shù)+1

 winner = self.chessboard.anyone_win(i, j) # 判斷輸贏
 if winner != EMPTY:
  self.mouse_point.clear()
  self.gameover(winner)

 def drawLines(self, qp): # 指示AI當(dāng)前下的棋子
 if self.step != 0:
  pen = QtGui.QPen(QtCore.Qt.black, 2, QtCore.Qt.SolidLine)
  qp.setPen(pen)
  qp.drawLine(self.x - 5, self.y - 5, self.x + 3, self.y + 3)
  qp.drawLine(self.x + 3, self.y, self.x + 3, self.y + 3)
  qp.drawLine(self.x, self.y + 3, self.x + 3, self.y + 3)

 def coordinate_transform_map2pixel(self, i, j):
 # 從 chessMap 里的邏輯坐標(biāo)到 UI 上的繪制坐標(biāo)的轉(zhuǎn)換
 return MARGIN + j * GRID - PIECE / 2, MARGIN + i * GRID - PIECE / 2

 def coordinate_transform_pixel2map(self, x, y):
 # 從 UI 上的繪制坐標(biāo)到 chessMap 里的邏輯坐標(biāo)的轉(zhuǎn)換
 i, j = int(round((y - MARGIN) / GRID)), int(round((x - MARGIN) / GRID))
 # 有MAGIN, 排除邊緣位置導(dǎo)致 i,j 越界
 if i < 0 or i >= 15 or j < 0 or j >= 15:
  return None, None
 else:
  return i, j

 def gameover(self, winner):
 if winner == BLACK:
  self.sound_win.play()
  reply = QMessageBox.question(self, 'You Win!', 'Continue?',
      QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
 else:
  self.sound_defeated.play()
  reply = QMessageBox.question(self, 'You Lost!', 'Continue?',
      QMessageBox.Yes | QMessageBox.No, QMessageBox.No)

 if reply == QMessageBox.Yes: # 復(fù)位
  self.piece_now = BLACK
  self.mouse_point.setPixmap(self.black)
  self.step = 0
  for piece in self.pieces:
  piece.clear()
  self.chessboard.reset()
  self.update()
 else:
  self.close()


if __name__ == '__main__':
 app = QApplication(sys.argv)
 ex = GoBang()
 sys.exit(app.exec_())

class AI(QtCore.QThread):
 finishSignal = QtCore.pyqtSignal(int, int)

 # 構(gòu)造函數(shù)里增加形參
 def __init__(self, board, parent=None):
 super(AI, self).__init__(parent)
 self.board = board

 # 重寫 run() 函數(shù)
 def run(self):
 self.ai = searcher()
 self.ai.board = self.board
 score, x, y = self.ai.search(2, 2)
 self.finishSignal.emit(x, y)

class LaBel(QLabel):
 def __init__(self, parent):
 super().__init__(parent)
 self.setMouseTracking(True)

 def enterEvent(self, e):
 e.ignore()