Changeset d48cc19 in sasview for src/sas/qtgui/Perspectives/Fitting/FittingWidget.py

Timestamp:

May 2, 2017 9:04:48 AM (7 years ago)

Author:

Piotr Rozyczko <rozyczko@…>

Branches:

ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc

Children:

02ddfb4

Parents:

0a6e097

Message:

Compute/Show? Plot button logic: SASVIEW-271
Unit tests for plotting in fitting: SASVIEW-501

File:

• src/sas/qtgui/Perspectives/Fitting/FittingWidget.py (modified) (10 diffs)

src/sas/qtgui/Perspectives/Fitting/FittingWidget.py

@@ -74 +74 @@
         self.weighting = 0
 
+        # Data for chosen model
+        self.model_data = None
+
         # Which tab is this widget displayed in?
         self.tab_id = id
@@ -250 +253 @@
         """
         self.cmdFit.setEnabled(False)
-        self.cmdPlot.setEnabled(True)
+        self.cmdPlot.setEnabled(False)
         self.options_widget.cmdComputePoints.setVisible(False) # probably redundant
         self.chkPolydispersity.setEnabled(True)
@@ -305 +308 @@
         self.has_error_column = False
 
+        # Set enablement on calculate/plot
+        self.cmdPlot.setEnabled(True)
+
         # SasModel -> QModel
         self.SASModelToQModel(model)
 
         if self.data_is_loaded:
+            self.cmdPlot.setText("Show Plot")
             self.calculateQGridForModel()
         else:
+            self.cmdPlot.setText("Calculate")
             # Create default datasets if no data passed
             self.createDefaultDataset()
@@ -569 +577 @@
         Plot the current set of data
         """
+        # Regardless of previous state, this should now be `plot show` functionality only
+        self.cmdPlot.setText("Show Plot")
+        self.recalculatePlotData()
+        self.showPlot()
+
+    def recalculatePlotData(self):
+        """
+        Generate a new dataset for model
+        """
         if not self.data_is_loaded:
             self.createDefaultDataset()
         self.calculateQGridForModel()
+
+    def showPlot(self):
+        """
+        Show the current plot in MPL
+        """
+        # Show the chart if ready
+        data_to_show = self.data if self.data_is_loaded else self.model_data
+        if data_to_show is not None:
+            self.communicate.plotRequestedSignal.emit([data_to_show])
 
     def onOptionsUpdate(self):
@@ -582 +608 @@
         self.lblMinRangeDef.setText(str(self.q_range_min))
         self.lblMaxRangeDef.setText(str(self.q_range_max))
-        self.onPlot()
+        self.recalculatePlotData()
 
     def setDefaultStructureCombo(self):
@@ -766 +792 @@
         # Force the chart update when actual parameters changed
         if model_column == 1:
-            self.onPlot()
+            self.recalculatePlotData()
 
     def checkboxSelected(self, item):
@@ -891 +917 @@
         Plot the current 1D data
         """
-        fitted_plot = self.logic.new1DPlot(return_data, self.tab_id)
-        self.calculateResiduals(fitted_plot)
+        fitted_data = self.logic.new1DPlot(return_data, self.tab_id)
+        self.calculateResiduals(fitted_data)
+        self.model_data = fitted_data
 
     def complete2D(self, return_data):
@@ -900 +927 @@
         fitted_data = self.logic.new2DPlot(return_data)
         self.calculateResiduals(fitted_data)
+        self.model_data = fitted_data
 
     def calculateResiduals(self, fitted_data):
@@ -914 +942 @@
         self.lblChi2Value.setText(chi2_repr)
 
+        self.communicate.plotUpdateSignal.emit([fitted_data])
+
         # Plot residuals if actual data
         if self.data_is_loaded:
@@ -920 +950 @@
             self.createNewIndex(residuals_plot)
             self.communicate.plotUpdateSignal.emit([residuals_plot])
-
-        self.communicate.plotUpdateSignal.emit([fitted_data])
 
     def calcException(self, etype, value, tb):