Skip to content
Snippets Groups Projects
Commit a103925a authored by Guillaume Demesy's avatar Guillaume Demesy
Browse files

tweaks

parent 5105c833
No related branches found
No related tags found
No related merge requests found
Pipeline #10145 passed
Stage: test
Hide whitespace changes
Inline Side-by-side
ElectromagneticScattering/scattererTmatrix_post.py
+ 5
− 5
View file @ a103925a
import numpy as np import numpy as np
np.set_printoptions(precision=2) import sys
def load_getdp_integral(filename): def load_getdp_integral(filename):
return np.loadtxt(filename)[1]+1j*np.loadtxt(filename)[2] return np.loadtxt(filename)[1]+1j*np.loadtxt(filename)[2]
p_max = 3 n_max = int(sys.argv[1])
p_max = n_max*n_max+2*n_max
Tmatrix = np.zeros((2*p_max,2*p_max),dtype=complex) Tmatrix = np.zeros((2*p_max,2*p_max),dtype=complex)
for k1 in range(2*p_max): for k1 in range(2*p_max):
...@@ -18,6 +20,4 @@ for k1 in range(2*p_max): ...@@ -18,6 +20,4 @@ for k1 in range(2*p_max):
if k1>=p_max and k2>=p_max: if k1>=p_max and k2>=p_max:
Tmatrix[k1,k2] = load_getdp_integral('resT/feN_pe%gpo%g.dat'%(k1+1-p_max,k2+1-p_max)) Tmatrix[k1,k2] = load_getdp_integral('resT/feN_pe%gpo%g.dat'%(k1+1-p_max,k2+1-p_max))
print('|2*Tmatrix+1|\n',np.abs(2*Tmatrix+1)) np.save('Tmatrix.npy',Tmatrix)
print('Re Tmatrix\n',Tmatrix.real)
print('Im Tmatrix\n',Tmatrix.imag)
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment