source: sasview/sansview/perspectives/fitting/fitpage.py @ 6f023e8

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