-
Notifications
You must be signed in to change notification settings - Fork 587
/
test_oblique_source.py
69 lines (54 loc) · 2.97 KB
/
test_oblique_source.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
import meep as mp
import math
import unittest
class TestEigenmodeSource(unittest.TestCase):
def test_waveguide_flux(self):
cell_size = mp.Vector3(10,10)
pml_layers = [mp.PML(thickness=2.0)]
rot_angles = range(0,60,20) # rotation angle of waveguide, CCW around z-axis
fluxes = []
coeff_fluxes = []
for t in rot_angles:
rot_angle = math.radians(t)
kpoint = mp.Vector3(math.cos(rot_angle),math.sin(rot_angle),0)
sources = [mp.EigenModeSource(src=mp.GaussianSource(1.0,fwidth=0.1),
size=mp.Vector3(y=10),
center=mp.Vector3(x=-3),
direction=mp.NO_DIRECTION,
eig_kpoint=kpoint,
eig_band=1,
eig_parity=mp.ODD_Z,
eig_match_freq=True)]
geometry = [mp.Block(center=mp.Vector3(),
size=mp.Vector3(mp.inf,1,mp.inf),
e1 = mp.Vector3(1).rotate(mp.Vector3(z=1), rot_angle),
e2 = mp.Vector3(y=1).rotate(mp.Vector3(z=1), rot_angle),
material=mp.Medium(index=1.5))]
sim = mp.Simulation(cell_size=cell_size,
resolution=50,
boundary_layers=pml_layers,
sources=sources,
geometry=geometry)
tran = sim.add_flux(1.0, 0, 1, mp.FluxRegion(center=mp.Vector3(x=3), size=mp.Vector3(y=10)))
sim.run(until_after_sources=100)
res = sim.get_eigenmode_coefficients(tran,
[1],
eig_parity=mp.EVEN_Y+mp.ODD_Z if t == 0 else mp.ODD_Z,
direction=mp.NO_DIRECTION,
kpoint_func=lambda f,n: kpoint)
fluxes.append(mp.get_fluxes(tran)[0])
coeff_fluxes.append(abs(res.alpha[0,0,0])**2)
print("flux:, {:.2f}, {:.6f}".format(t,fluxes[-1]))
print("coef_flux:, {:.2f}, {:.6f}".format(t,coeff_fluxes[-1]))
self.assertAlmostEqual(fluxes[0], fluxes[1], places=0)
self.assertAlmostEqual(fluxes[1], fluxes[2], places=0)
for i in range(3):
self.assertAlmostEqual(fluxes[i], coeff_fluxes[i], places=0)
# self.assertAlmostEqual(fluxes[0], fluxes[2], places=0)
# sadly the above line requires a workaround due to the
# following annoying numerical accident:
# AssertionError: 100.33815231783535 != 99.81145343586365 within 0 places
f0,f2=fluxes[0],fluxes[2]
self.assertLess( abs(f0-f2), 0.5*max(abs(f0),abs(f2)) )
if __name__ == '__main__':
unittest.main()