我最近一直在使用这个公式作出轨迹模拟器:pygame的轨道模拟问题
这里是我的代码:
import pygame, math
from pygame.locals import *
from random import randint
pygame.init()
screen = pygame.display.set_mode([500,500])
clock = pygame.time.Clock()
class Planet():
def __init__(self, vel = [1, 1], mass = 100000, pos = [100, 100], pathLength = 100000):
self.v = vel
self.m = mass
self.size = mass/1000000
self.pos = pos
self.pL = pathLength
self.path = [[pos[0], pos[1]]]
def update(self):
self.pos[0] += self.v[0]
self.pos[1] += self.v[1]
self.path.append([self.pos[0], self.pos[1]])
if len(self.path) == self.pL:
self.path.pop(0)
class World():
def __init__(self, planetList, iterations, mass = 10000000, gravityConstant = (6 * 10 ** -9)):
self.plnt = planetList
self.iter = iterations
self.mass = mass
self.size = int(mass/1000000)
self.gC = gravityConstant
def draw(self):
pygame.draw.circle(screen, [0, 0, 0], [250, 250], self.size)
for p in self.plnt:
pygame.draw.rect(screen, [0, 0, 0], [p.pos[0], p.pos[1], p.size, p.size])
pygame.draw.lines(screen, [0, 0, 0], False, p.path)
def update(self):
for i in range(self.iter):
for p in self.plnt:
d = math.sqrt((p.pos[0] - 250) ** 2 + (p.pos[1] - 250) ** 2)
f = (self.gC * self.mass * p.m)/(d ** 2)
vect = [((250 - p.pos[0])/d) * f, ((250 - p.pos[1])/d) * f]
p.v[0] += vect[0]
p.v[1] += vect[1]
p.update()
self.draw()
a = Planet([4,0])
b = Planet([4, 0])
w = World([b], 100)
while 1:
screen.fill([255, 255, 255])
w.update()
for event in pygame.event.get():
if event.type == QUIT:
pygame.quit()
pygame.display.update()
clock.tick(60)
,如果我只是在它按预期工作模拟1个星球,但与此有问题
a = Planet([4,0])
b = Planet([4, 0])
w = World([a, b], 100)
行星飞了出去,屏幕和永远继续下去,我不能看到我犯了一个错误。
与您的特定问题无关您的实施并不完全是如何引力的作品。你有一个中心“世界”对“行星”施加重力,而不受自身的影响。而你的“行星”根本不会互相影响。实际上,所有物体都以成对的方式彼此施加重力,并且如果A在B上施加力F,则从A施加在B上的力是-F。所以技术上你需要一个嵌套循环并计算所有成对关系一次(如上所述,相反的方向是多余的)。确保正确选择符号,因为方程式不提供它们。 –
@MartinKrämer是的,这是我想要实现这一点的方法,但是如果在中央机构和其他机构之间施加力量,我可以轻易改变。 – adammoyle
[尝试使用IDE](http://sopython.com/wiki/Python_IDEs)。它将帮助你在编写更好的代码和学习编程方面都有极大的帮助。我个人使用PyCharm,它会通知您默认的可变参数,但其他IDE可能具有类似的功能。 –