flow.fields

flow.fields(f, chi, alpha_max)

Density and velocity from converged populations.

Parameters

Name Type Description Default
f jax.Array Converged populations, (19, NX, NY, NZ). required
chi jax.Array Solid fraction, (NX, NY, NZ). required
alpha_max float Drag coefficient at chi = 1. required

Returns

Name Type Description
tuple[jax.Array, jax.Array] (rho, u) with shapes (NX, NY, NZ) and (3, NX, NY, NZ).