-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathflare_script.py
101 lines (84 loc) · 3.6 KB
/
flare_script.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
import moment_tests as mt
import Salty2 as salty
import numpy as np
import gc
'''This file is really just a script that allows me to make a bunch of flared
and non-flared models and compare them to data. It is designed to be run in
/usr/users/eigenbrot/research/edgeOn/flares/sims
'''
def do():
h_r = 8.3
hzrlist = [12.45]#np.arange(0.25,1.75,0.25) * h_r
filelist = [
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z0_MgI_binz2.slay.fits',
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z1_MgI_binz2.slay.fits',
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z2_MgI_bin60.slay.fits',
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z4_MgI_binz2.slay.fits']
zlist = [0,0.43,0.86,1.72]
fliplist = [True,False,False,False]
print "hzrlist:"
print hzrlist
vclist = [216]*4#[salty.find_Vc(*salty.openslay(i),back=back)
#for (i,back) in zip(filelist,fliplist)]
salty.plt.close('all')
'''generate some sim files'''
for filename, z, flip, vc in zip(filelist,zlist,fliplist,vclist):
zx = filename.split('_')[2]
noname = 'sim_{}_noflare.fits'.format(zx)
salty.simcurve(1001,z,vc,5.45,scale=100/1001.,output = noname)
noprefix = 'noflare_comp_{}'.format(zx)
# mt.line_comp(noname,filename,noprefix,flip=flip)
for hzr in hzrlist:
flarename = 'sim_{:}_nflarea_{:3.0f}.fits'.format(zx,hzr*100.)
salty.simcurve(1001,z,vc,5.45,scale=100/1001.,
output=flarename,flarepars=dict(h_zR = hzr,
ftype='exp'))
prefix = 'nflare_comp_{:}_f{:04.0f}'.format(zx,hzr*100.)
# mt.line_comp(flarename,filename,prefix,flip=flip)
salty.plt.close('all')
mt.plt.close('all')
salty.gc.collect()
mt.gc.collect()
gc.collect()
return
def do_linear():
'''run in /d/monk/eigenbrot/EdgeOn/flare_test/linear
'''
h_r = 8.3
hzrlist = np.arange(0.1,1.0,0.25) * h_r
filelist = [
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z0_MgI_binz2.slay.fits',
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z1_MgI_binz2.slay.fits',
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z2_MgI_bin60.slay.fits',
'/d/monk/eigenbrot/EdgeOn/SALT_data/tiESO_z4_MgI_binz2.slay.fits']
zlist = [0,0.43,0.86,1.72]
fliplist = [True,False,False,False]
print "hzrlist:"
print hzrlist
vclist = [salty.find_Vc(*salty.openslay(i),back=back)
for (i,back) in zip(filelist,fliplist)]
print "vclist:"
print vclist
salty.plt.close('all')
'''generate some sim files'''
for filename, z, flip, vc in zip(filelist,zlist,fliplist,vclist):
print filename+':'
zx = filename.split('_')[2]
noname = 'sim_{}_noflare.fits'.format(zx)
salty.simcurve(1001,z,vc,5.45,scale=100/1001.,output = noname)
noprefix = 'noflare_comp_{}'.format(zx)
# mt.line_comp(noname,filename,noprefix,flip=flip)
for hzr in hzrlist:
print filename+'\t'+str(hzr)
flarename = 'sim_{:}_lflare_{:03.0f}.fits'.format(zx,hzr*100.)
salty.simcurve(1001,z,vc,5.45,scale=100/1001.,
output=flarename,
flarepars=dict(ftype='linear',h_zR = hzr))
prefix = 'lflare_comp_{:}_f{:04.0f}'.format(zx,hzr*100.)
# mt.line_comp(flarename,filename,prefix,flip=flip)
salty.plt.close('all')
mt.plt.close('all')
salty.gc.collect()
mt.gc.collect()
gc.collect()
return