⚠️ Warning: This is a draft ⚠️
This means it might contain formatting issues, incorrect code, conceptual problems, or other severe issues.
If you want to help to improve and eventually enable this page, please fork RosettaGit's repository and open a merge request on GitHub.
{{task}}{{wikipedia|Abelian sandpile model}}
Implement the '''Abelian sandpile model''' also known as '''Bak–Tang–Wiesenfeld model'''. It's history, mathematical definition and properties can be found under it's [https://en.wikipedia.org/wiki/Abelian_sandpile_model wikipedia article].
The task requires the creation of a 2D grid of arbitrary size on which "piles of sand" can be placed. Any "pile" that has 4 or more sand particles on it ''collapses'', resulting in '''four particles being subtracted from the pile''' and '''distributed among it's neighbors.'''
It is recommended to display the output in some kind of image format, as terminal emulators are usually too small to display images larger than a few dozen characters tall. As an example of how to accomplish this, see the [https://rosettacode.org/wiki/Bitmap/Write_a_PPM_file Bitmap/Write a PPM file] task.
'''Examples:'''
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 1 0 0
0 0 4 0 0 -> 0 1 0 1 0
0 0 0 0 0 0 0 1 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 1 0 0
0 0 6 0 0 -> 0 1 2 1 0
0 0 0 0 0 0 0 1 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 1 0 0
0 0 0 0 0 0 2 1 2 0
0 0 16 0 0 -> 1 1 0 1 1
0 0 0 0 0 0 2 1 2 0
0 0 0 0 0 0 0 1 0 0
=={{header|Fōrmulæ}}==
In [https://wiki.formulae.org/Abelian_sandpile_model this] page you can see the solution of this task.
Fōrmulæ programs are not textual, visualization/edition of programs is done showing/manipulating structures but not text ([http://wiki.formulae.org/Editing_F%C5%8Drmul%C3%A6_expressions more info]). Moreover, there can be multiple visual representations of the same program. Even though it is possible to have textual representation —i.e. XML, JSON— they are intended for transportation effects more than visualization and edition.
The option to show Fōrmulæ programs and their results is showing images. Unfortunately images cannot be uploaded in Rosetta Code.
Forth
{{works with|gforth|0.7.3}}
#! /usr/bin/gforth -d 20M
\ Abelian Sandpile Model
0 assert-level !
\ command-line
: parse-number s>number? invert throw drop ;
: parse-size ." size : " next-arg parse-number dup . cr ;
: parse-height ." height: " next-arg parse-number dup . cr ;
: parse-args cr parse-size parse-height ;
parse-args constant HEIGHT constant SIZE
: allot-erase create here >r dup allot r> swap erase ;
: size^2 SIZE dup * cells ;
: 2cells [ 2 cells ] literal ;
: -2cells [ 2cells negate ] literal ;
size^2 allot-erase arr
\ array processing
: ix swap SIZE * + cells arr + ;
: center SIZE 2/ dup ;
: write-cell ix @ u. ;
: write-row SIZE 0 ?do dup i write-cell loop drop cr ;
: arr. SIZE 0 ?do i write-row loop ;
\ stack processing
: stack-empty? dup -1 = ;
: stack-full? stack-empty? invert ;
\ pgm-handling
: concat { a1 l1 a2 l2 } l1 l2 + allocate throw dup dup a1 swap l1 cmove a2 swap l1 + l2 cmove l1 l2 + ;
: write-pgm ." P2" cr SIZE u. SIZE u. cr ." 3" cr arr. ;
: u>s 0 <# #s #> ;
: filename s" sandpile-" SIZE u>s concat s" -" concat HEIGHT u>s concat s" .pgm" concat ;
: to-pgm filename w/o create-file throw ['] write-pgm over outfile-execute close-file throw ;
\ sandpile
: prep-arr HEIGHT center ix ! ;
: prep-stack -1 HEIGHT 4 u>= if center then ;
: prepare prep-arr prep-stack ;
: ensure if else 2drop 0 2rdrop exit then ;
: col>=0 dup 0>= ensure ;
: col<SIZE dup SIZE < ensure ;
: row>=0 over 0>= ensure ;
: row<SIZE over SIZE < ensure ;
: legal? col>=0 col<SIZE row>=0 row<SIZE 2drop true ;
: north 1. d- ;
: east 1+ ;
: south 1. d+ ;
: west 1- ;
: reduce 2dup ix dup -4 swap +! @ 4 < if 2drop then ;
: increase 2dup legal? if 2dup ix dup 1 swap +! @ 4 = if 2swap else 2drop then else 2drop then ;
: inc-north 2dup north increase ;
: inc-east 2dup east increase ;
: inc-south 2dup south increase ;
: inc-west 2dup west increase ;
: inc-all inc-north inc-east inc-south inc-west 2drop ;
: simulate prepare begin stack-full? while 2dup 2>r reduce 2r> inc-all repeat drop to-pgm ." written to " filename type cr ;
simulate bye
{{out}} sandpile with 5000 grains of sand: ./sandpile.fs 61 5000: [http://commons.wikimedia.org/wiki/File:Sandpile-61-5000.png]
sandpile with 50000 grains of sand: ./sandpile.fs 201 50000: [http://commons.wikimedia.org/wiki/File:Sandpile-201-50000.png]
sandpile with 500000 grains of sand: ./sandpile.fs 601 500000: [http://commons.wikimedia.org/wiki/File:Sandpile-601-500000.png]
Go
{{trans|Rust}}
Stack management in Go is automatic, starting very small (2KB) for each goroutine and expanding as necessary until the maximum allowed size is reached.
package main import ( "fmt" "log" "os" "strings" ) const dim = 16 // image size func check(err error) { if err != nil { log.Fatal(err) } } // Outputs the result to the terminal using UTF-8 block characters. func drawPile(pile [][]uint) { chars:= []rune(" ░▓█") for _, row := range pile { line := make([]rune, len(row)) for i, elem := range row { if elem > 3 { // only possible when algorithm not yet completed. elem = 3 } line[i] = chars[elem] } fmt.Println(string(line)) } } // Creates a .ppm file in the current directory, which contains // a colored image of the pile. func writePile(pile [][]uint) { file, err := os.Create("output.ppm") check(err) defer file.Close() // Write the signature, image dimensions and maximum color value to the file. fmt.Fprintf(file, "P3\n%d %d\n255\n", dim, dim) bcolors := []string{"125 0 25 ", "125 80 0 ", "186 118 0 ", "224 142 0 "} var line strings.Builder for _, row := range pile { for _, elem := range row { line.WriteString(bcolors[elem]) } file.WriteString(line.String() + "\n") line.Reset() } } // Main part of the algorithm, a simple, recursive implementation of the model. func handlePile(x, y uint, pile [][]uint) { if pile[y][x] >= 4 { pile[y][x] -= 4 // Check each neighbor, whether they have enough "sand" to collapse and if they do, // recursively call handlePile on them. if y > 0 { pile[y-1][x]++ if pile[y-1][x] >= 4 { handlePile(x, y-1, pile) } } if x > 0 { pile[y][x-1]++ if pile[y][x-1] >= 4 { handlePile(x-1, y, pile) } } if y < dim-1 { pile[y+1][x]++ if pile[y+1][x] >= 4 { handlePile(x, y+1, pile) } } if x < dim-1 { pile[y][x+1]++ if pile[y][x+1] >= 4 { handlePile(x+1, y, pile) } } // Uncomment this line to show every iteration of the program. // Not recommended with large input values. // drawPile(pile) // Finally call the function on the current cell again, // in case it had more than 4 particles. handlePile(x, y, pile) } } func main() { // Create 2D grid and set size using the 'dim' constant. pile := make([][]uint, dim) for i := 0; i < dim; i++ { pile[i] = make([]uint, dim) } // Place some sand particles in the center of the grid and start the algorithm. hdim := uint(dim/2 - 1) pile[hdim][hdim] = 16 handlePile(hdim, hdim, pile) drawPile(pile) // Uncomment this to save the final image to a file // after the recursive algorithm has ended. // writePile(pile) }
{{out}}
░
▓░▓
░░ ░░
▓░▓
░
Haskell
{{works with|GHC|8.8.1}} {{libheader|base|4.13.0.0}} {{libheader|array|0.5.4.0}} {{libheader|mtl|2.2.2}}
Using a custom monad to make the code cleaner.
{-# LANGUAGE FlexibleContexts #-} {-# LANGUAGE GeneralizedNewtypeDeriving #-} {-# LANGUAGE ScopedTypeVariables #-} module Rosetta.AbelianSandpileModel.ST ( simulate , test , toPGM ) where import Control.Monad.Reader (asks, MonadReader (..), ReaderT, runReaderT) import Control.Monad.ST (runST, ST) import Control.Monad.State (evalStateT, forM_, lift, MonadState (..), StateT, modify, when) import Data.Array.ST (freeze, readArray, STUArray, thaw, writeArray) import Data.Array.Unboxed (array, assocs, bounds, UArray, (!)) import Data.Word (Word32) import System.IO (hPutStr, hPutStrLn, IOMode (WriteMode), withFile) import Text.Printf (printf) type Point = (Int, Int) type ArrayST s = STUArray s Point Word32 type ArrayU = UArray Point Word32 newtype M s a = M (ReaderT (S s) (StateT [Point] (ST s)) a) deriving (Functor, Applicative, Monad, MonadReader (S s), MonadState [Point]) data S s = S { bMin :: !Point , bMax :: !Point , arr :: !(ArrayST s) } runM :: M s a -> S s -> [Point]-> ST s a runM (M m) = evalStateT . runReaderT m liftST :: ST s a -> M s a liftST = M . lift . lift simulate :: ArrayU -> ArrayU simulate a = runST $ simulateST a simulateST :: forall s. ArrayU -> ST s ArrayU simulateST a = do let (p1, p2) = bounds a s = [p | (p, c) <- assocs a, c >= 4] b <- thaw a :: ST s (ArrayST s) let st = S { bMin = p1 , bMax = p2 , arr = b } runM simulateM st s simulateM :: forall s. M s ArrayU simulateM = do ps <- get case ps of [] -> asks arr >>= liftST . freeze p : ps' -> do c <- changeArr p $ \x -> x - 4 when (c < 4) $ put ps' forM_ [north, east, south, west] $ inc . ($ p) simulateM changeArr :: Point -> (Word32 -> Word32) -> M s Word32 changeArr p f = do a <- asks arr oldC <- liftST $ readArray a p let newC = f oldC liftST $ writeArray a p newC return newC inc :: Point -> M s () inc p = do b <- inBounds p when b $ do c <- changeArr p succ when (c == 4) $ modify $ (p :) inBounds :: Point -> M s Bool inBounds p = do st <- ask return $ p >= bMin st && p <= bMax st north, east, south, west :: Point -> Point north (x, y) = (x, y + 1) east (x, y) = (x + 1, y) south (x, y) = (x, y - 1) west (x, y) = (x - 1, y) toPGM :: ArrayU -> FilePath -> IO () toPGM a fp = withFile fp WriteMode $ \h -> do let ((x1, y1), (x2, y2)) = bounds a width = x2 - x1 + 1 height = y2 - y1 + 1 hPutStrLn h "P2" hPutStrLn h $ show width ++ " " ++ show height hPutStrLn h "3" forM_ [y1 .. y2] $ \y -> do forM_ [x1 .. x2] $ \x -> do let c = min 3 $ a ! (x, y) hPutStr h $ show c ++ " " hPutStrLn h "" initArray :: Int -> Word32 -> ArrayU initArray size height = array ((-size, -size), (size, size)) [((x, y), if x == 0 && y == 0 then height else 0) | x <- [-size .. size], y <- [-size .. size]] test :: Int -> Word32 -> IO () test size height = do printf "size = %d, height = %d\n" size height let a = initArray size height b = simulate a fp = printf "sandpile_%d_%d.pgm" size height toPGM b fp putStrLn $ "wrote image to " ++ fp
{{out}} sandpile with 1000 grains of sand: test 15 1000: [http://commons.wikimedia.org/wiki/File:Sandpile_15_1000.jpg]
sandpile with 10000 grains of sand: test 40 10000: [http://commons.wikimedia.org/wiki/File:Sandpile_40_10000.jpg]
sandpile with 100000 grains of sand: test 150 100000: [http://commons.wikimedia.org/wiki/File:Sandpile_150_100000.jpg]
sandpile with 1000000 grains of sand: test 400 1000000: [http://commons.wikimedia.org/wiki/File:Sandpile_400_1000000.jpg]
Julia
Modified from code by Hayk Aleksanyan, viewable at github.com/hayk314/Sandpiles, license viewable there.
module AbelSand # supports output functionality for the results of the sandpile simulations # outputs the final grid in CSV format, as well as an image file using CSV, DataFrames, Images function TrimZeros(A) # given an array A trims any zero rows/columns from its borders # returns a 4 tuple of integers, i1, i2, j1, j2, where the trimmed array corresponds to A[i1:i2, j1:j2] # A can be either numeric or a boolean array i1, j1 = 1, 1 i2, j2 = size(A) zz = typeof(A[1, 1])(0) # comparison of a value takes into account the type as well # i1 is the first row which has non zero element for i = 1:size(A, 1) q = false for k = 1:size(A, 2) if A[i, k] != zz q = true i1 = i break end end if q == true break end end # i2 is the first from below row with non zero element for i in size(A, 1):-1:1 q = false for k = 1:size(A, 2) if A[i, k] != zz q = true i2 = i break end end if q == true break end end # j1 is the first column with non zero element for j = 1:size(A, 2) q = false for k = 1:size(A, 1) if A[k, j] != zz j1 = j q = true break end end if q == true break end end # j2 is the last column with non zero element for j in size(A, 2):-1:1 q=false for k=1:size(A,1) if A[k, j] != zz j2 = j q=true break end end if q==true break end end return i1, i2, j1, j2 end function addLayerofZeros(A, extraLayer) # adds layer of zeros from all corners to the given array A if extraLayer <= 0 return A end N, M = size(A) Z = zeros( typeof(A[1,1]), N + 2*extraLayer, M + 2*extraLayer) Z[(extraLayer+1):(N + extraLayer ), (extraLayer+1):(M+extraLayer)] = A return Z end function printIntoFile(A, extraLayer, strFileName, TrimSmallValues = false) # exports a 2d matrix A into a csv file # @extraLayer is an integers adding layer of 0-s sorrounding the output matrix # trimming off very small values; tiny values affect the performance of CSV export if TrimSmallValues == true A = map(x -> if (abs(x - floor(x)) < 0.01) floor(x) else x end, A) end i1, i2, j1, j2 = TrimZeros( A ) A = A[i1:i2, j1:j2] A = addLayerofZeros(A, extraLayer) CSV.write(string(strFileName,".csv"), DataFrame(A), writeheader = false) return A end function Array_magnifier(A, cell_mag, border_mag) # A is the main array; @cell_mag is the magnifying size of the cell, # @border_mag is the magnifying size of the border between lattice cells # creates a new array where each cell of the original array A appears magnified by size = cell_mag total_factor = cell_mag + border_mag A1 = zeros(typeof(A[1, 1]), total_factor*size(A, 1), total_factor*size(A, 2)) for i = 1:size(A,1), j = 1:size(A,2), u = ((i-1)*total_factor+1):(i*total_factor), v = ((j-1)*total_factor+1):(j*total_factor) if(( u - (i - 1) * total_factor <= cell_mag) && (v - (j - 1) * total_factor <= cell_mag)) A1[u, v] = A[i, j] end end return A1 end function saveAsGrayImage(A, fileName, cell_mag, border_mag, TrimSmallValues = false) # given a 2d matrix A, we save it as a gray image after magnifying by the given factors A1 = Array_magnifier(A, cell_mag, border_mag) A1 = A1/maximum(maximum(A1)) # trimming very small values from A1 to improve performance if TrimSmallValues == true A1 = map(x -> if ( x < 0.01) 0.0 else round(x, digits = 2) end, A1) end save(string(fileName, ".png") , colorview(Gray, A1)) end function saveAsRGBImage(A, fileName, color_codes, cell_mag, border_mag) # color_codes is a dictionary, where key is a value in A and value is an RGB triplet # given a 2d array A, and color codes (mapping from values in A to RGB triples), save A # into fileName as png image after applying the magnifying factors A1 = Array_magnifier(A, cell_mag, border_mag) color_mat = zeros(UInt8, (3, size(A1, 1), size(A1, 2))) for i = 1:size(A1,1) for j = 1:size(A1,2) color_mat[:, i, j] = get(color_codes, A1[i, j] , [0, 0, 0]) end end save(string(fileName, ".png") , colorview(RGB, color_mat/255)) end const N_size = 700 # the radius of the lattice Z^2, the actual size becomes (2*N+1)x(2*N+1) const dx = [1, 0, -1, 0] # for a given (x,y) in Z^2, (x + dx, y + dy) for all (dx,dy) covers the neighborhood of (x,y) const dy = [0, 1, 0, -1] struct L_coord # represents a lattice coordinate x::Int y::Int end function FindCoordinate(Z::Array{L_coord,1}, a::Int, b::Int) # in the given array Z of coordinates finds the (first) index of the tuple (a,b) # if no match, returns -1 for i=1:length(Z) if (Z[i].x == a) && (Z[i].y == b) return i end end return -1 end function move(N) # the main function moving the pile sand grains of size N at the origin of Z^2 until the sandpile becomes stable Z_lat = zeros(UInt8, 2 * N_size + 1, 2 * N_size + 1) # models the integer lattice Z^2, we will have at most 4 sands on each vertex V_sites = falses(2 * N_size + 1, 2 * N_size + 1) # all sites which are visited by the sandpile process, are being marked here Odometer = zeros(UInt64, 2 * N_size + 1, 2 * N_size + 1) # stores the values of the odometer function walking = L_coord[] # the coordinates of sites which need to move V_sites[N_size + 1, N_size + 1] = true # i1, ... j2 -> show the boundaries of the box which is visited by the sandpile process i1, i2, j1, j2 = N_size + 1, N_size + 1, N_size + 1, N_size + 1 n = N t1 = time_ns() while n > 0 n -= 1 Z_lat[N_size + 1, N_size + 1] += 1 if (Z_lat[N_size + 1, N_size + 1] >= 4) push!(walking, L_coord(N_size + 1, N_size + 1)) end while(length(walking) > 0) w = pop!(walking) x = w.x y = w.y Z_lat[x, y] -= 4 Odometer[x, y] += 4 for k = 1:4 Z_lat[x + dx[k], y + dy[k]] += 1 V_sites[x + dx[k], y + dy[k]] = true if Z_lat[x + dx[k], y + dy[k]] >= 4 if FindCoordinate(walking, x + dx[k] , y + dy[k]) == -1 push!(walking, L_coord( x + dx[k], y + dy[k])) end end end i1 = min(i1, x - 1) i2 = max(i2, x + 1) j1 = min(j1, y - 1) j2 = max(j2, y + 1) end end #end of the main while t2 = time_ns() println("The final boundaries are:: ", (i2 - i1 + 1),"x",(j2 - j1 + 1), "\n") print("time elapsed: " , (t2 - t1) / 1.0e9, "\n") Z_lat = printIntoFile(Z_lat, 0, string("Abel_Z_", N)) Odometer = printIntoFile(Odometer, 1, string("Abel_OD_", N)) saveAsGrayImage(Z_lat, string("Abel_Z_", N), 20, 0) color_code = Dict(1=>[255, 128, 255], 2=>[255, 0, 0],3 => [0, 128, 255]) saveAsRGBImage(Z_lat, string("Abel_Z_color_", N), color_code, 20, 0) # for the total elapsed time, it's better to use the @time macros on the main call return Z_lat, Odometer # these are trimmed in output module end # end of function move end # module using .AbelSand Z_lat, Odometer = AbelSand.move(100000)
{{out}}
[http://alahonua.com/temp/Abel_Z_color_100000.png Link to PNG output file for N=100000 ie. AbelSand.move(100000)]
[http://alahonua.com/temp/Abel_Z_color_1000000.png Link to PNG output file (run time >90 min) for N=1000000 (move(1000000))]
Perl
#!/usr/bin/perl use strict; # http://www.rosettacode.org/wiki/Abelian_sandpile_model use warnings; my ($high, $wide) = split ' ', qx(stty size); my $mask = "\0" x $wide . ("\0" . "\177" x ($wide - 2) . "\0") x ($high - 5) . "\0" x $wide; my $pile = $mask =~ s/\177/ rand() < 0.02 ? chr 64 + rand 20 : "\0" /ger; for ( 1 .. 1e6 ) { print "\e[H", $pile =~ tr/\0-\177/ 1-~/r, "\n$_"; my $add = $pile =~ tr/\1-\177/\0\0\0\200/r; # set high bit for >=4 $add =~ /\200/ or last; $pile =~ tr/\4-\177/\0-\173/; # subtract 4 if >=4 for ("\0$add", "\0" x $wide . $add, substr($add, 1), substr $add, $wide) { $pile |= $_; $pile =~ tr/\200-\377/\1-\176/; # add one to each neighbor of >=4 $pile &= $mask; } select undef, undef, undef, 0.1; # comment out for full speed }
Phix
{{libheader|pGUI}} Generates moving images similar to the julia output. The distributed version also has variable speed, additional display modes, and a random dropping toggle.
-- demo\rosetta\Abelian_sandpile_model.exw
include pGUI.e
Ihandle dlg, canvas
cdCanvas cddbuffer
sequence board = {{0,0,0},
{0,0,0},
{0,0,0}}
procedure drop(integer y, x)
sequence moves = {}
while true do
board[y,x] += 1
if board[y,x]>=4 then
board[y,x] -= 4
moves &= {{y,x-1},{y,x+1},{y-1,x},{y+1,x}}
end if
-- extend board if rqd (maintain a border of zeroes)
if x=1 then -- extend left
for i=1 to length(board) do
board[i] = prepend(board[i],0)
end for
for i=1 to length(moves) do
moves[i][2] += 1
end for
elsif x=length(board[1]) then -- extend right
for i=1 to length(board) do
board[i] = append(board[i],0)
end for
end if
-- (copy the all-0 lines from the other end...)
if y=1 then -- extend up
board = prepend(board,board[$])
for i=1 to length(moves) do
moves[i][1] += 1
end for
elsif y=length(board) then -- extend down
board = append(board,board[1])
end if
if length(moves)=0 then exit end if
{y,x} = moves[$]
moves = moves[1..$-1]
end while
IupUpdate(canvas)
end procedure
function timer_cb(Ihandle /*ih*/)
integer y = floor(length(board)/2)+1,
x = floor(length(board[1])/2)+1
drop(y,x)
return IUP_DEFAULT
end function
function redraw_cb(Ihandle ih, integer /*posx*/, integer /*posy*/)
IupGLMakeCurrent(ih)
cdCanvasActivate(cddbuffer)
cdCanvasClear(cddbuffer)
for y=1 to length(board) do
for x=1 to length(board[1]) do
integer c = board[y][x]
if c!=0 then
integer colour = {CD_VIOLET,CD_RED,CD_BLUE}[c]
cdCanvasPixel(cddbuffer, x, y, colour)
end if
end for
end for
cdCanvasFlush(cddbuffer)
return IUP_DEFAULT
end function
function map_cb(Ihandle ih)
IupGLMakeCurrent(ih)
atom res = IupGetDouble(NULL, "SCREENDPI")/25.4
cddbuffer = cdCreateCanvas(CD_GL, "300x100 %g", {res})
cdCanvasSetBackground(cddbuffer, CD_PARCHMENT)
return IUP_DEFAULT
end function
procedure main()
IupOpen()
canvas = IupGLCanvas("RASTERSIZE=300x100")
IupSetCallbacks({canvas}, {"ACTION", Icallback("redraw_cb"),
"MAP_CB", Icallback("map_cb")})
dlg = IupDialog(canvas,"TITLE=\"Abelian sandpile model\"")
IupCloseOnEscape(dlg)
IupShow(dlg)
Ihandle timer = IupTimer(Icallback("timer_cb"), 10)
IupMainLoop()
IupClose()
end procedure
main()
Python
import numpy as np import matplotlib.pyplot as plt def iterate(grid): changed = False for ii, arr in enumerate(grid): for jj, val in enumerate(arr): if val > 3: grid[ii, jj] -= 4 if ii > 0: grid[ii - 1, jj] += 1 if ii < len(grid)-1: grid[ii + 1, jj] += 1 if jj > 0: grid[ii, jj - 1] += 1 if jj < len(grid)-1: grid[ii, jj + 1] += 1 changed = True return grid, changed def simulate(grid): while True: grid, changed = iterate(grid) if not changed: return grid if __name__ == '__main__': start_grid = np.zeros((10, 10)) start_grid[4:5, 4:5] = 64 final_grid = simulate(start_grid.copy()) plt.figure() plt.gray() plt.imshow(start_grid) plt.figure() plt.gray() plt.imshow(final_grid)
Output: Before:
[[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0.64. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]]
After:
[[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 1. 2. 1. 0. 0. 0. 0.] [0. 0. 2. 2. 2. 2. 2. 0. 0. 0.] [0. 1. 2. 2. 2. 2. 2. 1. 0. 0.] [0. 2. 2. 2. 0. 2. 2. 2. 0. 0.] [0. 1. 2. 2. 2. 2. 2. 1. 0. 0.] [0. 0. 2. 2. 2. 2. 2. 0. 0. 0.] [0. 0. 0. 1. 2. 1. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] [0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]]
Rust
// Set image size. const DIM: usize = 16; // This function outputs the result to the console using UTF-8 block characters. fn draw_pile(pile: &Vec<Vec<usize>>) { for row in pile { let mut line = String::with_capacity(row.len()); for elem in row { line.push(match elem { 0 => ' ', 1 => '░', 2 => '▒', 3 => '▓', _ => '█' }); } println!("{}", line); } } // This function creates a file called "output.ppm" in the directory the program was run, which contains // a colored image of the pile. fn write_pile(pile: &Vec<Vec<usize>>) { use std::fs::File; // Used for opening the file. use std::io::Write; // Used for writing to the file. // Learn more about PPM here: http://netpbm.sourceforge.net/doc/ppm.html let mut file = File::create("./output.ppm").unwrap(); // We write the signature, image dimensions and maximum color value to the file. let _ = write!(file, "P3\n {} {}\n255\n", DIM, DIM).unwrap(); for row in pile { let mut line = String::with_capacity(row.len()*6); for elem in row { line.push_str(match elem { 0 => "125 0 25 ", // Background color for cells that have no "sand" in them. // Depending on how many particles of sand is there in the cell we use a different shade of yellow. 1 => "125 80 0 ", 2 => "186 118 0 ", 3 => "224 142 0 ", // It is impossible to have more than 3 particles of sand in one cell after the program has run, // however, Rust demands that all branches have to be considered in a match statement, so we // explicitly tell the compiler, that this is an unreachable branch. _ => unreachable!() }); } let _ = write!(file, "{}", line).unwrap(); } } // This is the main part of the algorithm, a simple, recursive implementation of the model. fn handle_pile(x: usize, y: usize, pile: &mut Vec<Vec<usize>>) { if pile[y][x] >= 4 { pile[y][x] -= 4; // We check each neighbor, whether they have enough "sand" to collapse and if they do, // we recursively call handle_pile on them. if y > 0 { pile[y-1][x] += 1; if pile[y-1][x] >= 4 {handle_pile(x, y-1, pile)}} if x > 0 { pile[y][x-1] += 1; if pile[y][x-1] >= 4 {handle_pile(x-1, y, pile)}} if y < DIM-1 { pile[y+1][x] += 1; if pile[y+1][x] >= 4 {handle_pile(x, y+1, pile)}} if x < DIM-1 { pile[y][x+1] += 1; if pile[y][x+1] >= 4 {handle_pile(x+1, y, pile)}} // Uncomment this line to show every iteration of the program. Not recommended with large input values. //draw_pile(&pile); // Finally we call the function on the current cell again, in case it had more than 4 particles. handle_pile(x,y,pile); } } fn main() { use std::thread::Builder; // Used to spawn a new thread. /* Rust by default uses a 2Mb stack, which gets quickly filled (resulting in a stack overflow) if we use any value larger than * about 30,000 as our input value. To circumvent this, we spawn a thread with 32Mbs of stack memory, which can easily handle * hundreds of thousands of sand particles. I tested the program using 256,000, but it should theoretically work with larger * values too. */ let _ = Builder::new().stack_size(33554432).spawn(|| { // This is our 2D grid. It's size can be set using the DIM constant found at the top of the code. let mut pile: Vec<Vec<usize>> = vec![vec![0;DIM]; DIM]; // We place this much sand in the center of the grid. pile[DIM/2 - 1][DIM/2 - 1] = 16; // We start the algorithm on the pile we just created. handle_pile(DIM/2 - 1, DIM/2 - 1, &mut pile); draw_pile(&pile) // Uncomment this to save the image to a file after the recursive algorithm has ended. //write_pile(&pile) }).unwrap().join(); }
'''Output:'''
░
▒░▒
░░ ░░
▒░▒
░