/
HoleyWaveguide.py
149 lines (123 loc) · 6.25 KB
/
HoleyWaveguide.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
import sys
import argparse
import numpy as np
import meep as mp
from meep.adjoint import (OptimizationProblem, DFTCell, adjoint_options,
xHat, yHat, zHat, origin, FluxLine,
ParameterizedDielectric, FourierLegendreBasis)
##################################################
##################################################
##################################################
class HoleyWaveguide(OptimizationProblem):
##################################################
##################################################
##################################################
def add_args(self, parser):
# add new problem-specific arguments
parser.add_argument('--dair', type=float, default=-1.0, help='')
parser.add_argument('--w_wvg', type=float, default=3.0, help='')
parser.add_argument('--eps_wvg', type=float, default=6.0, help='')
parser.add_argument('--r_disc', type=float, default=0.5, help='')
parser.add_argument('--nr_max', type=int, default=0, help='')
parser.add_argument('--kphi_max', type=int, default=0, help='')
# set problem-specific defaults for existing (general) arguments
parser.set_defaults(fcen=0.5)
parser.set_defaults(df=0.2)
parser.set_defaults(dpml=1.0)
##################################################
##################################################
##################################################
def init_problem(self, args):
#----------------------------------------
# size of computational cell
#----------------------------------------
lcen = 1.0/args.fcen
dpml = 0.5*lcen if args.dpml==-1.0 else args.dpml
dair = 0.5*args.w_wvg if args.dair==-1.0 else args.dair
L = 3.0*lcen
Lmin = 6.0*dpml + 2.0*args.r_disc
L = max(L,Lmin)
sx = dpml+L+dpml
sy = dpml+dair+args.w_wvg+dair+dpml
cell_size = mp.Vector3(sx,sy)
#----------------------------------------
#- design region
#----------------------------------------
design_center = origin
design_size = mp.Vector3(2.0*args.r_disc, 2.0*args.r_disc)
design_region = mp.Volume(center=design_center, size=design_size)
#----------------------------------------
#- objective regions
#----------------------------------------
x0_east = args.r_disc + dpml
x0_west = -args.r_disc - dpml
y0 = 0.0
flux_length = 2.0*args.w_wvg
east = FluxLine(x0_east,y0,flux_length,mp.X,'east')
west = FluxLine(x0_west,y0,flux_length,mp.X,'west')
objective_regions = [east, west]
#----------------------------------------
#- optional extra regions for visualization
#----------------------------------------
extra_regions = [mp.Volume(center=origin, size=cell_size)] if args.full_dfts else []
#----------------------------------------
# basis set
#----------------------------------------
basis = FourierLegendreBasis(radius=args.r_disc, nr_max=args.nr_max, kphi_max=args.kphi_max)
#----------------------------------------
#- source location
#----------------------------------------
source_center = (x0_west - dpml)*xHat
source_size = flux_length*yHat
#----------------------------------------
#- objective function
#----------------------------------------
fstr='Abs(P1_east)**2+0.0*(P2_east+P1_west+P2_west+M1_east+M2_east+M1_west+M2_west+S_east+S_west)'
#----------------------------------------
#- internal storage for variables needed later
#----------------------------------------
self.args = args
self.dpml = dpml
self.cell_size = cell_size
self.basis = basis
self.design_center = design_center
self.source_center = source_center
self.source_size = source_size
return fstr, objective_regions, extra_regions, design_region, basis
##############################################################
##############################################################
##############################################################
def create_sim(self, beta_vector, vacuum=False):
args=self.args
sx=self.cell_size.x
wvg=mp.Block(center=origin, material=mp.Medium(epsilon=args.eps_wvg),
size=mp.Vector3(self.cell_size.x,args.w_wvg))
disc=mp.Cylinder(center=self.design_center, radius=args.r_disc,
epsilon_func=ParameterizedDielectric(self.design_center,
self.basis,
beta_vector))
geometry=[wvg] if vacuum else [wvg, disc]
envelope = mp.GaussianSource(args.fcen,fwidth=args.df)
amp=1.0
if callable(getattr(envelope, "fourier_transform", None)):
amp /= envelope.fourier_transform(args.fcen)
sources=[mp.EigenModeSource(src=envelope,
center=self.source_center,
size=self.source_size,
eig_band=self.args.source_mode,
amplitude=amp
)
]
sim=mp.Simulation(resolution=args.res, cell_size=self.cell_size,
boundary_layers=[mp.PML(args.dpml)], geometry=geometry,
sources=sources)
if args.complex_fields:
sim.force_complex_fields=True
return sim
######################################################################
# if executed as a script, we look at our own filename to figure out
# the name of the class above, create an instance of this class called
# opt_prob, and call its run() method.
######################################################################
if __name__ == '__main__':
opt_prob=globals()[__file__.split('/')[-1].split('.')[0]]()