refactor: Deprecate DataContainer and replace with xarray (#168)

- The custom `DataContainer` class has been deprecated and replaced with `xarray.Dataset`.
- An `xarray` accessor `fleks` has been created to house the domain-specific logic for calculating derived variables and plotting.
---------

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Author: henry2004y

docs/amrex_data.ipynb

@@ -578,7 +578,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Flekspy provides convenient wrappers for creating plots from these slices."
+    "`flekspy` provides convenient wrappers for creating plots from these slices. You can use query syntax to plot with value range limits:"
    ]
   },
   {
@@ -587,9 +587,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "f, axes = dc.plot(\"By\", pcolor=True)\n",
-    "dc.add_stream(axes[0], \"Bx\", \"By\", color=\"w\")\n",
-    "dc.add_contour(axes[0], \"Bx\")"
+    "f, axes = dc.fleks.plot(\"Bx>(2.2e5)<(3e5) Ex\", pcolor=True, figsize=(12, 6))\n",
+    "dc.fleks.add_stream(axes[0], \"Bx\", \"By\", color=\"w\")\n",
+    "dc.fleks.add_contour(axes[1], \"Bx\")\n",
+    "\n",
+    "plt.show()"
    ]
   },
   {
@@ -611,19 +613,19 @@
     "\n",
     "fields = [\"By\", \"Bz\"]\n",
     "for ivar in range(2):\n",
-    "    v = dc.evaluate_expression(fields[ivar])\n",
+    "    v = dc.fleks.evaluate_expression(fields[ivar])\n",
     "    vmin = v.min().v\n",
     "    vmax = v.max().v\n",
     "    ax = axes[ivar]\n",
     "    ax.set_title(fields[ivar], fontsize=16)\n",
     "    ax.set_ylabel(\"Y\", fontsize=16)\n",
     "    ax.set_xlabel(\"X\", fontsize=16)\n",
-    "    c = ax.pcolormesh(dc.x.value, dc.y.value, np.array(v.T), cmap=\"turbo\")\n",
+    "    c = ax.pcolormesh(dc.X.values, dc.Y.values, np.array(v.T), cmap=\"turbo\")\n",
     "    cb = f.colorbar(c, ax=ax, pad=0.01)\n",
     "\n",
-    "    ax.set_xlim(np.min(dc.x.value), np.max(dc.x.value))\n",
-    "    ax.set_xlim(np.min(dc.y.value), np.max(dc.y.value))\n",
-    "    dc.add_stream(ax, \"Bx\", \"By\", density=0.5, color=\"w\")\n",
+    "    ax.set_xlim(np.min(dc.X.values), np.max(dc.X.values))\n",
+    "    ax.set_xlim(np.min(dc.Y.values), np.max(dc.Y.values))\n",
+    "    dc.fleks.add_stream(ax, \"Bx\", \"By\", density=0.5, color=\"w\")\n",
     "\n",
     "plt.show()"
    ]
@@ -760,29 +762,6 @@
     "pp_phase.show()"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Advanced Example: Slicing with Range Limits\n",
-    "\n",
-    "Now we'll use the second dataset to show a more advanced slicing feature. Inheriting from IDL procedures, you can pass strings to `plot` to limit the plotting range (note: this syntax is experimental)."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "ds2 = flekspy.load(\"data/fleks_particle_small/3d*amrex\")\n",
-    "dc2 = ds2.get_slice(\"z\", 0.001)\n",
-    "\n",
-    "f, axes = dc2.plot(\"Bx>(2.2e5)<(3e5) Ex\", figsize=(12, 6))\n",
-    "dc2.add_stream(axes[0], \"Bx\", \"By\", color=\"w\")\n",
-    "dc2.add_contour(axes[1], \"Bx\", color=\"k\")"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -850,4 +829,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}

src/flekspy/util/__init__.py

@@ -4,9 +4,3 @@
     get_unit,
     get_ticks,
 )
-from .data_container import (
-    DataContainer,
-    DataContainer1D,
-    DataContainer2D,
-    DataContainer3D,
-)