import pygame

from compute import matmul, rot_2D, transpose
from game import Game

EARTH_RADIUS = 30
MOON_RADIUS = 10
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
BLUE = (0, 0, 255)


class Moon(Game):
    """Der Mond kreist um die Erde."""

    def __init__(
            self,
            title: str = "Moon",
            fps: int = 60,
            size: tuple[int, int] = (640, 400),
    ) -> None:
        """Legt Startposition, Drehwinkel und Rotationsmatrix fest."""
        super().__init__(title, fps, size)

        # Startposition des Mondes relativ zum Mittelpunkt (z. B. 120 Pixel nach rechts)
        self.position_moon = (120.0, 0.0)

        # Kleiner Drehwinkel im Bogenmaß und Erstellung der Rotationsmatrix
        theta = 0.01
        self.rot_moon = rot_2D(theta)

    def update_game(self) -> bool:
        """Dreht den Mond ein Stück weiter."""
        # Mondposition mit der Rotationsmatrix weiterdrehen
        point = transpose([list(self.position_moon)])  # [x, y] wird zur Spalte [[x], [y]]
        result = matmul(self.rot_moon, point)  # Ergebnis: [[x'], [y']]
        self.position_moon = (result[0][0], result[1][0])
        return True

    def draw_game(self) -> None:
        """Zeichnet Erde und Mond."""
        # Bildschirm löschen
        self.screen.fill(BLACK)

        # Mittelpunkt des Fensters ermitteln
        center_x = self.screen.get_width() // 2
        center_y = self.screen.get_height() // 2

        # Erde als blauen Kreis in der Mitte zeichnen
        pygame.draw.circle(self.screen, BLUE, (center_x, center_y), EARTH_RADIUS)

        # Mond als weißen Kreis auf der Kreisbahn zeichnen
        moon_x = int(center_x + self.position_moon[0])
        moon_y = int(center_y + self.position_moon[1])
        pygame.draw.circle(self.screen, WHITE, (moon_x, moon_y), MOON_RADIUS)

        # Bild anzeigen
        pygame.display.flip()


if __name__ == "__main__":
    # Moon-Objekt erzeugen und starten
    game = Moon()
    game.run()