"Percolation" dans une grille en 2D

case en cours = case en haut à gauche
tant que la case en cours est dans le dessin:
	si couleur de la case en cours est une couleur d'origine:
		remplirforme(case en cours)
	case en cours = case suivant la case en cours

remplirforme(lacasededepart):
	couleurforme = couleur de la case de départ
	codeforme=i+j+3
	couleur de case de départ = codeforme
	ajouter les 4 voisins de la case de départ dans liste à faire
	tant que liste à faire n'est pas vide:
		retirer une case de la liste à faire, appellons-là casedetravail
		si couleur de casedetravail = couleurforme :
			couleur de case de travail = codeforme
			ajouter les 4 voisins de la case de travail dans liste à faire

##importation des modules
import numpy.random as rd
import numpy as np
import copy


##définition des fonctions

def init_tab(nb_b,N) :
    tab=np.zeros((N,N),dtype=int)
    nb=(N**2-nb_b)/2
    for i in np.arange(nb*2) :
        x=rd.randint(0,N)
        y=rd.randint(0,N)
        while tab[x][y]==1 or tab[x][y]==2 :
            x=rd.randint(0,N)
            y=rd.randint(0,N)
        if i%2==0 :
            tab[x][y]=1
        else :
            tab[x][y]=2
    return tab


def test_voisin(tab,i,j):
    a=i+j+3
    valeur=tab[i][j]
    tab[i][j]=a
    g=0
    d=0
    h=0
    b=0
    if i>0 and tab[i-1][j]==valeur :
        g+=1
        tab[i-1][j]=a
    if i<N-1 and tab[i+1][j]==valeur :
        d+=1
        tab[i+1][j]=a
    if j>0 and tab[i][j-1]==valeur:
        h+=1
        tab[i][j-1]=a
    if j<N-1 and tab[i][j+1]==valeur :
        b+=1
        tab[i][j+1]=a
    return tab,a,g,d,h,b

def coloration(tab) :
    N=len(tab)
    for i in range(N):
        for j in range(N):
            valeur=tab[i][j]
            if valeur==0 or valeur==1 or valeur==2 :
                tab,a,g,d,h,b=test_voisin(tab,i,j)
                while g!=0 and d!=0 and h!=0 and b!=0 :
                    if g!=0 :
                        tab,a,g,d,h,b=test_voisin(tab,i-1,j)
                    if d!=0 :
                        tab,a,g,d,h,b=test_voisin(tab,i-1,j)
                    if h!=0 :
                        tab,a,g,d,h,b=test_voisin(tab,i,j+1)
                    if b!=0 :
                        tab,a,g,d,h,b=test_voisin(tab,i,j-1)

##programme principale
N=10
b=10

tab_test=init_tab(b,N)
tab=percolation(tab_test)
print(tab)