m-labs · Mar 29, 2016 · Mar 29, 2016 · Mar 29, 2016
Showing with 55 additions and 32 deletions.

+2 −0 RELEASE_NOTES.rst

+11 −9 artiq/frontend/artiq_gui.py

+4 −1 artiq/gui/applets.py

+16 −6 artiq/language/environment.py

+14 −8 artiq/master/worker_db.py

+8 −8 examples/master/repository/flopping_f_simulation.py
diff --git a/RELEASE_NOTES.rst b/RELEASE_NOTES.rst
@@ -9,6 +9,8 @@ unreleased
 * The CPU speed in the pipistrello gateware has been reduced from 83 1/3 MHz to
   75 MHz. This will reduce the achievable sustained pulse rate and latency
   accordingly. ISE was intermittently failing to meet timing (#341).
+* set_dataset in broadcast mode no longer returns a Notifier. Mutating datasets
+  should be done with mutate_dataset instead (#345).
 
 
 1.0rc1

diff --git a/artiq/frontend/artiq_gui.py b/artiq/frontend/artiq_gui.py
@@ -37,9 +37,14 @@ def get_argparser():
 class MainWindow(QtWidgets.QMainWindow):
     def __init__(self, server):
         QtWidgets.QMainWindow.__init__(self)
+
         icon = QtGui.QIcon(os.path.join(artiq_dir, "gui", "logo.svg"))
         self.setWindowIcon(icon)
         self.setWindowTitle("ARTIQ - {}".format(server))
+
+        qfm = QtGui.QFontMetrics(self.font())
+        self.resize(140*qfm.averageCharWidth(), 38*qfm.lineSpacing())
+
         self.exit_request = asyncio.Event()
 
     def closeEvent(self, *args):
@@ -149,17 +154,14 @@ def main():
     smgr.register(logmgr)
 
     # lay out docks
+    right_docks = [d_explorer, d_shortcuts]
     if os.name != "nt":
-        main_window.addDockWidget(QtCore.Qt.RightDockWidgetArea, d_ttl_dds.dds_dock)
-        main_window.tabifyDockWidget(d_ttl_dds.dds_dock, d_ttl_dds.ttl_dock)
-        main_window.tabifyDockWidget(d_ttl_dds.ttl_dock, d_applets)
-        main_window.tabifyDockWidget(d_applets, d_datasets)
-    else:
-        main_window.addDockWidget(QtCore.Qt.RightDockWidgetArea, d_applets)
-        main_window.tabifyDockWidget(d_applets, d_datasets)
-    main_window.tabifyDockWidget(d_datasets, d_shortcuts)
+        right_docks += [d_ttl_dds.ttl_dock, d_ttl_dds.dds_dock]
+    right_docks += [d_datasets, d_applets]
+    main_window.addDockWidget(QtCore.Qt.RightDockWidgetArea, right_docks[0])
+    for d1, d2 in zip(right_docks, right_docks[1:]):
+        main_window.tabifyDockWidget(d1, d2)
     main_window.addDockWidget(QtCore.Qt.BottomDockWidgetArea, d_schedule)
-    main_window.addDockWidget(QtCore.Qt.LeftDockWidgetArea, d_explorer)
 
     # load/initialize state
     if os.name == "nt":

diff --git a/artiq/gui/applets.py b/artiq/gui/applets.py
@@ -89,7 +89,10 @@ class _AppletDock(QDockWidgetCloseDetect):
     def __init__(self, datasets_sub, uid, name, command):
         QDockWidgetCloseDetect.__init__(self, "Applet: " + name)
         self.setObjectName("applet" + str(uid))
-        self.setMinimumSize(QtCore.QSize(100, 100))
+
+        qfm = QtGui.QFontMetrics(self.font())
+        self.setMinimumSize(20*qfm.averageCharWidth(), 5*qfm.lineSpacing())
+        self.resize(40*qfm.averageCharWidth(), 10*qfm.lineSpacing())
 
         self.datasets_sub = datasets_sub
         self.applet_name = name

diff --git a/artiq/language/environment.py b/artiq/language/environment.py
@@ -221,13 +221,11 @@ def set_dataset(self, key, value,
                     broadcast=False, persist=False, save=True):
         """Sets the contents and handling modes of a dataset.
 
-        If the dataset is broadcasted, it must be PYON-serializable.
-        If the dataset is saved, it must be a scalar (``bool``, ``int``,
-        ``float`` or NumPy scalar) or a NumPy array.
+        Datasets must be scalars (``bool``, ``int``, ``float`` or NumPy scalar)
+        or NumPy arrays.
 
         :param broadcast: the data is sent in real-time to the master, which
-            dispatches it. Returns a Notifier that can be used to mutate the
-            dataset.
+            dispatches it.
         :param persist: the master should store the data on-disk. Implies
             broadcast.
         :param save: the data is saved into the local storage of the current
@@ -238,7 +236,19 @@ def set_dataset(self, key, value,
             return
         if self.__dataset_mgr is None:
             raise ValueError("Dataset manager not present")
-        return self.__dataset_mgr.set(key, value, broadcast, persist, save)
+        self.__dataset_mgr.set(key, value, broadcast, persist, save)
+
+    def mutate_dataset(self, key, index, value):
+        """Mutate an existing dataset at the given index (e.g. set a value at
+        a given position in a NumPy array)
+
+        If the dataset was created in broadcast mode, the modification is
+        immediately transmitted."""
+        if self.__parent is not None:
+            self.__parent.mutate_dataset(key, index, value)
+        if self.__dataset_mgr is None:
+            raise ValueError("Dataset manager not present")
+        self.__dataset_mgr.mutate(key, index, value)
 
     def get_dataset(self, key, default=NoDefault):
         """Returns the contents of a dataset.

diff --git a/artiq/master/worker_db.py b/artiq/master/worker_db.py
@@ -218,20 +218,26 @@ def __init__(self, ddb):
     def set(self, key, value, broadcast=False, persist=False, save=True):
         if persist:
             broadcast = True
-        r = None
         if broadcast:
-            self.broadcast[key] = (persist, value)
-            r = self.broadcast[key][1]
+            self.broadcast[key] = persist, value
         if save:
             self.local[key] = value
-        return r
+
+    def mutate(self, key, index, value):
+        target = None
+        if key in self.local:
+            target = self.local[key]
+        if key in self.broadcast.read:
+            target = self.broadcast[key][1]
+        if target is None:
+            raise KeyError("Cannot mutate non-existing dataset")
+        target[index] = value
 
     def get(self, key):
-        try:
+        if key in self.local:
             return self.local[key]
-        except KeyError:
-            pass
-        return self.ddb.get(key)
+        else:
+            return self.ddb.get(key)
 
     def write_hdf5(self, f):
         result_dict_to_hdf5(f, self.local)
diff --git a/examples/master/repository/flopping_f_simulation.py b/examples/master/repository/flopping_f_simulation.py
@@ -38,19 +38,19 @@ def build(self):
 
     def run(self):
         l = len(self.frequency_scan)
-        frequency = self.set_dataset("flopping_f_frequency",
-                                     np.full(l, np.nan),
-                                     broadcast=True, save=False)
-        brightness = self.set_dataset("flopping_f_brightness",
-                                      np.full(l, np.nan),
-                                     broadcast=True)
+        self.set_dataset("flopping_f_frequency",
+                         np.full(l, np.nan),
+                         broadcast=True, save=False)
+        self.set_dataset("flopping_f_brightness",
+                         np.full(l, np.nan),
+                         broadcast=True)
         self.set_dataset("flopping_f_fit", np.full(l, np.nan),
                          broadcast=True, save=False)
 
         for i, f in enumerate(self.frequency_scan):
             m_brightness = model(f, self.F0) + self.noise_amplitude*random.random()
-            frequency[i] = f
-            brightness[i] = m_brightness
+            self.mutate_dataset("flopping_f_frequency", i, f)
+            self.mutate_dataset("flopping_f_brightness", i, m_brightness)
             time.sleep(0.1)
         self.scheduler.submit(self.scheduler.pipeline_name, self.scheduler.expid,
                               self.scheduler.priority, time.time() + 20, False)