File: test_dft_energy.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 (80 lines) | stat: -rw-r--r-- 2,553 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
import unittest

import meep as mp

# compute group velocity of a waveguide mode using two different methods
# (1) ratio of Poynting flux to energy density
# (2) via MPB from get-eigenmode-coefficients


class TestDftEnergy(unittest.TestCase):
    def test_dft_energy(self):
        resolution = 20
        cell = mp.Vector3(10, 5)
        geom = [
            mp.Block(size=mp.Vector3(mp.inf, 1, mp.inf), material=mp.Medium(epsilon=12))
        ]
        pml = [mp.PML(1)]
        fsrc = 0.15
        sources = [
            mp.EigenModeSource(
                src=mp.GaussianSource(frequency=fsrc, fwidth=0.2 * fsrc),
                center=mp.Vector3(-3),
                size=mp.Vector3(y=5),
                eig_band=1,
                eig_parity=mp.ODD_Z + mp.EVEN_Y,
                eig_match_freq=True,
            )
        ]

        sim = mp.Simulation(
            resolution=resolution,
            cell_size=cell,
            geometry=geom,
            boundary_layers=pml,
            sources=sources,
            symmetries=[mp.Mirror(direction=mp.Y)],
        )

        flux = sim.add_flux(
            fsrc,
            0,
            1,
            mp.FluxRegion(center=mp.Vector3(3), size=mp.Vector3(y=5)),
            decimation_factor=1,
        )
        energy = sim.add_energy(
            fsrc,
            0,
            1,
            mp.EnergyRegion(center=mp.Vector3(3), size=mp.Vector3(y=5)),
            decimation_factor=1,
        )
        energy_decimated = sim.add_energy(
            fsrc,
            0,
            1,
            mp.EnergyRegion(center=mp.Vector3(3), size=mp.Vector3(y=5)),
            decimation_factor=10,
        )
        sim.run(until_after_sources=100)

        res = sim.get_eigenmode_coefficients(flux, [1], eig_parity=mp.ODD_Z + mp.EVEN_Y)
        mode_vg = res.vgrp[0]
        poynting_flux = mp.get_fluxes(flux)[0]
        e_energy = mp.get_electric_energy(energy)[0]
        ratio_vg = (0.5 * poynting_flux) / e_energy
        m_energy = mp.get_magnetic_energy(energy)[0]
        t_energy = mp.get_total_energy(energy)[0]

        self.assertAlmostEqual(m_energy + e_energy, t_energy)
        self.assertAlmostEqual(ratio_vg, mode_vg, places=3)

        e_energy_decimated = mp.get_electric_energy(energy_decimated)[0]
        m_energy_decimated = mp.get_magnetic_energy(energy_decimated)[0]
        self.assertAlmostEqual(e_energy, e_energy_decimated, places=1)
        self.assertAlmostEqual(m_energy, m_energy_decimated, places=1)


if __name__ == "__main__":
    unittest.main()