import pygame
from compute import matmul, rot_2D, transpose
from moon import Moon

# Colors
WHITE = (255, 255, 255)
RED = (255, 0, 0)

class Apollo(Moon):
    """Die Raumkapsel Apollo kreist um den Mond"""

    def __init__(self, title: str = "Apollo", fps: int = 60, size: tuple[int, int] = (640, 400)):
        super().__init__(title, fps, size)
        # Apollo orbit radius around the moon (relative to moon position)
        self.apollo_radius = 30.0
        # Initial position relative to moon
        self.apollo_offset = (0.0, self.apollo_radius)
        # Apollo angular speed: double moon speed, opposite direction
        self.theta = -2 * self.rot_moon[0][0]  # since rot_2D(theta) second element is cos=rot[0][0]
        self.rot_apollo = rot_2D(self.theta)

    def update_game(self) -> bool:
        # First let the moon move
        super().update_game()
        # Rotate Apollo around the moon
        point = transpose([list(self.apollo_offset)])
        result = matmul(self.rot_apollo, point)
        self.apollo_offset = (result[0][0], result[1][0])
        return True

    def draw_game(self) -> None:
        # Draw background and Earth and Moon via super()
        super().draw_game()
        # Compute Apollo absolute position
        moon_abs = (self.center[0] + self.position_moon[0], self.center[1] + self.position_moon[1])
        apollo_pos = (int(moon_abs[0] + self.apollo_offset[0]), int(moon_abs[1] + self.apollo_offset[1]))
        screen = pygame.display.get_surface()
        pygame.draw.circle(screen, RED, apollo_pos, 5)
        pygame.display.flip()

if __name__ == "__main__":
    Apollo().run()