In this notebook, we will use a multi-objective optimization to design a wavelength division multiplexer (WDM).
In short, this device takes in broadband light and directs light of different wavelengths to different output ports.
This demo combines the basic setup of our 3rd tutorial of a mode converter with the multi-frequency feature introduced in Tidy3D version 2.5.
If you are unfamiliar with inverse design, we also recommend our intro to inverse design tutorials and our primer on automatic differentiation with tidy3d.
import matplotlib.pylab as plt
# first import tidy3d, its adjoint plugin, numpy, and jax.
import tidy3d as td
import tidy3d.plugins.adjoint as tda
import numpy as np
import jax.numpy as jnp
import jax
np.random.seed(2)
Setup¶
First we set up our basic simulation.
We have an input waveguide connected to a square design region, which has two output waveguides.
The square design region is a custom medium with a pixellated permittivity grid that we wish to optimize such that input light of different wavelengths get directed to different output ports.
As this is a SOI device, we typically define the design region and waveguides as Silicon sitting on an SiO2 substrate. For this demo, we make a 2D simulation, but it can be easily made 3D by changing the Lz parameter, adding dimension to the structures, and adding a substrate.
# material information
n_si = 3.49
n_sio2 = 1.45 # not used in 2D
n_air = 1
# design output wavelengths
wavelength_top = 1.300
wavelength_bot = 1.550
# and their corresponding frequencies and spectral information
freq_top = td.C_0 / wavelength_top
freq_bot = td.C_0 / wavelength_bot
freq0 = (freq_top + freq_bot) / 2.0
fwidth = abs(freq_bot - freq_top)
run_time = 100 / fwidth
# create dictionaries to reference these later by string key 'top' or 'bot'
freqs = dict(top=freq_top, bot=freq_bot)
wavelengths = dict(top=wavelength_top, bot=wavelength_bot)
# size of design region
lx = 2.8
ly = 2.8
lz = td.inf # in 2D, we say the size of components is inf but the size of simulation is 0.
# size of waveguides
wg_width = 0.3
wg_length = 1.5
wg_spacing = 0.8
# spacing between design region and PML in y
buffer = 1.5
# size of simulation
Lx = lx + wg_length * 2
Ly = ly + buffer * 2
Lz = 0.0
# fabrication constraint (feature size and projection strength)
radius = 0.200
beta = 30
# resolution information
min_steps_per_wvl = 25
# define the waveguide ports
wg_in = td.Structure(
geometry=td.Box(
center=(-Lx/2, 0, 0),
size=(wg_length * 2, wg_width, lz),
),
medium=td.Medium(permittivity=n_si**2)
)
wg_top = td.Structure(
geometry=td.Box(
center=(+Lx/2, +wg_width/2+wg_spacing/2, 0),
size=(wg_length * 2, wg_width, lz),
),
medium=td.Medium(permittivity=n_si**2)
)
wg_bot = td.Structure(
geometry=td.Box(
center=(+Lx/2, -wg_width/2-wg_spacing/2, 0),
size=(wg_length * 2, wg_width, lz),
),
medium=td.Medium(permittivity=n_si**2)
)
# and a field monitor that measures fields on the z=0 plane
fld_mnt = td.FieldMonitor(
center=(0,0,0),
size=(td.inf, td.inf, 0),
freqs=[freq_top, freq_bot],
name="field",
)
Note we can ignore this warning as it will be resolved after 2.4.0
Define design region¶
Here we define the design region as a pixellated grid of permittivity values.
We first define the overall geometry as a JaxBox and also the number of pixels in x and y.
nx = 55
ny = 55
design_region_geo = tda.JaxBox(
size=(lx, ly, lz),
center=(0,0,0)
)
design_region_dl=lx/nx
Next we write a function to give us the pixellated array as a function of our parameters through our filtering and projection methods, which are used to make the resulting structures easier to fabricate. For more details, refer to our 4th lecture in the inverse design 101 lecture series, which focuses on fabrication constraints.
We also wrap this function in another one that generates the entire JaxStructure corresponding to the design region, for convenience later.
from tidy3d.plugins.adjoint.utils.filter import ConicFilter
conic_filter = ConicFilter(radius=radius, design_region_dl=design_region_dl)
# note: params is an array of shape (nx, ny) that stores values between -inf (air) and +inf (silicon)
def tanh_projection(x, beta, eta=0.5):
tanhbn = jnp.tanh(beta * eta)
num = tanhbn + jnp.tanh(beta * (x - eta))
den = tanhbn + jnp.tanh(beta * (1 - eta))
return num / den
def filter_project(x, beta, eta=0.5):
x = conic_filter.evaluate(x)
return tanh_projection(x, beta=beta, eta=eta)
# number of times to filter -> project. Two times with a lower beta (~30) seems to give decent results.
num_projections = 2
def pre_process(params, beta):
"""Get the permittivity values (1, eps_wg) array as a funciton of the parameters (0,1)"""
for _ in range(num_projections):
params = filter_project(params, beta=beta)
return params
def make_eps(params, beta):
params = pre_process(params, beta=beta)
eps_values = 1 + (n_si**2 - 1) * params
return eps_values
def make_custom_medium(params, beta):
"""Make JaxCustomMedium as a function of provided parameters."""
eps = make_eps(params, beta).reshape((nx, ny, 1, 1))
eps = jnp.where(eps < 1, 1, eps)
eps = jnp.where(eps > n_si**2, n_si**2, eps)
xs = list(jnp.linspace(-lx/2, lx/2, nx))
ys = list(jnp.linspace(-ly/2, ly/2, ny))
zs = [0]
freqs = [freq0]
coords = dict(x=xs, y=ys, z=zs, f=freqs)
eps_dataset = tda.JaxDataArray(values=eps, coords=coords)
medium = tda.JaxCustomMedium(
eps_dataset=tda.JaxPermittivityDataset(
eps_xx=eps_dataset,
eps_yy=eps_dataset,
eps_zz=eps_dataset,
)
)
struct = tda.JaxStructure(
geometry=design_region_geo,
medium=medium
)
return struct
Define base simulation¶
With all of these functions and variables defined, we can write a single function to return our "base" JaxSimulation as a function of our design parameters. This function first constructs the design region and then creates a JaxSimulation with all of the basic parameters.
Note, we don't yet have a source or monitors for injecting and measuring our fields, but will add those next after running the mode solver.
def make_sim_base(params, beta):
input_struct = make_custom_medium(params, beta=beta)
return tda.JaxSimulation(
size=(Lx, Ly, Lz),
grid_spec=td.GridSpec.auto(min_steps_per_wvl=min_steps_per_wvl, wavelength=wavelength_top),
structures=[wg_in, wg_top, wg_bot],
monitors=[fld_mnt],
input_structures=[input_struct],
boundary_spec=td.BoundarySpec.pml(x=True, y=True, z=True if Lz else False),
run_time=run_time,
)
Let's test out our function. We'll make an initially random array of parameters between 0 and 1 and generate the base simulation to plot and inspect.
params0 = np.random.random((nx, ny))
sim_base = make_sim_base(params0, beta=1)
WARNING:jax._src.xla_bridge:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
ax = sim_base.plot_eps(z=0, monitor_alpha=0.0)
It all looks good, so now we add the bits that define the optimization.
Adding Mode Sources and Monitors¶
Solving modes¶
First, we need to create our ModeSource and ModeMonitor objects that inject and measure the modes that we are interested in optimizing.
We'll use tidy3d's ModeSolver and use the remote run function that gets more accurate results by running on Flexcompute's servers.
from tidy3d.plugins.mode import ModeSolver
from tidy3d.plugins.mode.web import run as run_mode_solver
# we'll ask for 4 modes just to inspect
num_modes = 4
# let's define how large the mode planes are and how far they are from the PML relative to the design region
mode_size = (0, 1.8 * wg_spacing + wg_width, max([Lz, lz, 3]))
space_fraction = 0.2
# make a plane corresponding to where we wish to measure the input mode
plane_in = td.Box(
center=(-Lx/2 + space_fraction * wg_length, 0, 0),
size=mode_size,
)
# construct the mode solver using our base sim (converted from `JaxSimulation` to regular `Simulation`) + our plane
mode_solver = ModeSolver(
simulation=sim_base.to_simulation()[0],
plane=plane_in,
freqs=[freq_top],
mode_spec=td.ModeSpec(num_modes=num_modes)
)
Next we run the mode solver on the servers.
mode_data = run_mode_solver(mode_solver, reduce_simulation=True)
10:59:18 PST Mode solver created with task_id='fdve-9a68c85b-b233-4b9c-918f-aec0bf5cdcd1', solver_id='mo-4bbb01f6-7a1f-490a-a82f-7d81cbd26a0f'.
Output()
Output()
10:59:19 PST Mode solver status: queued
11:01:30 PST Mode solver status: running
11:01:33 PST Mode solver status: success
Output()
And visualize the results.
fig, axs = plt.subplots(num_modes, 3, figsize=(12, 12), tight_layout=True)
for mode_index in range(num_modes):
vmax = 1.1 * max(abs(mode_data.field_components[n].sel(mode_index=mode_index)).max() for n in ("Ex", "Ey", "Ez"))
for field_name, ax in zip(("Ex", "Ey", "Ez"), axs[mode_index]):
field = mode_data.field_components[field_name].sel(mode_index=mode_index)
field.real.plot(label="Real", ax=ax)
field.imag.plot(ls="--", label="Imag", ax=ax)
ax.set_title(f'index={mode_index}, {field_name}')
ax.set_ylim(-vmax, vmax)
axs[0, 0].legend()
print("Effective index of computed modes: ", np.array(mode_data.n_eff))
Effective index of computed modes: [[3.141611 2.806403 1.953868 1.062474]]
We identify mode_index=0 as the first order mode that is out of plane of the device. Let's choose to optimize our device with respect to this as the mode of interest for both the input and output.
We re-set the ModeSpec to only compute the number of modes we need (1) and also update our ModeSolver accordingly.
mode_index = 0
mode_spec = td.ModeSpec(num_modes=mode_index+1)
mode_solver = mode_solver.updated_copy(mode_spec=mode_spec)
Make input and output mode sources and monitors¶
Next, we will generate the input ModeSource and output ModeMonitor objects using the convenience methods defined in the ModeSolver.
Because our plane was defined at the input port, we'll modify the centers of the ModeMonitors to place them at the output ports to the right of the device.
# make source
mode_src = mode_solver.to_source(
source_time=td.GaussianPulse(
freq0=freq0,
fwidth=fwidth,
),
direction="+",
mode_index=mode_index,
)
# make a basic monitor
mode_mnt = mode_solver.to_monitor(
freqs=[freq0],
name="_"
)
# construct the proper centers for the monitors at the 'top' and 'bot' ports
mnt_center_top = list(plane_in.center)
mnt_center_bot = list(plane_in.center)
mnt_center_top[0] = -plane_in.center[0]
mnt_center_bot[0] = -plane_in.center[0]
mnt_center_top[1] = wg_top.geometry.center[1]
mnt_center_bot[1] = wg_bot.geometry.center[1]
# make a dictionary of names and frequencies to refer to later by key
mnt_names = dict(top="mode_top", bot="mode_bot")
mnt_freqs = dict(top=freq_top, bot=freq_bot)
# make two updated copies of the mode monitor with the proper frequencies, centers, and names
mode_mnt_top = mode_mnt.updated_copy(freqs=[mnt_freqs["top"]], center=mnt_center_top, name=mnt_names["top"])
mode_mnt_bot = mode_mnt.updated_copy(freqs=[mnt_freqs["bot"]], center=mnt_center_bot, name=mnt_names["bot"])
# make another dictionary mapping the keys to the monitors
mode_mnts = dict(top=mode_mnt_top, bot=mode_mnt_bot)
Add flux monitors¶
For plotting later, we'll add a couple of FluxMonitor objects at the output ports to measure the total flux over a large spectrum. With this data, we should be able to clearly see the difference in transmission for each of the ports at the design region and get an idea about the device bandwidth.
Nf = 121
freqs_flux = np.linspace(freq_bot - fwidth/10, freq_top + fwidth/10, Nf)
flux_mnt_names = dict(top="flux_top", bot="flux_bot")
flux_mnt_top = td.FluxMonitor(
center=mode_mnt_top.center,
size=mode_mnt_top.size,
name=flux_mnt_names["top"],
freqs=list(freqs_flux),
)
flux_mnt_bot = td.FluxMonitor(
center=mode_mnt_bot.center,
size=mode_mnt_bot.size,
name=flux_mnt_names["bot"],
freqs=list(freqs_flux),
)
Add to simulation¶
Finally, we will wrap our previous make_sim_base() function in a new one that adds our new objects to this base simulation.
def make_sim(params, beta):
output_monitors = [mode_mnts["top"], mode_mnts["bot"]]
sim_base = make_sim_base(params, beta=beta)
return sim_base.updated_copy(
output_monitors=output_monitors,
sources=[mode_src],
monitors=tuple(list(sim_base.monitors) + [flux_mnt_top, flux_mnt_bot])
)
Let's make the final simulation and visualize it with the sources and monitors added.
sim = make_sim(params0, beta=1)
Note: the FluxMonitor objects are overlaying the output ModeMonitor objects.
ax = sim.plot_eps(z=0.01)
plt.show()
Defining objective function¶
With our simulation fully defined as a function of our parameters, we are ready to define our objective function.
Computing power transmission¶
In this case, it is quite simple, we simply measure the transmitted power in our output waveguide mode. We wish to maximize transmission to the top port at the "top" wavelength (1330 nm) and maximize transmission to the bottom port at the "bot" wavelength (1550 nm).
def measure_power(sim_data) -> float:
"""Extract power from simulation data."""
def get_power(mnt_key: str, freq_key: str) -> float:
"""Get the power at monitor 'mnt_key' at frequency 'freq_key' (both either 'top' or 'bot')."""
mnt_name = mnt_names[mnt_key]
freq = freqs[freq_key]
mnt_data = sim_data[mnt_name]
amp = mnt_data.amps.sel(direction="+", mode_index=0, f=freq)
return jnp.abs(amp) ** 2
power_max = get_power("top", "top") + get_power("bot", "bot")
return power_max / 2.0
Next we add a penalty to produce structures that are invariant under erosion and dilation, which is a useful approach to implementing minimum length scale features.
from tidy3d.plugins.adjoint.utils.penalty import ErosionDilationPenalty
def penalty(params, beta) -> float:
"""Penalty based on the amount of change after erosion and dilation of structures."""
params = pre_process(params, beta=beta)
ed_penalty = ErosionDilationPenalty(length_scale=radius, pixel_size=design_region_dl)
return ed_penalty.evaluate(params)
Writing objective function¶
Then we write an objective function that constructs our simulation, runs it, measures the power, and subtracts our penalty.
Note, we return a
JaxSimulationDataas the second output. The reason for this is that we might wish to access our flux and field data later on.jaxgives an optionhas_auxto use only the first output for differentiation while letting the user have access to the 2nd "auxiliary" output, which we will make use of.
def objective(params, beta, verbose=False) -> float:
sim = make_sim(params, beta=beta)
sim_data = tda.web.run(sim, task_name="WDM_MULTIFREQ", verbose=verbose)
power = measure_power(sim_data)
J = power - penalty(params, beta=beta)
return J, sim_data
Differentiating objective¶
Finally, we can simply use jax to transform this objective function into a function that returns our objective function value, the auxiliary data, and our gradient, which we will feed to the optimizer.
grad_fn = jax.value_and_grad(objective, has_aux=True)
Let's try out our gradient function with verbosity on for just this run.
(J, sim_data), grad = grad_fn(params0, beta=1, verbose=True)
11:02:20 PST Created task 'WDM_MULTIFREQ' with task_id 'fdve-c66b7065-34a3-4a23-b70a-1cae0a9791e3' and task_type 'FDTD'.
View task using web UI at 'https://tidy3d.simulation.cloud/workbench?taskId=fdve-c66b7065-34a 3-4a23-b70a-1cae0a9791e3'.
Output()
Output()
11:02:21 PST status = success
Output()
11:02:22 PST loading simulation from simulation_data.hdf5
11:02:23 PST WARNING: 2 unique frequencies detected in the output monitors with a minimum spacing of 3.720e+13 (Hz). Setting the 'fwidth' of the adjoint sources to 0.1 times this value = 3.720e+12 (Hz) to avoid spectral overlap. To account for this, the corresponding 'run_time' in the adjoint simulation is will be set to 2.688527e-11 compared to 2.688527e-12 in the forward simulation. If the adjoint 'run_time' is large due to small frequency spacing, it could be better to instead run one simulation per frequency, which can be done in parallel using 'tidy3d.plugins.adjoint.web.run_async'.
Created task 'WDM_MULTIFREQ_adj' with task_id 'fdve-7c7a36f3-2152-4379-82f1-707760603828' and task_type 'FDTD'.
View task using web UI at 'https://tidy3d.simulation.cloud/workbench?taskId=fdve-7c7a36f3-215 2-4379-82f1-707760603828'.
Output()
Output()
11:02:24 PST status = queued
To cancel the simulation, use 'web.abort(task_id)' or 'web.delete(task_id)' or abort/delete the task in the web UI. Terminating the Python script will not stop the job running on the cloud.
Output()
11:02:32 PST status = preprocess
11:02:36 PST Maximum FlexCredit cost: 0.031. Use 'web.real_cost(task_id)' to get the billed FlexCredit cost after a simulation run.
starting up solver
running solver
Output()
11:02:48 PST early shutoff detected at 4%, exiting.
status = postprocess
Output()
11:02:50 PST status = success
11:02:51 PST View simulation result at 'https://tidy3d.simulation.cloud/workbench?taskId=fdve-7c7a36f3-215 2-4379-82f1-707760603828'.
Output()
Run Opimization¶
Finally, we are ready to optimize our device. We will make use the optax package to define an optimizer using the adam method, as we've done in the previous adjoint tutorials.
We record a history of objective function values, simulation data, and parameters for visualization later.
import optax
# we know that the source fwidth will be set automatically due to multi-freq adjoint, so suppress warnings
td.config.logging_level = "ERROR"
# hyperparameters
num_steps = 50
learning_rate = 5e-2
beta_min = 1
beta_max = 30
# initialize adam optimizer with starting parameters
params = params0
optimizer = optax.adam(learning_rate=learning_rate)
opt_state = optimizer.init(params)
# store history
Js = []
params_history = [params0]
data_history = []
beta_history = []
for i in range(num_steps):
perc_done = i / (num_steps - 1)
beta_i = beta_min * (1 - perc_done) + beta_max * perc_done
# compute gradient and current objective funciton value
(value, data), gradient = grad_fn(params, beta=beta_i)
# outputs
print(f"step = {i + 1}")
print(f"\tJ = {value:.4e}")
print(f"\tbeta = {beta_i:.2f}")
print(f"\tgrad_norm = {np.linalg.norm(gradient):.4e}")
# compute and apply updates to the optimizer based on gradient (-1 sign to maximize obj_fn)
updates, opt_state = optimizer.update(-gradient, opt_state, params)
params = optax.apply_updates(params, updates)
# keep params between 0 and 1
params = jnp.minimum(1.0, params)
params = jnp.maximum(0.0, params)
# save history
Js.append(value)
params_history.append(params)
beta_history.append(beta_i)
data_history.append(data)
step = 1 J = 1.1304e-01 beta = 1.00 grad_norm = 1.3148e-01 step = 2 J = 1.7998e-01 beta = 1.59 grad_norm = 2.2417e-01 step = 3 J = 1.9274e-01 beta = 2.18 grad_norm = 2.6165e-01 step = 4 J = 5.7359e-02 beta = 2.78 grad_norm = 3.2920e-01 step = 5 J = -2.7653e-02 beta = 3.37 grad_norm = 5.1749e-01 step = 6 J = -1.0726e-01 beta = 3.96 grad_norm = 5.9930e-01 step = 7 J = 5.4190e-02 beta = 4.55 grad_norm = 2.8124e-01 step = 8 J = 1.3695e-01 beta = 5.14 grad_norm = 3.0855e-01 step = 9 J = 1.9039e-01 beta = 5.73 grad_norm = 3.4029e-01 step = 10 J = 2.1604e-01 beta = 6.33 grad_norm = 5.2777e-01 step = 11 J = 2.6796e-01 beta = 6.92 grad_norm = 5.9430e-01 step = 12 J = 2.0278e-01 beta = 7.51 grad_norm = 1.1882e+00 step = 13 J = 3.2679e-01 beta = 8.10 grad_norm = 6.3214e-01 step = 14 J = 3.6340e-01 beta = 8.69 grad_norm = 2.8383e-01 step = 15 J = 3.7389e-01 beta = 9.29 grad_norm = 2.3837e-01 step = 16 J = 3.9022e-01 beta = 9.88 grad_norm = 1.6353e-01 step = 17 J = 4.0384e-01 beta = 10.47 grad_norm = 1.3485e-01 step = 18 J = 4.1686e-01 beta = 11.06 grad_norm = 1.2928e-01 step = 19 J = 4.2954e-01 beta = 11.65 grad_norm = 1.1254e-01 step = 20 J = 4.3786e-01 beta = 12.24 grad_norm = 1.2386e-01 step = 21 J = 4.4402e-01 beta = 12.84 grad_norm = 1.3647e-01 step = 22 J = 4.4862e-01 beta = 13.43 grad_norm = 3.4103e-01 step = 23 J = 4.5884e-01 beta = 14.02 grad_norm = 1.4109e-01 step = 24 J = 4.6795e-01 beta = 14.61 grad_norm = 8.9034e-02 step = 25 J = 4.7021e-01 beta = 15.20 grad_norm = 1.6547e-01 step = 26 J = 4.8514e-01 beta = 15.80 grad_norm = 1.6582e-01 step = 27 J = 5.0607e-01 beta = 16.39 grad_norm = 1.3549e-01 step = 28 J = 5.2328e-01 beta = 16.98 grad_norm = 1.7107e-01 step = 29 J = 5.5337e-01 beta = 17.57 grad_norm = 1.8353e-01 step = 30 J = 5.8120e-01 beta = 18.16 grad_norm = 2.6351e-01 step = 31 J = 6.0239e-01 beta = 18.76 grad_norm = 2.1676e-01 step = 32 J = 6.1259e-01 beta = 19.35 grad_norm = 4.3566e-01 step = 33 J = 5.4708e-01 beta = 19.94 grad_norm = 9.3248e-01 step = 34 J = 6.1207e-01 beta = 20.53 grad_norm = 6.7785e-01 step = 35 J = 6.4012e-01 beta = 21.12 grad_norm = 2.7947e-01 step = 36 J = 6.2275e-01 beta = 21.71 grad_norm = 6.0492e-01 step = 37 J = 6.6233e-01 beta = 22.31 grad_norm = 1.6761e-01 step = 38 J = 6.5566e-01 beta = 22.90 grad_norm = 6.2367e-01 step = 39 J = 6.5845e-01 beta = 23.49 grad_norm = 6.8979e-01 step = 40 J = 6.9423e-01 beta = 24.08 grad_norm = 3.0279e-01 step = 41 J = 7.0369e-01 beta = 24.67 grad_norm = 2.5525e-01 step = 42 J = 6.9838e-01 beta = 25.27 grad_norm = 5.4912e-01 step = 43 J = 6.9214e-01 beta = 25.86 grad_norm = 7.5892e-01 step = 44 J = 7.1860e-01 beta = 26.45 grad_norm = 4.9100e-01 step = 45 J = 7.4005e-01 beta = 27.04 grad_norm = 2.2158e-01 step = 46 J = 7.3577e-01 beta = 27.63 grad_norm = 5.3908e-01 step = 47 J = 7.4667e-01 beta = 28.22 grad_norm = 4.1632e-01 step = 48 J = 7.5887e-01 beta = 28.82 grad_norm = 2.1059e-01 step = 49 J = 7.6499e-01 beta = 29.41 grad_norm = 3.3670e-01 step = 50 J = 7.6311e-01 beta = 30.00 grad_norm = 4.7441e-01
Visualize Results¶
Let's visualize the results of our optimization.
Objective function vs Iteration¶
First we inspect the objective function value as a function of optimization iteration number. We see that it steadily increases as expected.
The presence of fabrication constraints tends to create some minor bumps in the optimization, which can be a signal that one needs to reduce the step size, but these results are sufficient for our purposes.
plt.plot(Js)
plt.xlabel('iteration number')
plt.ylabel('objective function')
plt.show()
Final Simulation¶
Let's take a look at the final simulation, which we grab from our history.
sim_data_final = data_history[-1]
sim_final = sim_data_final.simulation.to_simulation()[0]
We notice that the structure has reasonably large feature sizes but is not well binarized. This could be improved by increasing the beta projection value slowly over iteration number, as was done in the grating coupler tutorial.
ax = sim_final.plot_eps(z=0.01, monitor_alpha=0, source_alpha=0)
Flux¶
Let's inspect the flux over each of the output ports as a function of wavelength.
We notice that the top and bottom ports have peaks in transmission at their corresponding design wavelengths, as expected!
# plot flux
for key, color in zip(("top", "bot"), ('royalblue', 'firebrick')):
freq = freqs[key]
flux_data = sim_data_final[flux_mnt_names[key]]
wvl_nm = 1000 * td.C_0 / freq
wavelengths_nm = 1000 * td.C_0 / np.array(flux_data.flux.f)
flux = np.array(flux_data.flux.values)
flux_db = 10 * np.log(flux)
label = f"{key} ({int(wvl_nm)} nm)"
plt.plot(wavelengths_nm, flux_db, label=label, color=color)
plt.scatter([wvl_nm], [0], 100, marker="*", color=color)
plt.xlabel('wavelength (nm)')
plt.ylabel('transmission (dB)')
plt.legend()
plt.show()
Fields¶
Let's also plot the field intensity patterns at the two design wavelengths. We see from this plot the expected result that the power is directed to the design port at each frequency.
# plot fields at the two design wavelengths
fig, axes = plt.subplots(1, 2, tight_layout=True, figsize=(7, 3))
for key, ax in zip(("top", "bot"), axes):
freq = freqs[key]
sim_data_final.plot_field("field", "E", "abs^2", f=freq, ax=ax, vmax=1200)
wvl = 1000 * td.C_0 / freq
ax.set_title(f"wavelength = {int(wvl)} nm")
Animation¶
Finally, we animate this plot as a function of iteration number. The animation shows the device quickly accomplishing our design objective.
Note: can take a few minutes to complete
import matplotlib.animation as animation
from IPython.display import HTML
fig, (ax1, ax2, ax3) = fig, axes = plt.subplots(1, 3, tight_layout=False, figsize=(9, 4))
def animate(i):
sim_data_i = data_history[i]
sim_i = sim_data_i.simulation.to_simulation()[0]
sim_i.plot_eps(z=0.01, monitor_alpha=0, source_alpha=0, ax=ax1)
ax1.set_aspect('equal')
for key, ax in zip(("top", "bot"), (ax2, ax3)):
freq = freqs[key]
wvl = 1000 * td.C_0 / freq
int_i = sim_data_i.get_intensity("field").sel(f=freq)
int_i.squeeze().plot.pcolormesh(x='x', y='y', ax=ax, add_colorbar=False, cmap="magma", vmax=1000)
ax.set_aspect('equal')
ax.set_title(f"wavelength = {int(wvl)} nm")
# create animation
ani = animation.FuncAnimation(fig, animate, frames=len(data_history))
plt.close()
# display the animation (press "play" to start)
HTML(ani.to_jshtml())
<Figure size 640x480 with 0 Axes>
To save the animation as a file, uncomment the line below
Note: can take several more minutes to complete
# ani.save('img/animation_wdm_adjoint.gif', fps=60)
Export to GDS¶
The Simulation object has the .to_gds_file convenience function to export the final design to a GDS file. In addition to a file name, it is necessary to set a cross-sectional plane (z = 0 in this case) on which to evaluate the geometry, a frequency to evaluate the permittivity, and a permittivity_threshold to define the shape boundaries in custom mediums. See the GDS export notebook for a detailed example on using .to_gds_file and other GDS related functions.
sim_final = data_history[-1].simulation.to_simulation()[0]
sim_final.to_gds_file(
fname="./misc/inv_des_wdm.gds",
z=0,
permittivity_threshold=(n_si**2 + 1) / 2,
frequency=freq0,
)
