aima-python/gui/eight_puzzle.py at master · binlecode/aima-python · GitHub

History

138 lines (111 loc) · 4.51 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

# author ad71

from tkinter import *

from functools import partial

import time

import random

import numpy as np

import sys

import os.path

sys.path.append(os.path.join(os.path.dirname(__file__), '..'))

from search import astar_search, EightPuzzle

import utils

root = Tk()

state = [1, 2, 3, 4, 5, 6, 7, 8, 0]

puzzle = EightPuzzle(tuple(state))

solution = None

b = [None]*9

# TODO: refactor into OOP, remove global variables

def scramble():

""" Scrambles the puzzle starting from the goal state """

global state

global puzzle

possible_actions = ['UP', 'DOWN', 'LEFT', 'RIGHT']

scramble = []

for _ in range(60):

scramble.append(random.choice(possible_actions))

for move in scramble:

if move in puzzle.actions(state):

state = list(puzzle.result(state, move))

puzzle = EightPuzzle(tuple(state))

create_buttons()

def solve():

""" Solves the puzzle using astar_search """

return astar_search(puzzle).solution()

def solve_steps():

""" Solves the puzzle step by step """

global puzzle

global solution

global state

solution = solve()

print(solution)

for move in solution:

state = puzzle.result(state, move)

create_buttons()

root.update()

root.after(1, time.sleep(0.75))

def exchange(index):

""" Interchanges the position of the selected tile with the zero tile under certain conditions """

global state

global solution

global puzzle

zero_ix = list(state).index(0)

actions = puzzle.actions(state)

current_action = ''

i_diff = index//3 - zero_ix//3

j_diff = index%3 - zero_ix%3

if i_diff == 1:

current_action += 'DOWN'

elif i_diff == -1:

current_action += 'UP'

if j_diff == 1:

current_action += 'RIGHT'

elif j_diff == -1:

current_action += 'LEFT'

if abs(i_diff) + abs(j_diff) != 1:

current_action = ''

if current_action in actions:

b[zero_ix].grid_forget()

b[zero_ix] = Button(root, text=f'{state[index]}', width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, zero_ix))

b[zero_ix].grid(row=zero_ix//3, column=zero_ix%3, ipady=40)

b[index].grid_forget()

b[index] = Button(root, text=None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, index))

b[index].grid(row=index//3, column=index%3, ipady=40)

state[zero_ix], state[index] = state[index], state[zero_ix]

puzzle = EightPuzzle(tuple(state))

def create_buttons():

""" Creates dynamic buttons """

# TODO: Find a way to use grid_forget() with a for loop for initialization

b[0] = Button(root, text=f'{state[0]}' if state[0] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 0))

b[0].grid(row=0, column=0, ipady=40)

b[1] = Button(root, text=f'{state[1]}' if state[1] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 1))

b[1].grid(row=0, column=1, ipady=40)

b[2] = Button(root, text=f'{state[2]}' if state[2] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 2))

b[2].grid(row=0, column=2, ipady=40)

b[3] = Button(root, text=f'{state[3]}' if state[3] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 3))

b[3].grid(row=1, column=0, ipady=40)

b[4] = Button(root, text=f'{state[4]}' if state[4] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 4))

b[4].grid(row=1, column=1, ipady=40)

b[5] = Button(root, text=f'{state[5]}' if state[5] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 5))

b[5].grid(row=1, column=2, ipady=40)

b[6] = Button(root, text=f'{state[6]}' if state[6] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 6))

b[6].grid(row=2, column=0, ipady=40)

b[7] = Button(root, text=f'{state[7]}' if state[7] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 7))

b[7].grid(row=2, column=1, ipady=40)

b[8] = Button(root, text=f'{state[8]}' if state[8] != 0 else None, width=6, font=('Helvetica', 40, 'bold'), command=partial(exchange, 8))

b[8].grid(row=2, column=2, ipady=40)

def create_static_buttons():

""" Creates scramble and solve buttons """

scramble_btn = Button(root, text='Scramble', font=('Helvetica', 30, 'bold'), width=8, command=partial(init))

scramble_btn.grid(row=3, column=0, ipady=10)

solve_btn = Button(root, text='Solve', font=('Helvetica', 30, 'bold'), width=8, command=partial(solve_steps))

solve_btn.grid(row=3, column=2, ipady=10)

def init():

""" Calls necessary functions """

global state

global solution

state = [1, 2, 3, 4, 5, 6, 7, 8, 0]

scramble()

create_buttons()

create_static_buttons()

init()

root.mainloop()

aima-python/gui/eight_puzzle.py at master · binlecode/aima-python · GitHub

Expand file tree

Breadcrumbs

eight_puzzle.py

Latest commit

History

Breadcrumbs

eight_puzzle.py

File metadata and controls