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

fix mass dtype

Browse source
This commit is contained in:
rstiskalek 2022-10-11 13:34:59 +01:00
parent 075536b839
commit c0947c8a4a
1 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
galomatch/io/readsim.py
View file

@ -6,6 +6,12 @@ from os.path import join
from tqdm import tqdm
F16 = numpy.float16
F32 = numpy.float32
F64 = numpy.float64
INT32 = numpy.int32
def get_sim_path(n, fname="ramses_out_{}", srcdir="/mnt/extraspace/hdesmond"):
"""
Get a path to a CSiBORG simulation.
@ -159,11 +165,13 @@ def read_particle(pars_extract, n, simpath, verbose=True):
"""
# Open the particle files
nparts, partfiles = open_particle(n, simpath)
if verbose:
print("Opened {} particle files.".format(nparts.size))
ncpu = nparts.size
# Order in which the particles are written in the FortranFile
forder = [("x", "float16"), ("y", "float16"), ("z", "float16"),
("vx", "float16"), ("vy", "float16"), ("vz", "float16"),
("M", "float16"), ("ID", "int32"), ("level", "int32")]
forder = [("x", F16), ("y", F16), ("z", F16),
("vx", F16), ("vy", F16), ("vz", F16),
("M", F32), ("ID", INT32), ("level", INT32)]
fnames = [fp[0] for fp in forder]
fdtypes = [fp[1] for fp in forder]
# Check there are no strange parameters
@ -192,3 +200,7 @@ def read_particle(pars_extract, n, simpath, verbose=True):
dum[i:i + j] = read_sp(fdtype, partfiles[cpu])
return out
def read_unbibding():
pass