source: sasview/src/sas/perspectives/calculator/aperture_editor.py @ c22c5e3

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.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since c22c5e3 was 2f732b0, checked in by Doucet, Mathieu <doucetm@…>, 10 years ago

pylint fixes
  • Property mode set to 100644
File size: 16.2 KB
Line 
1
2import wx
3import sys
4from copy import deepcopy
5from sas.guiframe.utils import check_float
6
7_BOX_WIDTH = 60
8if sys.platform.count("win32") > 0:
9    _STATICBOX_WIDTH = 450
10    PANEL_WIDTH = 500
11    PANEL_HEIGHT = 290
12    FONT_VARIANT = 0
13else:
14    _STATICBOX_WIDTH = 480
15    PANEL_WIDTH = 530
16    PANEL_HEIGHT = 320
17    FONT_VARIANT = 1
18
19class ApertureDialog(wx.Dialog):
20    def __init__(self, parent=None, manager=None, aperture=None, *args, **kwds):
21        """
22        Dialog allows to enter values for aperture
23        """
24        kwds['size'] = (PANEL_WIDTH, PANEL_HEIGHT)
25        kwds['title'] = "Aperture Editor"
26        wx.Dialog.__init__(self, parent=parent, *args, **kwds)
27        self.parent = parent
28        self.manager = manager
29        self._aperture = aperture
30        self._reset_aperture = deepcopy(aperture)
31        self._notes = ""
32
33        #Attributes for panel
34        self.aperture_name_tcl = None
35        self.main_sizer = None
36        self.box_aperture = None
37        self.boxsizer_aperture = None
38        self.name_sizer = None
39        self.name_sizer = None
40        self.size_name_tcl = None
41        self.type_sizer = None
42        self.distance_sizer = None
43        self.size_name_sizer = None
44        self.aperture_size_unit_tcl = None
45        self.aperture_size_sizer = None
46        self.button_sizer = None
47        self.aperture_name_tcl = None
48        self.type_tcl = None
49        self.distance_tcl = None
50        self.distance_unit_tcl = None
51        self.x_aperture_size_tcl = None
52        self.y_aperture_size_tcl = None
53        self.z_aperture_size_tcl = None
54        self.bt_apply = None
55        self.bt_cancel = None
56        self.bt_close = None
57
58        self._do_layout()
59        self.set_values()
60
61    def _define_structure(self):
62        """
63        define initial sizer
64        """
65        self.main_sizer = wx.BoxSizer(wx.VERTICAL)
66        self.box_aperture = wx.StaticBox(self, -1, str("Aperture"))
67        self.boxsizer_aperture = wx.StaticBoxSizer(self.box_aperture, wx.VERTICAL)
68        self.name_sizer = wx.BoxSizer(wx.HORIZONTAL)
69        self.type_sizer = wx.BoxSizer(wx.HORIZONTAL)
70        self.distance_sizer = wx.BoxSizer(wx.HORIZONTAL)
71        self.size_name_sizer = wx.BoxSizer(wx.HORIZONTAL)
72        self.aperture_size_sizer = wx.BoxSizer(wx.HORIZONTAL)
73        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
74
75    def _layout_name(self):
76        """
77        Do the layout for aperture name related widgets
78        """
79        #Aperture name [string]
80        aperture_name_txt = wx.StaticText(self, -1, 'Aperture Name : ')
81        self.aperture_name_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH * 5, 20), style=0)
82        self.name_sizer.AddMany([(aperture_name_txt, 0, wx.LEFT | wx.RIGHT, 10),
83                                 (self.aperture_name_tcl, 0, wx.EXPAND)])
84    def _layout_type(self):
85        """
86        Do the  layout for aperture type  related widgets
87        """
88        #Aperture type [string]
89        type_txt = wx.StaticText(self, -1, 'Type: ')
90        self.type_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
91        self.type_sizer.AddMany([(type_txt, 0, wx.LEFT | wx.RIGHT, 10),
92                                 (self.type_tcl, 0, wx.LEFT, 20)])
93
94    def _layout_distance(self):
95        """
96        Do the  layout for aperture distance related widgets
97        """
98        #Aperture distance [float]
99        distance_txt = wx.StaticText(self, -1, 'Distance:')
100        self.distance_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20),
101                                        style=0)
102        distance_unit_txt = wx.StaticText(self, -1, 'Unit: ')
103        self.distance_unit_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
104        self.distance_sizer.AddMany([(distance_txt, 0, wx.LEFT | wx.RIGHT, 10),
105                                     (self.distance_tcl, 0, wx.LEFT, 10),
106                                     (distance_unit_txt, 0, wx.LEFT | wx.RIGHT, 10),
107                                     (self.distance_unit_tcl, 0, wx.EXPAND)])
108    def _layout_size_name(self):
109        """
110        Do the  layout for size name related widgets
111        """
112        # Size name [string]
113        size_name_txt = wx.StaticText(self, -1, 'Size Name : ')
114        self.size_name_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH * 5, 20), style=0)
115        self.size_name_sizer.AddMany([(size_name_txt, 0, wx.LEFT | wx.RIGHT, 10),
116                                      (self.size_name_tcl, 0, wx.EXPAND)])
117
118    def _layout_size(self):
119        """
120        Do the  layout for aperture size related widgets
121        """
122        #Aperture size [Vector]
123        aperture_size_txt = wx.StaticText(self, -1, 'Size:')
124        x_aperture_size_txt = wx.StaticText(self, -1, 'x = ')
125        self.x_aperture_size_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
126        y_aperture_size_txt = wx.StaticText(self, -1, 'y = ')
127        self.y_aperture_size_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
128        z_aperture_size_txt = wx.StaticText(self, -1, 'z = ')
129        self.z_aperture_size_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
130        aperture_size_unit_txt = wx.StaticText(self, -1, 'Unit: ')
131        self.aperture_size_unit_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
132        self.aperture_size_sizer.AddMany([(aperture_size_txt, 0, wx.LEFT | wx.RIGHT, 10),
133                                          (x_aperture_size_txt, 0, wx.LEFT, 17),
134                                          (self.x_aperture_size_tcl, 0, wx.RIGHT, 10),
135                                          (y_aperture_size_txt, 0, wx.EXPAND),
136                                          (self.y_aperture_size_tcl, 0, wx.RIGHT, 10),
137                                          (z_aperture_size_txt, 0, wx.EXPAND),
138                                          (self.z_aperture_size_tcl, 0, wx.RIGHT, 10),
139                                          (aperture_size_unit_txt, 0, wx.EXPAND),
140                                          (self.aperture_size_unit_tcl, 0, wx.RIGHT, 10)])
141
142    def _layout_button(self):
143        """
144        Do the layout for the button widgets
145        """
146        self.bt_apply = wx.Button(self, -1, 'Apply')
147        self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
148        self.bt_apply.SetToolTipString("Apply current changes to aperture.")
149        self.bt_cancel = wx.Button(self, -1, 'Cancel')
150        self.bt_cancel.SetToolTipString("Cancel current changes.")
151        self.bt_cancel.Bind(wx.EVT_BUTTON, self.on_click_cancel)
152        self.bt_close = wx.Button(self, wx.ID_CANCEL, 'Close')
153        self.bt_close.SetToolTipString("Close window.")
154        self.button_sizer.AddMany([(self.bt_apply, 0, wx.LEFT, 200),
155                                   (self.bt_cancel, 0, wx.LEFT, 10),
156                                   (self.bt_close, 0, wx.LEFT, 10)])
157
158    def _do_layout(self):#, data=None):
159        """
160        Draw the current panel
161        """
162        self._define_structure()
163        self._layout_name()
164        self._layout_type()
165        self._layout_distance()
166        self._layout_size_name()
167        self._layout_size()
168        self._layout_button()
169        self.boxsizer_aperture.AddMany([(self.name_sizer, 0, wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
170                                        (self.type_sizer, 0,
171                                         wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
172                                        (self.distance_sizer, 0,
173                                         wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
174                                        (self.size_name_sizer, 0,
175                                         wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
176                                        (self.aperture_size_sizer, 0,
177                                         wx.EXPAND | wx.TOP | wx.BOTTOM, 5)])
178        self.main_sizer.AddMany([(self.boxsizer_aperture, 0, wx.ALL, 10),
179                                 (self.button_sizer, 0,
180                                  wx.EXPAND | wx.TOP | wx.BOTTOM, 5)])
181        self.SetSizer(self.main_sizer)
182        self.SetAutoLayout(True)
183
184    def set_manager(self, manager):
185        """
186        Set manager of this window
187        """
188        self.manager = manager
189
190    def reset_aperture(self):
191        """
192        put the default value of the detector back to the current aperture
193        """
194        self._aperture.name = self._reset_aperture.name
195        self._aperture.type = self._reset_aperture.type
196        self._aperture.size_name = self._reset_aperture.size_name
197        self._aperture.size.x = self._reset_aperture.size.x
198        self._aperture.size.y = self._reset_aperture.size.y
199        self._aperture.size.z = self._reset_aperture.size.z
200        self._aperture.size_unit = self._reset_aperture.size_unit
201        self._aperture.distance = self._reset_aperture.distance
202        self._aperture.distance_unit = self._reset_aperture.distance_unit
203
204    def set_values(self):
205        """
206        take the aperture values of the current data and display them
207        through the panel
208        """
209        aperture = self._aperture
210        #Name
211        self.aperture_name_tcl.SetValue(str(aperture.name))
212        #Type
213        self.type_tcl.SetValue(str(aperture.type))
214        #distance
215        self.distance_tcl.SetValue(str(aperture.distance))
216        #distance unit
217        self.distance_unit_tcl.SetValue(str(aperture.distance_unit))
218        #Size name
219        self.size_name_tcl.SetValue(str(aperture.size_name))
220        #Aperture size as a vector
221        x, y, z = aperture.size.x, aperture.size.y, aperture.size.z
222        self.x_aperture_size_tcl.SetValue(str(x))
223        self.y_aperture_size_tcl.SetValue(str(y))
224        self.z_aperture_size_tcl.SetValue(str(z))
225        self.aperture_size_unit_tcl.SetValue(str(aperture.size_unit))
226
227    def get_aperture(self):
228        """
229        return the current aperture
230        """
231        return self._aperture
232
233    def get_notes(self):
234        """
235        return notes
236        """
237        return self._notes
238
239    def on_change_name(self):
240        """
241        Change name
242        """
243        #Change the name of the aperture
244        name = self.aperture_name_tcl.GetValue().lstrip().rstrip()
245        if name == "":
246            name = str(None)
247        if self._aperture.name != name:
248            self._notes += "Change sample 's "
249            self._notes += "name from %s to %s \n" % (self._aperture.name, name)
250            self._aperture.name = name
251
252    def on_change_type(self):
253        """
254        Change aperture type
255        """
256        #Change type
257        type_value = self.type_tcl.GetValue().lstrip().rstrip()
258        self._aperture.type = type_value
259        self._notes += " Change type from"
260        self._notes += " %s to %s \n" % (self._aperture.type, type_value)
261
262    def on_change_distance(self):
263        """
264        Change distance of the aperture
265        """
266        #Change distance
267        distance = self.distance_tcl.GetValue().lstrip().rstrip()
268        if distance == "" or distance == str(None):
269            distance = None
270            self._aperture.distance = distance
271        else:
272            if check_float(self.distance_tcl):
273                if self._aperture.distance != float(distance):
274                    self._notes += "Change distance from "
275                    self._notes += "%s to %s \n" % (self._aperture.distance, distance)
276                    self._aperture.distance = float(distance)
277            else:
278                self._notes += "Error: Expected a float for distance  "
279                self._notes += "won't changes distance from "
280                self._notes += "%s to %s" % (self._aperture.distance, distance)
281        #change the distance unit
282        unit = self.distance_unit_tcl.GetValue().lstrip().rstrip()
283        if self._aperture.distance_unit != unit:
284            self._notes += " Change distance 's unit from "
285            self._notes += "%s to %s" % (self._aperture.distance_unit, unit)
286
287    def on_change_size_name(self):
288        """
289        Change the size's name
290        """
291        #Change size name
292        size_name = self.size_name_tcl.GetValue().lstrip().rstrip()
293        self._aperture.size_name = size_name
294        self._notes += " Change size name from"
295        self._notes += " %s to %s \n" % (self._aperture.size_name, size_name)
296
297    def on_change_size(self):
298        """
299        Change aperture size
300        """
301        #Change x coordinate
302        x_aperture_size = self.x_aperture_size_tcl.GetValue().lstrip().rstrip()
303        if x_aperture_size == "" or x_aperture_size == str(None):
304            x_aperture_size = None
305        else:
306            if check_float(self.x_aperture_size_tcl):
307                if self._aperture.size.x != float(x_aperture_size):
308                    self._notes += "Change x of aperture size from "
309                    self._notes += "%s to %s \n" % (self._aperture.size.x, x_aperture_size)
310                    self._aperture.aperture_size.x = float(x_aperture_size)
311            else:
312                self._notes += "Error: Expected a"
313                self._notes += " float for the aperture size 's x "
314                self._notes += "won't changes x aperture size from "
315                self._notes += "%s to %s" % (self._aperture.size.x, x_aperture_size)
316        #Change y coordinate
317        y_aperture_size = self.y_aperture_size_tcl.GetValue().lstrip().rstrip()
318        if y_aperture_size == "" or y_aperture_size == str(None):
319            y_aperture_size = None
320            self._aperture.size.y = y_aperture_size
321        else:
322            if check_float(self.y_aperture_size_tcl):
323                if self._aperture.size.y != float(y_aperture_size):
324                    self._notes += "Change y of aperture size from "
325                    self._notes += "%s to %s \n" % (self._aperture.size.y, y_aperture_size)
326                    self._aperture.size.y = float(y_aperture_size)
327            else:
328                self._notes += "Error: Expected a float for the"
329                self._notes += " aperture size's y "
330                self._notes += "won't changes y aperture size from "
331                self._notes += "%s to %s" % (self._aperture.size.y, y_aperture_size)
332        #Change z coordinate
333        z_aperture_size = self.z_aperture_size_tcl.GetValue().lstrip().rstrip()
334        if z_aperture_size == "" or z_aperture_size == str(None):
335            z_aperture_size = None
336            self._aperture.size.z = z_aperture_size
337        else:
338            if check_float(self.z_aperture_size_tcl):
339                if self._aperture.size.z != float(z_aperture_size):
340                    self._notes += "Change z of aperture size from "
341                    self._notes += "%s to %s \n" % (self._aperture.size.z, z_aperture_size)
342                    self._aperture.size.z = float(z_aperture_size)
343            else:
344                self._notes += "Error: Expected a float for the offset 's x "
345                self._notes += "won't changes z aperture size from "
346                self._notes += "%s to %s" % (self._aperture.size.z, z_aperture_size)
347        #change the aperture center unit
348        unit = self.aperture_size_unit_tcl.GetValue().lstrip().rstrip()
349        if self._aperture.size_unit != unit:
350            self._notes += " Change aperture size's unit from "
351            self._notes += "%s to %s" % (self._aperture.size_unit, unit)
352            self._aperture.size_unit = unit
353
354    def on_click_apply(self, event):
355        """
356        Apply user values to the aperture
357        """
358        self.on_change_name()
359        self.on_change_type()
360        self.on_change_distance()
361        self.on_change_size_name()
362        self.on_change_size()
363        self.set_values()
364        if self.manager is not None:
365            self.manager.set_aperture(self._aperture)
366        if event is not None:
367            event.Skip()
368
369    def on_click_cancel(self, event):
370        """
371        reset the current aperture to its initial values
372        """
373        self.reset_aperture()
374        self.set_values()
375        if self.manager is not None:
376            self.manager.set_aperture(self._aperture)
377        if event is not None:
378            event.Skip()
Note: See TracBrowser for help on using the repository browser.