guilhem_lavaux/cosmotool
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixes for gadget IO

Browse source
This commit is contained in:
Guilhem Lavaux 2020-08-04 10:43:45 +02:00
parent 689cc70daa
commit 46831ba9ab
4 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
python/_cosmotool.pyx
View file

@ -513,7 +513,8 @@ def writeGadget(str filename, object simulation):
simdata.TotalNumPart = NumPart
simdata.NumPart = NumPart
cxx_writeGadget(filename, &simdata)
filename_b = bytes(filename, 'utf-8')
cxx_writeGadget(filename_b, &simdata)
def loadRamses(str basepath, int snapshot_id, int cpu_id, bool doublePrecision = False, bool loadPosition = True, bool loadVelocity = False, bool loadId = False, bool loadMass = False):
""" loadRamses(basepath, snapshot_id, cpu_id, doublePrecision = False, loadPosition = True, loadVelocity = False)

4
python/cosmotool/simu.py
View file

@ -98,12 +98,12 @@ def simpleWriteGadget(filename, positions, boxsize=1.0, Hubble=100, Omega_M=0.30
s.positions = positions
if velocities:
if velocities is not None:
s.velocities = velocities
else:
s.velocities = [np.zeros(positions[0].size,dtype=np.float32)]*3
if identifiers:
if identifiers is not None:
s.identifiers = identifiers
else:
s.identifiers = np.arange(positions[0].size, dtype=np.int64)