Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools.git synced 2024-12-22 21:58:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add the maximum snapshot

Browse source
This commit is contained in:
rstiskalek 2022-11-14 16:22:45 +00:00
parent 76349bab3f
commit f33bf0c3b2
2 changed files with 18 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
csiborgtools/io/__init__.py
View file

@ -14,7 +14,7 @@
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
from .readsim import (get_csiborg_ids, get_sim_path, get_snapshots, # noqa
get_snapshot_path, read_info, nparts_to_start_ind, # noqa
get_snapshot_path, get_maximum_snapshot, read_info, nparts_to_start_ind, # noqa
open_particle, open_unbinding, read_particle, # noqa
drop_zero_indx, # noqa
read_clumpid, read_clumps, read_mmain, # noqa

17
csiborgtools/io/readsim.py
View file

@ -106,6 +106,23 @@ def get_snapshots(simpath):
return numpy.sort(snaps)
def get_maximum_snapshot(simpath):
"""
Return the maximum snapshot of an IC realisation stored at `simpath`.
Parameters
----------
simpath : str
Path to the CSiBORG IC realisation.
Returns
-------
maxsnap : float
The maximum snapshot.
"""
return max(get_snapshots(simpath))
def get_snapshot_path(Nsnap, simpath):
"""
Get a path to a CSiBORG IC realisation snapshot.