pygame编写的万有引力效果演示

黎明zxc

pygame编写的万有引力效果演示

GMm_mian.py

1. import pygame
   
2. from GMm_ import *
   
3. pygame.init()
   
4. 
   
5. m1 = [1, 700, 300, 0, 0]
   
6. m2 = [1e14, 500, 300, 0, 0]
   
7. m3 = [1, 300, 300, 0, 0]
   
8. m1[4] = speed_v(m2[0], m1[1]-m2[1], m1[2]-m2[2])
   
9. m3[4] = -speed_v(m2[0], m3[1]-m2[1], m3[2]-m2[2])
   
10. 
    
11. window = pygame.display.set_mode((WIN_WIDTH, WIN_HEIGHT))
    
12. pygame.display.set_caption("star")
    
13. 
    
14. zhong = star(m1)
    
15. zhong1 = star(m2)
    
16. diqiou = star(m3)
    
17. stars_ = stars()
    
18. stars_.add(zhong)
    
19. stars_.add(zhong1)
    
20. stars_.add(diqiou)
    
21. 
    
22. window.fill((255, 255, 255))
    
23. FPS = 60
    
24. clock = pygame.time.Clock()
    
25. CREATE_ENEMY_EVENT = pygame.USEREVENT
    
26. pygame.time.set_timer(CREATE_ENEMY_EVENT, 1000)
    
27. 
    
28. 
    
29. while True:
    
30. 
    
31.     clock.tick(FPS)
    
32. 
    
33.     pygame.draw.circle(window, (211, 211, 211), (zhong.x, zhong.y), 10)
    
34.     pygame.draw.circle(window, (0, 0, 0), (zhong.x, zhong.y), 5)
    
35.     pygame.draw.circle(window, (211, 211, 211), (zhong1.x, zhong1.y), 10)
    
36.     pygame.draw.circle(window, (0, 0, 0), (zhong1.x, zhong1.y), 5)
    
37.     pygame.draw.circle(window, (211, 211, 211), (diqiou.x, diqiou.y), 10)
    
38.     pygame.draw.circle(window, (0, 0, 0), (diqiou.x, diqiou.y), 5)
    
39. 
    
40.     pygame.display.update()
    
41. 
    
42.     pygame.draw.circle(window, (211, 211, 211), (zhong.x, zhong.y), 5)
    
43.     pygame.draw.circle(window, (211, 211, 211), (zhong1.x, zhong1.y), 5)
    
44.     pygame.draw.circle(window, (211, 211, 211), (diqiou.x, diqiou.y), 5)
    
45. 
    
46.     stars_.gravitation(zhong)
    
47.     stars_.gravitation(zhong1)
    
48.     stars_.gravitation(diqiou)
    
49. 
    
50.     for event in pygame.event.get():
    
51. 
    
52.         if event.type == pygame.QUIT:
    
53.             exit()
    
54. 
    
55.         elif event.type == CREATE_ENEMY_EVENT:
    
56.             print(zhong.speed_x, zhong.speed_y, zhong1.speed_x, zhong1.speed_y, diqiou.speed_x, diqiou.speed_y)
    

复制代码

GMm_.py

1. G = 6.67e-11
   
2. WIN_WIDTH = 1000
   
3. WIN_HEIGHT = 600
   
4. 
   
5. 
   
6. def speed_v(M, x, y):
   
7.     return (G * M / ((x ** 2 + y ** 2) ** 0.5)) ** 0.5
   
8. 
   
9. 
   
10. class star:
    
11.     def __init__(self, li):
    
12.         self.m = li[0]
    
13.         self.x = li[1]
    
14.         self.y = li[2]
    
15.         self.speed_x = li[3]
    
16.         self.speed_y = li[4]
    
17. 
    
18.     def motion(self):
    
19.         self.x += self.speed_x
    
20.         self.y += self.speed_y
    
21. 
    
22.     def acceleration(self, q_x, q_y):
    
23.         self.speed_x += q_x
    
24.         self.speed_y += q_y
    
25. 
    
26. 
    
27. class stars:
    
28.     def __init__(self):
    
29.         self.li = []
    
30. 
    
31.     def add(self, star_):
    
32.         self.li.append(star_)
    
33. 
    
34.     def gravitation(self, star_):
    
35.         a = 0
    
36.         b = 0
    
37.         for i in self.li:
    
38.             if i == star_:
    
39.                 continue
    
40.             else:
    
41.                 x = i.x - star_.x
    
42.                 y = i.y - star_.y
    
43.                 r = (x ** 2 + y ** 2) ** 0.5
    
44.                 q = G * i.m / r ** 2
    
45.                 a += x / r * q
    
46.                 b += y / r * q
    
47.         star_.acceleration(a, b)
    
48.         star_.motion()
    

复制代码

                               
登录/注册后可看大图

Python初学者8号

牛蛙，哈哈，坐等更新三体