mirror/rebound-collisions
1
0
Fork 0
mirror of https://github.com/Findus23/rebound-collisions.git synced 2024-09-19 15:53:48 +02:00
Code Releases Activity

don't save plots in CI

Browse source
This commit is contained in:
Lukas Winkler 2021-05-03 16:31:30 +02:00
parent 7c7d9198b3
commit b50e894280
Signed by: lukas
GPG key ID: 54DE4D798D244853
5 changed files with 18 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
analyze.py
View file

@ -34,5 +34,6 @@ for name, d in data.items():
# plt.legend()
# OrbitPlot(sim, slices=1)
plt.tight_layout()
plt.savefig("/home/lukas/tmp/time.pdf", transparent=True)
if not is_ci():
plt.savefig("/home/lukas/tmp/time.pdf", transparent=True)
plt.show()

5
collisionhistory.py
View file

@ -4,7 +4,7 @@ from matplotlib.figure import Figure
from rebound import SimulationArchive, Simulation
from extradata import ExtraData, CollisionMeta
from utils import filename_from_argv, earth_mass, earth_water_mass, plot_settings
from utils import filename_from_argv, earth_mass, earth_water_mass, plot_settings, is_ci
plot_settings()
@ -72,5 +72,6 @@ for particle in last_sim.particles:
twin_ax.set_yscale("log")
ax_wmfs.axhline(earth_water_mass/earth_mass,linestyle="dotted")
fig.tight_layout()
fig.savefig("/home/lukas/tmp/collisionhistory.pdf", transparent=True)
if not is_ci():
fig.savefig("/home/lukas/tmp/collisionhistory.pdf", transparent=True)
plt.show()

7
particle_numbers.py
View file

@ -6,7 +6,7 @@ from matplotlib.figure import Figure
from rebound import SimulationArchive, Simulation
from extradata import ExtraData
from utils import filename_from_argv, plot_settings
from utils import filename_from_argv, plot_settings, is_ci
plot_settings()
@ -32,12 +32,13 @@ for file in argv[1:]:
Ns.append(N)
ts.append(sim.t)
perfect_merging = "pm" in str(fn)
ax.step(ts, Ns, label=fn, where="post", linestyle="dashed" if perfect_merging else "solid",linewidth=0.7)
ax.step(ts, Ns, label=fn, where="post", linestyle="dashed" if perfect_merging else "solid", linewidth=0.7)
ax.set_xlabel("time [yr]")
ax.set_ylabel("number of objects")
# ax.set_xscale("log")
plt.legend()
plt.tight_layout()
plt.savefig("/home/lukas/tmp/particle_numbers.pdf", transparent=True)
if not is_ci():
plt.savefig("/home/lukas/tmp/particle_numbers.pdf", transparent=True)
plt.show()

7
utils/os_tools.py
View file

@ -1,11 +1,12 @@
import os
import socket
import subprocess
from pathlib import Path
import socket
from sys import argv
from setproctitle import setproctitle
def filename_from_argv(argument: str = None) -> Path:
if len(argv) < 2:
raise ValueError("specify filename")
@ -37,3 +38,7 @@ def process_friendlyness(fn: Path) -> None:
return
setproctitle(f"[{fn.stem}] [rebound-watersim] read /home/winklerl23/sim-info.txt for more information")
os.nice(5)
def is_ci() -> bool:
return "CI" in os.environ

5
visualize_collision.py
View file

@ -5,7 +5,7 @@ from matplotlib.patches import FancyArrowPatch
from mpl_toolkits.mplot3d import Axes3D, proj3d
from extradata import ExtraData, CollisionMeta
from utils import filename_from_argv, plot_settings
from utils import filename_from_argv, plot_settings, is_ci
fn = filename_from_argv()
ed = ExtraData.load(fn)
@ -66,5 +66,6 @@ for collision in ed.tree.get_tree().values():
ax.set_zlim3d((zmin, zmin - diff))
# ax.set_ylim3d(XYZlim)
# ax.set_zlim3d(XYZlim * 3/4)
plt.savefig("/home/lukas/tmp/3d.pdf")
if not is_ci():
plt.savefig("/home/lukas/tmp/3d.pdf")
plt.show()