mirror/rebound-collisions
1
0
Fork 0
mirror of https://github.com/Findus23/rebound-collisions.git synced 2024-09-18 14:43:49 +02:00
Code Releases Activity

improve visualize_collision.py

Browse source
This commit is contained in:
Lukas Winkler 2022-01-30 19:52:11 +01:00
parent 51d5aac68d
commit 5201b83da9
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
visualize_collision.py
View file

@ -47,12 +47,13 @@ for collision in ed.tree.get_tree().values():
for pos, vel, r in zip(meta.collision_positions, meta.collision_velocities, meta.collision_radii):
print(pos, vel, r)
circle_coords = get_circle(*pos, r)
ax.plot_wireframe(*circle_coords, color=f"C{i}")
ax.plot_wireframe(*circle_coords, color=f"C{i}", linewidths=.75)
a = Arrow3D(*[(p, p + v / 100000) for p, v in zip(pos, vel)], mutation_scale=20,
lw=1, arrowstyle="-|>", color="k")
ax.add_artist(a)
i += 1
ax.set_title(f"angle={meta.input.alpha:.2f}, v/v_esc={meta.input.velocity_esc:.2f}")
print("title", f"angle={meta.input.alpha:.2f}, v/v_esc={meta.input.velocity_esc:.2f}")
# ax.set_title(f"angle={meta.input.alpha:.2f}, v/v_esc={meta.input.velocity_esc:.2f}")
xyzlim = np.array([ax.get_xlim3d(), ax.get_ylim3d(), ax.get_zlim3d()]).T
diff = min(xyzlim[0] - xyzlim[1])
@ -66,6 +67,10 @@ for collision in ed.tree.get_tree().values():
ax.set_zlim3d((zmin, zmin - diff))
# ax.set_ylim3d(XYZlim)
# ax.set_zlim3d(XYZlim * 3/4)
ax.set_zticklabels([])
ax.set_yticklabels([])
ax.set_xticklabels([])
fig.tight_layout()
if not is_ci():
plt.savefig("/home/lukas/tmp/3d.pdf")
plt.show()