astronomix.data_classes.simulation_snapshot_data

astronomix.data_classes.simulation_snapshot_data#

Snapshot return container for the time integration.

Holds the per-snapshot diagnostics (states, energies, spectra, …) plus run-level metadata (runtime, iteration count, compiled-step memory usage) returned by astronomix.time_stepping.time_integration.time_integration() when config.return_snapshots is set.

Module Contents#

Classes#

SnapshotData

Return format for the time integration, when snapshots are requested.

API#

class astronomix.data_classes.simulation_snapshot_data.SnapshotData[source]#

Bases: typing.NamedTuple

Return format for the time integration, when snapshots are requested.

time_points: jax.numpy.ndarray = None#
states: jax.numpy.ndarray = None#
final_state: jax.numpy.ndarray = None#
total_mass: jax.numpy.ndarray = None#
total_energy: jax.numpy.ndarray = None#
internal_energy: jax.numpy.ndarray = None#
kinetic_energy: jax.numpy.ndarray = None#
gravitational_energy: jax.numpy.ndarray = None#
radial_momentum: jax.numpy.ndarray = None#
magnetic_divergence: jax.numpy.ndarray = None#
kinetic_energy_spectrum: jax.numpy.ndarray = None#
magnetic_energy_spectrum: jax.numpy.ndarray = None#
helicity_spectrum: jax.numpy.ndarray = None#
k_spectra: jax.numpy.ndarray = None#
temperature_pdf: jax.numpy.ndarray = None#
runtime: float = 0.0#
num_iterations: int = 0#
temporary_memory_bytes: int = 0#
argument_memory_bytes: int = 0#
total_memory_bytes: int = 0#
current_checkpoint: int = 0#
Contents