source: sasview/src/sas/sasgui/guiframe/local_perspectives/plotting/detector_dialog.py @ 7432acb

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 7432acb was 7432acb, checked in by andyfaff, 7 years ago

MAINT: search+replace '!= None' by 'is not None'

  • Property mode set to 100644
File size: 11.0 KB
Line 
1"""
2    Widget to display a 2D map of the detector
3"""
4import wx
5import sys
6from sas.sasgui.guiframe.utils import format_number
7from sas.sasgui.guiframe.events import StatusEvent
8from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as Canvas
9import matplotlib as mpl
10from matplotlib import pylab
11# FONT size
12if sys.platform.count("win32") > 0:
13    FONT_VARIANT = 0
14else:
15    FONT_VARIANT = 1
16
17DEFAULT_CMAP = pylab.cm.get_cmap('jet')
18
19class DetectorDialog(wx.Dialog):
20    """
21    Dialog box to let the user edit detector settings
22    """
23
24    def __init__(self, parent, id=1, base=None, dpi=None,
25                 cmap=DEFAULT_CMAP, reset_zmin_ctl=None,
26                 reset_zmax_ctl=None, *args, **kwds):
27        """
28        """
29        kwds["style"] = wx.DEFAULT_DIALOG_STYLE
30        wx.Dialog.__init__(self, parent, id=1, *args, **kwds)
31        self.SetWindowVariant(variant=FONT_VARIANT)
32        self.parent = base
33        self.dpi = dpi
34        self.cmap = cmap
35        self.reset_zmin_ctl = reset_zmin_ctl
36        self.reset_zmax_ctl = reset_zmax_ctl
37        self.label_xnpts = wx.StaticText(self, -1, "Detector width in pixels")
38        self.label_ynpts = wx.StaticText(self, -1, "Detector Height in pixels")
39        self.label_qmax = wx.StaticText(self, -1, "Q max")
40        self.label_zmin = wx.StaticText(self, -1,
41                                        "Min amplitude for color map (optional)")
42        self.label_zmax = wx.StaticText(self, -1,
43                                        "Max amplitude for color map (optional)")
44        self.label_beam = wx.StaticText(self, -1,
45                                        "Beam stop radius in units of q")
46        self.xnpts_ctl = wx.StaticText(self, -1, "")
47        self.ynpts_ctl = wx.StaticText(self, -1, "")
48        self.qmax_ctl = wx.StaticText(self, -1, "")
49        self.beam_ctl = wx.StaticText(self, -1, "")
50        self.zmin_ctl = wx.TextCtrl(self, -1, size=(60, 20))
51        self.zmin_ctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
52        self.zmax_ctl = wx.TextCtrl(self, -1, size=(60, 20))
53        self.zmax_ctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
54        self.static_line_3 = wx.StaticLine(self, -1)
55        self.button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
56        self.button_reset = wx.Button(self, wx.NewId(), "Reset")
57        self.Bind(wx.EVT_BUTTON, self.resetValues, self.button_reset)
58        self.button_ok = wx.Button(self, wx.ID_OK, "OK")
59        self.Bind(wx.EVT_BUTTON, self.checkValues, self.button_ok)
60        self.__set_properties()
61        self.__do_layout()
62        self.Fit()
63
64    class Event(object):
65        """
66        """
67        xnpts = 0
68        ynpts = 0
69        qpax = 0
70        beam = 0
71        zmin = 0
72        zmax = 0
73        cmap = None
74        sym4 = False
75
76    def onSetFocus(self, event):
77        """
78        Highlight the txtcrtl
79        """
80        # Get a handle to the TextCtrl
81        widget = event.GetEventObject()
82        # Select the whole control, after this event resolves
83        wx.CallAfter(widget.SetSelection, -1, -1)
84
85    def resetValues(self, event):
86        """
87        reset detector info
88        """
89        try:
90            zmin = self.reset_zmin_ctl
91            zmax = self.reset_zmax_ctl
92            if zmin is None:
93                zmin = ""
94            if zmax is None:
95                zmax = ""
96            self.zmin_ctl.SetValue(str(zmin))
97            self.zmax_ctl.SetValue(str(zmax))
98            self.cmap = DEFAULT_CMAP
99            self.cmap_selector.SetStringSelection("jet")
100            self._on_select_cmap(event=None)
101        except:
102            msg = "error occurs while resetting Detector: %s" % sys.exc_value
103            wx.PostEvent(self.parent, StatusEvent(status=msg))
104
105    def checkValues(self, event):
106        """
107        Check the valitidity of zmin and zmax value
108        zmax should be a float and zmin less than zmax
109        """
110        flag = True
111        try:
112            value = self.zmin_ctl.GetValue()
113            self.zmin_ctl.SetBackgroundColour(wx.WHITE)
114            self.zmin_ctl.Refresh()
115        except:
116            flag = False
117            wx.PostEvent(self.parent, StatusEvent(status="Enter float value"))
118            self.zmin_ctl.SetBackgroundColour("pink")
119            self.zmin_ctl.Refresh()
120        try:
121            value = self.zmax_ctl.GetValue()
122            if value and float(value) == 0.0:
123                flag = False
124                wx.PostEvent(self.parent,
125                             StatusEvent(status="Enter number greater than zero"))
126                self.zmax_ctl.SetBackgroundColour("pink")
127                self.zmax_ctl.Refresh()
128            else:
129                self.zmax_ctl.SetBackgroundColour(wx.WHITE)
130                self.zmax_ctl.Refresh()
131        except:
132            flag = False
133            wx.PostEvent(self.parent, StatusEvent(status="Enter Integer value"))
134            self.zmax_ctl.SetBackgroundColour("pink")
135            self.zmax_ctl.Refresh()
136        if flag:
137            event.Skip(True)
138
139    def setContent(self, xnpts, ynpts, qmax, beam,
140                   zmin=None, zmax=None, sym=False):
141        """
142        received value and displayed them
143
144        :param xnpts: the number of point of the x_bins of data
145        :param ynpts: the number of point of the y_bins of data
146        :param qmax: the maxmimum value of data pixel
147        :param beam: the radius of the beam
148        :param zmin:  the value to get the minimum color
149        :param zmax:  the value to get the maximum color
150        :param sym:
151
152        """
153        self.xnpts_ctl.SetLabel(str(format_number(xnpts)))
154        self.ynpts_ctl.SetLabel(str(format_number(ynpts)))
155        self.qmax_ctl.SetLabel(str(format_number(qmax)))
156        self.beam_ctl.SetLabel(str(format_number(beam)))
157        if zmin is not None:
158            self.zmin_ctl.SetValue(str(format_number(zmin)))
159        if zmax is not None:
160            self.zmax_ctl.SetValue(str(format_number(zmax)))
161
162    def getContent(self):
163        """
164        return event containing value to reset the detector of a given data
165        """
166        event = self.Event()
167        t_min = self.zmin_ctl.GetValue()
168        t_max = self.zmax_ctl.GetValue()
169        v_min = None
170        v_max = None
171        if len(t_min.lstrip()) > 0:
172            try:
173                v_min = float(t_min)
174            except:
175                v_min = None
176        if len(t_max.lstrip()) > 0:
177            try:
178                v_max = float(t_max)
179            except:
180                v_max = None
181        event.zmin = v_min
182        event.zmax = v_max
183        event.cmap = self.cmap
184        return event
185
186    def __set_properties(self):
187        """
188        set proprieties of the dialog window
189        """
190        self.SetTitle("2D Color Map")
191        self.SetSize((600, 595))
192
193    def __do_layout(self):
194        """
195        fill the dialog window .
196        """
197        sizer_main = wx.BoxSizer(wx.VERTICAL)
198        sizer_button = wx.BoxSizer(wx.HORIZONTAL)
199        sizer_params = wx.GridBagSizer(5, 5)
200        sizer_colormap = wx.BoxSizer(wx.VERTICAL)
201        sizer_selection = wx.BoxSizer(wx.HORIZONTAL)
202
203        iy = 0
204        sizer_params.Add(self.label_xnpts, (iy, 0), (1, 1),
205                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
206        sizer_params.Add(self.xnpts_ctl, (iy, 1), (1, 1),
207                         wx.EXPAND | wx.ADJUST_MINSIZE, 0)
208        iy += 1
209        sizer_params.Add(self.label_ynpts, (iy, 0), (1, 1),
210                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
211        sizer_params.Add(self.ynpts_ctl, (iy, 1), (1, 1),
212                         wx.EXPAND | wx.ADJUST_MINSIZE, 0)
213        iy += 1
214        sizer_params.Add(self.label_qmax, (iy, 0), (1, 1),
215                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
216        sizer_params.Add(self.qmax_ctl, (iy, 1), (1, 1),
217                         wx.EXPAND | wx.ADJUST_MINSIZE, 0)
218        iy += 1
219        sizer_params.Add(self.label_beam, (iy, 0), (1, 1),
220                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
221        sizer_params.Add(self.beam_ctl, (iy, 1), (1, 1),
222                         wx.EXPAND | wx.ADJUST_MINSIZE, 0)
223        iy += 1
224        sizer_params.Add(self.label_zmin, (iy, 0), (1, 1),
225                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
226        sizer_params.Add(self.zmin_ctl, (iy, 1), (1, 1),
227                         wx.EXPAND | wx.ADJUST_MINSIZE, 0)
228        iy += 1
229        sizer_params.Add(self.label_zmax, (iy, 0), (1, 1),
230                         wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
231        sizer_params.Add(self.zmax_ctl, (iy, 1), (1, 1),
232                         wx.EXPAND | wx.ADJUST_MINSIZE, 0)
233        iy += 1
234        self.fig = mpl.figure.Figure(dpi=self.dpi, figsize=(4, 1))
235        self.ax1 = self.fig.add_axes([0.05, 0.65, 0.9, 0.15])
236        self.norm = mpl.colors.Normalize(vmin=0, vmax=100)
237        self.cb1 = mpl.colorbar.ColorbarBase(self.ax1, cmap=self.cmap,
238                                             norm=self.norm,
239                                             orientation='horizontal')
240        self.cb1.set_label('Detector Colors')
241        self.canvas = Canvas(self, -1, self.fig)
242        sizer_colormap.Add(self.canvas, 0, wx.LEFT | wx.EXPAND, 5)
243        self.cmap_selector = wx.ComboBox(self, -1)
244        self.cmap_selector.SetValue(str(self.cmap.name))
245        maps = sorted(m for m in pylab.cm.datad if not m.endswith("_r"))
246
247        for i, m in enumerate(maps):
248            self.cmap_selector.Append(str(m), pylab.get_cmap(m))
249
250        wx.EVT_COMBOBOX(self.cmap_selector, -1, self._on_select_cmap)
251        sizer_selection.Add(wx.StaticText(self, -1, "Select Cmap: "), 0,
252                            wx.LEFT | wx.ADJUST_MINSIZE, 5)
253        sizer_selection.Add(self.cmap_selector, 0, wx.EXPAND | wx.ALL, 10)
254        sizer_main.Add(sizer_params, 0, wx.EXPAND | wx.ALL, 5)
255        sizer_main.Add(sizer_selection, 0, wx.EXPAND | wx.ALL, 5)
256        note = "   Note: This is one time option. " + \
257               "It will be reset on updating the image."
258        note_txt = wx.StaticText(self, -1, note)
259        sizer_main.Add(note_txt, 0, wx.EXPAND | wx.ALL, 5)
260        sizer_main.Add(sizer_colormap, 1, wx.EXPAND | wx.ALL, 5)
261        sizer_main.Add(self.static_line_3, 0, wx.EXPAND, 0)
262        sizer_button.Add(self.button_reset, 0, wx.LEFT | wx.ADJUST_MINSIZE, 100)
263        sizer_button.Add(self.button_ok, 0, wx.LEFT | wx.ADJUST_MINSIZE, 10)
264        sizer_button.Add(self.button_cancel, 0,
265                         wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10)
266        sizer_main.Add(sizer_button, 0, wx.EXPAND | wx.BOTTOM | wx.TOP, 10)
267        self.SetAutoLayout(True)
268        self.SetSizer(sizer_main)
269        self.Layout()
270        self.Centre()
271        # end wxGlade
272
273    def _on_select_cmap(self, event):
274        """
275        display a new cmap
276        """
277        cmap_name = self.cmap_selector.GetCurrentSelection()
278        current_cmap = self.cmap_selector.GetClientData(cmap_name)
279        self.cmap = current_cmap
280        self.cb1 = mpl.colorbar.ColorbarBase(self.ax1, cmap=self.cmap,
281                                             norm=self.norm, orientation='horizontal')
282        self.canvas.draw()
Note: See TracBrowser for help on using the repository browser.