Skip to content

fixes delta lengths and netlist patching for notebook workflow_3_cascaded_mzi.ipynb #678

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
joamatab merged 2 commits into from
Feb 28, 2026
Merged

fixes delta lengths and netlist patching for notebook workflow_3_cascaded_mzi.ipynb #678

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
aebf13c
fixes delta lengths and netlist patching
jonbenfri Feb 26, 2026
c38dafe
rewrites ``patch_netlist`` for flat or recursive cases
jonbenfri Feb 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
91 changes: 59 additions & 32 deletions notebooks/workflow_3_cascaded_mzi.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
"\n",
"Each filter stage is formed by 4 cascaded Mach-Zenhder Interferometers (MZIs) with predefined delays for the central wavelength. Symmetrical Direction Couplers (DCs) are used to mix the signals at the ends of the MZI arms. In order to facilitate fabrication, all DC gaps are kept equal, so the power transfer ratios are defined by the coupling length of the DCs.\n",
"\n",
"We will design each DC through 3D FDTD simulations to guarantee the desired power ratios, which have been calculated to provide maximally flat response. The S parameters computed through FDTD are latter used in the full circuit simulation along with models for staight and curved waveguide sections, leading to an accurate model that exhibits features similar to those found in experimental data."
"We will design each DC through 3D FDTD simulations to guarantee the desired power ratios, which have been calculated to provide maximally flat response. The S parameters computed through FDTD are later used in the full circuit simulation along with models for staight and curved waveguide sections, leading to an accurate model that exhibits features similar to those found in experimental data."
]
},
{
Expand Down Expand Up @@ -311,7 +311,7 @@
"\n",
"We calculate the group index for our waveguides through `tidy3d`'s local mode solver. Because we're interested in precise dispersion, we use a dense mesh and high precision in these calculations.\n",
"\n",
"The path length differences for the MZIs are $\\Delta L$, $2\\Delta L$, $L_\\pi - 2\\Delta L$, and $-2\\Delta L$, with $L_\\pi$ the length required for $\\pi$ phase shift (negative values indicate a delay in the opposite arm to positive values).\n"
"The path length differences for the MZIs are -$\\Delta L$, -$2\\Delta L$, $2\\Delta L + L_\\pi$, and $2\\Delta L$, with $L_\\pi$ the length required for $\\pi$ phase shift (negative values indicate a delay in the opposite arm to positive values).\n"
]
},
{
Expand Down Expand Up @@ -352,10 +352,10 @@
"length_delta = mzi_path_difference(waveguide_solver, ng, fsr)\n",
"length_pi = lda_c / (2 * ne)\n",
"mzi_deltas = (\n",
" length_delta,\n",
" 2 * length_delta,\n",
" length_pi - 2 * length_delta,\n",
" -length_delta,\n",
" -2 * length_delta,\n",
" 2 * length_delta + length_pi,\n",
" 2 * length_delta,\n",
")\n",
"print(f\"Path difference (ΔL = {length_delta}, Lπ = {length_pi}):\", mzi_deltas)"
]
Expand All @@ -380,7 +380,7 @@
"layout = gf.c.mzi_lattice(\n",
" coupler_gaps=(gap,) * len(lengths),\n",
" coupler_lengths=tuple(lengths),\n",
" delta_lengths=tuple([abs(x) for x in mzi_deltas]),\n",
" delta_lengths=mzi_deltas,\n",
" cross_section=\"strip\",\n",
")\n",
"layout.plot()"
Expand Down Expand Up @@ -626,13 +626,13 @@
" i += 1\n",
" new_component = f\"{component}_v{i}\"\n",
" settings = model[\"settings\"]\n",
" settings_fitered = {\n",
" settings_filtered = {\n",
" k: v\n",
" for k, v in settings.items()\n",
" if k in inspect.signature(models_to_patch[component]).parameters\n",
" }\n",
" models[new_component] = models_to_patch[model[\"component\"]](\n",
" **settings_fitered\n",
" **settings_filtered\n",
" )\n",
" del model[\"settings\"]\n",
" model[\"component\"] = new_component\n",
Expand Down Expand Up @@ -705,6 +705,40 @@
"netlist[\"instances\"][instance_name]"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "f9a0edfe-9c44-43d3-a806-d25f790820f5",
"metadata": {},
"outputs": [],
"source": [
"def patch_netlist_mod(netlist, models, models_to_patch):\n",
" # For recursive netlist\n",
" for model in netlist.keys():\n",
" instances = netlist[model][\"instances\"]\n",
" for name in instances:\n",
" model = instances[name]\n",
" if model[\"component\"] in models_to_patch:\n",
" component = model[\"component\"]\n",
" i = 0\n",
" new_component = f\"{component}_v{i}\"\n",
" while new_component in models:\n",
" i += 1\n",
" new_component = f\"{component}_v{i}\"\n",
" settings = model[\"settings\"]\n",
" settings_filtered = {\n",
Comment thread
sourcery-ai[bot] marked this conversation as resolved.
Outdated
" k: v\n",
" for k, v in settings.items()\n",
" if k in inspect.signature(models_to_patch[component]).parameters\n",
" }\n",
" models[new_component] = models_to_patch[model[\"component\"]](\n",
" **settings_filtered\n",
" )\n",
" del model[\"settings\"]\n",
" model[\"component\"] = new_component\n",
" return netlist, models"
]
},
{
"cell_type": "markdown",
"id": "34",
Expand All @@ -720,29 +754,22 @@
"metadata": {},
"outputs": [],
"source": [
"# fig, ax = plt.subplots(1, 1, figsize=(12, 4))\n",
"# netlist, models = patch_netlist(\n",
"# netlist=layout.get_netlist(recursive=True),\n",
"# models={\"straight\": straight_model, \"bend_euler\": bend_model(cross_section=cross_section)},\n",
"# models_to_patch={\"coupler\": coupler_model},\n",
"# )\n",
"# circuit, _ = sax.circuit(netlist, models)\n",
"# lda = np.linspace(1.5, 1.6, 1001)\n",
"# s = circuit(wl=lda)\n",
"# ax.plot(lda, 20 * jnp.log10(jnp.abs(s[(\"o1\", \"o3\")])), label=\"Cross\")\n",
"# ax.plot(lda, 20 * jnp.log10(jnp.abs(s[(\"o1\", \"o4\")])), label=\"Thru\")\n",
"# ax.set_ylim(-30, 0)\n",
"# ax.set_xlabel(\"λ (µm)\")\n",
"# ax.legend()"
"fig, ax = plt.subplots(1, 1, figsize=(12, 4))\n",
"netlist, models = patch_netlist_mod(\n",
" netlist=layout.get_netlist(recursive=True),\n",
" models={\"straight\": straight_model, \"bend_euler\": bend_model(cross_section=cross_section)},\n",
" models_to_patch={\"coupler\": coupler_model},\n",
")\n",
"circuit, _ = sax.circuit(netlist, models)\n",
"lda = np.linspace(1.5, 1.6, 1001)\n",
"s = circuit(wl=lda)\n",
"ax.plot(lda, 20 * jnp.log10(jnp.abs(s[(\"o1\", \"o3\")])), label=\"Cross\")\n",
"ax.plot(lda, 20 * jnp.log10(jnp.abs(s[(\"o1\", \"o4\")])), label=\"Thru\")\n",
"ax.axvline(lda_c, linestyle=\"dotted\")\n",
"ax.set_ylim(-30, 0)\n",
"ax.set_xlabel(\"λ (µm)\")\n",
"ax.legend()"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "36",
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
Expand All @@ -751,7 +778,7 @@
"custom_cell_magics": "kql"
},
"kernelspec": {
"display_name": "base",
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "python3"
},
Expand All @@ -765,7 +792,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.11.9"
"version": "3.12.8"
}
},
"nbformat": 4,
Expand Down
Loading