mirror of
https://github.com/Findus23/rebound-collisions.git
synced 2024-09-18 14:43:49 +02:00
39 lines
1 KiB
Python
39 lines
1 KiB
Python
import matplotlib.pyplot as plt
|
|
from rebound import SimulationArchive, Particle, Simulation
|
|
|
|
from extradata import ExtraData
|
|
from utils import filename_from_argv, plot_settings, is_ci
|
|
|
|
plot_settings()
|
|
|
|
fn = filename_from_argv()
|
|
sa = SimulationArchive(str(fn.with_suffix(".bin")))
|
|
ed = ExtraData.load(fn)
|
|
print(ed.meta)
|
|
|
|
data = {}
|
|
sim: Simulation
|
|
print(f"{len(sa)} Snapshots found")
|
|
for sim in sa:
|
|
t = sim.t
|
|
for pn in range(1, sim.N):
|
|
part: Particle = sim.particles[pn]
|
|
hash = part.hash.value
|
|
if hash not in data:
|
|
data[hash] = ([], [])
|
|
data[hash][0].append(t)
|
|
data[hash][1].append(part.a)
|
|
|
|
for name, d in data.items():
|
|
times, values = d
|
|
print(list(map(len, [times, values])))
|
|
if False:
|
|
plt.scatter(times, values, label=name, s=.9)
|
|
else:
|
|
plt.plot(times, values, label=name, linewidth=0.6)
|
|
# plt.legend()
|
|
# OrbitPlot(sim, slices=1)
|
|
plt.tight_layout()
|
|
if not is_ci():
|
|
plt.savefig("/home/lukas/tmp/time.pdf", transparent=True)
|
|
plt.show()
|