add argo

Author: malmans2

notebooks/wp5/steric_sea_level_ora5.ipynb

@@ -53,6 +53,7 @@
     "import cacholote\n",
     "import gsw_xarray as gsw\n",
     "import matplotlib.pyplot as plt\n",
+    "import pooch\n",
     "import xarray as xr\n",
     "from c3s_eqc_automatic_quality_control import diagnostics, download, utils\n",
     "\n",
@@ -334,7 +335,7 @@
    "id": "13",
    "metadata": {},
    "source": [
-    "## Download and transform satellite"
+    "## Download and transform ARGO"
    ]
   },
   {
@@ -343,6 +344,77 @@
    "id": "14",
    "metadata": {},
    "outputs": [],
+   "source": [
+    "def preprocess(ds):\n",
+    "    # Naming\n",
+    "    ds = ds.rename({var: var.lower() for var in ds.variables})\n",
+    "    # Time\n",
+    "    ds[\"time\"].attrs[\"calendar\"] = \"360_day\"\n",
+    "    ds = xr.decode_cf(ds)\n",
+    "    # Depth\n",
+    "    ds[\"depth\"] = -gsw.z_from_p(ds[\"pressure\"], ds[\"latitude\"]).mean(\n",
+    "        \"latitude\", keep_attrs=True\n",
+    "    )\n",
+    "    ds[\"depth\"].attrs.update({\"positive\": \"down\", \"long_name\": \"Depth from pressure\"})\n",
+    "    return ds.swap_dims(pressure=\"depth\")\n",
+    "\n",
+    "\n",
+    "# First dataset\n",
+    "filenames = []\n",
+    "for var in [\"Temperature\", \"Salinity\"]:\n",
+    "    url = f\"https://sio-argo.ucsd.edu/RG/RG_ArgoClim_{var}_2019.nc.gz\"\n",
+    "    filename = pooch.retrieve(url=url, known_hash=None, processor=pooch.Decompress())\n",
+    "    filenames.append(filename)\n",
+    "with xr.set_options(use_new_combine_kwarg_defaults=True):\n",
+    "    ds_argo_1 = xr.open_mfdataset(filenames, preprocess=preprocess, decode_times=False)\n",
+    "\n",
+    "# Second dataset\n",
+    "filenames = []\n",
+    "for year in range(2019, year_stop + 1):\n",
+    "    for month in range(1, 13):\n",
+    "        url = f\"https://sio-argo.ucsd.edu/RG/RG_ArgoClim_{year}{month:02d}_2019.nc.gz\"\n",
+    "        filename = pooch.retrieve(\n",
+    "            url=url, known_hash=None, processor=pooch.Decompress()\n",
+    "        )\n",
+    "        filenames.append(filename)\n",
+    "with xr.set_options(use_new_combine_kwarg_defaults=True):\n",
+    "    ds_argo_2 = xr.open_mfdataset(filename, preprocess=preprocess, decode_times=False)\n",
+    "\n",
+    "# Combine\n",
+    "dataarrays = []\n",
+    "for var in [\"salinity\", \"temperature\"]:\n",
+    "    da = ds_argo_1[f\"argo_{var}_mean\"]\n",
+    "    units = da.units\n",
+    "    da = xr.combine_by_coords(\n",
+    "        [\n",
+    "            da + ds_argo_1[f\"argo_{var}_anomaly\"],\n",
+    "            da + ds_argo_2[f\"argo_{var}_anomaly\"],\n",
+    "        ]\n",
+    "    )\n",
+    "    da.attrs[\"units\"] = units\n",
+    "    dataarrays.append(da.rename(var))\n",
+    "with xr.set_options(use_new_combine_kwarg_defaults=True):\n",
+    "    ds_argo = xr.merge(dataarrays)\n",
+    "\n",
+    "# Selection\n",
+    "ds_argo = ds_argo.sel(time=slice(str(year_start), str(year_stop)))\n",
+    "ds_argo = utils.regionalise(ds_argo, lon_slice=lon_slice, lat_slice=lat_slice)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "15",
+   "metadata": {},
+   "source": [
+    "## Download and transform satellite"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "16",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "ds_satellite = download.download_and_transform(*request_satellite, **download_kwargs)"
    ]