Index

_

__all__ (in module astronomix.setup_helpers)

__truediv__() (astronomix.option_classes.simulation_config.StaticFloatVector method)

A

activate_snapshot_callback (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] ADAPTIVE_CHECKPOINTED (in module astronomix.time_stepping._time_loop) ADAPTIVE_WHILE (in module astronomix.time_stepping._time_loop) add_inset_box() (in module astronomix.plotting_helpers.inset_box), [1] add_power_law_indicators() (in module astronomix.plotting_helpers.power_law_indicators), [1] AM_HLLC (in module astronomix.option_classes.simulation_config) argument_memory_bytes (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] astronomix module, [1] astronomix._fluid_equations module astronomix._fluid_equations._eigen_hydro module astronomix._fluid_equations._eigen_hydro_iso module astronomix._fluid_equations._eigen_mhd module astronomix._fluid_equations._eigen_mhd_iso module astronomix._fluid_equations._enforce_positivity module astronomix._fluid_equations._enforce_positivity_pallas module astronomix._fluid_equations._equations module astronomix._fluid_equations._equations_mhd module astronomix._fluid_equations._fluxes module astronomix._fluid_equations._fluxes_mhd module astronomix._fluid_equations.total_quantities module astronomix._geometry module astronomix._geometry.boundaries module astronomix._geometry.boundaries_legacy module astronomix._geometry.geometric_terms module astronomix._geometry.geometry module astronomix._integrators module astronomix._integrators._explicit_rk module astronomix._modules module astronomix._modules._conduction module astronomix._modules._conduction._conduction module astronomix._modules._cosmic_rays module astronomix._modules._cosmic_rays.cosmic_ray_options module astronomix._modules._cosmic_rays.cr_fluid_equations module astronomix._modules._cosmic_rays.cr_injection module astronomix._modules._iteration_level_updates module astronomix._modules._stellar_wind module astronomix._modules._stellar_wind.stellar_wind module astronomix._modules._stellar_wind.stellar_wind_options module astronomix._modules._stellar_wind.weaver module astronomix._modules._time_integrator_sources module astronomix._modules._viscosity module astronomix._modules._viscosity._viscosity module

astronomix._pallas_helpers module astronomix._snapshotting module astronomix._snapshotting._orbax_storage module astronomix._snapshotting._snapshot_diagnostics module astronomix._spatial_operators module astronomix._spatial_operators._differencing module astronomix._spatial_operators._interpolate module astronomix._stencil_operations module astronomix._stencil_operations._stencil_operations module astronomix.analysis_helpers module, [1] astronomix.analysis_helpers._exposed_rhs module astronomix.analysis_helpers.energy_spectrum module, [1] astronomix.analysis_helpers.jacobians module, [1] astronomix.data_classes module, [1] astronomix.data_classes.simulation_helper_data module, [1] astronomix.data_classes.simulation_snapshot_data module, [1] astronomix.data_classes.simulation_state_struct module, [1] astronomix.initial_condition_generation module, [1] astronomix.initial_condition_generation.construct_primitive_state module, [1] astronomix.initial_condition_generation.magnetic_field_from_vector_potential module, [1] astronomix.initial_condition_generation.turbulent_ic_generator module, [1] astronomix.option_classes module, [1] astronomix.option_classes.simulation_config module, [1] astronomix.option_classes.simulation_params module, [1] astronomix.plotting_helpers module, [1] astronomix.plotting_helpers.inset_box module, [1] astronomix.plotting_helpers.power_law_indicators module, [1] astronomix.setup_helpers module, [1] astronomix.setup_helpers.restart module, [1] astronomix.shock_finder module, [1] astronomix.shock_finder.shock_finder module, [1] astronomix.time_stepping module, [1] astronomix.time_stepping._progress_bar module astronomix.time_stepping._time_loop module astronomix.time_stepping._utils module astronomix.time_stepping.time_integration module, [1] astronomix.units module, [1] astronomix.units.unit_helpers module, [1]

B

backend (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] BACKWARDS (in module astronomix.option_classes.simulation_config) boundary_handling (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] boundary_settings (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

BoundarySettings (class in astronomix.option_classes.simulation_config), [1] BoundarySettings1D (class in astronomix.option_classes.simulation_config), [1] box_size (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] build_snapshot_store() (in module astronomix._snapshotting._snapshot_diagnostics)

C

C_cfl (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] calculate_gravitational_energy() (in module astronomix._fluid_equations.total_quantities) calculate_internal_energy() (in module astronomix._fluid_equations.total_quantities) calculate_kinetic_energy() (in module astronomix._fluid_equations.total_quantities) calculate_radial_momentum() (in module astronomix._fluid_equations.total_quantities) calculate_shell_profiles() (astronomix._modules._stellar_wind.weaver.Weaver method) calculate_total_energy() (in module astronomix._fluid_equations.total_quantities) calculate_total_mass() (in module astronomix._fluid_equations.total_quantities) CARTESIAN (in module astronomix.option_classes.simulation_config) cell_centers_x (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] cell_centers_y (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] cell_centers_z (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] cell_volumes (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] clamp_in_estimates (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] cnn_mhd_corrector_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] cnn_mhd_corrector_params (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] CodeUnits (class in astronomix.units.unit_helpers), [1] compute (astronomix._snapshotting._snapshot_diagnostics.SnapshotQuantity attribute) config_to_string() (in module astronomix.option_classes.simulation_config), [1]

cons_activate (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] cons_coeff (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] cons_passes (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] CONSERVATIVE_GAS_STATE (in module astronomix.option_classes.simulation_config) conserved_state_from_primitive() (in module astronomix._fluid_equations._equations) conserved_state_from_primitive_isothermal() (in module astronomix._fluid_equations._equations_mhd) conserved_state_from_primitive_mhd() (in module astronomix._fluid_equations._equations_mhd) construct_primitive_state() (in module astronomix.initial_condition_generation.construct_primitive_state), [1] cooling_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] cooling_params (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] cosmic_ray_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] cosmic_ray_params (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] cosmic_rays (astronomix._modules._cosmic_rays.cosmic_ray_options.CosmicRayConfig attribute) CosmicRayConfig (class in astronomix._modules._cosmic_rays.cosmic_ray_options) CosmicRayParams (class in astronomix._modules._cosmic_rays.cosmic_ray_options) create_turb_field() (in module astronomix.initial_condition_generation.turbulent_ic_generator), [1] current_checkpoint (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] custom_roll() (in module astronomix._stencil_operations._stencil_operations) CYLINDRICAL (in module astronomix.option_classes.simulation_config)

D

deepvoid_blend (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] deepvoid_blend_factor (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] default_positivity_protection (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] diff_safe_sqrt() (in module astronomix._fluid_equations._eigen_hydro) (in module astronomix._fluid_equations._eigen_hydro_iso) (in module astronomix._fluid_equations._eigen_mhd) (in module astronomix._fluid_equations._eigen_mhd_iso) diffable_pallas_call() (in module astronomix._pallas_helpers) diffable_pallas_call_n() (in module astronomix._pallas_helpers) differentiation_mode (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

diffusion (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] diffusive_shock_acceleration (astronomix._modules._cosmic_rays.cosmic_ray_options.CosmicRayConfig attribute) diffusive_shock_acceleration_efficiency (astronomix._modules._cosmic_rays.cosmic_ray_options.CosmicRayParams attribute) diffusive_shock_acceleration_start_time (astronomix._modules._cosmic_rays.cosmic_ray_options.CosmicRayParams attribute) dimensionality (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] donate_state (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] DOUBLE_MINMOD (in module astronomix.option_classes.simulation_config) DOUBLE_PRECISION (in module astronomix.option_classes.simulation_config) dt_max (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] dummy_multi_star_wind() (in module astronomix._modules._stellar_wind.stellar_wind) DYNAMIC_VISCOSITY (in module astronomix.option_classes.simulation_config)

E

EI (in module astronomix._modules._stellar_wind.stellar_wind_options) enabled_snapshot_quantities() (in module astronomix._snapshotting._snapshot_diagnostics)

equation_of_state (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] exact_end_time (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] external_potential (astronomix.option_classes.simulation_config.GravityConfig attribute), [1]

F

fd_conduction_source() (in module astronomix._modules._conduction._conduction) fd_viscosity_source() (in module astronomix._modules._viscosity._viscosity) field (astronomix._snapshotting._snapshot_diagnostics.SnapshotQuantity attribute) FIELD_TYPE (in module astronomix.option_classes.simulation_config) final_index (astronomix.time_stepping._time_loop.SnapshotSpec attribute) final_state (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] finalize_config() (in module astronomix.option_classes.simulation_config), [1] find_shock_zone() (in module astronomix.shock_finder.shock_finder), [1] FINITE_DIFFERENCE (in module astronomix.option_classes.simulation_config) finite_difference_int6() (in module astronomix._spatial_operators._differencing) FINITE_VOLUME (in module astronomix.option_classes.simulation_config) first_order_fallback (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] FIXED_BOUNDARY (in module astronomix.option_classes.simulation_config)

FIXED_BOUNDARY_OPEN_MOMENTUM (in module astronomix.option_classes.simulation_config) fixed_boundary_state (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] FIXED_STEP (in module astronomix.time_stepping._time_loop) fixed_timestep (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] FixedBoundaryState (class in astronomix.option_classes.simulation_params), [1] FixedBoundaryState1D (class in astronomix.option_classes.simulation_params), [1] forcing (astronomix._snapshotting._orbax_storage.LoopCheckpoint attribute) (astronomix.time_stepping.time_integration.LoopState attribute), [1], [2] FORWARDS (in module astronomix.option_classes.simulation_config) FOURTH_ORDER_CONSERVATIVE (in module astronomix.option_classes.simulation_config) frame_tracking (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] fv_magnetic_integrator (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] fv_viscosity_update() (in module astronomix._modules._viscosity._viscosity)

G

gamma (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] gamma_cr (in module astronomix._modules._cosmic_rays.cr_fluid_equations) gamma_gas (in module astronomix._modules._cosmic_rays.cr_fluid_equations) gas_pressure_from_primitives_with_crs() (in module astronomix._modules._cosmic_rays.cr_fluid_equations) geometric_centers (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] geometry (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] get_absolute_velocity() (in module astronomix._fluid_equations._equations) get_critical_radius() (astronomix._modules._stellar_wind.weaver.Weaver method) get_cross_helicity_spectrum() (in module astronomix.analysis_helpers.energy_spectrum), [1] get_density_profile() (astronomix._modules._stellar_wind.weaver.Weaver method) get_helper_data() (in module astronomix.data_classes.simulation_helper_data), [1] get_inner_shock_radius() (astronomix._modules._stellar_wind.weaver.Weaver method) get_kinetic_energy_spectrum() (in module astronomix.analysis_helpers.energy_spectrum), [1] get_magnetic_energy_spectrum() (in module astronomix.analysis_helpers.energy_spectrum), [1] get_magnetic_helicity_spectrum() (in module astronomix.analysis_helpers.energy_spectrum), [1]

get_outer_shock_radius() (astronomix._modules._stellar_wind.weaver.Weaver method) get_pressure_profile() (astronomix._modules._stellar_wind.weaver.Weaver method) get_radial_range_free_wind() (astronomix._modules._stellar_wind.weaver.Weaver method) get_radial_range_undisturbed_ism() (astronomix._modules._stellar_wind.weaver.Weaver method) get_radial_range_wind_interior() (astronomix._modules._stellar_wind.weaver.Weaver method) get_temperature_from_internal_energy() (astronomix.units.unit_helpers.CodeUnits method), [1] get_velocity_profile() (astronomix._modules._stellar_wind.weaver.Weaver method) GHOST_CELLS (in module astronomix.option_classes.simulation_config) gpu_compute_capability_at_least_80() (in module astronomix.option_classes.simulation_config), [1] gravitational_constant (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] gravitational_energy (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] gravitational_potential (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] gravity (astronomix.option_classes.simulation_config.GravityConfig attribute), [1] gravity_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] GravityConfig (class in astronomix.option_classes.simulation_config), [1] grid_spacing (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

H

helicity_spectrum (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] HelperData (class in astronomix.data_classes.simulation_helper_data), [1] HelperDataRequirements (class in astronomix.data_classes.simulation_helper_data), [1] HLL (in module astronomix.option_classes.simulation_config)

HLLC (in module astronomix.option_classes.simulation_config) HLLC_LM (in module astronomix.option_classes.simulation_config) host_helper_data (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] HYBRID_HLLC (in module astronomix.option_classes.simulation_config)

I

IDEAL_GAS (in module astronomix.option_classes.simulation_config) IMPLICIT_EULER (in module astronomix.option_classes.simulation_config) IMPLICIT_MIDPOINT (in module astronomix.option_classes.simulation_config) init_from_unit_params() (astronomix.units.unit_helpers.CodeUnits method), [1] inject_crs_at_strongest_shock() (in module astronomix._modules._cosmic_rays.cr_injection) inner_cell_boundaries (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] INTEGER_BINNING (in module astronomix.analysis_helpers.energy_spectrum)

integrate() (in module astronomix.time_stepping._time_loop) internal_energy (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] internal_energy_from_energy() (in module astronomix._fluid_equations._equations) interp_center_to_face() (in module astronomix._spatial_operators._interpolate) interp_face_to_center() (in module astronomix._spatial_operators._interpolate) is_enabled (astronomix._snapshotting._snapshot_diagnostics.SnapshotQuantity attribute) ISOTHERMAL (in module astronomix.option_classes.simulation_config) isothermal_sound_speed (astronomix.option_classes.simulation_params.SimulationParams attribute), [1]

K

k_spectra (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] key (astronomix._snapshotting._orbax_storage.LoopCheckpoint attribute) (astronomix.time_stepping.time_integration.LoopState attribute), [1], [2]

KINEMATIC_VISCOSITY (in module astronomix.option_classes.simulation_config) kinetic_energy (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] kinetic_energy_spectrum (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1]

L

latest_checkpoint_step() (in module astronomix.setup_helpers) (in module astronomix.setup_helpers.restart), [1] latest_step() (in module astronomix._snapshotting._orbax_storage) LAX_FRIEDRICHS (in module astronomix.option_classes.simulation_config) left_boundary (astronomix.option_classes.simulation_config.BoundarySettings1D attribute), [1] left_state (astronomix.option_classes.simulation_params.FixedBoundaryState1D attribute), [1] limiter (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

limiter_to_string() (in module astronomix.option_classes.simulation_config), [1] load_loop_checkpoint() (in module astronomix._snapshotting._orbax_storage) LOG_BINNING (in module astronomix.analysis_helpers.energy_spectrum) loop_checkpointer() (in module astronomix._snapshotting._orbax_storage) LoopCheckpoint (class in astronomix._snapshotting._orbax_storage) LoopState (class in astronomix.time_stepping.time_integration), [1] lsrk4() (in module astronomix._integrators._explicit_rk)

M

magnetic_divergence (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] magnetic_energy_spectrum (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] MAGNETIC_FIELD_ONLY (in module astronomix.option_classes.simulation_config) MEI (in module astronomix._modules._stellar_wind.stellar_wind_options) memory_analysis (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] MEO (in module astronomix._modules._stellar_wind.stellar_wind_options) mhd (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] MHD_JET_BOUNDARY (in module astronomix.option_classes.simulation_config) minimum_density (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] minimum_pressure (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] MINMOD (in module astronomix.option_classes.simulation_config) module astronomix, [1] astronomix._fluid_equations astronomix._fluid_equations._eigen_hydro astronomix._fluid_equations._eigen_hydro_iso astronomix._fluid_equations._eigen_mhd astronomix._fluid_equations._eigen_mhd_iso astronomix._fluid_equations._enforce_positivity astronomix._fluid_equations._enforce_positivity_pallas astronomix._fluid_equations._equations astronomix._fluid_equations._equations_mhd astronomix._fluid_equations._fluxes astronomix._fluid_equations._fluxes_mhd astronomix._fluid_equations.total_quantities astronomix._geometry astronomix._geometry.boundaries astronomix._geometry.boundaries_legacy astronomix._geometry.geometric_terms astronomix._geometry.geometry astronomix._integrators astronomix._integrators._explicit_rk astronomix._modules astronomix._modules._conduction astronomix._modules._conduction._conduction astronomix._modules._cosmic_rays astronomix._modules._cosmic_rays.cosmic_ray_options astronomix._modules._cosmic_rays.cr_fluid_equations astronomix._modules._cosmic_rays.cr_injection astronomix._modules._iteration_level_updates astronomix._modules._stellar_wind astronomix._modules._stellar_wind.stellar_wind astronomix._modules._stellar_wind.stellar_wind_options astronomix._modules._stellar_wind.weaver astronomix._modules._time_integrator_sources astronomix._modules._viscosity astronomix._modules._viscosity._viscosity astronomix._pallas_helpers astronomix._snapshotting astronomix._snapshotting._orbax_storage astronomix._snapshotting._snapshot_diagnostics astronomix._spatial_operators astronomix._spatial_operators._differencing astronomix._spatial_operators._interpolate astronomix._stencil_operations astronomix._stencil_operations._stencil_operations astronomix.analysis_helpers, [1] astronomix.analysis_helpers._exposed_rhs astronomix.analysis_helpers.energy_spectrum, [1] astronomix.analysis_helpers.jacobians, [1] astronomix.data_classes, [1] astronomix.data_classes.simulation_helper_data, [1] astronomix.data_classes.simulation_snapshot_data, [1] astronomix.data_classes.simulation_state_struct, [1] astronomix.initial_condition_generation, [1] astronomix.initial_condition_generation.construct_primitive_state, [1] astronomix.initial_condition_generation.magnetic_field_from_vector_potential, [1] astronomix.initial_condition_generation.turbulent_ic_generator, [1] astronomix.option_classes, [1] astronomix.option_classes.simulation_config, [1] astronomix.option_classes.simulation_params, [1] astronomix.plotting_helpers, [1] astronomix.plotting_helpers.inset_box, [1] astronomix.plotting_helpers.power_law_indicators, [1] astronomix.setup_helpers, [1] astronomix.setup_helpers.restart, [1] astronomix.shock_finder, [1] astronomix.shock_finder.shock_finder, [1] astronomix.time_stepping, [1] astronomix.time_stepping._progress_bar astronomix.time_stepping._time_loop astronomix.time_stepping._utils astronomix.time_stepping.time_integration, [1] astronomix.units, [1] astronomix.units.unit_helpers, [1]

MUSCL (in module astronomix.option_classes.simulation_config)

N

nan_safe (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] NATIVE_JAX (in module astronomix.option_classes.simulation_config) needs_cell_centers_x (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_cell_centers_y (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_cell_centers_z (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_cell_volumes (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_geometric_centers (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_inner_cell_boundaries (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_outer_cell_boundaries (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_r (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_r_hat_alpha (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1] needs_volumetric_centers (astronomix.data_classes.simulation_helper_data.HelperDataRequirements attribute), [1]

neural_net_force_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] neural_net_force_params (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] num_cells (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] num_checkpoints (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] num_ghost_cells (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] num_injection_cells (astronomix._modules._stellar_wind.stellar_wind_options.WindConfig attribute) num_iterations (astronomix._snapshotting._orbax_storage.LoopCheckpoint attribute) (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] num_snapshots (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] num_temperature_bins (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] num_timesteps (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] numerical_precision (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

O

ON_DEVICE (in module astronomix.option_classes.simulation_config) OPEN_BOUNDARY (in module astronomix.option_classes.simulation_config)

OPTIMAL_BACKEND (in module astronomix.option_classes.simulation_config), [1] OSHER (in module astronomix.option_classes.simulation_config) outer_cell_boundaries (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1]

P

PALLAS (in module astronomix.option_classes.simulation_config) pallas_block_shape (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] pallas_ct (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] pallas_interpret (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] pallas_mesh_context() (in module astronomix._pallas_helpers) pallas_num_warps (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] pallas_use_triton (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] pallas_vjp_call() (in module astronomix._pallas_helpers) per_snapshot_shape (astronomix._snapshotting._snapshot_diagnostics.SnapshotQuantity attribute) per_stage_mode (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] per_step_mode (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] PERIODIC_BOUNDARY (in module astronomix.option_classes.simulation_config) PERIODIC_ROLL (in module astronomix.option_classes.simulation_config) PHYSICAL_BINNING (in module astronomix.analysis_helpers.energy_spectrum) point_values_to_averages() (in module astronomix._spatial_operators._interpolate) point_values_to_averages_single_axis() (in module astronomix._spatial_operators._interpolate) poisson_manual_open_boundaries (astronomix.option_classes.simulation_config.GravityConfig attribute), [1] positivity_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] POSITIVITY_CONSERVATIVE (in module astronomix.option_classes.simulation_config), [1]

POSITIVITY_HARD_FLOOR (in module astronomix.option_classes.simulation_config) positivity_max_velocity (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] POSITIVITY_NONE (in module astronomix.option_classes.simulation_config) POSITIVITY_REDISTRIBUTE (in module astronomix.option_classes.simulation_config) PositivityConfig (class in astronomix.option_classes.simulation_config), [1] preserving_flux (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] pressure_floor (astronomix._modules._stellar_wind.stellar_wind_options.WindParams attribute) pressure_from_energy() (in module astronomix._fluid_equations._equations) pressure_from_internal_energy() (in module astronomix._fluid_equations._equations) PRIMITIVE_GAS_STATE (in module astronomix.option_classes.simulation_config) primitive_state (astronomix._snapshotting._orbax_storage.LoopCheckpoint attribute) (astronomix.data_classes.simulation_state_struct.StateStruct attribute), [1] (astronomix.time_stepping.time_integration.LoopState attribute), [1], [2] primitive_state_from_conserved() (in module astronomix._fluid_equations._equations) primitive_state_from_conserved_isothermal() (in module astronomix._fluid_equations._equations_mhd) primitive_state_from_conserved_mhd() (in module astronomix._fluid_equations._equations_mhd) print_elapsed_time (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] print_simulation_parameters() (astronomix.units.unit_helpers.CodeUnits method), [1] progress_bar (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

R

r (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] r_hat_alpha (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1] radial_momentum (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] random_seed (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] reconstruction_order (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] record (astronomix.time_stepping._time_loop.SnapshotSpec attribute) record_final (astronomix.time_stepping._time_loop.SnapshotSpec attribute) record_snapshot() (in module astronomix._snapshotting._snapshot_diagnostics) REFLECTIVE_BOUNDARY (in module astronomix.option_classes.simulation_config) restart_from_latest_checkpoint() (in module astronomix.setup_helpers) (in module astronomix.setup_helpers.restart), [1] return_final_state (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_gravitational_energy (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_helicity_spectrum (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_internal_energy (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_kinetic_energy (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_kinetic_energy_spectrum (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1]

return_magnetic_divergence (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_magnetic_energy_spectrum (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_radial_momentum (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_snapshots (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] return_states (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_temperature_pdf (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_total_energy (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] return_total_mass (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] riemann_solver (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] riemann_solver_to_string() (in module astronomix.option_classes.simulation_config), [1] right_boundary (astronomix.option_classes.simulation_config.BoundarySettings1D attribute), [1] right_state (astronomix.option_classes.simulation_params.FixedBoundaryState1D attribute), [1] RK2_SSP (in module astronomix.option_classes.simulation_config) rk2_ssp() (in module astronomix._integrators._explicit_rk) RK4_LSRK (in module astronomix.option_classes.simulation_config) RK4_SSP (in module astronomix.option_classes.simulation_config) runtime (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] runtime_debugging (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

S

save_loop_checkpoint() (in module astronomix._snapshotting._orbax_storage) SECOND_ORDER_CONSERVATIVE (in module astronomix.option_classes.simulation_config) self_gravity (astronomix.option_classes.simulation_config.GravityConfig attribute), [1] self_gravity_version (astronomix.option_classes.simulation_config.GravityConfig attribute), [1] setup_magnetic_fields_from_vector_potential() (in module astronomix.initial_condition_generation.magnetic_field_from_vector_potential), [1] shock_criteria() (in module astronomix.shock_finder.shock_finder), [1] shock_sensor() (in module astronomix.shock_finder.shock_finder), [1] should_record (astronomix.time_stepping._time_loop.SnapshotSpec attribute) SIMPLE_SOURCE (in module astronomix.option_classes.simulation_config) SimulationConfig (class in astronomix.option_classes.simulation_config), [1] SimulationParams (class in astronomix.option_classes.simulation_params), [1] SINGLE_PRECISION (in module astronomix.option_classes.simulation_config) single_xmode_jacobian2Dt() (in module astronomix.analysis_helpers.jacobians), [1] single_xmode_rhs_jacobian2D() (in module astronomix.analysis_helpers.jacobians), [1] SNAPSHOT_QUANTITIES (in module astronomix._snapshotting._snapshot_diagnostics) snapshot_settings (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] snapshot_storage_mode (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] snapshot_storage_path (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] snapshot_timepoints (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] SnapshotData (class in astronomix.data_classes.simulation_snapshot_data), [1] SnapshotQuantity (class in astronomix._snapshotting._snapshot_diagnostics) SnapshotSettings (class in astronomix.option_classes.simulation_config), [1]

SnapshotSpec (class in astronomix.time_stepping._time_loop) solver_mode (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] solver_mode_to_string() (in module astronomix.option_classes.simulation_config), [1] source_term_aware_timestep (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] spectra_requested() (in module astronomix._snapshotting._snapshot_diagnostics) spectral_wavenumbers() (in module astronomix._snapshotting._snapshot_diagnostics) speed_of_sound() (in module astronomix._fluid_equations._equations) speed_of_sound_crs() (in module astronomix._modules._cosmic_rays.cr_fluid_equations) SPHERICAL (in module astronomix.option_classes.simulation_config) split (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] SPLIT (in module astronomix.option_classes.simulation_config) ssprk4() (in module astronomix._integrators._explicit_rk) state_struct (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] STATE_TYPE (in module astronomix.option_classes.simulation_config) STATE_TYPE_ALTERED (in module astronomix.option_classes.simulation_config) states (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] StateStruct (class in astronomix.data_classes.simulation_state_struct), [1] StaticFloatVector (class in astronomix.option_classes.simulation_config), [1] StaticIntVector (class in astronomix.option_classes.simulation_config), [1] stellar_wind (astronomix._modules._stellar_wind.stellar_wind_options.WindConfig attribute) step (astronomix._snapshotting._orbax_storage.LoopCheckpoint attribute) store (astronomix.time_stepping._time_loop.SnapshotSpec attribute) SUPERBEE (in module astronomix.option_classes.simulation_config)

T

t_end (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] t_start (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] temperature_pdf (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] temperature_pdf_max (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] temperature_pdf_min (astronomix.option_classes.simulation_config.SnapshotSettings attribute), [1] temporary_memory_bytes (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] thermal_conduction (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] thermal_conductivity (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] thermal_pressure_from_energy_mhd() (in module astronomix._fluid_equations._equations_mhd) time (astronomix._snapshotting._orbax_storage.LoopCheckpoint attribute) time_integration() (in module astronomix.time_stepping.time_integration), [1] time_integrator (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] time_points (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1]

times_close() (in module astronomix.time_stepping._time_loop) TO_DISK (in module astronomix.option_classes.simulation_config) total_energy (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] total_energy_from_primitives() (in module astronomix._fluid_equations._equations) total_energy_from_primitives_mhd() (in module astronomix._fluid_equations._equations_mhd) total_energy_from_primitives_with_crs() (in module astronomix._modules._cosmic_rays.cr_fluid_equations) total_mass (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] total_memory_bytes (astronomix.data_classes.simulation_snapshot_data.SnapshotData attribute), [1] total_pressure_from_conserved_mhd() (in module astronomix._fluid_equations._equations_mhd) total_pressure_from_conserved_with_crs() (in module astronomix._modules._cosmic_rays.cr_fluid_equations) trace_wind_density (astronomix._modules._stellar_wind.stellar_wind_options.WindConfig attribute) turbulent_forcing_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] turbulent_forcing_params (astronomix.option_classes.simulation_params.SimulationParams attribute), [1]

U

UNSPLIT (in module astronomix.option_classes.simulation_config)

use_max_adaptive_timestep (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] use_specific_snapshot_timepoints (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1]

V

vacuum_rest (astronomix.option_classes.simulation_config.PositivityConfig attribute), [1] VAN_ALBADA (in module astronomix.option_classes.simulation_config) VAN_ALBADA_PP (in module astronomix.option_classes.simulation_config) VARAXIS (in module astronomix.option_classes.simulation_config) vector_field_cross_spectrum() (in module astronomix.analysis_helpers.energy_spectrum), [1]

vector_field_energy_spectrum() (in module astronomix.analysis_helpers.energy_spectrum), [1] VELOCITY_ONLY (in module astronomix.option_classes.simulation_config) viscosity (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] viscosity_type (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] volumetric_centers (astronomix.data_classes.simulation_helper_data.HelperData attribute), [1]

W

Weaver (class in astronomix._modules._stellar_wind.weaver) wind_config (astronomix.option_classes.simulation_config.SimulationConfig attribute), [1] wind_final_velocity (astronomix._modules._stellar_wind.stellar_wind_options.WindParams attribute) wind_injection_scheme (astronomix._modules._stellar_wind.stellar_wind_options.WindConfig attribute)

wind_mass_loss_rate (astronomix._modules._stellar_wind.stellar_wind_options.WindParams attribute) wind_params (astronomix.option_classes.simulation_params.SimulationParams attribute), [1] WindConfig (class in astronomix._modules._stellar_wind.stellar_wind_options) WindParams (class in astronomix._modules._stellar_wind.stellar_wind_options)

X

x (astronomix.option_classes.simulation_config.BoundarySettings attribute), [1] (astronomix.option_classes.simulation_config.StaticFloatVector attribute), [1] (astronomix.option_classes.simulation_config.StaticIntVector attribute), [1] (astronomix.option_classes.simulation_params.FixedBoundaryState attribute), [1]

XAXIS (in module astronomix.option_classes.simulation_config)

Y

y (astronomix.option_classes.simulation_config.BoundarySettings attribute), [1] (astronomix.option_classes.simulation_config.StaticFloatVector attribute), [1] (astronomix.option_classes.simulation_config.StaticIntVector attribute), [1] (astronomix.option_classes.simulation_params.FixedBoundaryState attribute), [1]

YAXIS (in module astronomix.option_classes.simulation_config)

Z

z (astronomix.option_classes.simulation_config.BoundarySettings attribute), [1] (astronomix.option_classes.simulation_config.StaticFloatVector attribute), [1] (astronomix.option_classes.simulation_config.StaticIntVector attribute), [1] (astronomix.option_classes.simulation_params.FixedBoundaryState attribute), [1]

ZAXIS (in module astronomix.option_classes.simulation_config)