Creating an advanced model setup¶
Note
This guide is still work in progress.
This is a step-by-step guide that illustrates how even complicated setups can be created with relative ease (thanks to the tools provided by the scientific Python community). As an example, we will re-create the wave propagation setup, which is a global ocean model with an idealized Atlantic.
The vision¶
The purpose of this model is to examine wave propagation along the eastern boundary of the North Atlantic. Since it is difficult to track propagating waves along ragged geometry or through uneven forcing fields, we will idealize the representation of the North Atlantic; and as the presence of the Pacific in the model is crucial to achieve a realistic ocean circulation, we want to use a global model.
This leaves us with the following requirements for the final wave propagation model:
- A global model with a resolution of around 2 degrees and meridional stretching.
- Convert the eastern boundary of the Atlantic to a straight line, so analytically derived wave properties hold.
- A refined grid resolution at the eastern boundary of the Atlantic.
- Zonally averaged forcings in the Atlantic.
- A somehow interpolated initial state for cells that have been converted from land to ocean in the North Atlantic.
- Options for shelf and continental slope.
- A multiplier setting for the Southern Ocean wind stress.
Model skeleton¶
Instead of starting from scratch, we can use the global one degree model as a template, which looks like this:
#!/usr/bin/env python
import os
from netCDF4 import Dataset
from veros import Veros, veros_method, time, tools
BASE_PATH = os.path.dirname(os.path.realpath(__file__))
FORCING_FILE = os.path.join(BASE_PATH, "forcing_1deg_global.nc")
class GlobalOneDegree(Veros):
"""Global 1 degree model with 115 vertical levels.
`Adapted from pyOM2 <https://wiki.zmaw.de/ifm/TO/pyOM2/1x1%20global%20model>`_.
"""
@veros_method
def set_parameter(self):
"""
set main parameters
"""
if not os.path.isfile(FORCING_FILE):
raise RuntimeError("{} data file {} not found".format(name, filepath))
self.nx = 360
self.ny = 160
self.nz = 115
self.dt_mom = 1800.0
self.dt_tracer = 1800.0
self.runlen = 0.
self.coord_degree = True
self.enable_cyclic_x = True
self.congr_epsilon = 1e-10
self.congr_max_iterations = 10000
self.enable_hor_friction = True
self.A_h = 5e4
self.enable_hor_friction_cos_scaling = True
self.hor_friction_cosPower = 1
self.enable_tempsalt_sources = True
self.enable_implicit_vert_friction = True
self.eq_of_state_type = 5
# isoneutral
self.enable_neutral_diffusion = True
self.K_iso_0 = 1000.0
self.K_iso_steep = 50.0
self.iso_dslope = 0.005
self.iso_slopec = 0.005
self.enable_skew_diffusion = True
# tke
self.enable_tke = True
self.c_k = 0.1
self.c_eps = 0.7
self.alpha_tke = 30.0
self.mxl_min = 1e-8
self.tke_mxl_choice = 2
self.enable_tke_superbee_advection = True
# eke
self.enable_eke = True
self.eke_k_max = 1e4
self.eke_c_k = 0.4
self.eke_c_eps = 0.5
self.eke_cross = 2.
self.eke_crhin = 1.0
self.eke_lmin = 100.0
self.enable_eke_superbee_advection = True
self.enable_eke_isopycnal_diffusion = True
# idemix
self.enable_idemix = True
self.enable_eke_diss_surfbot = True
self.eke_diss_surfbot_frac = 0.2
self.enable_idemix_superbee_advection = True
self.enable_idemix_hor_diffusion = True
@veros_method
def _read_forcing(self, var):
with Dataset(FORCING_FILE, "r") as infile:
return np.array(infile.variables[var][...]).T
@veros_method
def set_grid(self):
dz_data = self._read_forcing("dz")
self.dzt[...] = dz_data[::-1]
self.dxt[...] = 1.0
self.dyt[...] = 1.0
self.y_origin = -79.
self.x_origin = 91.
@veros_method
def set_coriolis(self):
self.coriolis_t[...] = 2 * self.omega * np.sin(self.yt[np.newaxis, :] / 180. * self.pi)
@veros_method
def set_topography(self):
bathymetry_data = self._read_forcing("bathymetry")
salt_data = self._read_forcing("salinity")[:, :, ::-1]
mask_salt = salt_data == 0.
self.kbot[2:-2, 2:-2] = 1 + np.sum(mask_salt.astype(np.int), axis=2)
mask_bathy = bathymetry_data == 0
self.kbot[2:-2, 2:-2][mask_bathy] = 0
self.kbot[self.kbot >= self.nz] = 0
# close some channels
i, j = np.indices((self.nx, self.ny))
mask_channel = (i >= 207) & (i < 214) & (j < 5) # i = 208,214; j = 1,5
self.kbot[2:-2, 2:-2][mask_channel] = 0
# Aleutian islands
mask_channel = (i == 104) & (j == 134) # i = 105; j = 135
self.kbot[2:-2, 2:-2][mask_channel] = 0
# Engl channel
mask_channel = (i >= 269) & (i < 271) & (j == 130) # i = 270,271; j = 131
self.kbot[2:-2, 2:-2][mask_channel] = 0
@veros_method
def set_initial_conditions(self):
self.t_star = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
self.s_star = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
self.qnec = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
self.qnet = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
self.qsol = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
self.divpen_shortwave = np.zeros(self.nz, dtype=self.default_float_type)
self.taux = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
self.tauy = np.zeros((self.nx + 4, self.ny + 4, 12), dtype=self.default_float_type)
rpart_shortwave = 0.58
efold1_shortwave = 0.35
efold2_shortwave = 23.0
# initial conditions
temp_data = self._read_forcing("temperature")
self.temp[2:-2, 2:-2, :, 0] = temp_data[..., ::-1] * self.maskT[2:-2, 2:-2, :]
self.temp[2:-2, 2:-2, :, 1] = temp_data[..., ::-1] * self.maskT[2:-2, 2:-2, :]
salt_data = self._read_forcing("salinity")
self.salt[2:-2, 2:-2, :, 0] = salt_data[..., ::-1] * self.maskT[2:-2, 2:-2, :]
self.salt[2:-2, 2:-2, :, 1] = salt_data[..., ::-1] * self.maskT[2:-2, 2:-2, :]
# wind stress on MIT grid
taux_data = self._read_forcing("tau_x")
self.taux[2:-2, 2:-2, :] = taux_data / self.rho_0
tauy_data = self._read_forcing("tau_y")
self.tauy[2:-2, 2:-2, :] = tauy_data / self.rho_0
# Qnet and dQ/dT and Qsol
qnet_data = self._read_forcing("q_net")
self.qnet[2:-2, 2:-2, :] = -qnet_data * self.maskT[2:-2, 2:-2, -1, np.newaxis]
qnec_data = self._read_forcing("dqdt")
self.qnec[2:-2, 2:-2, :] = qnec_data * self.maskT[2:-2, 2:-2, -1, np.newaxis]
qsol_data = self._read_forcing("swf")
self.qsol[2:-2, 2:-2, :] = -qsol_data * self.maskT[2:-2, 2:-2, -1, np.newaxis]
# SST and SSS
sst_data = self._read_forcing("sst")
self.t_star[2:-2, 2:-2, :] = sst_data * self.maskT[2:-2, 2:-2, -1, np.newaxis]
sss_data = self._read_forcing("sss")
self.s_star[2:-2, 2:-2, :] = sss_data * self.maskT[2:-2, 2:-2, -1, np.newaxis]
if self.enable_idemix:
tidal_energy_data = self._read_forcing("tidal_energy")
mask = np.maximum(0, self.kbot[2:-2, 2:-2] - 1)[:, :, np.newaxis] == np.arange(self.nz)[np.newaxis, np.newaxis, :]
tidal_energy_data[:, :] *= self.maskW[2:-2, 2:-2, :][mask].reshape(self.nx, self.ny) / self.rho_0
self.forc_iw_bottom[2:-2, 2:-2] = tidal_energy_data
wind_energy_data = self._read_forcing("wind_energy")
wind_energy_data[:, :] *= self.maskW[2:-2, 2:-2, -1] / self.rho_0 * 0.2
self.forc_iw_surface[2:-2, 2:-2] = wind_energy_data
"""
Initialize penetration profile for solar radiation and store divergence in divpen
note that pen is set to 0.0 at the surface instead of 1.0 to compensate for the
shortwave part of the total surface flux
"""
swarg1 = self.zw / efold1_shortwave
swarg2 = self.zw / efold2_shortwave
pen = rpart_shortwave * np.exp(swarg1) + (1.0 - rpart_shortwave) * np.exp(swarg2)
pen[-1] = 0.
self.divpen_shortwave = np.zeros(self.nz, dtype=self.default_float_type)
self.divpen_shortwave[1:] = (pen[1:] - pen[:-1]) / self.dzt[1:]
self.divpen_shortwave[0] = pen[0] / self.dzt[0]
@veros_method
def set_forcing(self):
t_rest = 30. * 86400.
cp_0 = 3991.86795711963 # J/kg /K
year_in_seconds = time.convert_time(self, 1., "years", "seconds")
(n1, f1), (n2, f2) = tools.get_periodic_interval(self.time, year_in_seconds,
year_in_seconds / 12., 12)
# linearly interpolate wind stress and shift from MITgcm U/V grid to this grid
self.surface_taux[:-1, :] = f1 * self.taux[1:, :, n1] + f2 * self.taux[1:, :, n2]
self.surface_tauy[:, :-1] = f1 * self.tauy[:, 1:, n1] + f2 * self.tauy[:, 1:, n2]
if self.enable_tke:
self.forc_tke_surface[1:-1, 1:-1] = np.sqrt((0.5 * (self.surface_taux[1:-1, 1:-1] \
+ self.surface_taux[:-2, 1:-1])) ** 2 \
+ (0.5 * (self.surface_tauy[1:-1, 1:-1] \
+ self.surface_tauy[1:-1, :-2])) ** 2) ** (3. / 2.)
# W/m^2 K kg/J m^3/kg = K m/s
t_star_cur = f1 * self.t_star[..., n1] + f2 * self.t_star[..., n2]
self.qqnec = f1 * self.qnec[..., n1] + f2 * self.qnec[..., n2]
self.qqnet = f1 * self.qnet[..., n1] + f2 * self.qnet[..., n2]
self.forc_temp_surface[...] = (self.qqnet + self.qqnec * (t_star_cur - self.temp[..., -1, self.tau])) \
* self.maskT[..., -1] / cp_0 / self.rho_0
s_star_cur = f1 * self.s_star[..., n1] + f2 * self.s_star[..., n2]
self.forc_salt_surface[...] = 1. / t_rest * \
(s_star_cur - self.salt[..., -1, self.tau]) * self.maskT[..., -1] * self.dzt[-1]
# apply simple ice mask
mask1 = self.temp[:, :, -1, self.tau] * self.maskT[:, :, -1] <= -1.8
mask2 = self.forc_temp_surface <= 0
ice = ~(mask1 & mask2)
self.forc_temp_surface *= ice
self.forc_salt_surface *= ice
# solar radiation
if self.enable_tempsalt_sources:
self.temp_source[..., :] = (f1 * self.qsol[..., n1, None] + f2 * self.qsol[..., n2, None]) \
* self.divpen_shortwave[None, None, :] * ice[..., None] \
* self.maskT[..., :] / cp_0 / self.rho_0
@veros_method
def set_diagnostics(self):
average_vars = ["surface_taux", "surface_tauy", "forc_temp_surface", "forc_salt_surface",
"psi", "temp", "salt", "u", "v", "w", "Nsqr", "Hd", "rho",
"K_diss_v", "P_diss_v", "P_diss_nonlin", "P_diss_iso", "kappaH"]
if self.enable_skew_diffusion:
average_vars += ["B1_gm", "B2_gm"]
if self.enable_TEM_friction:
average_vars += ["kappa_gm", "K_diss_gm"]
if self.enable_tke:
average_vars += ["tke", "Prandtlnumber", "mxl", "tke_diss",
"forc_tke_surface", "tke_surf_corr"]
if self.enable_idemix:
average_vars += ["E_iw", "forc_iw_surface", "forc_iw_bottom", "iw_diss",
"c0", "v0"]
if self.enable_eke:
average_vars += ["eke", "K_gm", "L_rossby", "L_rhines"]
self.diagnostics["averages"].output_variables = average_vars
self.diagnostics["cfl_monitor"].output_frequency = 86400.0
self.diagnostics["snapshot"].output_frequency = 365 * 86400 / 24.
self.diagnostics["overturning"].output_frequency = 365 * 86400
self.diagnostics["overturning"].sampling_frequency = 365 * 86400 / 24.
self.diagnostics["energy"].output_frequency = 365 * 86400
self.diagnostics["energy"].sampling_frequency = 365 * 86400 / 24.
self.diagnostics["averages"].output_frequency = 365 * 86400
self.diagnostics["averages"].sampling_frequency = 365 * 86400 / 24.
if __name__ == "__main__":
simulation = GlobalOneDegree()
simulation.setup()
simulation.run()
The biggest changes in the new wave propgation setup will be located in the
set_grid()
set_topography()
and set_initial_conditions()
methods to accomodate for the new geometry and the interpolation of initial
conditions to the modified grid, so we can concentrate on implementing those
first.
Step 1: Setup grid¶
Warning
When using a non-uniform grid,
Step 2: Create idealized topography¶
Usually, to create an idealized topography, one would simply hand-craft some input and forcing files that reflect the desired changes. However, since we want our setup to have flexible resolution, we will have to write an algorithm that creates these input files for any given number of grid cells. One convenient way to achieve this is by creating some high-resolution masks representing the target topography by hand, and then interpolate these masks to the desired resolution.
Create a mask image¶
Before we can start, we need to download a high-resolution topography dataset. There are many freely available topographical data sets on the internet; one of them is ETOPO5 (with a resolution of 5 arc-minutes), which we will be using throughout this tutorial. To create a mask image from the topography file, you can use the command line tool veros create-mask, e.g. like
$ veros create-mask ETOPO5_Ice_g_gmt4.nc
This creates a one-to-one representation of the topography file as a PNG image. However, in the case of the 5 arc-minute topography, the resulting image includes a lot of small islands and complicated coastlines that might cause problems when being interpolated to a numerical grid with a much lower resolution. To address this, the create-mask script accepts a scale argument. When given, a Gaussian filter with standard deviation scale (in grid cells) is applied to the resulting image, smoothing out small features. The command
$ veros create-mask ETOPO5_Ice_g_gmt4 --scale 3 3
results in the following mask:
which looks good enough to serve as a basis for horizontal resolutions of around one degree.
Modify the mask¶
We can now proceed to mold this realistic version of the global topography into the desired idealized shape. You can use any image editor you have availble; one possibility is the free software GIMP. Inside the editor, we can use the pencil tools to create a modified version of the topography mask:
In this modified version, I have
- replaced the eastern boundary of the North Atlantic by a meridional line;
- removed all lakes and inland seas;
- thickened Central America (to prevent North and South America to become disconnected due to interpolation artifacts); and
- removed the Arctic Ocean and Hudson Bay.
Now that our topography mask is finished, we can go ahead and implement it in the Veros setup!
Import to Veros¶
To read the mask in PNG format, we are going to use the Python Imaging Library (PIL).