Initial commit

main.py

@@ -0,0 +1,51 @@
+# import math
+import pygame
+from planet import Planet
+
+pygame.init()
+
+WIDTH, HEIGHT = 800, 800
+
+WINDOW = pygame.display.set_mode((WIDTH, HEIGHT))
+pygame.display.set_caption("Planet simulation")
+
+YELLOW = (255, 255, 0)
+BLUE = (0, 0, 255)
+RED = (255, 0, 0)
+GRAY = (80, 78, 81)
+WHITE = (255, 255, 255)
+
+def main():
+    run = True
+    clock = pygame.time.Clock()
+
+
+    sun = Planet(0, 0, 30, YELLOW, 1.98892e30, True)
+    earth = Planet(-1 * Planet.AU, 0, 16, BLUE, 5.9742e24)
+    earth.y_vel = 29.783e3
+    mars = Planet(-1.524 * Planet.AU, 0, 12, RED, 6.39e23)
+    mars.y_vel = 24.077e3
+    mercury = Planet(0.387 * Planet.AU, 0, 8, GRAY, 3.3e23)
+    mercury.y_vel = -47.4e3
+    venus = Planet(0.723 * Planet.AU, 0, 14, WHITE, 4.8685e24)
+    venus.y_vel = -35.02e3
+
+    planets = [sun, earth, mars, mercury, venus]
+
+    while run:
+        # Cap the loop at 60 Hz
+        clock.tick(60)
+        WINDOW.fill((0, 0, 0))
+
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT: run = False
+
+        for planet in planets:
+            planet.update_position(planets)
+            planet.draw(WINDOW)
+
+        pygame.display.update()
+
+    pygame.quit()
+
+main()