2.4.2 Heterogeneous boundary conditions

This example implements a spatially coupled SIR model with the following dynamics for the density of susceptible, infected, and recovered individuals:

\[\begin{split}\partial_t s &= D \nabla^2 s - \beta is \\ \partial_t i &= D \nabla^2 i + \beta is - \gamma i \\ \partial_t r &= D \nabla^2 r + \gamma i\end{split}\]

Here, \(D\) is the diffusivity, \(\beta\) the infection rate, and \(\gamma\) the recovery rate.

  0%|          | 0/10.0 [00:00<?, ?it/s]
Initializing:   0%|          | 0/10.0 [00:00<?, ?it/s]
  0%|          | 0/10.0 [00:00<?, ?it/s]
  3%|▎         | 0.3/10.0 [00:00<00:04,  1.97it/s]
 10%|█         | 1.04/10.0 [00:00<00:01,  5.31it/s]
 46%|████▋     | 4.63/10.0 [00:00<00:00, 11.26it/s]
 46%|████▋     | 4.63/10.0 [00:00<00:00,  6.32it/s]
100%|██████████| 10.0/10.0 [00:00<00:00, 13.64it/s]
100%|██████████| 10.0/10.0 [00:00<00:00, 13.64it/s]

import numpy as np

from pde import CartesianGrid, DiffusionPDE, ScalarField

# define grid and an initial state
grid = CartesianGrid([[-5, 5], [-5, 5]], 32)
field = ScalarField(grid)


# define the boundary conditions, which here are calculated from a function
def bc_value(adjacent_value, dx, x, y, t):
    """Return boundary value."""
    return np.sign(x)


# define and solve a simple diffusion equation
eq = DiffusionPDE(bc={"*": {"derivative": 0}, "y+": {"value_expression": bc_value}})
res = eq.solve(field, t_range=10, dt=0.01, backend="numpy")
res.plot()