src.equations

Functions

compute_dfm_dt(f, k, nu, v_e, Nn, index, q)

Computes the time derivative of a function representing a system described in the

objective(t, f, args)

Computes the time derivative of the Fourier mode dynamics based on the provided input

Module Contents

src.equations.compute_dfm_dt(f, k, nu, v_e, Nn, index, q)

Computes the time derivative of a function representing a system described in the Fourier and physical space, taking into account effects like wave interactions, damping, and external forces.

Parameters:

  • f – jnp.ndarray The wave function represented as an array in the Fourier or physical space.

  • k – float The wave number that determines the spatial frequency of the wave.

  • nu – float The viscosity coefficient impacting energy dissipation.

  • v_e – jnp.ndarray An array specifying velocity factors for the external forces.

  • Nn – int The maximum number of wave modes in the system.

  • index – int The current index in the array representing the mode being evaluated.

  • q – jnp.ndarray An array of coefficients that modify the coupling between modes.

Returns:

jnp.complex128 The computed time derivative of the specified wave mode, taking into account interactions between different modes and external forces.

src.equations.objective(t, f, args)

Computes the time derivative of the Fourier mode dynamics based on the provided input parameters and a set of arguments. This function utilizes jax.vmap for vectorized map computation, applying the compute_dfm_dt function over a specific range of modes.

The computation involves iterating over a sequence of values determined by the provided maximum number of modes (m_max) and processing elements using the arguments supplied in args. JAX’s vmap is employed to enable efficient parallel application of the calculation across multiple data points or modes.

The function encapsulates a physics-oriented calculation typically used in time-evolution simulations, spectrum modeling, or Fourier transform-based computational frameworks.

Parameters:

  • t (float) – Current time, treated as a placeholder in this function.

  • f (jax.numpy.ndarray) – Array of Fourier modes at the current timestep. Represents the state or amplitude values of the modes.

  • args (tuple) – Tuple of additional parameters passed to define the dynamics: - k: Wavenumber or characteristic parameter affecting dynamics. - v_e: Constant or coefficient influencing velocity evolution. - nu: Diffusion or damping coefficient used in the computation. - m_max: Maximum number of Fourier modes utilized in the summation. - q: External or forcing parameter influencing mode amplitudes.

Returns:

Array containing the time derivatives of the Fourier modes computed over the range of indices specified by m_max.

Return type:

jax.numpy.ndarray