Changes in / [44880ac:f331852] in sasview

Location:

src/sas/qtgui

Files:

• Calculators/DataOperationUtilityPanel.py (added)
• Calculators/UI/DataOperationUtilityUI.ui (added)
• Calculators/UnitTesting/DataOperationUtilityTest.py (added)
• GUITests.py (modified) (2 diffs)
• MainWindow/DataExplorer.py (modified) (5 diffs)
• MainWindow/DataManager.py (modified) (2 diffs)
• MainWindow/GuiManager.py (modified) (7 diffs)
• Plotting/Plotter.py (modified) (1 diff)
• Plotting/Plotter2D.py (modified) (4 diffs)
• Utilities/GuiUtils.py (modified) (6 diffs)
• Utilities/UnitTesting/GuiUtilsTest.py (modified) (2 diffs)

src/sas/qtgui/GUITests.py

@@ -38 +38 @@
 from Calculators.UnitTesting import SlitSizeCalculatorTest
 from Calculators.UnitTesting import ResolutionCalculatorPanelTest
+from Calculators.UnitTesting import DataOperationUtilityTest
 
 # Utilities
@@ -94 +95 @@
         unittest.makeSuite(SlitSizeCalculatorTest.SlitSizeCalculatorTest, 'test'),
         unittest.makeSuite(ResolutionCalculatorPanelTest.ResolutionCalculatorPanelTest, 'test'),
+        unittest.makeSuite(DataOperationUtilityTest.DataOperationUtilityTest, 'test'),
 
         # Perspectives

src/sas/qtgui/MainWindow/DataExplorer.py

@@ -93 +93 @@
         self.communicator.activeGraphName.connect(self.updatePlotName)
         self.communicator.plotUpdateSignal.connect(self.updatePlot)
+
         self.cbgraph.editTextChanged.connect(self.enableGraphCombo)
         self.cbgraph.currentIndexChanged.connect(self.enableGraphCombo)
@@ -280 +281 @@
         # Use 'while' so the row count is forced at every iteration
         deleted_indices = []
+        deleted_names = []
         while ind < self.model.rowCount():
             ind += 1
             item = self.model.item(ind)
+
             if item and item.isCheckable() and item.checkState() == QtCore.Qt.Checked:
                 # Delete these rows from the model
+                deleted_names.append(str(self.model.item(ind).text()))
                 deleted_indices.append(item)
+
                 self.model.removeRow(ind)
                 # Decrement index since we just deleted it
@@ -292 +297 @@
         # Let others know we deleted data
         self.communicator.dataDeletedSignal.emit(deleted_indices)
+
+        # update stored_data
+        self.manager.update_stored_data(deleted_names)
 
     def deleteTheory(self, event):
@@ -869 +877 @@
         self.txt_widget.setWindowIcon(QtGui.QIcon(":/res/ball.ico"))
         self.txt_widget.setWindowTitle("Data Info: %s" % data.filename)
+        self.txt_widget.clear()
         self.txt_widget.insertPlainText(text_to_show)
 
@@ -1012 +1021 @@
         self.model.appendRow(checkbox_item)
 
-
     def updateModelFromPerspective(self, model_item):
         """

src/sas/qtgui/MainWindow/DataManager.py

@@ -275 +275 @@
                 del self.stored_data[id]
 
-
     def get_by_name(self, name_list=None):
         """
@@ -295 +294 @@
                 if data_state.data.name == selected_name:
                     del self.stored_data[id]
+
+    def update_stored_data(self, name_list=None):
+        """ update stored data after deleting files in Data Explorer """
+        for selected_name in name_list:
+            for idx in self.stored_data.keys():
+                if str(selected_name) in str(idx):
+                    print selected_name, idx
+                    del self.stored_data[idx]
 
     def get_data_state(self, data_id):

src/sas/qtgui/MainWindow/GuiManager.py

@@ -31 +31 @@
 from sas.qtgui.Calculators.GenericScatteringCalculator import GenericScatteringCalculator
 from sas.qtgui.Calculators.ResolutionCalculatorPanel import ResolutionCalculatorPanel
-
+from sas.qtgui.Calculators.DataOperationUtilityPanel import DataOperationUtilityPanel
 
 # Perspectives
@@ -47 +47 @@
     Main SasView window functionality
     """
+
     def __init__(self, parent=None):
         """
@@ -142 +143 @@
         self.GENSASCalculator = GenericScatteringCalculator(self)
         self.ResolutionCalculator = ResolutionCalculatorPanel(self)
+        self.DataOperation = DataOperationUtilityPanel(self)
 
     def statusBarSetup(self):
@@ -337 +339 @@
         self.communicate.updateTheoryFromPerspectiveSignal.connect(self.updateTheoryFromPerspective)
         self.communicate.plotRequestedSignal.connect(self.showPlot)
+        self.communicate.updateModelFromDataOperationPanelSignal.connect(self.updateModelFromDataOperationPanel)
 
     def addTriggers(self):
@@ -531 +534 @@
         """
         """
-        print("actionData_Operation TRIGGERED")
-        pass
+        self.communicate.sendDataToPanelSignal.emit(self._data_manager.get_all_data())
+
+        self.DataOperation.show()
 
     def actionSLD_Calculator(self):
@@ -731 +735 @@
         self.filesWidget.updateTheoryFromPerspective(index)
 
+    def updateModelFromDataOperationPanel(self, new_item, new_datalist_item):
+        """
+        :param new_item: item to be added to list of loaded files
+        :param new_datalist_item:
+        """
+        if not isinstance(new_item, QtGui.QStandardItem) or \
+                not isinstance(new_datalist_item, dict):
+            msg = "Wrong data type returned from calculations."
+            raise AttributeError, msg
+
+        self.filesWidget.model.appendRow(new_item)
+        self._data_manager.add_data(new_datalist_item)
+
     def showPlot(self, plot):
         """
@@ -737 +754 @@
         if hasattr(self, "filesWidget"):
             self.filesWidget.displayData(plot)
-

src/sas/qtgui/Plotting/Plotter.py

@@ -145 +145 @@
 
         # Now add the legend with some customizations.
+
         self.legend = ax.legend(loc='upper right', shadow=True)
         if self.legend:

src/sas/qtgui/Plotting/Plotter2D.py

@@ -87 +87 @@
         self._item = item
 
-    def plot(self, data=None, marker=None):
+    def plot(self, data=None, marker=None, show_colorbar=True):
         """
         Plot 2D self._data
@@ -109 +109 @@
                       ymin=self.ymin, ymax=self.ymax,
                       cmap=self.cmap, zmin=zmin_2D_temp,
-                      zmax=zmax_2D_temp)
+                      zmax=zmax_2D_temp, show_colorbar=show_colorbar)
 
     def calculateDepth(self):
@@ -379 +379 @@
 
     def showPlot(self, data, qx_data, qy_data, xmin, xmax, ymin, ymax,
-                 zmin, zmax, label='data2D', cmap=DEFAULT_CMAP):
+                 zmin, zmax, label='data2D', cmap=DEFAULT_CMAP, show_colorbar=True):
         """
         Render and show the current data
@@ -457 +457 @@
             self.vmax = cb.vmax
 
+            if show_colorbar is False:
+                cb.remove()
+
         else:
             # clear the previous 2D from memory

src/sas/qtgui/Utilities/GuiUtils.py

@@ -144 +144 @@
     # custom open_path
     open_folder = custom_config.DEFAULT_OPEN_FOLDER
-    if open_folder != None and os.path.isdir(open_folder):
+    if open_folder is not None and os.path.isdir(open_folder):
         DEFAULT_OPEN_FOLDER = os.path.abspath(open_folder)
     else:
@@ -230 +230 @@
     dataDeletedSignal = QtCore.pyqtSignal(list)
 
+    # Send data to Data Operation Utility panel
+    sendDataToPanelSignal = QtCore.pyqtSignal(dict)
+
+    # Send result of Data Operation Utility panel to Data Explorer
+    updateModelFromDataOperationPanelSignal = QtCore.pyqtSignal(QtGui.QStandardItem, dict)
 
 def updateModelItemWithPlot(item, update_data, name=""):
@@ -430 +435 @@
     text += 'X_min = %s:  X_max = %s\n' % (xmin, max(data.x))
     text += 'Y_min = %s:  Y_max = %s\n' % (ymin, max(data.y))
-    if data.dy != None:
+    if data.dy is not None:
         text += 'dY_min = %s:  dY_max = %s\n' % (min(data.dy), max(data.dy))
     text += '\nData Points:\n'
     x_st = "X"
     for index in range(len(data.x)):
-        if data.dy != None and len(data.dy) > index:
+        if data.dy is not None and len(data.dy) > index:
             dy_val = data.dy[index]
         else:
             dy_val = 0.0
-        if data.dx != None and len(data.dx) > index:
+        if data.dx is not None and len(data.dx) > index:
             dx_val = data.dx[index]
         else:
             dx_val = 0.0
-        if data.dxl != None and len(data.dxl) > index:
+        if data.dxl is not None and len(data.dxl) > index:
             if index == 0:
                 x_st = "Xl"
             dx_val = data.dxl[index]
-        elif data.dxw != None and len(data.dxw) > index:
+        elif data.dxw is not None and len(data.dxw) > index:
             if index == 0:
                 x_st = "Xw"
@@ -486 +491 @@
         y_val = data.qy_data[index]
         i_val = data.data[index]
-        if data.err_data != None:
+        if data.err_data is not None:
             di_val = data.err_data[index]
-        if data.dqx_data != None:
+        if data.dqx_data is not None:
             dx_val = data.dqx_data[index]
-        if data.dqy_data != None:
+        if data.dqy_data is not None:
             dy_val = data.dqy_data[index]
 
@@ -523 +528 @@
                 has_errors = False
         if has_errors:
-            if data.dx != None and data.dx != []:
+            if data.dx is not None and data.dx != []:
                 out.write("<X>   <Y>   <dY>   <dX>\n")
             else:
@@ -532 +537 @@
         for i in range(len(data.x)):
             if has_errors:
-                if data.dx != None and data.dx != []:
+                if data.dx is not None and data.dx != []:
                     if  data.dx[i] != None:
                         out.write("%g  %g  %g  %g\n" % (data.x[i],

src/sas/qtgui/Utilities/UnitTesting/GuiUtilsTest.py

@@ -69 +69 @@
             'progressBarUpdateSignal',
             'activeGraphName',
+            'sendDataToPanelSignal',
+            'updateModelFromDataOperationPanelSignal'
         ]
 
@@ -412 +414 @@
         self.assertEqual(yscale, "log")
 
+
 class FormulaValidatorTest(unittest.TestCase):
     """ Test the formula validator """