Commit 43410cda authored by Wouter Kessels's avatar Wouter Kessels
parents 0f016165 3f614fce
......@@ -317,6 +317,7 @@ def dynamics(bodies, D, h, t_max, barnes_hut, theta, approximation, j, startsave
E_kin.append(0.5*mass*(np.linalg.norm(velo, axis=1)**2))
times.append(h*i)
all_pos_corr = np.transpose(np.transpose(all_pos, (1,0,2)) - np.transpose(all_pos, (1,0,2))[0], (1,0,2))
# return
return all_pos, velo, times, E_kin
return all_pos_corr, velo, times, E_kin
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment