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 & Apollo",
            fps: int = 60,
            size: tuple[int, int] = (640, 400),
    ) -> None:
        """Legt Startposition, Drehwinkel und Rotationsmatrix für den Mond fest."""
        super().__init__(title, fps, size)

        # Mond-Parameter
        self.theta_moon = 0.01
        self.rot_moon = rot_2D(self.theta_moon)
        self.position_moon = (120.0, 0.0)

    def update_game(self) -> bool:
        """Aktualisiert die Positionen aller Himmelskörper."""
        self.update_moon()
        return True

    def update_moon(self) -> None:
        """Dreht den Mond ein Stück weiter um die Erde."""
        point = transpose([list(self.position_moon)])
        result = matmul(self.rot_moon, point)
        self.position_moon = (result[0][0], result[1][0])

    def draw_game(self) -> None:
        """Zeichnet den Hintergrund, die Erde und den Mond."""
        self.screen.fill(BLACK)

        center_x = self.size[0] // 2
        center_y = self.size[1] // 2

        # Erde zeichnen
        pygame.draw.circle(self.screen, BLUE, (center_x, center_y), EARTH_RADIUS)

        # Mond 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)

        # Hier können Unterklassen (wie Apollo) weiteres Zeichnen anhängen
        self.draw_objects(center_x, center_y)

        pygame.display.flip()

    def draw_objects(self, center_x: int, center_y: int) -> None:
        """Erweiterungspunkt für Unterklassen, um zusätzliche Objekte zu zeichnen."""
        pass


if __name__ == "__main__":
    app = Moon()
    app.run()