Ignore:
Timestamp:
Apr 27, 2012 2:20:33 PM (12 years ago)
Author:
Mathieu Doucet <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:
f992a06
Parents:
9e07bc4
Message:

Pep-8-ification

File:
1 edited

Legend:

Unmodified
Added
Removed

  • fittingview/src/sans/perspectives/fitting/fitpanel.py

    r4316e85 rf32d144  
     1""" 
     2FitPanel class contains fields allowing to fit  models and  data 
    13 
    2 import numpy 
    3 import string  
     4:note: For Fit to be performed the user should check at least one parameter 
     5    on fit Panel window. 
     6    
     7""" 
    48import wx 
    5 import sys 
    6 #from wx.lib.flatnotebook import FlatNotebook as nb 
    79from wx.aui import AuiNotebook as nb 
    810 
    … …  
    1719 
    1820 
    19 class FitPanel(nb, PanelBase):     
    20  
     21class FitPanel(nb, PanelBase): 
    2122    """ 
    2223    FitPanel class contains fields allowing to fit  models and  data 
    … …  
    3637        """ 
    3738        nb.__init__(self, parent, -1, 
    38                     style= wx.aui.AUI_NB_WINDOWLIST_BUTTON| 
     39                    style=wx.aui.AUI_NB_WINDOWLIST_BUTTON| 
    3940                    wx.aui.AUI_NB_DEFAULT_STYLE| 
    4041                    wx.CLIP_CHILDREN) 
    … …  
    4748        self.menu_mng = models.ModelManager() 
    4849        self.model_list_box = self.menu_mng.get_model_list() 
    49         #pageClosedEvent = nb.EVT_FLATNOTEBOOK_PAGE_CLOSING  
     50        #pageClosedEvent = nb.EVT_FLATNOTEBOOK_PAGE_CLOSING 
    5051        self.pageClosedEvent = wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE 
    5152         
    5253        self.Bind(self.pageClosedEvent, self.on_close_page) 
    53          ## save the title of the last page tab added 
     54        ## save the title of the last page tab added 
    5455        self.fit_page_name = {} 
    5556        ## list of existing fit page 
    5657        self.opened_pages = {} 
    57         #index of fit page  
     58        #index of fit page 
    5859        self.fit_page_index = 0 
    5960        #index of batch page 
    6061        self.batch_page_index = 0 
    61         #page of simultaneous fit  
     62        #page of simultaneous fit 
    6263        self.sim_page = None 
    6364        self.batch_page = None 
    … …  
    9394                state = page.get_state().clone() 
    9495                if data is not None: 
    95                     new_doc = self._manager.state_reader.write_toXML(data, state) 
     96                    new_doc = self._manager.state_reader.write_toXML(data, 
     97                                                                     state) 
    9698                    if doc != None and hasattr(doc, "firstChild"): 
    9799                        child = new_doc.firstChild.firstChild 
    98                         doc.firstChild.appendChild(child)   
     100                        doc.firstChild.appendChild(child) 
    99101                    else: 
    100                         doc = new_doc  
     102                        doc = new_doc 
    101103        if msg.strip() != "": 
    102104            temp = "Save Project is not supported for Batch page.\n" 
    … …  
    105107            wx.PostEvent(self._manager.parent, StatusEvent(status=message, 
    106108                                                            info="warning")) 
    107         return doc     
     109        return doc 
    108110     
    109111    def _on_engine_change(self, name='scipy'): 
    … …  
    136138        return temp 
    137139     
    138     def get_page_by_id(self, uid):   
     140    def get_page_by_id(self, uid): 
    139141        """ 
    140142        """ 
    … …  
    151153        """ 
    152154        wx.CallAfter(self.helper_on_page_change) 
    153          
    154155         
    155156    def helper_on_page_change(self): 
    … …  
    197198            return selected_page.get_state() 
    198199     
    199    
    200    
    201200    def close_all(self): 
    202201        """ 
    … …  
    207206        nop = self.GetPageCount() 
    208207        #use while-loop, for-loop will not do the job well. 
    209         while (nop>0): 
     208        while (nop > 0): 
    210209            #delete the first page until no page exists 
    211210            page = self.GetPage(0) 
    … …  
    219218        self.fit_page_name = {} 
    220219        ## list of existing fit page 
    221         self.opened_pages = {}   
     220        self.opened_pages = {} 
    222221          
    223          
    224222    def set_state(self, state): 
    225223        """ 
    … …  
    233231                    # the page is still opened 
    234232                    panel.reset_page(state=state) 
    235                     panel.save_current_state()  
     233                    panel.save_current_state() 
    236234                    page_is_opened = True 
    237235            if not page_is_opened: 
    … …  
    244242                panel = self._manager.add_fit_page(data=list_data) 
    245243                # add data associated to the page created 
    246                 if panel is not None:   
    247                     self._manager.store_data(uid=panel.uid,  
     244                if panel is not None: 
     245                    self._manager.store_data(uid=panel.uid, 
    248246                                             data_list=list_data, 
    249247                                             caption=panel.window_caption) 
    … …  
    260258        self._manager.mypanels = [] 
    261259         
    262                         
    263260    def on_close_page(self, event=None): 
    264261        """ 
    … …  
    271268                if event is not None: 
    272269                    event.Veto() 
    273                 return  
     270                return 
    274271        self._close_helper(selected_page=selected_page) 
    275272         
    … …  
    281278            return 
    282279        for index in range(self.GetPageCount()): 
    283             selected_page = self.GetPage(index)  
    284             if hasattr(selected_page,"get_data"): 
     280            selected_page = self.GetPage(index) 
     281            if hasattr(selected_page, "get_data"): 
    285282                data = selected_page.get_data() 
    286283                 
    287284                if data is None: 
    288                     #the fitpanel exists and only the initial fit page is open  
     285                    #the fitpanel exists and only the initial fit page is open 
    289286                    #with no selected data 
    290287                    return 
    … …  
    308305 
    309306    def set_model_list(self, dict): 
    310          """  
    311          copy a dictionary of model into its own dictionary 
    312           
    313          :param dict: dictionnary made of model name as key and model class 
    314              as value 
    315          """ 
    316          self.model_list_box = dict 
     307        """ 
     308        copy a dictionary of model into its own dictionary 
     309         
     310        :param dict: dictionnary made of model name as key and model class 
     311        as value 
     312        """ 
     313        self.model_list_box = dict 
    317314         
    318315    def get_current_page(self): 
    … …  
    330327        page_finder = self._manager.get_page_finder() 
    331328        if caption == "Simultaneous Fit": 
    332             self.sim_page = SimultaneousFitPage(self,page_finder=page_finder, 
     329            self.sim_page = SimultaneousFitPage(self, page_finder=page_finder, 
    333330                                                 id=-1, batch_on=False) 
    334331            self.sim_page.window_caption = caption 
    … …  
    349346            self.enable_close_button() 
    350347            return self.batch_page 
    351          
    352348  
    353349    def add_empty_page(self): 
    … …  
    388384        self.enable_close_button() 
    389385        panel.on_set_focus(None) 
    390         return panel  
     386        return panel 
    391387     
    392388    def enable_close_button(self): 
    393389        """ 
    394         display the close button on tab for more than 1 tabs else remove the  
     390        display the close button on tab for more than 1 tabs else remove the 
    395391        close button 
    396392        """ 
    397393        if self.GetPageCount() <= 1: 
    398             style = self.GetWindowStyleFlag()  
     394            style = self.GetWindowStyleFlag() 
    399395            flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB 
    400396            if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB == flag: 
    … …  
    424420                    temp = self.GetSelection() 
    425421                    self.DeletePage(temp) 
    426             if self.GetPageCount()== 0: 
     422            if self.GetPageCount() == 0: 
    427423                self._manager.on_add_new_page(event=None) 
    428424         
    … …  
    432428        However all data in the set of data must be either 1D or 2D type.  
    433429        This method presents option to select the data type before creating a  
    434         tab.  
     430        tab. 
    435431        """ 
    436432        data_1d_list = [] 
    … …  
    447443        page = None 
    448444        for p in self.opened_pages.values(): 
    449             #check if there is an empty page to fill up  
     445            #check if there is an empty page to fill up 
    450446            if not check_data_validity(p.get_data()) and p.batch_on: 
    451447                 
    … …  
    473469            else: 
    474470                #the batch analysis is canceled 
    475                  dlg.Destroy() 
    476                  return None 
     471                dlg.Destroy() 
     472                return None 
    477473        else: 
    478474            if page is None: 
    479475                page = self.add_empty_page() 
    480476            if data_1d_list and not data_2d_list: 
    481                 #only on type of data  
     477                #only on type of data 
    482478                page.fill_data_combobox(data_1d_list) 
    483479            elif not data_1d_list and data_2d_list: 
    … …  
    492488     
    493489    def set_data(self, data_list): 
    494         """  
     490        """ 
    495491        Add a fitting page on the notebook contained by fitpanel 
    496492         
    497493        :param data: data to fit 
    498494         
    499         :return panel : page just added for further used.  
     495        :return panel : page just added for further used. 
    500496        is used by fitting module 
    501497         
    … …  
    504500            return None 
    505501        if self.batch_on: 
    506            return self.set_data_on_batch_mode(data_list) 
     502            return self.set_data_on_batch_mode(data_list) 
    507503        else: 
    508504            data = None 
    … …  
    546542        page = event.page 
    547543        if page.uid in self.fit_page_name: 
    548            self.fit_page_name[page.uid].appendItem(page.createMemento())  
     544            self.fit_page_name[page.uid].appendItem(page.createMemento()) 
    549545             
    550546    def _onUndo(self, event): 
    … …  
    552548        return the previous state of a given page is available 
    553549        """ 
    554         page = event.page  
     550        page = event.page 
    555551        if page.uid in self.fit_page_name: 
    556             if self.fit_page_name[page.uid].getCurrentPosition()==0: 
     552            if self.fit_page_name[page.uid].getCurrentPosition() == 0: 
    557553                state = None 
    558554            else: 
    … …  
    561557            page.reset_page(state) 
    562558         
    563     def _onRedo(self, event):  
     559    def _onRedo(self, event): 
    564560        """ 
    565561        return the next state available 
    566         """        
    567         page = event.page  
     562        """ 
     563        page = event.page 
    568564        if page.uid in self.fit_page_name: 
    569             length= len(self.fit_page_name[page.uid]) 
    570             if self.fit_page_name[page.uid].getCurrentPosition()== length -1: 
     565            length = len(self.fit_page_name[page.uid]) 
     566            if self.fit_page_name[page.uid].getCurrentPosition() == length - 1: 
    571567                state = None 
    572568                page._redo.Enable(False) 
    573569                page._redo.Enable(True) 
    574570            else: 
    575                 state =self.fit_page_name[page.uid].getNextItem() 
    576             page.reset_page(state)   
     571                state = self.fit_page_name[page.uid].getNextItem() 
     572            page.reset_page(state) 
    577573                  
    578574    def _close_helper(self, selected_page): 
    … …  
    584580        #    return 
    585581        ## removing sim_page 
    586         if selected_page == self.sim_page : 
    587             self._manager.sim_page=None  
     582        if selected_page == self.sim_page: 
     583            self._manager.sim_page = None 
    588584            return 
    589         if selected_page == self.batch_page : 
    590             self._manager.batch_page=None  
     585        if selected_page == self.batch_page: 
     586            self._manager.batch_page = None 
    591587            return 
    592588        """ 
    … …  
    599595        ## closing other pages 
    600596        state = selected_page.createMemento() 
    601         page_name = selected_page.window_name 
    602         page_finder = self._manager.get_page_finder()  
    603         fitproblem = None 
     597        page_finder = self._manager.get_page_finder() 
    604598        ## removing fit page 
    605599        data = selected_page.get_data() 
    … …  
    615609                if flag and selected_page.uid == uid: 
    616610                    self._manager.remove_plot(uid, theory=False) 
    617                     break  
     611                    break 
    618612            del page_finder[selected_page] 
    619  
    620                  
     613   
    621614        #Delete the name of the page into the list of open page 
    622615        for uid, list in self.opened_pages.iteritems(): 
    … …  
    624617            if selected_page.uid == uid: 
    625618                del self.opened_pages[selected_page.uid] 
    626                 break  
     619                break 
    627620        ##remove the check box link to the model name of this page (selected_page) 
    628621        try: 
    … …  
    631624            ## that page is already deleted no need to remove check box on 
    632625            ##non existing page 
    633             pass      
     626            pass 
    634627        try: 
    635628            self.batch_page.draw_page() 
    … …  
    637630            ## that page is already deleted no need to remove check box on 
    638631            ##non existing page 
    639             pass      
    640          
     632            pass 
Note: See TracChangeset for help on using the changeset viewer.