Index

A

acceleration_delta_fn_wrapper() (in module jax_sph.solver) acceleration_riemann_fn_wrapper() (in module jax_sph.solver)

acceleration_standard_fn_wrapper() (in module jax_sph.solver) acceleration_tvf_fn_wrapper() (in module jax_sph.solver) artificial_viscosity_fn_wrapper() (in module jax_sph.solver)

B

BaseEoS (class in jax_sph.eos)

BaseKernel (class in jax_sph.kernel)

C

check_cfg() (in module jax_sph.defaults) compute_nws_jax_wrapper() (in module jax_sph.utils)

compute_nws_scipy() (in module jax_sph.utils) CubicKernel (class in jax_sph.kernel)

F

forward_wrapper() (jax_sph.solver.WCSPH method)

G

GaussianKernel (class in jax_sph.kernel) get_array_stats() (in module jax_sph.utils) get_ekin() (in module jax_sph.utils) get_noise_masked() (in module jax_sph.utils)

get_particle_cells() (in module jax_sph.partition) get_stats() (in module jax_sph.utils) grad_w() (jax_sph.kernel.BaseKernel method) gwbc_fn_riemann_wrapper() (in module jax_sph.solver) gwbc_fn_wrapper() (in module jax_sph.solver)

I

initialize() (jax_sph.case_setup.SimulationSetup method)

io_setup() (in module jax_sph.io_state)

J

jax_sph.case_setup module jax_sph.defaults module jax_sph.eos module jax_sph.integrator module jax_sph.io_state module

jax_sph.kernel module jax_sph.partition module jax_sph.simulate module jax_sph.solver module jax_sph.utils module

L

limiter_fn_wrapper() (in module jax_sph.solver)

load_case() (in module jax_sph.case_setup) Logger (class in jax_sph.utils)

M

module jax_sph.case_setup jax_sph.defaults jax_sph.eos jax_sph.integrator jax_sph.io_state jax_sph.kernel jax_sph.partition jax_sph.simulate jax_sph.solver jax_sph.utils

N

neighbor_list() (in module jax_sph.partition)

P

p_fn() (jax_sph.eos.BaseEoS method) (jax_sph.eos.RIEMANNEoS method) (jax_sph.eos.TaitEoS method)

pos_box_2d() (in module jax_sph.utils) pos_box_3d() (in module jax_sph.utils) pos_init_cartesian_2d() (in module jax_sph.utils) pos_init_cartesian_3d() (in module jax_sph.utils)

Q

QuinticKernel (class in jax_sph.kernel)

R

read_h5() (in module jax_sph.io_state) rho_evol_fn() (in module jax_sph.solver) rho_evol_fn_delta_wrapper() (in module jax_sph.solver) rho_evol_riemann_fn_wrapper() (in module jax_sph.solver) rho_fn() (jax_sph.eos.BaseEoS method) (jax_sph.eos.RIEMANNEoS method) (jax_sph.eos.TaitEoS method)

rho_renorm_fn() (in module jax_sph.solver) rho_summation_fn() (in module jax_sph.solver) RIEMANNEoS (class in jax_sph.eos)

S

set_defaults() (in module jax_sph.defaults) set_relaxation() (in module jax_sph.case_setup) si_euler() (in module jax_sph.integrator)

simulate() (in module jax_sph.simulate) SimulationSetup (class in jax_sph.case_setup) sph_interpolator() (in module jax_sph.utils) SuperGaussianKernel (class in jax_sph.kernel)

T

Tag (class in jax_sph.utils) TaitEoS (class in jax_sph.eos)

temperature_derivative_wrapper() (in module jax_sph.solver) tvf_stress_fn() (in module jax_sph.solver)

W

w() (jax_sph.kernel.BaseKernel method) (jax_sph.kernel.CubicKernel method) (jax_sph.kernel.GaussianKernel method) (jax_sph.kernel.QuinticKernel method) (jax_sph.kernel.SuperGaussianKernel method) (jax_sph.kernel.WendlandC2Kernel method) (jax_sph.kernel.WendlandC4Kernel method) (jax_sph.kernel.WendlandC6Kernel method)

wall_phi_vec_wrapper() (in module jax_sph.solver) WCSPH (class in jax_sph.solver) WendlandC2Kernel (class in jax_sph.kernel) WendlandC4Kernel (class in jax_sph.kernel) WendlandC6Kernel (class in jax_sph.kernel) write_h5() (in module jax_sph.io_state) write_state() (in module jax_sph.io_state) write_vtk() (in module jax_sph.io_state) write_vtks_from_h5s() (in module jax_sph.io_state)