compute rho

Author: malmans2

notebooks/wp5/steric_sea_level_ora5.ipynb

@@ -23,6 +23,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import gsw_xarray as gsw\n",
+    "import xarray as xr\n",
     "from c3s_eqc_automatic_quality_control import download"
    ]
   },
@@ -42,7 +44,7 @@
    "outputs": [],
    "source": [
     "year_start = 2004\n",
-    "year_stop = 2023"
+    "year_stop = 2004  # 2023"
    ]
   },
   {
@@ -90,7 +92,7 @@
    "id": "7",
    "metadata": {},
    "source": [
-    "## Download and transform reanalysis"
+    "## Define functions to cache"
    ]
   },
   {
@@ -100,15 +102,20 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ds_reanalysis = download.download_and_transform(*request_reanalysis, **download_kwargs)"
+    "def compute_insitu_density(ds):\n",
+    "    p = gsw.p_from_z(-ds[\"deptht\"], ds[\"latitude\"])\n",
+    "    SA = gsw.SA_from_SP(ds[\"vosaline\"], p, ds[\"longitude\"], ds[\"latitude\"])\n",
+    "    CT = gsw.CT_from_pt(SA, ds[\"votemper\"])\n",
+    "    rho = gsw.rho(SA, CT, p)\n",
+    "    return xr.merge([p, SA, CT, rho])"
    ]
   },
   {
    "cell_type": "markdown",
    "id": "9",
    "metadata": {},
    "source": [
-    "## Download and transform satellite"
+    "## Download and transform reanalysis"
    ]
   },
   {
@@ -117,6 +124,28 @@
    "id": "10",
    "metadata": {},
    "outputs": [],
+   "source": [
+    "ds_reanalysis = download.download_and_transform(\n",
+    "    *request_reanalysis,\n",
+    "    transform_func=compute_insitu_density,\n",
+    "    **download_kwargs,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "11",
+   "metadata": {},
+   "source": [
+    "## Download and transform satellite"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "12",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "ds_satellite = download.download_and_transform(*request_satellite, **download_kwargs)"
    ]