Last active
December 19, 2021 17:11
-
-
Save oskooi/429e94e4c90b43c07f5fa46357b9f65a to your computer and use it in GitHub Desktop.
convergence test for adjoint gradient
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import numpy as np | |
from autograd import numpy as npa | |
from autograd import tensor_jacobian_product | |
import meep as mp | |
import meep.adjoint as mpa | |
silicon = mp.Medium(epsilon=12.) | |
sxy = 5.0 | |
cell_size = mp.Vector3(sxy,sxy,0) | |
dpml = 1.0 | |
pml = [mp.PML(thickness=dpml,direction=mp.X)] | |
eig_parity = mp.EVEN_Y + mp.ODD_Z | |
design_region_size = mp.Vector3(1.5,1.5) | |
design_region_resolution = 1280. | |
Nx = int(design_region_resolution*design_region_size.x) + 1 | |
Ny = int(design_region_resolution*design_region_size.y) + 1 | |
fcen = 1/1.55 | |
df = 0.23*fcen | |
source = [mp.Source(src=mp.GaussianSource(fcen,fwidth=df,is_integrated=True), | |
center=mp.Vector3(-0.5*sxy+dpml,0), | |
size=mp.Vector3(0,sxy), | |
component=mp.Ez)] | |
x = np.linspace(-0.5*design_region_size.x,0.5*design_region_size.x,Nx) | |
y = np.linspace(-0.5*design_region_size.y,0.5*design_region_size.y,Ny) | |
xv, yv = np.meshgrid(x,y) | |
rad = 0.538948295 | |
wdt = 0.194838432 | |
weights = np.where(np.logical_and(np.sqrt(np.square(xv) + np.square(yv)) > rad, | |
np.sqrt(np.square(xv) + np.square(yv)) < rad+wdt), | |
1., | |
0.) | |
filter_radius = 32/design_region_resolution | |
def mapping(x): | |
filtered_weights = mpa.conic_filter(x, | |
filter_radius, | |
design_region_size.x, | |
design_region_size.y, | |
design_region_resolution) | |
return filtered_weights.flatten() | |
ring_weights = mapping(weights) | |
def adjoint_solver(resolution): | |
matgrid = mp.MaterialGrid(mp.Vector3(Nx,Ny), | |
mp.air, | |
silicon, | |
weights=np.ones((Nx,Ny)), | |
do_averaging=True, | |
beta=np.inf) | |
matgrid_region = mpa.DesignRegion(matgrid, | |
volume=mp.Volume(center=mp.Vector3(), | |
size=mp.Vector3(design_region_size.x, | |
design_region_size.y, | |
0))) | |
geometry = [mp.Block(center=matgrid_region.center, | |
size=matgrid_region.size, | |
material=matgrid)] | |
sim = mp.Simulation(resolution=resolution, | |
cell_size=cell_size, | |
boundary_layers=pml, | |
k_point=mp.Vector3(), | |
sources=source, | |
geometry=geometry) | |
obj_list = [mpa.FourierFields(sim, | |
mp.Volume(center=mp.Vector3(1.25), | |
size=mp.Vector3()), | |
mp.Ez)] | |
def J(mode_mon): | |
return npa.power(npa.abs(mode_mon),2) | |
opt = mpa.OptimizationProblem( | |
simulation=sim, | |
objective_functions=J, | |
objective_arguments=obj_list, | |
design_regions=[matgrid_region], | |
maximum_run_time=100., | |
frequencies=[fcen]) | |
f, dJ_du = opt([ring_weights]) | |
bp_dJ_du = tensor_jacobian_product(mapping,0)(weights,dJ_du) | |
sim.reset_meep() | |
return f, bp_dJ_du | |
for r in [20., 40., 80., 160., 320., 640.]: | |
adjsol_obj, adjsol_grad = adjoint_solver(r) | |
print("grad:, {}, {}, {}".format(r,adjsol_obj[0],np.sum(adjsol_grad))) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment