chess.py

#!/usr/bin/python3

from Piece import Piece
from engine import *
import curses
import time

BG_BLACK = 1
BG_WHITE = 2
FG_BLACK = 3
FG_WHITE = 4
BG_SELECTION = 5

BLACK_ON_BLACK = 1
BLACK_ON_WHITE = 2
WHITE_ON_BLACK = 3
WHITE_ON_WHITE = 4
BLACK_ON_SELECTION = 5
WHITE_ON_SELECTION = 6

BOARD_MARGIN_TOP = 3
SCREEN_CENTERED = True

destination = None

def print_square(i,j,margin_left):
    p = board[i][j]
    bg = get_square_color(i,j,p)
    if p:
        screen.addstr(BOARD_MARGIN_TOP+i,margin_left + 3*j,f" {p.icon} ",curses.color_pair(bg))
    else:
        screen.addstr(BOARD_MARGIN_TOP+i,margin_left + 3*j,f"   ",curses.color_pair(bg))


def get_square_color(i,j,p):
    if p and p.position == get_selected_square():
        if p.color == WHITE:
            return WHITE_ON_SELECTION
        else:
            return BLACK_ON_SELECTION
    if((i % 2 == 1 and j % 2 == 1) or (i % 2 == 0 and j % 2 == 0)):
        if (p and p.color == WHITE):
                return WHITE_ON_WHITE
        return BLACK_ON_WHITE
    if (p and p.color == WHITE):
        return WHITE_ON_BLACK
    return BLACK_ON_BLACK

def get_margin_left():
    if SCREEN_CENTERED:
        margin = max(curses.COLS / 2 - 4*3, 0)
        return int(margin)
    return 0

def print_board():
    margin_left = get_margin_left()

    for i in range(8):
        for j in range(8):
            print_square(i,j,margin_left)  
    screen.addstr(BOARD_MARGIN_TOP + 9,margin_left,f"{get_turn().capitalize()}'s turn")
    screen.addstr(BOARD_MARGIN_TOP + 10,margin_left,f"{get_state().capitalize()}")
    screen.addstr(BOARD_MARGIN_TOP + 11,margin_left,f"[q] Quit")
    screen.addstr(BOARD_MARGIN_TOP + 12,margin_left,f"[r] Reset")

def get_square_location(x,y):
    margin_left = get_margin_left()
    i = y-BOARD_MARGIN_TOP
    j = int((x-margin_left)/3)
    return i, j

def init_colors():
    curses.start_color()
    curses.use_default_colors()

    curses.init_color(BG_BLACK, 627, 321, 176)
    curses.init_color(BG_WHITE, 770, 721, 529)
    curses.init_color(FG_BLACK, 0, 0, 0)
    curses.init_color(FG_WHITE, 1000, 1000, 1000)
    curses.init_color(BG_SELECTION, 800, 300, 300)

    curses.init_pair(BLACK_ON_BLACK, FG_BLACK, BG_BLACK)
    curses.init_pair(BLACK_ON_WHITE, FG_BLACK, BG_WHITE)
    curses.init_pair(WHITE_ON_BLACK, FG_WHITE, BG_BLACK)
    curses.init_pair(WHITE_ON_WHITE, FG_WHITE, BG_WHITE)
    curses.init_pair(WHITE_ON_SELECTION, FG_WHITE, BG_SELECTION)
    curses.init_pair(BLACK_ON_SELECTION, FG_BLACK, BG_SELECTION)

if __name__ == '__main__':
    screen = curses.initscr()
    curses.noecho()
    curses.cbreak()
    curses.curs_set(0)
    screen.keypad(1) 
    curses.mousemask(1)

    init_colors()
    init_board()
    print_board()
    screen.refresh()

    try:
        while True:
            event = screen.getch() 
            if event == ord("q"): break 
            if event == ord("r"):
               init_board()  
            if event == curses.KEY_MOUSE:
                _, mx, my, _, _ = curses.getmouse()
                i,j = get_square_location(mx, my)
                click(i,j)

            print_board()
            screen.refresh()

    finally:
        curses.echo()
        curses.nocbreak()
        curses.endwin()