from pygame import *
font.init()
f1 = font.Font(font.get_default_font(),40)
f2 = font.Font(font.get_default_font(),80)
from random import choice
scr = display.set_mode((500,500))
import pygame.event
direction = {K_LEFT  : (  0,  4, 1, 4), K_RIGHT : (-13, -4,-1, 4),
             K_UP    : (  0, 15, 4, 1), K_DOWN  : ( -4,-13,-4, 1)}
matrice = [0,0,0,0,
           0,0,0,0,
           0,0,0,0,
           0,0,0,0]
def game():
    def redraw():
        for e,v in enumerate(matrice):
            x = (e%4)*120+20
            y = (e//4)*120+20
            r = scr.fill(-1,(x,y,100,100))
            if not v: v = ''
            v = f1.render(str(v),1,(100,100,100))
            scr.blit(v,v.get_rect(center=r.center))
        display.flip()

    def MoveAndAdd(key):
        n1,n2,s,p = direction[key]
        for foo in 1,2,3,4:
            t = [i for i in matrice[n1:n1+n2:s] if i]
            i = 0
            while i<len(t)-1:
                if t[i] == t[i+1]:
                    t[i] += t[i] #t[i] = t[i]+t[i]
                    del(t[i+1])# supprimer t[i+1]
                i += 1
            matrice[n1:n1+n2:s] = (t+[0,0,0,0])[:4]# 0~4
            n1 += p
            #print(matrice[i],matrice[i+1],matrice[i+4])
            print(matrice[i],i,t)


    c  = None
    matrice[choice([e for e,i in enumerate(matrice) if not i])] = choice((2,4))   #[e for e,i in enumerate(matrice) if not i]=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
    while True:
        if c != matrice:   #(!=) differente
            tv = [e for e,i in enumerate(matrice) if not i]
            if tv:
                matrice[choice(tv)] = choice((2,4))
                print (len(tv),tv)
            if len(tv)<2:
                a = not any(matrice[i] == matrice[i+1]for i in (0,1,2,4,5,6,8,9,10,12,13,14))
                b = not any(matrice[i] == matrice[i+4]for i in range(12))
                if a and b:# si a et b sont vrai,,
                    redraw()
                    return 0 #perdu
            redraw()
        ev = event.wait()
        c = matrice[:] #[0~15]
        if ev.type == KEYDOWN and ev.key in direction:
            MoveAndAdd(ev.key)
        if 2048 in matrice:
            redraw()
            return 1 #gagne

while True:
    event.clear()
    g = game()
    if g == 0:
        message = f2.render('PERDU',1,(200,100,100))
        scr.blit(message,message.get_rect(center=(250,250)))
    else:
        message = f2.render('GAGNE',1,(255,100,100))
        scr.blit(message,message.get_rect(center=(250,250)))
    display.flip()
    while event.wait().type != QUIT: pass
    break