source: sasview/sansview/perspectives/fitting/fitpage.py @ 6355f4d

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

compute chiqrs

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