source: sasview/sansview/perspectives/fitting/fitpage.py @ 376916c

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

resettting gaussian dispersity —fixed
reducing access to molde multplication —-done as request

  • Property mode set to 100644
File size: 34.4 KB
Line 
1import sys
2import wx
3import wx.lib.newevent
4import numpy
5import copy
6import math
7from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
8
9from sans.guicomm.events import StatusEvent   
10from sans.guiframe.utils import format_number
11(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
12_BOX_WIDTH = 80
13
14import basepage
15from basepage import BasicPage
16
17
18class FitPage(BasicPage):
19    """
20        FitPanel class contains fields allowing to display results when
21        fitting  a model and one data
22        @note: For Fit to be performed the user should check at least one parameter
23        on fit Panel window.
24 
25    """
26    def __init__(self,parent, page_info, name=""):
27        BasicPage.__init__(self, parent, page_info,name)
28        """
29            Initialization of the Panel
30        """
31        self.name= self.data.name
32        page_info.page_name= self.data.name
33       
34        ## fit page does not content npts txtcrtl
35        self.npts=None
36        ## if no dispersity parameters is avaible
37        self.text_disp_1=None
38       
39        self._fill_datainfo_sizer()
40        self._fill_model_sizer( self.sizer1)
41        self._fill_range_sizer() 
42        self.model = self.formfactorbox.GetClientData(0)()
43        self.page_info.model= self.model
44        self._on_select_model(event=None)
45           
46       
47    def _on_display_description(self, event):
48        """
49            Show or Hide description
50            @param event: wx.EVT_RADIOBUTTON
51        """
52        self._on_display_description_helper()
53       
54       
55       
56    def _on_display_description_helper(self):
57        """
58            Show or Hide description
59            @param event: wx.EVT_RADIOBUTTON
60        """
61        ## save state of radiobox
62        self.page_info. save_radiobox_state( self.description_hide )
63        self.page_info. save_radiobox_state( self.description_show )
64        ## Show description
65        if not self.description_show.GetValue():
66            self.sizer_description.Clear(True)
67           
68        else:
69            model=self.page_info.model
70            description=""
71            if model!=None:
72                description = self.page_info.model.description
73            self.description = wx.StaticText( self,-1,str(description) )
74            self.sizer_description.Add( self.description, 1, wx.EXPAND | wx.ALL, 10 )
75           
76        self.Layout()
77   
78   
79    def _fill_range_sizer(self):
80        """
81            Fill the sizer containing the plotting range
82            add  access to npts
83        """
84        sizer_fit = wx.GridSizer(1, 1,0, 0)
85   
86        self.btFit = wx.Button(self,wx.NewId(),'Fit')
87        self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
88        self.btFit.SetToolTipString("Perform fit.")
89       
90        #self.btStopFit = wx.Button(self,wx.NewId(),'Stop Fit')
91        #self.btStopFit.Bind(wx.EVT_BUTTON, self.onStopFit,id= self.btStopFit.GetId())
92        #self.btStopFit.SetToolTipString("Stop Current fit job.")
93        #self.btStopFit.Hide()
94       
95        sizer_fit.Add((5,5),1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)       
96        sizer_fit.Add(self.btFit,0, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5) 
97        #sizer_fit.Add(self.btStopFit,1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
98       
99        sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
100        #Filling the sizer containing instruments smearing info.
101        self.disable_smearer = wx.RadioButton(self, -1, 'No', style=wx.RB_GROUP)
102        self.enable_smearer = wx.RadioButton(self, -1, 'Yes')
103        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
104        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
105       
106        sizer_smearer.Add(wx.StaticText(self,-1,'Instrument Smearing? '))
107        sizer_smearer.Add((10, 10))
108        sizer_smearer.Add( self.enable_smearer )
109        sizer_smearer.Add((10,10))
110        sizer_smearer.Add( self.disable_smearer )
111       
112        self._set_range_sizer( title="Fitted Q Range",
113                               object1=sizer_smearer, object= sizer_fit)
114       
115       
116    def _fill_datainfo_sizer(self):
117        """
118            fill sizer 0 with data info
119        """
120        self.sizer0.Clear(True)
121        ## no loaded data , don't fill the sizer
122        if self.data== None:
123            self.sizer0.Layout()
124            return
125       
126        box_description= wx.StaticBox(self, -1, 'Loaded Data')
127        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
128        #----------------------------------------------------------
129        sizer_data = wx.GridSizer(3, 3,5, 5)
130        #Filling the sizer containing data related fields
131        DataSource  =wx.StaticText(self, -1,str(self.data.name))
132
133        sizer_data.Add(wx.StaticText(self, -1, 'Data Source Name : '))
134        sizer_data.Add(DataSource )
135        sizer_data.Add( (5,5) )
136       
137        #---------sizer 2 draw--------------------------------
138        #set maximum range for x in linear scale
139        if not hasattr(self.data,"data"): #Display only for 1D data fit
140            # Minimum value of data   
141            data_min = str(format_number(numpy.min(self.data.x)))
142            # Maximum value of data 
143            data_max = str(format_number(numpy.max(self.data.x)))
144            text4_3 = wx.StaticText(self, -1, 'Maximum Q Range(Linear)',
145                                     style=wx.ALIGN_LEFT)
146            sizer_data.Add( text4_3 )
147            sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%data_min))
148            sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%data_max))
149           
150        else:
151            radius_min= 0
152            x= numpy.max(self.data.xmin, self.data.xmax)
153            y= numpy.max(self.data.ymin, self.data.ymax)
154            radius_max = math.sqrt(x*x + y*y)
155            # Minimum value of data   
156            data_min = str(format_number(radius_min))
157            # Maximum value of data 
158            data_max = str(format_number(radius_max))
159            text4_3 = wx.StaticText(self, -1, 'Maximum Q Range',
160                                     style=wx.ALIGN_LEFT)
161            sizer_data.Add( text4_3 )
162            sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%data_min))
163            sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%data_max))
164           
165        boxsizer1.Add(sizer_data)
166        #------------------------------------------------------------
167        self.sizer0.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
168        self.sizer0.Layout()
169       
170        self.qmin_x= data_min
171        self.qmax_x= data_max
172       
173       
174    def _fill_model_sizer(self, sizer):
175        """
176            fill sizer containing model info
177        """
178        sizer_tcChi = wx.GridSizer(2, 2,5, 5)
179        self.tcChi    =  wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
180       
181       
182        self.text1_1 = wx.StaticText(self, -1, 'Chi2/dof', style=wx.ALIGN_LEFT)
183       
184       
185        self.btChi = wx.Button(self,wx.NewId(),'View Chi')
186        self.btChi.Bind(wx.EVT_BUTTON, self._onFit,id= self.btChi.GetId())
187        self.btChi.SetToolTipString("View data - model.")
188       
189        sizer_tcChi.Add(self.text1_1,1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
190        sizer_tcChi.Add(self.btChi,1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)       
191        sizer_tcChi.Add(self.tcChi,1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)       
192        sizer_tcChi.Add((5,5),1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
193           
194        ## class base method  to add view 2d button   
195        self._set_model_sizer(sizer=sizer, title="Model",object=  sizer_tcChi )   
196   
197   
198    def _set_sizer_gaussian(self):
199        """
200            draw sizer with gaussian dispersity parameters
201        """
202        self.fittable_param=[]
203        self.fixed_param=[]
204   
205        self.sizer4_4.Clear(True)
206       
207        if self.model==None:
208            ##no model is selected
209            return
210        if not self.enable_disp.GetValue():
211            ## the user didn't select dispersity display
212            return 
213       
214        self._reset_dispersity()
215        # Create the dispersion objects
216        for item in self.model.dispersion.keys():
217            disp_model =  GaussianDispersion()
218            self._disp_obj_dict[item] = disp_model
219            self.model.set_dispersion(item, disp_model)
220
221        ix=0
222        iy=1
223        disp = wx.StaticText(self, -1, 'Names')
224        self.sizer4_4.Add(disp,( iy, ix),(1,1), 
225                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
226        ix += 1 
227        values = wx.StaticText(self, -1, 'Values')
228        self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
229        ix +=2 
230        self.text_disp_1 = wx.StaticText(self, -1, 'Errors')
231        self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
232                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
233        self.text_disp_1.Hide()
234        ix += 1 
235        npts = wx.StaticText(self, -1, 'Npts')
236        self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
237        ix += 1 
238        nsigmas = wx.StaticText(self, -1, 'Nsigmas')
239        self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
240       
241        for item in self.model.dispersion.keys():
242            self.disp_cb_dict[item]= None
243            name1=item+".width"
244            name2=item+".npts"
245            name3=item+".nsigmas"
246            iy += 1
247            for p in self.model.dispersion[item].keys():
248   
249                if p=="width":
250                    ix = 0
251                    cb = wx.CheckBox(self, -1, name1, (10, 10))
252                    wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
253                   
254                    self.sizer4_4.Add( cb,( iy, ix),(1,1), 
255                                       wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
256                    ix = 1
257                    value= self.model.getParam(name1)
258                    ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
259                                        style=wx.TE_PROCESS_ENTER)
260                   
261                    ctl1.SetValue(str (format_number(value)))
262                    ctl1.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
263                    ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
264                    self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
265                   
266                    ## text to show error sign
267                    ix = 2
268                    text2=wx.StaticText(self, -1, '+/-')
269                    self.sizer4_4.Add(text2,(iy, ix),(1,1),
270                                      wx.EXPAND|wx.ADJUST_MINSIZE, 0)
271                    text2.Hide() 
272                    ## txtcrtl to add error from fit
273                    ix = 3
274                    ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
275                    self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
276                    ctl2.Hide()
277                   
278                    self.fittable_param.append([cb,name1,ctl1,text2,
279                                                ctl2, None, None,None])
280                elif p=="npts":
281                        ix = 4
282                        value= self.model.getParam(name2)
283                        Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
284                                            style=wx.TE_PROCESS_ENTER)
285                       
286                        Tctl.SetValue(str (format_number(value)))
287                        Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
288                        Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
289                        self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
290                                           wx.EXPAND|wx.ADJUST_MINSIZE, 0)
291                       
292                        self.fixed_param.append([None,name2, Tctl,None,None,
293                                                  None, None,None])
294               
295                elif p=="nsigmas":
296                        ix = 5
297                        value= self.model.getParam(name3)
298                        Tctl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
299                                            style=wx.TE_PROCESS_ENTER)
300                        Tctl.SetValue(str (format_number(value)))
301                        Tctl.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
302                        Tctl.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
303                        self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
304                                           wx.EXPAND|wx.ADJUST_MINSIZE, 0)
305                        ix +=1
306                        self.sizer4_4.Add((20,20), (iy,ix),(1,1),
307                                           wx.EXPAND|wx.ADJUST_MINSIZE, 0)
308                       
309                        self.fixed_param.append([None,name3, Tctl
310                                                 ,None,None, None, None,None])
311               
312        wx.PostEvent(self.parent, StatusEvent(status=\
313                        " Selected Distribution: Gaussian"))   
314        ix =0
315        iy +=1 
316        self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)       
317        self.sizer4_4.Layout()
318        self.sizer4.Layout()
319        self.SetScrollbars(20,20,200,100)
320     
321       
322    def _onFit(self, event):     
323        """
324            Allow to fit
325        """
326        #self.btFit.SetLabel("Stop")
327        from sans.guiframe.utils import check_value
328        flag = check_value( self.qmin, self.qmax) 
329       
330        if not flag:
331            msg= "Fitting range invalid"
332            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
333            return 
334       
335        if len(self.param_toFit) <= 0:
336            msg= "Select at least one parameter to fit"
337            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
338            return 
339       
340        self.qmin_x=float(self.qmin.GetValue())
341        self.qmax_x =float( self.qmax.GetValue())
342        self.manager._reset_schedule_problem( value=0)
343        self.manager.schedule_for_fit( value=1,page=self,fitproblem =None) 
344        self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
345        #single fit
346        #self.manager.on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x)
347        self.manager.onFit()
348           
349        self.sizer5.Layout()
350        self.SetScrollbars(20,20,55,40)
351       
352       
353    def _on_select_model(self, event): 
354        """
355             call back for model selection
356        """   
357        self._on_select_model_helper() 
358        self.set_model_param_sizer(self.model)
359        self._set_sizer_gaussian()
360       
361        evt = ModelEventbox(model=self.model)
362        wx.PostEvent(self.event_owner, evt)   
363        try:
364            self.compute_chisqr()
365            self.text1_1.Show()
366            self.tcChi.Show()
367        except:
368            ## error occured on chisqr computation
369            pass
370   
371    def _onparamEnter(self,event):
372        """
373            when enter value on panel redraw model according to changed
374        """
375       
376        self._onparamEnter_helper()
377        self.compute_chisqr()
378       
379    def get_range(self):
380        """
381            return the fitting range
382        """
383        return float(self.qmin_x) , float(self.qmax_x)
384       
385    def get_param_list(self):
386        """
387            @return self.param_toFit: list containing  references to TextCtrl
388            checked.Theses TextCtrl will allow reference to parameters to fit.
389            @raise: if return an empty list of parameter fit will nnote work
390            properly so raise ValueError,"missing parameter to fit"
391        """
392        if self.param_toFit !=[]:
393            return self.param_toFit
394        else:
395            raise ValueError,"missing parameter to fit"   
396     
397    def onsetValues(self,chisqr, out,cov):
398        """
399            Build the panel from the fit result
400            @param chisqr:Value of the goodness of fit metric
401            @param out:list of parameter with the best value found during fitting
402            @param cov:Covariance matrix
403       
404        """
405        self.tcChi.SetLabel(format_number(chisqr))
406        params = {}
407        is_modified = False
408        has_error = False
409        self.text2_3.Hide()
410        if self.text_disp_1 !=None:
411            self.text_disp_1.Hide()
412        #set the panel when fit result are float not list
413        if out.__class__==numpy.float64:
414            self.param_toFit[0][2].SetValue(format_number(out))
415            self.param_toFit[0][2].Refresh()
416           
417            self.param_toFit[0][4].Clear()
418            self.param_toFit[0][4].Hide()
419            if cov !=None :
420                self.text2_3.Show(True)
421                if self.text_disp_1 !=None:
422                    self.text_disp_1.Show(True)
423                   
424                self.param_toFit[0][3].Show(True)
425                self.param_toFit[0][4].Clear()
426                self.param_toFit[0][4].SetValue(format_number(cov[0]))
427                self.param_toFit[0][4].Show(True)
428        else:
429            i=0
430            j=0
431            #Set the panel when fit result are list
432            for item in self.param_toFit:
433                ## reset error value to initial state
434                item[4].Clear()
435                item[4].Hide()
436                item[4].Refresh()
437                if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
438                    item[2].SetValue(format_number(self.model.getParam(item[1])))
439                    item[2].Refresh()
440                if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
441                    self.text2_3.Show(True) 
442                    if self.text_disp_1!=None:
443                        self.text_disp_1.Show(True)
444                    item[3].Show(True)
445                    item[4].Clear()
446                    for j in range(len(out)):
447                        if out[j]==self.model.getParam(item[1]):
448                            break
449                    item[4].SetValue(format_number(cov[j]))
450                    item[4].Refresh()
451                    item[4].Show(True)   
452                i+=1
453       
454        self.sizer3.Layout()
455        self.sizer4.Layout()
456        self.SetScrollbars(20,20,200,100)
457       
458       
459     
460    def onSmear(self, event):
461        """
462            Create a smear object that will change the way residuals
463            are compute when fitting
464        """
465        smear =None
466        msg=""
467        if self.enable_smearer.GetValue():
468            from DataLoader.qsmearing import smear_selection
469            smear = smear_selection( self.data )
470            if hasattr(self.data,"dxl"):
471                msg= ": Resolution smearing parameters"
472            if hasattr(self.data,"dxw"):
473                msg= ": Slit smearing parameters"
474            if smear ==None:
475                wx.PostEvent(self.manager.parent, StatusEvent(status=\
476                            "Data contains no smearing information"))
477            else:
478                wx.PostEvent(self.manager.parent, StatusEvent(status=\
479                            "Data contains smearing information %s"%msg))
480            self.manager.set_smearer(smear, qmin= self.qmin_x, qmax= self.qmax_x)   
481             
482 
483    def compute_chisqr2D(self):
484        """
485            compute chi square given a model and data 2D and set the value
486            to the tcChi txtcrl
487        """
488        from sans.guiframe.utils import check_value
489        flag = check_value( self.qmin, self.qmax)
490       
491        res=[]
492        if flag== True:
493            try:
494                self.qmin_x = float(self.qmin.GetValue())
495                self.qmax_x = float(self.qmax.GetValue())
496                for i in range(len(self.data.x_bins)):
497                    for j in range(len(self.data.y_bins)):
498                        #Check the range containing data between self.qmin_x and self.qmax_x
499                        if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)>=math.pow(self.qmin_x,2):
500                            if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)<=math.pow(self.qmax_x,2):
501                                chisqrji=(self.data.data[j][i]- self.model.runXY(\
502                                                                                    [self.data.x_bins[i],self.data.y_bins[j]]))\
503                                                                                    /self.data.err_data[j][i]
504                                #Vector containing residuals
505                                res.append( math.pow(chisqrji,2) )
506                # compute sum of residual
507                sum=0
508                for item in res:
509                    if numpy.isfinite(item):
510                        sum +=item
511                self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
512            except:
513                wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
514                            "Chisqr cannot be compute: %s"% sys.exc_value))
515                return
516   
517       
518    def compute_chisqr(self):
519        """
520            compute chi square given a model and data 1D and set the value
521            to the tcChi txtcrl
522        """
523        from sans.guiframe.utils import check_value
524        flag = check_value( self.qmin, self.qmax)
525       
526        if flag== True:
527            try:
528                if hasattr(self.data,"data"):
529                    self.compute_chisqr2D()
530                    return
531                else:
532                    self.qmin_x = float(self.qmin.GetValue())
533                    self.qmax_x = float(self.qmax.GetValue())
534                    # return residuals within self.qmin_x and self.qmax_x
535                    x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
536                    if self.qmin_x==None and self.qmax_x==None: 
537                        fx =numpy.asarray([self.model.run(v) for v in x])
538                        temp=(y - fx)/dy
539                        res= temp*temp
540                    else:
541                        idx = (x>= self.qmin_x) & (x <=self.qmax_x)
542                        fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
543                        temp=(y[idx] - fx)/dy[idx]
544                        res= temp*temp
545                    #sum of residuals
546                    sum=0
547                    for item in res:
548                        if numpy.isfinite(item):
549                            sum +=item
550                    self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
551            except:
552                wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
553                            "Chisqr cannot be compute: %s"% sys.exc_value))
554                return 
555           
556   
557    def select_all_param(self,event): 
558        """
559             set to true or false all checkBox given the main checkbox value cb1
560        """
561        self.param_toFit=[]
562        if  self.parameters !=[]:
563            if  self.cb1.GetValue()==True:
564                for item in self.parameters:
565                    item[0].SetValue(True)
566                    self.param_toFit.append(item )
567                if len(self.fittable_param)>0:
568                    for item in self.fittable_param:
569                        item[0].SetValue(True)
570                        self.param_toFit.append(item )
571            else:
572                for item in self.parameters:
573                    item[0].SetValue(False)
574                for item in self.fittable_param:
575                    item[0].SetValue(False)
576                self.param_toFit=[]
577               
578               
579               
580    def select_param(self,event):
581        """
582            Select TextCtrl  checked for fitting purpose and stores them
583            in  self.param_toFit=[] list
584        """
585        self.param_toFit=[]
586        for item in self.parameters:
587            #Select parameters to fit for list of primary parameters
588            if item[0].GetValue()==True:
589                if not (item in self.param_toFit):
590                    self.param_toFit.append(item ) 
591            else:
592                #remove parameters from the fitting list
593                if item in self.param_toFit:
594                    self.param_toFit.remove(item)
595        #Select parameters to fit for list of fittable parameters with dispersion         
596        for item in self.fittable_param:
597            if item[0].GetValue()==True:
598                if not (item in self.param_toFit):
599                    self.param_toFit.append(item) 
600            else:
601                #remove parameters from the fitting list
602                if item in self.param_toFit:
603                    self.param_toFit.remove(item)           
604        #Set the value of checkbox that selected every checkbox or not           
605        if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
606            self.cb1.SetValue(True)
607        else:
608            self.cb1.SetValue(False)
609   
610   
611    def set_model_description(self,description,sizer):
612        """
613            fill a sizer with description
614            @param description: of type string
615            @param sizer: wx.BoxSizer()
616        """
617   
618        sizer.Clear(True)
619        box_description= wx.StaticBox(self, -1, 'Model Description')
620        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
621               
622        self.description = wx.StaticText( self,-1,str(description) )
623        sizer_selection=wx.BoxSizer(wx.HORIZONTAL)
624       
625       
626        self.description_show = wx.RadioButton(self, -1, 'Show', style=wx.RB_GROUP)
627        self.description_hide = wx.RadioButton(self, -1, 'Hide')
628       
629        self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
630                   id=self.description_hide.GetId() )
631       
632        self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
633                   id=self.description_show.GetId() )
634       
635        self.model_description = wx.Button(self, -1, "More Details")
636     
637       
638        self.page_info.save_radiobox_state( self.description_hide )
639        self.page_info.save_radiobox_state( self.description_show )
640       
641        sizer_selection.Add( self.description_show )
642        sizer_selection.Add( (20,20)) 
643        sizer_selection.Add( self.description_hide )
644        sizer_selection.Add( (20,20)) 
645        sizer_selection.Add( self.model_description )
646       
647         
648        self.sizer_description=wx.BoxSizer(wx.HORIZONTAL)
649        self.sizer_description.Add( self.description, 1, wx.EXPAND | wx.ALL, 10 )
650       
651        boxsizer1.Add( sizer_selection) 
652        boxsizer1.Add( (20,20)) 
653        boxsizer1.Add( self.sizer_description) 
654   
655
656        sizer.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
657        sizer.Layout()
658   
659   
660       
661       
662    def set_model_param_sizer(self, model, engine_type="park"):
663        """
664            Build the panel from the model content
665            @param model: the model selected in combo box for fitting purpose
666        """
667        self.sizer3.Clear(True)
668        self.parameters = []
669        self.param_toFit=[]
670        self.fittable_param=[]
671        self.fixed_param=[]
672       
673        if model ==None:
674            self.sizer3.Layout()
675            self.SetScrollbars(20,20,200,100)
676            return
677        box_description= wx.StaticBox(self, -1,str("Model Parameters"))
678        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
679        sizer = wx.GridBagSizer(5,5)
680        ## save the current model
681        self.model = model
682           
683        keys = self.model.getParamList()
684        #list of dispersion paramaters
685        self.disp_list=self.model.getDispParamList()
686       
687        keys.sort()
688   
689        iy = 1
690        ix = 0
691        self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
692        wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
693        self.cb1.SetValue(False)
694       
695        sizer.Add(self.cb1,(iy, ix),(1,1),\
696                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
697        ix +=1
698        self.text2_2 = wx.StaticText(self, -1, 'Values')
699        sizer.Add(self.text2_2,(iy, ix),(1,1),\
700                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
701        ix +=2 
702        self.text2_3 = wx.StaticText(self, -1, 'Errors')
703        sizer.Add(self.text2_3,(iy, ix),(1,1),\
704                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
705        self.text2_3.Hide()
706        ix +=1 
707        self.text2_min = wx.StaticText(self, -1, 'Min')
708        sizer.Add(self.text2_min,(iy, ix),(1,1),\
709                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
710        self.text2_min.Hide()
711        ix +=1 
712        self.text2_max = wx.StaticText(self, -1, 'Max')
713        sizer.Add(self.text2_max,(iy, ix),(1,1),\
714                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
715        self.text2_max.Hide()
716        ix += 1
717        self.text2_4 = wx.StaticText(self, -1, 'Units')
718        sizer.Add(self.text2_4,(iy, ix),(1,1),\
719                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
720        self.text2_4.Hide()
721        if  engine_type == "park":
722            self.text2_max.Show(True)
723            self.text2_min.Show(True)
724
725        for item in keys:
726            if not item in self.disp_list:
727                iy += 1
728                ix = 0
729                ## add parameters name with checkbox for selecting to fit
730                cb = wx.CheckBox(self, -1, item )
731                cb.SetValue(False)
732                wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
733                sizer.Add( cb,( iy, ix),(1,1),
734                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
735                ## add parameter value
736                ix += 1
737                value= self.model.getParam(item)
738                ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
739                                    style=wx.TE_PROCESS_ENTER)
740               
741                ctl1.SetValue(str (format_number(value)))
742                ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
743                ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
744                sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
745                ## text to show error sign
746                ix += 1
747                text2=wx.StaticText(self, -1, '+/-')
748                sizer.Add(text2,(iy, ix),(1,1),\
749                                wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
750                text2.Hide() 
751                ## txtcrtl to add error from fit
752                ix += 1
753                ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
754                sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
755                ctl2.Hide()
756               
757                ix += 1
758                ctl3 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
759                ctl3.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
760                ctl3.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
761                sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
762                ctl3.Hide()
763               
764                ix += 1
765                ctl4 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
766                ctl4.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
767                ctl4.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
768                sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
769                ctl4.Hide()
770                if  engine_type == "park":
771                    ctl3.Show(True)
772                    ctl4.Show(True)
773                   
774                ix +=1
775                # Units
776                try:
777                    units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
778                except:
779                    units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
780                sizer.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
781               
782                ##[cb state, name, value, "+/-", error of fit, min, max , units]
783                self.parameters.append([cb,item, ctl1,
784                                        text2,ctl2, ctl3, ctl4,None])
785               
786        iy+=1
787        sizer.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
788       
789        #Display units text on panel
790        for item in keys:   
791            if self.model.details[item][0]!='':
792                self.text2_4.Show()
793                break
794            else:
795                self.text2_4.Hide()
796   
797        boxsizer1.Add(sizer)
798       
799        self.sizer3.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
800        self.sizer3.Layout()
801        self.SetScrollbars(20,20,200,100)
802       
803   
804           
805       
806class HelpWindow(wx.Frame):
807    def __init__(self, parent, id, title):
808        wx.Frame.__init__(self, parent, id, title, size=(570, 400))
809       
810        from sans.models.CylinderModel import CylinderModel
811        model = CylinderModel()
812        #from sans.models.LineModel import LineModel
813        #model = LineModel()
814        from danse.common.plottools.plottables import Data1D
815        data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
816   
817        from pageInfo import PageInfo
818        myinfo = PageInfo(self,  model, data=data )
819       
820        ## add data
821       
822        from models import ModelList
823        mylist= ModelList()
824
825        from sans.models.SphereModel import SphereModel
826        from sans.models.SquareWellStructure import SquareWellStructure
827        from sans.models.DebyeModel import DebyeModel
828        from sans.models.LineModel import LineModel
829        name= "shapes"
830        list1= [SphereModel]
831        mylist.set_list( name, list1)
832       
833        name= "Shape-independent"
834        list1= [DebyeModel]
835        mylist.set_list( name, list1)
836       
837        name= "Structure Factors"
838        list1= [SquareWellStructure]
839        mylist.set_list( name, list1)
840       
841        name= "Added models"
842        list1= [LineModel]
843        mylist.set_list( name, list1)
844       
845        myinfo.model_list_box = mylist.get_list()
846       
847        self.page = FitPage(self, myinfo) 
848       
849       
850       
851        self.Centre()
852        self.Show(True)
853
854
855   
856if __name__=="__main__":
857    app = wx.App()
858    HelpWindow(None, -1, 'HelpWindow')
859    app.MainLoop()
860               
Note: See TracBrowser for help on using the repository browser.