Create xy_vacuum_environment.py

Author: Cat-Gawr

src_A2/gui/xy_vacuum_environment.py

@@ -0,0 +1,191 @@
+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')
+        # Place the agent in the centre of the grid.
+        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))  # 1-right, 2-left, others-forward
+
+    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")
+
+
+# TODO: Check the coordinate system.
+# TODO: Give manual choice for agent's location.
+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()