[已解决]碰撞检测

歌者文明清理员

源代码
只要看第 129 到 133 行就行了

1. # -*- coding: utf-8 -*-
   
2. import pygame
   
3. import math
   
4. import numpy as np
   
5. import random
   
6. from trail import TrailSprite
   
7. from tkinter import *
   
8. 
   
9. pygame.init()
   
10. G = 1
    
11. pygame.time.set_timer(pygame.USEREVENT, 5000)
    
12. 
    
13. 
    
14. class Star(pygame.sprite.Sprite, TrailSprite):
    
15.     def __init__(self, radius, mass, init_speed, pos):
    
16.         pygame.sprite.Sprite.__init__(self)
    
17.         self.radius = radius
    
18.         self.mass = mass
    
19.         self.color = rand_color()
    
20.         if init_speed:
    
21.             self.speed = init_speed
    
22.         self.image = pygame.Surface((radius * 2, radius * 2)).convert_alpha()
    
23.         self.image.fill((0, 0, 0, 0))
    
24.         pygame.draw.circle(self.image, self.color, (radius, radius), radius)
    
25.         TrailSprite.__init__(self, self.color, self.image, 1)
    
26.         TrailSprite.set_screen(self, screen)
    
27.         self.rect.center = pos
    
28. 
    
29.     def move(self):
    
30.         TrailSprite.update(self)
    
31.         self.rect.centerx += self.speed[0]
    
32.         self.rect.centery += self.speed[1]
    
33.         # 跑出窗口从另一边回来
    
34.         if self.rect.left <= 0 or self.rect.right >= width:
    
35.             self.rect.centerx %= width
    
36.             self.clear_trail()
    
37.         if self.rect.top <= 0 or self.rect.bottom >= height:
    
38.             self.rect.centery %= height
    
39.             self.clear_trail()
    
40. 
    
41. 
    
42. def all_clear_trail():
    
43.     for st in stars:
    
44.         st.clear_trail()
    
45. 
    
46. 
    
47. def rand_color():
    
48.     r = random.randint(0, 255)
    
49.     g = random.randint(0, 255)
    
50.     b = random.randint(0, 255)
    
51.     return r, g, b
    
52. 
    
53. 
    
54. def set_speed():
    
55.     global step
    
56.     if not entry.get().isdigit():
    
57.         return
    
58.     step = entry.get()
    
59.     root.destroy()
    
60. 
    
61. 
    
62. pygame.init()
    
63. 
    
64. size = width, height = 640, 480
    
65. bg = (0, 0, 0)
    
66. 
    
67. screen = pygame.display.set_mode(size)
    
68. pygame.display.set_caption("StarMotionSimulator")
    
69. 
    
70. stars = []  # 随机生成天体
    
71. 
    
72. clock = pygame.time.Clock()
    
73. 
    
74. step = 1
    
75. stop_moving = False
    
76. down = False
    
77. star_pos = (0, 0)
    
78. rel = [0, 0]
    
79. running = True
    
80. while running:
    
81.     clock.tick(60)
    
82.     for event in pygame.event.get():
    
83.         if event.type == pygame.QUIT:
    
84.             running = False
    
85.         elif event.type == pygame.KEYDOWN:
    
86.             if event.key == pygame.K_SPACE:
    
87.                 stop_moving = not stop_moving
    
88.                 all_clear_trail()
    
89.             elif event.key == pygame.K_r:
    
90.                 stars.clear()
    
91.             elif event.key == pygame.K_s:
    
92.                 root = Tk()
    
93.                 root.title('SpeedEditor')
    
94.                 root.geometry('100x50')
    
95.                 label = Label(root, text='Speed')
    
96.                 label.pack()
    
97.                 entry = Entry(root)
    
98.                 entry.pack()
    
99.                 submit = Button(root, text='Confirm')
    
100.                 submit['command'] = set_speed
     
101.                 submit.pack()
     
102.                 root.mainloop()
     
103.         elif event.type == pygame.MOUSEBUTTONDOWN:
     
104.             if not down:
     
105.                 star_pos = event.pos
     
106.                 down = True
     
107.         elif event.type == pygame.MOUSEBUTTONUP:
     
108.             down = False
     
109.             star = Star(5, 100, (-rel[0] // 4, -rel[1] // 4), event.pos)
     
110.             stars.append(star)
     
111.             rel.clear()
     
112.             rel.extend((0, 0))
     
113.         elif event.type == pygame.MOUSEMOTION:
     
114.             if down:  # 按下 + 移动 = 拖动
     
115.                 rel[0] += event.rel[0]
     
116.                 rel[1] += event.rel[1]
     
117.         elif event.type == pygame.USEREVENT:
     
118.             all_clear_trail()
     
119. 
     
120.     screen.fill((0, 0, 0))
     
121.     for star in stars:
     
122.         screen.blit(star.image, star.rect)
     
123.     if not stop_moving:
     
124.         for i in range(len(stars)):
     
125.             star = stars[i]
     
126.             for star2 in stars[(i + 1):]:
     
127.                 distance = math.dist(star.rect.center, star2.rect.center)
     
128.                 #### 从这里开始看
     
129.                 if pygame.sprite.collide_circle(star, star2):
     
130.                     star.mass += star2.mass
     
131.                     stars.remove(star2)
     
132.                     continue
     
133.                 #### 就是上面这段代码，一碰撞就报错
     
134.                 if distance < 100:
     
135.                     distance = 100
     
136.                 # 万有引力公式F=GMm/r**2，0.1是为了适当缩小万有引力，否则会出现天体一直在转圈，你可以自己调整数值
     
137.                 f = (G * (star.mass * star2.mass)) / distance ** 2 * 0.1 * step
     
138.                 relative_position = np.subtract(star.rect.center, star2.rect.center)
     
139.                 angle = math.atan2(*relative_position[::-1])
     
140.                 dx = math.cos(angle) * f
     
141.                 dy = math.sin(angle) * f
     
142.                 star.speed = np.subtract(star.speed, [dx, dy])
     
143.                 star2.speed = np.add(star2.speed, [dx, dy])
     
144.             star.speed = [x for x in star.speed]  # 虽然天体是在做匀速直线运动，但不做衰减的话速度会一直叠加导致过快
     
145.             star.move()
     
146.     pygame.display.flip()
     
147. pygame.quit()
     

复制代码

一碰撞就报错
IndexError: index out of range
链接放后面
感谢@鱼cpython学习者

最佳答案

月排行榜 / 总排行榜

鱼cpython学习者

2023-2-12 13:41:13

永远不要尝试一边迭代一个容器一边删除容器里面的东西，我吃过很多次亏
我的建议是把

1. stars.remove(star2)

复制代码

换成

1. star2.mass = 0
   
2. destroyed.append(star2)

复制代码

等到全部弄完再遍历destroyed，删除star

跳转到最佳答案楼层

鱼cpython学习者

永远不要尝试一边迭代一个容器一边删除容器里面的东西，我吃过很多次亏
我的建议是把

1. stars.remove(star2)

复制代码

换成

1. star2.mass = 0
   
2. destroyed.append(star2)

复制代码

等到全部弄完再遍历destroyed，删除star