patchwork

Install from GitHub with

pip install git+https://github.com/max-models/patchwork

Patchwork stores cell-centered or node-centered fields on nested rectangular patches using mpiarray. The same API works with and without MPI; under mpiexec, each patch lives on the ranks you give it, and exchange_halos fills every patch’s halo from its neighbours and the level below.

from patchwork import AMRHierarchy, plot_patches

mesh = AMRHierarchy(base_shape=(32, 24))
fine = mesh.add_patch(level=1, box=((16, 48), (12, 36)))
mesh.allocate_field("density")
fine.field("density").local[:] = 1.0
fig, ax = plot_patches(mesh, field="density")

It includes composite Poisson solvers (cell-centered finite volumes, and a node-centered solver with Dirichlet and Neumann faces, periodic axes and embedded bodies, factorized once for particle-in-cell codes), and regrid, which places properly nested patches over tagged cells with Berger-Rigoutsos clustering.

You can find the documentation here.

Check out the code on GitHub!