uncomment next cell

mfschubert · Jan 8, 2025 · 9c8b1bf · 9c8b1bf
1 parent 19191f5
commit 9c8b1bf
Showing 1 changed file with 25 additions and 25 deletions.
diff --git a/docs/notebooks/crystal_bz.ipynb b/docs/notebooks/crystal_bz.ipynb
@@ -500,37 +500,37 @@
     }
    ],
    "source": [
-    "# from skimage import measure\n",
-    "# import numpy as onp\n",
-    "\n",
-    "# section_xy, section_xz, section_yz = crystal_cross_sections(\n",
-    "#     thickness_ambient=float(thickness_ambient),\n",
-    "#     thickness_slab=float(thickness_slab),\n",
-    "#     pitch=pitch,\n",
-    "#     diameter=diameter,\n",
-    "#     resolution=resolution,\n",
-    "#     num_unit_cells=brillouin_grid_shape,\n",
-    "# )\n",
+    "from skimage import measure\n",
+    "import numpy as onp\n",
+    "\n",
+    "section_xy, section_xz, section_yz = crystal_cross_sections(\n",
+    "    thickness_ambient=float(thickness_ambient),\n",
+    "    thickness_slab=float(thickness_slab),\n",
+    "    pitch=pitch,\n",
+    "    diameter=diameter,\n",
+    "    resolution=resolution,\n",
+    "    num_unit_cells=brillouin_grid_shape,\n",
+    ")\n",
     "\n",
-    "# xplot, zplot = jnp.meshgrid(x, z, indexing=\"ij\")\n",
-    "# field_plot = ex[:, :, 0].real\n",
+    "xplot, zplot = jnp.meshgrid(x, z, indexing=\"ij\")\n",
+    "field_plot = ex[:, :, 0].real\n",
     "\n",
-    "# plt.figure(figsize=(float(jnp.amax(xplot)), float(jnp.amax(zplot))), dpi=80)\n",
-    "# ax = plt.subplot(111)\n",
-    "# im = plt.pcolormesh(xplot, zplot, field_plot, shading=\"nearest\", cmap=\"bwr\")\n",
+    "plt.figure(figsize=(float(jnp.amax(xplot)), float(jnp.amax(zplot))), dpi=80)\n",
+    "ax = plt.subplot(111)\n",
+    "im = plt.pcolormesh(xplot, zplot, field_plot, shading=\"nearest\", cmap=\"bwr\")\n",
     "\n",
-    "# im.set_clim((-float(jnp.amax(field_plot)), float(jnp.amax(field_plot))))\n",
+    "im.set_clim((-float(jnp.amax(field_plot)), float(jnp.amax(field_plot))))\n",
     "\n",
-    "# contours = measure.find_contours(onp.array(section_xz))\n",
-    "# scale_factor = pitch / resolution\n",
-    "# for c in contours:\n",
-    "#     ax.plot(c[:, 0] / scale_factor, c[:, 1] / scale_factor, \"k\")\n",
+    "contours = measure.find_contours(onp.array(section_xz))\n",
+    "scale_factor = pitch / resolution\n",
+    "for c in contours:\n",
+    "    ax.plot(c[:, 0] / scale_factor, c[:, 1] / scale_factor, \"k\")\n",
     "\n",
-    "# ax.axis(\"equal\")\n",
-    "# ax.axis(\"off\")\n",
-    "# ax.set_ylim(ax.get_ylim()[::-1])\n",
+    "ax.axis(\"equal\")\n",
+    "ax.axis(\"off\")\n",
+    "ax.set_ylim(ax.get_ylim()[::-1])\n",
     "\n",
-    "# plt.subplots_adjust(left=0, bottom=0, right=1, top=1)"
+    "plt.subplots_adjust(left=0, bottom=0, right=1, top=1)"
    ]
   },
   {