separated out graph code into more file and its own sub package

coretk/coretk/graph/node.py

@@ -0,0 +1,163 @@
+import logging
+import tkinter as tk
+from tkinter import font
+
+from coretk.dialogs.emaneconfig import EmaneConfigDialog
+from coretk.dialogs.mobilityconfig import MobilityConfigDialog
+from coretk.dialogs.nodeconfig import NodeConfigDialog
+from coretk.dialogs.wlanconfig import WlanConfigDialog
+from coretk.graph.canvastooltip import CanvasTooltip
+from coretk.graph.enums import GraphMode
+from coretk.graph.graph_helper import WlanAntennaManager
+
+NODE_TEXT_OFFSET = 5
+
+
+class CanvasNode:
+    def __init__(self, app, core_node, image):
+        self.app = app
+        self.canvas = app.canvas
+        self.image = image
+        self.core_node = core_node
+        x = self.core_node.position.x
+        y = self.core_node.position.y
+        self.id = self.canvas.create_image(
+            x, y, anchor=tk.CENTER, image=self.image, tags="node"
+        )
+        image_box = self.canvas.bbox(self.id)
+        y = image_box[3] + NODE_TEXT_OFFSET
+        text_font = font.Font(family="TkIconFont", size=12)
+        self.text_id = self.canvas.create_text(
+            x,
+            y,
+            text=self.core_node.name,
+            tags="nodename",
+            font=text_font,
+            fill="#0000CD",
+        )
+        self.antenna_draw = WlanAntennaManager(self.canvas, self.id)
+        self.tooltip = CanvasTooltip(self.canvas)
+        self.canvas.tag_bind(self.id, "<ButtonPress-1>", self.click_press)
+        self.canvas.tag_bind(self.id, "<ButtonRelease-1>", self.click_release)
+        self.canvas.tag_bind(self.id, "<B1-Motion>", self.motion)
+        self.canvas.tag_bind(self.id, "<Double-Button-1>", self.double_click)
+        self.canvas.tag_bind(self.id, "<Control-1>", self.select_multiple)
+        self.canvas.tag_bind(self.id, "<Enter>", self.on_enter)
+        self.canvas.tag_bind(self.id, "<Leave>", self.on_leave)
+        self.edges = set()
+        self.interfaces = []
+        self.wireless_edges = set()
+        self.moving = None
+
+    def redraw(self):
+        self.canvas.itemconfig(self.id, image=self.image)
+        self.canvas.itemconfig(self.text_id, text=self.core_node.name)
+
+    def move(self, x, y, update=True):
+        old_x = self.core_node.position.x
+        old_y = self.core_node.position.y
+        x_offset = x - old_x
+        y_offset = y - old_y
+        self.core_node.position.x = int(x)
+        self.core_node.position.y = int(y)
+        self.canvas.move(self.id, x_offset, y_offset)
+        self.canvas.move(self.text_id, x_offset, y_offset)
+        self.antenna_draw.update_antennas_position(x_offset, y_offset)
+        self.canvas.canvas_management.node_drag(self, x_offset, y_offset)
+        for edge in self.edges:
+            x1, y1, x2, y2 = self.canvas.coords(edge.id)
+            if edge.src == self.id:
+                self.canvas.coords(edge.id, x, y, x2, y2)
+            else:
+                self.canvas.coords(edge.id, x1, y1, x, y)
+            edge.link_info.recalculate_info()
+        for edge in self.wireless_edges:
+            x1, y1, x2, y2 = self.canvas.coords(edge.id)
+            if edge.src == self.id:
+                self.canvas.coords(edge.id, x, y, x2, y2)
+            else:
+                self.canvas.coords(edge.id, x1, y1, x, y)
+        if self.app.core.is_runtime() and update:
+            self.app.core.edit_node(self.core_node.id, int(x), int(y))
+
+    def on_enter(self, event):
+        if self.app.core.is_runtime() and self.app.core.observer:
+            self.tooltip.text.set("waiting...")
+            self.tooltip.on_enter(event)
+            output = self.app.core.run(self.core_node.id)
+            self.tooltip.text.set(output)
+
+    def on_leave(self, event):
+        self.tooltip.on_leave(event)
+
+    def click(self, event):
+        print("click")
+
+    def double_click(self, event):
+        if self.app.core.is_runtime():
+            self.canvas.core.launch_terminal(self.core_node.id)
+        else:
+            self.show_config()
+
+    def update_coords(self):
+        x, y = self.canvas.coords(self.id)
+        self.core_node.position.x = int(x)
+        self.core_node.position.y = int(y)
+
+    def click_press(self, event):
+        logging.debug(f"node click press {self.core_node.name}: {event}")
+        self.moving = self.canvas.canvas_xy(event)
+        if self.id not in self.canvas.canvas_management.selected:
+            self.canvas.canvas_management.node_select(self)
+            self.canvas.selected = self.id
+
+    def click_release(self, event):
+        logging.debug(f"node click release {self.core_node.name}: {event}")
+        self.update_coords()
+        self.moving = None
+
+    def motion(self, event):
+        if self.canvas.mode == GraphMode.EDGE:
+            return
+        x, y = self.canvas.canvas_xy(event)
+        my_x = self.core_node.position.x
+        my_y = self.core_node.position.y
+        self.move(x, y)
+        # move other selected components
+        for nid, bboxid in self.canvas.canvas_management.selected.items():
+            if nid != self.id and nid in self.canvas.nodes:
+                other_old_x = self.canvas.nodes[nid].core_node.position.x
+                other_old_y = self.canvas.nodes[nid].core_node.position.y
+                other_new_x = x + other_old_x - my_x
+                other_new_y = y + other_old_y - my_y
+                self.canvas.nodes[nid].move(other_new_x, other_new_y)
+            if nid != self.id and nid in self.canvas.shapes:
+                self.canvas.shapes[nid].motion(None, x - my_x, y - my_y)
+
+    def select_multiple(self, event):
+        self.canvas.canvas_management.node_select(self, True)
+
+    def show_config(self):
+        self.canvas.context = None
+        dialog = NodeConfigDialog(self.app, self.app, self)
+        dialog.show()
+
+    def show_wlan_config(self):
+        self.canvas.context = None
+        dialog = WlanConfigDialog(self.app, self.app, self)
+        dialog.show()
+
+    def show_mobility_config(self):
+        self.canvas.context = None
+        dialog = MobilityConfigDialog(self.app, self.app, self)
+        dialog.show()
+
+    def show_mobility_player(self):
+        self.canvas.context = None
+        mobility_player = self.app.core.mobility_players[self.core_node.id]
+        mobility_player.show()
+
+    def show_emane_config(self):
+        self.canvas.context = None
+        dialog = EmaneConfigDialog(self.app, self.app, self)
+        dialog.show()