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

Added pixel shifting in ICgen

Browse source
This commit is contained in:
Guilhem Lavaux 2014-06-04 09:13:34 +02:00
parent a82688b0b1
commit 7258dd20cb
4 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

14
python_sample/icgen/test_ic_from_borg.py
View file

@ -1,25 +1,27 @@
import numpy as np
import cosmotool as ct
import borgicgen as bic
import sys
cosmo={'omega_M_0':0.3175, 'h':0.6711}
cosmo['omega_lambda_0']=1-cosmo['omega_M_0']
cosmo['omega_k_0'] = 0
cosmo['omega_B_0']=0.049
cosmo['SIGMA8']=0.8344
N0=128
snap_id=11
snap_id=int(sys.argv[1])
s = ct.loadRamsesAll("/nethome/lavaux/remote2/borgsim/", snap_id, doublePrecision=True)
astart=s.getTime()
pos,_,density,N,L,_ = bic.run_generation("initial_condition_borg.dat", 0.001, astart, cosmo, supersample=1, do_lpt2=True)
pos,_,density,N,L,_ = bic.run_generation("initial_condition_borg.dat", 0.001, astart, cosmo, supersample=2, do_lpt2=True)
dcic = ct.cicParticles(pos, L, N)
dcic = ct.cicParticles(pos, L, N0)
dcic /= np.average(np.average(np.average(dcic, axis=0), axis=0), axis=0)
dcic -= 1
dsim = ct.cicParticles(s.getPositions(), L, N)
dsim = ct.cicParticles(s.getPositions(), L, N0)
dsim /= np.average(np.average(np.average(dsim, axis=0), axis=0), axis=0)
dsim -= 1
@ -27,8 +29,8 @@ dsim -= 1
dcic_hat = np.fft.rfftn(dcic)*(L/N)**3
dsim_hat = np.fft.rfftn(dsim)*(L/N)**3
Pcic, bcic = bic.bin_power(np.abs(dcic_hat)**2/L**3, L, bins=50)
Psim, bsim = bic.bin_power(np.abs(dsim_hat)**2/L**3, L, bins=50)
Pcic, bcic = bic.bin_power(np.abs(dcic_hat)**2/L**3, L, range=(0,1.), bins=150)
Psim, bsim = bic.bin_power(np.abs(dsim_hat)**2/L**3, L, range=(0,1.), bins=150)
borg_evolved = ct.read_borg_vol("final_density_1380.dat")