等大球体重力模拟程序过近卡顿问题的解决方法求助

你的程序在球体靠近时卡顿，核心原因是距离过近时引力加速度瞬间爆炸，导致速度突变，进而引发画面异常。以下是具体问题分析和修复方案：

问题分析 #

1. 引力无上限：当两球距离d趋近于0时，引力公式g = (6.674e-11*m*m)/pow(d, 2)会让引力趋近于无穷大，加速度a也随之飙升，速度v短时间内变得异常巨大，导致球体运动失控、画面卡顿。
2. 角度计算精度缺陷：用atan(slope)计算角度，不仅会在水平距离为0时触发除以0错误，还会因斜率四舍五入丢失精度，放大近距离下的运动误差。
3. 变量未初始化：代码中X_1、Y_1、X_2、Y_2无初始值，容错性差，容易引发隐性错误。
4. 速度更新逻辑不稳定：依赖moveEvent事件更新速度，主循环帧率不固定会导致速度更新的时间步长混乱，加剧运动异常。

修复后的完整代码 #

from random import randint
import math
import pygame
import sys

pygame.init()
screen = pygame.display.set_mode((800, 600))
clock = pygame.time.Clock()  # 用于固定帧率和计算时间增量

# 初始化球体初始位置
X_1, Y_1 = 200, 300
X_2, Y_2 = 600, 300

v = 0
# 球体参数
m = 2e15
dia = 20
min_distance = dia  # 最小距离限制，防止引力爆炸

while True:
    delta_time = clock.tick(60) / 1000  # 固定60帧率，转换为秒级时间增量
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()

    # 计算两球的水平、垂直距离
    dx = X_2 - X_1
    dy = Y_2 - Y_1
    d = math.hypot(dx, dy)

    # 避免距离过小导致引力爆炸
    if d < min_distance:
        d = min_distance
        # 可选：添加排斥力防止球体重叠
        # g = (6.674e-11*m*m)/pow(min_distance, 2) * -1
    else:
        # 计算引力和加速度
        g = (6.674e-11 * m * m) / pow(d, 2)
        a = g / m
        # 用时间增量稳定更新速度
        v += a * delta_time

    # 用atan2直接计算角度（返回弧度，无需转换为角度）
    angle = math.atan2(dy, dx)
    v_X = v * math.cos(angle)
    v_Y = v * math.sin(angle)

    # 用时间增量稳定更新位置
    X_1 += v_X * delta_time
    Y_1 += v_Y * delta_time
    X_2 -= v_X * delta_time
    Y_2 -= v_Y * delta_time

    # 绘制画面
    screen.fill((0, 0, 0))
    pygame.draw.circle(screen, (100, 100, 100), (int(X_1), int(Y_1)), dia, 0)
    pygame.draw.circle(screen, (100, 100, 100), (int(X_2), int(Y_2)), dia, 0)
    pygame.display.update()

关键修复点 #

• 最小距离限制：当两球距离小于球体直径时，强制将距离设为直径，避免引力无限大，也可按需添加排斥力防止球体重叠。
• atan2计算角度：直接用水平、垂直距离计算角度，避免斜率计算的除以0问题和精度损失，弧度值可直接用于三角函数计算。
• 固定帧率与时间增量：用clock.tick(60)固定帧率，通过delta_time统一时间步长，确保速度和位置更新的稳定性，避免帧率波动引发的运动异常。
• 初始化球体位置：添加初始位置定义，让代码可独立运行。

内容的提问来源于stack exchange，提问作者Adam Closs