MeridionalAdvectionDiffusion

General solver of the 1D meridional advection-diffusion equation on the sphere:
for a state variable \(\psi(\phi,t)\), arbitrary diffusivity \(K(\phi)\) in units of \(x^2 ~ t^{-1}\), and advecting velocity \(U(\phi)\). \(\phi\) is latitude and \(a\) is the Earth’s radius (in meters).
\(K\) and \(U\) can be scalars, or optionally vector specified at grid cell boundaries (so their lengths must be exactly 1 greater than the length of \(\phi\)).
\(K\) and \(U\) can be modified by the user at any time (e.g., after each timestep, if they depend on other state variables).
A fully implicit timestep is used for computational efficiency. Thus the computed tendency \(\frac{\partial \psi}{\partial t}\) will depend on the timestep.
In addition to the tendency over the implicit timestep, the solver also calculates several diagnostics from the updated state:
diffusive_fluxgiven by \(-\frac{K(\phi)}{a} ~ \frac{\partial \psi}{\partial \phi}\) in units of \([\psi]~[x]\)/sadvective_fluxgiven by \(U(\phi) \psi(\phi)\) (same units)total_flux, the sum of advective, diffusive and prescribed fluxesflux_convergence(or instantanous scalar tendency) given by the right hand side of the first equation above, in units of \([\psi]\)/s
Non-uniform grid spacing is supported.
The state variable \(\psi\) may be multi-dimensional, but the diffusion will operate along the latitude dimension only.
- class climlab.dynamics.meridional_advection_diffusion.MeridionalAdvectionDiffusion(K=0.0, U=0.0, use_banded_solver=False, prescribed_flux=0.0, **kwargs)[source]
Bases:
AdvectionDiffusionA parent class for meridional advection-diffusion processes.
- Attributes:
- K
- U
- current_time
depthDepth at grid centers (m)
depth_boundsDepth at grid interfaces (m)
diagnosticsDictionary access to all diagnostic variables
- elapsed_time
inputDictionary access to all input variables
latLatitude of grid centers (degrees North)
lat_boundsLatitude of grid interfaces (degrees North)
levPressure levels at grid centers (hPa or mb)
lev_boundsPressure levels at grid interfaces (hPa or mb)
lonLongitude of grid centers (degrees)
lon_boundsLongitude of grid interfaces (degrees)
- prescribed_flux
timestepThe amount of time over which
step_forward()is integrating in unit seconds.timestep_in_secondsReturn a float value representing the timestep in units of seconds
Methods
add_diagnostic(name[, value])Create a new diagnostic variable called
namefor this process and initialize it with the givenvalue.add_input(name[, value])Create a new input variable called
namefor this process and initialize it with the givenvalue.add_subprocess(name, proc[, verbose])Adds a single subprocess to this process.
add_subprocesses(procdict)Adds a dictionary of subproceses to this process.
compute()Computes the tendencies for all state variables given current state and specified input.
compute_diagnostics([num_iter])Compute all tendencies and diagnostics, but don't update model state.
declare_diagnostics(diaglist)Add the variable names in
inputlistto the list of diagnostics.declare_input(inputlist)Add the variable names in
inputlistto the list of necessary inputs.integrate_converge([crit, verbose])Integrates the model until model states are converging.
integrate_days([days, verbose])Integrates the model forward for a specified number of days.
integrate_years([years, verbose])Integrates the model by a given number of years.
remove_diagnostic(name)Removes a diagnostic from the
process.diagnosticdictionary and also delete the associated process attribute.remove_subprocess(name[, verbose])Removes a single subprocess from this process.
set_state(name, value)Sets the variable
nameto a new statevalue.step_forward()Updates state variables with computed tendencies.
to_xarray([diagnostics, timeave])Convert process variables to
xarray.Datasetformat.
- class climlab.dynamics.meridional_advection_diffusion.MeridionalDiffusion(K=0.0, use_banded_solver=False, prescribed_flux=0.0, **kwargs)[source]
Bases:
MeridionalAdvectionDiffusionA parent class for meridional diffusion-only processes, with advection set to zero.
Otherwise identical to the parent class.
- Attributes:
- K
- U
- current_time
depthDepth at grid centers (m)
depth_boundsDepth at grid interfaces (m)
diagnosticsDictionary access to all diagnostic variables
- elapsed_time
inputDictionary access to all input variables
latLatitude of grid centers (degrees North)
lat_boundsLatitude of grid interfaces (degrees North)
levPressure levels at grid centers (hPa or mb)
lev_boundsPressure levels at grid interfaces (hPa or mb)
lonLongitude of grid centers (degrees)
lon_boundsLongitude of grid interfaces (degrees)
- prescribed_flux
timestepThe amount of time over which
step_forward()is integrating in unit seconds.timestep_in_secondsReturn a float value representing the timestep in units of seconds
Methods
add_diagnostic(name[, value])Create a new diagnostic variable called
namefor this process and initialize it with the givenvalue.add_input(name[, value])Create a new input variable called
namefor this process and initialize it with the givenvalue.add_subprocess(name, proc[, verbose])Adds a single subprocess to this process.
add_subprocesses(procdict)Adds a dictionary of subproceses to this process.
compute()Computes the tendencies for all state variables given current state and specified input.
compute_diagnostics([num_iter])Compute all tendencies and diagnostics, but don't update model state.
declare_diagnostics(diaglist)Add the variable names in
inputlistto the list of diagnostics.declare_input(inputlist)Add the variable names in
inputlistto the list of necessary inputs.integrate_converge([crit, verbose])Integrates the model until model states are converging.
integrate_days([days, verbose])Integrates the model forward for a specified number of days.
integrate_years([years, verbose])Integrates the model by a given number of years.
remove_diagnostic(name)Removes a diagnostic from the
process.diagnosticdictionary and also delete the associated process attribute.remove_subprocess(name[, verbose])Removes a single subprocess from this process.
set_state(name, value)Sets the variable
nameto a new statevalue.step_forward()Updates state variables with computed tendencies.
to_xarray([diagnostics, timeave])Convert process variables to
xarray.Datasetformat.