Add files via upload

python_symb/visual.py

@@ -0,0 +1,92 @@
+import tkinter as tk
+from expr import Expr
+
+class Visual:
+    def __init__(self):
+        self.root = tk.Tk()
+        self.root.title("Visual")
+        self.root.geometry("800x800")
+
+        # Add a canvas on the left side of the window
+        self.tree_canvas = tk.Canvas(self.root, width=500, height=500, bg="white")
+        self.tree_canvas.pack(side=tk.LEFT, expand=False)
+
+        # Create right frame
+        self.right_frame = tk.Frame(self.root, width=300, height=500, background="grey")
+
+
+        # Add a label with the text "Input" on top of the right frame, should not take all the space
+        self.input_label = tk.Label(self.right_frame, text="Input (infix string):")
+
+
+        # Add a entry below the label with default text "(5+2)*3"
+        self.input_entry = tk.Entry(self.right_frame, width=50)
+        self.input_entry.insert(0, "(5+2)*3")
+
+
+
+        # Add a button below the entry with the text "To Tree"
+        self.input_button = tk.Button(self.right_frame, text="To Tree", command=self.show_tree)
+
+        # Pack the widgets, make sure they are not expanding, and all are fixed size
+
+
+        self.input_label.pack()
+        self.input_entry.pack()
+        self.input_button.pack()
+        self.right_frame.pack(side=tk.LEFT, expand=False)
+
+
+
+
+
+
+        self.root.mainloop()
+
+    def show_tree(self):
+        # Get the text from the entry
+        text = self.input_entry.get()
+
+        # Clear the canvas
+        self.tree_canvas.delete("all")
+        tree = Expr.from_infix_str(text)
+
+        # Draw the tree
+        self.draw_tree(tree)
+
+    def create_circle(self, x, y, r, color):  # center coordinates, radius
+        x0 = x - r
+        y0 = y - r
+        x1 = x + r
+        y1 = y + r
+        return self.tree_canvas.create_oval(x0, y0, x1, y1, fill=color)
+
+    def draw_tree(self, tree, first_x=250, first_y=50, x_offset=100, y_offset=100):
+
+        children = tree.children
+        n = len(children)
+        for i in range(n):
+            child = children[i]
+            x = first_x + (i - (n - 1) / 2) * x_offset
+            y = first_y + y_offset
+
+            #Link the node to the parent
+            self.tree_canvas.create_line(first_x, first_y, x, y)
+            # Draw the node
+            self.draw_tree(child, int(x), y, x_offset, y_offset)
+
+        # Draw the root node
+        self.create_circle(first_x, first_y, 20, "red")
+        self.tree_canvas.create_text(first_x, first_y, text=str(tree.value))
+
+if __name__ == "__main__":
+    Visual()
+
+
+
+
+
+
+
+
+