window = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT)) pygame.display.set_caption("Venus, Earth, Mars and Sun") clock = pygame.time.Clock() rate = 10000000 """BodyName = Body('Name', Mass, rotation_speed, x, y, vx, vy, radius, color)""" Sun = Body('Sun', S_MASS, 0, 0, 0, 0, 0, PIX2M * 30, (255, 165, 0)) Mars = Body('Mars', MA_MASS, 0, -MA_DIST, 0, 0, -MA_SPEED * rate, PIX2M * 8, (180, 0, 0)) Earth = Body('Earth', E_MASS, 0, AU, 0, 0, E_SPEED * rate, PIX2M * 10, (0, 0, 165)) Venus = Body('Venus', V_MASS, 0, 0, -V_DIST, V_SPEED * rate, 0, PIX2M * 10, (90, 90, 90)) system = System(G2 * (rate**2), 2) system.add_body(Sun) system.add_body(Mars) system.add_body(Venus) system.add_body(Earth) sim = Simulation(system) run = True while run: clock.tick(FREQUENCY) for event in pygame.event.get(): if event.type == pygame.QUIT: run = False sim.update() draw(sim, window)