Files
matrixmania/moon.py
T

68 lines
2.0 KiB
Python

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)
# Kleiner Drehwinkel im Bogenmaß und zugehörige Rotationsmatrix
theta = 0.01
self.rot_moon = rot_2D(theta)
# Startposition des Mondes als relativer Abstand zur Erde (z. B. 120 Pixel nach rechts)
self.position_moon = (120.0, 0.0)
def update_game(self) -> bool:
"""Dreht den Mond ein Stück weiter."""
# 1. Position in Spaltenvektor [[x], [y]] umwandeln
point = transpose([list(self.position_moon)])
# 2. Mit der Rotationsmatrix multiplizieren
result = matmul(self.rot_moon, point)
# 3. Ergebnis zurück in ein Tupel konvertieren und speichern
self.position_moon = (result[0][0], result[1][0])
return True
def draw_game(self) -> None:
"""Zeichnet Erde und Mond."""
# Bildschirm mit Schwarz füllen
self.screen.fill(BLACK)
# Fenstermittelpunkt ermitteln, damit die Erde exakt in der Mitte sitzt
center_x = self.size[0] // 2
center_y = self.size[1] // 2
# Erde als blauen Kreis in der Mitte zeichnen
pygame.draw.circle(self.screen, BLUE, (center_x, center_y), EARTH_RADIUS)
# Mondposition relativ zum Mittelpunkt berechnen und 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)
# Anzeige aktualisieren
pygame.display.flip()
if __name__ == "__main__":
app = Moon()
app.run()