37 lines
888 B
Python
37 lines
888 B
Python
"""Compiling with pythran."""
|
|
|
|
#compiling: pythran sandpile_pythran.py
|
|
|
|
import numpy as np
|
|
|
|
|
|
#pythran export apply_gravity(int[][])
|
|
|
|
def apply_gravity(terrain):
|
|
shape = np.shape(terrain)
|
|
while True:
|
|
did_someting = False
|
|
for x, y in np.ndindex(shape):
|
|
if terrain[x, y] >= 4:
|
|
div, terrain[x, y] = divmod(terrain[x][y], 4)
|
|
terrain[x - 1][y] += div
|
|
terrain[x + 1][y] += div
|
|
terrain[x][y + 1] += div
|
|
terrain[x][y - 1] += div
|
|
did_someting = True
|
|
|
|
if not did_someting:
|
|
return
|
|
|
|
|
|
|
|
#pythran export main(int, bool)
|
|
|
|
def main(height, show):
|
|
width = int(height ** .5) + 1
|
|
terrain = np.zeros((width, width), dtype=np.int64)
|
|
terrain[width // 2, width // 2] = height
|
|
apply_gravity(terrain)
|
|
if show:
|
|
print(np.array(terrain))
|