source: sasview/sansview/perspectives/fitting/fitpanel.py @ 10f3b23

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 10f3b23 was 10f3b23, checked in by Gervaise Alina <gervyh@…>, 14 years ago

coming remove hintpage combbox
  • Property mode set to 100644
File size: 17.8 KB
Line 
1
2import numpy
3import string 
4import wx
5import wx.aui
6from sans.guicomm.events import NewPlotEvent, StatusEvent 
7import basepage
8
9_BOX_WIDTH = 80
10
11
12class StateIterator(object):
13    """
14    Contains all saved state of a given page.
15    Provide position of the current state of a page, the first save state
16    and the last state for a given page.
17    Allow easy undo or redo for a given page 
18    """
19    def __init__(self):
20        """
21        """
22        self._current=0
23       
24    def __iter__(self):
25        """
26        """
27        return self
28   
29    def first(self):
30        """
31        """
32        self._current =0
33        return self._current
34   
35    def next(self, max ):
36        """
37        """
38        if self._current < max:
39            self._current += 1
40        return self._current
41   
42    def previous(self):
43        """
44        """
45        if self._current > 0:
46            self._current = self._current -1
47        return self._current
48   
49    def currentPosition(self):
50        """
51        """
52        return self._current
53   
54    def setPosition(self, value):
55        """
56        """
57        if value >=0:
58            self._current = int(value)
59       
60   
61class ListOfState(list):   
62    """
63    """ 
64    def __init__(self, *args, **kw):
65        list.__init__(self, *args, **kw)
66        self.iterator = StateIterator()
67       
68    def appendItem(self, x):
69        """
70        """
71        self.append(x)
72        self.iterator.setPosition(value= len(self)-1)
73       
74    def removeItem(self, x):
75        """
76        """
77        self.iterator.previous()
78        self.remove(x)
79       
80    def getPreviousItem(self):
81        """
82        """
83        position = self.iterator.previous()
84       
85        if position < 0:
86            return None
87        else:
88            return self[position]
89       
90    def getNextItem(self):
91        """
92        """
93        position = self.iterator.next(max= len(self)-1)
94        if position >= len(self):
95            return None
96        else:
97            return self[position]
98       
99    def getCurrentItem(self):
100        """
101        """
102        postion = self.iterator.currentPosition()
103        if postion >= 0 and position < len(self):
104            return self[postion]
105        else:
106            return None
107       
108    def getCurrentPosition(self):
109        """
110        """
111        return self.iterator.currentPosition()
112         
113
114class PageInfo(object):
115    """
116    this class contains the minimum numbers of data members
117    a fitpage or model page need to be initialized.
118    """
119    data = None
120    model =  None
121    manager = None
122    event_owner= None
123    model_list_box = None
124    name = None
125    ## Internal name for the AUI manager
126    window_name = "Page"
127    ## Title to appear on top of the window
128    window_caption = "Page"
129    #type of page can be real data , theory 1D or therory2D
130    type = "Data"
131    def __init__(self, model=None, data=None, manager=None,
132                  event_owner=None, model_list_box=None, name=None):
133        """
134        Initialize data members
135        """
136        self.data = data
137        self.model= model
138        self.manager= manager
139        self.event_owner= event_owner
140        self.model_list_box = model_list_box
141        self.name=None
142        self.window_name = "Page"
143        self.window_caption = "Page"
144        self.type = "Data"
145       
146class FitPanel(wx.aui.AuiNotebook):   
147
148    """
149    FitPanel class contains fields allowing to fit  models and  data
150   
151    :note: For Fit to be performed the user should check at least one parameter
152        on fit Panel window.
153       
154    """
155    ## Internal name for the AUI manager
156    window_name = "Fit panel"
157    ## Title to appear on top of the window
158    window_caption = "Fit Panel "
159    CENTER_PANE = True
160   
161    def __init__(self, parent, *args, **kwargs):
162        """
163        """
164        wx.aui.AuiNotebook.__init__(self, parent, -1,
165                    style= wx.aui.AUI_NB_WINDOWLIST_BUTTON|
166                    wx.aui.AUI_NB_DEFAULT_STYLE|
167                    wx.CLIP_CHILDREN)
168   
169        self.manager=None
170        self.parent=parent
171        self.event_owner=None
172       
173        pageClosedEvent = wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE
174        self.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE, self.on_close_page)
175       
176        #dictionary of miodel {model class name, model class}
177        self.model_list_box = {}
178        ## save the title of the last page tab added
179        self.fit_page_name = {}
180        ## list of existing fit page
181        self.opened_pages = {}
182        #page of simultaneous fit
183        self.sim_page = None
184        ## get the state of a page
185        self.Bind(basepage.EVT_PAGE_INFO, self._onGetstate)
186        self.Bind(basepage.EVT_PREVIOUS_STATE, self._onUndo)
187        self.Bind(basepage.EVT_NEXT_STATE, self._onRedo)
188       
189       #add default page
190        from hint_fitpage import HintFitPage
191        self.hint_page = HintFitPage(self) 
192        self.AddPage(page=self.hint_page, caption="Hint")
193        #Add the first fit page
194        self.add_empty_page()
195       
196        # increment number for model name
197        self.count = 0
198        #updating the panel
199        self.Update()
200        self.Center()
201       
202    def set_state(self, state):
203        """
204        Restore state of the panel
205        """
206        page_is_opened = False
207        if state is not None:
208            page_info = self.get_page_info(data=state.data)
209            if hasattr(state.data,"title"):
210                title = str(state.data.title.lstrip().rstrip())
211                if title == "":
212                    title = str(state.data.name)
213            else:
214                title = str(state.data.name)
215            for name, panel, type in self.opened_pages.values():
216                #Don't return any panel is the exact same page is created
217                if name == page_info.window_name:
218                    # the page is still opened
219                    panel.set_manager(self.manager)
220                    panel.set_owner(self.event_owner)
221                    if panel.model_list_box is None or len(panel.model_list_box) == 0: 
222                        page_info.model_list_box = self.model_list_box.get_list()
223                        panel.populate_box(dict=page_info.model_list_box)
224                        panel.initialize_combox()
225                        panel.set_page_info(page_info=page_info)
226                        self.opened_pages[page_info.window_name] = [page_info.window_name, panel, page_info.type]
227                    if panel is not None: 
228                        self.manager.store_page(page=panel, data=state.data)
229                    panel.reset_page(state=state)
230                    wx.PostEvent(self.parent, NewPlotEvent(plot=state.data, title=title))
231                    page_is_opened = True
232                   
233            if not page_is_opened:
234                panel = self.add_fit_page(data=state.data)
235                # add data associated to the page created
236                if panel is not None: 
237                    self.manager.store_page(page=panel, data=state.data)
238                    panel.reset_page(state=state)
239               
240                   
241    def on_close_page(self, event):
242        """
243        close page and remove all references to the closed page
244        """
245        nbr_page = self.GetPageCount()
246        pos = self.GetSelection()
247        if nbr_page == 1 or self.GetPage(pos)== self.hint_page:
248            event.Veto()
249            return 
250        selected_page = self.GetPage(self.GetSelection())
251        self._close_helper(selected_page=selected_page)
252       
253    def close_page_with_data(self, deleted_data):
254        """
255        close a fit page when its data is completely remove from the graph
256        """
257        if deleted_data is None:
258            return
259        for index in range(self.GetPageCount()):
260            selected_page = self.GetPage(index) 
261            if hasattr(selected_page,"get_data"):
262                data = selected_page.get_data()
263                if data is None:
264                    #the fitpanel exists and only the initial fit page is open
265                    #with no selected data
266                    return
267                if data.name == deleted_data.name:
268                    self._close_helper(selected_page)
269                    self.DeletePage(index)
270                    break
271       
272    def set_manager(self, manager):
273        """
274        set panel manager
275       
276        :param manager: instance of plugin fitting
277       
278        """
279        self.manager = manager
280   
281       
282    def set_owner(self,owner):
283        """
284        set and owner for fitpanel
285       
286        :param owner: the class responsible of plotting
287       
288        """
289        self.event_owner = owner
290   
291    def set_model_list(self, dict):
292         """
293         copy a dictionary of model into its own dictionary
294         
295         :param dict: dictionnary made of model name as key and model class
296             as value
297         """
298         self.model_list_box = dict
299       
300    def get_current_page(self):
301        """
302        :return: the current page selected
303       
304        """
305        return self.GetPage(self.GetSelection() )
306   
307    def add_sim_page(self):
308        """
309        Add the simultaneous fit page
310        """
311        from simfitpage import SimultaneousFitPage
312        page_finder= self.manager.get_page_finder()
313        self.sim_page = SimultaneousFitPage(self,page_finder=page_finder, id=-1)
314       
315        self.AddPage(self.sim_page,caption="Simultaneous Fit",select=True)
316        self.sim_page.set_manager(self.manager)
317        return self.sim_page
318       
319    def get_page_info(self, data=None):
320        """
321        fill information required to add a page in the fit panel
322        """
323        name = "Fit Page"
324        type = 'empty'
325        if data is not None and hasattr(data, "is_data"):
326            if data.is_data:
327                name = data.name
328                type = 'Data'
329            else:
330                if data.__class__.__name__ == "Data2D":
331                    name = 'Model 2D Fit'
332                    type = 'Theory2D'
333                else:
334                    name = 'Model 1D Fit'
335                    type = 'Theory1D'
336        page_info = PageInfo(data=data, name=name)
337        page_info.event_owner = self.event_owner
338        page_info.manager = self.manager
339        page_info.window_name = name
340        page_info.window_caption = name
341        page_info.type = type
342        return page_info
343   
344    def add_empty_page(self):
345        """
346        add an empty page
347        """
348        page_info = self.get_page_info()
349        from fitpage import FitPage
350        panel = FitPage(parent=self, page_info=page_info)
351        self.AddPage(page=panel, caption=page_info.window_name, select=True)
352        self.opened_pages[page_info.window_name] = [page_info.window_name, panel, page_info.type]
353        return panel
354   
355    def add_page(self, page_info):
356        """
357        add a new page
358        """
359        from fitpage import FitPage
360        panel = FitPage(parent=self, page_info=page_info)
361        self.AddPage(page=panel, caption=page_info.window_name, select=True)
362        index = self.GetPageIndex(panel)
363        self.change_page_content(data=page_info.data, index=index)
364        return panel
365   
366    def change_page_content(self, data, index):
367        """
368        replace the contains of an existing page
369        """
370        page_info = self.get_page_info(data=data)
371        self.SetPageText(index, page_info.window_name)
372        panel = self.GetPage(index)
373        panel.set_data(data)
374        if panel.model_list_box is None or len(panel.model_list_box) == 0: 
375            page_info.model_list_box = self.model_list_box.get_list()
376            panel.populate_box(dict=page_info.model_list_box)
377            panel.initialize_combox()
378        panel.set_page_info(page_info=page_info)
379        self.opened_pages[page_info.window_name] = [page_info.window_name, panel, page_info.type]
380        return panel
381   
382    def replace_page(self, index, page_info, type):
383        """
384        replace an existing page
385        """
386        self.DeletePage(index)
387        del self.opened_pages[page_info.window_name]
388        return self.add_page(page_info=page_info)
389       
390    def add_fit_page(self, data, reset=False):
391        """
392        Add a fitting page on the notebook contained by fitpanel
393       
394        :param data: data to fit
395       
396        :return panel : page just added for further used. is used by fitting module
397       
398        """
399        if data is None:
400            return None
401        page_info = self.get_page_info(data=data)
402        type = page_info.type
403        npages = len(self.opened_pages.keys())
404        if npages == 0:
405            # create new type of page
406            panel = self.add_page(page_info=page_info)
407            return panel
408        else:
409            if page_info.window_name not in self.opened_pages.keys() and \
410                page_info.type.lower() in ['data']:
411                #create new data type page
412                panel = self.add_page(page_info=page_info)
413                return panel
414            elif page_info.type.lower() not in ['data']:
415                #create new theory page
416                for open_tab_caption, value in self.opened_pages.iteritems():
417                    open_tab = value[1]
418                    open_type = value[2]
419                    if page_info.type == open_type:
420                        #replace theory fit tab
421                        existing_value = self.opened_pages[open_tab_caption]
422                        panel = existing_value[1]
423                        #delete the existing page and replace it
424                        index = self.GetPageIndex(panel)
425                        panel = self.replace_page(index=index, page_info=page_info,
426                                                   type=type)
427                        return panel
428                panel = self.add_page(page_info=page_info)
429                return panel
430            else:
431                _, panel, _ = self.opened_pages[page_info.window_name]
432                index = self.GetPageIndex(panel)
433                self.SetSelection(index)
434                #page already existing
435                return None
436             
437    def  _onGetstate(self, event):
438        """
439        copy the state of a page
440        """
441        page= event.page
442        if page.window_name in self.fit_page_name:
443            self.fit_page_name[page.window_name].appendItem(page.createMemento()) 
444           
445    def _onUndo(self, event ):
446        """
447        return the previous state of a given page is available
448        """
449        page = event.page
450        if page.window_name in self.fit_page_name:
451            if self.fit_page_name[page.window_name].getCurrentPosition()==0:
452                state = None
453            else:
454                state = self.fit_page_name[page.window_name].getPreviousItem()
455                page._redo.Enable(True)
456            page.reset_page(state)
457       
458    def _onRedo(self, event): 
459        """
460        return the next state available
461        """       
462        page = event.page
463        if page.window_name in self.fit_page_name:
464            length= len(self.fit_page_name[page.window_name])
465            if self.fit_page_name[page.window_name].getCurrentPosition()== length -1:
466                state = None
467                page._redo.Enable(False)
468                page._redo.Enable(True)
469            else:
470                state = self.fit_page_name[page.window_name].getNextItem()
471            page.reset_page(state) 
472                 
473    def _close_helper(self, selected_page):
474        """
475        Delete the given page from the notebook
476        """
477        #remove hint page
478        if selected_page == self.hint_page:
479            return
480        ## removing sim_page
481        if selected_page == self.sim_page :
482            self.manager.sim_page=None 
483            return
484       
485        ## closing other pages
486        state = selected_page.createMemento()
487        page_name = selected_page.window_name
488        page_finder = self.manager.get_page_finder() 
489        fitproblem = None
490        ## removing fit page
491        data = selected_page.get_data()
492        #Don' t remove plot for 2D
493        flag = True
494        if data.__class__.__name__ == 'Data2D':
495            flag = False
496        if selected_page in page_finder:
497            #Delete the name of the page into the list of open page
498            for type, list in self.opened_pages.iteritems():
499                #Don't return any panel is the exact same page is created
500                name = str(list[0])
501                if flag and selected_page.window_name == name:
502                    if type.lower() in ['theory1d', 'theory2d']:
503                        self.manager.remove_plot(selected_page, theory=True)
504                    else:
505                        self.manager.remove_plot(selected_page, theory=False)
506                    break 
507            del page_finder[selected_page]
508        ##remove the check box link to the model name of this page (selected_page)
509        try:
510            self.sim_page.draw_page()
511        except:
512            ## that page is already deleted no need to remove check box on
513            ##non existing page
514            pass
515               
516        #Delete the name of the page into the list of open page
517        for type, list in self.opened_pages.iteritems():
518            #Don't return any panel is the exact same page is created
519            name = str(list[0])
520            if selected_page.window_name == name:
521                del self.opened_pages[type]
522                break 
523     
524 
Note: See TracBrowser for help on using the repository browser.