Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools_public.git synced 2025-05-20 17:41:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify print statements

Browse source
This commit is contained in:
rstiskalek 2023-10-18 19:37:45 +01:00
parent ab3ad46128
commit 58b47547e3
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/sort_initsnap.py
View file

@ -74,11 +74,11 @@ def process_snapshot(nsim, simname, halo_finder, verbose):
# Right away we dump the halo IDs to a HDF5 file and clear up memory.
fprint(f"loading PIDs of IC {nsim}.", verbose)
hids = partreader.read_halo_id(nsnap, nsim, halo_finder, verbose)
fprint("sorting PIDs of IC {nsim}.")
fprint(f"sorting PIDs of IC {nsim}.")
sort_indxs = numpy.argsort(hids).astype(numpy.uint64)
# Dump halo IDs
fprint("Loading and dumping halo IDs", verbose)
fprint("loading and dumping halo IDs", verbose)
with h5py.File(fname, "w") as f:
group = f.create_group("snapshot_final")
group.attrs["header"] = "Snapshot data at z = 0."
@ -89,7 +89,7 @@ def process_snapshot(nsim, simname, halo_finder, verbose):
collect()
# Dump particle positions
fprint("Loading and dumping particle positions", verbose)
fprint("loading and dumping particle positions", verbose)
pos = partreader.read_snapshot(nsnap, nsim, "pos")[sort_indxs]
with h5py.File(fname, "r+") as f:
dset = f["snapshot_final"].create_dataset("pos", data=pos)
@ -110,7 +110,7 @@ def process_snapshot(nsim, simname, halo_finder, verbose):
collect()
# Dump masses
fprint("Loading and dumping particle masses", verbose)
fprint("loading and dumping particle masses", verbose)
mass = partreader.read_snapshot(nsnap, nsim, "mass")[sort_indxs]
mass = box.box2solarmass(mass) if simname == "csiborg" else mass
with h5py.File(fname, "r+") as f:
@ -121,7 +121,7 @@ def process_snapshot(nsim, simname, halo_finder, verbose):
collect()
# Dump particle IDs
fprint("Loading and dumping particle IDs", verbose)
fprint("loading and dumping particle IDs", verbose)
pid = partreader.read_snapshot(nsnap, nsim, "pid")[sort_indxs]
with h5py.File(fname, "r+") as f:
dset = f["snapshot_final"].create_dataset("pid", data=pid)