43 lines
815 B
Python
43 lines
815 B
Python
|
from qpms import *
|
||
|
|
import numpy as np
|
||
|
|
R = 40e-9
|
||
|
|
lMax = 2
|
||
|
|
ω = 1.5*eV/ℏ
|
||
|
|
|
||
|
|
spec = BaseSpec(lMax = lMax)
|
||
|
|
|
||
|
|
inside = EpsMuGenerator(lorentz_drude['Au'])
|
||
|
|
outside = EpsMuGenerator(EpsMu(2.3104,1))
|
||
|
|
|
||
|
|
gensphere_arc = TMatrixGenerator.sphere_asarc(outside=outside, inside=inside, r=R, lMax_extend=lMax)
|
||
|
|
|
||
|
|
np.set_printoptions(precision=3, suppress=True,linewidth=1000)
|
||
|
|
|
||
|
|
QT = gensphere_arc.Q_transposed(ω, spec.norm)
|
||
|
|
RT = gensphere_arc.R_transposed(ω, spec.norm)
|
||
|
|
T = gensphere_arc(spec,ω)
|
||
|
|
|
||
|
|
|
||
|
|
QT_corrected = np.array(QT)
|
||
|
|
RT_corrected = np.array(RT)
|
||
|
|
QT_corrected[8:,:8] = 0
|
||
|
|
QT_corrected[:8,8:] = 0
|
||
|
|
RT_corrected[8:,:8] = 0
|
||
|
|
RT_corrected[:8,8:] = 0
|
||
|
|
T_corrected = np.dot(np.linalg.inv(QT_corrected), RT_corrected)
|
||
|
|
|
||
|
|
print("QT:")
|
||
|
|
print(QT)
|
||
|
|
|
||
|
|
print("RT:")
|
||
|
|
print(RT)
|
||
|
|
|
||
|
|
print("T:")
|
||
|
|
print(T[:])
|
||
|
|
|
||
|
|
print("T_corrected:")
|
||
|
|
print(T_corrected)
|
||
|
|
|
||
|
|
|
||
|
|
|