[+] A2 Starter files

2022-02-10 18:43:51 -05:00
parent d17d902fb4
commit 4e8b7e8748
30 changed files with 4252 additions and 0 deletions
@@ -0,0 +1,95 @@
+"""
+    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)