import Pk_library as PKL ################################## INPUT ###################################### snapshot_fname = ['../ics', '../snapdir_000/snap_000', '../snapdir_001/snap_001', '../snapdir_002/snap_002', '../snapdir_003/snap_003'] dims = 1024 particle_type = [1,2] #list with particle types. [-1] for total matter cpus = 14 ############################################################################### # do a loop over the different snapshots for snapshot in snapshot_fname: ######## REAL-SPACE ######## do_RSD = False; axis = 0 PKL.Pk_Gadget(snapshot,dims,particle_type,do_RSD,axis,cpus) ###### REDSHIFT-SPACE ###### do_RSD = True for axis in [0,1,2]: PKL.Pk_Gadget(snapshot,dims,particle_type,do_RSD,axis,cpus)
import numpy as np import Pk_library as PKL ####################################### INPUT ######################################### # parameters root = '/mnt/ceph/users/fvillaescusa/Euclid/Euclid_neutrino_comparison/Sims/0.15eV' grid = 1024 #grid size particle_type = [1, 2] #use dark matter [1] do_RSD = False #move particles to redshift-space and calculate Pk in redshift-space axis = 1 #RSD placed along the y-axis cpus = 8 #number of openmp threads folder_out = './' #folder where to write results ####################################################################################### # do a loop over the different snapshots for snapnum in [1, 2, 3, 4]: # snapshot name snapshot = '%s/snap_%03d.hdf5' % (root, snapnum) # compute power spectrum of the snapshot PKL.Pk_Gadget(snapshot, grid, particle_type, do_RSD, axis, cpus, folder_out)