source: sasview/src/sas/perspectives/calculator/data_operator.py @ d5419f7f

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 d5419f7f was d5419f7f, checked in by butler, 9 years ago
  • Fixed problem with mac vs pc positioning of tool panels for now -

opened ticket to completely rethink panels in mac vs pc

  • Added close button/activated close method for tool panels that did not

have it. Tool panels are now much more consistant.

  • Property mode set to 100644
File size: 35.3 KB
Line 
1"""
2GUI for the data operations panel (sum and multiply)
3"""
4import wx
5import sys
6import time
7import numpy
8from sas.dataloader.data_info import Data1D
9from sas.plottools.PlotPanel import PlotPanel
10from sas.plottools.plottables import Graph
11from sas.plottools import transform
12from matplotlib.font_manager import FontProperties
13from sas.guiframe.events import StatusEvent
14from sas.perspectives.calculator import calculator_widgets as widget
15from sas.guiframe.documentation_window import DocumentationWindow
16
17#Control panel width
18if sys.platform.count("win32") > 0:
19    PANEL_TOP = 0
20    PANEL_WIDTH = 790
21    PANEL_HEIGTH = 370
22    FONT_VARIANT = 0
23    _BOX_WIDTH = 200
24    ON_MAC = False
25else:
26    PANEL_TOP = 60
27    _BOX_WIDTH = 230
28    PANEL_WIDTH = 900
29    PANEL_HEIGTH = 430
30    FONT_VARIANT = 1
31    ON_MAC = True
32
33class DataOperPanel(wx.ScrolledWindow):
34    """
35    """
36    def __init__(self, parent, *args, **kwds):
37        kwds['name'] = "Data Operation"
38        kwds["size"] = (PANEL_WIDTH, PANEL_HEIGTH)
39        wx.ScrolledWindow.__init__(self, parent, *args, **kwds)
40        self.parent = parent
41        #sizers etc.
42        self.main_sizer = None
43        self.name_sizer = None
44        self.button_sizer = None
45        self.data_namectr = None
46        self.numberctr = None
47        self.data1_cbox = None
48        self.operator_cbox = None
49        self.data2_cbox = None
50        self.data_title_tcl = None
51        self.out_pic = None
52        self.equal_pic = None
53        self.data1_pic = None
54        self.operator_pic = None
55        self.data2_pic = None
56        self.output = None
57        self._notes = None
58        #text grayed color
59        self.color = wx.SystemSettings.GetColour(wx.SYS_COLOUR_BACKGROUND)
60        #data
61        self._data = self.get_datalist()
62        self._do_layout()
63        self.fill_data_combox()
64        self.fill_oprator_combox()
65        self.Bind(wx.EVT_PAINT, self.set_panel_on_focus)
66
67    def _define_structure(self):
68        """
69        define initial sizer
70        """
71        self.main_sizer = wx.BoxSizer(wx.VERTICAL)
72        title = "Data Operation "
73        title += "[ + (add); - (subtract); "
74        title += "* (multiply); / (divide); "
75        title += "| (append) ]"
76        name_box = wx.StaticBox(self, -1, title)
77        self.name_sizer = wx.StaticBoxSizer(name_box, wx.HORIZONTAL)
78        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
79
80    def _layout_name(self):
81        """
82        Do the layout for data name related widgets
83        """
84        new_data_sizer = wx.BoxSizer(wx.VERTICAL)
85        equal_sizer = wx.BoxSizer(wx.VERTICAL)
86        old_data1_sizer = wx.BoxSizer(wx.VERTICAL)
87        operator_sizer = wx.BoxSizer(wx.VERTICAL)
88        old_data2_sizer = wx.BoxSizer(wx.VERTICAL)
89        data2_hori_sizer = wx.BoxSizer(wx.HORIZONTAL)
90        data_name = wx.StaticText(self, -1, 'Output Data Name')
91        equal_name = wx.StaticText(self, -1, ' =', size=(50, 25))
92        data1_name = wx.StaticText(self, -1, 'Data1')
93        operator_name = wx.StaticText(self, -1, 'Operator')
94        data2_name = wx.StaticText(self, -1, 'Data2 (or Number)')
95        self.data_namectr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 25), style=wx.TE_PROCESS_ENTER)
96        self.data_namectr.SetToolTipString("Hit 'Enter' key after typing.")
97        self.data_namectr.SetValue(str('MyNewDataName'))
98        self.numberctr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH / 3, 25), style=wx.TE_PROCESS_ENTER)
99        self.numberctr.SetToolTipString("Hit 'Enter' key after typing.")
100        self.numberctr.SetValue(str(1.0))
101        self.data1_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH, 25),
102                                      style=wx.CB_READONLY)
103        self.operator_cbox = wx.ComboBox(self, -1, size=(70, 25),
104                                         style=wx.CB_READONLY)
105        operation_tip = "Add: +, Subtract: -, "
106        operation_tip += "Multiply: *, Divide: /, "
107        operation_tip += "Append(Combine): | "
108        self.operator_cbox.SetToolTipString(operation_tip)
109        self.data2_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH * 2 / 3, 25),
110                                       style=wx.CB_READONLY)
111
112        self.out_pic = SmallPanel(self, -1, True,
113                                    size=(_BOX_WIDTH, _BOX_WIDTH),
114                                    style=wx.NO_BORDER)
115        self.equal_pic = SmallPanel(self, -1, True, '=',
116                                    size=(50, _BOX_WIDTH),
117                                    style=wx.NO_BORDER)
118        self.data1_pic = SmallPanel(self, -1, True,
119                                    size=(_BOX_WIDTH, _BOX_WIDTH),
120                                    style=wx.NO_BORDER)
121        self.operator_pic = SmallPanel(self, -1, True, '+',
122                                    size=(70, _BOX_WIDTH),
123                                    style=wx.NO_BORDER)
124        self.data2_pic = SmallPanel(self, -1, True,
125                                    size=(_BOX_WIDTH, _BOX_WIDTH),
126                                    style=wx.NO_BORDER)
127        for ax in self.equal_pic.axes:
128            ax.set_frame_on(False)
129        for ax in self.operator_pic.axes:
130            ax.set_frame_on(False)
131
132        new_data_sizer.AddMany([(data_name, 0, wx.LEFT, 3),
133                                       (self.data_namectr, 0, wx.LEFT, 3),
134                                       (self.out_pic, 0, wx.LEFT, 3)])
135        equal_sizer.AddMany([(13, 13), (equal_name, 0, wx.LEFT, 3),
136                                       (self.equal_pic, 0, wx.LEFT, 3)])
137        old_data1_sizer.AddMany([(data1_name, 0, wx.LEFT, 3),
138                                       (self.data1_cbox, 0, wx.LEFT, 3),
139                                       (self.data1_pic, 0, wx.LEFT, 3)])
140        operator_sizer.AddMany([(operator_name, 0, wx.LEFT, 3),
141                                 (self.operator_cbox, 0, wx.LEFT, 3),
142                                 (self.operator_pic, 0, wx.LEFT, 3)])
143        data2_hori_sizer.AddMany([(self.data2_cbox, 0, wx.LEFT, 0),
144                                       (self.numberctr, 0, wx.RIGHT, 0)])
145        old_data2_sizer.AddMany([(data2_name, 0, wx.LEFT, 3),
146                                       (data2_hori_sizer, 0, wx.LEFT, 3),
147                                       (self.data2_pic, 0, wx.LEFT, 3)])
148        self.name_sizer.AddMany([(new_data_sizer, 0, wx.LEFT | wx.TOP, 5),
149                                       (equal_sizer, 0, wx.TOP, 5),
150                                       (old_data1_sizer, 0, wx.TOP, 5),
151                                       (operator_sizer, 0, wx.TOP, 5),
152                                       (old_data2_sizer, 0, wx.TOP, 5)])
153        self.data2_cbox.Show(True)
154
155        self._show_numctrl(self.numberctr, False)
156
157        wx.EVT_TEXT_ENTER(self.data_namectr, -1, self.on_name)
158        wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number)
159        wx.EVT_COMBOBOX(self.data1_cbox, -1, self.on_select_data1)
160        wx.EVT_COMBOBOX(self.operator_cbox, -1, self.on_select_operator)
161        wx.EVT_COMBOBOX(self.data2_cbox, -1, self.on_select_data2)
162
163    def _show_numctrl(self, ctrl, enable=True):
164        """
165        Show/Hide on Win
166        Enable/Disable on MAC
167        """
168        if ON_MAC:
169            ctrl.Enable(enable)
170            children = ctrl.GetChildren()
171            if len(children) > 0:
172                ctrl.GetChildren()[0].SetBackGroundColour(self.color)
173            if enable:
174                wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number)
175        else:
176            if not ctrl.IsEnabled():
177                ctrl.Enable(True)
178            ctrl.Show(enable)
179
180    def on_name(self, event=None):
181        """
182        On data name typing
183        """
184        if event != None:
185            event.Skip()
186        item = event.GetEventObject()
187        if item.IsEnabled():
188            self._set_textctrl_color(item, 'white')
189        else:
190            self._set_textctrl_color(item, self.color)
191        text = item.GetValue().strip()
192        self._check_newname(text)
193
194    def _check_newname(self, name=None):
195        """
196        Check name ctr strings
197        """
198        self.send_warnings('')
199        msg = ''
200        if name == None:
201            text = self.data_namectr.GetValue().strip()
202        else:
203            text = name
204        state_list = self.get_datalist().values()
205        name_list = []
206        for state in state_list:
207            if state.data == None:
208                theory_list = state.get_theory()
209                theory, _ = theory_list.values()[0]
210                d_name = str(theory.name)
211            else:
212                d_name = str(state.data.name)
213            name_list.append(d_name)
214        if text in name_list:
215            self._set_textctrl_color(self.data_namectr, 'pink')
216            msg = "DataOperation: The name already exists."
217        if len(text) == 0:
218            self._set_textctrl_color(self.data_namectr, 'pink')
219            msg = "DataOperation: Type the data name first."
220        if self._notes:
221            self.send_warnings(msg, 'error')
222        self.name_sizer.Layout()
223        self.Refresh()
224
225    def _set_textctrl_color(self, ctrl, color):
226        """
227        Set TextCtrl color
228        """
229        if ON_MAC:
230            children = ctrl.GetChildren()
231            if len(children) > 0:
232                children[0].SetBackgroundColour(color)
233        else:
234            ctrl.SetBackgroundColour(color)
235        self.name_sizer.Layout()
236
237    def on_number(self, event=None):
238        """
239        On selecting Number for Data2
240        """
241        self.send_warnings('')
242        item = event.GetEventObject()
243        text = item.GetValue().strip()
244        if self.numberctr.IsShown():
245            if self.numberctr.IsEnabled():
246                self._set_textctrl_color(self.numberctr, 'white')
247                try:
248                    val = float(text)
249                    pos = self.data2_cbox.GetCurrentSelection()
250                    self.data2_cbox.SetClientData(pos, val)
251                except:
252                    self._set_textctrl_color(self.numberctr, 'pink')
253                    msg = "DataOperation: Number requires a float number."
254                    self.send_warnings(msg, 'error')
255                    return
256            else:
257                self._set_textctrl_color(self.numberctr, self.color)
258
259        self.put_text_pic(self.data2_pic, content=str(val))
260        self.check_data_inputs()
261        if self.output != None:
262            self.output.name = str(self.data_namectr.GetValue())
263        self.draw_output(self.output)
264        self.Refresh()
265
266    def on_select_data1(self, event=None):
267        """
268        On select data1
269        """
270        self.send_warnings('')
271        item = event.GetEventObject()
272        pos = item.GetCurrentSelection()
273        data = item.GetClientData(pos)
274        if data == None:
275            content = "?"
276            self.put_text_pic(self.data1_pic, content)
277        else:
278            self.data1_pic.add_image(data)
279        self.check_data_inputs()
280        if self.output != None:
281            self.output.name = str(self.data_namectr.GetValue())
282        self.draw_output(self.output)
283
284    def on_select_operator(self, event=None):
285        """
286        On Select an Operator
287        """
288        self.send_warnings('')
289        item = event.GetEventObject()
290        text = item.GetValue().strip()
291        self.put_text_pic(self.operator_pic, content=text)
292        self.check_data_inputs()
293        if self.output != None:
294            self.output.name = str(self.data_namectr.GetValue())
295        self.draw_output(self.output)
296
297    def on_select_data2(self, event=None):
298        """
299        On Selecting Data2
300        """
301        self.send_warnings('')
302        item = event.GetEventObject()
303        text = item.GetValue().strip().lower()
304        self._show_numctrl(self.numberctr, text == 'number')
305        pos = item.GetCurrentSelection()
306        data = item.GetClientData(pos)
307        content = "?"
308        if not (self.numberctr.IsShown() and self.numberctr.IsEnabled()):
309            if data == None:
310                content = "?"
311                self.put_text_pic(self.data2_pic, content)
312            else:
313                self.data2_pic.add_image(data)
314            self.check_data_inputs()
315        else:
316            content = str(self.numberctr.GetValue().strip())
317            try:
318                content = float(content)
319                data = content
320            except:
321                self._set_textctrl_color(self.numberctr, 'pink')
322                content = "?"
323                data = None
324            item.SetClientData(pos, data)
325            if data != None:
326                self.check_data_inputs()
327
328            self.put_text_pic(self.data2_pic, content)
329
330        if self.output != None:
331            self.output.name = str(self.data_namectr.GetValue())
332        self.draw_output(self.output)
333
334    def put_text_pic(self, pic=None, content=''):
335        """
336        Put text to the pic
337        """
338        pic.set_content(content)
339        pic.add_text()
340        pic.draw()
341
342    def check_data_inputs(self):
343        """
344        Check data1 and data2 whether or not they are ready for operation
345        """
346        self._set_textctrl_color(self.data1_cbox, 'white')
347        self._set_textctrl_color(self.data2_cbox, 'white')
348        flag = False
349        pos1 = self.data1_cbox.GetCurrentSelection()
350        data1 = self.data1_cbox.GetClientData(pos1)
351        if data1 == None:
352            self.output = None
353            return flag
354        pos2 = self.data2_cbox.GetCurrentSelection()
355        data2 = self.data2_cbox.GetClientData(pos2)
356
357        if data2 == None:
358            self.output = None
359            return flag
360        if self.numberctr.IsShown():
361            if self.numberctr.IsEnabled():
362                self._set_textctrl_color(self.numberctr, 'white')
363                try:
364                    float(data2)
365                    if self.operator_cbox.GetValue().strip() == '|':
366                        msg = "DataOperation: This operation can not accept "
367                        msg += "a float number."
368                        self.send_warnings(msg, 'error')
369                        self._set_textctrl_color(self.numberctr, 'pink')
370                        self.output = None
371                        return flag
372                except:
373                    msg = "DataOperation: Number requires a float number."
374                    self.send_warnings(msg, 'error')
375                    self._set_textctrl_color(self.numberctr, 'pink')
376                    self.output = None
377                    return flag
378            else:
379                self._set_textctrl_color(self.numberctr, self.color)
380        elif data1.__class__.__name__ != data2.__class__.__name__:
381            self._set_textctrl_color(self.data1_cbox, 'pink')
382            self._set_textctrl_color(self.data2_cbox, 'pink')
383            msg = "DataOperation: Data types must be same."
384            self.send_warnings(msg, 'error')
385            self.output = None
386            return flag
387        try:
388            self.output = self.make_data_out(data1, data2)
389        except:
390            self._check_newname()
391            self._set_textctrl_color(self.data1_cbox, 'pink')
392            self._set_textctrl_color(self.data2_cbox, 'pink')
393            msg = "DataOperation: Data types must be same."
394            self.send_warnings(msg, 'error')
395            self.output = None
396            return flag
397        return True
398
399    def make_data_out(self, data1, data2):
400        """
401        Make a temp. data output set
402        """
403        output = None
404        pos = self.operator_cbox.GetCurrentSelection()
405        operator = self.operator_cbox.GetClientData(pos)
406        try:
407            exec "output = data1 %s data2" % operator
408        except:
409            raise
410        return output
411
412
413    def draw_output(self, output):
414        """
415        Draw output data(temp)
416        """
417        out = self.out_pic
418        if output == None:
419            content = "?"
420            self.put_text_pic(out, content)
421        else:
422            out.add_image(output)
423        wx.CallAfter(self.name_sizer.Layout)
424        self.Layout()
425        self.Refresh()
426
427    def _layout_button(self):
428        """
429            Do the layout for the button widgets
430        """
431        self.bt_apply = wx.Button(self, -1, "Apply", size=(_BOX_WIDTH / 2, -1))
432        app_tip = "Generate the Data and send to Data Explorer."
433        self.bt_apply.SetToolTipString(app_tip)
434        self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
435
436        self.bt_help = wx.Button(self, -1, "HELP")
437        app_tip = "Get help on Data Operations."
438        self.bt_help.SetToolTipString(app_tip)
439        self.bt_help.Bind(wx.EVT_BUTTON, self.on_help)
440
441        self.bt_close = wx.Button(self, -1, 'Close', size=(_BOX_WIDTH / 2, -1))
442        self.bt_close.Bind(wx.EVT_BUTTON, self.on_close)
443        self.bt_close.SetToolTipString("Close this panel.")
444
445        self.button_sizer.AddMany([(PANEL_WIDTH / 2, 25),
446                                   (self.bt_apply, 0, wx.RIGHT, 10),
447                                   (self.bt_help, 0, wx.RIGHT, 10),
448                                   (self.bt_close, 0, wx.RIGHT, 10)])
449
450    def _do_layout(self):
451        """
452        Draw the current panel
453        """
454        self._define_structure()
455        self._layout_name()
456        self._layout_button()
457        self.main_sizer.AddMany([(self.name_sizer, 0, wx.EXPAND | wx.ALL, 10),
458                                (self.button_sizer, 0,
459                                          wx.EXPAND | wx.TOP | wx.BOTTOM, 5)])
460        self.SetSizer(self.main_sizer)
461        self.SetScrollbars(20, 20, 25, 65)
462        self.SetAutoLayout(True)
463
464    def set_panel_on_focus(self, event):
465        """
466        On Focus at this window
467        """
468        if event != None:
469            event.Skip()
470        self._data = self.get_datalist()
471        if ON_MAC:
472            self.fill_data_combox()
473        else:
474            children = self.GetChildren()
475            # update the list only when it is on the top
476            if self.FindFocus() in children:
477                self.fill_data_combox()
478
479    def fill_oprator_combox(self):
480        """
481        fill the current combobox with the operator
482        """
483        operator_list = [' +', ' -', ' *', " /", " |"]
484        for oper in operator_list:
485            pos = self.operator_cbox.Append(str(oper))
486            self.operator_cbox.SetClientData(pos, str(oper.strip()))
487        self.operator_cbox.SetSelection(0)
488
489
490    def fill_data_combox(self):
491        """
492        fill the current combobox with the available data
493        """
494        pos_pre1 = self.data1_cbox.GetCurrentSelection()
495        pos_pre2 = self.data2_cbox.GetCurrentSelection()
496        current1 = self.data1_cbox.GetLabel()
497        current2 = self.data2_cbox.GetLabel()
498        if pos_pre1 < 0:
499            pos_pre1 = 0
500        if pos_pre2 < 0:
501            pos_pre2 = 0
502        self.data1_cbox.Clear()
503        self.data2_cbox.Clear()
504
505        if not self._data:
506            pos = self.data1_cbox.Append('No Data Available')
507            self.data1_cbox.SetSelection(pos)
508            self.data1_cbox.SetClientData(pos, None)
509            pos2 = self.data2_cbox.Append('No Data Available')
510            self.data2_cbox.SetSelection(pos2)
511            self.data2_cbox.SetClientData(pos2, None)
512            return
513        pos1 = self.data1_cbox.Append('Select Data')
514        self.data1_cbox.SetSelection(pos1)
515        self.data1_cbox.SetClientData(pos1, None)
516        pos2 = self.data2_cbox.Append('Select Data')
517        self.data2_cbox.SetSelection(pos2)
518        self.data2_cbox.SetClientData(pos2, None)
519        pos3 = self.data2_cbox.Append('Number')
520        val = None
521        if (self.numberctr.IsShown() and self.numberctr.IsEnabled()):
522            try:
523                val = float(self.numberctr.GetValue())
524            except:
525                val = None
526        self.data2_cbox.SetClientData(pos3, val)
527        dnames = []
528        ids = self._data.keys()
529        for id in ids:
530            if id != None:
531                if self._data[id].data != None:
532                    dnames.append(self._data[id].data.name)
533                else:
534                    theory_list = self._data[id].get_theory()
535                    theory, _ = theory_list.values()[0]
536                    dnames.append(theory.name)
537        ind = numpy.argsort(dnames)
538        if len(ind) > 0:
539            val_list = numpy.array(self._data.values())[ind]
540            for datastate in val_list:
541                data = datastate.data
542                if data != None:
543                    name = data.name
544                    pos1 = self.data1_cbox.Append(str(name))
545                    self.data1_cbox.SetClientData(pos1, data)
546                    pos2 = self.data2_cbox.Append(str(name))
547                    self.data2_cbox.SetClientData(pos2, data)
548                    if str(current1) == str(name):
549                      pos_pre1 = pos1
550                    if str(current2) == str(name):
551                      pos_pre2 = pos2
552                try:
553                    theory_list = datastate.get_theory()
554                    for theory, _ in theory_list.values():
555                        th_name = theory.name
556                        posth1 = self.data1_cbox.Append(str(th_name))
557                        self.data1_cbox.SetClientData(posth1, theory)
558                        posth2 = self.data2_cbox.Append(str(th_name))
559                        self.data2_cbox.SetClientData(posth2, theory)
560                        if str(current1) == str(th_name):
561                            pos_pre1 = posth1
562                        if str(current2) == str(th_name):
563                            pos_pre2 = posth2
564                except:
565                    continue
566        self.data1_cbox.SetSelection(pos_pre1)
567        self.data2_cbox.SetSelection(pos_pre2)
568
569    def get_datalist(self):
570        """
571        """
572        data_manager = self.parent.parent.get_data_manager()
573        if data_manager != None:
574            return  data_manager.get_all_data()
575        else:
576            return {}
577
578    def on_click_apply(self, event):
579        """
580        changes are saved in data object imported to edit
581        """
582        self.send_warnings('')
583        self.data_namectr.SetBackgroundColour('white')
584        state_list = self.get_datalist().values()
585        name = self.data_namectr.GetValue().strip()
586        name_list = []
587        for state in state_list:
588            if state.data == None:
589                theory_list = state.get_theory()
590                theory, _ = theory_list.values()[0]
591                d_name = str(theory.name)
592            else:
593                d_name = str(state.data.name)
594            name_list.append(d_name)
595        if name in name_list:
596            self._set_textctrl_color(self.data_namectr, 'pink')
597            msg = "The Output Data Name already exists...   "
598            wx.MessageBox(msg, 'Error')
599            return
600        if name == '':
601            self._set_textctrl_color(self.data_namectr, 'pink')
602            msg = "Please type the output data name first...   "
603            wx.MessageBox(msg, 'Error')
604            return
605        if self.output == None:
606            msg = "No Output Data has been generated...   "
607            wx.MessageBox(msg, 'Error')
608            return
609        # send data to data manager
610        self.output.name = name
611        self.output.run = "Data Operation"
612        self.output.instrument = "SasView"
613        self.output.id = str(name) + str(time.time())
614        data = {self.output.id :self.output}
615        self.parent.parent.add_data(data)
616        self.name_sizer.Layout()
617        self.Refresh()
618        #must post event here
619        event.Skip()
620
621    def on_help(self, event):
622        """
623        Bring up the Data Operations Panel Documentation whenever
624        the HELP button is clicked.
625
626        Calls DocumentationWindow with the path of the location within the
627        documentation tree (after /doc/ ....".  Note that when using old
628        versions of Wx (before 2.9) and thus not the release version of
629        installers, the help comes up at the top level of the file as
630        webbrowser does not pass anything past the # to the browser when it is
631        running "file:///...."
632
633    :param evt: Triggers on clicking the help button
634    """
635
636        _TreeLocation = "user/perspectives/calculator/data_operator_help.html"
637        _doc_viewer = DocumentationWindow(self, -1, \
638             _TreeLocation, "Data Operation Help")
639
640    def disconnect_panels(self):
641        """
642        """
643        self.out_pic.connect.disconnect()
644        self.equal_pic.connect.disconnect()
645        self.data1_pic.connect.disconnect()
646        self.operator_pic.connect.disconnect()
647        self.data2_pic.connect.disconnect()
648
649    def on_close(self, event):
650        """
651        leave data as it is and close
652        """
653        self.parent.OnClose()
654
655    def set_plot_unfocus(self):
656        """
657        Unfocus on right click
658        """
659
660    def send_warnings(self, msg='', info='info'):
661        """
662        Send warning to status bar
663        """
664        wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info=info))
665
666class SmallPanel(PlotPanel):
667    """
668    PlotPanel for Quick plot and masking plot
669    """
670    def __init__(self, parent, id= -1, is_number=False, content='?', **kwargs):
671        """
672        """
673        PlotPanel.__init__(self, parent, id=id, **kwargs)
674        self.is_number = is_number
675        self.content = content
676        self.point = None
677        self.position = (0.4, 0.5)
678        self.scale = 'linear'
679        self.prevXtrans = "x"
680        self.prevYtrans = "y"
681        self.viewModel = "--"
682        self.subplot.set_xticks([])
683        self.subplot.set_yticks([])
684        self.add_text()
685        self.figure.subplots_adjust(left=0.1, bottom=0.1)
686
687    def set_content(self, content=''):
688        """
689        Set text content
690        """
691        self.content = str(content)
692
693    def add_toolbar(self):
694        """
695        Add toolbar
696        """
697        # Not implemented
698        pass
699
700    def on_set_focus(self, event):
701        """
702        send to the parenet the current panel on focus
703        """
704        pass
705
706    def add_image(self, plot):
707        """
708        Add Image
709        """
710        self.content = ''
711        self.textList = []
712        self.plots = {}
713        self.clear()
714        self.point = plot
715        try:
716            self.figure.delaxes(self.figure.axes[0])
717            self.subplot = self.figure.add_subplot(111)
718            #self.figure.delaxes(self.figure.axes[1])
719        except:
720            pass
721        try:
722            name = plot.name
723        except:
724            name = plot.filename
725        self.plots[name] = plot
726
727        #init graph
728        self.graph = Graph()
729
730        #add plot
731        self.graph.add(plot)
732        #draw       
733        self.graph.render(self)
734
735        try:
736            self.figure.delaxes(self.figure.axes[1])
737        except:
738            pass
739        self.subplot.figure.canvas.resizing = False
740        self.subplot.tick_params(axis='both', labelsize=9)
741        # Draw zero axis lines
742        self.subplot.axhline(linewidth=1, color='r')
743        self.subplot.axvline(linewidth=1, color='r')
744
745        self.erase_legend()
746        try:
747            # mpl >= 1.1.0
748            self.figure.tight_layout()
749        except:
750            self.figure.subplots_adjust(left=0.1, bottom=0.1)
751        self.subplot.figure.canvas.draw()
752
753    def add_text(self):
754        """
755        Text in the plot
756        """
757        if not self.is_number:
758            return
759
760        self.clear()
761        try:
762            self.figure.delaxes(self.figure.axes[0])
763            self.subplot = self.figure.add_subplot(111)
764            self.figure.delaxes(self.figure.axes[1])
765        except:
766            pass
767        self.subplot.set_xticks([])
768        self.subplot.set_yticks([])
769        label = self.content
770        FONT = FontProperties()
771        xpos, ypos = (0.4, 0.5)
772        font = FONT.copy()
773        font.set_size(14)
774
775        self.textList = []
776        self.subplot.set_xlim((0, 1))
777        self.subplot.set_ylim((0, 1))
778
779        try:
780            if self.content != '?':
781                float(label)
782        except:
783            self.subplot.set_frame_on(False)
784        try:
785            # mpl >= 1.1.0
786            self.figure.tight_layout()
787        except:
788            self.figure.subplots_adjust(left=0.1, bottom=0.1)
789        if len(label) > 0 and xpos > 0 and ypos > 0:
790            new_text = self.subplot.text(str(xpos), str(ypos), str(label),
791                                           fontproperties=font)
792            self.textList.append(new_text)
793
794    def erase_legend(self):
795        """
796        Remove Legend
797        """
798        #for ax in self.axes:
799        self.remove_legend(self.subplot)
800
801    def onMouseMotion(self, event):
802        """
803        Disable dragging 2D image
804        """
805
806    def onWheel(self, event):
807        """
808        """
809
810    def onLeftDown(self, event):
811        """
812        Disables LeftDown
813        """
814
815    def onPick(self, event):
816        """
817        Remove Legend
818        """
819        for ax in self.axes:
820            self.remove_legend(ax)
821
822
823    def draw(self):
824        """
825        Draw
826        """
827        if self.dimension == 3:
828            pass
829        else:
830            self.subplot.figure.canvas.resizing = False
831            self.subplot.tick_params(axis='both', labelsize=9)
832            self.erase_legend()
833            self.subplot.figure.canvas.draw_idle()
834            try:
835                self.figure.delaxes(self.figure.axes[1])
836            except:
837                pass
838
839
840    def onContextMenu(self, event):
841        """
842        Default context menu for a plot panel
843        """
844        id = wx.NewId()
845        slicerpop = wx.Menu()
846        data = self.point
847        if issubclass(data.__class__, Data1D):
848            slicerpop.Append(id, '&Change Scale')
849            wx.EVT_MENU(self, id, self._onProperties)
850        else:
851            slicerpop.Append(id, '&Toggle Linear/Log Scale')
852            wx.EVT_MENU(self, id, self.ontogglescale)
853        try:
854            # mouse event
855            pos_evt = event.GetPosition()
856            pos = self.ScreenToClient(pos_evt)
857        except:
858            # toolbar event
859            pos_x, pos_y = self.toolbar.GetPositionTuple()
860            pos = (pos_x, pos_y + 5)
861        self.PopupMenu(slicerpop, pos)
862
863    def ontogglescale(self, event):
864        """
865        On toggle 2d scale
866        """
867        self._onToggleScale(event)
868        try:
869            # mpl >= 1.1.0
870            self.figure.tight_layout()
871        except:
872            self.figure.subplots_adjust(left=0.1, bottom=0.1)
873        try:
874            self.figure.delaxes(self.figure.axes[1])
875        except:
876            pass
877
878    def _onProperties(self, event):
879        """
880        when clicking on Properties on context menu ,
881        The Property dialog is displayed
882        The user selects a transformation for x or y value and
883        a new plot is displayed
884        """
885        list = []
886        list = self.graph.returnPlottable()
887        if len(list.keys()) > 0:
888            first_item = list.keys()[0]
889            if first_item.x != []:
890                from sas.plottools.PropertyDialog import Properties
891                dial = Properties(self, -1, 'Change Scale')
892                # type of view or model used
893                dial.xvalue.Clear()
894                dial.yvalue.Clear()
895                dial.view.Clear()
896                dial.xvalue.Insert("x", 0)
897                dial.xvalue.Insert("log10(x)", 1)
898                dial.yvalue.Insert("y", 0)
899                dial.yvalue.Insert("log10(y)", 1)
900                dial.view.Insert("--", 0)
901                dial.view.Insert("Linear y vs x", 1)
902                dial.setValues(self.prevXtrans, self.prevYtrans, self.viewModel)
903                dial.Update()
904                if dial.ShowModal() == wx.ID_OK:
905                    self.xLabel, self.yLabel, self.viewModel = dial.getValues()
906                    if self.viewModel == "Linear y vs x":
907                        self.xLabel = "x"
908                        self.yLabel = "y"
909                        self.viewModel = "--"
910                        dial.setValues(self.xLabel, self.yLabel, self.viewModel)
911                    self._onEVT_FUNC_PROPERTY()
912                dial.Destroy()
913
914    def _onEVT_FUNC_PROPERTY(self, remove_fit=True):
915        """
916        Receive the x and y transformation from myDialog,
917        Transforms x and y in View
918        and set the scale
919        """
920        list = []
921        list = self.graph.returnPlottable()
922        # Changing the scale might be incompatible with
923        # currently displayed data (for instance, going
924        # from ln to log when all plotted values have
925        # negative natural logs).
926        # Go linear and only change the scale at the end.
927        self.set_xscale("linear")
928        self.set_yscale("linear")
929        _xscale = 'linear'
930        _yscale = 'linear'
931        for item in list:
932            item.setLabel(self.xLabel, self.yLabel)
933            # control axis labels from the panel itself
934            yname, yunits = item.get_yaxis()
935            xname, xunits = item.get_xaxis()
936            # Goes through all possible scales
937            # Goes through all possible scales
938            if(self.xLabel == "x"):
939                item.transformX(transform.toX, transform.errToX)
940                self.graph._xaxis_transformed("%s" % xname, "%s" % xunits)
941            if(self.xLabel == "log10(x)"):
942                item.transformX(transform.toX_pos, transform.errToX_pos)
943                _xscale = 'log'
944                self.graph._xaxis_transformed("%s" % xname, "%s" % xunits)
945            if(self.yLabel == "y"):
946                item.transformY(transform.toX, transform.errToX)
947                self.graph._yaxis_transformed("%s" % yname, "%s" % yunits)
948            if(self.yLabel == "log10(y)"):
949                item.transformY(transform.toX_pos, transform.errToX_pos)
950                _yscale = 'log'
951                self.graph._yaxis_transformed("%s" % yname, "%s" % yunits)
952            item.transformView()
953        self.prevXtrans = self.xLabel
954        self.prevYtrans = self.yLabel
955        self.set_xscale(_xscale)
956        self.set_yscale(_yscale)
957        self.draw()
958
959class DataOperatorWindow(widget.CHILD_FRAME):
960    def __init__(self, parent, manager, *args, **kwds):
961        kwds["size"] = (PANEL_WIDTH, PANEL_HEIGTH)
962        widget.CHILD_FRAME.__init__(self, parent, *args, **kwds)
963        self.parent = parent
964        self.manager = manager
965        self.panel = DataOperPanel(parent=self)
966        wx.EVT_CLOSE(self, self.OnClose)
967        self.SetPosition((wx.LEFT, PANEL_TOP))
968        self.Show()
969
970    def OnClose(self, event=None):
971        """
972        On close event
973        """
974        if self.manager != None:
975            self.manager.data_operator_frame = None
976        self.panel.disconnect_panels()
977        self.Destroy()
978
979
980if __name__ == "__main__":
981
982    app = wx.App()
983    widget.CHILD_FRAME = wx.Frame
984    window = DataOperatorWindow(parent=None, data=[], title="Data Editor")
985    app.MainLoop()
986
Note: See TracBrowser for help on using the repository browser.