import os.path
from tkinter import *
from agents import *
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
class Gui(VacuumEnvironment):
"""This is a two-dimensional GUI environment. Each location may be
dirty, clean or can have a wall. The user can change these at each step.
"""
xi, yi = (0, 0)
perceptible_distance = 1
def __init__(self, root, width=7, height=7, elements=None):
super().__init__(width, height)
if elements is None:
elements = ['D', 'W']
self.root = root
self.create_frames()
self.create_buttons()
self.create_walls()
self.elements = elements
def create_frames(self):
"""Adds frames to the GUI environment."""
self.frames = []
for _ in range(7):
frame = Frame(self.root, bg='grey')
frame.pack(side='bottom')
self.frames.append(frame)
def create_buttons(self):
"""Adds buttons to the respective frames in the GUI."""
self.buttons = []
for frame in self.frames:
button_row = []
for _ in range(7):
button = Button(frame, height=3, width=5, padx=2, pady=2)
button.config(
command=lambda btn=button: self.display_element(btn))
button.pack(side='left')
button_row.append(button)
self.buttons.append(button_row)
def create_walls(self):
"""Creates the outer boundary walls which do not move."""
for row, button_row in enumerate(self.buttons):
if row == 0 or row == len(self.buttons) - 1:
for button in button_row:
button.config(text='W', state='disabled',
disabledforeground='black')
else:
button_row[0].config(
text='W', state='disabled', disabledforeground='black')
button_row[len(button_row) - 1].config(text='W',
state='disabled', disabledforeground='black')
self.buttons[3][3].config(
text='A', state='disabled', disabledforeground='black')
def display_element(self, button):
"""Show the things on the GUI."""
txt = button['text']
if txt != 'A':
if txt == 'W':
button.config(text='D')
elif txt == 'D':
button.config(text='')
elif txt == '':
button.config(text='W')
def execute_action(self, agent, action):
"""Determines the action the agent performs."""
xi, yi = (self.xi, self.yi)
if action == 'Suck':
dirt_list = self.list_things_at(agent.location, Dirt)
if dirt_list:
dirt = dirt_list[0]
agent.performance += 100
self.delete_thing(dirt)
self.buttons[xi][yi].config(text='', state='normal')
xf, yf = agent.location
self.buttons[xf][yf].config(
text='A', state='disabled', disabledforeground='black')
else:
agent.bump = False
if action == 'TurnRight':
agent.direction += Direction.R
elif action == 'TurnLeft':
agent.direction += Direction.L
elif action == 'Forward':
agent.bump = self.move_to(agent, agent.direction.move_forward(agent.location))
if not agent.bump:
self.buttons[xi][yi].config(text='', state='normal')
xf, yf = agent.location
self.buttons[xf][yf].config(
text='A', state='disabled', disabledforeground='black')
if action != 'NoOp':
agent.performance -= 1
def read_env(self):
"""Reads the current state of the GUI environment."""
for i, btn_row in enumerate(self.buttons):
for j, btn in enumerate(btn_row):
if (i != 0 and i != len(self.buttons) - 1) and (j != 0 and j != len(btn_row) - 1):
agt_loc = self.agents[0].location
if self.some_things_at((i, j)) and (i, j) != agt_loc:
for thing in self.list_things_at((i, j)):
self.delete_thing(thing)
if btn['text'] == self.elements[0]:
self.add_thing(Dirt(), (i, j))
elif btn['text'] == self.elements[1]:
self.add_thing(Wall(), (i, j))
def update_env(self):
"""Updates the GUI environment according to the current state."""
self.read_env()
agt = self.agents[0]
previous_agent_location = agt.location
self.xi, self.yi = previous_agent_location
self.step()
xf, yf = agt.location
def reset_env(self, agt):
"""Resets the GUI environment to the initial state."""
self.read_env()
for i, btn_row in enumerate(self.buttons):
for j, btn in enumerate(btn_row):
if (i != 0 and i != len(self.buttons) - 1) and (j != 0 and j != len(btn_row) - 1):
if self.some_things_at((i, j)):
for thing in self.list_things_at((i, j)):
self.delete_thing(thing)
btn.config(text='', state='normal')
self.add_thing(agt, location=(3, 3))
self.buttons[3][3].config(
text='A', state='disabled', disabledforeground='black')
def XYReflexAgentProgram(percept):
"""The modified SimpleReflexAgentProgram for the GUI environment."""
status, bump = percept
if status == 'Dirty':
return 'Suck'
if bump == 'Bump':
value = random.choice((1, 2))
else:
value = random.choice((1, 2, 3, 4))
if value == 1:
return 'TurnRight'
elif value == 2:
return 'TurnLeft'
else:
return 'Forward'
class XYReflexAgent(Agent):
"""The modified SimpleReflexAgent for the GUI environment."""
def __init__(self, program=None):
super().__init__(program)
self.location = (3, 3)
self.direction = Direction("up")
if __name__ == "__main__":
root = Tk()
root.title("Vacuum Environment")
root.geometry("420x440")
root.resizable(0, 0)
frame = Frame(root, bg='black')
reset_button = Button(frame, text='Reset', height=2,
width=6, padx=2, pady=2)
reset_button.pack(side='left')
next_button = Button(frame, text='Next', height=2,
width=6, padx=2, pady=2)
next_button.pack(side='left')
frame.pack(side='bottom')
env = Gui(root)
agt = XYReflexAgent(program=XYReflexAgentProgram)
env.add_thing(agt, location=(3, 3))
next_button.config(command=env.update_env)
reset_button.config(command=lambda: env.reset_env(agt))
root.mainloop()
