analogchessV4.py

import pygame
from differentfiles.pieces import *
from differentfiles.colors import *
from differentfiles.utils import to_game_coords
from differentfiles.drawing import (
    see_through,
    see_through2,
    draw_line_round_corners_polygon,
    draw_checkers,
    screen,
)

pygame.init()

# fucking awful global variable that is is changed by the confirm() function of the pieces
whites_turn = True
radius = 0.7

pieces = [
    Rook(0.5, 0.5, white),
    Rook(7.5, 0.5, white),
    Knight(1.5, 0.5, white),
    Knight(6.5, 0.5, white),
    Bishop(5.5, 0.5, white),
    Bishop(2.5, 0.5, white),
    King(4.5, 0.5, white),
    Queen(3.5, 0.5, white),
    Pawn(0.5, 1.5, white),
    Pawn(1.5, 1.5, white),
    Pawn(2.5, 1.5, white),
    Pawn(3.5, 1.5, white),
    Pawn(4.5, 1.5, white),
    Pawn(5.5, 1.5, white),
    Pawn(6.5, 1.5, white),
    Pawn(7.5, 1.5, white),
    Rook(0.5, 7.5, black),
    Rook(7.5, 7.5, black),
    Knight(1.5, 7.5, black),
    Knight(6.5, 7.5, black),
    Bishop(5.5, 7.5, black),
    Bishop(2.5, 7.5, black),
    King(4.5, 7.5, black),
    Queen(3.5, 7.5, black),
    Pawn(0.5, 6.5, black),
    Pawn(1.5, 6.5, black),
    Pawn(2.5, 6.5, black),
    Pawn(3.5, 6.5, black),
    Pawn(4.5, 6.5, black),
    Pawn(5.5, 6.5, black),
    Pawn(6.5, 6.5, black),
    Pawn(7.5, 6.5, black),
]


done = False
clock = pygame.time.Clock()
confirmed = True


pygame.display.set_caption("Analog Chess")


draw_line_round_corners_polygon(
    see_through, (120, 120), (220, 220), RED_HIGHLIGHT, 0.7 * 640 / 8
)
grabbed_piece = None

for piece in pieces:
    piece.calc_paths(pieces)
                
while not done:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
            for piece in pieces:
                piece.try_grab(to_game_coords(pygame.mouse.get_pos()))
                """
                if whites_turn:
                    if piece.color == white:
                        piece.try_grab(to_game_coords(pygame.mouse.get_pos()))
                else:
                    if piece.color != white:
                        piece.try_grab(to_game_coords(pygame.mouse.get_pos()))
                """
        elif event.type == pygame.MOUSEMOTION:
            for piece in pieces:
                piece.drag(to_game_coords(pygame.mouse.get_pos()), pieces)

        elif event.type == pygame.MOUSEBUTTONUP:

            new_pieces = []
            for piece in pieces:
                piece.ungrab(pieces)
                if piece.can_promote():
                    new_pieces.append(Queen(piece.x, piece.y, piece.color))
                else:
                    new_pieces.append(piece)

            pieces = new_pieces
            for piece in pieces:
                piece.calc_paths(pieces)

    """
    if not pygame.mouse.get_focused():
        for piece in pieces:
            piece.ungrab(pieces)
    """

    draw_checkers()


    prev_grabbed_piece = grabbed_piece
    grabbed_piece = None
    for piece in pieces:
        if piece.grabbed:
            grabbed_piece = piece   
    
    if grabbed_piece:
        for piece in pieces:
            if piece.color != grabbed_piece.color:
                piece.draw_paths(pieces)
    if grabbed_piece:
        grabbed_piece.draw_moves(pieces)

    screen.blit(see_through, (0, 0))

    screen.blit(see_through2, (0, 0))

    for piece in pieces:
        piece.draw()

    # draw grabbed piece last so it will show up on top
    if grabbed_piece:
        grabbed_piece.draw()

    pygame.display.flip()
    clock.tick(30)

    see_through.fill((0, 0, 0, 0))

    see_through2.fill((0, 0, 0, 0))