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

Add manticore params

Browse source
This commit is contained in:
rstiskalek 2024-09-19 00:49:56 +01:00
parent 4ecc792ebb
commit ebdb95b70b
1 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
csiborgtools/read/paths.py
View file

@ -50,6 +50,8 @@ class Paths:
Path to the CSiBORG2 random simulation directory. Path to the CSiBORG2 random simulation directory.
csiborg2_varysmall_srcdir : str csiborg2_varysmall_srcdir : str
Path to the CSiBORG2 varysmall simulation directory. Path to the CSiBORG2 varysmall simulation directory.
manticore_srcdir : str
Path to the MANTICORE simulation directory.
postdir : str postdir : str
Path to the CSiBORG post-processing directory. Path to the CSiBORG post-processing directory.
quijote_dir : str quijote_dir : str
@ -70,6 +72,7 @@ class Paths:
csiborg2_main_srcdir, csiborg2_main_srcdir,
csiborg2_random_srcdir, csiborg2_random_srcdir,
csiborg2_varysmall_srcdir, csiborg2_varysmall_srcdir,
manticore_dir,
postdir, postdir,
quijote_dir, quijote_dir,
borg1_dir, borg1_dir,
@ -82,6 +85,7 @@ class Paths:
self.csiborg2_main_srcdir = csiborg2_main_srcdir self.csiborg2_main_srcdir = csiborg2_main_srcdir
self.csiborg2_random_srcdir = csiborg2_random_srcdir self.csiborg2_random_srcdir = csiborg2_random_srcdir
self.csiborg2_varysmall_srcdir = csiborg2_varysmall_srcdir self.csiborg2_varysmall_srcdir = csiborg2_varysmall_srcdir
self.manticore_dir = manticore_dir
self.quijote_dir = quijote_dir self.quijote_dir = quijote_dir
self.borg1_dir = borg1_dir self.borg1_dir = borg1_dir
self.borg2_dir = borg2_dir self.borg2_dir = borg2_dir
@ -108,6 +112,12 @@ class Paths:
files = glob(join(self.csiborg2_varysmall_srcdir, "chain_*")) files = glob(join(self.csiborg2_varysmall_srcdir, "chain_*"))
files = [int(search(r'chain_16417_(\d+)', f).group(1)) files = [int(search(r'chain_16417_(\d+)', f).group(1))
for f in files] for f in files]
elif "manticore" in simname:
fdir = self.manticore_dir
if simname == "manticore_2MPP_N128_DES_V1":
fdir = join(fdir, "2MPP_N128_DES_V1", "resimulations", "R512")
files = glob(join(fdir, "mcmc_*"))
files = [int(search(r'mcmc_(\d+)', f).group(1)) for f in files]
elif simname == "csiborg2X": elif simname == "csiborg2X":
# NOTE this too is preliminary # NOTE this too is preliminary
fname = "/mnt/extraspace/rstiskalek/MANTICORE/resimulations/fields/2MPP_N128_DES_PROD/R512" # noqa fname = "/mnt/extraspace/rstiskalek/MANTICORE/resimulations/fields/2MPP_N128_DES_PROD/R512" # noqa
@ -173,6 +183,8 @@ class Paths:
snaps = sorted(snaps) snaps = sorted(snaps)
elif simname == "csiborg2X": elif simname == "csiborg2X":
raise ValueError("Snapshots not available for CSiBORG2X.") raise ValueError("Snapshots not available for CSiBORG2X.")
elif "manticore" in simname:
raise ValueError("Snapshots not available for MANTICORE.")
elif simname == "quijote": elif simname == "quijote":
snaps = glob(join(self.quijote_dir, "fiducial_processed", snaps = glob(join(self.quijote_dir, "fiducial_processed",
f"chain_{nsim}", "snapshot_*")) f"chain_{nsim}", "snapshot_*"))
@ -203,6 +215,8 @@ class Paths:
f"snapshot_{str(nsnap).zfill(3)}.hdf5") f"snapshot_{str(nsnap).zfill(3)}.hdf5")
elif simname == "csiborg2X": elif simname == "csiborg2X":
raise ValueError("Snapshots not available for CSiBORG2X.") raise ValueError("Snapshots not available for CSiBORG2X.")
elif "manticore" in simname:
raise ValueError("Snapshots not available for MANTICORE.")
elif simname == "quijote": elif simname == "quijote":
fpath = join(self.quijote_dir, "fiducial_processed", fpath = join(self.quijote_dir, "fiducial_processed",
f"chain_{nsim}", f"chain_{nsim}",