warpSPH

(formerly compressibleSPH; the SPH core previously hosted at this URL now lives at wi-re/warpSPHCore)

A GPU-oriented Python package for compressible Smoothed Particle Hydrodynamics (SPH) experiments and benchmarks.

The project is organized around reusable builders for simulation configuration, interchangeable SPH schemes, adaptive time integration, case setup helpers, and export/import utilities for trajectories and postprocessing.

What This Package Implements

Core SPH Schemes

The package currently exposes three compressible SPH schemes via CompressibleSPHScheme:

• Monaghan
• CompSPH
• CRKSPH

Scheme dispatch is handled by buildScheme(...) in src/warpSPH/schemes/builder.py.

Kernels

Kernel selection is part of SimulationConfig.kernel (type KernelFunctions, imported from warpSPHCore).

The parser helper parseKernelFunctions(...) accepts any kernel enum name exposed by your installed warpSPHCore version.

Integrators

Time integration is selected with SimulationConfig.integrationScheme (type IntegrationSchemeType, from warpSPHIntegrators).

The helper buildConfig(...) returns both:

• a configured SimulationConfig
• the integrator object from getIntegrator(...)

Spatial Operators and Support Modes

Simulation configuration includes selectable operator and support modes:

• supportMode (SupportScheme)
• gradientMode (GradientScheme)
• laplacianMode (LaplacianScheme)
• samplingScheme (SamplingScheme)

Viscosity Switches

ViscositySwitch includes:

• Balsara1995
• Colagrossi2004
• CullenDehnen2010
• CullenHopkins
• MorrisMonaghan1997
• Rosswog2000
• NoneSwitch

Adaptive Support Schemes

AdaptiveSupportScheme includes:

• NoScheme
• Monaghan
• Owen

Energy Schemes

EnergyScheme includes:

• equalWork
• PdV
• diminishing
• monotonic
• hybrid
• CRK

I/O and Reproducibility Utilities

The package provides trajectory/state serialization utilities:

• prepExport(...)
• exportSimulationSystem(...)
• importSimulationSystem(...)
• importConfigs(...)

These write/read HDF5 state data and JSON config metadata for reproducible runs.

Package Layout

• src/warpSPH/configurations/: simulation and scheme configuration dataclasses/builders
• src/warpSPH/schemes/: scheme implementations and scheme builder
• src/warpSPH/systems/: state/system containers
• src/warpSPH/modules/: runtime modules (for example timestep support)
• src/warpSPH/sample/: sampling utilities
• src/warpSPH/caseUtils/: case setup helpers used by examples
• src/warpSPH/io.py: export/import and parsing helpers
• examples/compressible/: benchmark notebooks and generated media

Environment Setup

Recommended setup uses a dedicated conda environment and editable installs for backend repositories.

mkdir warp-sph-stack
cd warp-sph-stack
git clone https://github.com/wi-re/warpSPHIntegrators
git clone https://github.com/wi-re/warpSPHPlotting
git clone https://github.com/wi-re/warpSPHCore
git clone https://github.com/wi-re/warpSPH
conda create -n warp_env python=3.14
conda activate warp_env
pip install tqdm seaborn pandas pyglet dill numba scipy scikit-image h5py ipykernel ipympl
pip install -e warpSPHIntegrators/ warpSPHPlotting/ warpSPHCore/ 
cd warpSPH
pip install -e .

Current status: cloning and editable installs are the accepted setup path for now.

Quick Start

If you are running directly from this repository, ensure src is on your Python path:

export PYTHONPATH="$PWD/src:$PYTHONPATH"

Minimal usage pattern:

import torch
from warpSPH import *

# Build global simulation config + integrator
config, integrator = buildConfig(
    dim=1,
    dt=1e-3,
    adaptiveDt=True,
    cflFactor=0.3,
)

# Pick an SPH scheme
scheme = CompressibleSPHScheme.CRKSPH
SimulationSystem, SimulationState, SimulationConfig, SimulationUpdate, fn, export_fn, import_fn = buildScheme(scheme)

# Build scheme-specific config
schemeConfig = SimulationConfig()
schemeConfig.gamma = 5.0 / 3.0

# Case setup is typically done with helpers in warpSPH.caseUtils

Compressible Examples

A full gallery page with previews and embedded videos is available at:

• examples/compressible/EXAMPLES_SUMMARY.md

Table of Cases

Case	Notebook	Preview (PNG)	Video (MP4)
01. Sod Shock Tube (1D)	01-Sod_Shock_Tube_1D.ipynb		MP4
02. Linear Wave	02-Linear_Wave.ipynb		MP4
03. Kidder Isentropic Compression	03-Kidder_Isentropic_Compression.ipynb		MP4
04. Noh Implosion	04-Noh_Implosion.ipynb		MP4
05. Woodward-Colella Double Blastwave	05-Woodward_Colella.ipynb		MP4
06. Sedov-Taylor Blastwave (1D)	06-Sedov_Taylor_Blastwave_1D.ipynb		MP4
07. Sedov-Taylor Blastwave (2D)	07-Sedov_Taylor_Blastwave_2D.ipynb		MP4
08. Hydrostatic	08-Hydrostatic.ipynb		MP4
09. Gresho-Chan Vortex	09-Gresho_Chan_Vortex.ipynb		MP4
10. Yee Vortex	10-Yee_Vortex.ipynb		MP4
11. Shearing Noh Implosion (2D)	11-Shearing_Noh_Implosion_2D.ipynb		MP4
12. Kelvin-Helmholtz	12-Kelvin-Helmholtz.ipynb		MP4
13. Rayleigh-Taylor	13-Rayleigh_Taylor.ipynb		MP4
14. Triple Point (Equal Resolution)	14-Triple_point.ipynb		MP4
15. Triple Point (Equal Mass)	15-Triple_point_equalMass.ipynb

Precision Note (Notebook Workflows)

In the example notebooks, floating-point precision is configured early in the import/setup cell. If precision is changed (for example float32 to float64), restart the Jupyter kernel before rerunning.