Fix: correctly sort instruments in device annotator (#16)

* Fix: correctly sort instruments in device annotator * Fix: device annotator improve formatting * Fix: use pyqt5 directly
jenshnielsen · Jun 26, 2017 · 15897e6 · 15897e6
1 parent b729f1a
commit 15897e6
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/qcodes/utils/qcodes_device_annotator.py b/qcodes/utils/qcodes_device_annotator.py
@@ -4,9 +4,9 @@
 import os
 import json
 import glob
-import qtpy.QtWidgets as qt
-import qtpy.QtGui as gui
-import qtpy.QtCore as core
+import PyQt5.QtWidgets as qt
+import PyQt5.QtGui as gui
+import PyQt5.QtCore as core
 
 from shutil import copyfile
 
@@ -49,7 +49,8 @@ def __init__(self, folder, station):
         self.model.setHorizontalHeaderLabels([self.tr("Instruments")])
         self.addStation(self.model, station)
         self.treeView.setModel(self.model)
-
+        self.treeView.sortByColumn(0, core.Qt.AscendingOrder)
+        self.treeView.setSortingEnabled(True)
         grid.addWidget(self.imageCanvas, 0, 0, 4, 6)
         grid.addWidget(self.loadButton, 4, 0)
         grid.addWidget(self.labelButton, 4, 1)
@@ -73,7 +74,6 @@ def addStation(self, parent, station):
                 paramitem = gui.QStandardItem(param)
                 paramitem.setEditable(False)
                 item.appendRow(paramitem)
-            item.sortChildren(0)
 
     def loadimage(self):
         """
@@ -262,7 +262,16 @@ def updateValues(self, station):
 
         for instrument, parameters in self._data.items():
             for parameter in parameters.keys():
-                self._data[instrument][parameter]['value'] = str(station.components[instrument][parameter].get_latest())
+                value = station.components[instrument][parameter].get_latest()
+                try:
+                    floatvalue = float(station.components[instrument][parameter].get_latest())
+                    if floatvalue > 1000 or floatvalue < 0.1:
+                        valuestr = "{:.2e}".format(floatvalue)
+                    else:
+                        valuestr = "{:.2f}".format(floatvalue)
+                except ValueError:
+                    valuestr = str(value)
+                self._data[instrument][parameter]['value'] = valuestr
 
     def makePNG(self, counter, path=None):
         """