File: test_oblique_source.py

package info (click to toggle)
meep-openmpi 1.25.0-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 64,556 kB
  • sloc: cpp: 32,214; python: 27,958; lisp: 1,225; makefile: 505; sh: 249; ansic: 131; javascript: 5
file content (84 lines) | stat: -rw-r--r-- 2,876 bytes parent folder | download | duplicates (5)
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
import math
import unittest

import meep as mp


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(f"flux:, {t:.2f}, {fluxes[-1]:.6f}")
            print(f"coef_flux:, {t:.2f}, {coeff_fluxes[-1]:.6f}")

        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()