sy_full1 = sy_full1.astype('float32')
sz_full1 = sz_full1.astype('float32')
sx_full_zoom1 = sx_full_zoom1.astype('float32')
sy_full_zoom1 = sy_full_zoom1.astype('float32')
sz_full_zoom1 = sz_full_zoom1.astype('float32')
NPART_zoom = list(sx_full_zoom1.shape)
print "Starting 2LPT on full box."
#Get bounding boxes for full box with 1 refinement level for MUSIC.
## this seems to be purely parameter setting.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, [0,0], gridscale)
print "cellsize", cellsize, cellsize_zoom ## both are zero
print "bbox", BBox_out
print BBox_out_zoom ## same size as BB_out - [0, 256]^3
print "grid", ngrid_x, gridcellsize ### 768 ngrid, and gridcellsize is 1/3. I presume that's because I set gridscale=3. That's not related to anything in Music.
### perhaps I can set these params by hand to give a non-existance zoom?
#BBox_out_zoom = np.array( [[0,0], [0,0], [0,0]], dtype='int32')
#cellsize_zoom = 0.0 ### ic_2lpt needs to have sy2_full_zoom values removed if I do this, as it will try to return 3 values if cellsize_zoom=0
### but, might just be easiest to not evolve the zoom particles?
sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, \
def run(music_file):
import numpy as np
import matplotlib.pyplot as plt
from aux import boundaries
from ic import ic_2lpt, import_music_snapshot
from evolve import evolve
from cic import CICDeposit_3
from potential import initialize_density
# Set up the parameters from the MUSIC ic snapshot:
# Set up according to instructions for
# aux.boundaries()
boxsize = 100.0 # in Mpc/h
level = 7
level_zoom = 8
gridscale = 1
# Set up according to instructions for
# ic.import_music_snapshot()
level0 = '07' # should match level above
level1 = '08' # should match level_zoom above
# Set how much to cut from the sides of the full box.
# This makes the COLA box to be of the following size in Mpc/h:
# (2.**level-(cut_from_sides[0]+cut_from_sides[1]))/2.**level*boxsize
# This is the full box. Set FULL=True in evolve() below
#cut_from_sides=[0,0]# 100Mpc/h.
#
# These are the interesting cases:
#cut_from_sides=[64,64]# 75Mpc/h
#cut_from_sides=[128,128] # 50Mpc/h
# cut_from_sides=[192,192] # 25Mpc/h
cut_from_sides = [192 / 4, 192 / 4] # 25Mpc/h with level/zoom = 7/8
sx_full1, sy_full1, sz_full1, sx_full_zoom1, sy_full_zoom1, \
sz_full_zoom1, offset_from_code \
= import_music_snapshot(music_file, \
boxsize,level0=level0,level1=level1)
sx_full1 = sx_full1.astype('float32')
sy_full1 = sy_full1.astype('float32')
sz_full1 = sz_full1.astype('float32')
sx_full_zoom1 = sx_full_zoom1.astype('float32')
sy_full_zoom1 = sy_full_zoom1.astype('float32')
sz_full_zoom1 = sz_full_zoom1.astype('float32')
NPART_zoom = list(sx_full_zoom1.shape)
print "Starting 2LPT on full box."
#Get bounding boxes for full box with 1 refinement level for MUSIC.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, [0,0], gridscale)
sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, \
sy2_full_zoom1, sz2_full_zoom1 \
= ic_2lpt(
cellsize,
sx_full1 ,
sy_full1 ,
sz_full1 ,
cellsize_zoom=cellsize_zoom,
sx_zoom = sx_full_zoom1,
sy_zoom = sy_full_zoom1,
sz_zoom = sz_full_zoom1,
boxsize=100.00,
ngrid_x_lpt=ngrid_x,ngrid_y_lpt=ngrid_y,ngrid_z_lpt=ngrid_z,
offset_zoom=offset_zoom,BBox_in=BBox_in)
#Get bounding boxes for the COLA box with 1 refinement level for MUSIC.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(
boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, cut_from_sides, gridscale)
# Trim full-box displacement fields down to COLA volume.
sx_full = sx_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sy_full = sy_full1[BBox_out[0,0]:BBox_out[0,1],
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sz_full = sz_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sx_full_zoom = sx_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sy_full_zoom = sy_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sz_full_zoom = sz_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
del sx_full1, sy_full1, sz_full1, sx_full_zoom1, sy_full_zoom1, sz_full_zoom1
sx2_full = sx2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sy2_full = sy2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sz2_full = sz2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sx2_full_zoom = sx2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sy2_full_zoom = sy2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sz2_full_zoom = sz2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
del sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, sy2_full_zoom1, sz2_full_zoom1
print "2LPT on full box is done."
print "Starting COLA!"
px, py, pz, vx, vy, vz, \
px_zoom, py_zoom, pz_zoom, vx_zoom, vy_zoom, vz_zoom \
= evolve(
cellsize,
sx_full, sy_full, sz_full,
sx2_full, sy2_full, sz2_full,
FULL=False,
cellsize_zoom=cellsize_zoom,
sx_full_zoom = sx_full_zoom ,
sy_full_zoom = sy_full_zoom ,
sz_full_zoom = sz_full_zoom ,
sx2_full_zoom = sx2_full_zoom,
sy2_full_zoom = sy2_full_zoom,
sz2_full_zoom = sz2_full_zoom,
offset_zoom=offset_zoom,
BBox_in=BBox_in,
ngrid_x=ngrid_x,
ngrid_y=ngrid_y,
ngrid_z=ngrid_z,
gridcellsize=gridcellsize,
ngrid_x_lpt=ngrid_x,
ngrid_y_lpt=ngrid_y,
ngrid_z_lpt=ngrid_z,
gridcellsize_lpt=gridcellsize,
a_final=1.,
a_initial=1./10.,
n_steps=10,
save_to_file=False, # set this to True to output the snapshot to a file
file_npz_out='tmp.npz',
)
del vx_zoom, vy_zoom, vz_zoom
del vx, vy, vz
print "Making a figure ..."
# grid size for figure array
ngrid = 2 * 128
# physical size of figure array
cutsize = 12.0 #Mpc/h
# offset vector [Mpc/h]:
com = [1.30208333, 1.10677083, 0.944]
com[0] += offset_zoom[0]+cellsize_zoom * \
(BBox_out_zoom[0,1]-BBox_out_zoom[0,0])/2.0-cutsize/2.0
com[1] += offset_zoom[1]+cellsize_zoom * \
(BBox_out_zoom[1,1]-BBox_out_zoom[1,0])/2.0-cutsize/2.0
com[2] += offset_zoom[2]+cellsize_zoom * \
(BBox_out_zoom[2,1]-BBox_out_zoom[2,0])/2.0-cutsize/2.0
density, den_k, den_fft, _ = initialize_density(ngrid, ngrid, ngrid)
density.fill(0.0)
# Lay down fine particles on density array with CiC:
CICDeposit_3(
px_zoom - com[0],
py_zoom - com[1],
pz_zoom - com[2],
px_zoom,
py_zoom,
pz_zoom, #dummies
density,
cellsize_zoom,
cutsize / float(ngrid),
0,
0, # dummy
0, # dummy
np.array([[0, 0], [0, 0], [0, 0]], dtype='int32'),
np.array([0.0, 0.0, 0.0], dtype='float32'),
0)
# Lay down any present crude particles on density array with CiC:
CICDeposit_3(
px - com[0],
py - com[1],
pz - com[2],
px,
py,
pz, #dummies
density,
cellsize,
cutsize / float(ngrid),
0,
0, # dummy
0, # dummy
BBox_in,
np.array([0.0, 0.0, 0.0], dtype='float32'),
0)
# make the figure:
plt.imshow(np.arcsinh(
(density.mean(axis=2)) * np.sinh(1.0) / 10.0)**(1. / 3.),
vmin=0.0,
vmax=1.75,
interpolation='bicubic',
cmap='CMRmap_r')
plt.axis('off')
plt.show()
def runit(infile, outfile, omM):
import numpy as np
import matplotlib.pyplot as plt
from aux import boundaries
from ic import ic_2lpt,import_music_snapshot
from evolve import evolve
from cic import CICDeposit_3
from potential import initialize_density
# Set up the parameters from the MUSIC ic snapshot:
music_file=infile
# Set up according to instructions for
# aux.boundaries()
boxsize=512.0 # in Mpc/h
level=9
level_zoom=9
gridscale=3
# Set up according to instructions for
# ic.import_music_snapshot()
level0='09' # should match level above
level1='09' # should match level_zoom above
# Set how much to cut from the sides of the full box.
# This makes the COLA box to be of the following size in Mpc/h:
# (2.**level-(cut_from_sides[0]+cut_from_sides[1]))/2.**level*boxsize
# This is the full box. Set FULL=True in evolve() below
cut_from_sides=[0,0]# 100Mpc/h.
#
# These are the interesting cases:
#cut_from_sides=[64,64]# 75Mpc/h
#cut_from_sides=[128,128] # 50Mpc/h
#cut_from_sides=[192,192] # 25Mpc/h
### some nonsens eto convert types from float64 to float32 - I think this is the cause of an error later?
sx_full1, sy_full1, sz_full1, sx_full_zoom1, sy_full_zoom1, \
sz_full_zoom1, offset_from_code \
= import_music_snapshot(music_file, \
boxsize,level0=level0,level1=level1)
NPART_zoom=list(sx_full_zoom1.shape)
print "Starting 2LPT on full box."
#Get bounding boxes for full box with 1 refinement level for MUSIC.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, [0,0], gridscale)
sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, \
sy2_full_zoom1, sz2_full_zoom1 \
= ic_2lpt(
cellsize,
sx_full1 ,
sy_full1 ,
sz_full1 ,
cellsize_zoom=cellsize_zoom,
sx_zoom = sx_full_zoom1,
sy_zoom = sy_full_zoom1,
sz_zoom = sz_full_zoom1,
boxsize=boxsize,
ngrid_x_lpt=ngrid_x,ngrid_y_lpt=ngrid_y,ngrid_z_lpt=ngrid_z,
offset_zoom=offset_zoom,BBox_in=BBox_in)
#Get bounding boxes for the COLA box with 1 refinement level for MUSIC.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(
boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, cut_from_sides, gridscale)
# Trim full-box displacement fields down to COLA volume.
sx_full = sx_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sy_full = sy_full1[BBox_out[0,0]:BBox_out[0,1],
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sz_full = sz_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sx_full_zoom = sx_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sy_full_zoom = sy_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sz_full_zoom = sz_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
del sx_full1, sy_full1, sz_full1, sx_full_zoom1, sy_full_zoom1, sz_full_zoom1
sx2_full = sx2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sy2_full = sy2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sz2_full = sz2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sx2_full_zoom = sx2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sy2_full_zoom = sy2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sz2_full_zoom = sz2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
del sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, sy2_full_zoom1, sz2_full_zoom1
print "2LPT on full box is done."
print "Starting COLA!"
print "cellsize:", cellsize
px, py, pz, vx, vy, vz, \
px_zoom, py_zoom, pz_zoom, vx_zoom, vy_zoom, vz_zoom \
= evolve(
cellsize,
sx_full, sy_full, sz_full,
sx2_full, sy2_full, sz2_full,
FULL=True,
cellsize_zoom=cellsize_zoom,
sx_full_zoom = sx_full_zoom ,
sy_full_zoom = sy_full_zoom ,
sz_full_zoom = sz_full_zoom ,
sx2_full_zoom = sx2_full_zoom,
sy2_full_zoom = sy2_full_zoom,
sz2_full_zoom = sz2_full_zoom,
offset_zoom=offset_zoom,
BBox_in=BBox_in,
ngrid_x=ngrid_x,
ngrid_y=ngrid_y,
ngrid_z=ngrid_z,
gridcellsize=gridcellsize,
ngrid_x_lpt=ngrid_x,
ngrid_y_lpt=ngrid_y,
ngrid_z_lpt=ngrid_z,
gridcellsize_lpt=gridcellsize,
Om = float(omM),
Ol = 1.0-float(omM),
a_final=1.,
a_initial=1./10.,
n_steps=10,
save_to_file=True, # set this to True to output the snapshot to a file
file_npz_out=outfile,
)
del vx_zoom,vy_zoom,vz_zoom
del vx,vy,vz
def pycola_evolve(input_file, output_file, box_length, level, gridscale=3) :
# Set up the parameters from the MUSIC ic snapshot:
# music_file="/home/yren/EXA/code/cosmoflow-sims/MUSIC/peter_ics_bl_128.hdf5"
#music_file="/home/yren/EXA/code/cosmoflow-sims/MUSIC/peter_ics_bl128_level3.hdf5"
# Set up according to instructions for
# aux.boundaries()
## boxsize=256.0 # in Mpc/h
## level=8
## level_zoom=8
## gridscale=3
boxsize=float(box_length)
# boxsize=128.0 # in Mpc/h
level=level
level_zoom=level
#gridscale=3
# Set up according to instructions for
# ic.import_music_snapshot()
# level0='08' # should match level above
# level1='08' # should match level_zoom above
# level0='09' # should match level above
# level1='09' # should match level_zoom above
level0="%02d"%(level)
level1="%02d"%(level_zoom)
# Set how much to cut from the sides of the full box.
# This makes the COLA box to be of the following size in Mpc/h:
# (2.**level-(cut_from_sides[0]+cut_from_sides[1]))/2.**level*boxsize
# This is the full box. Set FULL=True in evolve() below
cut_from_sides=[0,0]# 100Mpc/h.
#
# These are the interesting cases:
#cut_from_sides=[64,64]# 75Mpc/h
#cut_from_sides=[128,128] # 50Mpc/h
#cut_from_sides=[192,192] # 25Mpc/h
sx_full1, sy_full1, sz_full1, sx_full_zoom1, sy_full_zoom1, \
sz_full_zoom1, offset_from_code \
= import_music_snapshot(input_file, \
boxsize,level0=level0,level1=level1)
# sx_full1 = sx_full1.astype('float32')
# sy_full1 = sy_full1.astype('float32')
# sz_full1 = sz_full1.astype('float32')
# sx_full_zoom1 = sx_full_zoom1.astype('float32')
# sy_full_zoom1 = sy_full_zoom1.astype('float32')
# sz_full_zoom1 = sz_full_zoom1.astype('float32')
NPART_zoom=list(sx_full_zoom1.shape)
# print "Starting 2LPT on full box."
#Get bounding boxes for full box with 1 refinement level for MUSIC.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, [0,0], gridscale)
sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, \
sy2_full_zoom1, sz2_full_zoom1 \
= ic_2lpt(
cellsize,
sx_full1 ,
sy_full1 ,
sz_full1 ,
cellsize_zoom=cellsize_zoom,
sx_zoom = sx_full_zoom1,
sy_zoom = sy_full_zoom1,
sz_zoom = sz_full_zoom1,
boxsize=boxsize,
ngrid_x_lpt=ngrid_x,ngrid_y_lpt=ngrid_y,ngrid_z_lpt=ngrid_z,
offset_zoom=offset_zoom,BBox_in=BBox_in)
#Get bounding boxes for the COLA box with 1 refinement level for MUSIC.
BBox_in, offset_zoom, cellsize, cellsize_zoom, \
offset_index, BBox_out, BBox_out_zoom, \
ngrid_x, ngrid_y, ngrid_z, gridcellsize \
= boundaries(
boxsize, level, level_zoom, \
NPART_zoom, offset_from_code, cut_from_sides, gridscale)
# Trim full-box displacement fields down to COLA volume.
sx_full = sx_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sy_full = sy_full1[BBox_out[0,0]:BBox_out[0,1],
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sz_full = sz_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sx_full_zoom = sx_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sy_full_zoom = sy_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sz_full_zoom = sz_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
del sx_full1, sy_full1, sz_full1, sx_full_zoom1, sy_full_zoom1, sz_full_zoom1
sx2_full = sx2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sy2_full = sy2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sz2_full = sz2_full1[BBox_out[0,0]:BBox_out[0,1], \
BBox_out[1,0]:BBox_out[1,1], \
BBox_out[2,0]:BBox_out[2,1]]
sx2_full_zoom = sx2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sy2_full_zoom = sy2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
sz2_full_zoom = sz2_full_zoom1[BBox_out_zoom[0,0]:BBox_out_zoom[0,1], \
BBox_out_zoom[1,0]:BBox_out_zoom[1,1], \
BBox_out_zoom[2,0]:BBox_out_zoom[2,1]]
del sx2_full1, sy2_full1, sz2_full1, sx2_full_zoom1, sy2_full_zoom1, sz2_full_zoom1
#print "2LPT on full box is done."
#print "Starting COLA!"
px, py, pz, vx, vy, vz, \
px_zoom, py_zoom, pz_zoom, vx_zoom, vy_zoom, vz_zoom \
= evolve(
cellsize,
sx_full, sy_full, sz_full,
sx2_full, sy2_full, sz2_full,
FULL=True,
cellsize_zoom=cellsize_zoom,
sx_full_zoom = sx_full_zoom ,
sy_full_zoom = sy_full_zoom ,
sz_full_zoom = sz_full_zoom ,
sx2_full_zoom = sx2_full_zoom,
sy2_full_zoom = sy2_full_zoom,
sz2_full_zoom = sz2_full_zoom,
offset_zoom=offset_zoom,
BBox_in=BBox_in,
ngrid_x=ngrid_x,
ngrid_y=ngrid_y,
ngrid_z=ngrid_z,
gridcellsize=gridcellsize,
ngrid_x_lpt=ngrid_x,
ngrid_y_lpt=ngrid_y,
ngrid_z_lpt=ngrid_z,
gridcellsize_lpt=gridcellsize,
a_final=1.,
a_initial=1./10.,
n_steps=10,
save_to_file=True, # set this to True to output the snapshot to a file
file_npz_out=output_file,
)
del vx_zoom,vy_zoom,vz_zoom
del vx,vy,vz