GitHub

One dimensional column forced by external data with carbonate chemistry

In this example we setup a simple 1D column with the LOBSTER biogeochemical model and observe its evolution. This example demonstrates:

  • How to setup OceanBioME's biogeochemical models
  • How to load external forcing data
  • How to run with optional tracer sets such as carbonate chemistry
  • How to setup a non-uniform grid for better near surface resolution
  • How to visualise results

For this example we use force by mixing layer depth and surface photosynthetically available radiation (PAR) data from the Mercator Ocean model and NASA VIIRS observations.

Install dependencies

First we check we have the dependencies installed

using Pkg
pkg"add OceanBioME, Oceananigans, NetCDF, Interpolations, DataDeps, CairoMakie"

Model setup

First load the required packages

using OceanBioME
using Oceananigans, Random, Printf, NetCDF, Interpolations, DataDeps
using Oceananigans.Units
using Oceananigans.Fields: FunctionField

import Oceananigans.TurbulenceClosures: maximum_numeric_diffusivity

maximum_numeric_diffusivity(κ::NamedTuple) = maximum(maximum.(values(κ)))
maximum_numeric_diffusivity(κ::FunctionField) = maximum(κ)

const year = years = 365days # just for these idealised cases

Load external forcing data

Loading the forcing data from our online copy

dd = DataDep(
    "example_data",
    "example data from subpolar re-analysis and observational products",
    "https://github.com/OceanBioME/OceanBioME_example_data/raw/main/subpolar.nc"
)
register(dd)
filename = datadep"example_data/subpolar.nc"
times = ncread(filename, "time")
temp = ncread(filename, "temp")
salinity = ncread(filename, "so")
mld = ncread(filename, "mld")
par = ncread(filename, "par")

temperature_itp = LinearInterpolation(times, temp)
salinity_itp = LinearInterpolation(times, salinity)
mld_itp = LinearInterpolation(times, mld)
PAR_itp = LinearInterpolation(times, par)

t_function(x, y, z, t) = temperature_itp(mod(t, 364days))
s_function(x, y, z, t) = salinity_itp(mod(t, 364days))
surface_PAR(x, y, t) = PAR_itp(mod(t, 364days))
κₜ(x, y, z, t) = 2e-2 * max(1 - (z + mld_itp(mod(t, 364days)) / 2)^2 / (mld_itp(mod(t, 364days)) / 2)^2, 0) + 1e-4

Grid and diffusivity field

Define the grid (in this case a non uniform grid for better resolution near the surface) and an extra Oceananigans field for the PAR to be stored in

Nz = 33
Lz = 600meters
refinement = 10
stretching = 5.754
h(k) = (k - 1) / Nz
ζ₀(k) = 1 + (h(k) - 1) / refinement
Σ(k) = (1 - exp(-stretching * h(k))) / (1 - exp(-stretching))
z_faces(k) = Lz * (ζ₀(k) * Σ(k) - 1)

grid = RectilinearGrid(size = (1, 1, Nz), x = (0, 20meters), y = (0, 20meters), z = z_faces)

clock = Clock(; time = 0.0)

κ = FunctionField{Center, Center, Center}(κₜ, grid; clock)
FunctionField located at (Center, Center, Center)
├── func: κₜ (generic function with 1 method)
├── grid: 1×1×33 RectilinearGrid{Float64, Oceananigans.Grids.Periodic, Oceananigans.Grids.Periodic, Oceananigans.Grids.Bounded} on Oceananigans.Architectures.CPU with 1×1×3 halo
├── clock: Clock{Float64, Float64}(time=0 seconds, iteration=0, last_Δt=Inf days)
└── parameters: nothing

Biogeochemical and Oceananigans model

Here we instantiate the LOBSTER model with carbonate chemistry and a surface flux of DIC (CO₂)

biogeochemistry = LOBSTER(; grid,
                            surface_photosynthetically_active_radiation = surface_PAR,
                            carbonates = true,
                            scale_negatives = true)

CO₂_flux = CarbonDioxideGasExchangeBoundaryCondition()

T = FunctionField{Center, Center, Center}(t_function, grid; clock)
S = FunctionField{Center, Center, Center}(s_function, grid; clock)

model = NonhydrostaticModel(; grid, clock,
                              closure = ScalarDiffusivity(ν = κ, κ = κ),
                              biogeochemistry,
                              boundary_conditions = (DIC = FieldBoundaryConditions(top = CO₂_flux),),
                              auxiliary_fields = (; T, S))

set!(model, P = 0.03, Z = 0.03, NO₃ = 11.0, NH₄ = 0.05, DIC = 2200.0, Alk = 2400.0)

Simulation

Next we setup the simulation along with some callbacks that:

  • Show the progress of the simulation
  • Store the output
  • Prevent the tracers from going negative from numerical error (see discussion of this in the positivity preservation implementation page)
  • Adapt the timestep length to reduce the run time
simulation = Simulation(model, Δt = 1minutes, stop_time = 100days)

progress_message(sim) = @printf("Iteration: %04d, time: %s, Δt: %s, wall time: %s\n",
                                iteration(sim),
                                prettytime(sim),
                                prettytime(sim.Δt),
                                prettytime(sim.run_wall_time))

simulation.callbacks[:progress] = Callback(progress_message, IterationInterval(500))

filename = "data_forced"
simulation.output_writers[:profiles] = JLD2Writer(model,
                                                  model.tracers,
                                                  filename = "$filename.jld2",
                                                  schedule = TimeInterval(1day),
                                                  overwrite_existing = true)

wizard = TimeStepWizard(cfl = 0.2, diffusive_cfl = 0.2,
                        max_change = 1.5, min_change = 0.75,
                        cell_advection_timescale = column_advection_timescale)

simulation.callbacks[:wizard] = Callback(wizard, IterationInterval(10))
┌ Warning: Attempting to store typeof(Main.var"##234".κₜ).
│ JLD2 only stores functions by name.
│  This may not be useful for anonymous functions.
└ @ JLD2 ~/.julia/packages/JLD2/SgtOb/src/data/writing_datatypes.jl:447
┌ Warning: Attempting to store typeof(Main.var"##234".κₜ).
│ JLD2 only stores functions by name.
│  This may not be useful for anonymous functions.
└ @ JLD2 ~/.julia/packages/JLD2/SgtOb/src/data/writing_datatypes.jl:447

Run!

We are ready to run the simulation

run!(simulation)
[ Info: Initializing simulation...
Iteration: 0000, time: 0 seconds, Δt: 1 minute, wall time: 0 seconds
[ Info:     ... simulation initialization complete (31.346 seconds)
[ Info: Executing initial time step...
[ Info:     ... initial time step complete (12.278 seconds).
Iteration: 0500, time: 6.598 hours, Δt: 47.460 seconds, wall time: 47.281 seconds
Iteration: 1000, time: 13.187 hours, Δt: 47.420 seconds, wall time: 50.681 seconds
Iteration: 1500, time: 19.771 hours, Δt: 47.386 seconds, wall time: 54.131 seconds
Iteration: 2000, time: 1.098 days, Δt: 47.380 seconds, wall time: 57.519 seconds
Iteration: 2500, time: 1.372 days, Δt: 47.427 seconds, wall time: 1.013 minutes
Iteration: 3000, time: 1.647 days, Δt: 47.487 seconds, wall time: 1.070 minutes
Iteration: 3500, time: 1.921 days, Δt: 47.481 seconds, wall time: 1.125 minutes
Iteration: 4000, time: 2.196 days, Δt: 47.312 seconds, wall time: 1.183 minutes
Iteration: 4500, time: 2.470 days, Δt: 47.360 seconds, wall time: 1.240 minutes
Iteration: 5000, time: 2.744 days, Δt: 47.449 seconds, wall time: 1.298 minutes
Iteration: 5500, time: 3.018 days, Δt: 47.405 seconds, wall time: 1.356 minutes
Iteration: 6000, time: 3.292 days, Δt: 47.445 seconds, wall time: 1.410 minutes
Iteration: 6500, time: 3.566 days, Δt: 47.337 seconds, wall time: 1.469 minutes
Iteration: 7000, time: 3.841 days, Δt: 47.312 seconds, wall time: 1.526 minutes
Iteration: 7500, time: 4.115 days, Δt: 47.427 seconds, wall time: 1.585 minutes
Iteration: 8000, time: 4.389 days, Δt: 47.347 seconds, wall time: 1.640 minutes
Iteration: 8500, time: 4.663 days, Δt: 47.314 seconds, wall time: 1.696 minutes
Iteration: 9000, time: 4.937 days, Δt: 47.320 seconds, wall time: 1.751 minutes
Iteration: 9500, time: 5.210 days, Δt: 47.335 seconds, wall time: 1.807 minutes
Iteration: 10000, time: 5.484 days, Δt: 47.349 seconds, wall time: 1.862 minutes
Iteration: 10500, time: 5.758 days, Δt: 47.366 seconds, wall time: 1.918 minutes
Iteration: 11000, time: 6.032 days, Δt: 47.388 seconds, wall time: 1.976 minutes
Iteration: 11500, time: 6.307 days, Δt: 47.442 seconds, wall time: 2.032 minutes
Iteration: 12000, time: 6.581 days, Δt: 47.510 seconds, wall time: 2.090 minutes
Iteration: 12500, time: 6.856 days, Δt: 47.494 seconds, wall time: 2.147 minutes
Iteration: 13000, time: 7.131 days, Δt: 47.500 seconds, wall time: 2.205 minutes
Iteration: 13500, time: 7.406 days, Δt: 47.481 seconds, wall time: 2.260 minutes
Iteration: 14000, time: 7.680 days, Δt: 47.401 seconds, wall time: 2.314 minutes
Iteration: 14500, time: 7.954 days, Δt: 47.345 seconds, wall time: 2.366 minutes
Iteration: 15000, time: 8.228 days, Δt: 47.337 seconds, wall time: 2.416 minutes
Iteration: 15500, time: 8.502 days, Δt: 47.337 seconds, wall time: 2.466 minutes
Iteration: 16000, time: 8.776 days, Δt: 47.337 seconds, wall time: 2.516 minutes
Iteration: 16500, time: 9.049 days, Δt: 47.337 seconds, wall time: 2.565 minutes
Iteration: 17000, time: 9.323 days, Δt: 47.337 seconds, wall time: 2.615 minutes
Iteration: 17500, time: 9.597 days, Δt: 47.337 seconds, wall time: 2.666 minutes
Iteration: 18000, time: 9.871 days, Δt: 47.337 seconds, wall time: 2.717 minutes
Iteration: 18500, time: 10.145 days, Δt: 47.337 seconds, wall time: 2.770 minutes
Iteration: 19000, time: 10.419 days, Δt: 47.337 seconds, wall time: 2.820 minutes
Iteration: 19500, time: 10.693 days, Δt: 47.337 seconds, wall time: 2.870 minutes
Iteration: 20000, time: 10.966 days, Δt: 47.337 seconds, wall time: 2.920 minutes
Iteration: 20500, time: 11.240 days, Δt: 47.338 seconds, wall time: 2.971 minutes
Iteration: 21000, time: 11.514 days, Δt: 47.340 seconds, wall time: 3.021 minutes
Iteration: 21500, time: 11.788 days, Δt: 47.341 seconds, wall time: 3.071 minutes
Iteration: 22000, time: 12.061 days, Δt: 47.330 seconds, wall time: 3.122 minutes
Iteration: 22500, time: 12.335 days, Δt: 47.314 seconds, wall time: 3.172 minutes
Iteration: 23000, time: 12.609 days, Δt: 47.376 seconds, wall time: 3.222 minutes
Iteration: 23500, time: 12.884 days, Δt: 47.531 seconds, wall time: 3.271 minutes
Iteration: 24000, time: 13.158 days, Δt: 47.431 seconds, wall time: 3.323 minutes
Iteration: 24500, time: 13.432 days, Δt: 47.385 seconds, wall time: 3.372 minutes
Iteration: 25000, time: 13.707 days, Δt: 47.349 seconds, wall time: 3.422 minutes
Iteration: 25500, time: 13.981 days, Δt: 47.325 seconds, wall time: 3.472 minutes
Iteration: 26000, time: 14.254 days, Δt: 47.332 seconds, wall time: 3.523 minutes
Iteration: 26500, time: 14.528 days, Δt: 47.345 seconds, wall time: 3.573 minutes
Iteration: 27000, time: 14.802 days, Δt: 47.360 seconds, wall time: 3.622 minutes
Iteration: 27500, time: 15.076 days, Δt: 47.408 seconds, wall time: 3.675 minutes
Iteration: 28000, time: 15.351 days, Δt: 47.467 seconds, wall time: 3.725 minutes
Iteration: 28500, time: 15.625 days, Δt: 47.334 seconds, wall time: 3.776 minutes
Iteration: 29000, time: 15.899 days, Δt: 47.317 seconds, wall time: 3.826 minutes
Iteration: 29500, time: 16.173 days, Δt: 47.337 seconds, wall time: 3.876 minutes
Iteration: 30000, time: 16.447 days, Δt: 47.337 seconds, wall time: 3.926 minutes
Iteration: 30500, time: 16.720 days, Δt: 47.337 seconds, wall time: 3.976 minutes
Iteration: 31000, time: 16.994 days, Δt: 47.337 seconds, wall time: 4.025 minutes
Iteration: 31500, time: 17.268 days, Δt: 47.333 seconds, wall time: 4.076 minutes
Iteration: 32000, time: 17.542 days, Δt: 47.329 seconds, wall time: 4.129 minutes
Iteration: 32500, time: 17.816 days, Δt: 47.326 seconds, wall time: 4.179 minutes
Iteration: 33000, time: 18.089 days, Δt: 47.323 seconds, wall time: 4.230 minutes
Iteration: 33500, time: 18.363 days, Δt: 47.321 seconds, wall time: 4.280 minutes
Iteration: 34000, time: 18.637 days, Δt: 47.319 seconds, wall time: 4.330 minutes
Iteration: 34500, time: 18.911 days, Δt: 47.318 seconds, wall time: 4.380 minutes
Iteration: 35000, time: 19.185 days, Δt: 47.320 seconds, wall time: 4.434 minutes
Iteration: 35500, time: 19.458 days, Δt: 47.325 seconds, wall time: 4.485 minutes
Iteration: 36000, time: 19.732 days, Δt: 47.330 seconds, wall time: 4.536 minutes
Iteration: 36500, time: 20.006 days, Δt: 47.337 seconds, wall time: 4.590 minutes
Iteration: 37000, time: 20.280 days, Δt: 47.337 seconds, wall time: 4.642 minutes
Iteration: 37500, time: 20.554 days, Δt: 47.337 seconds, wall time: 4.692 minutes
Iteration: 38000, time: 20.828 days, Δt: 47.337 seconds, wall time: 4.742 minutes
Iteration: 38500, time: 21.101 days, Δt: 47.337 seconds, wall time: 4.793 minutes
Iteration: 39000, time: 21.375 days, Δt: 47.337 seconds, wall time: 4.844 minutes
Iteration: 39500, time: 21.649 days, Δt: 47.337 seconds, wall time: 4.898 minutes
Iteration: 40000, time: 21.923 days, Δt: 47.337 seconds, wall time: 4.952 minutes
Iteration: 40500, time: 22.197 days, Δt: 47.335 seconds, wall time: 5.007 minutes
Iteration: 41000, time: 22.471 days, Δt: 47.332 seconds, wall time: 5.059 minutes
Iteration: 41500, time: 22.745 days, Δt: 47.329 seconds, wall time: 5.111 minutes
Iteration: 42000, time: 23.018 days, Δt: 47.328 seconds, wall time: 5.164 minutes
Iteration: 42500, time: 23.292 days, Δt: 47.400 seconds, wall time: 5.214 minutes
Iteration: 43000, time: 23.567 days, Δt: 47.522 seconds, wall time: 5.263 minutes
Iteration: 43500, time: 23.842 days, Δt: 47.420 seconds, wall time: 5.314 minutes
Iteration: 44000, time: 24.116 days, Δt: 47.336 seconds, wall time: 5.365 minutes
Iteration: 44500, time: 24.389 days, Δt: 47.312 seconds, wall time: 5.415 minutes
Iteration: 45000, time: 24.663 days, Δt: 47.345 seconds, wall time: 5.466 minutes
Iteration: 45500, time: 24.938 days, Δt: 47.428 seconds, wall time: 5.523 minutes
Iteration: 46000, time: 25.211 days, Δt: 47.461 seconds, wall time: 5.578 minutes
Iteration: 46500, time: 25.486 days, Δt: 47.468 seconds, wall time: 5.628 minutes
Iteration: 47000, time: 25.761 days, Δt: 47.475 seconds, wall time: 5.678 minutes
Iteration: 47500, time: 26.035 days, Δt: 47.482 seconds, wall time: 5.729 minutes
Iteration: 48000, time: 26.310 days, Δt: 47.481 seconds, wall time: 5.780 minutes
Iteration: 48500, time: 26.585 days, Δt: 47.481 seconds, wall time: 5.830 minutes
Iteration: 49000, time: 26.859 days, Δt: 47.480 seconds, wall time: 5.880 minutes
Iteration: 49500, time: 27.134 days, Δt: 47.481 seconds, wall time: 5.931 minutes
Iteration: 50000, time: 27.409 days, Δt: 47.483 seconds, wall time: 5.981 minutes
Iteration: 50500, time: 27.684 days, Δt: 47.485 seconds, wall time: 6.031 minutes
Iteration: 51000, time: 27.958 days, Δt: 47.487 seconds, wall time: 6.081 minutes
Iteration: 51500, time: 28.233 days, Δt: 47.489 seconds, wall time: 6.132 minutes
Iteration: 52000, time: 28.508 days, Δt: 47.491 seconds, wall time: 6.182 minutes
Iteration: 52500, time: 28.783 days, Δt: 47.492 seconds, wall time: 6.232 minutes
Iteration: 53000, time: 29.057 days, Δt: 47.497 seconds, wall time: 6.284 minutes
Iteration: 53500, time: 29.332 days, Δt: 47.515 seconds, wall time: 6.334 minutes
Iteration: 54000, time: 29.607 days, Δt: 47.533 seconds, wall time: 6.385 minutes
Iteration: 54500, time: 29.882 days, Δt: 47.550 seconds, wall time: 6.435 minutes
Iteration: 55000, time: 30.157 days, Δt: 47.502 seconds, wall time: 6.488 minutes
Iteration: 55500, time: 30.432 days, Δt: 47.443 seconds, wall time: 6.539 minutes
Iteration: 56000, time: 30.706 days, Δt: 47.396 seconds, wall time: 6.589 minutes
Iteration: 56500, time: 30.981 days, Δt: 47.360 seconds, wall time: 6.639 minutes
Iteration: 57000, time: 31.254 days, Δt: 47.358 seconds, wall time: 6.691 minutes
Iteration: 57500, time: 31.528 days, Δt: 47.359 seconds, wall time: 6.743 minutes
Iteration: 58000, time: 31.802 days, Δt: 47.360 seconds, wall time: 6.794 minutes
Iteration: 58500, time: 32.076 days, Δt: 47.372 seconds, wall time: 6.846 minutes
Iteration: 59000, time: 32.350 days, Δt: 47.426 seconds, wall time: 6.896 minutes
Iteration: 59500, time: 32.625 days, Δt: 47.498 seconds, wall time: 6.947 minutes
Iteration: 60000, time: 32.900 days, Δt: 47.519 seconds, wall time: 6.997 minutes
Iteration: 60500, time: 33.174 days, Δt: 47.480 seconds, wall time: 7.048 minutes
Iteration: 61000, time: 33.449 days, Δt: 47.466 seconds, wall time: 7.099 minutes
Iteration: 61500, time: 33.724 days, Δt: 47.451 seconds, wall time: 7.149 minutes
Iteration: 62000, time: 33.998 days, Δt: 47.438 seconds, wall time: 7.200 minutes
Iteration: 62500, time: 34.272 days, Δt: 47.454 seconds, wall time: 7.252 minutes
Iteration: 63000, time: 34.547 days, Δt: 47.471 seconds, wall time: 7.304 minutes
Iteration: 63500, time: 34.822 days, Δt: 47.489 seconds, wall time: 7.355 minutes
Iteration: 64000, time: 35.096 days, Δt: 47.434 seconds, wall time: 7.408 minutes
Iteration: 64500, time: 35.370 days, Δt: 47.317 seconds, wall time: 7.459 minutes
Iteration: 65000, time: 35.644 days, Δt: 47.361 seconds, wall time: 7.511 minutes
Iteration: 65500, time: 35.919 days, Δt: 47.488 seconds, wall time: 7.563 minutes
Iteration: 66000, time: 36.193 days, Δt: 47.399 seconds, wall time: 7.618 minutes
Iteration: 66500, time: 36.467 days, Δt: 47.376 seconds, wall time: 7.671 minutes
Iteration: 67000, time: 36.741 days, Δt: 47.357 seconds, wall time: 7.723 minutes
Iteration: 67500, time: 37.015 days, Δt: 47.344 seconds, wall time: 7.776 minutes
Iteration: 68000, time: 37.290 days, Δt: 47.499 seconds, wall time: 7.827 minutes
Iteration: 68500, time: 37.564 days, Δt: 47.383 seconds, wall time: 7.878 minutes
Iteration: 69000, time: 37.838 days, Δt: 47.312 seconds, wall time: 7.930 minutes
Iteration: 69500, time: 38.112 days, Δt: 47.336 seconds, wall time: 7.983 minutes
Iteration: 70000, time: 38.386 days, Δt: 47.367 seconds, wall time: 8.034 minutes
Iteration: 70500, time: 38.660 days, Δt: 47.410 seconds, wall time: 8.085 minutes
Iteration: 71000, time: 38.935 days, Δt: 47.464 seconds, wall time: 8.136 minutes
Iteration: 71500, time: 39.209 days, Δt: 47.480 seconds, wall time: 8.187 minutes
Iteration: 72000, time: 39.484 days, Δt: 47.480 seconds, wall time: 8.238 minutes
Iteration: 72500, time: 39.758 days, Δt: 47.480 seconds, wall time: 8.288 minutes
Iteration: 73000, time: 40.033 days, Δt: 47.480 seconds, wall time: 8.340 minutes
Iteration: 73500, time: 40.308 days, Δt: 47.480 seconds, wall time: 8.390 minutes
Iteration: 74000, time: 40.583 days, Δt: 47.480 seconds, wall time: 8.442 minutes
Iteration: 74500, time: 40.857 days, Δt: 47.480 seconds, wall time: 8.493 minutes
Iteration: 75000, time: 41.132 days, Δt: 47.480 seconds, wall time: 8.547 minutes
Iteration: 75500, time: 41.407 days, Δt: 47.480 seconds, wall time: 8.600 minutes
Iteration: 76000, time: 41.681 days, Δt: 47.480 seconds, wall time: 8.652 minutes
Iteration: 76500, time: 41.956 days, Δt: 47.480 seconds, wall time: 8.704 minutes
Iteration: 77000, time: 42.231 days, Δt: 47.480 seconds, wall time: 8.758 minutes
Iteration: 77500, time: 42.506 days, Δt: 47.480 seconds, wall time: 8.811 minutes
Iteration: 78000, time: 42.780 days, Δt: 47.480 seconds, wall time: 8.864 minutes
Iteration: 78500, time: 43.055 days, Δt: 47.480 seconds, wall time: 8.919 minutes
Iteration: 79000, time: 43.330 days, Δt: 47.480 seconds, wall time: 8.971 minutes
Iteration: 79500, time: 43.604 days, Δt: 47.480 seconds, wall time: 9.024 minutes
Iteration: 80000, time: 43.879 days, Δt: 47.480 seconds, wall time: 9.075 minutes
Iteration: 80500, time: 44.154 days, Δt: 47.421 seconds, wall time: 9.127 minutes
Iteration: 81000, time: 44.428 days, Δt: 47.344 seconds, wall time: 9.178 minutes
Iteration: 81500, time: 44.702 days, Δt: 47.312 seconds, wall time: 9.230 minutes
Iteration: 82000, time: 44.976 days, Δt: 47.332 seconds, wall time: 9.283 minutes
Iteration: 82500, time: 45.249 days, Δt: 47.312 seconds, wall time: 9.339 minutes
Iteration: 83000, time: 45.523 days, Δt: 47.333 seconds, wall time: 9.394 minutes
Iteration: 83500, time: 45.797 days, Δt: 47.400 seconds, wall time: 9.448 minutes
Iteration: 84000, time: 46.071 days, Δt: 47.480 seconds, wall time: 9.504 minutes
Iteration: 84500, time: 46.346 days, Δt: 47.480 seconds, wall time: 9.561 minutes
Iteration: 85000, time: 46.621 days, Δt: 47.480 seconds, wall time: 9.616 minutes
Iteration: 85500, time: 46.896 days, Δt: 47.480 seconds, wall time: 9.672 minutes
Iteration: 86000, time: 47.170 days, Δt: 47.480 seconds, wall time: 9.725 minutes
Iteration: 86500, time: 47.445 days, Δt: 47.480 seconds, wall time: 9.778 minutes
Iteration: 87000, time: 47.720 days, Δt: 47.480 seconds, wall time: 9.838 minutes
Iteration: 87500, time: 47.995 days, Δt: 47.480 seconds, wall time: 9.899 minutes
Iteration: 88000, time: 48.269 days, Δt: 47.475 seconds, wall time: 9.960 minutes
Iteration: 88500, time: 48.544 days, Δt: 47.470 seconds, wall time: 10.022 minutes
Iteration: 89000, time: 48.819 days, Δt: 47.466 seconds, wall time: 10.084 minutes
Iteration: 89500, time: 49.093 days, Δt: 47.461 seconds, wall time: 10.144 minutes
Iteration: 90000, time: 49.367 days, Δt: 47.456 seconds, wall time: 10.202 minutes
Iteration: 90500, time: 49.642 days, Δt: 47.451 seconds, wall time: 10.259 minutes
Iteration: 91000, time: 49.917 days, Δt: 47.446 seconds, wall time: 10.318 minutes
Iteration: 91500, time: 50.191 days, Δt: 47.448 seconds, wall time: 10.377 minutes
Iteration: 92000, time: 50.466 days, Δt: 47.454 seconds, wall time: 10.432 minutes
Iteration: 92500, time: 50.740 days, Δt: 47.459 seconds, wall time: 10.490 minutes
Iteration: 93000, time: 51.015 days, Δt: 47.464 seconds, wall time: 10.547 minutes
Iteration: 93500, time: 51.290 days, Δt: 47.462 seconds, wall time: 10.606 minutes
Iteration: 94000, time: 51.564 days, Δt: 47.460 seconds, wall time: 10.661 minutes
Iteration: 94500, time: 51.839 days, Δt: 47.458 seconds, wall time: 10.716 minutes
Iteration: 95000, time: 52.113 days, Δt: 47.456 seconds, wall time: 10.774 minutes
Iteration: 95500, time: 52.388 days, Δt: 47.452 seconds, wall time: 10.831 minutes
Iteration: 96000, time: 52.662 days, Δt: 47.449 seconds, wall time: 10.886 minutes
Iteration: 96500, time: 52.937 days, Δt: 47.445 seconds, wall time: 10.941 minutes
Iteration: 97000, time: 53.211 days, Δt: 47.445 seconds, wall time: 10.999 minutes
Iteration: 97500, time: 53.486 days, Δt: 47.445 seconds, wall time: 11.053 minutes
Iteration: 98000, time: 53.761 days, Δt: 47.445 seconds, wall time: 11.107 minutes
Iteration: 98500, time: 54.035 days, Δt: 47.445 seconds, wall time: 11.161 minutes
Iteration: 99000, time: 54.309 days, Δt: 47.451 seconds, wall time: 11.216 minutes
Iteration: 99500, time: 54.584 days, Δt: 47.457 seconds, wall time: 11.270 minutes
Iteration: 100000, time: 54.858 days, Δt: 47.463 seconds, wall time: 11.327 minutes
Iteration: 100500, time: 55.133 days, Δt: 47.468 seconds, wall time: 11.385 minutes
Iteration: 101000, time: 55.408 days, Δt: 47.472 seconds, wall time: 11.443 minutes
Iteration: 101500, time: 55.682 days, Δt: 47.475 seconds, wall time: 11.500 minutes
Iteration: 102000, time: 55.957 days, Δt: 47.479 seconds, wall time: 11.559 minutes
Iteration: 102500, time: 56.232 days, Δt: 47.480 seconds, wall time: 11.619 minutes
Iteration: 103000, time: 56.507 days, Δt: 47.480 seconds, wall time: 11.678 minutes
Iteration: 103500, time: 56.781 days, Δt: 47.480 seconds, wall time: 11.736 minutes
Iteration: 104000, time: 57.056 days, Δt: 47.480 seconds, wall time: 11.795 minutes
Iteration: 104500, time: 57.331 days, Δt: 47.480 seconds, wall time: 11.854 minutes
Iteration: 105000, time: 57.606 days, Δt: 47.480 seconds, wall time: 11.912 minutes
Iteration: 105500, time: 57.880 days, Δt: 47.480 seconds, wall time: 11.970 minutes
Iteration: 106000, time: 58.155 days, Δt: 47.457 seconds, wall time: 12.029 minutes
Iteration: 106500, time: 58.429 days, Δt: 47.418 seconds, wall time: 12.088 minutes
Iteration: 107000, time: 58.704 days, Δt: 47.385 seconds, wall time: 12.144 minutes
Iteration: 107500, time: 58.978 days, Δt: 47.358 seconds, wall time: 12.198 minutes
Iteration: 108000, time: 59.252 days, Δt: 47.379 seconds, wall time: 12.253 minutes
Iteration: 108500, time: 59.526 days, Δt: 47.411 seconds, wall time: 12.310 minutes
Iteration: 109000, time: 59.800 days, Δt: 47.448 seconds, wall time: 12.371 minutes
Iteration: 109500, time: 60.075 days, Δt: 47.480 seconds, wall time: 12.431 minutes
Iteration: 110000, time: 60.350 days, Δt: 47.480 seconds, wall time: 12.492 minutes
Iteration: 110500, time: 60.624 days, Δt: 47.480 seconds, wall time: 12.552 minutes
Iteration: 111000, time: 60.899 days, Δt: 47.480 seconds, wall time: 12.613 minutes
Iteration: 111500, time: 61.174 days, Δt: 47.480 seconds, wall time: 12.673 minutes
Iteration: 112000, time: 61.448 days, Δt: 47.480 seconds, wall time: 12.733 minutes
Iteration: 112500, time: 61.723 days, Δt: 47.480 seconds, wall time: 12.790 minutes
Iteration: 113000, time: 61.998 days, Δt: 47.480 seconds, wall time: 12.848 minutes
Iteration: 113500, time: 62.273 days, Δt: 47.480 seconds, wall time: 12.909 minutes
Iteration: 114000, time: 62.547 days, Δt: 47.480 seconds, wall time: 12.971 minutes
Iteration: 114500, time: 62.822 days, Δt: 47.480 seconds, wall time: 13.032 minutes
Iteration: 115000, time: 63.097 days, Δt: 47.480 seconds, wall time: 13.095 minutes
Iteration: 115500, time: 63.371 days, Δt: 47.480 seconds, wall time: 13.156 minutes
Iteration: 116000, time: 63.646 days, Δt: 47.480 seconds, wall time: 13.217 minutes
Iteration: 116500, time: 63.921 days, Δt: 47.480 seconds, wall time: 13.275 minutes
Iteration: 117000, time: 64.196 days, Δt: 47.480 seconds, wall time: 13.333 minutes
Iteration: 117500, time: 64.470 days, Δt: 47.480 seconds, wall time: 13.392 minutes
Iteration: 118000, time: 64.745 days, Δt: 47.480 seconds, wall time: 13.453 minutes
Iteration: 118500, time: 65.020 days, Δt: 47.472 seconds, wall time: 13.515 minutes
Iteration: 119000, time: 65.294 days, Δt: 47.353 seconds, wall time: 13.577 minutes
Iteration: 119500, time: 65.568 days, Δt: 47.311 seconds, wall time: 13.638 minutes
Iteration: 120000, time: 65.842 days, Δt: 47.364 seconds, wall time: 13.699 minutes
Iteration: 120500, time: 66.116 days, Δt: 47.453 seconds, wall time: 13.761 minutes
Iteration: 121000, time: 66.390 days, Δt: 47.345 seconds, wall time: 13.820 minutes
Iteration: 121500, time: 66.664 days, Δt: 47.332 seconds, wall time: 13.878 minutes
Iteration: 122000, time: 66.939 days, Δt: 47.455 seconds, wall time: 13.935 minutes
Iteration: 122500, time: 67.213 days, Δt: 47.313 seconds, wall time: 13.994 minutes
Iteration: 123000, time: 67.487 days, Δt: 47.312 seconds, wall time: 14.053 minutes
Iteration: 123500, time: 67.761 days, Δt: 47.355 seconds, wall time: 14.113 minutes
Iteration: 124000, time: 68.035 days, Δt: 47.480 seconds, wall time: 14.176 minutes
Iteration: 124500, time: 68.310 days, Δt: 47.480 seconds, wall time: 14.237 minutes
Iteration: 125000, time: 68.585 days, Δt: 47.480 seconds, wall time: 14.298 minutes
Iteration: 125500, time: 68.859 days, Δt: 47.480 seconds, wall time: 14.357 minutes
Iteration: 126000, time: 69.134 days, Δt: 47.480 seconds, wall time: 14.419 minutes
Iteration: 126500, time: 69.409 days, Δt: 47.480 seconds, wall time: 14.479 minutes
Iteration: 127000, time: 69.684 days, Δt: 47.480 seconds, wall time: 14.536 minutes
Iteration: 127500, time: 69.958 days, Δt: 47.480 seconds, wall time: 14.594 minutes
Iteration: 128000, time: 70.233 days, Δt: 47.480 seconds, wall time: 14.654 minutes
Iteration: 128500, time: 70.508 days, Δt: 47.480 seconds, wall time: 14.708 minutes
Iteration: 129000, time: 70.783 days, Δt: 47.480 seconds, wall time: 14.762 minutes
Iteration: 129500, time: 71.057 days, Δt: 47.480 seconds, wall time: 14.816 minutes
Iteration: 130000, time: 71.332 days, Δt: 47.480 seconds, wall time: 14.871 minutes
Iteration: 130500, time: 71.607 days, Δt: 47.480 seconds, wall time: 14.926 minutes
Iteration: 131000, time: 71.881 days, Δt: 47.480 seconds, wall time: 14.982 minutes
Iteration: 131500, time: 72.156 days, Δt: 47.480 seconds, wall time: 15.037 minutes
Iteration: 132000, time: 72.431 days, Δt: 47.480 seconds, wall time: 15.090 minutes
Iteration: 132500, time: 72.706 days, Δt: 47.480 seconds, wall time: 15.142 minutes
Iteration: 133000, time: 72.980 days, Δt: 47.480 seconds, wall time: 15.196 minutes
Iteration: 133500, time: 73.255 days, Δt: 47.480 seconds, wall time: 15.253 minutes
Iteration: 134000, time: 73.530 days, Δt: 47.480 seconds, wall time: 15.307 minutes
Iteration: 134500, time: 73.805 days, Δt: 47.480 seconds, wall time: 15.360 minutes
Iteration: 135000, time: 74.079 days, Δt: 47.480 seconds, wall time: 15.416 minutes
Iteration: 135500, time: 74.354 days, Δt: 47.480 seconds, wall time: 15.468 minutes
Iteration: 136000, time: 74.629 days, Δt: 47.480 seconds, wall time: 15.519 minutes
Iteration: 136500, time: 74.903 days, Δt: 47.480 seconds, wall time: 15.570 minutes
Iteration: 137000, time: 75.178 days, Δt: 47.480 seconds, wall time: 15.623 minutes
Iteration: 137500, time: 75.453 days, Δt: 47.480 seconds, wall time: 15.673 minutes
Iteration: 138000, time: 75.728 days, Δt: 47.480 seconds, wall time: 15.724 minutes
Iteration: 138500, time: 76.002 days, Δt: 47.480 seconds, wall time: 15.775 minutes
Iteration: 139000, time: 76.277 days, Δt: 47.480 seconds, wall time: 15.826 minutes
Iteration: 139500, time: 76.552 days, Δt: 47.480 seconds, wall time: 15.876 minutes
Iteration: 140000, time: 76.827 days, Δt: 47.480 seconds, wall time: 15.926 minutes
Iteration: 140500, time: 77.101 days, Δt: 47.480 seconds, wall time: 15.978 minutes
Iteration: 141000, time: 77.376 days, Δt: 47.480 seconds, wall time: 16.028 minutes
Iteration: 141500, time: 77.651 days, Δt: 47.480 seconds, wall time: 16.078 minutes
Iteration: 142000, time: 77.925 days, Δt: 47.480 seconds, wall time: 16.128 minutes
Iteration: 142500, time: 78.200 days, Δt: 47.480 seconds, wall time: 16.180 minutes
Iteration: 143000, time: 78.475 days, Δt: 47.480 seconds, wall time: 16.230 minutes
Iteration: 143500, time: 78.750 days, Δt: 47.480 seconds, wall time: 16.281 minutes
Iteration: 144000, time: 79.024 days, Δt: 47.480 seconds, wall time: 16.335 minutes
Iteration: 144500, time: 79.299 days, Δt: 47.480 seconds, wall time: 16.387 minutes
Iteration: 145000, time: 79.574 days, Δt: 47.480 seconds, wall time: 16.439 minutes
Iteration: 145500, time: 79.848 days, Δt: 47.480 seconds, wall time: 16.490 minutes
Iteration: 146000, time: 80.123 days, Δt: 47.480 seconds, wall time: 16.542 minutes
Iteration: 146500, time: 80.398 days, Δt: 47.480 seconds, wall time: 16.593 minutes
Iteration: 147000, time: 80.673 days, Δt: 47.480 seconds, wall time: 16.644 minutes
Iteration: 147500, time: 80.947 days, Δt: 47.480 seconds, wall time: 16.695 minutes
Iteration: 148000, time: 81.222 days, Δt: 47.480 seconds, wall time: 16.748 minutes
Iteration: 148500, time: 81.497 days, Δt: 47.480 seconds, wall time: 16.800 minutes
Iteration: 149000, time: 81.772 days, Δt: 47.480 seconds, wall time: 16.853 minutes
Iteration: 149500, time: 82.046 days, Δt: 47.480 seconds, wall time: 16.906 minutes
Iteration: 150000, time: 82.321 days, Δt: 47.480 seconds, wall time: 16.958 minutes
Iteration: 150500, time: 82.596 days, Δt: 47.480 seconds, wall time: 17.010 minutes
Iteration: 151000, time: 82.870 days, Δt: 47.480 seconds, wall time: 17.062 minutes
Iteration: 151500, time: 83.145 days, Δt: 47.480 seconds, wall time: 17.117 minutes
Iteration: 152000, time: 83.420 days, Δt: 47.480 seconds, wall time: 17.170 minutes
Iteration: 152500, time: 83.695 days, Δt: 47.480 seconds, wall time: 17.222 minutes
Iteration: 153000, time: 83.969 days, Δt: 47.480 seconds, wall time: 17.275 minutes
Iteration: 153500, time: 84.244 days, Δt: 47.480 seconds, wall time: 17.328 minutes
Iteration: 154000, time: 84.519 days, Δt: 47.480 seconds, wall time: 17.380 minutes
Iteration: 154500, time: 84.794 days, Δt: 47.480 seconds, wall time: 17.432 minutes
Iteration: 155000, time: 85.068 days, Δt: 47.479 seconds, wall time: 17.485 minutes
Iteration: 155500, time: 85.343 days, Δt: 47.477 seconds, wall time: 17.536 minutes
Iteration: 156000, time: 85.618 days, Δt: 47.475 seconds, wall time: 17.587 minutes
Iteration: 156500, time: 85.892 days, Δt: 47.473 seconds, wall time: 17.638 minutes
Iteration: 157000, time: 86.167 days, Δt: 47.321 seconds, wall time: 17.690 minutes
Iteration: 157500, time: 86.441 days, Δt: 47.321 seconds, wall time: 17.742 minutes
Iteration: 158000, time: 86.715 days, Δt: 47.392 seconds, wall time: 17.793 minutes
Iteration: 158500, time: 86.989 days, Δt: 47.355 seconds, wall time: 17.845 minutes
Iteration: 159000, time: 87.263 days, Δt: 47.459 seconds, wall time: 17.900 minutes
Iteration: 159500, time: 87.538 days, Δt: 47.489 seconds, wall time: 17.960 minutes
Iteration: 160000, time: 87.812 days, Δt: 47.528 seconds, wall time: 18.026 minutes
Iteration: 160500, time: 88.085 days, Δt: 47.325 seconds, wall time: 18.097 minutes
Iteration: 161000, time: 88.359 days, Δt: 47.316 seconds, wall time: 18.149 minutes
Iteration: 161500, time: 88.633 days, Δt: 47.312 seconds, wall time: 18.201 minutes
Iteration: 162000, time: 88.907 days, Δt: 47.313 seconds, wall time: 18.252 minutes
Iteration: 162500, time: 89.180 days, Δt: 47.346 seconds, wall time: 18.305 minutes
Iteration: 163000, time: 89.455 days, Δt: 47.465 seconds, wall time: 18.356 minutes
Iteration: 163500, time: 89.729 days, Δt: 47.448 seconds, wall time: 18.407 minutes
Iteration: 164000, time: 90.003 days, Δt: 47.340 seconds, wall time: 18.459 minutes
Iteration: 164500, time: 90.277 days, Δt: 47.339 seconds, wall time: 18.513 minutes
Iteration: 165000, time: 90.551 days, Δt: 47.338 seconds, wall time: 18.565 minutes
Iteration: 165500, time: 90.825 days, Δt: 47.338 seconds, wall time: 18.616 minutes
Iteration: 166000, time: 91.098 days, Δt: 47.337 seconds, wall time: 18.669 minutes
Iteration: 166500, time: 91.372 days, Δt: 47.337 seconds, wall time: 18.724 minutes
Iteration: 167000, time: 91.646 days, Δt: 47.338 seconds, wall time: 18.776 minutes
Iteration: 167500, time: 91.920 days, Δt: 47.338 seconds, wall time: 18.828 minutes
Iteration: 168000, time: 92.193 days, Δt: 47.312 seconds, wall time: 18.884 minutes
Iteration: 168500, time: 92.467 days, Δt: 47.363 seconds, wall time: 18.936 minutes
Iteration: 169000, time: 92.742 days, Δt: 47.527 seconds, wall time: 18.992 minutes
Iteration: 169500, time: 93.016 days, Δt: 47.352 seconds, wall time: 19.046 minutes
Iteration: 170000, time: 93.290 days, Δt: 47.333 seconds, wall time: 19.102 minutes
Iteration: 170500, time: 93.564 days, Δt: 47.319 seconds, wall time: 19.154 minutes
Iteration: 171000, time: 93.838 days, Δt: 47.400 seconds, wall time: 19.206 minutes
Iteration: 171500, time: 94.112 days, Δt: 47.404 seconds, wall time: 19.259 minutes
Iteration: 172000, time: 94.386 days, Δt: 47.383 seconds, wall time: 19.312 minutes
Iteration: 172500, time: 94.660 days, Δt: 47.376 seconds, wall time: 19.364 minutes
Iteration: 173000, time: 94.934 days, Δt: 47.830 seconds, wall time: 19.417 minutes
Iteration: 173500, time: 95.210 days, Δt: 47.431 seconds, wall time: 19.471 minutes
Iteration: 174000, time: 95.484 days, Δt: 47.442 seconds, wall time: 19.524 minutes
Iteration: 174500, time: 95.758 days, Δt: 47.337 seconds, wall time: 19.579 minutes
Iteration: 175000, time: 96.032 days, Δt: 47.312 seconds, wall time: 19.633 minutes
Iteration: 175500, time: 96.306 days, Δt: 47.348 seconds, wall time: 19.685 minutes
Iteration: 176000, time: 96.580 days, Δt: 47.354 seconds, wall time: 19.738 minutes
Iteration: 176500, time: 96.854 days, Δt: 47.323 seconds, wall time: 19.790 minutes
Iteration: 177000, time: 97.129 days, Δt: 47.430 seconds, wall time: 19.844 minutes
Iteration: 177500, time: 97.403 days, Δt: 47.352 seconds, wall time: 19.896 minutes
Iteration: 178000, time: 97.677 days, Δt: 47.316 seconds, wall time: 19.948 minutes
Iteration: 178500, time: 97.950 days, Δt: 47.316 seconds, wall time: 20.000 minutes
Iteration: 179000, time: 98.223 days, Δt: 47.312 seconds, wall time: 20.053 minutes
Iteration: 179500, time: 98.497 days, Δt: 47.320 seconds, wall time: 20.105 minutes
Iteration: 180000, time: 98.771 days, Δt: 47.351 seconds, wall time: 20.157 minutes
Iteration: 180500, time: 99.045 days, Δt: 47.497 seconds, wall time: 20.210 minutes
Iteration: 181000, time: 99.319 days, Δt: 47.312 seconds, wall time: 20.262 minutes
Iteration: 181500, time: 99.593 days, Δt: 47.311 seconds, wall time: 20.314 minutes
Iteration: 182000, time: 99.868 days, Δt: 47.369 seconds, wall time: 20.367 minutes
[ Info: Simulation is stopping after running for 20.393 minutes.
[ Info: Simulation time 100.000 days equals or exceeds stop time 100 days.

Load output and plot

Now we can visualise the results with some post processing to diagnose the air-sea CO₂ flux

   P = FieldTimeSeries("$filename.jld2", "P")
 NO₃ = FieldTimeSeries("$filename.jld2", "NO₃")
   Z = FieldTimeSeries("$filename.jld2", "Z")
sPOM = FieldTimeSeries("$filename.jld2", "sPOM")
bPOM = FieldTimeSeries("$filename.jld2", "bPOM")
 DIC = FieldTimeSeries("$filename.jld2", "DIC")
 Alk = FieldTimeSeries("$filename.jld2", "Alk")

x, y, z = nodes(P)
times = P.times

We compute the air-sea CO₂ flux at the surface (corresponding to vertical index k = grid.Nz) and the carbon export by computing how much carbon sinks below some arbitrary depth; here we use depth that corresponds to k = grid.Nz - 20.

air_sea_CO₂_flux = zeros(length(times))
carbon_export = zeros(length(times))

using Oceananigans.Biogeochemistry: biogeochemical_drift_velocity

for (n, t) in enumerate(times)
    clock.time = t

    air_sea_CO₂_flux[n] = CO₂_flux.condition.func(1, 1, grid, clock, (; DIC = DIC[n], Alk = Alk[n], T, S))

    carbon_export[n] = (sPOM[n][1, 1, grid.Nz-20] * biogeochemical_drift_velocity(model.biogeochemistry, Val(:sPOM)).w[1, 1, grid.Nz-20] +
                        bPOM[n][1, 1, grid.Nz-20] * biogeochemical_drift_velocity(model.biogeochemistry, Val(:bPOM)).w[1, 1, grid.Nz-20]) * redfield(Val(:sPOM), model.biogeochemistry)
end

Both air_sea_CO₂_flux and carbon_export are in units mmol CO₂ / (m² s).

using CairoMakie

fig = Figure(size = (1000, 1500), fontsize = 20)

axis_kwargs = (xlabel = "Time (days)", ylabel = "z (m)", limits = ((0, times[end] / days), (-150meters, 0)))

axP = Axis(fig[1, 1]; title = "Phytoplankton concentration (mmol N/m³)", axis_kwargs...)
hmP = heatmap!(times / days, z, interior(P, 1, 1, :, :)', colormap=:batlow)
Colorbar(fig[1, 2], hmP)

axNO₃ = Axis(fig[2, 1]; title = "Nitrate concentration (mmol N/m³)", axis_kwargs...)
hmNO₃ = heatmap!(times / days, z, interior(NO₃, 1, 1, :, :)', colormap=:batlow)
Colorbar(fig[2, 2], hmNO₃)

axZ = Axis(fig[3, 1]; title = "Zooplankton concentration (mmol N/m³)", axis_kwargs...)
hmZ = heatmap!(times / days, z, interior(Z, 1, 1, :, :)', colormap=:batlow)
Colorbar(fig[3, 2], hmZ)

axD = Axis(fig[4, 1]; title = "Detritus concentration (mmol N/m³)", axis_kwargs...)
hmD = heatmap!(times / days, z, interior(sPOM, 1, 1, :, :)' .+ interior(bPOM, 1, 1, :, :)', colormap=:batlow)
Colorbar(fig[4, 2], hmD)

CO₂_molar_mass = (12 + 2 * 16) * 1e-3 # kg / mol

axfDIC = Axis(fig[5, 1], xlabel = "Time (days)", ylabel = "Flux (kgCO₂/m²/year)",
                         title = "Air-sea CO₂ flux and Sinking", limits = ((0, times[end] / days), nothing))
lines!(axfDIC, times / days, cumsum(air_sea_CO₂_flux) /1e3 * CO₂_molar_mass * year, linewidth = 3, label = "Air-sea flux")
lines!(axfDIC, times / days, cumsum(carbon_export) /1e3    * CO₂_molar_mass * year, linewidth = 3, label = "Sinking export")
Legend(fig[5, 2], axfDIC, framevisible = false)

fig

This page was generated using Literate.jl.