source: sasview/sansview/perspectives/fitting/simfitpage.py @ 05f14dd

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

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