ui.py

import sys
import copy
import time

from utils import infinity
from ai import Problem
from ai import astar_search
from ai import iterative_deepening_search
from ai import depth_first_tree_search
from ai import depth_first_graph_search
from ai import breadth_first_tree_search
from ai import breadth_first_graph_search
from ai import depth_limited_search

import Tkinter as tkinter
import tkMessageBox as messagebox
from easygui import ccbox
from easygui import codebox

from tpIAsokobanparser import obtenerMapa
from sokoban import main
from sokoban import generateGoalState
from sokoban import SokobanProblem
from constantes import *

printTableFather=False
MONOSPACE_FONT_SIZE =  10



class tables():
    tabs= []
    cont=0
    excTime=""
    expand=0
    buttons=[]
    filas=0
    columnas=0
    

def tableToStr(tableS):
    tableR=[]
    for i in tableS:
        tableR.append("".join(i))
        tableR.append("\n")
    return tableR



    

def callmove(i):
    def wrap():
        move(i)
    return wrap


def printText(text, eTime ,cont, textArea):
    try:
        text = "".join(text)  # convert a list or a tuple to a string
    except:
        sys.exit(16)
    textArea.configure(state=tkinter.NORMAL) 
    textArea.delete(1.0, tkinter.END)
    textArea.insert(tkinter.END,"SOKOBAN\n\n", "normal")    
    textArea.insert(tkinter.END,text, "normal")
    textArea.insert(tkinter.END,eTime, "normal")
    textArea.insert(tkinter.END,"\nPaso nro:" + str(cont), "normal")
    textArea.insert(tkinter.END,"\nCantidad de Pasos:" + str(len(tables.tabs)), "normal")
    textArea.insert(tkinter.END,"\nCantidad de Nodos Expandidos:" + str(tables.expand), "normal")
    textArea.configure(state=tkinter.DISABLED)


def changeTableNext(textA, tables):
    if tables.cont >= len(tables.tabs)-1:
        messagebox.showinfo("END", "BINGO!")
        tables.cont = -1
        text = tableToStr(tables.tabs[tables.cont])
    else:
        tables.cont=tables.cont + 1
        text = tableToStr(tables.tabs[tables.cont])
        printText(text, tables.excTime,tables.cont ,  textA)
        row=0
        for i in tables.tabs[tables.cont]:
            for j in i:
                show = getImage(j)
                tables.buttons[row].configure(image=show)
                row+=1
        
    return

def changeTablePrev(textA, tables):
    if tables.cont <= 0:
        messagebox.showinfo("START", "PRESS NEXT STEP!")
       
    else:
        tables.cont=tables.cont-1
        text = tableToStr(tables.tabs[tables.cont])
        printText(text, tables.excTime ,tables.cont,  textA)
        row=0
        for i in tables.tabs[tables.cont]:
            for j in i:
                show = getImage(j)
                tables.buttons[row].configure(image=show)
                row+=1
    return

def tableNext(textA,  table):
    def wrapNext():
        changeTableNext( textA, table)
    return wrapNext

def tablePrev(textA,  table):
    def wrapPrev():
        changeTablePrev( textA, table)
    return wrapPrev
    
TOP = "top"
YES = True
BOTH = "both"
BOTTOM = "bottom"
NO = False
X = "x"
root = tkinter.Tk()
screen_width = root.winfo_screenwidth()
screen_height = root.winfo_screenheight()
root_width    = int((screen_width * 0.8))
btnframe = tkinter.Frame(root)
btnframe.pack(side=TOP, expand=YES, fill=BOTH)

character_width = int((root_width * 0.6) / MONOSPACE_FONT_SIZE)
textArea = tkinter.Text(root,height=10,width=character_width, padx="1m", pady="1m")
textArea.pack(side=tkinter.LEFT, fill=BOTH, expand=YES)

nextbtn = tkinter.Button(root, command=tableNext(textArea, tables), text="NEXT STEP")
nextbtn.pack(side=BOTTOM, expand=NO, fill=X)

prevbtn = tkinter.Button(root, command=tablePrev(textArea, tables), text="PREV STEP")
prevbtn.pack(side=BOTTOM, expand=NO, fill=X)

soko = tkinter.PhotoImage(file="images/sokoban.gif")
brick = tkinter.PhotoImage(file="images/brick.gif")
player = tkinter.PhotoImage(file="images/player.gif")
blank = tkinter.PhotoImage(file="images/blank.gif")
box = tkinter.PhotoImage(file="images/box.gif")
box2 = tkinter.PhotoImage(file="images/box2.gif")
goalImage = tkinter.PhotoImage(file="images/goal.gif")
label = tkinter.Label(image=soko)


label.pack(side=BOTTOM, expand=NO, fill=X)



def main():
    columna = 0
    fila = 0
    if len(sys.argv)==2 or len(sys.argv)==3:
        if Debug():
            printTableFather = True
            initial = obtenerMapa(sys.argv[2], fila, columna)
        else:
            initial = obtenerMapa(sys.argv[1], fila, columna)
            
        goal = generateGoalState(initial) #encontrar el estado final
        sokoban = SokobanProblem(initial, goal)
        x1= time.time()
        search = astar_search(sokoban)
        x2= time.time()
        tables.filas = initial.matrixX
        tables.columnas = initial.matrixY

        for i in initial.matrix:
            for j in i:
                show = getImage(j)
                tables.buttons.append(tkinter.Label(btnframe, image=show)) 
    
        for i, button in enumerate(tables.buttons):
            button.grid(row=i//tables.columnas, column=i%tables.columnas, sticky="N"+"E"+"W"+"S")
            
        if search:
            
            pathS = search.path()

            timediff = x2 - x1
            
            #Genera la secuencia de estados
            path = [node.state for node in pathS]
            states = []
            
            for state in path:
                states.append(state)
            
            i = len(states) - 1
            while i >= 0:
                tables.tabs.append(states[i].matrix)
                i -= 1
            tables.excTime= 'Tiempo:', round(timediff,4),'segundos'
            tables.expand = sokoban.expanded

        else:
            print "No se pudo encontrar la solucion al problema"
    else:
        if len(sys.argv)<2:
            print "Debe recibir el nombre del archivo..."
        if len(sys.argv)>3:
            print "Demasiados arguementos..."
    
    text = tableToStr(tables.tabs[0])
    printText(text,tables.excTime, tables.cont,textArea)
    root.mainloop()

def getImage(j):
    if j == CHAR_WALL:
        return brick
    elif j == CHAR_PLAYER:
        return player
    elif j == CHAR_PLAYER_S:
        return player2
    elif j == CHAR_SPACE_S:
        return goalImage
    elif j == CHAR_BOX:
        return box              
    elif j == CHAR_BOX_S:
        return box2
    else:
        return blank

if __name__ == "__main__":
    main()