Added Planet Simulation Python Script

Step-By-Step-Guides/PlanetSimulation.py

@@ -0,0 +1,146 @@
+# importing the required Packages
+import pygame
+import math
+
+# To initialize all imported pygame modules
+pygame.init()
+
+# Setting the Width & Height of the Output Window
+WIDTH , HEIGHT = 1000,800
+WIN = pygame.display.set_mode((WIDTH,HEIGHT))
+
+# Setting the Caption of the Output Window
+pygame.display.set_caption("Planet Simulation")
+
+# RGB Color Codes of 5 Different Colors
+white = (255, 255, 255)
+yellow = (255, 255, 0)
+blue = (100, 149, 237)
+red = (188, 39, 50)
+dark_grey = (80, 78, 81)
+
+# Setting the Font to be displayed in the Output Window
+FONT = pygame.font.SysFont("comicsans", 16)
+
+# Creation of Class "Planet"
+class Planet:
+    # Setting the required Scale Parameters
+    AU = 149.6e6 * 1000
+    G = 6.67428e-11
+    SCALE = 250 / AU   # 1AU = 100 pixels
+    timestep = 3600 * 24 # 1 day
+
+    # Defining the Construtor to Initialize all the class variables
+    def __init__(self,x,y,radius,color,mass):
+        self.x = x
+        self.y = y
+        self.radius = radius
+        self.color = color
+        self.mass = mass
+
+        self.orbit = []
+        self.sun = False
+        self.distance_to_sun = 0
+
+        self.x_vel = 0
+        self.y_vel = 0
+
+    # Function to draw the Orbit Path of Planets
+    def draw(self, win):
+        x = self.x * self.SCALE + (WIDTH / 2)
+        y = self.y * self.SCALE + (HEIGHT / 2)
+
+        if len(self.orbit) >2 :
+            Updated_points = []
+            for point in self.orbit:
+                x, y = point
+                x = x *self.SCALE + WIDTH/2
+                y = y *self.SCALE + HEIGHT/2
+                Updated_points.append((x,y))
+
+            pygame.draw.lines(win, self.color, False, Updated_points, 2)
+
+        pygame.draw.circle(win, self.color, (x, y), self.radius)
+
+        if not self.sun:
+            distance_text = FONT.render(f"{round(self.distance_to_sun/1000, 1)} km", 1, white )
+            win.blit(distance_text,(x - distance_text.get_width()/2, y - distance_text.get_width()/2))
+
+    # Function to set the distance of Planets with respect to Sun
+    def attraction(self, other):
+        other_x, other_y = other.x, other.y
+        distance_x = other_x - self.x
+        distance_y = other_y - self.y
+        distance = math.sqrt(distance_x ** 2 + distance_y ** 2)
+
+        if other.sun :
+            self.distance_to_sun = distance
+
+        force = self.G * self.mass * other.mass / distance**2
+        theta = math.atan2(distance_y, distance_x)
+        force_X = math.cos(theta) * force
+        force_Y = math.sin(theta) * force
+        return force_X, force_Y
+
+    # Function to Update the Positions of Planets with respect to Time
+    def update_position(self,planets):
+        total_fx = total_fy = 0
+        for planet in planets:
+
+            if self == planet:
+                continue
+
+            fx , fy = self.attraction(planet)
+            total_fx += fx
+            total_fy += fy
+
+        self.x_vel += total_fx / self.mass * self.timestep
+        self.y_vel += total_fy / self.mass * self.timestep
+
+        self.x += self.x_vel * self.timestep
+        self.y += self.y_vel * self.timestep
+        self.orbit.append((self.x,self.y))
+
+
+def main():
+    run = True
+    clock = pygame.time.Clock()
+
+    # creating 5 objects for 5 planets
+
+    sun = Planet(0, 0, 30,  yellow, 1.98892 * 10**30)
+    sun.sun = True
+
+    earth = Planet(-1 * Planet.AU, 0, 16, blue, 5.9742 * 10**24)
+    earth.y_vel = 29.738 * 1000
+
+    mars = Planet(-1.524 * Planet.AU, 0, 12, red, 6.39 * 10**23)
+    mars.y_vel = 24.077 *1000
+
+    mercury = Planet(0.387 * Planet.AU, 0, 8, dark_grey, 3.30 * 10**23)
+    mercury.y_vel = -47.4 * 1000
+
+    venus = Planet(0.732 * Planet.AU, 0, 14, white, 4.8685 * 10**24 )
+    venus.y_vel = -35.02 * 1000
+
+    planets = [sun,earth,mars,mercury,venus]
+
+    while run:
+        clock.tick(60)
+        WIN.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(WIN)
+
+        pygame.display.update()
+
+    # exiting the game
+    pygame.quit()
+
+# Running the Main Function 
+main()