sweep

Author: flaport

README.md

@@ -34,7 +34,7 @@ The documentation is available at
 ## Contributors
 
 - [@flaport](https://github.com/flaport): creator of MEOW
-- [@jan-david-black](https://github.com/jan-david-black): fixing mode overlaps and more
+- [@jan-david-fischbach](https://github.com/jan-david-fischbach): fixing mode overlaps and more
 
 ## Credits
 

notebooks/02_taper_length_sweep.ipynb

@@ -82,7 +82,10 @@
     "    )\n",
     "\n",
     "    structures = [oxide, core]\n",
-    "    return structures"
+    "    return structures\n",
+    "\n",
+    "\n",
+    "structures = create_structures()"
    ]
   },
   {
@@ -92,7 +95,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "mw.visualize(create_structures())"
+    "mw.visualize(structures)"
    ]
   },
   {
@@ -110,37 +113,22 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def create_cells(length, cell_length):\n",
-    "    num_cells = round(length / cell_length)\n",
-    "    structures = create_structures(length=length)\n",
-    "    cells = mw.create_cells(\n",
-    "        structures=structures,\n",
-    "        mesh=mw.Mesh2D(\n",
-    "            x=np.linspace(-2, 2, 101),\n",
-    "            y=np.linspace(-2, 2, 101),\n",
-    "            # specify possible conformal mesh specifications here:\n",
-    "            # bend_radius=2.0,\n",
-    "            # bend_axis=1,\n",
-    "        ),\n",
-    "        Ls=np.array([length / num_cells for _ in range(num_cells)]),\n",
-    "    )\n",
-    "    return cells"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "9",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "cells = create_cells(10, 1)\n",
+    "num_cells = 20\n",
+    "\n",
+    "cells = mw.create_cells(\n",
+    "    structures=structures,\n",
+    "    mesh=mw.Mesh2D(\n",
+    "        x=np.linspace(-2, 2, 101),\n",
+    "        y=np.linspace(-2, 2, 101),\n",
+    "    ),\n",
+    "    Ls=np.array([10.0 / num_cells] * num_cells),\n",
+    ")\n",
     "mw.visualize(cells[-1])"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "10",
+   "id": "9",
    "metadata": {},
    "source": [
     "### 3. Cross Sections\n"
@@ -149,31 +137,18 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "11",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "def create_cross_sections(length, cell_length, wl=1.55, T=25.0):\n",
-    "    cells = create_cells(length=length, cell_length=cell_length)\n",
-    "    env = mw.Environment(wl=1.55, T=25.0)\n",
-    "    css = [mw.CrossSection.from_cell(cell=cell, env=env) for cell in cells]\n",
-    "    return css"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "12",
+   "id": "10",
    "metadata": {},
    "outputs": [],
    "source": [
-    "css = create_cross_sections(10, 1)\n",
+    "env = mw.Environment(wl=1.55, T=25.0)\n",
+    "css = [mw.CrossSection.from_cell(cell=cell, env=env) for cell in cells]\n",
     "mw.visualize(css[0])"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "13",
+   "id": "11",
    "metadata": {},
    "source": [
     "### 4. Find Modes (FDE)"
@@ -182,22 +157,18 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "14",
+   "id": "12",
    "metadata": {},
    "outputs": [],
    "source": [
-    "def find_modes(length, cell_length, num_modes, wl=1.55, T=25.0):\n",
-    "    css = create_cross_sections(length=length, cell_length=cell_length, wl=wl, T=T)\n",
-    "    modes: list[list[mw.Mode]] = []\n",
-    "    for cs in tqdm(css):\n",
-    "        modes_in_cs = mw.compute_modes(cs, num_modes=num_modes)\n",
-    "        modes.append(modes_in_cs)\n",
-    "    return modes"
+    "num_modes = 20\n",
+    "\n",
+    "modes = [mw.compute_modes(cs, num_modes=num_modes) for cs in tqdm(css)]"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "15",
+   "id": "13",
    "metadata": {},
    "source": [
     "### 5. Calculate S-matrix (EME)"
@@ -206,40 +177,17 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "16",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "def find_s_matrix(length, cell_length, num_modes, wl=1.55, T=25.0):\n",
-    "    modes = find_modes(\n",
-    "        length=length, cell_length=cell_length, wl=wl, T=T, num_modes=num_modes\n",
-    "    )\n",
-    "    S, port_map = mw.compute_s_matrix(modes, cell_lengths=[cell_length for _ in modes])\n",
-    "    return S, port_map"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "17",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "num_modes = 20\n",
-    "cell_length = 0.5\n",
-    "S1, pm1 = find_s_matrix(10.0, cell_length, num_modes)\n",
-    "S2, pm2 = find_s_matrix(5.0, cell_length, num_modes)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "18",
+   "id": "14",
    "metadata": {},
    "outputs": [],
    "source": [
-    "print(f\"long taper transmission: {abs(S1[0, num_modes]) ** 2}\")\n",
-    "print(f\"short taper transmission: {abs(S2[0, num_modes]) ** 2}\")"
+    "taper_lengths = [5.0, 10.0, 15.0, 20.0]\n",
+    "\n",
+    "for length in taper_lengths:\n",
+    "    cell_lengths = [length / num_cells] * num_cells\n",
+    "    S, pm = mw.compute_s_matrix(modes, cell_lengths=cell_lengths)\n",
+    "    T_00 = abs(S[0, num_modes]) ** 2\n",
+    "    print(f\"taper length {length:5.1f} | transmission: {T_00:.6f}\")"
    ]
   }
  ],