pygui: added menu buttons to add/delete canvases

daemon/core/gui/app.py

@@ -138,22 +138,8 @@ class Application(ttk.Frame):
         label.grid(sticky=tk.EW, pady=PADY)
 
     def draw_canvas(self) -> None:
-        canvas_frame = ttk.Frame(self.right_frame)
-        canvas_frame.rowconfigure(0, weight=1)
-        canvas_frame.columnconfigure(0, weight=1)
-        canvas_frame.grid(row=0, column=0, sticky=tk.NSEW, pady=1)
-        self.manager = CanvasManager(canvas_frame, self, self.core)
+        self.manager = CanvasManager(self.right_frame, self, self.core)
         self.manager.notebook.grid(sticky=tk.NSEW)
-        # self.canvas = CanvasGraph(canvas_frame, self, self.core)
-        # self.canvas.grid(sticky=tk.NSEW)
-        # scroll_y = ttk.Scrollbar(canvas_frame, command=self.canvas.yview)
-        # scroll_y.grid(row=0, column=1, sticky=tk.NS)
-        # scroll_x = ttk.Scrollbar(
-        #     canvas_frame, orient=tk.HORIZONTAL, command=self.canvas.xview
-        # )
-        # scroll_x.grid(row=1, column=0, sticky=tk.EW)
-        # self.canvas.configure(xscrollcommand=scroll_x.set)
-        # self.canvas.configure(yscrollcommand=scroll_y.set)
 
     def draw_status(self) -> None:
         self.statusbar = StatusBar(self.right_frame, self)

daemon/core/gui/graph/manager.py

@@ -1,6 +1,6 @@
 import logging
 import tkinter as tk
-from tkinter import BooleanVar, ttk
+from tkinter import BooleanVar, messagebox, ttk
 from typing import TYPE_CHECKING, Dict, Optional, Set, Tuple
 
 from core.api.grpc.wrappers import Session
@@ -70,11 +70,55 @@ class CanvasManager:
 
         # widget
         self.notebook: Optional[ttk.Notebook] = None
+        self.unique_ids: Dict[str, int] = {}
         self.draw()
 
     def draw(self) -> None:
         self.notebook = ttk.Notebook(self.master)
-        self.notebook.grid(sticky=tk.NSEW)
+        self.notebook.grid(sticky=tk.NSEW, pady=1)
+
+    def next_id(self) -> int:
+        _id = 1
+        tab_ids = set(self.unique_ids.values())
+        while _id in tab_ids:
+            _id += 1
+        return _id
+
+    def add_canvas(self) -> CanvasGraph:
+        # create tab frame
+        tab = ttk.Frame(self.notebook, padding=0)
+        tab.grid(sticky=tk.NSEW)
+        tab.columnconfigure(0, weight=1)
+        tab.rowconfigure(0, weight=1)
+        tab_id = self.next_id()
+        self.notebook.add(tab, text=f"Canvas {tab_id}")
+        unique_id = self.notebook.tabs()[-1]
+        logging.info("tab(%s) is %s", unique_id, tab_id)
+        self.unique_ids[unique_id] = tab_id
+
+        # create canvas
+        canvas = CanvasGraph(tab, self.app, self, self.core, tab_id)
+        canvas.grid(sticky=tk.NSEW)
+        self.canvases[tab_id] = canvas
+
+        # add scrollbars
+        scroll_y = ttk.Scrollbar(tab, command=canvas.yview)
+        scroll_y.grid(row=0, column=1, sticky=tk.NS)
+        scroll_x = ttk.Scrollbar(tab, orient=tk.HORIZONTAL, command=canvas.xview)
+        scroll_x.grid(row=1, column=0, sticky=tk.EW)
+        canvas.configure(xscrollcommand=scroll_x.set)
+        canvas.configure(yscrollcommand=scroll_y.set)
+        return canvas
+
+    def delete_canvas(self) -> None:
+        if len(self.notebook.tabs()) == 1:
+            messagebox.showinfo("Canvas", "Cannot delete last canvas", parent=self.app)
+            return
+        unique_id = self.notebook.select()
+        self.notebook.forget(unique_id)
+        tab_id = self.unique_ids.pop(unique_id)
+        self.canvases.pop(tab_id)
+        # TODO: handle clearing out canvas related nodes and links
 
     def join(self, session: Session) -> None:
         # clear out all canvas
@@ -95,18 +139,11 @@ class CanvasManager:
         self.annotation_type = None
         self.node_draw = None
 
+        # TODO: create and add nodes to all associated canvases
         # draw initial tab(s) and session
-        tab = ttk.Frame(self.notebook, padding=0)
-        tab.grid(sticky=tk.NSEW)
-        tab.columnconfigure(0, weight=1)
-        tab.rowconfigure(0, weight=1)
-        tab_id = len(self.notebook.tabs())
-        self.notebook.add(tab, text=f"Canvas {tab_id}")
-        logging.info("canvas tab id: %s", tab_id)
-        canvas = CanvasGraph(tab, self.app, self, self.core, tab_id)
-        canvas.grid(sticky=tk.NSEW)
-        self.canvases[tab_id] = canvas
+        canvas = self.add_canvas()
 
+        # draw session on canvas
         canvas.reset_and_redraw(session)
         self.core.parse_metadata(canvas)
         canvas.organize()

daemon/core/gui/menubar.py

@@ -128,6 +128,8 @@ class Menubar(tk.Menu):
         Create canvas menu
         """
         menu = tk.Menu(self)
+        menu.add_command(label="New", command=self.click_canvas_add)
+        menu.add_command(label="Delete", command=self.click_canvas_delete)
         menu.add_command(label="Size / Scale", command=self.click_canvas_size_and_scale)
         menu.add_command(label="Wallpaper", command=self.click_canvas_wallpaper)
         self.add_cascade(label="Canvas", menu=menu)
@@ -372,6 +374,12 @@ class Menubar(tk.Menu):
         dialog = PreferencesDialog(self.app)
         dialog.show()
 
+    def click_canvas_add(self) -> None:
+        self.canvas_manager.add_canvas()
+
+    def click_canvas_delete(self) -> None:
+        self.canvas_manager.delete_canvas()
+
     def click_canvas_size_and_scale(self) -> None:
         dialog = SizeAndScaleDialog(self.app)
         dialog.show()