source: sasview/src/sas/sasgui/perspectives/calculator/detector_editor.py @ fa81e94

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 fa81e94 was fa81e94, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago

Initial commit of the P(r) inversion perspective.
Code merged from Jeff Krzywon's ESS_GUI_Pr branch.
Also, minor 2to3 mods to sascalc/sasgui to enble error free setup.

  • Property mode set to 100755
File size: 39.5 KB
Line 
1
2
3import wx
4import sys
5from copy import deepcopy
6from sas.sascalc.dataloader.data_info import Detector
7from sas.sasgui.guiframe.utils import check_float
8
9_BOX_WIDTH = 60
10if sys.platform.count("win32") > 0:
11    _STATICBOX_WIDTH = 465
12    PANEL_WIDTH = 500
13    PANEL_HEIGHT = 450
14    FONT_VARIANT = 0
15else:
16    _STATICBOX_WIDTH = 480
17    PANEL_WIDTH = 550
18    PANEL_HEIGHT = 480
19    FONT_VARIANT = 1
20
21class DetectorDialog(wx.Dialog):
22    def __init__(self, parent=None, manager=None, detector=None,
23                        title="Detector Editor",
24                        size=(PANEL_WIDTH, PANEL_HEIGHT)):
25        wx.Dialog.__init__(self, parent=parent, id=id, title=title, size=size)
26        try:
27            self.parent = parent
28            self.manager = manager
29            self._detector = detector
30
31            self._reset_detector = deepcopy(detector)
32            self._notes = ""
33            self._description = "Edit Detector"
34            self._do_layout()
35            self.set_values()
36        except:
37            print("error", sys.exc_info()[1])
38
39    def _define_structure(self):
40        """
41            define initial sizer
42        """
43        self.main_sizer = wx.BoxSizer(wx.VERTICAL)
44        self.box_detector = wx.StaticBox(self, -1, str("Edit Selected Detector"))
45        self.boxsizer_detector = wx.StaticBoxSizer(self.box_detector,
46                                                    wx.VERTICAL)
47        detector_box = wx.StaticBox(self, -1, "Edit Number of Detectors")
48        self.detector_sizer = wx.StaticBoxSizer(detector_box, wx.VERTICAL)
49        self.detector_sizer.SetMinSize((_STATICBOX_WIDTH, -1))
50        self.detector_button_sizer = wx.BoxSizer(wx.HORIZONTAL)
51        self.detector_hint_sizer = wx.BoxSizer(wx.HORIZONTAL)
52
53        self.detector_button_sizer = wx.BoxSizer(wx.HORIZONTAL)
54        self.detector_hint_sizer = wx.BoxSizer(wx.HORIZONTAL)
55        self.instrument_sizer = wx.BoxSizer(wx.HORIZONTAL)
56        self.distance_sizer = wx.BoxSizer(wx.HORIZONTAL)
57        self.offset_sizer = wx.BoxSizer(wx.HORIZONTAL)
58        self.orientation_sizer = wx.BoxSizer(wx.HORIZONTAL)
59        self.beam_sizer = wx.BoxSizer(wx.HORIZONTAL)
60        self.pixel_sizer = wx.BoxSizer(wx.HORIZONTAL)
61        self.slit_sizer = wx.BoxSizer(wx.HORIZONTAL)
62        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
63
64    def _layout_detector(self):
65        """
66            Do the layout for detector related widgets
67        """
68        detector_name_txt = wx.StaticText(self, -1, "Detector:")
69        hint_detector_txt = 'Current available detector.'
70        detector_name_txt.SetToolTipString(hint_detector_txt)
71        self.detector_cbox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
72        hint_detector_name_txt = 'Name of detectors.'
73        self.detector_cbox.SetToolTipString(hint_detector_name_txt)
74
75        self.bt_add_detector = wx.Button(self, -1, "Add")
76        self.bt_add_detector.SetToolTipString("Add data's detector.")
77        self.bt_add_detector.Bind(wx.EVT_BUTTON, self.add_detector)
78
79        self.bt_remove_detector = wx.Button(self, -1, "Remove")
80        self.bt_remove_detector.SetToolTipString("Remove data's detector.")
81        self.bt_remove_detector.Bind(wx.EVT_BUTTON, self.remove_detector)
82
83        self.detector_button_sizer.AddMany([(detector_name_txt, 0, wx.LEFT, 15),
84                                     (self.detector_cbox, 0, wx.LEFT, 5),
85                                     (self.bt_add_detector, 0, wx.LEFT, 10),
86                                     (self.bt_remove_detector, 0, wx.LEFT, 5)])
87        detector_hint_txt = 'No detector is available for this data.'
88        self.detector_txt = wx.StaticText(self, -1, detector_hint_txt)
89        self.detector_hint_sizer.Add(self.detector_txt, 0, wx.LEFT, 10)
90        self.detector_sizer.AddMany([(self.detector_button_sizer, 0,
91                                       wx.ALL, 10),
92                                     (self.detector_hint_sizer, 0, wx.ALL, 10)])
93
94        self.fill_detector_combox()
95        self.enable_detector()
96
97    def _layout_instrument_sizer(self):
98        """
99            Do the layout for instrument related widgets
100        """
101        #Instrument
102        instrument_name_txt = wx.StaticText(self, -1, 'Instrument Name : ')
103        self.instrument_name_tcl = wx.TextCtrl(self, -1,
104                                             size=(_BOX_WIDTH * 5, 20), style=0)
105        self.instrument_sizer.AddMany([(instrument_name_txt, 0,
106                                        wx.LEFT | wx.RIGHT, 10),
107                                    (self.instrument_name_tcl, 0, wx.EXPAND)])
108    def _layout_distance(self):
109        """
110            Do the  layout for distance related widgets
111        """
112        distance_txt = wx.StaticText(self, -1, 'Sample to Detector Distance : ')
113        self.distance_tcl = wx.TextCtrl(self, -1,
114                                         size=(_BOX_WIDTH, 20), style=0)
115        distance_unit_txt = wx.StaticText(self, -1, 'Unit: ')
116        self.distance_unit_tcl = wx.TextCtrl(self, -1,
117                                              size=(_BOX_WIDTH, 20), style=0)
118        self.distance_sizer.AddMany([(distance_txt, 0, wx.LEFT | wx.RIGHT, 10),
119                                     (self.distance_tcl, 0, wx.RIGHT, 10),
120                                     (distance_unit_txt, 0, wx.EXPAND),
121                                     (self.distance_unit_tcl, 0, wx.RIGHT, 10)])
122
123    def _layout_offset(self):
124        """
125            Do the  layout for offset related widgets
126        """
127        #Offset
128        offset_txt = wx.StaticText(self, -1, 'Offset:')
129        x_offset_txt = wx.StaticText(self, -1, 'x = ')
130        self.x_offset_tcl = wx.TextCtrl(self, -1,
131                                        size=(_BOX_WIDTH, 20), style=0)
132        y_offset_txt = wx.StaticText(self, -1, 'y = ')
133        self.y_offset_tcl = wx.TextCtrl(self, -1,
134                                         size=(_BOX_WIDTH, 20), style=0)
135        z_offset_txt = wx.StaticText(self, -1, 'z = ')
136        self.z_offset_tcl = wx.TextCtrl(self, -1,
137                                         size=(_BOX_WIDTH, 20), style=0)
138        offset_unit_txt = wx.StaticText(self, -1, 'Unit: ')
139        self.offset_unit_tcl = wx.TextCtrl(self, -1,
140                                           size=(_BOX_WIDTH, 20), style=0)
141        self.offset_sizer.AddMany([(offset_txt, 0, wx.LEFT | wx.RIGHT, 10),
142                                     (x_offset_txt, 0, wx.LEFT, 30),
143                                     (self.x_offset_tcl, 0, wx.RIGHT, 10),
144                                     (y_offset_txt, 0, wx.EXPAND),
145                                     (self.y_offset_tcl, 0, wx.RIGHT, 10),
146                                     (z_offset_txt, 0, wx.EXPAND),
147                                     (self.z_offset_tcl, 0, wx.RIGHT, 10),
148                                     (offset_unit_txt, 0, wx.EXPAND),
149                                     (self.offset_unit_tcl, 0, wx.RIGHT, 10)])
150
151    def _layout_orientation(self):
152        """
153            Do the  layout for orientation related widgets
154        """
155        #Orientation
156        orientation_txt = wx.StaticText(self, -1, 'Orientation:')
157        x_orientation_txt = wx.StaticText(self, -1, 'x = ')
158        self.x_orientation_tcl = wx.TextCtrl(self, -1,
159                                              size=(_BOX_WIDTH, 20), style=0)
160        y_orientation_txt = wx.StaticText(self, -1, 'y = ')
161        self.y_orientation_tcl = wx.TextCtrl(self, -1,
162                                             size=(_BOX_WIDTH, 20), style=0)
163        z_orientation_txt = wx.StaticText(self, -1, 'z = ')
164        self.z_orientation_tcl = wx.TextCtrl(self, -1,
165                                              size=(_BOX_WIDTH, 20), style=0)
166        orientation_unit_txt = wx.StaticText(self, -1, 'Unit: ')
167        self.orientation_unit_tcl = wx.TextCtrl(self, -1,
168                                             size=(_BOX_WIDTH, 20), style=0)
169        self.orientation_sizer.AddMany([(orientation_txt, 0,
170                                          wx.LEFT | wx.RIGHT, 10),
171                                     (x_orientation_txt, 0, wx.LEFT, 7),
172                                     (self.x_orientation_tcl, 0, wx.RIGHT, 10),
173                                     (y_orientation_txt, 0, wx.EXPAND),
174                                     (self.y_orientation_tcl, 0, wx.RIGHT, 10),
175                                     (z_orientation_txt, 0, wx.EXPAND),
176                                     (self.z_orientation_tcl, 0, wx.RIGHT, 10),
177                                     (orientation_unit_txt, 0, wx.EXPAND),
178                            (self.orientation_unit_tcl, 0, wx.RIGHT, 10)])
179
180    def _layout_beam_center(self):
181        """
182            Do the  layout for beam center related widgets
183        """
184        #Beam center
185        beam_center_txt = wx.StaticText(self, -1, 'Beam Center:')
186        x_beam_center_txt = wx.StaticText(self, -1, 'x = ')
187        self.x_beam_center_tcl = wx.TextCtrl(self, -1,
188                                              size=(_BOX_WIDTH, 20), style=0)
189        y_beam_center_txt = wx.StaticText(self, -1, 'y = ')
190        self.y_beam_center_tcl = wx.TextCtrl(self, -1,
191                                              size=(_BOX_WIDTH, 20), style=0)
192        z_beam_center_txt = wx.StaticText(self, -1, 'z = ')
193        self.z_beam_center_tcl = wx.TextCtrl(self, -1,
194                                              size=(_BOX_WIDTH, 20), style=0)
195        beam_center_unit_txt = wx.StaticText(self, -1, 'Unit: ')
196        self.beam_center_unit_tcl = wx.TextCtrl(self, -1,
197                                                 size=(_BOX_WIDTH, 20), style=0)
198        self.beam_sizer.AddMany([(beam_center_txt, 0, wx.LEFT | wx.RIGHT, 10),
199                                     (x_beam_center_txt, 0, wx.EXPAND),
200                                     (self.x_beam_center_tcl, 0, wx.RIGHT, 10),
201                                     (y_beam_center_txt, 0, wx.EXPAND),
202                                     (self.y_beam_center_tcl, 0, wx.RIGHT, 10),
203                                     (z_beam_center_txt, 0, wx.EXPAND),
204                                     (self.z_beam_center_tcl, 0, wx.RIGHT, 10),
205                                     (beam_center_unit_txt, 0, wx.EXPAND),
206                                (self.beam_center_unit_tcl, 0, wx.RIGHT, 10)])
207
208    def _layout_pixel_size(self):
209        """
210            Do the  layout for pixel size related widgets
211        """
212        #Pixel Size
213        pixel_size_txt = wx.StaticText(self, -1, 'Pixel Size:')
214        x_pixel_size_txt = wx.StaticText(self, -1, 'x = ')
215        self.x_pixel_size_tcl = wx.TextCtrl(self, -1,
216                                             size=(_BOX_WIDTH, 20), style=0)
217        y_pixel_size_txt = wx.StaticText(self, -1, 'y = ')
218        self.y_pixel_size_tcl = wx.TextCtrl(self, -1,
219                                            size=(_BOX_WIDTH, 20), style=0)
220        z_pixel_size_txt = wx.StaticText(self, -1, 'z = ')
221        self.z_pixel_size_tcl = wx.TextCtrl(self, -1,
222                                             size=(_BOX_WIDTH, 20), style=0)
223        pixel_size_unit_txt = wx.StaticText(self, -1, 'Unit: ')
224        self.pixel_size_unit_tcl = wx.TextCtrl(self, -1,
225                                             size=(_BOX_WIDTH, 20), style=0)
226        self.pixel_sizer.AddMany([(pixel_size_txt, 0, wx.LEFT | wx.RIGHT, 10),
227                                     (x_pixel_size_txt, 0, wx.LEFT, 17),
228                                     (self.x_pixel_size_tcl, 0, wx.RIGHT, 10),
229                                     (y_pixel_size_txt, 0, wx.EXPAND),
230                                     (self.y_pixel_size_tcl, 0, wx.RIGHT, 10),
231                                     (z_pixel_size_txt, 0, wx.EXPAND),
232                                     (self.z_pixel_size_tcl, 0, wx.RIGHT, 10),
233                                     (pixel_size_unit_txt, 0, wx.EXPAND),
234                                (self.pixel_size_unit_tcl, 0, wx.RIGHT, 10)])
235
236    def _layout_slit_length(self):
237        """
238            Do the  layout for slit length related widgets
239        """
240        #slit length
241        slit_length_txt = wx.StaticText(self, -1, 'Slit Length: ')
242        self.slit_length_tcl = wx.TextCtrl(self, -1,
243                                            size=(_BOX_WIDTH, 20), style=0)
244        slit_length_unit_txt = wx.StaticText(self, -1, 'Unit: ')
245        self.slit_length_unit_tcl = wx.TextCtrl(self, -1,
246                                                 size=(_BOX_WIDTH, 20), style=0)
247        self.slit_sizer.AddMany([(slit_length_txt, 0, wx.LEFT, 10),
248                                     (self.slit_length_tcl, 0, wx.RIGHT, 10),
249                                     (slit_length_unit_txt, 0, wx.EXPAND),
250                            (self.slit_length_unit_tcl, 0, wx.RIGHT, 10)])
251
252    def _layout_button(self):
253        """
254            Do the layout for the button widgets
255        """
256        self.bt_apply = wx.Button(self, -1, 'Apply')
257        self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
258        self.bt_apply.SetToolTipString("Apply current changes to the detector.")
259        self.bt_cancel = wx.Button(self, -1, 'Cancel')
260        self.bt_cancel.SetToolTipString("Cancel current changes.")
261        self.bt_cancel.Bind(wx.EVT_BUTTON, self.on_click_cancel)
262        self.bt_close = wx.Button(self, wx.ID_CANCEL, 'Close')
263        self.bt_close.SetToolTipString("Close window.")
264        self.button_sizer.AddMany([(self.bt_apply, 0, wx.LEFT, 200),
265                                   (self.bt_cancel, 0, wx.LEFT, 10),
266                                   (self.bt_close, 0, wx.LEFT, 10)])
267
268    def _do_layout(self, data=None):
269        """
270             Draw the current panel
271        """
272        self._define_structure()
273        self._layout_detector()
274        self._layout_instrument_sizer()
275        self._layout_distance()
276        self._layout_offset()
277        self._layout_orientation()
278        self._layout_beam_center()
279        self._layout_pixel_size()
280        self._layout_slit_length()
281        self._layout_button()
282
283        self.boxsizer_detector.AddMany([(self.instrument_sizer, 0,
284                                          wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
285                                   (self.distance_sizer, 0,
286                                     wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
287                                   (self.offset_sizer, 0,
288                                     wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
289                                   (self.orientation_sizer, 0,
290                                    wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
291                                   (self.beam_sizer, 0,
292                                    wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
293                                   (self.pixel_sizer, 0,
294                                    wx.EXPAND | wx.TOP | wx.BOTTOM, 5),
295                                   (self.slit_sizer, 0,
296                                     wx.EXPAND | wx.TOP | wx.BOTTOM, 5)])
297        self.main_sizer.AddMany([(self.detector_sizer, 0, wx.ALL, 10),
298                                 (self.boxsizer_detector, 0, wx.ALL, 10),
299                                  (self.button_sizer, 0,
300                                    wx.EXPAND | wx.TOP | wx.BOTTOM, 5)])
301
302        self.SetSizer(self.main_sizer)
303        self.SetAutoLayout(True)
304
305    def reset_detector(self):
306        """
307            put the default value of the detector back to the current detector
308        """
309        self._detector = deepcopy(self._reset_detector)
310        self.detector_cbox.Clear()
311        self.fill_detector_combox()
312        self.set_values()
313
314    def set_manager(self, manager):
315        """
316            Set manager of this window
317        """
318        self.manager = manager
319
320    def fill_detector_combox(self):
321        """
322            fill the current combobox with the available detector
323        """
324        for detector in self._detector:
325            pos = self.detector_cbox.Append(str(detector.name))
326            self.detector_cbox.SetClientData(pos, detector)
327            self.detector_cbox.SetSelection(pos)
328            self.detector_cbox.SetStringSelection(str(detector.name))
329
330    def reset_detector_combobox(self, edited_detector):
331        """
332            take all edited editor and reset clientdata of detector combo box
333        """
334        for position in range(self.detector_cbox.GetCount()):
335            detector = self.detector_cbox.GetClientData(position)
336            if detector == edited_detector:
337                detector = edited_detector
338                self.detector_cbox.SetString(position, str(detector.name))
339                self.detector_cbox.SetClientData(position, detector)
340                self.detector_cbox.SetStringSelection(str(detector.name))
341
342    def add_detector(self, event):
343        """
344            Append empty detector to data's list of detector
345        """
346
347        if not self.detector_cbox.IsEnabled():
348            self.detector_cbox.Enable()
349        detector = Detector()
350        self._detector.append(detector)
351        position = self.detector_cbox.Append(str(detector.name))
352        self.detector_cbox.SetClientData(position, detector)
353        self.detector_cbox.SetSelection(position)
354        self.enable_detector()
355        self.set_values()
356
357    def remove_detector(self, event):
358        """
359            Remove detector to data's list of detector
360        """
361        if self.detector_cbox.IsEnabled():
362            if self.detector_cbox.GetCount() > 1:
363                position = self.detector_cbox.GetCurrentSelection()
364                detector = self.detector_cbox.GetClientData(position)
365                if detector in self._detector:
366                    self._detector.remove(detector)
367                    self.detector_cbox.Delete(position)
368                    #set the combo box box the next available item
369                    position = self.detector_cbox.GetCount()
370                    if position > 0:
371                        position -= 1
372                    self.detector_cbox.SetSelection(position)
373                    self.set_values()
374        #disable or enable the combo box when necessary
375        self.enable_detector()
376
377    def enable_detector(self):
378        """
379            Enable /disable widgets crelated to detector
380        """
381        if self._detector is not None and self.detector_cbox.GetCount() > 0:
382            self.detector_cbox.Enable()
383            self.bt_remove_detector.Enable()
384            n_detector = self.detector_cbox.GetCount()
385            detector_hint_txt = 'Detectors available: %s ' % str(n_detector)
386            if len(self._detector) > 1:
387                self.bt_remove_detector.Enable()
388            else:
389                self.bt_remove_detector.Disable()
390        else:
391            self.detector_cbox.Disable()
392            self.bt_remove_detector.Disable()
393            detector_hint_txt = 'No detector is available for this data.'
394        self.detector_txt.SetLabel(detector_hint_txt)
395
396    def set_detector(self, detector):
397        """
398            set detector for data
399        """
400        if self._detector is None:
401            return
402        if self._detector:
403            for item in self._detector:
404                if item == detector:
405                    item = detector
406                    self.reset_detector_combobox(edited_detector=detector)
407                    return
408
409    def get_current_detector(self):
410        """
411        """
412        if not self.detector_cbox.IsEnabled():
413            return None, None, None
414        position = self.detector_cbox.GetSelection()
415        if position == wx.NOT_FOUND:
416            return None, None, None
417        detector_name = self.detector_cbox.GetStringSelection()
418        detector = self.detector_cbox.GetClientData(position)
419        return detector, detector_name, position
420
421    def set_values(self):
422        """
423            take the detector values of the current data and display them
424            through the panel
425        """
426        detector, _, _ = self.get_current_detector()
427        if detector is None:
428            return
429        self.instrument_name_tcl.SetValue(str(detector.name))
430        #Distance
431        distance = detector.distance
432        self.distance_tcl.SetValue(str(distance))
433        self.distance_unit_tcl.SetValue(str(detector.distance_unit))
434        #Offset
435        x, y, z = detector.offset.x, detector.offset.y , detector.offset.z
436        self.x_offset_tcl.SetValue(str(x))
437        self.y_offset_tcl.SetValue(str(y))
438        self.z_offset_tcl.SetValue(str(z))
439        self.offset_unit_tcl.SetValue(str(detector.offset_unit))
440        #Orientation
441        x, y = detector.orientation.x, detector.orientation.y
442        z = detector.orientation.z
443        self.x_orientation_tcl.SetValue(str(x))
444        self.y_orientation_tcl.SetValue(str(y))
445        self.z_orientation_tcl.SetValue(str(z))
446        self.orientation_unit_tcl.SetValue(str(detector.orientation_unit))
447        #Beam center
448        x, y = detector.beam_center.x, detector.beam_center.y
449        z = detector.beam_center.z
450        self.x_beam_center_tcl.SetValue(str(x))
451        self.y_beam_center_tcl.SetValue(str(y))
452        self.z_beam_center_tcl.SetValue(str(z))
453        self.beam_center_unit_tcl.SetValue(str(detector.beam_center_unit))
454        #Pixel size
455        x, y = detector.pixel_size.x, detector.pixel_size.y
456        z = detector.pixel_size.z
457        self.x_pixel_size_tcl.SetValue(str(x))
458        self.y_pixel_size_tcl.SetValue(str(y))
459        self.z_pixel_size_tcl.SetValue(str(z))
460        self.pixel_size_unit_tcl.SetValue(str(detector.pixel_size_unit))
461        #Slit length
462        slit_length = detector.slit_length
463        self.slit_length_tcl.SetValue(str(detector.slit_length))
464        self.slit_length_unit_tcl.SetValue(str(detector.slit_length_unit))
465
466    def get_detector(self):
467        """
468            return the current detector
469        """
470        return self._detector
471
472    def get_notes(self):
473        """
474            return notes
475        """
476        return self._notes
477
478    def on_change_instrument(self):
479        """
480            Change instrument
481        """
482        detector, detector_name, position = self.get_current_detector()
483        if detector is None:
484            return
485        #Change the name of the detector
486        name = self.instrument_name_tcl.GetValue().lstrip().rstrip()
487        if name == "" or name == str(None):
488            name = None
489        if detector_name != name:
490            self._notes += " Instrument's "
491            self._notes += "name from %s to %s \n" % (detector_name, name)
492            detector.name = name
493            self.detector_cbox.SetString(position, str(detector.name))
494            self.detector_cbox.SetClientData(position, detector)
495            self.detector_cbox.SetStringSelection(str(detector.name))
496
497    def on_change_distance(self):
498        """
499            Change distance of the sample to the detector
500        """
501        detector, _, position = self.get_current_detector()
502        if detector is None:
503            return
504        #Change the distance
505        distance = self.distance_tcl.GetValue().lstrip().rstrip()
506        if distance == "" or distance == str(None):
507            distance = None
508            detector.distance = distance
509        else:
510            if check_float(self.distance_tcl):
511                if detector.distance != float(distance):
512                    self._notes += " Change Distance from"
513                    self._notes += " %s to %s \n" % (detector.distance, distance)
514                    detector.distance = float(distance)
515            else:
516                self._notes += "Error: Expected a float for "
517                self._notes += " the distance won't changes "
518                self._notes += "%s to %s" % (detector.distance, distance)
519        #change the distance unit
520        unit = self.distance_unit_tcl.GetValue().lstrip().rstrip()
521        if detector.distance_unit != unit:
522            self._notes += " Change distance's unit from "
523            self._notes += "%s to %s" % (detector.distance_unit, unit)
524
525        self.detector_cbox.SetString(position, str(detector.name))
526        self.detector_cbox.SetClientData(position, detector)
527        self.detector_cbox.SetStringSelection(str(detector.name))
528
529    def on_change_offset(self):
530        """
531            Change the detector offset
532        """
533        detector, _, position = self.get_current_detector()
534        if detector is None:
535            return
536        #Change x coordinate
537        x_offset = self.x_offset_tcl.GetValue().lstrip().rstrip()
538        if x_offset == "" or x_offset == str(None):
539            x_offset = None
540            detector.offset.x = x_offset
541        else:
542            if check_float(self.x_offset_tcl):
543                if detector.offset.x != float(x_offset):
544                    self._notes += "Change x of offset from"
545                    self._notes += " %s to %s \n" % (detector.offset.x,
546                                                      x_offset)
547                    detector.offset.x = float(x_offset)
548            else:
549                self._notes += "Error: Expected a float for the offset 's x "
550                self._notes += "won't changes x offset"
551                self._notes += " from %s to %s" % (detector.offset.x, x_offset)
552        #Change y coordinate
553        y_offset = self.y_offset_tcl.GetValue().lstrip().rstrip()
554        if y_offset == "" or y_offset == str(None):
555            y_offset = None
556            detector.offset.y = y_offset
557        else:
558            if check_float(self.y_offset_tcl):
559                if detector.offset.y != float(y_offset):
560                    self._notes += "Change y of offset from "
561                    self._notes += "%s to %s \n" % (detector.offset.y, y_offset)
562                    detector.offset.y = float(y_offset)
563            else:
564                self._notes += "Error: Expected a float for the offset 's y "
565                self._notes += "won't changes y "
566                self._notes += "offset from %s to %s" % (detector.offset.y,
567                                                         y_offset)
568        #Change z coordinate
569        z_offset = self.z_offset_tcl.GetValue().lstrip().rstrip()
570        if z_offset == "" or z_offset == str(None):
571            z_offset = None
572            detector.offset.z = z_offset
573        else:
574            if check_float(self.z_offset_tcl):
575                if detector.offset.z != float(z_offset):
576                    self._notes += "Change z of offset from"
577                    self._notes += " %s to %s \n" % (detector.offset.z,
578                                                              z_offset)
579                    detector.offset.z = float(z_offset)
580            else:
581                self._notes += "Error: Expected a float for the offset 's x "
582                self._notes += "won't changes z"
583                self._notes += "offset from %s to %s" % (detector.offset.z,
584                                                       z_offset)
585        #change the offset unit
586        unit = self.offset_unit_tcl.GetValue().lstrip().rstrip()
587        if detector.offset_unit != unit:
588            self._notes += " Change Offset's"
589            self._notes += "unit from %s to %s" % (detector.offset_unit, unit)
590
591        self.detector_cbox.SetString(position, str(detector.name))
592        self.detector_cbox.SetClientData(position, detector)
593        self.detector_cbox.SetStringSelection(str(detector.name))
594
595    def on_change_orientation(self):
596        """
597            Change the detector orientation
598        """
599        detector, _, position = self.get_current_detector()
600        if detector is None:
601            return
602        #Change x coordinate
603        x_orientation = self.x_orientation_tcl.GetValue().lstrip().rstrip()
604        if x_orientation == "" or x_orientation == str(None):
605            x_orientation = None
606            detector.orientation.x = x_orientation
607        else:
608            if check_float(self.x_orientation_tcl):
609                if detector.orientation.x != float(x_orientation):
610                    self._notes += "Change x of orientation from "
611                    self._notes += "%s to %s \n" % (detector.orientation.x,
612                                                   x_orientation)
613                    detector.orientation.x = float(x_orientation)
614            else:
615                self._notes += "Error: Expected a float for the orientation "
616                self._notes += "'s x  won't changes x orientation from "
617                self._notes += "%s to %s" % (detector.orientation.x,
618                                              x_orientation)
619        #Change y coordinate
620        y_orientation = self.y_orientation_tcl.GetValue().lstrip().rstrip()
621        if y_orientation == "" or y_orientation == str(None):
622            y_orientation = None
623            detector.orientation.y = y_orientation
624        else:
625            if check_float(self.y_orientation_tcl):
626                if detector.orientation.y != float(y_orientation):
627                    self._notes += "Change y of orientation from "
628                    self._notes += "%s to %s \n" % (detector.orientation.y,
629                                                     y_orientation)
630                    detector.orientation.y = float(y_orientation)
631            else:
632                self._notes += "Error: Expected a float for the orientation's "
633                self._notes += " y won't changes y orientation from "
634                self._notes += "%s to %s" % (detector.orientation.y,
635                                            y_orientation)
636        #Change z coordinate
637        z_orientation = self.z_orientation_tcl.GetValue().lstrip().rstrip()
638        if z_orientation == "" or z_orientation == str(None):
639            z_orientation = None
640            detector.orientation.z = z_orientation
641        else:
642            if check_float(self.z_orientation_tcl):
643                if detector.orientation.z != float(z_orientation):
644                    self._notes += "Change z of offset from "
645                    self._notes += "%s to %s \n" % (detector.orientation.z,
646                                                   z_orientation)
647                    detector.orientation.z = float(z_orientation)
648            else:
649                self._notes += "Error: Expected a float for the orientation 's"
650                self._notes += " x won't changes z orientation from "
651                self._notes += "%s to %s" % (detector.orientation.z,
652                                              z_orientation)
653        #change the orientation unit
654        unit = self.orientation_unit_tcl.GetValue().lstrip().rstrip()
655        if detector.orientation_unit != unit:
656            self._notes += " Change orientation's unit from "
657            self._notes += "%s to %s" % (detector.orientation_unit, unit)
658
659        self.detector_cbox.SetString(position, str(detector.name))
660        self.detector_cbox.SetClientData(position, detector)
661        self.detector_cbox.SetStringSelection(str(detector.name))
662
663    def on_change_beam_center(self):
664        """
665            Change the detector beam center
666        """
667
668        detector, _, position = self.get_current_detector()
669        if detector is None:
670            return
671        #Change x coordinate
672        x_beam_center = self.x_beam_center_tcl.GetValue().lstrip().rstrip()
673        if x_beam_center == "" or x_beam_center == str(None):
674            x_beam_center = None
675            detector.beam_center.x = x_beam_center
676        else:
677            if check_float(self.x_beam_center_tcl):
678                if detector.beam_center.x != float(x_beam_center):
679                    self._notes += "Change x of offset from "
680                    self._notes += "%s to %s \n" % (detector.beam_center.x,
681                                                     x_beam_center)
682                    detector.beam_center.x = float(x_beam_center)
683            else:
684                self._notes += "Error: Expected a float for the beam "
685                self._notes += "center 's x won't changes x beam center from "
686                self._notes += "%s to %s" % (detector.beam_center.x,
687                                            x_beam_center)
688        #Change y coordinate
689        y_beam_center = self.y_beam_center_tcl.GetValue().lstrip().rstrip()
690        if y_beam_center == "" or y_beam_center == str(None):
691            y_beam_center = None
692            detector.beam_center.y = y_beam_center
693        else:
694            if check_float(self.y_beam_center_tcl):
695                if detector.beam_center.y != float(y_beam_center):
696                    self._notes += "Change y of beam center from "
697                    self._notes += "%s to %s \n" % (detector.beam_center.y,
698                                                     y_beam_center)
699                    detector.beam_center.y = float(y_beam_center)
700            else:
701                self._notes += "Error: Expected a float for the beam "
702                self._notes += "center 's y won't changes y beam center from "
703                self._notes += "%s to %s" % (detector.beam_center.y,
704                                              y_beam_center)
705        #Change z coordinate
706        z_beam_center = self.z_beam_center_tcl.GetValue().lstrip().rstrip()
707        if z_beam_center == "" or z_beam_center == str(None):
708            z_beam_center = None
709            detector.beam_center.z = z_beam_center
710        else:
711            if check_float(self.z_beam_center_tcl):
712                if detector.beam_center.z != float(z_beam_center):
713                    self._notes += "Change z of beam center from "
714                    self._notes += "%s to %s \n" % (detector.beam_center.z,
715                                                     z_beam_center)
716                    detector.beam_center.z = float(z_beam_center)
717            else:
718                self._notes += "Error: Expected a float for the offset 's x "
719                self._notes += "won't changes z beam center from "
720                self._notes += "%s to %s" % (detector.beam_center.z,
721                                            z_beam_center)
722        #change the beam center unit
723        unit = self.beam_center_unit_tcl.GetValue().lstrip().rstrip()
724        if detector.beam_center_unit != unit:
725            self._notes += " Change beam center's unit from "
726            self._notes += "%s to %s" % (detector.beam_center_unit, unit)
727
728        self.detector_cbox.SetString(position, str(detector.name))
729        self.detector_cbox.SetClientData(position, detector)
730        self.detector_cbox.SetStringSelection(str(detector.name))
731
732    def on_change_pixel_size(self):
733        """
734            Change the detector pixel size
735        """
736        detector, _, position = self.get_current_detector()
737        if detector is None:
738            return
739        #Change x coordinate
740        x_pixel_size = self.x_pixel_size_tcl.GetValue().lstrip().rstrip()
741        if x_pixel_size == "" or x_pixel_size == str(None):
742            x_pixel_size = None
743        else:
744            if check_float(self.x_pixel_size_tcl):
745                if detector.pixel_size.x != float(x_pixel_size) :
746                    self._notes += "Change x of pixel size from "
747                    self._notes += "%s to %s \n" % (detector.pixel_size.x,
748                                                   x_pixel_size)
749                    detector.pixel_size.x = float(x_pixel_size)
750            else:
751                self._notes += "Error: Expected a float for the pixel"
752                self._notes += " size 's x  won't changes x pixel size from "
753                self._notes += "%s to %s" % (detector.pixel_size.x,
754                                             x_pixel_size)
755        #Change y coordinate
756        y_pixel_size = self.y_pixel_size_tcl.GetValue().lstrip().rstrip()
757        if y_pixel_size == "" or y_pixel_size == str(None):
758            y_pixel_size = None
759            detector.pixel_size.y = y_pixel_size
760        else:
761            if check_float(self.y_pixel_size_tcl):
762                if detector.pixel_size.y != float(y_pixel_size):
763                    self._notes += "Change y of pixel size from "
764                    self._notes += "%s to %s \n" % (detector.pixel_size.y,
765                                                   y_pixel_size)
766                    detector.pixel_size.y = float(y_pixel_size)
767            else:
768                self._notes += "Error: Expected a float for the pixel "
769                self._notes += "size's y  won't changes y pixel size from "
770                self._notes += "%s to %s" % (detector.pixel_size.y,
771                                              y_pixel_size)
772        #Change z coordinate
773        z_pixel_size = self.z_pixel_size_tcl.GetValue().lstrip().rstrip()
774        if z_pixel_size == "" or z_pixel_size == str(None):
775            z_pixel_size = None
776            detector.pixel_size.z = z_pixel_size
777        else:
778            if check_float(self.z_pixel_size_tcl):
779                if detector.pixel_size.z != float(z_pixel_size):
780                    self._notes += "Change z of pixel size from "
781                    self._notes += "%s to %s \n" % (detector.pixel_size.z,
782                                                   z_pixel_size)
783                    detector.pixel_size.z = float(z_pixel_size)
784            else:
785                self._notes += "Error: Expected a float for the offset 's x "
786                self._notes += "won't changes z pixel size from "
787                self._notes += "%s to %s" % (detector.pixel_size.z, z_pixel_size)
788        #change the beam center unit
789        unit = self.pixel_size_unit_tcl.GetValue().lstrip().rstrip()
790        if detector.pixel_size_unit != unit:
791            self._notes += " Change pixel size's unit from "
792            self._notes += "%s to %s" % (detector.pixel_size_unit, unit)
793
794        self.detector_cbox.SetString(position, str(detector.name))
795        self.detector_cbox.SetClientData(position, detector)
796        self.detector_cbox.SetStringSelection(str(detector.name))
797
798    def on_change_slit_length(self):
799        """
800            Change slit length of the detector
801        """
802        detector, _, position = self.get_current_detector()
803        if detector is None:
804            return
805        #Change the distance
806        slit_length = self.slit_length_tcl.GetValue().lstrip().rstrip()
807        if slit_length == "" or slit_length == str(None):
808            slit_length = None
809            detector.slit_length = slit_length
810        else:
811            if check_float(self.slit_length_tcl):
812                if detector.slit_length != float(slit_length):
813                    self._notes += " Change slit length from"
814                    self._notes += " %s to %s \n" % (detector.slit_length,
815                                                    slit_length)
816                    detector.slit_length = float(slit_length)
817            else:
818                self._notes += "Error: Expected a float"
819                self._notes += " for the slit length won't changes "
820                self._notes += "%s to %s" % (detector.slit_length, slit_length)
821        #change the distance unit
822        unit = self.slit_length_unit_tcl.GetValue().lstrip().rstrip()
823        if detector.slit_length_unit != unit:
824            self._notes += " Change slit length's unit from "
825            self._notes += "%s to %s" % (detector.slit_length_unit_tcl, unit)
826
827        self.detector_cbox.SetString(position, str(detector.name))
828        self.detector_cbox.SetClientData(position, detector)
829        self.detector_cbox.SetStringSelection(str(detector.name))
830
831    def on_click_cancel(self, event):
832        """
833            reset the current detector to its initial values
834        """
835        self.reset_detector()
836        self.set_values()
837        if self.manager is not None:
838             self.manager.set_detector(self._detector)
839
840    def on_click_apply(self, event):
841        """
842            Apply user values to the detector
843        """
844        self.on_change_instrument()
845        self.on_change_distance()
846        self.on_change_instrument()
847        self.on_change_beam_center()
848        self.on_change_offset()
849        self.on_change_orientation()
850        self.on_change_pixel_size()
851        self.on_change_slit_length()
852        for detector in self._detector:
853            self.manager.set_detector(self._detector, self._notes)
854
855if __name__ == "__main__":
856    app = wx.App()
857    test_detector = Detector()
858    dlg = DetectorDialog(detector=[test_detector])
859    dlg.ShowModal()
860    app.MainLoop()
Note: See TracBrowser for help on using the repository browser.