aart/magneticangle.py at main · iAART/aart · GitHub

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
from aart_func import *
from params import *

print("Magnetic Angle")

fnbands=path+"LensingBands_a_%s_i_%s.h5"%(spin_case,i_case)

print("Reading file: ",fnbands)

h5f = h5py.File(fnbands,'r')

supergrid0=h5f['grid0'][:]
mask0=h5f['mask0'][:]
N0=int(h5f["N0"][0])

supergrid1=h5f['grid1'][:]
mask1=h5f['mask1'][:]
N1=int(h5f["N1"][0])

supergrid2=h5f['grid2'][:]
mask2=h5f['mask2'][:]
N2=int(h5f["N2"][0])

h5f.close()

fnbands=path+"Rays_a_%s_i_%s.h5"%(spin_case,i_case)

print("Reading file: ",fnbands)

h5f = h5py.File(fnbands,'r')

rs0=h5f['rs0'][:]
sign0=h5f['sign0'][:]

rs1=h5f['rs1'][:]
sign1=h5f['sign1'][:]

rs2=h5f['rs2'][:]
sign2=h5f['sign2'][:]

h5f.close()

filename=path+"MagneticAngle_a_%s_i_%s.h5"%(spin_case,i_case)

h5f = h5py.File(filename, 'w')

h5f.create_dataset('cos2angB_n0',data=magneticf.cos2angB_f(supergrid0,mask0,N0,rs0,sign0,spin_case,thetao).reshape(N0,N0).T)
h5f.create_dataset('cos2angB_n1',data=magneticf.cos2angB_f(supergrid1,mask1,N1,rs1,sign1,spin_case,thetao).reshape(N1,N1).T)
h5f.create_dataset('cos2angB_n2',data=magneticf.cos2angB_f(supergrid2,mask2,N2,rs2,sign2,spin_case,thetao).reshape(N2,N2).T)

h5f.close()

print("File ",filename," created.")