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

White phase exporter (for mpgrafic e.g.)

Browse source
This commit is contained in:
Guilhem Lavaux 2014-06-02 10:41:39 +02:00
parent 1aaf5f4a13
commit ddb66ab436
3 changed files with 17 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
python/cosmotool/__init__.py
View file

@ -1,5 +1,5 @@
from _cosmotool import *
from grafic import writeGrafic
from grafic import writeGrafic, writeWhitePhase
from borg import read_borg_vol

15
python/cosmotool/grafic.py
View file

@ -22,3 +22,18 @@ def writeGrafic(filename, field, BoxSize, scalefac, **cosmo):
f.write(field[i].astype(np.float32).tostring())
f.write(struct.pack("I", checkPoint))
def writeWhitePhase(filename, field):
with file(filename, mode="wb") as f:
Nx,Ny,Nz = field.shape
checkPoint = 4*4
f.write(struct.pack("IIIIII", checkPoint, Nx, Ny, Nz, 0, checkPoint))
checkPoint = struct.pack("I", 4*Ny*Nz)
for i in xrange(Nx):
f.write(checkPoint)
f.write(field[i].astype(np.float32).tostring())
f.write(checkPoint)