88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
import os
|
|
from typing import override
|
|
import pygame
|
|
from src.net import NetGame
|
|
|
|
# pyright: reportUnusedCallResult=false, reportAny=false
|
|
|
|
LEFT_TURN = pygame.USEREVENT + 1
|
|
RIGHT_TURN = pygame.USEREVENT + 2
|
|
|
|
|
|
class PieceSprite(pygame.sprite.Sprite):
|
|
image: pygame.Surface
|
|
rect: pygame.Rect
|
|
x: int
|
|
y: int
|
|
|
|
def __init__(self, x: int, y: int) -> None:
|
|
super().__init__()
|
|
self.image = pygame.Surface((28, 28))
|
|
self.rect = self.image.get_rect()
|
|
self.rect.x = 30 * x + 1
|
|
self.rect.y = 30 * y + 1
|
|
self.x = x
|
|
self.y = y
|
|
|
|
@override
|
|
def update(self, game: NetGame, events: list[pygame.event.Event]):
|
|
piece = game.get_piece(self.x, self.y)
|
|
image = pygame.image.load(os.path.join("assets", f"{piece.type}.bmp")).convert()
|
|
image = pygame.transform.rotate(image, -90 * piece.direction)
|
|
self.image.blit(image, (0, 0))
|
|
for event in events:
|
|
if event.type == pygame.MOUSEBUTTONUP:
|
|
if not self.rect.collidepoint(*event.pos):
|
|
continue
|
|
if event.button == 1:
|
|
pygame.event.post(
|
|
pygame.event.Event(LEFT_TURN, {"x": self.x, "y": self.y})
|
|
)
|
|
elif event.button == 3:
|
|
pygame.event.post(
|
|
pygame.event.Event(RIGHT_TURN, {"x": self.x, "y": self.y})
|
|
)
|
|
|
|
|
|
class NetGUI:
|
|
game: NetGame
|
|
window: pygame.Surface
|
|
pieceSprites: pygame.sprite.Group[PieceSprite]
|
|
|
|
def __init__(self):
|
|
self.game = NetGame(5, 5)
|
|
pygame.init()
|
|
self.window = pygame.display.set_mode((5 * 30, 5 * 30))
|
|
self.window.fill("#ffffff")
|
|
pygame.display.set_caption("Net")
|
|
|
|
self.pieceSprites = pygame.sprite.Group()
|
|
for x in range(5):
|
|
for y in range(5):
|
|
self.pieceSprites.add(PieceSprite(x, y))
|
|
|
|
def run_game(self):
|
|
while True:
|
|
events = pygame.event.get()
|
|
|
|
for event in events:
|
|
if event.type == pygame.QUIT:
|
|
raise SystemExit
|
|
elif event.type == LEFT_TURN:
|
|
self.game.turn_ccw(event.x, event.y)
|
|
if self.game.solved():
|
|
raise SystemExit
|
|
elif event.type == RIGHT_TURN:
|
|
self.game.turn_cw(event.x, event.y)
|
|
if self.game.solved():
|
|
raise SystemExit
|
|
|
|
self.pieceSprites.update(self.game, events)
|
|
for sprite in self.pieceSprites:
|
|
self.window.blit(sprite.image, sprite.rect)
|
|
|
|
pygame.display.flip()
|
|
|
|
|
|
NetGUI().run_game()
|