-
Notifications
You must be signed in to change notification settings - Fork 1
/
make_h102054_mr.py
152 lines (134 loc) · 5.27 KB
/
make_h102054_mr.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
#!/usr/bin/env python
# Script to set up a renormalized cluster.
#
# The assumption is that this will be drawn from a uniform volume of 600 Mpc,
# and the high res. region will be XXX 16/h Mpc in size.
# Planck cosmology will be used.
# Waves: Big cube used 768**3 waves
# Renormalized: Ultimate resolution is 768(gas)/1152(dm)^3 in 25Mpc cube.
# Mpc cube, or 1536/2304 in 50Mpc cube. Of the 1536, the top 24**3 waves will
# overlap the 768**3. From 24 to 768 (Nyquist) will be filled by the high res.
# Group 102054 from cosmo600
#
# Med res run: down to 384**3 in 25Mpc
#
import sys, os, glob, string
from math import *
# Cosmology
# Note that these parameters need to be consistent with the CMBFAST data
# file
############################################
#
pboxsize = 600.0 # In Mpc
h = 0.6777
omega = 0.3086
om_lambda = .6914
tilt = 0.9611
gamma = -1 # power spectrum gets read in from CMBFAST
sigma8 = .8288
bias = 1.0/sigma8
omegab = 0.04825
zstart = 139.0
dmgasratio = (omega - omegab)/omegab
gastemp = 1e4
#
############################################
name='h102054mr'
cpart = 'cpartt'
refine = 'refinet'
# markfile = 'vir3.mark'
def mysystem(command):
print command
os.system(command)
def main():
basegrid = 288 # base particle grid (need 64 x to get romulus resolution)
nwaves_base = 768
basegrid3 = basegrid**3
pmass = omega/basegrid3
pboxsize_hinv=pboxsize*h
pboxsize2 = 0.5*pboxsize
basedr=pboxsize/basegrid
epsMpc = 0.00035155674140696647*64 # .351 kpc is romulus softening
baseeps=epsMpc/pboxsize
# make base particle grid
os.system('%s %d %g %g %g %g > %s.g1.bin' % (cpart, basegrid3, pmass, -pboxsize2, pboxsize2, baseeps, name))
#
# parameters for refinement
# refine 5 times for this test run
#
refsize2 = 12. # Radius of inner refine in Mpc
rmin= refsize2
rmax= pboxsize2
rfac= exp(log(rmax/rmin)/5.0)
r_refine=rmax
dr=basedr
r_refine=r_refine/rfac
print 'r_refine', r_refine
os.system('%s %g %g 2 %s.g2.bin < %s.g1.bin' % (refine, dr, r_refine, name, name))
dr= dr*0.5
r_refine=r_refine/rfac
print 'r_refine', r_refine
os.system('%s %g %g 2 %s.g3.bin < %s.g2.bin' % (refine, dr, r_refine, name, name))
r_refine=r_refine/rfac
dr= dr*0.5
print 'r_refine', r_refine
os.system('%s %g %g 2 %s.g4.bin < %s.g3.bin' % (refine, dr, r_refine, name, name))
r_refine=r_refine/rfac
dr= dr*0.5
print 'r_refine', r_refine
os.system('%s %g %g 2 %s.g5.bin < %s.g4.bin' % (refine, dr, r_refine, name, name))
r_refine=r_refine/rfac
dr= dr*0.5
print 'r_refine', r_refine
os.system('%s %g %g 2 %s.g6.bin < %s.g5.bin' % (refine, dr, r_refine, name, name))
# Following is for high res.
# r_refine=r_refine/rfac
# dr= dr*0.5
# print 'r_refine', r_refine
# os.system('%s %g %g 2 1 %g %g %s %s.g6.bin n%s < %s.g5.bin'
# % ('gasrefmarks', dr, r_refine, gastemp, dmgasratio, markfile,
# name, markfile, name))
# make waves
if 0 :
mysystem('echo %d %s 1.0 %s 1. 0.0 -7678301 %s %s %s | kgen_mt > out1' % (nwaves_base, pboxsize_hinv, h, omega, tilt, gamma))
mysystem('invfftr< out1 > out2')
mysystem('fft <out2 > out3')
kmax = nwaves_base*6.28*sqrt(3.0)/pboxsize
mysystem('powk 1000 %g < %s.rfft > %s.pow' % (kmax, name, name))
mysystem('rhotophik < out3 > base.phfft')
mysystem('gradrhok x < base.phfft > gr_tmp.fft; invfftr < gr_tmp.fft > base.fx' % (name, ))
mysystem('gradrhok y < base.phfft > gr_tmp.fft; invfftr < gr_tmp.fft > base.fy' % (name, ))
mysystem('gradrhok z < base.phfft > gr_tmp.fft; invfftr < gr_tmp.fft > base.fz' % (name, ))
# make high res waves
hrboxsize = 50.
hrboxsize_hinv = hrboxsize*h
nwaves_hr = 1536
# for low res only: "unpad" the fft to this number
nwaves_keep = 768
# mysystem('echo %d %g 1.0 %g 1.0 0.0 -50 %g %g %g | kgen_mt > hr.fft0'
# % (nwaves_hr, hrboxsize_hinv, h, omega, tilt, gamma))
# mysystem('export OMP_NUM_THREADS=4; invfftr < hr.fft0 > hr.rg')
# mysystem('fft < hr.rg > hr.rfft')
kmax = nwaves_hr*6.28*sqrt(3.0)/hrboxsize
# mysystem('powk 1000 %g < hr.rfft > hr.pow' % (kmax))
# zero out waves up to the Nyquist of the LR run
# mysystem('speczero 0 %g < hr.rfft > hrz.fft'
# % ((hrboxsize_hinv/pboxsize_hinv)*nwaves_base/2))
# mysystem('powk 1000 %g < hrz.fft > hrz.pow' % (kmax))
# mysystem('unpad %d < hrz.fft > hrz_lr.fft'
# % (nwaves_keep,))
# mysystem('rhotophik < hrz_lr.fft > hr_lr.phfft')
# mysystem('gradrhok x < hr_lr.phfft | invfftr > hr_lr.fx')
# mysystem('gradrhok y < hr_lr.phfft | invfftr > hr_lr.fy')
# mysystem('gradrhok z < hr_lr.phfft | invfftr > hr_lr.fz')
# Center of group (in real Mpc)
mx = -79.6875
my = 191.40625
mz = 284.375
mysystem('pmovefrgbg2t %s.g6.bin base.fx base.fy base.fz hr_lr.fx hr_lr.fy hr_lr.fz %g %g %g %g %g %g %g %g > %s.sbin'
% (name, zstart, bias, omega, om_lambda, 0.0, mx, my, mz, name))
# Unit conversion from Mpc, 10^15 M_sun, Gyr to natural units
tscale=1.0/sqrt(4.498*h*h*.0002776)
mysystem('scalet %g %g < %s.sbin > %s.tbin' % (pboxsize, tscale, name, name))
if __name__ == '__main__':
main()