source: sasview/sansview/perspectives/fitting/fitpage.py @ 0a518e4c

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

after fitting remove error txtcrtl

  • Property mode set to 100644
File size: 33.7 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        ## fit page does not content npts txtcrtl
32        self.npts=None
33        ## if no dispersity parameters is avaible
34        self.text_disp_1=None
35       
36        self._fill_datainfo_sizer()
37        self._fill_model_sizer( self.sizer1)
38        self._fill_range_sizer() 
39        if hasattr(self.page_info,"model"):
40            self.set_model_param_sizer(self.model)
41           
42       
43    def _on_display_description(self, event):
44        """
45            Show or Hide description
46            @param event: wx.EVT_RADIOBUTTON
47        """
48        self._on_display_description_helper()
49       
50       
51       
52    def _on_display_description_helper(self):
53        """
54            Show or Hide description
55            @param event: wx.EVT_RADIOBUTTON
56        """
57        ## save state of radiobox
58        self.page_info. save_radiobox_state( self.description_hide )
59        self.page_info. save_radiobox_state( self.description_show )
60        ## Show description
61        if not self.description_show.GetValue():
62            self.sizer_description.Clear(True)
63           
64        else:
65            model=self.page_info.model
66            description=""
67            if model!=None:
68                description = self.page_info.model.description
69            self.description = wx.StaticText( self,-1,str(description) )
70            self.sizer_description.Add( self.description, 1, wx.EXPAND | wx.ALL, 10 )
71           
72        self.Layout()
73   
74   
75    def _fill_range_sizer(self):
76        """
77            Fill the sizer containing the plotting range
78            add  access to npts
79        """
80        sizer_fit = wx.GridSizer(2, 2,0, 0)
81   
82        self.btFit = wx.Button(self,wx.NewId(),'Fit')
83        self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
84        self.btFit.SetToolTipString("Perform fit.")
85       
86        self.btStopFit = wx.Button(self,wx.NewId(),'Stop Fit')
87        #self.btStopFit.Bind(wx.EVT_BUTTON, self.onStopFit,id= self.btStopFit.GetId())
88        self.btStopFit.SetToolTipString("Stop Current fit job.")
89        #self.btStopFit.Hide()
90       
91        sizer_fit.Add((5,5),1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)       
92        sizer_fit.Add((5,5))
93        sizer_fit.Add(self.btFit,1, 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        from sans.guiframe.utils import check_value
316        flag = check_value( self.qmin, self.qmax) 
317       
318        if not flag:
319            msg= "Fitting range invalid"
320            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
321            return 
322       
323        if len(self.param_toFit) <= 0:
324            msg= "Select at least one parameter to fit"
325            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
326            return 
327       
328        self.qmin_x=float(self.qmin.GetValue())
329        self.qmax_x =float( self.qmax.GetValue())
330       
331        self.manager.schedule_for_fit( value=1,fitproblem =None) 
332       
333        #single fit
334        self.manager.on_single_fit(qmin=self.qmin_x,qmax=self.qmax_x)
335           
336        self.sizer5.Layout()
337        self.SetScrollbars(20,20,55,40)
338       
339       
340    def _on_select_model(self, event): 
341        """
342             call back for model selection
343        """   
344        self._on_select_model_helper(event) 
345        self.set_model_param_sizer(self.model)
346        self._set_sizer_gaussian()
347       
348        evt = ModelEventbox(model=self.model)
349        wx.PostEvent(self.event_owner, evt)   
350        try:
351            self.compute_chisqr()
352            self.text1_1.Show()
353            self.tcChi.Show()
354        except:
355            ## error occured on chisqr computation
356            pass
357    def get_range(self):
358        """
359            return the fitting range
360        """
361        return self.qmin_x , self.qmax_x
362       
363    def get_param_list(self):
364        """
365            @return self.param_toFit: list containing  references to TextCtrl
366            checked.Theses TextCtrl will allow reference to parameters to fit.
367            @raise: if return an empty list of parameter fit will nnote work
368            properly so raise ValueError,"missing parameter to fit"
369        """
370        if self.param_toFit !=[]:
371            return self.param_toFit
372        else:
373            raise ValueError,"missing parameter to fit"   
374     
375    def onsetValues(self,chisqr, out,cov):
376        """
377            Build the panel from the fit result
378            @param chisqr:Value of the goodness of fit metric
379            @param out:list of parameter with the best value found during fitting
380            @param cov:Covariance matrix
381       
382        """
383        self.tcChi.SetLabel(format_number(chisqr))
384        params = {}
385        is_modified = False
386        has_error = False
387        self.text2_3.Hide()
388        if self.text_disp_1 !=None:
389            self.text_disp_1.Hide()
390        #set the panel when fit result are float not list
391        if out.__class__==numpy.float64:
392            self.param_toFit[0][2].SetValue(format_number(out))
393            self.param_toFit[0][2].Refresh()
394           
395            self.param_toFit[0][4].Clear()
396            self.param_toFit[0][4].Hide()
397            if cov !=None :
398                self.text2_3.Show(True)
399                if self.text_disp_1 !=None:
400                    self.text_disp_1.Show(True)
401                   
402                self.param_toFit[0][3].Show(True)
403                self.param_toFit[0][4].Clear()
404                self.param_toFit[0][4].SetValue(format_number(cov[0]))
405                self.param_toFit[0][4].Show(True)
406        else:
407            i=0
408            j=0
409            #Set the panel when fit result are list
410            for item in self.param_toFit:
411                ## reset error value to initial state
412                item[4].Clear()
413                item[4].Hide()
414                item[4].Refresh()
415                if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
416                    item[2].SetValue(format_number(self.model.getParam(item[1])))
417                    item[2].Refresh()
418                if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
419                    self.text2_3.Show(True) 
420                    if self.text_disp_1!=None:
421                        self.text_disp_1.Show(True)
422                    item[3].Show(True)
423                    item[4].Clear()
424                    for j in range(len(out)):
425                        if out[j]==self.model.getParam(item[1]):
426                            break
427                    item[4].SetValue(format_number(cov[j]))
428                    item[4].Refresh()
429                    item[4].Show(True)   
430                i+=1
431       
432        self.sizer3.Layout()
433        self.sizer4.Layout()
434        self.SetScrollbars(20,20,200,100)
435       
436       
437     
438    def onSmear(self, event):
439        """
440            Create a smear object that will change the way residuals
441            are compute when fitting
442        """
443        smear =None
444        msg=""
445        if self.enable_smearer.GetValue():
446            from DataLoader.qsmearing import smear_selection
447            smear = smear_selection( self.data )
448            if hasattr(self.data,"dxl"):
449                msg= ": Resolution smearing parameters"
450            if hasattr(self.data,"dxw"):
451                msg= ": Slit smearing parameters"
452            if smear ==None:
453                wx.PostEvent(self.manager.parent, StatusEvent(status=\
454                            "Data contains no smearing information"))
455            else:
456                wx.PostEvent(self.manager.parent, StatusEvent(status=\
457                            "Data contains smearing information %s"%msg))
458            self.manager.set_smearer(smear, qmin= self.qmin_x, qmax= self.qmax_x)   
459             
460       
461       
462    def compute_chisqr2D(self):
463        """
464            compute chi square given a model and data 2D and set the value
465            to the tcChi txtcrl
466        """
467        from sans.guiframe.utils import check_value
468        flag = check_value( self.qmin, self.qmax)
469       
470        res=[]
471        if flag== True:
472            try:
473                self.qmin_x = float(self.qmin.GetValue())
474                self.qmax_x = float(self.qmax.GetValue())
475                for i in range(len(self.data.x_bins)):
476                    for j in range(len(self.data.y_bins)):
477                        #Check the range containing data between self.qmin_x and self.qmax_x
478                        if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)>=math.pow(self.qmin_x,2):
479                            if math.pow(self.data.x_bins[i],2)+math.pow(self.data.y_bins[j],2)<=math.pow(self.qmax_x,2):
480                                chisqrji=(self.data.data[j][i]- self.model.runXY(\
481                                                                                    [self.data.y_bins[j],self.data.x_bins[i]]))\
482                                                                                    /self.data.err_data[j][i]
483                                #Vector containing residuals
484                                res.append( math.pow(chisqrji,2) )
485                # compute sum of residual
486                sum=0
487                for item in res:
488                    if numpy.isfinite(item):
489                        sum +=item
490                self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
491            except:
492                wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
493                            "Chisqr cannot be compute: %s"% sys.exc_value))
494                return
495   
496       
497    def compute_chisqr(self):
498        """
499            compute chi square given a model and data 1D and set the value
500            to the tcChi txtcrl
501        """
502        from sans.guiframe.utils import check_value
503        flag = check_value( self.qmin, self.qmax)
504       
505        if flag== True:
506            try:
507                if hasattr(self.data,"data"):
508                    self.compute_chisqr2D()
509                    return
510                else:
511                    self.qmin_x = float(self.qmin.GetValue())
512                    self.qmax_x = float(self.qmax.GetValue())
513                    # return residuals within self.qmin_x and self.qmax_x
514                    x,y,dy = [numpy.asarray(v) for v in (self.data.x,self.data.y,self.data.dy)]
515                    if self.qmin_x==None and self.qmax_x==None: 
516                        fx =numpy.asarray([self.model.run(v) for v in x])
517                        temp=(y - fx)/dy
518                        res= temp*temp
519                    else:
520                        idx = (x>= self.qmin_x) & (x <=self.qmax_x)
521                        fx = numpy.asarray([self.model.run(item)for item in x[idx ]])
522                        temp=(y[idx] - fx)/dy[idx]
523                        res= temp*temp
524                    #sum of residuals
525                    sum=0
526                    for item in res:
527                        if numpy.isfinite(item):
528                            sum +=item
529                    self.tcChi.SetLabel(format_number(math.fabs(sum/ len(res))))
530            except:
531                wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
532                            "Chisqr cannot be compute: %s"% sys.exc_value))
533                return 
534           
535   
536    def select_all_param(self,event): 
537        """
538             set to true or false all checkBox given the main checkbox value cb1
539        """
540        self.param_toFit=[]
541        if  self.parameters !=[]:
542            if  self.cb1.GetValue()==True:
543                for item in self.parameters:
544                    item[0].SetValue(True)
545                    self.param_toFit.append(item )
546                if len(self.fittable_param)>0:
547                    for item in self.fittable_param:
548                        item[0].SetValue(True)
549                        self.param_toFit.append(item )
550            else:
551                for item in self.parameters:
552                    item[0].SetValue(False)
553                for item in self.fittable_param:
554                    item[0].SetValue(False)
555                self.param_toFit=[]
556               
557               
558               
559    def select_param(self,event):
560        """
561            Select TextCtrl  checked for fitting purpose and stores them
562            in  self.param_toFit=[] list
563        """
564        self.param_toFit=[]
565        for item in self.parameters:
566            #Select parameters to fit for list of primary parameters
567            if item[0].GetValue()==True:
568                if not (item in self.param_toFit):
569                    self.param_toFit.append(item ) 
570            else:
571                #remove parameters from the fitting list
572                if item in self.param_toFit:
573                    self.param_toFit.remove(item)
574        #Select parameters to fit for list of fittable parameters with dispersion         
575        for item in self.fittable_param:
576            if item[0].GetValue()==True:
577                if not (item in self.param_toFit):
578                    self.param_toFit.append(item) 
579            else:
580                #remove parameters from the fitting list
581                if item in self.param_toFit:
582                    self.param_toFit.remove(item)           
583        #Set the value of checkbox that selected every checkbox or not           
584        if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
585            self.cb1.SetValue(True)
586        else:
587            self.cb1.SetValue(False)
588   
589   
590    def set_model_description(self,description,sizer):
591        """
592            fill a sizer with description
593            @param description: of type string
594            @param sizer: wx.BoxSizer()
595        """
596   
597        sizer.Clear(True)
598        box_description= wx.StaticBox(self, -1, 'Model Description')
599        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
600               
601        self.description = wx.StaticText( self,-1,str(description) )
602        sizer_selection=wx.BoxSizer(wx.HORIZONTAL)
603       
604       
605        self.description_show = wx.RadioButton(self, -1, 'Show', style=wx.RB_GROUP)
606        self.description_hide = wx.RadioButton(self, -1, 'Hide')
607       
608        self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
609                   id=self.description_hide.GetId() )
610       
611        self.Bind( wx.EVT_RADIOBUTTON, self._on_display_description,
612                   id=self.description_show.GetId() )
613       
614        self.model_description = wx.Button(self, -1, "More Details")
615     
616       
617        self.page_info.save_radiobox_state( self.description_hide )
618        self.page_info.save_radiobox_state( self.description_show )
619       
620        sizer_selection.Add( self.description_show )
621        sizer_selection.Add( (20,20)) 
622        sizer_selection.Add( self.description_hide )
623        sizer_selection.Add( (20,20)) 
624        sizer_selection.Add( self.model_description )
625       
626         
627        self.sizer_description=wx.BoxSizer(wx.HORIZONTAL)
628        self.sizer_description.Add( self.description, 1, wx.EXPAND | wx.ALL, 10 )
629       
630        boxsizer1.Add( sizer_selection) 
631        boxsizer1.Add( (20,20)) 
632        boxsizer1.Add( self.sizer_description) 
633   
634
635        sizer.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
636        sizer.Layout()
637   
638   
639       
640       
641    def set_model_param_sizer(self, model, engine_type="park"):
642        """
643            Build the panel from the model content
644            @param model: the model selected in combo box for fitting purpose
645        """
646        self.sizer3.Clear(True)
647        self.parameters = []
648        self.param_toFit=[]
649        self.fittable_param=[]
650        self.fixed_param=[]
651       
652        if model ==None:
653            self.sizer3.Layout()
654            self.SetScrollbars(20,20,200,100)
655            return
656        box_description= wx.StaticBox(self, -1,str("Model Parameters"))
657        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
658        sizer = wx.GridBagSizer(5,5)
659        ## save the current model
660        self.model = model
661           
662        keys = self.model.getParamList()
663        #list of dispersion paramaters
664        self.disp_list=self.model.getDispParamList()
665       
666        keys.sort()
667   
668        iy = 1
669        ix = 0
670        self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
671        wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
672        self.cb1.SetValue(False)
673       
674        sizer.Add(self.cb1,(iy, ix),(1,1),\
675                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
676        ix +=1
677        self.text2_2 = wx.StaticText(self, -1, 'Values')
678        sizer.Add(self.text2_2,(iy, ix),(1,1),\
679                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
680        ix +=2 
681        self.text2_3 = wx.StaticText(self, -1, 'Errors')
682        sizer.Add(self.text2_3,(iy, ix),(1,1),\
683                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
684        self.text2_3.Hide()
685        ix +=1 
686        self.text2_min = wx.StaticText(self, -1, 'Min')
687        sizer.Add(self.text2_min,(iy, ix),(1,1),\
688                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
689        self.text2_min.Hide()
690        ix +=1 
691        self.text2_max = wx.StaticText(self, -1, 'Max')
692        sizer.Add(self.text2_max,(iy, ix),(1,1),\
693                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
694        self.text2_max.Hide()
695        ix += 1
696        self.text2_4 = wx.StaticText(self, -1, 'Units')
697        sizer.Add(self.text2_4,(iy, ix),(1,1),\
698                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
699        self.text2_4.Hide()
700        if  engine_type == "park":
701            self.text2_max.Show(True)
702            self.text2_min.Show(True)
703
704        for item in keys:
705            if not item in self.disp_list:
706                iy += 1
707                ix = 0
708                ## add parameters name with checkbox for selecting to fit
709                cb = wx.CheckBox(self, -1, item )
710                cb.SetValue(False)
711                wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
712                sizer.Add( cb,( iy, ix),(1,1),
713                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
714                ## add parameter value
715                ix += 1
716                value= self.model.getParam(item)
717                ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20),
718                                    style=wx.TE_PROCESS_ENTER)
719               
720                ctl1.SetValue(str (format_number(value)))
721                ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
722                ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
723                sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
724                ## text to show error sign
725                ix += 1
726                text2=wx.StaticText(self, -1, '+/-')
727                sizer.Add(text2,(iy, ix),(1,1),\
728                                wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
729                text2.Hide() 
730                ## txtcrtl to add error from fit
731                ix += 1
732                ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
733                sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
734                ctl2.Hide()
735               
736                ix += 1
737                ctl3 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
738                ctl3.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
739                ctl3.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
740                sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
741                ctl3.Hide()
742               
743                ix += 1
744                ctl4 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER)
745                ctl4.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
746                ctl4.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
747                sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
748                ctl4.Hide()
749                if  engine_type == "park":
750                    ctl3.Show(True)
751                    ctl4.Show(True)
752                   
753                ix +=1
754                # Units
755                try:
756                    units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
757                except:
758                    units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
759                sizer.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
760               
761                ##[cb state, name, value, "+/-", error of fit, min, max , units]
762                self.parameters.append([cb,item, ctl1,
763                                        text2,ctl2, ctl3, ctl4,None])
764               
765        iy+=1
766        sizer.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
767       
768        #Display units text on panel
769        for item in keys:   
770            if self.model.details[item][0]!='':
771                self.text2_4.Show()
772                break
773            else:
774                self.text2_4.Hide()
775   
776        boxsizer1.Add(sizer)
777       
778        self.sizer3.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
779        self.sizer3.Layout()
780        self.SetScrollbars(20,20,200,100)
781       
782   
783           
784       
785class HelpWindow(wx.Frame):
786    def __init__(self, parent, id, title):
787        wx.Frame.__init__(self, parent, id, title, size=(570, 400))
788       
789        from sans.models.CylinderModel import CylinderModel
790        model = CylinderModel()
791        #from sans.models.LineModel import LineModel
792        #model = LineModel()
793        from danse.common.plottools.plottables import Data1D
794        data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
795   
796        from pageInfo import PageInfo
797        myinfo = PageInfo(self,  model, data=data )
798       
799        ## add data
800       
801        from models import ModelList
802        mylist= ModelList()
803
804        from sans.models.SphereModel import SphereModel
805        from sans.models.SquareWellStructure import SquareWellStructure
806        from sans.models.DebyeModel import DebyeModel
807        from sans.models.LineModel import LineModel
808        name= "shapes"
809        list1= [SphereModel]
810        mylist.set_list( name, list1)
811       
812        name= "Shape-independent"
813        list1= [DebyeModel]
814        mylist.set_list( name, list1)
815       
816        name= "Structure Factors"
817        list1= [SquareWellStructure]
818        mylist.set_list( name, list1)
819       
820        name= "Added models"
821        list1= [LineModel]
822        mylist.set_list( name, list1)
823       
824        myinfo.model_list_box = mylist.get_list()
825       
826        self.page = FitPage(self, myinfo) 
827       
828       
829       
830        self.Centre()
831        self.Show(True)
832
833
834   
835if __name__=="__main__":
836    app = wx.App()
837    HelpWindow(None, -1, 'HelpWindow')
838    app.MainLoop()
839               
Note: See TracBrowser for help on using the repository browser.