wuliaozhiji
96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
"""
|
|
Ahira Justice, ADEFOKUN
|
|
justiceahira@gmail.com
|
|
"""
|
|
|
|
|
|
import os
|
|
import pygame
|
|
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
IMAGE_DIR = os.path.join(BASE_DIR, 'images')
|
|
|
|
BLACK = 'BLACK'
|
|
WHITE = 'WHITE'
|
|
|
|
BISHOP = 'BISHOP'
|
|
KING = 'KING'
|
|
KNGHT = 'KNIGHT'
|
|
PAWN = 'PAWN'
|
|
QUEEN = 'QUEEN'
|
|
ROOK = 'ROOK'
|
|
|
|
|
|
class Piece:
|
|
bBishop = pygame.image.load(os.path.join(IMAGE_DIR, 'bB.png'))
|
|
bKing = pygame.image.load(os.path.join(IMAGE_DIR, 'bK.png'))
|
|
bKnight = pygame.image.load(os.path.join(IMAGE_DIR, 'bN.png'))
|
|
bPawn = pygame.image.load(os.path.join(IMAGE_DIR, 'bP.png'))
|
|
bQueen = pygame.image.load(os.path.join(IMAGE_DIR, 'bQ.png'))
|
|
bRook = pygame.image.load(os.path.join(IMAGE_DIR, 'bR.png'))
|
|
|
|
wBishop = pygame.image.load(os.path.join(IMAGE_DIR, 'wB.png'))
|
|
wKing = pygame.image.load(os.path.join(IMAGE_DIR, 'wK.png'))
|
|
wKnight = pygame.image.load(os.path.join(IMAGE_DIR, 'wN.png'))
|
|
wPawn = pygame.image.load(os.path.join(IMAGE_DIR, 'wP.png'))
|
|
wQueen = pygame.image.load(os.path.join(IMAGE_DIR, 'wQ.png'))
|
|
wRook = pygame.image.load(os.path.join(IMAGE_DIR, 'wR.png'))
|
|
|
|
def __init__(self, color, piece, DISPLAYSURF, size):
|
|
self.position = None
|
|
self.sprite = None
|
|
self.DISPLAYSURF = DISPLAYSURF
|
|
self.size = size
|
|
|
|
self.color = color
|
|
self.piece = piece
|
|
|
|
self.setSprite()
|
|
|
|
def setPosition(self, position):
|
|
self.position = position
|
|
|
|
|
|
def setSprite(self):
|
|
if self.piece == BISHOP:
|
|
if self.color == BLACK:
|
|
self.sprite = Piece.bBishop
|
|
elif self.color == WHITE:
|
|
self.sprite = Piece.wBishop
|
|
|
|
elif self.piece == KING:
|
|
if self.color == BLACK:
|
|
self.sprite = Piece.bKing
|
|
elif self.color == WHITE:
|
|
self.sprite = Piece.wKing
|
|
|
|
elif self.piece == KNGHT:
|
|
if self.color == BLACK:
|
|
self.sprite = Piece.bKnight
|
|
if self.color == WHITE:
|
|
self.sprite = Piece.wKnight
|
|
|
|
elif self.piece == PAWN:
|
|
if self.color == BLACK:
|
|
self.sprite = Piece.bPawn
|
|
elif self.color == WHITE:
|
|
self.sprite = Piece.wPawn
|
|
|
|
elif self.piece == QUEEN:
|
|
if self.color == BLACK:
|
|
self.sprite = Piece.bQueen
|
|
elif self.color == WHITE:
|
|
self.sprite = Piece.wQueen
|
|
|
|
elif self.piece == ROOK:
|
|
if self.color == BLACK:
|
|
self.sprite = Piece.bRook
|
|
elif self.color == WHITE:
|
|
self.sprite = Piece.wRook
|
|
|
|
|
|
def displayPiece(self):
|
|
sprite = pygame.transform.scale(self.sprite, (self.size, self.size))
|
|
self.DISPLAYSURF.blit(sprite, self.position)
|