Changeset 55d89f8 in sasview for src/sas

Timestamp:

Dec 6, 2016 8:35:10 AM (8 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:

64f1e93

Parents:

6d05e1d

Message:

Context menu 3D plotting

Location:

src/sas/qtgui

Files:

• DataExplorer.py (modified) (2 diffs)
• Plotter2D.py (modified) (13 diffs)
• PlotterBase.py (modified) (1 diff, 1 prop)
• UnitTesting/Plotter2DTest.py (added)
• run_tests.bat (modified) (1 diff)
• run_tests.sh (modified) (1 diff)

src/sas/qtgui/DataExplorer.py

@@ -768 +768 @@
             method_name='Plotter2D'
 
-        # TODO: Replace this with the proper MaskPlotPanel from plottools
         new_plot = globals()[method_name](self, quickplot=True)
         new_plot.data = data
@@ -782 +781 @@
     def quickData3DPlot(self):
         """
-        """
-        print "quickData3DPlot"
-        pass
+        Slowish 3D plot
+        """
+        index = self.treeView.selectedIndexes()[0]
+        model_item = self.model.itemFromIndex(self.data_proxy.mapToSource(index))
+        data = GuiUtils.dataFromItem(model_item)
+
+        new_plot = Plotter2D(self, quickplot=True, dimension=3)
+        new_plot.data = data
+        new_plot.plot(marker='o', linestyle='')
+
+        # Update the global plot counter
+        title = "Plot " + data.name
+        new_plot.setWindowTitle(title)
+
+        # Show the plot
+        new_plot.show()
 
     def showEditDataMask(self):

src/sas/qtgui/Plotter2D.py

@@ -11 +11 @@
 
 class Plotter2D(PlotterBase):
-    def __init__(self, parent=None, quickplot=False):
+    def __init__(self, parent=None, quickplot=False, dimension=2):
+        self.dimension = dimension
         super(Plotter2D, self).__init__(parent, quickplot=quickplot)
 
@@ -31 +32 @@
         self.zmax=data.zmax
         self.label=data.name
-        self.dimension=2
         self.xLabel="%s(%s)"%(data._xaxis, data._xunit)
         self.yLabel="%s(%s)"%(data._yaxis, data._yunit)
@@ -41 +41 @@
         """
         # create an axis object
-        ax = self.ax
+        zmin_2D_temp = self.zmin
+        zmax_2D_temp = self.zmax
+        if self.scale == 'log_{10}':
+            self.scale = 'linear'
+            if not self.zmin is None:
+                zmin_2D_temp = numpy.pow(10, self.zmin)
+            if not self.zmax is None:
+                zmax_2D_temp = numpy.pow(10, self.zmax)
+        else:
+            self.scale = 'log_{10}'
+            if not self.zmin is None:
+                # min log value: no log(negative)
+                if self.zmin <= 0:
+                    zmin_2D_temp = -32
+                else:
+                    zmin_2D_temp = numpy.log10(self.zmin)
+            if not self.zmax is None:
+                zmax_2D_temp = numpy.log10(self.zmax)
 
-        # graph properties
-        ax.set_xlabel(self.x_label)
-        ax.set_ylabel(self.y_label)
-        # Title only for regular charts
-        if not self.quickplot:
-            ax.set_title(label=self._title)
-
-        # Re-adjust colorbar
-        self.figure.subplots_adjust(left=0.2, right=.8, bottom=.2)
-
-        output = PlotUtilities.build_matrix(self._data.data, self.qx_data, self.qy_data)
-
-        im = ax.imshow(output,
-                       interpolation='nearest',
-                       origin='lower',
-                       vmin=self.zmin, vmax=self.zmax,
-                       cmap=self.cmap,
-                       extent=(self.xmin, self.xmax,
-                               self.ymin, self.ymax))
-
-        cbax = self.figure.add_axes([0.84, 0.2, 0.02, 0.7])
-        cb = self.figure.colorbar(im, cax=cbax)
-        cb.update_bruteforce(im)
-        cb.set_label('$' + self.scale + '$')
-
-        # Schedule the draw for the next time the event loop is idle.
-        self.canvas.draw_idle()
+        self.image(data=self.data.data, qx_data=self.qx_data,
+                   qy_data=self.qy_data, xmin=self.xmin,
+                   xmax=self.xmax,
+                   ymin=self.ymin, ymax=self.ymax,
+                   cmap=self.cmap, zmin=zmin_2D_temp,
+                   zmax=zmax_2D_temp)
 
     def contextMenuQuickPlot(self):
@@ -81 +77 @@
         self.actionCopyToClipboard = self.contextMenu.addAction("Copy to Clipboard")
         self.contextMenu.addSeparator()
-        self.actionToggleGrid = self.contextMenu.addAction("Toggle Grid On/Off")
-        self.contextMenu.addSeparator()
+        if self.dimension == 2:
+            self.actionToggleGrid = self.contextMenu.addAction("Toggle Grid On/Off")
+            self.contextMenu.addSeparator()
         self.actionChangeScale = self.contextMenu.addAction("Toggle Linear/Log Scale")
 
@@ -89 +86 @@
         self.actionPrintImage.triggered.connect(self.onImagePrint)
         self.actionCopyToClipboard.triggered.connect(self.onClipboardCopy)
-        self.actionToggleGrid.triggered.connect(self.onGridToggle)
+        if self.dimension == 2:
+            self.actionToggleGrid.triggered.connect(self.onGridToggle)
         self.actionChangeScale.triggered.connect(self.onToggleScale)
 
@@ -95 +93 @@
         """
         Toggle axis and replot image
-
         """
-        zmin_2D_temp = self.zmin
-        zmax_2D_temp = self.zmax
-        if self.scale == 'log_{10}':
-            self.scale = 'linear'
-            if not self.zmin is None:
-                zmin_2D_temp = math.pow(10, self.zmin)
-            if not self.zmax is None:
-                zmax_2D_temp = math.pow(10, self.zmax)
-        else:
-            self.scale = 'log_{10}'
-            if not self.zmin is None:
-                # min log value: no log(negative)
-                if self.zmin <= 0:
-                    zmin_2D_temp = -32
-                else:
-                    zmin_2D_temp = math.log10(self.zmin)
-            if not self.zmax is None:
-                zmax_2D_temp = math.log10(self.zmax)
-
-        self.image(data=self.data.data, qx_data=self.qx_data,
-                   qy_data=self.qy_data, xmin=self.xmin,
-                   xmax=self.xmax,
-                   ymin=self.ymin, ymax=self.ymax,
-                   cmap=self.cmap, zmin=zmin_2D_temp,
-                   zmax=zmax_2D_temp)
-
+        self.plot()
 
     def image(self, data, qx_data, qy_data, xmin, xmax, ymin, ymax,
@@ -129 +101 @@
         """
         Render the current data
-
         """
-        #self.data = data
         self.qx_data = qx_data
         self.qy_data = qy_data
@@ -140 +110 @@
         self.zmin = zmin
         self.zmax = zmax
-        #c = self.color(color)
         # If we don't have any data, skip.
         if data == None:
             return
         if data.ndim == 1:
-            #output = self._build_matrix()
             output = PlotUtilities.build_matrix(data, self.qx_data, self.qy_data)
         else:
@@ -157 +125 @@
                     zmin_temp = self.zmin_2D
                     output[output > 0] = numpy.log10(output[output > 0])
-                    #In log scale Negative values are not correct in general
-                    #output[output<=0] = math.log(numpy.min(output[output>0]))
                 elif self.zmin <= 0:
                     zmin_temp = self.zmin
@@ -166 +132 @@
                     zmin_temp = self.zmin
                     output[output > 0] = numpy.log10(output[output > 0])
-                    #In log scale Negative values are not correct in general
-                    #output[output<=0] = math.log(numpy.min(output[output>0]))
             except:
                 #Too many problems in 2D plot with scale
@@ -193 +157 @@
             self.figure.subplots_adjust(left=0.1, right=.8, bottom=.1)
 
-            X = self.x_bins[0:-1]
-            Y = self.y_bins[0:-1]
+            X = self._data.x_bins[0:-1]
+            Y = self._data.y_bins[0:-1]
             X, Y = numpy.meshgrid(X, Y)
 
@@ -200 +164 @@
                 # mpl >= 1.0.0
                 ax = self.figure.gca(projection='3d')
-                #ax.disable_mouse_rotation()
                 cbax = self.figure.add_axes([0.84, 0.1, 0.02, 0.8])
                 if len(X) > 60:
@@ -218 +181 @@
             im = ax.plot_surface(X, Y, output, rstride=1, cstride=1, cmap=cmap,
                                  linewidth=0, antialiased=False)
-            self.set_axis_off()
+            self.ax.set_axis_off()
 
         if cbax == None:

src/sas/qtgui/PlotterBase.py

@@ -52 +52 @@
         self.color=0
         self.symbol=0
-        self.dimension=1
         self.grid_on = False
         self.scale = 'linear'

src/sas/qtgui/run_tests.bat

@@ -12 +12 @@
 python -m UnitTesting.PlotHelperTest
 python -m UnitTesting.KiessigCalculatorTest
+python -m UnitTesting.PlotterTest
+python -m UnitTesting.Plotter2DTest
+python -m UnitTesting.ScalePropertiesTest

src/sas/qtgui/run_tests.sh

@@ -9 +9 @@
 python -m UnitTesting.PlotHelperTest
 python -m UnitTesting.SasviewLoggerTest
+python -m UnitTesting.PlotterTest
+python -m UnitTesting.Plotter2DTest
+python -m UnitTesting.ScalePropertiesTest
+python -m UnitTesting.KiessigCalculatorTest
 python -m UnitTesting.DensityCalculatorTest
-python -m UnitTesting.KiessigCalculatorTest