water_level_comparison.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import matplotlib.pyplot as plt\n",
    "import cartopy.crs as ccrs\n",
    "import numpy as np\n",
    "import matplotlib.tri as tri\n",
    "import cartopy.feature as cfeature\n",
    "import netCDF4 as nc\n",
    "\n",
    "# Load your datasets\n",
    "withoutoffset = nc.Dataset('/work2/07174/soelem/hopper/fort.63.nc')\n",
    "withoffset = nc.Dataset('/work2/07174/soelem/validation/combined/fort.63.nc')\n",
    "\n",
    "# Extract necessary variables\n",
    "lon = withoutoffset.variables['x'][:]\n",
    "lat = withoutoffset.variables['y'][:]\n",
    "\n",
    "# Specify the timestep you're interested in\n",
    "timestep = 2400  # Replace with your specific timestep\n",
    "\n",
    "# Define the levels for contouring\n",
    "levels = np.arange(-0.9, 0.9, 0.01)  # Adjust as needed\n",
    "\n",
    "# Create the figure and subplots\n",
    "fig, axs = plt.subplots(1, 2, figsize=(14, 8), subplot_kw={'projection': ccrs.PlateCarree()}, gridspec_kw={'wspace': 0.05})\n",
    "\n",
    "# Plot for 'withoutoffset' dataset on the left\n",
    "ssh_without = withoutoffset.variables['zeta'][timestep, :].data\n",
    "triang_without = tri.Triangulation(lon, lat, triangles=withoutoffset.variables['element'][:, :] - 1)\n",
    "cf_without = axs[0].tricontourf(triang_without, ssh_without, levels=levels, cmap='viridis', extend='both')\n",
    "axs[0].coastlines()\n",
    "axs[0].add_feature(cfeature.LAND, facecolor='darkgrey')\n",
    "gl_without = axs[0].gridlines(draw_labels=True, xlocs=np.arange(-180, 181, 10), ylocs=np.arange(-90, 91, 10))\n",
    "gl_without.ylabels_right = False\n",
    "gl_without.xlabels_top = False\n",
    "gl_without.xlabel_style = {'size': 12}\n",
    "gl_without.ylabel_style = {'rotation': 90, 'size': 12}\n",
    "axs[0].set_title('Plain Simulation')\n",
    "\n",
    "# Plot for 'withoffset' dataset on the right\n",
    "ssh_with = withoffset.variables['zeta'][timestep, :].data\n",
    "triang_with = tri.Triangulation(lon, lat, triangles=withoffset.variables['element'][:, :] - 1)\n",
    "cf_with = axs[1].tricontourf(triang_with, ssh_with, levels=levels, cmap='viridis', extend='both')\n",
    "axs[1].coastlines()\n",
    "axs[1].add_feature(cfeature.LAND, facecolor='darkgrey')\n",
    "gl_with = axs[1].gridlines(draw_labels=True, xlocs=np.arange(-180, 181, 10), ylocs=np.arange(-90, 91, 10))\n",
    "gl_with.ylabels_left = False\n",
    "gl_with.ylabels_right = False\n",
    "gl_with.xlabels_top = False\n",
    "gl_with.xlabel_style = {'size': 12}\n",
    "axs[1].set_title('Corrected Simulation')\n",
    "\n",
    "# Position the colorbar outside and below the plot boundaries\n",
    "cbar = fig.colorbar(cf_with, ax=axs.ravel().tolist(), orientation='horizontal', pad=0.05, fraction=0.05)\n",
    "cbar.set_label('Water Level (m)', size=20)\n",
    "\n",
    "# Adjust layout\n",
    "plt.tight_layout()\n",
    "\n",
    "# Save the figure\n",
    "plt.savefig(\"water_levels_comparison.png\", dpi=600)\n",
    "\n",
    "# Show the plot\n",
    "plt.show()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "plot",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.18"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}