jf1uids._physics_modules._cosmic_rays.cr_fluid_equations

Module Contents

Functions

total_energy_from_primitives_with_crs
gas_pressure_from_primitives_with_crs
total_pressure_from_conserved_with_crs

API

jf1uids._physics_modules._cosmic_rays.cr_fluid_equations.total_energy_from_primitives_with_crs(primitive_state: jaxtyping.Float[jaxtyping.Array, num_vars num_cells], registered_variables: jf1uids.fluid_equations.registered_variables.RegisteredVariables) → jaxtyping.Float[jaxtyping.Array, num_cells]

jf1uids._physics_modules._cosmic_rays.cr_fluid_equations.gas_pressure_from_primitives_with_crs(primitive_state: jaxtyping.Float[jaxtyping.Array, num_vars num_cells], registered_variables: jf1uids.fluid_equations.registered_variables.RegisteredVariables) → jaxtyping.Float[jaxtyping.Array, num_cells]

jf1uids._physics_modules._cosmic_rays.cr_fluid_equations.total_pressure_from_conserved_with_crs(conserved_state: jaxtyping.Float[jaxtyping.Array, num_vars num_cells], registered_variables: jf1uids.fluid_equations.registered_variables.RegisteredVariables) → jaxtyping.Float[jaxtyping.Array, num_cells]