Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools_public.git synced 2025-05-21 01:51:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add flipping of X and Z

Browse source
This commit is contained in:
rstiskalek 2023-10-19 22:04:50 +01:00
parent 4edee593d0
commit 05650346fc
1 changed files with 17 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

28
csiborgtools/read/readsim.py
View file

@ -30,7 +30,7 @@ from tqdm import trange
from ..utils import fprint from ..utils import fprint
from .paths import Paths from .paths import Paths
from .utils import add_columns, cols_to_structured from .utils import add_columns, cols_to_structured, flip_cols
class BaseReader(ABC): class BaseReader(ABC):
@ -161,12 +161,18 @@ class CSiBORGReader(BaseReader):
def read_snapshot(self, nsnap, nsim, kind): def read_snapshot(self, nsnap, nsim, kind):
sim = pynbody.load(self.paths.snapshot(nsnap, nsim, "csiborg")) sim = pynbody.load(self.paths.snapshot(nsnap, nsim, "csiborg"))
if kind == "pid": if kind == "pid":
return numpy.array(sim["iord"], dtype=numpy.uint64) x = numpy.array(sim["iord"], dtype=numpy.uint64)
elif kind in ["pos", "vel", "mass"]: elif kind in ["pos", "vel", "mass"]:
return numpy.array(sim[kind], dtype=numpy.float32) x = numpy.array(sim[kind], dtype=numpy.float32)
else: else:
raise ValueError(f"Unknown kind `{kind}`.") raise ValueError(f"Unknown kind `{kind}`.")
# Because of a RAMSES bug x and z are flipped.
if x in ["pos", "vel"]:
x[:, [0, 2]] = x[:, [2, 0]]
return x
def read_halo_id(self, nsnap, nsim, halo_finder, verbose=True): def read_halo_id(self, nsnap, nsim, halo_finder, verbose=True):
if halo_finder == "PHEW": if halo_finder == "PHEW":
ids = self.read_phew_id(nsnap, nsim, verbose) ids = self.read_phew_id(nsnap, nsim, verbose)
@ -293,6 +299,8 @@ class CSiBORGReader(BaseReader):
out["x"] = pos[:, 0] * h + 677.7 / 2 out["x"] = pos[:, 0] * h + 677.7 / 2
out["y"] = pos[:, 1] * h + 677.7 / 2 out["y"] = pos[:, 1] * h + 677.7 / 2
out["z"] = pos[:, 2] * h + 677.7 / 2 out["z"] = pos[:, 2] * h + 677.7 / 2
# Because of a RAMSES bug x and z are flipped.
flip_cols(out, ["x", "z"])
out["totpartmass"] = totmass * 1e11 * h out["totpartmass"] = totmass * 1e11 * h
out["m200c"] = m200c * 1e11 * h out["m200c"] = m200c * 1e11 * h
return out return out
@ -350,24 +358,22 @@ class CSiBORGReader(BaseReader):
out['x'] *= 677.7 out['x'] *= 677.7
out['y'] *= 677.7 out['y'] *= 677.7
out['z'] *= 677.7 out['z'] *= 677.7
# Because of a RAMSES bug x and z are flipped.
flip_cols(out, ["x", "z"])
out["mass_cl"] *= 2.6543271649678946e+19 out["mass_cl"] *= 2.6543271649678946e+19
if find_ultimate_parent: if find_ultimate_parent:
ultimate_parent = self.find_parents(out, verbose) ultimate_parent = self.find_parents(out, verbose)
is_main = out["index"] == ultimate_parent
summed_mass = numpy.full(out.size, numpy.nan, dtype=numpy.float32) summed_mass = numpy.full(out.size, numpy.nan, dtype=numpy.float32)
for i in range(out.size): for i in range(out.size):
mask = [ultimate_parent == out["index"][i]] mask = [ultimate_parent == out["index"][i]]
summed_mass[i] = numpy.sum(out["mass_cl"][mask]) summed_mass[i] = numpy.sum(out["mass_cl"][mask])
subfrac = 1 - out["mass_cl"] / summed_mass add_columns(out,
[ultimate_parent, summed_mass],
add_columns( ["ultimate_parent", "summed_mass"]
out, )
[ultimate_parent, is_main, summed_mass, subfrac],
["ultimate_parent", "is_main", "summed_mass", "subfrac"]
)
return out return out