source: sasview/src/sas/qtgui/ColorMap.py @ 3bdbfcc

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc
Last change on this file since 3bdbfcc was 3bdbfcc, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago

Reimplementation of the slicer functionality

  • Property mode set to 100644
File size: 8.6 KB
Line 
1"""
2Allows users to change the range of the current graph
3"""
4from PyQt4 import QtGui
5from PyQt4 import QtCore
6import path_prepare
7
8import matplotlib as mpl
9from matplotlib import pylab
10import numpy
11
12from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
13from sas.sasgui.guiframe.dataFitting import Data2D
14from sas.qtgui.GuiUtils import formatNumber
15from rangeSlider import RangeSlider
16
17DEFAULT_MAP = 'jet'
18
19# Local UI
20from sas.qtgui.UI.ColorMapUI import Ui_ColorMapUI
21
22class ColorMap(QtGui.QDialog, Ui_ColorMapUI):
23    apply_signal = QtCore.pyqtSignal(tuple, str)
24    def __init__(self, parent=None, cmap=None, vmin=0.0, vmax=100.0, data=None):
25        super(ColorMap, self).__init__()
26
27        self.setupUi(self)
28        assert(isinstance(data, Data2D))
29
30        self.data = data
31        self._cmap_orig = self._cmap = cmap if cmap is not None else DEFAULT_MAP
32        self.all_maps = [m for m in pylab.cm.datad]
33        self.maps = sorted(m for m in self.all_maps if not m.endswith("_r"))
34        self.rmaps = sorted(set(self.all_maps) - set(self.maps))
35
36        self.vmin = self.vmin_orig = vmin
37        self.vmax = self.vmax_orig = vmax
38
39        # Initialize detector labels
40        self.initDetectorData()
41
42        # Initialize the combo box
43        self.initMapCombobox()
44
45        self.initRangeSlider()
46
47        # Add the color map component
48        self.initColorMap()
49
50        # Initialize validators on amplitude textboxes
51        validator_min = QtGui.QDoubleValidator(self.txtMinAmplitude)
52        validator_min.setNotation(0)
53        self.txtMinAmplitude.setValidator(validator_min)
54        validator_max = QtGui.QDoubleValidator(self.txtMaxAmplitude)
55        validator_max.setNotation(0)
56        self.txtMaxAmplitude.setValidator(validator_max)
57
58        # Set the initial amplitudes
59        self.txtMinAmplitude.setText(formatNumber(self.vmin))
60        self.txtMaxAmplitude.setText(formatNumber(self.vmax))
61
62        # Enforce constant size on the widget
63        self.setFixedSize(self.minimumSizeHint())
64
65        # Handle combobox changes
66        self.cbColorMap.currentIndexChanged.connect(self.onMapIndexChange)
67
68        # Handle checkbox changes
69        self.chkReverse.stateChanged.connect(self.onColorMapReversed)
70
71        # Handle the Reset button click
72        self.buttonBox.button(QtGui.QDialogButtonBox.Reset).clicked.connect(self.onReset)
73
74        # Handle the Apply button click
75        self.buttonBox.button(QtGui.QDialogButtonBox.Apply).clicked.connect(self.onApply)
76
77        # Handle the amplitude setup
78        self.txtMinAmplitude.editingFinished.connect(self.onAmplitudeChange)
79        self.txtMaxAmplitude.editingFinished.connect(self.onAmplitudeChange)
80
81    def cmap(self):
82        """
83        Getter for the color map
84        """
85        return self._cmap
86
87    def norm(self):
88        """
89        Getter for the color map norm
90        """
91        return (self._norm.vmin, self._norm.vmax)
92
93    def onReset(self):
94        """
95        Respond to the Reset button click
96        """
97        # Go back to original settings
98        self._cmap = self._cmap_orig
99        self.vmin = self.vmin_orig
100        self.vmax = self.vmax_orig
101        self._norm = mpl.colors.Normalize(vmin=self.vmin, vmax=self.vmax)
102        self.txtMinAmplitude.setText(formatNumber(self.vmin))
103        self.txtMaxAmplitude.setText(formatNumber(self.vmax))
104        self.initMapCombobox()
105        self.slider.setMinimum(self.vmin)
106        self.slider.setMaximum(self.vmax)
107        self.slider.setLowValue(self.vmin)
108        self.slider.setHighValue(self.vmax)
109        # Redraw the widget
110        self.redrawColorBar()
111        self.canvas.draw()
112
113    def initDetectorData(self):
114        """
115        Fill out the Detector labels
116        """
117        xnpts = len(self.data.x_bins)
118        ynpts = len(self.data.y_bins)
119        self.lblWidth.setText(formatNumber(xnpts))
120        self.lblHeight.setText(formatNumber(ynpts))
121        xmax = max(self.data.xmin, self.data.xmax)
122        ymax = max(self.data.ymin, self.data.ymax)
123        qmax = numpy.sqrt(numpy.power(xmax, 2) + numpy.power(ymax, 2))
124        self.lblQmax.setText(formatNumber(qmax))
125        self.lblStopRadius.setText(formatNumber(self.data.xmin))
126
127    def initMapCombobox(self):
128        """
129        Fill out the combo box with all available color maps
130        """
131        if self._cmap in self.rmaps:
132            maps = self.rmaps
133            # Assure correct state of the checkbox
134            self.chkReverse.setChecked(True)
135        else:
136            maps = self.maps
137            # Assure correct state of the checkbox
138            self.chkReverse.setChecked(False)
139
140        self.cbColorMap.addItems(maps)
141        # Set the default/passed map
142        self.cbColorMap.setCurrentIndex(self.cbColorMap.findText(self._cmap))
143
144    def initRangeSlider(self):
145        """
146        Create and display the double slider for data range mapping.
147        """
148        self.slider = RangeSlider()
149        self.slider.setMinimum(self.vmin)
150        self.slider.setMaximum(self.vmax)
151        self.slider.setLowValue(self.vmin)
152        self.slider.setHighValue(self.vmax)
153        self.slider.setOrientation(QtCore.Qt.Horizontal)
154
155        self.slider_label = QtGui.QLabel()
156        self.slider_label.setText("Drag the sliders to adjust color range.")
157
158        def set_vmin(value):
159            self.vmin = value
160            self.txtMinAmplitude.setText(str(value))
161            self.updateMap()
162        def set_vmax(value):
163            self.vmax = value
164            self.txtMaxAmplitude.setText(str(value))
165            self.updateMap()
166
167        self.slider.lowValueChanged.connect(set_vmin)
168        self.slider.highValueChanged.connect(set_vmin)
169
170    def updateMap(self):
171        self._norm = mpl.colors.Normalize(vmin=self.vmin, vmax=self.vmax)
172        self.redrawColorBar()
173        self.canvas.draw()
174
175    def initColorMap(self):
176        """
177        Prepare and display the color map
178        """
179        self.fig = mpl.figure.Figure(figsize=(4, 1))
180        self.ax1 = self.fig.add_axes([0.05, 0.65, 0.9, 0.15])
181
182        self._norm = mpl.colors.Normalize(vmin=self.vmin, vmax=self.vmax)
183        self.redrawColorBar()
184        self.canvas = FigureCanvas(self.fig)
185
186        layout = QtGui.QVBoxLayout()
187        layout.addWidget(self.slider_label)
188        layout.addWidget(self.slider)
189        layout.addWidget(self.canvas)
190
191        self.widget.setLayout(layout)
192
193    def onMapIndexChange(self, index):
194        """
195        Respond to the color map change event
196        """
197        new_map = str(self.cbColorMap.itemText(index))
198        self._cmap = new_map
199        self.redrawColorBar()
200        self.canvas.draw()
201
202    def redrawColorBar(self):
203        """
204        Call ColorbarBase with current values, effectively redrawing the widget
205        """
206        self.cb = mpl.colorbar.ColorbarBase(self.ax1, cmap=self._cmap,
207                                            norm=self._norm,
208                                            orientation='horizontal')
209        self.cb.set_label('Detector Colors')
210
211    def onColorMapReversed(self, isChecked):
212        """
213        Respond to ticking/unticking the color map reverse checkbox
214        """
215        current_map = str(self.cbColorMap.currentText())
216        if isChecked:
217            # Add "_r" to map name for the reversed version
218            new_map = current_map + "_r"
219            maps = self.rmaps
220            # Assure the reversed map exists.
221            if new_map not in maps:
222                new_map = maps[0]
223        else:
224            new_map = current_map[:-2] # "_r" = last two chars
225            maps = self.maps
226            # Base map for the reversed map should ALWAYS exist,
227            # but let's be paranoid here
228            if new_map not in maps:
229                new_map = maps[0]
230
231        self._cmap = new_map
232        # Clear the content of the combobox.
233        # Needs signal blocking, or else onMapIndexChange() spoils it all
234        self.cbColorMap.blockSignals(True)
235        self.cbColorMap.clear()
236        # Add the new set of maps
237        self.cbColorMap.addItems(maps)
238        # Remove the signal block before the proper index set
239        self.cbColorMap.blockSignals(False)
240        self.cbColorMap.setCurrentIndex(self.cbColorMap.findText(new_map))
241
242    def onAmplitudeChange(self):
243        """
244        Respond to editing the amplitude fields
245        """
246        min_amp = self.vmin
247        max_amp = self.vmax
248
249        try:
250            min_amp = float(self.txtMinAmplitude.text())
251        except ValueError:
252            pass
253        try:
254            max_amp = float(self.txtMaxAmplitude.text())
255        except ValueError:
256            pass
257
258        self._norm = mpl.colors.Normalize(vmin=min_amp, vmax=max_amp)
259        self.redrawColorBar()
260        self.canvas.draw()
Note: See TracBrowser for help on using the repository browser.