本文旨在帮助开发者解决在使用 Pymunk 结合 Pygame 创建物理模拟时，遇到的刚体位置变为 NaN 的问题。通过分析问题原因，并提供正确的代码示例和必要的补充说明，确保读者能够成功创建并控制 Pymunk 中的刚体，并在 Pygame 窗口中正确显示。本文将重点讲解刚体创建时质量和惯性的设置，以及 Pygame 窗口更新的必要步骤。
在 Pymunk 中，刚体（pymunk.Body）是物理模拟的核心元素。如果刚体的位置显示为 Vec2d(nan, nan)，通常是因为刚体在创建时没有正确设置质量（mass）和惯性（moment）。Pymunk 要求刚体的质量和惯性不能为 0，否则会导致计算错误，从而产生 NaN 值。此外，Pygame 窗口的正确刷新也是保证画面正常显示的关键。
以下是一个修正后的代码示例，展示了如何正确创建刚体并将其显示在 Pygame 窗口中：
import pymunk
import pygame
pygame.init()
# 初始化 Pymunk 空间
space = pymunk.Space()
space.gravity = (0, 100) # 添加重力
# 设置 Pygame 窗口
width, height = 600, 400
display = pygame.display.set_mode((width, height))
pygame.display.set_caption("Pymunk Ball")
# 创建刚体
body = pymunk.Body(1, 10) # 质量为 1，惯性为 10
body.position = 400, 400
# 创建形状
shape = pymunk.Circle(body, 10)
space.add(body, shape)
# 游戏主循环
FPS = 60
clock = pygame.time.Clock()
def main():
running = True
while running:
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
# 清空屏幕
display.fill((255, 255, 255))
# 获取刚体位置
x, y = body.position
pygame.draw.circle(display, (255, 0, 0), (int(x), int(y)), 10)
# 更新 Pygame 窗口
pygame.display.flip()
# 模拟物理世界
space.step(1/FPS)
# 控制帧率
clock.tick(FPS)
pygame.quit()
if __name__ == "__main__":
main()代码解释：
- 刚体创建： body = pymunk.Body(1, 10) 创建了一个质量为 1，惯性为 10 的刚体。务必确保质量和惯性不为 0。惯性的具体数值取决于物体的形状和质量分布，可以根据实际情况调整。
- 重力设置： space.gravity = (0, 100) 设置了重力，使得刚体在垂直方向上受到向下的力。
- 窗口更新： pygame.display.flip() 用于更新 Pygame 窗口，将绘制的内容显示出来。 每次绘制完所有物体后，都需要调用此函数。
注意事项：
- 质量和惯性： 创建 pymunk.Body 时，必须指定一个非零的质量和惯性。 惯性是物体抵抗旋转的能力，对于圆形物体，可以使用 pymunk.moment_for_circle 函数计算。
- Pygame 更新： pygame.display.flip() 是必要的，用于更新 Pygame 窗口。 缺少此步骤会导致画面无法显示或只显示静态内容。
- 空间步进： space.step(1/FPS) 用于模拟物理世界的运动。 FPS 是帧率，1/FPS 表示每次模拟的时间步长。
- 事件处理： 确保正确处理 Pygame 事件，例如退出事件，以避免程序崩溃。
- 调试： 使用 print(body.position) 可以在控制台输出刚体的位置，方便调试。
总结：
解决 Pymunk 刚体位置 NaN 问题的关键在于正确创建刚体，并确保质量和惯性不为 0。同时，正确更新 Pygame 窗口是保证画面正常显示的前提。通过遵循这些步骤，可以成功创建并控制 Pymunk 中的刚体，并在 Pygame 窗口中实现物理模拟效果。
