source: sasview/sansview/perspectives/fitting/fitpanel.py @ 8898e963

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

working on guiframe and combobox save sate
  • Property mode set to 100644
File size: 17.9 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="Loaded Data")
193        # increment number for model name
194        self.count = 0
195        #updating the panel
196        self.Update()
197        self.Center()
198       
199    def set_data(self, list=[], state=None):
200        """
201        Receive a list of data from data manager
202        """
203        if len(list) ==0:
204            return 
205        self.hint_page.set_data(list=list)
206     
207    def set_state(self, state):
208        """
209        Restore state of the panel
210        """
211        page_is_opened = False
212        if state is not None:
213            page_info = self.get_page_info(data=state.data)
214            if hasattr(state.data,"title"):
215                title = str(state.data.title.lstrip().rstrip())
216                if title == "":
217                    title = str(state.data.name)
218            else:
219                title = str(state.data.name)
220            for name, panel, type in self.opened_pages.values():
221                #Don't return any panel is the exact same page is created
222                if name == page_info.window_name:
223                    # the page is still opened
224                    panel.set_manager(self.manager)
225                    panel.set_owner(self.event_owner)
226                    if panel.model_list_box is None or len(panel.model_list_box) == 0: 
227                        page_info.model_list_box = self.model_list_box.get_list()
228                        panel.populate_box(dict=page_info.model_list_box)
229                        panel.initialize_combox()
230                        panel.set_page_info(page_info=page_info)
231                        self.opened_pages[page_info.window_name] = [page_info.window_name, panel, page_info.type]
232                    if panel is not None: 
233                        self.manager.store_page(page=panel, data=state.data)
234                    panel.reset_page(state=state)
235                    wx.PostEvent(self.parent, NewPlotEvent(plot=state.data, title=title))
236                    page_is_opened = True
237                   
238            if not page_is_opened:
239                panel = self.add_fit_page(data=state.data)
240                # add data associated to the page created
241                if panel is not None: 
242                    self.manager.store_page(page=panel, data=state.data)
243                    panel.reset_page(state=state)
244               
245                   
246    def on_close_page(self, event):
247        """
248        close page and remove all references to the closed page
249        """
250        nbr_page = self.GetPageCount()
251        pos = self.GetSelection()
252        if nbr_page == 1 or self.GetPage(pos)== self.hint_page:
253            event.Veto()
254            return 
255        selected_page = self.GetPage(self.GetSelection())
256        self._close_helper(selected_page=selected_page)
257       
258    def close_page_with_data(self, deleted_data):
259        """
260        close a fit page when its data is completely remove from the graph
261        """
262        if deleted_data is None:
263            return
264        for index in range(self.GetPageCount()):
265            selected_page = self.GetPage(index) 
266            if hasattr(selected_page,"get_data"):
267                data = selected_page.get_data()
268                if data is None:
269                    #the fitpanel exists and only the initial fit page is open
270                    #with no selected data
271                    return
272                if data.name == deleted_data.name:
273                    self._close_helper(selected_page)
274                    self.DeletePage(index)
275                    break
276       
277    def set_manager(self, manager):
278        """
279        set panel manager
280       
281        :param manager: instance of plugin fitting
282       
283        """
284        self.manager = manager
285
286       
287    def set_owner(self,owner):
288        """
289        set and owner for fitpanel
290       
291        :param owner: the class responsible of plotting
292       
293        """
294        self.event_owner = owner
295   
296    def set_model_list(self, dict):
297         """
298         copy a dictionary of model into its own dictionary
299         
300         :param dict: dictionnary made of model name as key and model class
301             as value
302         """
303         self.model_list_box = dict
304       
305    def get_current_page(self):
306        """
307        :return: the current page selected
308       
309        """
310        return self.GetPage(self.GetSelection() )
311   
312    def add_sim_page(self):
313        """
314        Add the simultaneous fit page
315        """
316        from simfitpage import SimultaneousFitPage
317        page_finder= self.manager.get_page_finder()
318        self.sim_page = SimultaneousFitPage(self,page_finder=page_finder, id=-1)
319       
320        self.AddPage(self.sim_page,caption="Simultaneous Fit",select=True)
321        self.sim_page.set_manager(self.manager)
322        return self.sim_page
323       
324    def get_page_info(self, data=None):
325        """
326        fill information required to add a page in the fit panel
327        """
328        name = "Fit Page"
329        type = 'empty'
330        if data is not None and hasattr(data, "is_data"):
331            if data.is_data:
332                name = data.name
333                type = 'Data'
334            else:
335                if data.__class__.__name__ == "Data2D":
336                    name = 'Model 2D Fit'
337                    type = 'Theory2D'
338                else:
339                    name = 'Model 1D Fit'
340                    type = 'Theory1D'
341        page_info = PageInfo(data=data, name=name)
342        page_info.event_owner = self.event_owner
343        page_info.manager = self.manager
344        page_info.window_name = name
345        page_info.window_caption = name
346        page_info.type = type
347        return page_info
348   
349    def add_empty_page(self):
350        """
351        add an empty page
352        """
353        page_info = self.get_page_info()
354        from fitpage import FitPage
355        panel = FitPage(parent=self, page_info=page_info)
356        self.AddPage(page=panel, caption=page_info.window_name, select=True)
357        self.opened_pages[page_info.window_name] = [page_info.window_name, panel, page_info.type]
358        return panel
359   
360    def add_page(self, page_info):
361        """
362        add a new page
363        """
364        from fitpage import FitPage
365        panel = FitPage(parent=self, page_info=page_info)
366        self.AddPage(page=panel, caption=page_info.window_name, select=True)
367        index = self.GetPageIndex(panel)
368        self.change_page_content(data=page_info.data, index=index)
369        return panel
370   
371    def change_page_content(self, data, index):
372        """
373        replace the contains of an existing page
374        """
375        page_info = self.get_page_info(data=data)
376        self.SetPageText(index, page_info.window_name)
377        panel = self.GetPage(index)
378        panel.set_data(data)
379        if panel.model_list_box is None or len(panel.model_list_box) == 0: 
380            page_info.model_list_box = self.model_list_box.get_list()
381            panel.populate_box(dict=page_info.model_list_box)
382            panel.initialize_combox()
383        panel.set_page_info(page_info=page_info)
384        self.opened_pages[page_info.window_name] = [page_info.window_name, panel, page_info.type]
385        return panel
386   
387    def replace_page(self, index, page_info, type):
388        """
389        replace an existing page
390        """
391        self.DeletePage(index)
392        del self.opened_pages[page_info.window_name]
393        return self.add_page(page_info=page_info)
394       
395    def add_fit_page(self, data, reset=False):
396        """
397        Add a fitting page on the notebook contained by fitpanel
398       
399        :param data: data to fit
400       
401        :return panel : page just added for further used. is used by fitting module
402       
403        """
404        if data is None:
405            return None
406        page_info = self.get_page_info(data=data)
407        type = page_info.type
408        npages = len(self.opened_pages.keys())
409        if npages == 0:
410            # create new type of page
411            panel = self.add_page(page_info=page_info)
412            return panel
413        else:
414            if page_info.window_name not in self.opened_pages.keys() and \
415                page_info.type.lower() in ['data']:
416                #create new data type page
417                panel = self.add_page(page_info=page_info)
418                return panel
419            elif page_info.type.lower() not in ['data']:
420                #create new theory page
421                for open_tab_caption, value in self.opened_pages.iteritems():
422                    open_tab = value[1]
423                    open_type = value[2]
424                    if page_info.type == open_type:
425                        #replace theory fit tab
426                        existing_value = self.opened_pages[open_tab_caption]
427                        panel = existing_value[1]
428                        #delete the existing page and replace it
429                        index = self.GetPageIndex(panel)
430                        panel = self.replace_page(index=index, page_info=page_info,
431                                                   type=type)
432                        return panel
433                panel = self.add_page(page_info=page_info)
434                return panel
435            else:
436                _, panel, _ = self.opened_pages[page_info.window_name]
437                index = self.GetPageIndex(panel)
438                self.SetSelection(index)
439                #page already existing
440                return None
441             
442    def  _onGetstate(self, event):
443        """
444        copy the state of a page
445        """
446        page= event.page
447        if page.window_name in self.fit_page_name:
448            self.fit_page_name[page.window_name].appendItem(page.createMemento()) 
449           
450    def _onUndo(self, event ):
451        """
452        return the previous state of a given page is available
453        """
454        page = event.page
455        if page.window_name in self.fit_page_name:
456            if self.fit_page_name[page.window_name].getCurrentPosition()==0:
457                state = None
458            else:
459                state = self.fit_page_name[page.window_name].getPreviousItem()
460                page._redo.Enable(True)
461            page.reset_page(state)
462       
463    def _onRedo(self, event): 
464        """
465        return the next state available
466        """       
467        page = event.page
468        if page.window_name in self.fit_page_name:
469            length= len(self.fit_page_name[page.window_name])
470            if self.fit_page_name[page.window_name].getCurrentPosition()== length -1:
471                state = None
472                page._redo.Enable(False)
473                page._redo.Enable(True)
474            else:
475                state = self.fit_page_name[page.window_name].getNextItem()
476            page.reset_page(state) 
477                 
478    def _close_helper(self, selected_page):
479        """
480        Delete the given page from the notebook
481        """
482        #remove hint page
483        if selected_page == self.hint_page:
484            return
485        ## removing sim_page
486        if selected_page == self.sim_page :
487            self.manager.sim_page=None 
488            return
489       
490        ## closing other pages
491        state = selected_page.createMemento()
492        page_name = selected_page.window_name
493        page_finder = self.manager.get_page_finder() 
494        fitproblem = None
495        ## removing fit page
496        data = selected_page.get_data()
497        #Don' t remove plot for 2D
498        flag = True
499        if data.__class__.__name__ == 'Data2D':
500            flag = False
501        if selected_page in page_finder:
502            #Delete the name of the page into the list of open page
503            for type, list in self.opened_pages.iteritems():
504                #Don't return any panel is the exact same page is created
505                name = str(list[0])
506                if flag and selected_page.window_name == name:
507                    if type.lower() in ['theory1d', 'theory2d']:
508                        self.manager.remove_plot(selected_page, theory=True)
509                    else:
510                        self.manager.remove_plot(selected_page, theory=False)
511                    break 
512            del page_finder[selected_page]
513        ##remove the check box link to the model name of this page (selected_page)
514        try:
515            self.sim_page.draw_page()
516        except:
517            ## that page is already deleted no need to remove check box on
518            ##non existing page
519            pass
520               
521        #Delete the name of the page into the list of open page
522        for type, list in self.opened_pages.iteritems():
523            #Don't return any panel is the exact same page is created
524            name = str(list[0])
525            if selected_page.window_name == name:
526                del self.opened_pages[type]
527                break 
528     
529 
Note: See TracBrowser for help on using the repository browser.