Private types and functions

Abstract supertype for diagnostics.

mutable struct Clock{T<:AbstractFloat}

Represents the clock of a problem.

• dt::AbstractFloat: the time-step

• t::AbstractFloat: the time

• step::Int64: the step number

struct EmptyParams <: AbstractParams

A placeholder struct for parameters.

struct EmptyVars <: AbstractVars

A placeholder struct for variables.

struct Equation{T, TL, G<:AbstractFloat}

The equation to be solved ∂u/∂t = L * u + N(u). Array L includes the coefficients of the linear term L * u and calcN! is a function which computes the nonlinear term N(u). The struct also includes the problem's grid and the float type of the state vector (and consequently of N(u)).

• L::Any: array with coefficient for the linear part of the equation

• calcN!::Function: function that computes the nonlinear part of the equation

• grid::AbstractGrid{G} where G<:AbstractFloat: the grid

• dims::Tuple: the dimensions of L

• T::Any: the float type for the state vector

Equation(L, calcN!, grid; dims=supersize(L), T=nothing)

The equation constructor from the array L of the coefficients of the linear term, the function calcN! that computes the nonlinear term and the grid for the problem.

struct Problem{T, A<:AbstractArray, Tg<:AbstractFloat, TL}

A problem that represents a partial differential equation.

• sol::AbstractArray: the state vector

• clock::FourierFlows.Clock: the problem's

• eqn::FourierFlows.Equation{T, TL, Tg} where {T, Tg<:AbstractFloat, TL}: the equation

• grid::AbstractGrid{Tg} where Tg<:AbstractFloat: the grid

• vars::AbstractVars: the variables

• params::AbstractParams: the parameters

• timestepper::AbstractTimeStepper{A} where A<:AbstractArray: the timestepper

Problem(eqn::Equation, stepper, dt, grid::AbstractGrid{T}, 
        vars=EmptyVars, params=EmptyParams; stepperkwargs...) where T

Construct a Problem for equation eqn using the timestepper with timestep dt, on grid. The device is inferred from the grid. Optionally, use the keyword arguments to provide variables with vars and parameters with params. The stepperkwargs are passed on to the time-stepper constructor.

getaliasedwavenumbers(nk, nkr, aliased_fraction)

Return the top aliased_fraction highest wavenumbers, both for and real FFTs, kalias and kralias respectively. For example, aliased_fraction = 1/3 should return the indices of the top-most 1/6-th (in absolute value) for both positive and negative wavenumbers (i.e., 1/3 total) that should be set to zero after performing an FFT.

makefilter(K; order=4, innerK=2/3, outerK=1, tol=1e-15)

Return a filter acting on the non-dimensional wavenumber K. For a one-dimensional grid, the non-dimensional wavenumber K is

K = k * dx / π

and thus take values in $[-1, 1]$.

For K ≤ innerK the filter is inactive, i.e., equal to 1. For K > innerK, the filter decays exponentially to remove high-wavenumber content from the solution, i.e.,

filter(K) = exp(- decay * (K - innerK)^order)

For a given order, the decay rate is determined so that the filter value at the outer wavenumber outerK is tol, where tol is a small number, close to machine precision.

decay = - log(tol) / (outerK - innerK)^order

jacobian(a, b, grid)

Return the Jacobian of a and b on grid.

jacobianh(a, b, grid)

Return the Fourier transform of the Jacobian of a and b on grid.

on_grid(func, grid)

Return an array, of the type compatible with the device that the grid lives on, that contains the values of function func evaluated on the grid.

parsevalsum(uh, grid)

Return the real part of the sum of uh on the grid. For example on a 2D grid, parsevalsum returns

\[ℜ [ \sum_{𝐤} û_{𝐤} L_x L_y ] , \]

where $û_{𝐤} =$ uh $/ (n_x e^{i 𝐤 ⋅ 𝐱₀})$. The elements of the vector $𝐱₀$ are the left-most position in each direction, e.g., for a 2D grid (grid.x[1], grid.y[1]).

When the input uh comes from a real-FFT transform, parsevalsum takes care to count the contribution from certain $k$-modes twice.

parsevalsum2(uh, grid)

Return the sum of |uh|² on the grid, which is equal to the domain integral of u². For example on a 2D grid, parsevalsum2 returns

\[\sum_{𝐤} |û_{𝐤}|² L_x L_y = \iint u² 𝖽x 𝖽y ,\]

where $û_{𝐤} =$ uh $/ (n_x e^{i 𝐤 ⋅ 𝐱₀})$. The elements of the vector $𝐱₀$ are the left-most position in each direction, e.g., for a 2D grid (grid.x[1], grid.y[1]).

When the input uh comes from a real-FFT transform, parsevalsum2 takes care to count the contribution from certain $k$-modes twice.

radialspectrum(fh, grid; n=nothing, m=nothing, refinement=2)

Return the radial spectrum of fh. fh lives on Cartesian wavenumber grid $(k, l)$. To compute the radial spectrum, we first interpolate $f̂(k, l)$ onto a radial wavenumber grid $(ρ, θ)$, where $ρ² = k² + l²$ and $θ = \arctan(l / k)$. Note here that $f̂ =$ fh $/ (n_x e^{i 𝐤 ⋅ 𝐱₀})$. The elements of the vector $𝐱₀$ are the left-most position in each direction, e.g., for a 2D grid (grid.x[1], grid.y[1]).

After interpolation, we integrate $f̂$ over angles $θ$ to get fρ,

\[ f̂_ρ = \int f̂(ρ, θ) ρ 𝖽ρ 𝖽θ .\]

The resolution (n, m) for the polar wavenumber grid is n = refinement * maximum(grid.nk, grid.nl), m = refinement * maximum(grid.nk, grid.nl), where refinement = 2 by default. If fh is in conjugate symmetric form then only the upper-half plane in $θ$ is represented on the polar grid.

e.g. plot(energydiag[:t], energydiag[:data]).

extend!(diag, n)

Extend the data, time, and steps vectors of the diagnostic diag by n.

extend!(diag::Diagnostic{T,N})

Double the extend of the data, time, and steps vectors of the diagnostic diag.

savefield(file, location, data)

Saves a particular field's data to file.

savefields(file, grid)

Saves some parameters of problem's grid to file.

uniquepath(path)

Return path with a number appended if isfile(path) == true. The number is incremented until path does not exist.

getetdcoeffs(dt, L; ncirc=32, rcirc=1)

Calculate the coefficients associated with the (diagonal) linear coefficient L for an ETDRK4 timestepper with timestep dt.

The calculation is done by integrating over a unit circle in the complex space. For more info refer to Kassam and Trefethen (2005).

struct Params{T} <: AbstractParams

The parameters for diffusion problem:

• κ::Any: diffusivity coefficient

struct Vars{Aphys, Atrans} <: AbstractVars

The variables for diffusion problem:

• c: tracer concentration

• cx: tracer concentration derivative

• ch: Fourier transform of tracer concentration

• cxh: Fourier transform of tracer concentration derivative

Vars(grid)

Return the variables vars for a constant diffusivity problem on grid.

Equation(grid, params)

Return the equation for a constant diffusivity problem on grid with diffusivity found inside params.

calcN!(N, sol, t, clock, vars, params, grid)

Calculate the nonlinear term for the 1D diffusion equation.