From 4c4da77cc805b21807dbee73e6ef5753246b5f68 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Mon, 5 Jan 2015 19:52:58 +0800
Subject: [PATCH] gui: 'remove' buttons functionality

---
 artiq/gui/scheduler.py | 30 +++++++++++++++++++++++-------
 frontend/artiq_gui.py  | 16 ++++++++++------
 2 files changed, 33 insertions(+), 13 deletions(-)

diff --git a/artiq/gui/scheduler.py b/artiq/gui/scheduler.py
index 68e2368b8..fc2bac505 100644
--- a/artiq/gui/scheduler.py
+++ b/artiq/gui/scheduler.py
@@ -32,7 +32,9 @@ class _PeriodicStoreSyncer(DictSyncer):
 
 
 class SchedulerWindow(Window):
-    def __init__(self):
+    def __init__(self, schedule_ctl):
+        self.schedule_ctl = schedule_ctl
+
         Window.__init__(self, title="Scheduler")
         self.set_default_size(720, 570)
 
@@ -50,13 +52,13 @@ class SchedulerWindow(Window):
         topvbox.pack_start(notebook, True, True, 0)
 
         self.queue_store = Gtk.ListStore(int, str, str, str)
-        tree = Gtk.TreeView(self.queue_store)
+        self.queue_tree = Gtk.TreeView(self.queue_store)
         for i, title in enumerate(["RID", "File", "Unit", "Timeout"]):
             renderer = Gtk.CellRendererText()
             column = Gtk.TreeViewColumn(title, renderer, text=i)
-            tree.append_column(column)
+            self.queue_tree.append_column(column)
         scroll = Gtk.ScrolledWindow()
-        scroll.add(tree)
+        scroll.add(self.queue_tree)
         vbox = Gtk.VBox(spacing=6)
         vbox.pack_start(scroll, True, True, 0)
         hbox = Gtk.HBox(spacing=6)
@@ -67,31 +69,45 @@ class SchedulerWindow(Window):
         button = Gtk.Button("Move down")
         hbox.pack_start(button, True, True, 0)
         button = Gtk.Button("Remove")
+        button.connect("clicked", self.remove_queue)
         hbox.pack_start(button, True, True, 0)
         vbox.pack_start(hbox, False, False, 0)
         vbox.set_border_width(6)
         notebook.insert_page(vbox, Gtk.Label("Queue"), -1)
 
         self.periodic_store = Gtk.ListStore(str, int, str, str, str, str)
-        tree = Gtk.TreeView(self.periodic_store)
+        self.periodic_tree = Gtk.TreeView(self.periodic_store)
         for i, title in enumerate(["Next run", "PRID", "File", "Unit",
                                    "Timeout", "Period"]):
             renderer = Gtk.CellRendererText()
             column = Gtk.TreeViewColumn(title, renderer, text=i)
-            tree.append_column(column)
+            self.periodic_tree.append_column(column)
         scroll = Gtk.ScrolledWindow()
-        scroll.add(tree)
+        scroll.add(self.periodic_tree)
         vbox = Gtk.VBox(spacing=6)
         vbox.pack_start(scroll, True, True, 0)
         hbox = Gtk.HBox(spacing=6)
         button = Gtk.Button("Change period")
         hbox.pack_start(button, True, True, 0)
         button = Gtk.Button("Remove")
+        button.connect("clicked", self.remove_periodic)
         hbox.pack_start(button, True, True, 0)
         vbox.pack_start(hbox, False, False, 0)
         vbox.set_border_width(6)
         notebook.insert_page(vbox, Gtk.Label("Periodic schedule"), -1)
 
+    def remove_queue(self, widget):
+        store, selected = self.queue_tree.get_selection().get_selected()
+        if selected is not None:
+            rid = store[selected][0]
+            asyncio.Task(self.schedule_ctl.cancel_once(rid))
+
+    def remove_periodic(self, widget):
+        store, selected = self.periodic_tree.get_selection().get_selected()
+        if selected is not None:
+            prid = store[selected][1]
+            asyncio.Task(self.schedule_ctl.cancel_periodic(prid))
+
     @asyncio.coroutine
     def sub_connect(self, host, port):
         self.queue_subscriber = Subscriber("queue", self.init_queue_store)
diff --git a/frontend/artiq_gui.py b/frontend/artiq_gui.py
index ea3a1db76..d72044ff1 100755
--- a/frontend/artiq_gui.py
+++ b/frontend/artiq_gui.py
@@ -7,6 +7,7 @@ import atexit
 import gbulb
 from gi.repository import Gtk
 
+from artiq.management.pc_rpc import AsyncioClient
 from artiq.gui.scheduler import SchedulerWindow
 from artiq.gui.parameters import ParametersWindow
 
@@ -32,19 +33,22 @@ def main():
     loop = asyncio.get_event_loop()
     atexit.register(lambda: loop.close())
 
-    scheduler_win = SchedulerWindow()
+    # share the schedule control connection
+    schedule_ctl = AsyncioClient()
+    loop.run_until_complete(schedule_ctl.connect_rpc(
+        args.server, args.port_control, "master_schedule"))
+
+    scheduler_win = SchedulerWindow(schedule_ctl)
     scheduler_win.connect("delete-event", Gtk.main_quit)
     scheduler_win.show_all()
-
-    parameters_win = ParametersWindow()
-    parameters_win.connect("delete-event", Gtk.main_quit)
-    parameters_win.show_all()
-
     loop.run_until_complete(scheduler_win.sub_connect(
         args.server, args.port_notify))
     atexit.register(
         lambda: loop.run_until_complete(scheduler_win.sub_close()))
 
+    parameters_win = ParametersWindow()
+    parameters_win.connect("delete-event", Gtk.main_quit)
+    parameters_win.show_all()
     loop.run_until_complete(parameters_win.sub_connect(
         args.server, args.port_notify))
     atexit.register(