class DisplayBoard(QWidget):
def __init__(self, board = None, parent = None):
super(QWidget, self).__init__()
policy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
self.setSizePolicy(policy)
if(board == None):
board = Board()
board.castling_rights = 0
self.board = board
self.parent = parent
self.borderWidth = 12
self.pieceImages = PieceImages()
self.selected_xy = None
self.pcs = [['P','R','B','N','Q','K'],['p','r','b','n','q','k']]
self.initUI()
def initUI(self):
self.show()
def calculateBoardSize(self):
size = self.size()
boardSize = min(size.width(), size.height())
squareSize = (boardSize-(2*self.borderWidth))//8
boardSize = 8 * squareSize + 2 * self.borderWidth
return (boardSize,squareSize)
def resizeEvent(self, e):
self.setMinimumWidth(self.height()*1.35)
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
self.drawBoard(event, qp)
qp.end()
def mousePressEvent(self, mouseEvent):
pos = self.getBoardPosition(mouseEvent.x(), mouseEvent.y())
def getBoardPosition(self,x,y):
(boardSize,squareSize) = self.calculateBoardSize()
# check if x,y are actually on the board
if(x > self.borderWidth and y > self.borderWidth
and x < (boardSize - self.borderWidth)
and y < (boardSize - self.borderWidth)):
x = x - self.borderWidth
y = y - self.borderWidth
x = x // squareSize
y = 7 - (y // squareSize)
return (x,y)
# check if x,y are on place where pieces
# are selected
if(x > self.borderWidth + 9*squareSize and x < self.borderWidth + 11*squareSize
and y > self.borderWidth and y < self.borderWidth + 6*squareSize):
x = x - (self.borderWidth + 9*squareSize)
y = y - self.borderWidth
x = (x // squareSize)+8
y = y//squareSize
return (x,y)
return None
def drawBoard(self, event, qp):
penZero = QPen(Qt.black, 1, Qt.NoPen)
qp.setPen(penZero)
darkBlue = QColor(56,66,91)
#Fritz 13
#lightBlue = QtGui.QColor(111,132,181)
#darkWhite = QtGui.QColor(239,239,239)
#Fritz 6
lightBlue = QColor(90,106,173)
lightBlue2 = QColor(166,188,231)
darkWhite = QColor(239,239,239)
qp.setBrush(darkBlue)
(boardSize,squareSize) = self.calculateBoardSize()
#draw rect (i.e. border) around board
qp.drawRect(1,1,boardSize,boardSize)
#draw rect (i.e. border) around pick up fields
qp.drawRect(9*squareSize,1,2*squareSize+2*self.borderWidth,6 * squareSize + 2 * self.borderWidth)
boardOffsetX = self.borderWidth;
boardOffsetY = self.borderWidth;
# draw Board
for i in range(0,8):
for j in range(0,8):
if((j%2 == 0 and i%2 ==1) or (j%2 == 1 and i%2 ==0)):
qp.setBrush(lightBlue2)
else:
qp.setBrush(lightBlue)
#draw Square
x = boardOffsetX+(i*squareSize)
# drawing coordinates are from top left
# whereas chess coords are from bottom left
y = boardOffsetY+((7-j)*squareSize)
qp.drawRect(x,y,squareSize,squareSize)
#draw Piece
piece = self.board.piece_at(j*8+i)
if(piece != None and piece.symbol() in ('P','R','N','B','Q','K','p','r','n','b','q','k')):
qp.drawImage(x,y,self.pieceImages.getWp(piece.symbol(), squareSize))
for i in range(0,6):
for j in range(0,2):
qp.setBrush(lightBlue2)
if(self.selected_xy != None and self.selected_xy == (j,i)):
qp.setBrush(lightBlue)
#draw Square
x = boardOffsetX+((9+j)*squareSize)
# drawing coordinates are from top left
# whereas chess coords are from bottom left
y = boardOffsetY+(i*squareSize)
qp.drawRect(x,y,squareSize,squareSize)
#draw Piece
qp.drawImage(x,y,self.pieceImages.getWp(self.pcs[j][i], squareSize))
qp.setPen(darkWhite)
qp.setFont(QFont('Decorative',8))
for i in range(0,8):
idx = str(chr(65+i))
qp.drawText(boardOffsetX+(i*squareSize)+(squareSize/2)-4,
boardOffsetY+(8*squareSize)+(self.borderWidth-3),idx)
qp.drawText(4,boardOffsetY+(i*squareSize)+(squareSize/2)+4,str(8-i))
def mousePressEvent(self, mouseEvent):
pos = self.getBoardPosition(mouseEvent.x(), mouseEvent.y())
if(pos):
x = pos[0]
y = pos[1]
if(x > 7):
self.selected_xy = (x-8,y)
else:
if(self.selected_xy != None):
(i,j) = self.selected_xy
piece = self.pcs[i][j]
square = y*8+x
current_piece = self.board.piece_at(square)
if(current_piece and current_piece.symbol() == piece):
self.board.remove_piece_at(square)
else:
self.board.set_piece_at(square,Piece.from_symbol(piece))
if(self.board.status() == 0):
self.parent.enable_ok_button()
else:
self.parent.disable_ok_button()
self.update()
class ChessboardView(QWidget):
def __init__(self,gamestate,engine):
super(ChessboardView, self).__init__()
#super(QWidget, self).__init__()
policy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
self.setSizePolicy(policy)
self.gs = gamestate
self.engine = engine
self.pieceImages = PieceImages()
self.borderWidth = 12
self.moveSrc = None
self.grabbedPiece = None
self.grabbedX = None
self.grabbedY = None
self.drawGrabbedPiece = False
self.flippedBoard = False
#self.cache = BoardCache()
self.initUI()
def initUI(self):
self.show()
def heightForWidth(self, width):
return width
def resizeEvent(self, e):
self.setMinimumWidth(self.height())
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
self.drawBoard(event, qp)
qp.end()
def getBoardPosition(self,x,y):
(boardSize,squareSize) = self.calculateBoardSize()
# check if x,y are actually on the board
if(x > self.borderWidth and y > self.borderWidth
and x < (boardSize - self.borderWidth)
and y < (boardSize - self.borderWidth)):
x = x - self.borderWidth
y = y - self.borderWidth
x = x // squareSize
y = 7 - (y // squareSize)
if(self.flippedBoard):
return Point(7-x,7-y)
else:
return Point(x,y)
return None
@pyqtSlot()
def flip_board(self):
if(self.flippedBoard):
self.flippedBoard = False
else:
self.flippedBoard = True
self.update()
def touchPiece(self, x, y, mouse_x, mouse_y):
self.moveSrc = Point(x,y)
piece = self.gs.current.board().piece_at(y*8+x).symbol()
self.grabbedPiece = piece
self.grabbedX = mouse_x
self.grabbedY = mouse_y
self.drawGrabbedPiece = True
def debug_msg(self,s):
msgBox = QMessageBox()
msgBox.setText(s)
msgBox.exec_()
#self.gt.current.board.set_at(x,y,'e')
def _make_uci(self,point_src,point_dst,promote_to=None):
uci = point_src.to_str() + point_dst.to_str()
if(promote_to != None):
uci += promote_to
return uci
def executeMove(self, uci):
temp = self.gs.current
self.gs.current.invalidate = True
move = chess.Move.from_uci(uci)
# check if move already exists
variation_moves = [ x.move for x in self.gs.current.variations ]
if(move in variation_moves):
for node in self.gs.current.variations:
if(node.move == move):
self.gs.current = node
# otherwise create a new node
else:
self.gs.current.add_variation(move)
# enforce repainting main variation if exists
if(len(self.gs.current.variations)>0 and len(self.gs.current.variations[0].variations)>0):
self.gs.current.variations[0].variations[0].invalidate = True
self.gs.current = self.gs.current.variation(move)
self.emit(SIGNAL("unsaved_changes()"))
#"+str(( self.gs.board().turn + 1)%2))
#new_node = chess.pgn.GameNode()
#new_node.parent = temp
#new_node.move = chess.Move.from_uci(uci)
self.moveSrc = None
self.grabbedPiece = None
self.drawGrabbedPiece = False
#self.movesEdit = MovesEdit(self)
#self.movesEdit.update_san()
# check if game is drawn, or checkmate due to various conditions
if(self.gs.current.board().is_checkmate()): # due to checkmate
display_mbox(self.trUtf8("Checkmate"),self.trUtf8("The game is over!"))
self.emit(SIGNAL("checkmate"))
elif(self.gs.current.board().is_stalemate()): # due to stalemate
display_mbox(self.trUtf8("Stalemate"),self.trUtf8("The game is drawn!"))
self.emit(SIGNAL("drawn"))
elif(self.gs.current.board().is_insufficient_material() and not self.gs.mode == MODE_ENTER_MOVES):
# due to insufficient material
display_mbox(self.trUtf8("Insufficient material to win."),self.trUtf8("The game is drawn!"))
self.emit(SIGNAL("drawn"))
elif(self.gs.current.board().can_claim_threefold_repetition()): # due to threefold repetition
display_mbox(self.trUtf8("Threefold repetition."),self.trUtf8("The game is drawn!"))
self.emit(SIGNAL("drawn"))
if(self.gs.mode == MODE_ANALYSIS):
fen, uci_string = self.gs.printer.to_uci(self.gs.current)
self.engine.send_fen(fen)
self.engine.uci_send_position(uci_string)
self.engine.uci_go_infinite()
#self.update()
if((self.gs.mode == MODE_PLAY_WHITE and self.gs.current.board().turn == chess.BLACK) or
(self.gs.mode == MODE_PLAY_BLACK and self.gs.current.board().turn == chess.WHITE)):
book_move = None
s = (str(os.listdir(".")))
#msgBox = QMessageBox()
#msgBox.setText(s)
#msgBox.exec_()
with open_reader("./books/varied.bin") as reader:
#self.debug_msg("ok, openend file")
entries = reader.find_all(self.gs.current.board())
moves = []
for entry in entries:
move = entry.move().uci()
moves.append(move)
#self.emit(SIGNAL("bestmove(QString)"),move)
l = len(moves)
if(l > 0):
book_move = True
n = random.randint(0,l-1)
book_move = moves[n]
if(book_move != None):
#self.debug_msg("ok, sending book move")
self.emit(SIGNAL("bestmove(QString)"),book_move)
else:
fen, uci_string = self.gs.printer.to_uci(self.gs.current)
self.engine.send_fen(fen)
self.engine.uci_send_position(uci_string)
self.engine.uci_go_movetime(self.gs.computer_think_time)
elif(self.gs.mode == MODE_PLAY_WHITE or self.gs.mode == MODE_PLAY_BLACK): pass
#self.engine.uci_go_infinite()
elif(self.gs.mode == MODE_PLAYOUT_POS):
fen, uci_string = self.gs.printer.to_uci(self.gs.current)
self.engine.send_fen(fen)
self.engine.uci_send_position(uci_string)
self.engine.uci_go_movetime(self.gs.computer_think_time)
self.update()
self.emit(SIGNAL("statechanged()"))
def resetMove(self):
#self.gt.current.board.set_at(self.moveSrc.x,self.moveSrc.y,self.grabbedPiece)
self.moveSrc = None
self.grabbedPiece = None
self.drawGrabbedPiece = False
def __players_turn(self):
if(self.gs.mode == MODE_PLAYOUT_POS):
return True
if(self.gs.mode == MODE_PLAY_BLACK and self.gs.current.board().turn == chess.WHITE):
return False
if(self.gs.mode == MODE_PLAY_WHITE and self.gs.current.board().turn == chess.BLACK):
return False
return True
def _is_valid_and_promotes(self,uci):
if self.gs.mode == MODE_GAME_ANALYSIS:
return False
if(not self.__players_turn()):
return False
legal_moves = self.gs.current.board().legal_moves
for lm in legal_moves:
if(uci == lm.uci()[0:4] and len(lm.uci())==5):
return True
return False
def _is_valid(self,uci):
if self.gs.mode == MODE_GAME_ANALYSIS:
return False
if(not self.__players_turn()):
return False
legal_moves = self.gs.current.board().legal_moves
return (len([x for x in legal_moves if x.uci() == uci]) > 0)
def mousePressEvent(self, mouseEvent):
pos = self.getBoardPosition(mouseEvent.x(), mouseEvent.y())
if(pos):
i = pos.x
j = pos.y
if(self.grabbedPiece):
#m = Point(i,j)
uci = self._make_uci(self.moveSrc,pos)
if(self._is_valid_and_promotes(uci)):
promDialog = DialogPromotion(self.gs.current.board().turn == chess.WHITE)
answer = promDialog.exec_()
if(answer):
uci += promDialog.final_piece.lower()
self.executeMove(uci)
elif(self._is_valid(uci)):
self.executeMove(uci)
else:
self.resetMove()
if(self.gs.current.board().piece_at(j*8+i) != None):
self.touchPiece(i,j,mouseEvent.x(),mouseEvent.y())
else:
if(self.gs.current.board().piece_at(j*8+i) != None):
self.touchPiece(i,j,mouseEvent.x(),mouseEvent.y())
self.update()
def mouseMoveEvent(self, mouseEvent):
button = mouseEvent.button()
if(button == 0 and (not self.grabbedPiece == None)):
self.grabbedX = mouseEvent.x()
self.grabbedY = mouseEvent.y()
self.drawGrabbedPiece = True
self.update()
def mouseReleaseEvent(self, mouseEvent):
self.drawGrabbedPiece = False
pos = self.getBoardPosition(mouseEvent.x(), mouseEvent.y())
if(pos and self.grabbedPiece != None):
if(pos != self.moveSrc):
uci = self._make_uci(self.moveSrc, pos)
if(self._is_valid_and_promotes(uci)):
promDialog = DialogPromotion(self.gs.current.board().turn == chess.WHITE)
answer = promDialog.exec_()
if(answer):
uci += promDialog.final_piece.lower()
self.executeMove(uci)
elif(self._is_valid(uci)):
self.executeMove(uci)
else:
self.resetMove()
self.update()
def calculateBoardSize(self):
size = self.size()
boardSize = min(size.width(), size.height())
squareSize = (boardSize-(2*self.borderWidth))//8
boardSize = 8 * squareSize + 2 * self.borderWidth
return (boardSize,squareSize)
def on_statechanged(self):
self.update()
def drawBoard(self, event, qp):
penZero = QPen(Qt.black, 1, Qt.NoPen)
qp.setPen(penZero)
darkBlue = QColor(56,66,91)
#Fritz 13
#lightBlue = QtGui.QColor(111,132,181)
#darkWhite = QtGui.QColor(239,239,239)
#Fritz 6
lightBlue = QColor(90,106,173)
lightBlue2 = QColor(166,188,231)
darkWhite = QColor(239,239,239)
qp.setBrush(darkBlue)
(boardSize,squareSize) = self.calculateBoardSize()
qp.drawRect(1,1,boardSize,boardSize)
boardOffsetX = self.borderWidth;
boardOffsetY = self.borderWidth;
# if the board is flipped, also
# flip colors so that the board
# loop works without any change/additional
# if-checks
dark = lightBlue2
light = lightBlue
if(self.flippedBoard):
dark = lightBlue
light = lightBlue2
# draw Board
#board = self.cache.get(self.gs.current)
board = self.gs.current.board()
for i in range(0,8):
for j in range(0,8):
if((j%2 == 0 and i%2 ==1) or (j%2 == 1 and i%2 ==0)):
qp.setBrush(dark)
else:
qp.setBrush(light)
#draw Square
if(self.flippedBoard):
x = boardOffsetX+((7-i)*squareSize)
else:
x = boardOffsetX+(i*squareSize)
# drawing coordinates are from top left
# whereas chess coords are from bottom left
y = boardOffsetY+((7-j)*squareSize)
qp.drawRect(x,y,squareSize,squareSize)
#draw Piece
piece = None
if(self.flippedBoard):
piece = board.piece_at((7-j)*8+i)
else:
piece = board.piece_at(j*8+i)
if(piece != None and piece.symbol() in ('P','R','N','B','Q','K','p','r','n','b','q','k')):
# skip piece that is currently picked up
if(not self.flippedBoard):
if(not (self.drawGrabbedPiece and i == self.moveSrc.x and j == self.moveSrc.y)):
qp.drawImage(x,y,self.pieceImages.getWp(piece.symbol(), squareSize))
else:
if(not (self.drawGrabbedPiece and i == self.moveSrc.x and (7-j) == self.moveSrc.y)):
qp.drawImage(x,y,self.pieceImages.getWp(piece.symbol(), squareSize))
if(self.drawGrabbedPiece):
offset = squareSize // 2
qp.drawImage(self.grabbedX-offset,self.grabbedY-offset,self.pieceImages.getWp(self.grabbedPiece, squareSize))
qp.setPen(darkWhite)
qp.setFont(QFont('Decorative',8))
for i in range(0,8):
if(self.flippedBoard):
idx = str(chr(65+(7-i)))
qp.drawText(boardOffsetX+(i*squareSize)+(squareSize/2)-4,
boardOffsetY+(8*squareSize)+(self.borderWidth-3),idx)
qp.drawText(4,boardOffsetY+(i*squareSize)+(squareSize/2)+4,str(i+1))
else:
idx = str(chr(65+i))
qp.drawText(boardOffsetX+(i*squareSize)+(squareSize/2)-4,
boardOffsetY+(8*squareSize)+(self.borderWidth-3),idx)
qp.drawText(4,boardOffsetY+(i*squareSize)+(squareSize/2)+4,str(8-i))
class DialogPromotion(QDialog):
def __init__(self, whitePromotes, parent=None):
super(DialogPromotion, self).__init__(parent)
self.setWindowTitle(self.trUtf8("Promotion"))
self.border = 10
h = self.size().height() - (2 * self.border)
w = (self.size().width() - (2 * self.border)) / 4
# piece size
self.ps = min(h, w)
self.resize(300, 90)
self.whitePromotes = whitePromotes
self.final_piece = "Q"
self.sel_idx = 0
self.pieceImages = PieceImages()
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
# draw images
h = self.size().height() - (2 * self.border)
w = (self.size().width() - (2 * self.border)) / 4
self.ps = min(h, w)
s = self.ps
lightBlue2 = QColor(166, 188, 231)
qp.setBrush(lightBlue2)
for i in range(0, 4):
if self.sel_idx == i:
qp.drawRect(self.border + i * s, self.border, s, s)
qp.drawImage(self.border + i * s, self.border,
self.pieceImages.getWp(self.piece_by_idx(i), s))
qp.end()
def piece_by_idx(self, idx):
if (self.whitePromotes):
if (idx == 0):
return "Q"
elif (idx == 1):
return "R"
elif (idx == 2):
return "B"
elif (idx == 3):
return "N"
if (not self.whitePromotes):
if (idx == 0):
return "q"
elif (idx == 1):
return "r"
elif (idx == 2):
return "b"
elif (idx == 3):
return "n"
def mousePressEvent(self, mouseEvent):
sel_idx = mouseEvent.x() // self.ps
self.sel_idx = sel_idx
self.update()
def mouseReleaseEvent(self, mouseEvent):
self.final_piece = self.piece_by_idx(self.sel_idx)
self.done(True)
def keyPressEvent(self, event):
key = event.key()
if key == Qt.Key_Left:
self.sel_idx = max(self.sel_idx - 1, 0)
self.update()
elif key == Qt.Key_Right:
self.sel_idx = min(self.sel_idx + 1, 3)
self.update()
elif key == Qt.Key_Return:
self.final_piece = self.piece_by_idx(self.sel_idx)
self.done(True)
class DisplayBoard(QWidget):
def __init__(self, board = None, parent = None):
super(QWidget, self).__init__()
policy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
self.setSizePolicy(policy)
if(board == None):
board = Bitboard()
board.castling_rights = CASTLING_NONE
self.board = board
self.parent = parent
self.borderWidth = 12
self.pieceImages = PieceImages()
self.selected_xy = None
self.pcs = [['P','R','B','N','Q','K'],['p','r','b','n','q','k']]
self.initUI()
def initUI(self):
self.show()
def calculateBoardSize(self):
size = self.size()
boardSize = min(size.width(), size.height())
squareSize = (boardSize-(2*self.borderWidth))//8
boardSize = 8 * squareSize + 2 * self.borderWidth
return (boardSize,squareSize)
def resizeEvent(self, e):
self.setMinimumWidth(self.height()*1.35)
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
self.drawBoard(event, qp)
qp.end()
def mousePressEvent(self, mouseEvent):
pos = self.getBoardPosition(mouseEvent.x(), mouseEvent.y())
#print(str(pos))
def getBoardPosition(self,x,y):
(boardSize,squareSize) = self.calculateBoardSize()
# check if x,y are actually on the board
if(x > self.borderWidth and y > self.borderWidth
and x < (boardSize - self.borderWidth)
and y < (boardSize - self.borderWidth)):
x = x - self.borderWidth
y = y - self.borderWidth
x = x // squareSize
y = 7 - (y // squareSize)
return (x,y)
# check if x,y are on place where pieces
# are selected
if(x > self.borderWidth + 9*squareSize and x < self.borderWidth + 11*squareSize
and y > self.borderWidth and y < self.borderWidth + 6*squareSize):
x = x - (self.borderWidth + 9*squareSize)
y = y - self.borderWidth
x = (x // squareSize)+8
y = y//squareSize
return (x,y)
return None
def drawBoard(self, event, qp):
penZero = QPen(Qt.black, 1, Qt.NoPen)
qp.setPen(penZero)
darkBlue = QColor(56,66,91)
#Fritz 13
#lightBlue = QtGui.QColor(111,132,181)
#darkWhite = QtGui.QColor(239,239,239)
#Fritz 6
lightBlue = QColor(90,106,173)
lightBlue2 = QColor(166,188,231)
darkWhite = QColor(239,239,239)
qp.setBrush(darkBlue)
(boardSize,squareSize) = self.calculateBoardSize()
#draw rect (i.e. border) around board
qp.drawRect(1,1,boardSize,boardSize)
#draw rect (i.e. border) around pick up fields
qp.drawRect(9*squareSize,1,2*squareSize+2*self.borderWidth,6 * squareSize + 2 * self.borderWidth)
boardOffsetX = self.borderWidth;
boardOffsetY = self.borderWidth;
# draw Board
for i in range(0,8):
for j in range(0,8):
if((j%2 == 0 and i%2 ==1) or (j%2 == 1 and i%2 ==0)):
qp.setBrush(lightBlue2)
else:
qp.setBrush(lightBlue)
#draw Square
x = boardOffsetX+(i*squareSize)
# drawing coordinates are from top left
# whereas chess coords are from bottom left
y = boardOffsetY+((7-j)*squareSize)
qp.drawRect(x,y,squareSize,squareSize)
#draw Piece
piece = self.board.piece_at(j*8+i)
if(piece != None and piece.symbol() in ('P','R','N','B','Q','K','p','r','n','b','q','k')):
qp.drawImage(x,y,self.pieceImages.getWp(piece.symbol(), squareSize))
for i in range(0,6):
for j in range(0,2):
qp.setBrush(lightBlue2)
if(self.selected_xy != None and self.selected_xy == (j,i)):
qp.setBrush(lightBlue)
#draw Square
x = boardOffsetX+((9+j)*squareSize)
# drawing coordinates are from top left
# whereas chess coords are from bottom left
y = boardOffsetY+(i*squareSize)
qp.drawRect(x,y,squareSize,squareSize)
#draw Piece
qp.drawImage(x,y,self.pieceImages.getWp(self.pcs[j][i], squareSize))
qp.setPen(darkWhite)
qp.setFont(QFont('Decorative',8))
for i in range(0,8):
idx = str(chr(65+i))
qp.drawText(boardOffsetX+(i*squareSize)+(squareSize/2)-4,
boardOffsetY+(8*squareSize)+(self.borderWidth-3),idx)
qp.drawText(4,boardOffsetY+(i*squareSize)+(squareSize/2)+4,str(8-i))
def mousePressEvent(self, mouseEvent):
pos = self.getBoardPosition(mouseEvent.x(), mouseEvent.y())
if(pos):
x = pos[0]
y = pos[1]
if(x > 7):
self.selected_xy = (x-8,y)
else:
if(self.selected_xy != None):
(i,j) = self.selected_xy
piece = self.pcs[i][j]
square = y*8+x
current_piece = self.board.piece_at(square)
if(current_piece and current_piece.symbol() == piece):
self.board.remove_piece_at(square)
else:
self.board.set_piece_at(square,Piece.from_symbol(piece))
if(self.board.status() == 0):
self.parent.enable_ok_button()
else:
self.parent.disable_ok_button()
self.update()
class DialogPromotion(QDialog):
def __init__(self, whitePromotes, parent=None):
super(DialogPromotion, self).__init__(parent)
self.setWindowTitle(self.trUtf8("Promotion"))
self.border = 10
h = self.size().height() - (2*self.border)
w = (self.size().width()-(2*self.border))/4
# piece size
self.ps = min(h,w)
self.resize(300, 90)
self.whitePromotes = whitePromotes
self.final_piece = "Q"
self.sel_idx = 0
self.pieceImages = PieceImages()
def paintEvent(self, event):
qp = QPainter()
qp.begin(self)
# draw images
h = self.size().height() - (2*self.border)
w = (self.size().width()-(2*self.border))/4
self.ps = min(h,w)
s = self.ps
lightBlue2 = QColor(166,188,231)
qp.setBrush(lightBlue2)
for i in range(0,4):
if self.sel_idx == i:
qp.drawRect(self.border+i*s,self.border,s,s)
qp.drawImage(self.border+i*s,self.border,self.pieceImages.getWp(self.piece_by_idx(i),s))
qp.end()
def piece_by_idx(self,idx):
if(self.whitePromotes):
if(idx == 0):
return "Q"
elif(idx == 1):
return "R"
elif(idx == 2):
return "B"
elif(idx == 3):
return "N"
if(not self.whitePromotes):
if(idx == 0):
return "q"
elif(idx == 1):
return "r"
elif(idx == 2):
return "b"
elif(idx == 3):
return "n"
def mousePressEvent(self, mouseEvent):
sel_idx = mouseEvent.x() // self.ps
self.sel_idx = sel_idx
self.update()
def mouseReleaseEvent(self, mouseEvent):
self.final_piece = self.piece_by_idx(self.sel_idx);
self.done(True)
def keyPressEvent(self, event):
key = event.key()
if key == Qt.Key_Left:
self.sel_idx = max(self.sel_idx-1,0)
self.update()
elif key == Qt.Key_Right:
self.sel_idx = min(self.sel_idx+1,3)
self.update()
elif key == Qt.Key_Return :
self.final_piece = self.piece_by_idx(self.sel_idx);
self.done(True)