import numpy as np
M = np.array([
[1]*4+[2]+[1]*4,
[1]+7*[0]+[1],
[1]*4+[2]+[1]*4,
[0]*4+[2]+[0]*4,
[0]*2+[1]*2+[2]+[1]*2+2*[0],
[0]*2+[1]+3*[0]+[1]+2*[0],
[2]+3*[1]+[2]+3*[1]+[2],
[1]+3*[0]+[1]+3*[0]+[1],
[1]*9])

pions=np.array([[0]*1+[3]*3+[0]*1+[4]*3+[0]*1])

arrivee=np.array([[0]*4+[5]*1+[0]*4])


plateau=np.concatenate((arrivee,M,pions),axis=0)





import matplotlib.pyplot as plt



def traceCercle(x,y,color='black'):
    r = 0.4
    rx=r*0.8
    ry=r
    T = np.linspace(0,2*np.pi,100)
    X = x+rx*np.cos(T)
    Y = y+ry*np.sin(T)
    plt.fill(X,Y,color)
    plt.draw()


def affiche(plateau):
    plt.figure(1)
    plt.clf()
    xmin, xmax, ymin, ymax = -1,9, -1, 11 # Delimitation de la fenetre
    plt.axis([xmin,xmax,ymin,ymax])
    for i in range(11):
        y = 10-i
        for j in range(9):
            x = j
            if plateau[i,j]==1:
                traceCercle(x,y,color='black')
            elif plateau[i,j]==2:
                traceCercle(x,y,'yellow')
            elif plateau[i,j]==3:
                traceCercle(x,y,'green')
            elif plateau[i,j]==4:
                traceCercle(x,y,'blue')
            elif plateau[i,j]==5:
                traceCercle(x,y,'red')
    plt.draw()
    
barricade = 2
bleu = 3
vert = 4

def adversaire(couleur):
    if couleur == bleu:
        return vert
    else:
        return bleu 

def voisinage(i,j,couleur):
    L = []
    for dx in (-1,1):
        if -1<i+dx<10:
            #print(i+dx,j)
            if plateau[i+dx,j] != 0 and plateau[i+dx,j] != couleur:
                    L.append((i+dx,j))
        if -1<j+dx<9:
            #print(i,j+dx)
            if plateau[i,j+dx] != 0 and plateau[i,j+dx] != couleur:
                    L.append((i,j+dx))
    return L
 

class pile:
    def __init__(self):
        self.main = []    
        
    def empiler(self,x):
        self.main.append(x)
        
    def depiler(self):
        return self.main.pop()
        
    def est_vide(self):
        return len(self.main)==0
        
    def peek(self):
        return self.main[-1]
        
    def affiche(self):
        print(self.main)
        
def Recherche_pileVer2(i,j,couleur,d):
    parcours = pile()
    visites = []
    voisins = voisinage(i,j,couleur)
    parcours.empiler([(i,j), voisins])
    Trouves = []
    coup = 0
    recherche = True
    while recherche:
        parcours.affiche()
        if parcours.est_vide():
            return Trouves
        (i,j), voisins = parcours.peek()
        visites.append((i,j))
        if coup == d:
            Trouves.append((i,j))
            visites.append((i,j))
            coup -= 1
            parcours.depiler()
            continue
        if voisins == []:
            parcours.depiler()
            coup -= 1
        else:
            (iv,jv) = voisins.pop()
            if (iv,jv) in visites:
                continue
            if coup== d-1:
                Trouves.append((iv,jv))
            elif coup <= d-2:
                if plateau[iv,jv] == 2:
                    Trouves.append((iv,jv))
                    
                else:
                    parcours.empiler([(iv,jv),voisinage(iv,jv,couleur)])
                    coup += 1
        
        

def changementplacebarricade(i,j,b1,b2):
    if plateau[b1,b2]==1:
        plateau[b1,b2]=2
        plateau[i,j]=1
        return plateau
    else:
        return "impossible"

from math import * 
from random import randint

def partie(plateau):
    couleur=input("Choississez votre couleur:")
    couleuradversaire=adversaire(couleur)
    i,j=9,6 #cases départ
    a,b=9,2
    plateau[10,6]=1 #met la case en noir
    plateau[10,2]=1 #idem
    print('BLEU commence')
    while plateau[i,j]!=5 or plateau[a,b]!=5 : #tant que case finale (rouge) pas atteinte
        print('C est à bleu')
        d=randint(1,6)
        print(d)
        choixpossiblesbleu=Recherche_pileVer2(i,j,couleur,d)
        print(choixpossiblesbleu)
        b1=int(input("N°ligne nouvelle case:"))
        b2=int(input("N°colonne nouvelle case:"))
        if plateau[b1,b2]==2:
            bb1=int(input("Choississez une nouvelle ligne pour la barricade:"))#choix de l'utilisateur de la mettre ou il veut
            bb2=int(input("Choississez une nouvelle colonne pour la barricade:"))
            plateau=changementplacebarricade(b1,b2,bb1,bb2)
        plateau[b1,b2]=4
        plateau[i,j]=1
        i,j=b1,b2
        print('Maintenant, c est à vert')
        d=randint(1,6)
        print(d)
        choixpossiblesvert=Recherche_pileVer2(a,b,couleuradversaire,d)
        print(choixpossiblesvert)
        v1=int(input("N°ligne nouvelle case:"))
        v2=int(input("N°colonne nouvelle case:"))
        if plateau[v1,v2]==2:
            bv1=int(input("Choississez une nouvelle ligne pour la barricade"))
            bv2=int(input("Choississez une nouvelle colonne pour la barricade"))
            plateau=changementplacebarricade(v1,v2,bv1,bv2)
        plateau[v1,v2]=3
        plateau[a,b]=1
        a,b=v1,v2
        B=affiche(plateau) # problème ici , si on met B dans la boucle bug figure"répond plus"
    return B