spectrax._initialize_maxwellian

Initialization of Hermite–Fourier coefficients for Maxwellian equilibria.

Functions

compute_C_nmp(Us_grid, alpha_s, u_s, Nn, Nm, Np, Ns)

Build the Hermite-Fourier coefficients for a drifting Maxwellian distribution function.

Module Contents

spectrax._initialize_maxwellian.compute_C_nmp(Us_grid, alpha_s, u_s, Nn, Nm, Np, Ns)

Build the Hermite-Fourier coefficients for a drifting Maxwellian distribution function.

Parameters:

  • Us_grid (jnp.ndarray) – Velocity grid for each species with shape (Ns, 3, Ny, Nx, Nz), containing the phase-space coordinates at which the Maxwellian is sampled.

  • alpha_s (array-like) – Sequence of thermal scaling parameters flattened as (3 * Ns,), grouped as (alpha_x, alpha_y, alpha_z) per species.

  • u_s (array-like) – Sequence of drift velocities flattened as (3 * Ns,), grouped as (u_x, u_y, u_z) per species.

  • Nn (int) – Number of Hermite modes retained along the x, y, and z velocity axes.

  • Nm (int) – Number of Hermite modes retained along the x, y, and z velocity axes.

  • Np (int) – Number of Hermite modes retained along the x, y, and z velocity axes.

  • Ns (int) – Number of species.

Returns:

Complex Hermite-Fourier coefficients with shape (Ns, Np, Nm, Nn, Ny, Nx, Nz) corresponding to the Maxwellian evaluated on the supplied grid.

Return type:

jnp.ndarray