pygui: added layer menu to nodes, simple toggle support for nodes alone

daemon/core/gui/graph/layers.py

@@ -1,10 +1,40 @@
 import tkinter as tk
+from functools import partial
 from typing import TYPE_CHECKING, Dict, Iterable, Set
 
+from core.gui import themes
+
 if TYPE_CHECKING:
+    from core.gui.app import Application
     from core.gui.graph.graph import CanvasGraph
 
 
+class LayersMenu(tk.Menu):
+    def __init__(self, master: tk.BaseWidget, app: "Application", item: int) -> None:
+        super().__init__(master)
+        themes.style_menu(self)
+        self.app: "Application" = app
+        self.item: int = item
+        self.buttons: Dict[str, tk.BooleanVar] = {}
+        self.draw()
+
+    def draw(self) -> None:
+        for name in self.app.canvas.layers.names():
+            value = self.app.canvas.layers.in_layer(name, self.item)
+            var = tk.BooleanVar(value=value)
+            self.buttons[name] = var
+            self.add_checkbutton(
+                label=name, variable=var, command=partial(self.click_layer, name)
+            )
+
+    def click_layer(self, name):
+        value = self.buttons[name].get()
+        if value:
+            self.app.canvas.layers.add_item(name, self.item)
+        else:
+            self.app.canvas.layers.delete_item(name, self.item)
+
+
 class CanvasLayers:
     def __init__(self, canvas: "CanvasGraph"):
         self.canvas: "CanvasGraph" = canvas
@@ -25,20 +55,24 @@ class CanvasLayers:
         items = self.layers.pop(name, set())
         hidden_items = self.all_hidden()
         items -= hidden_items
-        self.canvas.config(items, state=tk.NORMAL)
+        for item in items:
+            self.canvas.itemconfig(item, state=tk.NORMAL)
+
+    def in_layer(self, name: str, item: int) -> bool:
+        return item in self.layers.get(name, set())
 
     def add_item(self, name: str, item: int) -> None:
         if name in self.layers:
             self.layers[name].add(item)
             if name in self.hidden:
-                self.canvas.config(item, state=tk.HIDDEN)
+                self.canvas.itemconfig(item, state=tk.HIDDEN)
 
     def delete_item(self, name: str, item: int) -> None:
         if name in self.layers:
             self.layers[name].remove(item)
             hidden_items = self.all_hidden()
             if item not in hidden_items:
-                self.canvas.config(item, state=tk.NORMAL)
+                self.canvas.itemconfig(item, state=tk.NORMAL)
 
     def toggle_layer(self, name: str) -> None:
         items = self.layers[name]
@@ -46,10 +80,12 @@ class CanvasLayers:
             self.hidden.remove(name)
             hidden_items = self.all_hidden()
             items -= hidden_items
-            self.canvas.config(items, state=tk.NORMAL)
+            for item in items:
+                self.canvas.itemconfig(item, state=tk.NORMAL)
         else:
             self.hidden.add(name)
-            self.canvas.config(items, state=tk.HIDDEN)
+            for item in items:
+                self.canvas.itemconfig(item, state=tk.HIDDEN)
 
     def all_hidden(self) -> Set[int]:
         items = set()

daemon/core/gui/graph/node.py

@@ -19,6 +19,7 @@ from core.gui.dialogs.wlanconfig import WlanConfigDialog
 from core.gui.frames.node import NodeInfoFrame
 from core.gui.graph import tags
 from core.gui.graph.edges import CanvasEdge, CanvasWirelessEdge
+from core.gui.graph.layers import LayersMenu
 from core.gui.graph.tooltip import CanvasTooltip
 from core.gui.images import ImageEnum
 from core.gui.nodeutils import ANTENNA_SIZE, NodeUtils
@@ -271,6 +272,9 @@ class CanvasNode:
             edit_menu.add_command(label="Copy", command=self.canvas_copy)
             edit_menu.add_command(label="Delete", command=self.canvas_delete)
             self.context.add_cascade(label="Edit", menu=edit_menu)
+
+        layer_menu = LayersMenu(self.context, self.app, self.id)
+        self.context.add_cascade(label="Layer", menu=layer_menu)
         self.context.tk_popup(event.x_root, event.y_root)
 
     def click_cut(self) -> None: