plot.camera_auto_fit = False
plot.grid_auto_fit = False
G = 6.67E-11
lines = []
speeds = []
for i in range(bodies_count):
lines.append([])
speeds.append([])
for t in range(500):
for i in range(bodies_count):
sum_force = np.zeros(3)
for j in range(bodies_count):
if i == j:
continue
direction = bodies[j, 0:3] - bodies[i, 0:3]
force = G * bodies[i, 6] * bodies[j, 6] * direction
force = force / (np.linalg.norm(direction) ** 3)
sum_force = sum_force + force
bodies[i, 3:6] = bodies[i, 3:6] + sum_force / bodies[i, 6]
for i in range(bodies_count):
bodies[i, 0:3] = bodies[i, 0:3] + bodies[i, 3:6] * 0.2
lines[i].append(np.copy(bodies[i, 0:3]))
speeds[i].append(np.linalg.norm(bodies[i, 3:6]))
points.positions = np.copy(bodies[:, 0:3]).astype(np.float32)
for line, speed in zip(lines, speeds):
plot += k3d.line(line, attribute=speed, shader="mesh", width=0.0025,
color_range=[0,0.1], color_map=k3d.basic_color_maps.Jet)