2018-10-03 18:59:01 +00:00
|
|
|
# setup: import numpy as np ; grid_shape = (512, 512) ; grid = np.zeros(grid_shape) ; block_low = int(grid_shape[0] * .4) ; block_high = int(grid_shape[0] * .5) ; grid[block_low:block_high, block_low:block_high] = 0.005 # noqa
|
|
|
|
|
# run: evolve(grid, 0.1)
|
|
|
|
|
# from: High Performance Python by Micha Gorelick and Ian Ozsvald,
|
|
|
|
|
# http://shop.oreilly.com/product/0636920028963.do
|
2018-04-05 22:07:33 +00:00
|
|
|
|
2018-10-03 18:59:01 +00:00
|
|
|
# pythran export evolve(float64[][], float)
|
2018-10-03 12:38:48 +00:00
|
|
|
|
2018-04-05 22:07:33 +00:00
|
|
|
import numpy as np
|
2018-10-03 12:38:48 +00:00
|
|
|
|
|
|
|
|
|
2018-04-05 22:07:33 +00:00
|
|
|
def laplacian(grid):
|
2018-10-03 12:38:48 +00:00
|
|
|
return (np.roll(grid, +1, 0) + np.roll(grid, -1, 0)
|
|
|
|
|
+ np.roll(grid, +1, 1) + np.roll(grid, -1, 1) - 4 * grid)
|
|
|
|
|
|
2018-04-05 22:07:33 +00:00
|
|
|
|
|
|
|
|
def evolve(grid, dt, D=1):
|
|
|
|
|
return grid + dt * D * laplacian(grid)
|
