File: material_grid.py

package info (click to toggle)
meep-mpi-default 1.17.1-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 51,672 kB
  • sloc: cpp: 29,881; python: 17,210; lisp: 1,225; makefile: 477; sh: 249; ansic: 133; javascript: 5
file content (63 lines) | stat: -rw-r--r-- 2,917 bytes parent folder | download | duplicates (3)
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
import meep as mp
import numpy as np
from meep.materials import Si, SiO2
import unittest

class TestMaterialGrid(unittest.TestCase):
    def gen_sim(self):
        resolution = 10 # pixels/um
        cell_size = mp.Vector3(14,14)
        pml_layers = [mp.PML(thickness=2)]
        w = 3.0 # width of waveguide
        m1 = SiO2
        m2 = Si
        n = 10
        gs = mp.Vector3(n,n)
        np.random.seed(1)
        dp = np.random.rand(n*n)
        mg = mp.MaterialGrid(gs,m1,m2,dp,"U_SUM")
        geometry = []
        rot_angle = np.radians(0)
        geometry += [mp.Block(center=mp.Vector3(),
                            size=mp.Vector3(w,w,mp.inf),
                            e1=mp.Vector3(x=1).rotate(mp.Vector3(z=1), rot_angle),
                            e2=mp.Vector3(y=1).rotate(mp.Vector3(z=1), rot_angle),
                            material=mg)]
        geometry += [mp.Block(center=mp.Vector3(),
                            size=mp.Vector3(w,w,mp.inf),
                            e1=mp.Vector3(x=-1).rotate(mp.Vector3(z=1), np.pi/2),
                            e2=mp.Vector3(y=1).rotate(mp.Vector3(z=1), np.pi/2),
                            material=mg)]
        fsrc = 1/1.55 # frequency of eigenmode or constant-amplitude source
        bnum = 1    # band number of eigenmode
        kpoint = mp.Vector3(x=1).rotate(mp.Vector3(z=1), rot_angle)
        compute_flux = True # compute flux (True) or plot the field profile (False)
        eig_src = True # eigenmode (True) or constant-amplitude (False) source
        if eig_src:
            sources = [mp.EigenModeSource(src=mp.GaussianSource(fsrc,fwidth=0.2*fsrc) if compute_flux else mp.ContinuousSource(fsrc),
                                        center=mp.Vector3(),
                                        size=mp.Vector3(y=3*w),
                                        direction=mp.NO_DIRECTION,
                                        eig_kpoint=kpoint,
                                        eig_band=bnum,
                                        eig_parity=mp.EVEN_Y+mp.ODD_Z if rot_angle == 0 else mp.ODD_Z,
                                        eig_match_freq=True)]
        else:
            sources = [mp.Source(src=mp.GaussianSource(fsrc,fwidth=0.2*fsrc) if compute_flux else mp.ContinuousSource(fsrc),
                                center=mp.Vector3(),
                                size=mp.Vector3(y=3*w),
                                component=mp.Ez)]

        sim = mp.Simulation(cell_size=cell_size,
                            resolution=resolution,
                            boundary_layers=pml_layers,
                            sources=sources,
                            extra_materials = [m1,m2],
                            geometry=geometry
                            )
        return sim
    
    def test_eval(self):
        sim = self.gen_sim()
if __name__ == '__main__':
    unittest.main()