Changeset 49ab5d7 in sasview for src/sas/perspectives/calculator/data_operator.py

Timestamp:

Mar 4, 2015 1:28:39 PM (9 years ago)

Author:

Doucet, Mathieu <doucetm@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

2f732b0

Parents:

76aed53

Message:

Take care of white spaces (pylint)

File:

• src/sas/perspectives/calculator/data_operator.py (modified) (50 diffs)

src/sas/perspectives/calculator/data_operator.py

@@ -11 +11 @@
 from sas.plottools import transform
 from matplotlib.font_manager import FontProperties
-from sas.guiframe.events import StatusEvent 
+from sas.guiframe.events import StatusEvent
 from sas.perspectives.calculator import calculator_widgets as widget
 from sas.guiframe.documentation_window import DocumentationWindow
-   
+
 #Control panel width 
 if sys.platform.count("win32") > 0:
@@ -28 +28 @@
     FONT_VARIANT = 1
     ON_MAC = True
-      
+
 class DataOperPanel(wx.ScrolledWindow):
     """
@@ -62 +62 @@
         self.fill_oprator_combox()
         self.Bind(wx.EVT_PAINT, self.set_panel_on_focus)
-             
+
     def _define_structure(self):
         """
-        define initial sizer 
+        define initial sizer
         """
         self.main_sizer = wx.BoxSizer(wx.VERTICAL)
@@ -75 +75 @@
         self.name_sizer = wx.StaticBoxSizer(name_box, wx.HORIZONTAL)
         self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
-      
+
     def _layout_name(self):
         """
@@ -81 +81 @@
         """
         new_data_sizer = wx.BoxSizer(wx.VERTICAL)
-        equal_sizer =  wx.BoxSizer(wx.VERTICAL)
+        equal_sizer = wx.BoxSizer(wx.VERTICAL)
         old_data1_sizer = wx.BoxSizer(wx.VERTICAL)
         operator_sizer = wx.BoxSizer(wx.VERTICAL)
         old_data2_sizer = wx.BoxSizer(wx.VERTICAL)
         data2_hori_sizer = wx.BoxSizer(wx.HORIZONTAL)
-        data_name = wx.StaticText(self, -1, 'Output Data Name')  
-        equal_name = wx.StaticText(self, -1, ' =', size=(50, 25)) 
+        data_name = wx.StaticText(self, -1, 'Output Data Name')
+        equal_name = wx.StaticText(self, -1, ' =', size=(50, 25))
         data1_name = wx.StaticText(self, -1, 'Data1')
         operator_name = wx.StaticText(self, -1, 'Operator')
         data2_name = wx.StaticText(self, -1, 'Data2 (or Number)')
-        self.data_namectr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 25), style=wx.TE_PROCESS_ENTER) 
+        self.data_namectr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 25), style=wx.TE_PROCESS_ENTER)
         self.data_namectr.SetToolTipString("Hit 'Enter' key after typing.")
         self.data_namectr.SetValue(str('MyNewDataName'))
-        self.numberctr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/3, 25), style=wx.TE_PROCESS_ENTER) 
+        self.numberctr = wx.TextCtrl(self, -1, size=(_BOX_WIDTH / 3, 25), style=wx.TE_PROCESS_ENTER)
         self.numberctr.SetToolTipString("Hit 'Enter' key after typing.")
         self.numberctr.SetValue(str(1.0))
-        self.data1_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH, 25), 
+        self.data1_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH, 25),
                                       style=wx.CB_READONLY)
-        self.operator_cbox = wx.ComboBox(self, -1, size=(70, 25), 
+        self.operator_cbox = wx.ComboBox(self, -1, size=(70, 25),
                                          style=wx.CB_READONLY)
         operation_tip = "Add: +, Subtract: -, "
@@ -105 +105 @@
         operation_tip += "Append(Combine): | "
         self.operator_cbox.SetToolTipString(operation_tip)
-        self.data2_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH*2/3, 25),
+        self.data2_cbox = wx.ComboBox(self, -1, size=(_BOX_WIDTH * 2 / 3, 25),
                                        style=wx.CB_READONLY)
 
-        self.out_pic = SmallPanel(self, -1, True, 
-                                    size=(_BOX_WIDTH, _BOX_WIDTH), 
+        self.out_pic = SmallPanel(self, -1, True,
+                                    size=(_BOX_WIDTH, _BOX_WIDTH),
                                     style=wx.NO_BORDER)
-        self.equal_pic = SmallPanel(self, -1, True, '=',  
-                                    size=(50, _BOX_WIDTH), 
+        self.equal_pic = SmallPanel(self, -1, True, '=',
+                                    size=(50, _BOX_WIDTH),
                                     style=wx.NO_BORDER)
-        self.data1_pic = SmallPanel(self, -1, True, 
-                                    size=(_BOX_WIDTH, _BOX_WIDTH), 
+        self.data1_pic = SmallPanel(self, -1, True,
+                                    size=(_BOX_WIDTH, _BOX_WIDTH),
                                     style=wx.NO_BORDER)
         self.operator_pic = SmallPanel(self, -1, True, '+',
-                                    size=(70, _BOX_WIDTH), 
+                                    size=(70, _BOX_WIDTH),
                                     style=wx.NO_BORDER)
-        self.data2_pic = SmallPanel(self, -1, True, 
-                                    size=(_BOX_WIDTH, _BOX_WIDTH), 
+        self.data2_pic = SmallPanel(self, -1, True,
+                                    size=(_BOX_WIDTH, _BOX_WIDTH),
                                     style=wx.NO_BORDER)
         for ax in self.equal_pic.axes:
@@ -144 +144 @@
                                        (data2_hori_sizer, 0, wx.LEFT, 3),
                                        (self.data2_pic, 0, wx.LEFT, 3)])
-        self.name_sizer.AddMany([(new_data_sizer, 0, wx.LEFT|wx.TOP, 5),
+        self.name_sizer.AddMany([(new_data_sizer, 0, wx.LEFT | wx.TOP, 5),
                                        (equal_sizer, 0, wx.TOP, 5),
                                        (old_data1_sizer, 0, wx.TOP, 5),
@@ -152 +152 @@
 
         self._show_numctrl(self.numberctr, False)
-        
+
         wx.EVT_TEXT_ENTER(self.data_namectr, -1, self.on_name)
-        wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number) 
-        wx.EVT_COMBOBOX(self.data1_cbox, -1, self.on_select_data1) 
-        wx.EVT_COMBOBOX(self.operator_cbox, -1, self.on_select_operator) 
+        wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number)
+        wx.EVT_COMBOBOX(self.data1_cbox, -1, self.on_select_data1)
+        wx.EVT_COMBOBOX(self.operator_cbox, -1, self.on_select_operator)
         wx.EVT_COMBOBOX(self.data2_cbox, -1, self.on_select_data2)
-    
-    def _show_numctrl(self, ctrl, enable=True):  
+
+    def _show_numctrl(self, ctrl, enable=True):
         """
         Show/Hide on Win
@@ -170 +170 @@
                 ctrl.GetChildren()[0].SetBackGroundColour(self.color)
             if enable:
-                wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number) 
+                wx.EVT_TEXT_ENTER(self.numberctr, -1, self.on_number)
         else:
             if not ctrl.IsEnabled():
                 ctrl.Enable(True)
             ctrl.Show(enable)
-            
+
     def on_name(self, event=None):
         """
@@ -189 +189 @@
         text = item.GetValue().strip()
         self._check_newname(text)
-    
+
     def _check_newname(self, name=None):
         """
@@ -220 +220 @@
         self.name_sizer.Layout()
         self.Refresh()
-    
-    def _set_textctrl_color(self, ctrl, color): 
-        """
-        Set TextCtrl color 
+
+    def _set_textctrl_color(self, ctrl, color):
+        """
+        Set TextCtrl color
         """
         if ON_MAC:
             children = ctrl.GetChildren()
             if len(children) > 0:
-                children[0].SetBackgroundColour(color)  
-        else:
-            ctrl.SetBackgroundColour(color)  
+                children[0].SetBackgroundColour(color)
+        else:
+            ctrl.SetBackgroundColour(color)
         self.name_sizer.Layout()
-                     
+
     def on_number(self, event=None):
         """
@@ -254 +254 @@
             else:
                 self._set_textctrl_color(self.numberctr, self.color)
-        
+
         self.put_text_pic(self.data2_pic, content=str(val))
         self.check_data_inputs()
@@ -261 +261 @@
         self.draw_output(self.output)
         self.Refresh()
-        
+
     def on_select_data1(self, event=None):
         """
@@ -272 +272 @@
         if data == None:
             content = "?"
-            self.put_text_pic(self.data1_pic, content) 
+            self.put_text_pic(self.data1_pic, content)
         else:
             self.data1_pic.add_image(data)
@@ -279 +279 @@
             self.output.name = str(self.data_namectr.GetValue())
         self.draw_output(self.output)
-        
+
     def on_select_operator(self, event=None):
         """
@@ -287 +287 @@
         item = event.GetEventObject()
         text = item.GetValue().strip()
-        self.put_text_pic(self.operator_pic, content=text) 
+        self.put_text_pic(self.operator_pic, content=text)
         self.check_data_inputs()
         if self.output != None:
             self.output.name = str(self.data_namectr.GetValue())
         self.draw_output(self.output)
-        
+
     def on_select_data2(self, event=None):
         """
@@ -300 +300 @@
         item = event.GetEventObject()
         text = item.GetValue().strip().lower()
-        self._show_numctrl(self.numberctr, text=='number')
+        self._show_numctrl(self.numberctr, text == 'number')
         pos = item.GetCurrentSelection()
         data = item.GetClientData(pos)
@@ -310 +310 @@
             else:
                 self.data2_pic.add_image(data)
-            self.check_data_inputs() 
+            self.check_data_inputs()
         else:
             content = str(self.numberctr.GetValue().strip())
@@ -323 +323 @@
             if data != None:
                 self.check_data_inputs()
-                
-            self.put_text_pic(self.data2_pic, content)   
-        
+
+            self.put_text_pic(self.data2_pic, content)
+
         if self.output != None:
             self.output.name = str(self.data_namectr.GetValue())
         self.draw_output(self.output)
-        
-    def put_text_pic(self, pic=None, content=''):  
+
+    def put_text_pic(self, pic=None, content=''):
         """
         Put text to the pic
         """
-        pic.set_content(content) 
+        pic.set_content(content)
         pic.add_text()
         pic.draw()
-                  
+
     def check_data_inputs(self):
         """
@@ -352 +352 @@
         pos2 = self.data2_cbox.GetCurrentSelection()
         data2 = self.data2_cbox.GetClientData(pos2)
-        
+
         if data2 == None:
             self.output = None
@@ -375 +375 @@
                     return flag
             else:
-                self._set_textctrl_color(self.numberctr, self.color )
+                self._set_textctrl_color(self.numberctr, self.color)
         elif data1.__class__.__name__ != data2.__class__.__name__:
             self._set_textctrl_color(self.data1_cbox, 'pink')
@@ -394 +394 @@
             return flag
         return True
-    
+
     def make_data_out(self, data1, data2):
         """
@@ -403 +403 @@
         operator = self.operator_cbox.GetClientData(pos)
         try:
-            exec "output = data1 %s data2"% operator
+            exec "output = data1 %s data2" % operator
         except:
             raise
         return output
-    
-    
+
+
     def draw_output(self, output):
         """
@@ -416 +416 @@
         if output == None:
             content = "?"
-            self.put_text_pic(out, content) 
+            self.put_text_pic(out, content)
         else:
             out.add_image(output)
@@ -422 +422 @@
         self.Layout()
         self.Refresh()
-                    
-    def _layout_button(self):  
+
+    def _layout_button(self):
         """
             Do the layout for the button widgets
-        """ 
-        self.bt_apply = wx.Button(self, -1, "Apply", size=(_BOX_WIDTH/2, -1))
+        """
+        self.bt_apply = wx.Button(self, -1, "Apply", size=(_BOX_WIDTH / 2, -1))
         app_tip = "Generate the Data and send to Data Explorer."
         self.bt_apply.SetToolTipString(app_tip)
         self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
-        
+
         self.bt_help = wx.Button(self, -1, "HELP")
         app_tip = "Get help on Data Operations."
         self.bt_help.SetToolTipString(app_tip)
         self.bt_help.Bind(wx.EVT_BUTTON, self.on_help)
-        
-        self.bt_close = wx.Button(self, -1, 'Close', size=(_BOX_WIDTH/2, -1))
+
+        self.bt_close = wx.Button(self, -1, 'Close', size=(_BOX_WIDTH / 2, -1))
         self.bt_close.Bind(wx.EVT_BUTTON, self.on_close)
         self.bt_close.SetToolTipString("Close this panel.")
-        
-        self.button_sizer.AddMany([(PANEL_WIDTH/2, 25),
+
+        self.button_sizer.AddMany([(PANEL_WIDTH / 2, 25),
                                    (self.bt_apply, 0, wx.RIGHT, 10),
                                    (self.bt_help, 0, wx.RIGHT, 10),
                                    (self.bt_close, 0, wx.RIGHT, 10)])
-        
+
     def _do_layout(self):
         """
@@ -453 +453 @@
         self._layout_name()
         self._layout_button()
-        self.main_sizer.AddMany([(self.name_sizer, 0, wx.EXPAND|wx.ALL, 10),
+        self.main_sizer.AddMany([(self.name_sizer, 0, wx.EXPAND | wx.ALL, 10),
                                 (self.button_sizer, 0,
-                                          wx.EXPAND|wx.TOP|wx.BOTTOM, 5)])
+                                          wx.EXPAND | wx.TOP | wx.BOTTOM, 5)])
         self.SetSizer(self.main_sizer)
         self.SetScrollbars(20, 20, 25, 65)
         self.SetAutoLayout(True)
-    
+
     def set_panel_on_focus(self, event):
         """
@@ -474 +474 @@
             if self.FindFocus() in children:
                 self.fill_data_combox()
-         
+
     def fill_oprator_combox(self):
         """
         fill the current combobox with the operator
-        """   
+        """
         operator_list = [' +', ' -', ' *', " /", " |"]
         for oper in operator_list:
@@ -484 +484 @@
             self.operator_cbox.SetClientData(pos, str(oper.strip()))
         self.operator_cbox.SetSelection(0)
-        
-        
+
+
     def fill_data_combox(self):
         """
@@ -545 +545 @@
                     self.data2_cbox.SetClientData(pos2, data)
                     if str(current1) == str(name):
-                      pos_pre1 = pos1 
+                      pos_pre1 = pos1
                     if str(current2) == str(name):
                       pos_pre2 = pos2
@@ -561 +561 @@
                             pos_pre2 = posth2
                 except:
-                    continue 
+                    continue
         self.data1_cbox.SetSelection(pos_pre1)
         self.data2_cbox.SetSelection(pos_pre2)
-    
+
     def get_datalist(self):
         """
@@ -573 +573 @@
         else:
             return {}
-            
+
     def on_click_apply(self, event):
-        """   
+        """
         changes are saved in data object imported to edit
         """
@@ -616 +616 @@
         #must post event here
         event.Skip()
-    
-    def on_help(self, event):    
+
+    def on_help(self, event):
         """
         Bring up the Data Operations Panel Documentation whenever
-        the HELP button is clicked. 
-        
+        the HELP button is clicked.
+
         Calls DocumentationWindow with the path of the location within the
-        documentation tree (after /doc/ ....".  Note that when using old 
-        versions of Wx (before 2.9) and thus not the release version of 
-        installers, the help comes up at the top level of the file as 
+        documentation tree (after /doc/ ....".  Note that when using old
+        versions of Wx (before 2.9) and thus not the release version of
+        installers, the help comes up at the top level of the file as
         webbrowser does not pass anything past the # to the browser when it is
         running "file:///...."
-    
+
     :param evt: Triggers on clicking the help button
     """
-                
+
         _TreeLocation = "user/perspectives/calculator/data_operator_help.html"
         _doc_viewer = DocumentationWindow(self, -1, \
-             _TreeLocation,"Data Operation Help")
+             _TreeLocation, "Data Operation Help")
 
     def disconnect_panels(self):
@@ -644 +644 @@
         self.operator_pic.connect.disconnect()
         self.data2_pic.connect.disconnect()
-    
+
     def on_close(self, event):
         """
@@ -650 +650 @@
         """
         self.parent.OnClose()
-        
+
     def set_plot_unfocus(self):
         """
         Unfocus on right click
         """
-    
+
     def send_warnings(self, msg='', info='info'):
         """
@@ -661 +661 @@
         """
         wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info=info))
-          
+
 class SmallPanel(PlotPanel):
     """
     PlotPanel for Quick plot and masking plot
     """
-    def __init__(self, parent, id=-1, is_number=False, content='?', **kwargs):
-        """
-        """ 
+    def __init__(self, parent, id= -1, is_number=False, content='?', **kwargs):
+        """
+        """
         PlotPanel.__init__(self, parent, id=id, **kwargs)
         self.is_number = is_number
@@ -682 +682 @@
         self.add_text()
         self.figure.subplots_adjust(left=0.1, bottom=0.1)
-        
+
     def set_content(self, content=''):
         """
@@ -688 +688 @@
         """
         self.content = str(content)
-         
+
     def add_toolbar(self):
-        """ 
+        """
         Add toolbar
         """
         # Not implemented
         pass
-    
+
     def on_set_focus(self, event):
         """
@@ -730 +730 @@
         #draw        
         self.graph.render(self)
-        
+
         try:
             self.figure.delaxes(self.figure.axes[1])
@@ -738 +738 @@
         self.subplot.tick_params(axis='both', labelsize=9)
         # Draw zero axis lines
-        self.subplot.axhline(linewidth = 1, color='r')  
-        self.subplot.axvline(linewidth = 1, color='r')       
+        self.subplot.axhline(linewidth=1, color='r')
+        self.subplot.axvline(linewidth=1, color='r')
 
         self.erase_legend()
@@ -774 +774 @@
         self.subplot.set_xlim((0, 1))
         self.subplot.set_ylim((0, 1))
-        
+
         try:
             if self.content != '?':
@@ -788 +788 @@
             new_text = self.subplot.text(str(xpos), str(ypos), str(label),
                                            fontproperties=font)
-            self.textList.append(new_text)  
-        
+            self.textList.append(new_text)
+
     def erase_legend(self):
         """
@@ -796 +796 @@
         #for ax in self.axes:
         self.remove_legend(self.subplot)
-                     
+
     def onMouseMotion(self, event):
         """
         Disable dragging 2D image
         """
-    
+
     def onWheel(self, event):
         """
         """
-     
+
     def onLeftDown(self, event):
         """
         Disables LeftDown
         """
-    
+
     def onPick(self, event):
         """
@@ -817 +817 @@
         for ax in self.axes:
             self.remove_legend(ax)
-                        
-    
+
+
     def draw(self):
         """
@@ -827 +827 @@
         else:
             self.subplot.figure.canvas.resizing = False
-            self.subplot.tick_params(axis='both', labelsize=9)     
+            self.subplot.tick_params(axis='both', labelsize=9)
             self.erase_legend()
-            self.subplot.figure.canvas.draw_idle() 
+            self.subplot.figure.canvas.draw_idle()
             try:
                 self.figure.delaxes(self.figure.axes[1])
             except:
                 pass
-            
-       
+
+
     def onContextMenu(self, event):
         """
@@ -848 +848 @@
         else:
             slicerpop.Append(id, '&Toggle Linear/Log Scale')
-            wx.EVT_MENU(self, id, self.ontogglescale)     
+            wx.EVT_MENU(self, id, self.ontogglescale)
         try:
             # mouse event
@@ -858 +858 @@
             pos = (pos_x, pos_y + 5)
         self.PopupMenu(slicerpop, pos)
-        
-    def ontogglescale(self, event): 
+
+    def ontogglescale(self, event):
         """
         On toggle 2d scale
         """
-        self._onToggleScale(event)  
+        self._onToggleScale(event)
         try:
             # mpl >= 1.1.0
             self.figure.tight_layout()
         except:
-            self.figure.subplots_adjust(left=0.1, bottom=0.1) 
+            self.figure.subplots_adjust(left=0.1, bottom=0.1)
         try:
             self.figure.delaxes(self.figure.axes[1])
         except:
             pass
-        
+
     def _onProperties(self, event):
         """
@@ -909 +909 @@
                     self._onEVT_FUNC_PROPERTY()
                 dial.Destroy()
-         
+
     def _onEVT_FUNC_PROPERTY(self, remove_fit=True):
         """
@@ -954 +954 @@
         self.set_yscale(_yscale)
         self.draw()
-        
+
 class DataOperatorWindow(widget.CHILD_FRAME):
     def __init__(self, parent, manager, *args, **kwds):
@@ -965 +965 @@
         self.SetPosition((25, 10))
         self.Show()
-    
-    def OnClose(self, event=None):  
+
+    def OnClose(self, event=None):
         """
         On close event
@@ -975 +975 @@
         self.Destroy()
 
-        
+
 if __name__ == "__main__":
 
-    app  = wx.App()
+    app = wx.App()
     widget.CHILD_FRAME = wx.Frame
     window = DataOperatorWindow(parent=None, data=[], title="Data Editor")
     app.MainLoop()
- 
+