source: sasview/sansview/perspectives/fitting/simfitpage.py @ b28717b

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

working on simpage
  • Property mode set to 100644
File size: 21.4 KB
Line 
1
2import sys,re,string, wx 
3import wx.lib.newevent 
4from sans.guicomm.events import StatusEvent   
5
6class FitConstraint:
7    """
8        Contains info on selected model to fit
9        via simultaneous page
10    """
11    def __init__(self, model, page=None):
12        """
13            initialization
14        """
15        self.model= model
16        self.page = page
17        self.fittable_param =[]
18        self._set_fittableParam()
19       
20    def _set_fittableParam(self):
21        """
22            fill self.fittable_param
23        """
24        for item in self.model.getParamList():
25            if not item  in self.model.getDispParamList():
26                self.fittable_param.append(item)
27        for item in self.model.fixed:
28            self.fittable_param.append(item)
29     
30
31class SimultaneousFitPage(wx.ScrolledWindow):
32    """
33        Simultaneous fitting panel
34        All that needs to be defined are the
35        two data members window_name and window_caption
36    """
37    ## Internal name for the AUI manager
38    window_name = "simultaneous Fit page"
39    ## Title to appear on top of the window
40    window_caption = "Simultaneous Fit Page"
41   
42   
43    def __init__(self, parent,page_finder ={}, *args, **kwargs):
44        wx.ScrolledWindow.__init__(self, parent, *args, **kwargs)
45        """
46             Simultaneous page display
47        """
48        self.parent = parent
49        ## store page_finder
50        self.page_finder = page_finder
51        ## list contaning info to set constraint
52        ## look like self.constraint_dict[page]=FitConstraint(model, page)
53        self.constraint_dict={}
54        ## item list  self.constraints_list=[combobox1, combobox2,=,textcrtl, button ]
55        self.constraints_list=[]
56        ## list of current model
57        self.model_list=[]
58        ## selected mdoel to fit
59        self.model_toFit=[]
60        ## number of constraint
61        self.nb_constraint= 0
62        ## draw page
63        self.define_page_structure()
64        self.draw_page()
65        self.set_layout()
66       
67       
68       
69    def define_page_structure(self):
70        """
71            Create empty sizer for a panel
72        """
73        self.vbox  = wx.BoxSizer(wx.VERTICAL)
74        self.sizer1 = wx.BoxSizer(wx.VERTICAL)
75        self.sizer2 = wx.BoxSizer(wx.VERTICAL)
76     
77        self.vbox.Add(self.sizer1)
78        self.vbox.Add(self.sizer2)
79       
80    def set_scroll(self):
81        self.SetScrollbars(20,20,200,100)
82        self.Layout() 
83         
84    def set_layout(self):
85        """
86             layout
87        """
88        self.vbox.Layout()
89        self.vbox.Fit(self) 
90        self.SetSizer(self.vbox)
91       
92        self.set_scroll()
93        self.Centre()
94       
95    def onFit(self,event):
96        """ signal for fitting"""
97        ## making sure all parameters content a constraint
98        ## validity of the constraint expression is own by fit engine
99        if self.show_constraint.GetValue():
100            self._set_constraint()
101        ## get the fit range of very fit problem       
102        for page, value in self.page_finder.iteritems():
103            qmin, qmax= page.get_range()
104            value.set_range(qmin, qmax)
105        ## model was actually selected from this page to be fit
106        if len(self.model_toFit) >= 1 :
107            self.manager.on_simul_fit()
108        else:
109            msg= "Select at least one model to fit "
110            wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
111           
112           
113           
114    def set_manager(self, manager):
115        """
116            set panel manager
117            @param manager: instance of plugin fitting
118        """
119        self.manager = manager
120       
121       
122    def check_all_model_name(self,event):
123        """
124            check all models names
125        """
126        self.model_toFit=[] 
127        if self.cb1.GetValue()==True:
128            for item in self.model_list:
129                item[0].SetValue(True)
130                self.manager.schedule_for_fit( value=1,fitproblem =item[1]) 
131                self.model_toFit.append(item)
132               
133            ## constraint info
134            self._store_model()
135            ## display constraint fields
136            if self.show_constraint.GetValue():
137                self._show_constraint()
138                return
139        else:
140            for item in self.model_list:
141                item[0].SetValue(False) 
142                self.manager.schedule_for_fit( value=0,fitproblem =item[1]) 
143               
144            self.model_toFit=[]
145            ##constraint info
146            self._hide_constraint()
147       
148 
149    def check_model_name(self,event):
150        """
151            Save information related to checkbox and their states
152        """
153        self.model_toFit=[]
154        for item in self.model_list:
155            if item[0].GetValue()==True:
156                self.model_toFit.append(item)
157                self.manager.schedule_for_fit( value=1,fitproblem =item[1]) 
158            else:
159                self.manager.schedule_for_fit( value=0,fitproblem =item[1]) 
160                if item in self.model_toFit:
161                    self.model_toFit.remove(item)
162                    self.cb1.SetValue(False)
163       
164        ## display constraint fields
165        if len(self.model_toFit)==2:
166            self._store_model()
167            if self.show_constraint.GetValue() and len(self.constraints_list)==0:
168                self._show_constraint()
169        elif len(self.model_toFit)< 2:
170            ##constraint info
171            self._hide_constraint()             
172       
173        ## set the value of the main check button         
174        if len(self.model_list)==len(self.model_toFit):
175            self.cb1.SetValue(True)
176            return
177        else:
178            self.cb1.SetValue(False)
179           
180       
181 
182    def draw_page(self):     
183        """
184            Draw a sizer containing couples of data and model
185        """ 
186     
187        self.model_list=[]
188        self.model_toFit=[]
189        if len(self.model_list)>0:
190            for item in self.model_list:
191                item[0].SetValue(False) 
192                self.manager.schedule_for_fit( value=0,fitproblem =item[1])
193               
194        self.sizer1.Clear(True)
195       
196               
197        box_description= wx.StaticBox(self, -1,"Fit Couples")
198        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
199        sizer_title = wx.BoxSizer(wx.HORIZONTAL)
200        sizer_couples = wx.GridBagSizer(5,5)
201       
202        #------------------------------------------------------
203        if len(self.page_finder)==0:
204            sizer_title.Add(wx.StaticText(self,-1," No fit couple available !"))
205        else:
206            ## store model 
207            self._store_model()
208       
209            self.cb1 = wx.CheckBox(self, -1,'Select all Fit Couples')
210            self.cb1.SetValue(False)
211            wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.check_all_model_name)
212           
213            sizer_title.Add((15,15))
214            sizer_title.Add(self.cb1)
215            ## draw list of model name
216            self._fill_sizer_model_list(sizer_couples)
217            ## draw the sizer containing constraint info
218            self._fill_sizer_constraint()
219        #--------------------------------------------------------
220        boxsizer1.Add((5, 5)) 
221        boxsizer1.Add(sizer_title)
222        boxsizer1.Add((5, 5)) 
223        boxsizer1.Add(sizer_couples)
224        boxsizer1.Add((5,5))
225       
226        self.sizer1.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
227        self.sizer1.Layout()
228        self.SetScrollbars(20,20,200,100)
229        self.AdjustScrollbars()
230       
231    def _store_model(self):
232        """
233            Store selected model
234        """
235        if len(self.model_toFit) < 2:
236            return
237        for page, value in self.page_finder.iteritems():
238            model = value.get_model()
239            for item in self.model_toFit:
240                if model in item and not model in self.constraint_dict.keys():
241                    self.constraint_dict[model]=FitConstraint(model=model, page=page)
242                   
243       
244    def _display_constraint(self, event):
245        """
246            Show fields to add constraint
247        """
248        if len(self.model_toFit)< 2:
249            msg= "Select at least 2 models to add constraint "
250            wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
251            ## hide button
252            self._hide_constraint()
253            return
254        if self.show_constraint.GetValue():
255            self._show_constraint()
256            return
257        else:
258           self._hide_constraint()
259           return 
260           
261   
262    def _show_constraint(self):
263        """
264            Show constraint fields
265        """
266        if len(self.constraints_list)!= 0:
267            nb_fit_param = 0
268            for value in self.constraint_dict.values():
269                nb_fit_param += len(value.fittable_param)
270            ##Don't add anymore
271            if len(self.constraints_list) == nb_fit_param:
272                msg= "Cannot add another constraint .Maximum of number "
273                msg += "Parameters name reached %s"%str(nb_fit_param)
274                wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
275                self.sizer_constraints.Layout()
276                self.sizer2.Layout()
277                self.SetScrollbars(20,20,200,100)
278                return
279           
280        if len(self.model_toFit) < 2 :
281            msg= "Select at least 2 model to add constraint "
282            wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
283            self.sizer_constraints.Layout()
284            self.sizer2.Layout()
285            self.SetScrollbars(20,20,200,100)
286            return
287           
288        sizer_constraint =  wx.BoxSizer(wx.HORIZONTAL)
289       
290        model_cbox = wx.ComboBox(self, -1)
291        model_cbox.Clear()
292
293        param_cbox = wx.ComboBox(self, -1)
294        param_cbox.Hide()
295        wx.EVT_COMBOBOX(param_cbox,-1, self._on_select_param)
296       
297        ctl2 = wx.TextCtrl(self, -1)
298       
299        egal_txt= wx.StaticText(self,-1," = ")
300     
301       
302       
303        for model, value in self.constraint_dict.iteritems():
304            ## check if all parameters have been selected for constraint
305            ## then do not allow add constraint on parameters
306            model_cbox.Append( str(model.name), model)
307           
308           
309        wx.EVT_COMBOBOX(model_cbox,-1, self._on_select_model)
310       
311        ##[combobox1, combobox2,=,textcrtl ]
312        self.constraints_list.append([model_cbox, param_cbox, egal_txt, ctl2])
313       
314        sizer_constraint.Add(model_cbox, flag= wx.RIGHT|wx.EXPAND,border=10)
315        sizer_constraint.Add(param_cbox, flag= wx.RIGHT|wx.EXPAND,border=5)
316        sizer_constraint.Add(egal_txt, flag= wx.RIGHT|wx.EXPAND,border=5)
317        sizer_constraint.Add(ctl2, flag= wx.RIGHT|wx.EXPAND,border=10)
318     
319        self.sizer_constraints.Insert(before=self.nb_constraint,
320                                      item=sizer_constraint, flag= wx.TOP|wx.BOTTOM|wx.EXPAND,
321                                   border=5, userData=0)
322       
323        self.nb_constraint += 1
324        self.sizer_constraints.Layout()
325        self.sizer2.Layout()
326        self.SetScrollbars(20,20,200,100)
327       
328    def _hide_constraint(self): 
329        """
330            hide buttons related constraint
331        """ 
332        if len(self.constraint_dict)>0:
333            for item in self.constraints_list:
334                model = item[0].GetClientData(item[0].GetCurrentSelection())
335                if model  in self.constraint_dict.keys():
336                    page = self.constraint_dict[model].page
337                    self.page_finder[page].clear_model_param()
338               
339        self.nb_constraint =0     
340        self.constraint_dict={}
341        self._store_model()
342     
343        self.constraints_list=[]         
344        self.sizer_constraints.Clear(True) 
345        self.sizer_constraints.Layout()   
346        self.sizer2.Layout()
347        self.SetScrollbars(20,20,200,100)
348        self.AdjustScrollbars()   
349               
350   
351       
352    def _on_select_model(self, event):
353        """
354         fill combox box with list of parameters
355        """
356        model = event.GetClientData()
357        param_list= self.constraint_dict[model].fittable_param
358        length = len(self.constraints_list)
359        if length < 1:
360            return 
361       
362        param_cbox = self.constraints_list[length-1][1]
363        param_cbox.Clear()
364        ## insert only fittable paramaters
365        for param in param_list:
366            param_cbox.Append( str(param), model)
367           
368        param_cbox.Show(True)
369       
370       
371        self.sizer2.Layout()
372        self.SetScrollbars(20,20,200,100)
373       
374       
375    def _on_select_param(self, event):
376        """
377            Store the appropriate constraint in the page_finder
378        """
379        model = event.GetClientData()
380        param = event.GetString()
381     
382        length = len(self.constraints_list)
383        if length < 1:
384            return 
385        egal_txt = self.constraints_list[length-1][2]
386        egal_txt.Show(True)       
387       
388        ctl2 = self.constraints_list[length-1][3]
389        ctl2.Show(True)
390       
391       
392        self.sizer2.Layout()
393        self.SetScrollbars(20,20,200,100)
394       
395       
396    def _onAdd_constraint(self, event): 
397        """
398            Add another line for constraint
399        """
400        msg= " "
401        wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
402       
403        ## check that a constraint is added before allow to add another cosntraint
404       
405        for item in self.constraints_list:
406            model_cbox = item[0]
407            if model_cbox.GetString(0)=="":
408                msg= " Select a model Name! "
409                wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
410                return 
411            param_cbox = item[1]
412            if param_cbox.GetString(0)=="":
413                msg= " Select a parameter Name! "
414                wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
415                return 
416            ctl2 = item[3]
417            if ctl2.GetValue().lstrip().rstrip()=="":
418                model= param_cbox.GetClientData(param_cbox.GetCurrentSelection())
419                msg= " Enter a constraint for %s.%s! "%(model.name,param_cbox.GetString(0))           
420                wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
421                return 
422       
423        ## some model or parameters can be constrained
424        self._show_constraint()
425       
426       
427   
428       
429       
430    def _fill_sizer_constraint(self):
431        """
432            Fill sizer containing constraint info
433        """
434        msg= "Select at least 2 model to add constraint "
435        wx.PostEvent(self.parent.Parent, StatusEvent(status= msg ))
436       
437        self.sizer2.Clear(True)
438 
439        box_description= wx.StaticBox(self, -1,"Fit Constraints")
440        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
441        sizer_title = wx.BoxSizer(wx.HORIZONTAL)
442        self.sizer_constraints = wx.BoxSizer(wx.VERTICAL)
443        sizer_button = wx.BoxSizer(wx.HORIZONTAL)
444       
445        self.hide_constraint = wx.RadioButton(self, -1, 'No', (10, 10), style=wx.RB_GROUP)
446        self.show_constraint = wx.RadioButton(self, -1, 'Yes', (10, 30))
447       
448        self.Bind( wx.EVT_RADIOBUTTON, self._display_constraint,
449                    id= self.hide_constraint.GetId() )
450       
451        self.Bind(  wx.EVT_RADIOBUTTON, self._display_constraint,
452                         id= self.show_constraint.GetId()    )
453       
454        sizer_title.Add( wx.StaticText(self,-1," Model") )
455        sizer_title.Add(( 10,10) )
456        sizer_title.Add( wx.StaticText(self,-1," Parameter") )
457        sizer_title.Add(( 10,10) )
458        sizer_title.Add( wx.StaticText(self,-1," Add Constraint?") )
459        sizer_title.Add(( 10,10) )
460        sizer_title.Add( self.show_constraint )
461        sizer_title.Add( self.hide_constraint )
462        sizer_title.Add(( 10,10) )
463       
464        self.btAdd =wx.Button(self,wx.NewId(),'Add')
465        self.btAdd.Bind(wx.EVT_BUTTON, self._onAdd_constraint,id= self.btAdd.GetId())
466        self.btAdd.SetToolTipString("Add another constraint?")
467       
468       
469        self.btFit = wx.Button(self,wx.NewId(),'Fit')
470        self.btFit.Bind(wx.EVT_BUTTON, self.onFit,id= self.btFit.GetId())
471        self.btFit.SetToolTipString("Perform fit.")
472       
473        text_hint = wx.StaticText(self,-1,"Example: M0.paramter = M1.parameter") 
474        sizer_button.Add(text_hint, 0 , wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 10)
475        sizer_button.Add(self.btAdd, 0, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 10)
476        sizer_button.Add(self.btFit, 0, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 10)
477       
478        boxsizer1.Add(sizer_title)
479        boxsizer1.Add((10,10))
480        boxsizer1.Add(self.sizer_constraints)
481        boxsizer1.Add((10,10))
482        boxsizer1.Add(sizer_button)
483       
484        self.sizer2.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
485        self.sizer2.Layout()
486        self.SetScrollbars(20,20,200,100)
487       
488       
489       
490    def _set_constraint(self):
491        """
492            get values from the constrainst textcrtl ,parses them into model name
493            parameter name and parameters values.
494            store them in a list self.params .when when params is not empty set_model
495            uses it to reset the appropriate model and its appropriates parameters
496        """
497        for item in self.constraints_list:
498           
499            model = item[0].GetClientData(item[0].GetCurrentSelection())
500            param = item[1].GetString(item[1].GetCurrentSelection())
501            constraint = item[3].GetValue().lstrip().rstrip()
502            if model  in self.constraint_dict.keys():
503                page = self.constraint_dict[model].page
504                if constraint == "":
505                    constraint = None
506               
507                self.page_finder[page].set_model_param(param,constraint)
508       
509   
510             
511    def _fill_sizer_model_list(self,sizer):
512        """
513            Receive a dictionary containing information to display model name
514            @param page_finder: the dictionary containing models information
515        """
516        ix = 0
517        iy = 0
518        list=[]
519        sizer.Clear(True)
520       
521        new_name = wx.StaticText(self, -1, 'New Model Name', style=wx.ALIGN_CENTER)
522        new_name.SetBackgroundColour('orange')
523        sizer.Add(new_name,(iy, ix),(1,1),
524                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
525       
526        ix +=2 
527        model_type = wx.StaticText(self, -1, '  Model Type')
528        model_type.SetBackgroundColour('grey')
529        sizer.Add(model_type,(iy, ix),(1,1),
530                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
531        ix +=1 
532        data_used = wx.StaticText(self, -1, '  Used Data')
533        data_used.SetBackgroundColour('grey')
534        sizer.Add(data_used,(iy, ix),(1,1),
535                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
536       
537        for page, value in self.page_finder.iteritems():
538            try:
539                ix = 0
540                iy += 1 
541                model = value.get_model()
542                cb = wx.CheckBox(self, -1, str(model.name))
543                cb.SetValue(False)
544                sizer.Add( cb,( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
545                wx.EVT_CHECKBOX(self, cb.GetId(), self.check_model_name)
546               
547               
548                ix +=2 
549                type = model.__class__.__name__
550                model_type = wx.StaticText(self, -1, str(type))
551                sizer.Add(model_type,( iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
552               
553                ix +=1 
554                data = value.get_fit_data()
555                data_used= wx.StaticText(self, -1, str(data.name))
556                sizer.Add(data_used,( iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
557               
558               
559                self.model_list.append([cb,value,page,model])
560               
561            except:
562                pass
563        iy +=1
564        sizer.Add((20,20),( iy,ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
565        sizer.Layout()   
566       
567   
568 
569       
570class HelpWindow(wx.Frame):
571    def __init__(self, parent, id, title):
572        wx.Frame.__init__(self, parent, id, title, size=(570, 400))
573       
574        page_finder ={}
575        ## create random data
576        from danse.common.plottools.plottables import Data1D
577        data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
578        data.name="mydata.txt"
579        ## create model
580        from sans.models.CylinderModel import CylinderModel
581        model = CylinderModel()
582        model.name="M0"
583       
584       
585        from fitproblem import FitProblem
586        page_finder["page"]= FitProblem()
587        ## fill the page_finder
588        page_finder["page"].add_fit_data(data)
589        page_finder["page"].set_model(model)
590        self.page = SimultaneousFitPage(self, page_finder=page_finder) 
591       
592       
593       
594        self.Centre()
595        self.Show(True)
596
597
598   
599if __name__=="__main__":
600    app = wx.App()
601    HelpWindow(None, -1, 'HelpWindow')
602    app.MainLoop()
603   
Note: See TracBrowser for help on using the repository browser.