source: sasview/sansview/perspectives/fitting/fitpage.py @ 4043c96

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 4043c96 was 4043c96, checked in by Jae Cho <jhjcho@…>, 15 years ago

fixed a crashing-bug. Need to look at it tomorrow

  • Property mode set to 100644
File size: 63.5 KB
Line 
1
2
3import sys
4import wx
5import wx.lib.newevent
6import numpy
7import copy
8import math
9from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
10
11from sans.guicomm.events import StatusEvent   
12from sans.guiframe.utils import format_number,check_float
13
14## event to post model to fit to fitting plugins
15(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
16
17## event to know the selected fit engine
18(FitterTypeEvent, EVT_FITTER_TYPE)   = wx.lib.newevent.NewEvent()
19(FitStopEvent, EVT_FIT_STOP)   = wx.lib.newevent.NewEvent()
20_BOX_WIDTH = 76
21
22import basepage
23from basepage import BasicPage
24from basepage import PageInfoEvent
25from DataLoader.qsmearing import smear_selection
26
27class FitPage(BasicPage):
28    """
29        FitPanel class contains fields allowing to display results when
30        fitting  a model and one data
31        @note: For Fit to be performed the user should check at least one parameter
32        on fit Panel window.
33 
34    """
35    def __init__(self,parent, page_info):
36        BasicPage.__init__(self, parent, page_info)
37        """
38            Initialization of the Panel
39        """
40        ## fit page does not content npts txtcrtl
41        self.npts=None
42        ## thread for compute Chisqr
43        self.calc_Chisqr=None
44        ## default fitengine type
45        self.engine_type = None
46        ## draw sizer
47        self._fill_datainfo_sizer()
48       
49        self._fill_model_sizer( self.sizer1)
50 
51        self._fill_range_sizer() 
52        #self._on_select_model(event=None)
53        if self.data !=None:
54            self.smearer = smear_selection( self.data )
55            if self.smearer ==None:
56                self.enable_smearer.Disable()
57                self.disable_smearer.Disable()
58       
59        ## to update the panel according to the fit engine type selected
60        self.Bind(EVT_FITTER_TYPE,self._on_engine_change)
61        self.Bind(EVT_FIT_STOP,self._on_fit_complete)
62   
63    def _on_fit_complete(self, event):
64        """
65            When fit is complete ,reset the fit button label.
66        """
67        #self.btFit.SetLabel("Fit")
68        #self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
69        #self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
70        pass
71       
72       
73    def _on_engine_change(self, event):
74        """
75            get an event containing the current name of the fit engine type
76            @param event: FitterTypeEvent containing  the name of the current engine
77        """
78        self.engine_type = event.type
79         
80        if len(self.parameters)==0:
81            return
82        if event.type =="park":
83            self.btFit.SetLabel("Fit")
84           
85        for item in self.parameters:
86            if event.type =="scipy" :
87                item[5].SetValue("")
88                item[5].Hide()
89                item[6].SetValue("")
90                item[6].Hide()
91                self.text2_min.Hide()
92                self.text2_max.Hide()
93            else:
94                item[5].Show(True)
95                item[6].Show(True)
96                self.text2_min.Show(True)
97                self.text2_max.Show(True)
98        for item in self.fittable_param:
99            if item[5]!=None and item[6]!=None and not item in self.orientation_params_disp:
100                if event.type =="scipy" and not item in self.orientation_params:
101                    item[5].SetValue("")
102                    item[5].Hide()
103                    item[6].SetValue("")
104                    item[6].Hide()
105                    self.text2_min.Hide()
106                    self.text2_max.Hide()
107                    self.text_disp_min.Hide()
108                    self.text_disp_max.Hide()
109                else:
110                    item[5].Show(True)
111                    item[6].Show(True)
112                    self.text2_min.Show(True)
113                    self.text2_max.Show(True)
114                    self.text_disp_min.Show(True)
115                    self.text_disp_max.Show(True)
116           
117        for item in self.orientation_params:
118            if item[5]!=None and item[6]!=None:
119                if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
120                    item[5].SetValue("")
121                    item[5].Hide()
122                    item[6].SetValue("")
123                    item[6].Hide()
124                else:
125                    item[5].Show(True)
126                    item[6].Show(True)
127                   
128        for item in self.orientation_params_disp:         
129            if item[5]!=None and item[6]!=None:
130                if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
131                    item[5].SetValue("")
132                    item[5].Hide()
133                    item[6].SetValue("")
134                    item[6].Hide()
135                else:
136                    item[5].Show(True)
137                    item[6].Show(True)
138     
139        self.sizer3.Layout()
140        self.SetScrollbars(20,20,25,65)
141       
142   
143    def _fill_range_sizer(self):
144        """
145            Fill the sizer containing the plotting range
146            add  access to npts
147        """
148        sizer_fit = wx.GridSizer(1, 1,0, 0)
149   
150        self.btFit = wx.Button(self,wx.NewId(),'Fit')
151        self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
152        self.btFit.SetToolTipString("Perform fit.")
153     
154       
155        sizer_fit.Add((5,5),1, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)       
156        sizer_fit.Add(self.btFit,0, wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5) 
157       
158        sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
159        #Filling the sizer containing instruments smearing info.
160        self.disable_smearer = wx.RadioButton(self, -1, 'No', style=wx.RB_GROUP)
161        self.enable_smearer = wx.RadioButton(self, -1, 'Yes')
162        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
163        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
164       
165        sizer_smearer.Add(wx.StaticText(self,-1,'Instrument Smearing? '))
166        sizer_smearer.Add((10, 10))
167        sizer_smearer.Add( self.enable_smearer )
168        sizer_smearer.Add((10,10))
169        sizer_smearer.Add( self.disable_smearer )
170       
171        #Display Chi^2/dof
172        sizer_smearer.Add((68,10))
173        box_description= wx.StaticBox(self, -1,'Chi2/dof')
174        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
175        boxsizer1.SetMinSize((60,-1))
176        temp_smearer = None
177        if self.enable_smearer.GetValue():
178            temp_smearer= self.smearer
179       
180        self.tcChi    =  wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
181         
182        boxsizer1.Add( self.tcChi )   
183        sizer_smearer.Add( boxsizer1 )
184               
185        #Set sizer for Fitting section
186        self._set_range_sizer( title="Fitting",object1=sizer_smearer, object= sizer_fit)
187 
188       
189    def _fill_datainfo_sizer(self):
190        """
191            fill sizer 0 with data info
192        """
193        self.sizer0.Clear(True)
194        ## no loaded data , don't fill the sizer
195        if self.data== None:
196            self.sizer0.Layout()
197            return
198       
199        box_description= wx.StaticBox(self, -1, 'Data')
200        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
201        #----------------------------------------------------------
202        sizer_data = wx.GridSizer(3, 3,5, 5)
203        #Filling the sizer containing data related fields
204        DataSource  =wx.StaticText(self, -1,str(self.data.name))
205
206        sizer_data.Add(wx.StaticText(self, -1, 'Source Name : '))
207        sizer_data.Add(DataSource )
208        sizer_data.Add( (0,5) )
209       
210        #---------sizer 2 draw--------------------------------
211        #set maximum range for x in linear scale
212        if not hasattr(self.data,"data"): #Display only for 1D data fit
213            # Minimum value of data   
214            data_min = min(self.data.x)
215            # Maximum value of data 
216            data_max = max(self.data.x)
217            text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
218                                     style=wx.ALIGN_LEFT)
219            sizer_data.Add( text4_3 )
220            sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%str(data_min)))
221            sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%str(data_max)))
222           
223        else:
224            ## Minimum value of data
225            data_min= 0
226            x= max(math.fabs(self.data.xmin), math.fabs(self.data.xmax)) 
227            y= max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
228            ## Maximum value of data 
229            data_max = math.sqrt(x*x + y*y)
230         
231            #For qmin and qmax, do not use format_number
232            #.(If do, qmin and max could be different from what is in the data.)
233            text4_3 = wx.StaticText(self, -1, "Total Q Range (1/A)",
234                                     style=wx.ALIGN_LEFT)
235            sizer_data.Add( text4_3 )
236            sizer_data.Add(wx.StaticText(self, -1, "Min : %s"%str(data_min)))
237            sizer_data.Add(wx.StaticText(self, -1, "Max : %s"%str(data_max)))
238        ## set q range to plot
239        self.qmin_x= data_min
240        self.qmax_x= data_max
241
242        boxsizer1.Add(sizer_data)
243        #------------------------------------------------------------
244        self.sizer0.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
245        self.sizer0.Layout()
246       
247    def _fill_model_sizer(self, sizer):
248        """
249            fill sizer containing model info
250        """
251        ##Add model function Details button in fitpanel.
252        id = wx.NewId()
253        self.model_help =wx.Button(self,id,'Details')
254        self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,id=id)
255        self.model_help.SetToolTipString("Model Function Help")
256       
257        ## class base method  to add view 2d button   
258        self._set_model_sizer(sizer=sizer, title="Model",object=self.model_help )   
259
260   
261    #def _set_sizer_gaussian(self):
262    def _set_sizer_dispersion(self, dispersity):
263        """
264            draw sizer with gaussian dispersity parameters
265        """
266        self.fittable_param=[]
267        self.fixed_param=[]
268        self.orientation_params_disp=[]
269
270        self.sizer4_4.Clear(True)
271        if self.model==None:
272            ##no model is selected
273            return
274        if not self.enable_disp.GetValue():
275            ## the user didn't select dispersity display
276            return 
277
278        self._reset_dispersity()
279        # Create the dispersion objects
280        for item in self.model.dispersion.keys():
281            #disp_model =  GaussianDispersion()
282            disp_model = dispersity()
283            self._disp_obj_dict[item] = disp_model
284            self.model.set_dispersion(item, disp_model)
285            self.state._disp_obj_dict[item]= disp_model
286
287        ix=0
288        iy=1
289        disp = wx.StaticText(self, -1, 'Names')
290        self.sizer4_4.Add(disp,( iy, ix),(1,1), 
291                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
292        ix += 1 
293        values = wx.StaticText(self, -1, 'Sigmas (STD)')
294        self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
295        ix +=2 
296        self.text_disp_1 = wx.StaticText(self, -1, '')
297        self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
298                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
299        self.text_disp_1.Hide()
300       
301       
302        ix +=1 
303        self.text_disp_min = wx.StaticText(self, -1, 'Min')
304        self.sizer4_4.Add(self.text_disp_min,(iy, ix),(1,1),\
305                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
306        self.text_disp_min.Hide()
307        ix +=1 
308        self.text_disp_max = wx.StaticText(self, -1, 'Max')
309        self.sizer4_4.Add(self.text_disp_max,(iy, ix),(1,1),\
310                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
311        self.text_disp_max.Hide()
312                       
313       
314       
315       
316       
317       
318        ix += 1 
319        npts = wx.StaticText(self, -1, 'Npts')
320        self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
321        ix += 1 
322        nsigmas = wx.StaticText(self, -1, 'Nsigmas')
323        self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
324       
325        if self.engine_type=="park":
326            self.text_disp_max.Show(True)
327            self.text_disp_min.Show(True)
328
329        for item in self.model.dispersion.keys():
330            if not item in self.model.orientation_params:
331                self.disp_cb_dict[item]= None
332                name1=item+".width"
333                name2=item+".npts"
334                name3=item+".nsigmas"
335                iy += 1
336                for p in self.model.dispersion[item].keys(): 
337       
338                    if p=="width":
339                        ix = 0
340                        cb = wx.CheckBox(self, -1, name1, (10, 10))
341                        wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
342                        self.sizer4_4.Add( cb,( iy, ix),(1,1), 
343                                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
344                        ix = 1
345                        value= self.model.getParam(name1)
346                        ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
347                                            style=wx.TE_PROCESS_ENTER)
348                        ctl1.SetValue(str (format_number(value)))
349                        self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
350                        ## text to show error sign
351                        ix = 2
352                        text2=wx.StaticText(self, -1, '+/-')
353                        self.sizer4_4.Add(text2,(iy, ix),(1,1),
354                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
355                        text2.Hide() 
356                        ## txtcrtl to add error from fit
357                        ix = 3
358                        ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
359                        self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
360                        ctl2.Hide()
361                       
362                        #Prepare add min and max in model.details.(Maybe Not correct way)
363                        if not self.model.details.has_key(name1):
364                            self.model.details [name1] = ["",None, None]                           
365                        param_min, param_max= self.model.details[name1][1:]
366                        ix = 4
367                        ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
368                                                       kill_focus_callback = self._on_paramRangeEnter,
369                                                       set_focus_callback  = self._on_paramRangeEnter)
370                        if param_min ==None:
371                            ctl3.SetValue("")
372                        else:
373                            ctl3.SetValue(str(param_min))
374                        self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
375                        ctl3.Hide()
376               
377                        ix = 5
378                        ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
379                                                       kill_focus_callback = self._on_paramRangeEnter,
380                                                       set_focus_callback  = self._on_paramRangeEnter)
381                        self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
382                        if param_max==None:
383                            ctl4.SetValue("")
384                        else:
385                            ctl4.SetValue(str(param_max))
386                        ctl4.Hide()
387                       
388                        if self.engine_type=="park":
389                            ctl3.Show(True)
390                            ctl4.Show(True)
391                                         
392                        self.fittable_param.append([cb,name1,ctl1,text2,
393                                                    ctl2, ctl3, ctl4,None])
394                        #self.parameters.append([cb,name1, ctl1,text2,ctl2, ctl3, ctl4,None])
395                        #self.orientation_params.append([cb,name1, ctl1,text2,ctl2, ctl3, ctl4,None])
396                   
397                   
398                    elif p=="npts":
399                            ix = 6
400                            value= self.model.getParam(name2)
401                            Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
402                                                style=wx.TE_PROCESS_ENTER)
403                           
404                            Tctl.SetValue(str (format_number(value)))
405                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
406                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
407                            self.fixed_param.append([None,name2, Tctl,None,None,
408                                                      None, None,None])
409                    elif p=="nsigmas":
410                            ix = 7
411                            value= self.model.getParam(name3)
412                            Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
413                                                style=wx.TE_PROCESS_ENTER)
414                            Tctl.SetValue(str (format_number(value)))
415                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
416                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
417                            ix +=1
418                            self.sizer4_4.Add((20,20), (iy,ix),(1,1),
419                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
420                           
421                            self.fixed_param.append([None,name3, Tctl
422                                                     ,None,None,None, None,None])
423                           
424        ix =0
425        iy +=1 
426        self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
427        for item in self.model.dispersion.keys():
428            if  item in self.model.orientation_params:
429                self.disp_cb_dict[item]= None
430                name1=item+".width"
431                name2=item+".npts"
432                name3=item+".nsigmas"
433                iy += 1
434                for p in self.model.dispersion[item].keys(): 
435       
436                    if p=="width":
437                        ix = 0
438                        cb = wx.CheckBox(self, -1, name1, (10, 10))
439                        wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
440                        self.sizer4_4.Add( cb,( iy, ix),(1,1), 
441                                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
442                        if self.data.__class__.__name__ =="Data2D":
443                            cb.Show(True)
444                        else:
445                            cb.Hide()
446                        ix = 1
447                        value= self.model.getParam(name1)
448                        ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
449                                            style=wx.TE_PROCESS_ENTER)
450                        ctl1.SetValue(str (format_number(value)))
451                        if self.data.__class__.__name__ =="Data2D":
452                            ctl1.Show(True)
453                        else:
454                            ctl1.Hide()
455                        self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
456                        ## text to show error sign
457                        ix = 2
458                        text2=wx.StaticText(self, -1, '+/-')
459                        self.sizer4_4.Add(text2,(iy, ix),(1,1),
460                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
461                        text2.Hide() 
462                        ## txtcrtl to add error from fit
463                        ix = 3
464                        ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
465                        self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
466                        ctl2.Hide()
467                        if self.data.__class__.__name__ =="Data2D":
468                            ctl2.Enable()
469                        else:
470                            ctl2.Disable()
471                           
472                           
473                        if not self.model.details.has_key(name1):
474                            self.model.details [name1] = ["",None, None]                           
475               
476                        param_min, param_max= self.model.details[name1][1:]
477                        ix = 4
478                        ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
479                                                       kill_focus_callback = self._onparamRangeEnter,
480                                                       set_focus_callback  = self._onparamRangeEnter)
481                        if param_min ==None:
482                            ctl3.SetValue("")
483                        else:
484                            ctl3.SetValue(str(param_min))
485                        self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
486                        ctl3.Hide()
487                        if self.data.__class__.__name__ =="Data2D":
488                            ctl3.Enable()
489                        else:
490                            ctl3.Disable()
491               
492                        ix = 5
493                        ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
494                                                       kill_focus_callback = self._onparamRangeEnter,
495                                                       set_focus_callback  = self._onparamRangeEnter)
496                        self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
497                        if param_max==None:
498                            ctl4.SetValue("")
499                        else:
500                            ctl4.SetValue(str(param_max))
501                        ctl4.Hide()
502                        if self.data.__class__.__name__ =="Data2D":
503                            ctl4.Enable()
504                        else:
505                            ctl4.Disable()
506                       
507                        if self.engine_type=="park" and self.data.__class__.__name__ =="Data2D":
508                            ctl3.Show(True)
509                            ctl4.Show(True)                           
510                           
511                           
512                           
513                           
514                        self.fittable_param.append([cb,name1,ctl1,text2,
515                                                    ctl2, ctl3, ctl4,None])
516                        #self.parameters.append([cb,name1, ctl1,text2,ctl2, ctl3, ctl4,None])
517                        #self.orientation_params.append([cb,name1,ctl1,text2, ctl2, ctl3, ctl4,None])
518                        self.orientation_params_disp.append([cb,name1,ctl1,text2,
519                                                    ctl2, ctl3, ctl4,None])
520                    elif p=="npts":
521                            ix = 6
522                            value= self.model.getParam(name2)
523                            Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
524                                                style=wx.TE_PROCESS_ENTER)
525                           
526                            Tctl.SetValue(str (format_number(value)))
527                            if self.data.__class__.__name__ =="Data2D":
528                                Tctl.Show(True)
529                            else:
530                                Tctl.Hide()
531                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
532                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
533                            self.fixed_param.append([None,name2, Tctl,None,None,
534                                                      None, None,None])
535                            self.orientation_params_disp.append([None,name2, Tctl,None,None,
536                                                      None, None,None])
537                    elif p=="nsigmas":
538                            ix = 7
539                            value= self.model.getParam(name3)
540                            Tctl = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
541                                                style=wx.TE_PROCESS_ENTER)
542                            Tctl.SetValue(str (format_number(value)))
543                            if self.data.__class__.__name__ =="Data2D":
544                                Tctl.Show(True)
545                            else:
546                                Tctl.Hide()
547                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
548                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
549                            ix +=1
550                            #self.sizer4_4.Add((20,20), (iy,ix),(1,1),
551                                               #wx.EXPAND|wx.ADJUST_MINSIZE, 0)
552                            self.fixed_param.append([None,name3, Tctl
553                                                     ,None,None, None, None,None])   
554                                                       
555                            self.orientation_params_disp.append([None,name3, Tctl
556                                                     ,None,None, None, None,None])
557   
558        self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict)     
559        self.state.model = self.model.clone() 
560         ## save state into
561        self.state.cb1 = self.cb1.GetValue()
562        self._copy_parameters_state(self.parameters, self.state.parameters)
563        self._copy_parameters_state(self.orientation_params_disp,
564                                     self.state.orientation_params_disp)
565        self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
566        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
567
568
569        wx.PostEvent(self.parent, StatusEvent(status=\
570                        " Selected Distribution: Gaussian"))   
571        ix =0
572        iy +=1 
573        #self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)       
574        self.sizer4_4.Layout()
575        self.sizer4.Layout()
576        self.Layout()
577        self.SetScrollbars(20,20,25,65)
578        self.Refresh()
579     
580       
581    def _onFit(self, event):     
582        """
583            Allow to fit
584        """
585        from sans.guiframe.utils import check_value
586        flag = check_value( self.qmin, self.qmax) 
587       
588        if not flag:
589            msg= "Fitting range invalid"
590            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
591            return 
592       
593        if len(self.param_toFit) <= 0:
594            msg= "Select at least one parameter to fit"
595            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
596            return 
597
598        # Remove or do not allow fitting on the Q=0 point, especially when y(q=0)=None at x[0].
599        #ToDo: Fix this.               
600        self.qmin_x = float(self.qmin.GetValue())
601        self.qmax_x =float( self.qmax.GetValue())
602        self.manager._reset_schedule_problem( value=0)
603        self.manager.schedule_for_fit( value=1,page=self,fitproblem =None) 
604        self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
605       
606        #single fit
607        self.manager.onFit()
608        ## allow stopping the fit
609        #if self.engine_type=="scipy":
610        #    self.btFit.SetLabel("Stop")
611        #    self.btFit.Unbind(event=wx.EVT_BUTTON, id= self.btFit.GetId())
612        #    self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._StopFit, id=self.btFit.GetId())
613        #else:
614        #    self.btFit.SetLabel("Fit")
615        #    self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._onFit, id=self.btFit.GetId())
616        self.btFit.SetFocus()   
617        self.sizer5.Layout()
618        self.SetScrollbars(20,20,55,40)
619       
620    def _StopFit(self, event):
621        """
622            Stop fit
623        """
624        self.btFit.SetLabel("Fit")
625        if self.engine_type=="scipy":
626            self.manager.stop_fit()
627        self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
628        self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
629       
630           
631        self.btFit.SetFocus()   
632        self.sizer5.Layout()
633        self.SetScrollbars(20,20,55,40)   
634           
635    def _on_select_model(self, event): 
636        """
637             call back for model selection
638        """   
639        self._on_select_model_helper() 
640        self.set_model_param_sizer(self.model)                   
641       
642        self.enable_disp.SetValue(False)
643        self.disable_disp.SetValue(True)
644        try:
645            self.set_dispers_sizer()
646        except:
647            pass
648        if self.model !=None:
649            try:
650                temp_smear= None
651                if self.enable_smearer.GetValue():
652                    temp_smear= self.smearer
653                self.compute_chisqr(temp_smear)
654            except:
655                ## error occured on chisqr computation
656                pass
657            ## set smearing value whether or not the data contain the smearing info
658            self.manager.set_smearer(smearer=temp_smear, qmin= float(self.qmin_x),
659                                     qmax= float(self.qmax_x)) 
660            evt = ModelEventbox(model=self.model)
661            wx.PostEvent(self.event_owner, evt)   
662           
663        self.btFit.SetFocus() 
664        self.state.enable_disp = self.enable_disp.GetValue()
665        self.state.disable_disp = self.disable_disp.GetValue()
666   
667        self.state.structurecombobox = self.structurebox.GetCurrentSelection()
668        self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
669     
670       
671        if event !=None:
672            #self._undo.Enable(True)
673            ## post state to fit panel
674            event = PageInfoEvent(page = self)
675            wx.PostEvent(self.parent, event) 
676     
677   
678    def _onparamRangeEnter(self, event):
679        """
680            Check validity of value enter in the parameters range field
681        """
682        tcrtl= event.GetEventObject()
683        if tcrtl.GetValue().lstrip().rstrip()!="":
684            try:
685                value = float(tcrtl.GetValue())
686                tcrtl.SetBackgroundColour(wx.WHITE)
687                tcrtl.Refresh()
688            except:
689                tcrtl.SetBackgroundColour("pink")
690                tcrtl.Refresh()
691                return
692        else:
693           tcrtl.SetBackgroundColour(wx.WHITE)
694           tcrtl.Refresh() 
695        self._onparamEnter_helper() 
696       
697        ##compute chisqr for range change
698        temp_smearer = None
699        if self.enable_smearer.GetValue():
700            msg=""
701            temp_smearer= self.smearer   
702         ##Calculate chi2
703       
704        self.compute_chisqr(smearer= temp_smearer) 
705        ## new state posted
706        if self.state_change:
707            #self._undo.Enable(True)
708            event = PageInfoEvent(page = self)
709            wx.PostEvent(self.parent, event)
710            self.state_change= False
711       
712       
713    def _onparamEnter(self,event):
714        """
715            when enter value on panel redraw model according to changed
716        """
717        tcrtl= event.GetEventObject()
718       
719        if check_float(tcrtl):
720            self._onparamEnter_helper()
721            temp_smearer = None
722            if self.enable_smearer.GetValue():
723                temp_smearer= self.smearer
724           
725            self.compute_chisqr(smearer= temp_smearer)
726            ## new state posted
727            if self.state_change:
728                #self._undo.Enable(True)
729                event = PageInfoEvent(page = self)
730                wx.PostEvent(self.parent, event)
731                self.state_change= False
732        else:
733            msg= "Cannot Plot :Must enter a number!!!  "
734            wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
735            return 
736       
737    ### Same as _onparamRangeEnter but smear and chi^2 calculations are removed
738    ### and _onparamEnter_helper is not used, which causing mess sizer display
739    def _on_paramRangeEnter(self, event):
740        """
741            Check validity of value enter in the parameters range field
742        """
743        tcrtl= event.GetEventObject()
744        if tcrtl.GetValue().lstrip().rstrip()!="":
745            try:
746                value = float(tcrtl.GetValue())
747                tcrtl.SetBackgroundColour(wx.WHITE)
748                tcrtl.Refresh()
749            except:
750                tcrtl.SetBackgroundColour("pink")
751                tcrtl.Refresh()
752                return
753        else:
754           tcrtl.SetBackgroundColour(wx.WHITE)
755           tcrtl.Refresh() 
756        self._onparamEnter_helper() 
757       
758        ## new state posted
759        if self.state_change:
760            #self._undo.Enable(True)
761            event = PageInfoEvent(page = self)
762            wx.PostEvent(self.parent, event)
763            self.state_change= False
764       
765               
766       
767    def set_data(self, data ):
768        """
769            reset the current data
770        """
771        if data ==None:
772            return 
773        self.data =data
774        self.state.data= data
775        self._fill_datainfo_sizer()
776        self.SetScrollbars(20,20,25,65)
777        self.Layout()   
778       
779    def reset_page(self, state,first=False):
780        """
781            reset the state
782        """
783        #if first:
784        #    self._undo.Enable(False)
785           
786        self.reset_page_helper(state)
787        evt = ModelEventbox(model=state.model)
788        wx.PostEvent(self.event_owner, evt)   
789           
790           
791    def get_range(self):
792        """
793            return the fitting range
794        """
795        return float(self.qmin_x) , float(self.qmax_x)
796       
797    def get_param_list(self):
798        """
799            @return self.param_toFit: list containing  references to TextCtrl
800            checked.Theses TextCtrl will allow reference to parameters to fit.
801            @raise: if return an empty list of parameter fit will nnote work
802            properly so raise ValueError,"missing parameter to fit"
803        """
804        if self.param_toFit !=[]:
805            return self.param_toFit
806        else:
807            raise ValueError,"missing parameter to fit"   
808     
809    def onsetValues(self,chisqr, out,cov):
810        """
811            Build the panel from the fit result
812            @param chisqr:Value of the goodness of fit metric
813            @param out:list of parameter with the best value found during fitting
814            @param cov:Covariance matrix
815       
816        """
817        self.tcChi.SetLabel(format_number(chisqr))
818        params = {}
819        is_modified = False
820        has_error = False
821        self.text2_3.Hide()
822        try:
823            n = self.disp_box.GetCurrentSelection()
824            dispersity= self.disp_box.GetClientData(n)
825            name= dispersity.__name__
826            if name == "GaussianDispersion":
827                if hasattr(self,"text_disp_1" ):
828                    if self.text_disp_1 !=None:
829                        self.text_disp_1.Hide()
830        except:
831            pass
832        #set the panel when fit result are float not list
833        if out.__class__==numpy.float64:
834            self.param_toFit[0][2].SetValue(format_number(out))
835            self.param_toFit[0][2].Refresh()
836           
837            self.param_toFit[0][4].Clear()
838            self.param_toFit[0][4].Hide()
839            if cov !=None :
840                self.text2_3.Show(True)
841                try:
842                    name= dispersity.__name__
843                    if name == "GaussianDispersion":
844                        if hasattr(self,"text_disp_1" ):
845                            if self.text_disp_1 !=None:
846                                self.text_disp_1.Show(True)
847                except:
848                    pass
849                if cov[0]==None or  not numpy.isfinite(cov[0]): 
850                    self.param_toFit[0][3].Hide()
851                    self.param_toFit[0][4].Clear()
852                    self.param_toFit[0][4].Hide()
853                    self.param_toFit[0][4].Refresh()
854                else:
855                    self.param_toFit[0][3].Show(True)
856                    self.param_toFit[0][4].Clear()
857                    self.param_toFit[0][4].SetValue(format_number(cov[0]))
858                    self.param_toFit[0][4].Show(True)
859                    self.param_toFit[0][4].Refresh()
860        else:
861            i=0
862            j=0
863            #Set the panel when fit result are list
864            for item in self.param_toFit:
865                ## reset error value to initial state
866                item[4].Clear()
867                item[4].Hide()
868                item[4].Refresh()
869                if( out != None ) and len(out)<=len(self.param_toFit)and i < len(out):
870                    item[2].SetValue(format_number(self.model.getParam(item[1])))
871                    item[2].Refresh()
872                if(cov !=None)and len(cov)<=len(self.param_toFit)and i < len(cov):
873                    self.text2_3.Show(True) 
874                    try:
875                        name= dispersity.__name__
876                        if name == "GaussianDispersion":
877                            if hasattr(self,"text_disp_1" ):
878                                if self.text_disp_1!=None:
879                                    self.text_disp_1.Show(True)
880                    except:
881                        pass
882                    item[3].Show(True)
883                    item[4].Clear()
884                    for j in range(len(out)):
885                        if out[j]==self.model.getParam(item[1]):
886                            break
887                    ## unable to compare cov[j]==numpy.nan so switch to None
888                    if cov[j]==None or not numpy.isfinite(cov[j]):
889                        item[3].Hide()
890                        item[4].Refresh()
891                        item[4].Clear()
892                        item[4].Hide()
893                    else:
894                        item[4].SetValue(format_number(cov[j]))
895                        item[4].Refresh()
896                        item[4].Show(True)   
897                i+=1
898       
899        self.sizer3.Layout()
900        self.sizer4.Layout()
901        self.Layout()
902        self.Refresh()
903        self.SetScrollbars(20,20,25,65)
904       
905       
906    def onSmear(self, event):
907        """
908            Create a smear object that will change the way residuals
909            are compute when fitting
910        """
911        if self.model ==None:
912            msg="Need model and data to smear plot"
913            wx.PostEvent(self.manager.parent, StatusEvent(status=\
914                            "Smear: %s"%msg))
915            return
916        temp_smearer = None
917        if self.enable_smearer.GetValue():
918            temp_smearer= self.smearer
919            if hasattr(self.data,"dxl"):
920                msg= ": Resolution smearing parameters"
921            if hasattr(self.data,"dxw"):
922                msg= ": Slit smearing parameters"
923            if self.smearer ==None:
924                wx.PostEvent(self.manager.parent, StatusEvent(status=\
925                            "Data contains no smearing information"))
926            else:
927                wx.PostEvent(self.manager.parent, StatusEvent(status=\
928                            "Data contains smearing information %s"%msg))
929       
930        ## set smearing value whether or not the data contain the smearing info
931        self.manager.set_smearer(smearer=temp_smearer, qmin= float(self.qmin_x),
932                                     qmax= float(self.qmax_x)) 
933        ##Calculate chi2
934        self.compute_chisqr(smearer= temp_smearer) 
935       
936        self.state.enable_smearer=  self.enable_smearer.GetValue()
937        self.state.disable_smearer=self.disable_smearer.GetValue()
938   
939    def complete_chisqr(self, output, elapsed=None): 
940        """
941            print result chisqr
942        """
943        try:
944            if output ==None:
945                output= "-"
946            self.tcChi.SetLabel(str(format_number(output)))
947           
948            self.sizer5.Layout()
949            self.Layout()
950            self.Refresh
951            ## post state to fit panel
952            self.state.tcChi =output
953         
954        except:
955            pass
956       
957       
958    def compute_chisqr1D(self, smearer=None):
959        """
960            Compute chisqr for 1D
961        """
962        from sans.guiframe.utils import check_value
963        flag = check_value( self.qmin, self.qmax)
964       
965        if not flag:
966            return 
967       
968        try:
969            self.qmin_x = float(self.qmin.GetValue())
970            self.qmax_x = float(self.qmax.GetValue())
971            ##return residuals within self.qmin_x and self.qmax_x
972            from gui_thread import CalcChisqr1D
973            ## If a thread is already started, stop it
974            if self.calc_Chisqr!= None and self.calc_Chisqr.isrunning():
975                self.calc_Chisqr.stop()
976               
977            self.calc_Chisqr= CalcChisqr1D( data1d= self.data,
978                                            model= self.model,
979                                            smearer=smearer,
980                                            qmin=self.qmin_x,
981                                            qmax=self.qmax_x,
982                                            completefn = self.complete_chisqr,
983                                            updatefn   = None)
984   
985            self.calc_Chisqr.queue()
986           
987        except:
988            raise ValueError," Could not compute Chisqr for %s Model 2D: "%self.model.name
989           
990           
991   
992       
993       
994    def compute_chisqr2D(self):
995        """
996            compute chi square given a model and data 2D and set the value
997            to the tcChi txtcrl
998        """
999        from sans.guiframe.utils import check_value
1000        flag = check_value( self.qmin, self.qmax)
1001        if not flag:
1002            return 
1003     
1004        try:
1005            self.qmin_x = float(self.qmin.GetValue())
1006            self.qmax_x = float(self.qmax.GetValue())
1007           
1008            ##return residuals within self.qmin_x and self.qmax_x
1009            from gui_thread import CalcChisqr2D
1010            ## If a thread is already started, stop it
1011            if self.calc_Chisqr!= None and self.calc_Chisqr.isrunning():
1012                self.calc_Chisqr.stop()
1013           
1014            self.calc_Chisqr= CalcChisqr2D( data2d= self.data,
1015                                            model= self.model,
1016                                            qmin= self.qmin_x,
1017                                            qmax = self.qmax_x,
1018                                            completefn = self.complete_chisqr,
1019                                            updatefn   = None)
1020   
1021            self.calc_Chisqr.queue()
1022         
1023        except:
1024           raise
1025
1026       
1027    def compute_chisqr(self , smearer=None):
1028        """
1029            compute chi square given a model and data 1D and set the value
1030            to the tcChi txtcrl
1031        """
1032        from sans.guiframe.utils import check_value
1033        flag = check_value( self.qmin, self.qmax)
1034        if flag== True:
1035            try:
1036                if hasattr(self.data,"data"):
1037                    self.compute_chisqr2D()
1038                    return
1039                else:
1040                    self.compute_chisqr1D(smearer=smearer)
1041                    return
1042            except:
1043                wx.PostEvent(self.parent.parent, StatusEvent(status=\
1044                            "Chisqr Error: %s"% sys.exc_value))
1045                return 
1046           
1047   
1048    def select_all_param(self,event): 
1049        """
1050             set to true or false all checkBox given the main checkbox value cb1
1051        """           
1052
1053        self.param_toFit=[]
1054        if  self.parameters !=[]:
1055            if  self.cb1.GetValue():
1056                for item in self.parameters:
1057                    ## for data2D select all to fit
1058                    if self.data.__class__.__name__=="Data2D":
1059                        item[0].SetValue(True)
1060                        self.param_toFit.append(item )
1061                    else:
1062                        ## for 1D all parameters except orientation
1063                        if not item in self.orientation_params:
1064                            item[0].SetValue(True)
1065                            self.param_toFit.append(item )
1066                if len(self.fittable_param)>0:
1067                    for item in self.fittable_param:
1068                        if self.data.__class__.__name__=="Data2D":
1069                            item[0].SetValue(True)
1070                            self.param_toFit.append(item )
1071                        else:
1072                            ## for 1D all parameters except orientation
1073                            if not item in self.orientation_params_disp:
1074                                item[0].SetValue(True)
1075                                self.param_toFit.append(item )
1076            else:
1077                for item in self.parameters:
1078                    item[0].SetValue(False)
1079                for item in self.fittable_param:
1080                    item[0].SetValue(False)
1081                self.param_toFit=[]
1082        self._copy_parameters_state(self.orientation_params,
1083                                     self.state.orientation_params)
1084        self._copy_parameters_state(self.orientation_params_disp,
1085                                     self.state.orientation_params_disp)
1086        self._copy_parameters_state(self.parameters, self.state.parameters)
1087        self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1088        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1089           
1090        self.save_current_state() 
1091        if event !=None:
1092            #self._undo.Enable(True)
1093            ## post state to fit panel
1094            event = PageInfoEvent(page = self)
1095            wx.PostEvent(self.parent, event) 
1096     
1097               
1098               
1099    def select_param(self,event):
1100        """
1101            Select TextCtrl  checked for fitting purpose and stores them
1102            in  self.param_toFit=[] list
1103        """
1104        self.param_toFit=[]
1105        for item in self.parameters:
1106            #Select parameters to fit for list of primary parameters
1107            if item[0].GetValue():
1108                if not (item in self.param_toFit):
1109                    self.param_toFit.append(item ) 
1110            else:
1111                #remove parameters from the fitting list
1112                if item in self.param_toFit:
1113                    self.param_toFit.remove(item)
1114        #Select parameters to fit for list of fittable parameters with dispersion         
1115        for item in self.fittable_param:
1116            if item[0].GetValue():
1117                if not (item in self.param_toFit):
1118                    self.param_toFit.append(item) 
1119            else:
1120                #remove parameters from the fitting list
1121                if item in self.param_toFit:
1122                    self.param_toFit.remove(item)           
1123        #Set the value of checkbox that selected every checkbox or not           
1124        if len(self.parameters)+len(self.fittable_param) ==len(self.param_toFit):
1125            self.cb1.SetValue(True)
1126        else:
1127            self.cb1.SetValue(False)
1128        ## save current state of the page
1129        self.state.cb1 = self.cb1.GetValue()
1130        self._copy_parameters_state(self.orientation_params,
1131                                     self.state.orientation_params)
1132        self._copy_parameters_state(self.orientation_params_disp,
1133                                     self.state.orientation_params_disp)
1134        self._copy_parameters_state(self.parameters, self.state.parameters)
1135        self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1136        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1137       
1138        self.save_current_state()
1139        if event !=None:
1140            #self._undo.Enable(True)
1141            ## post state to fit panel
1142            event = PageInfoEvent(page = self)
1143            wx.PostEvent(self.parent, event) 
1144     
1145   
1146       
1147    def set_model_param_sizer(self, model):
1148        """
1149            Build the panel from the model content
1150            @param model: the model selected in combo box for fitting purpose
1151        """
1152        self.sizer3.Clear(True)
1153        self.parameters = []
1154        self.param_toFit=[]
1155        self.fittable_param=[]
1156        self.fixed_param=[]
1157        self.orientation_params=[]
1158        self.orientation_params_disp=[]
1159       
1160        if model ==None:
1161            self.sizer3.Layout()
1162            self.SetScrollbars(20,20,25,65)
1163            return
1164        ## the panel is drawn using the current value of the fit engine
1165        if self.engine_type==None and self.manager !=None:
1166            self.engine_type= self.manager._return_engine_type()
1167           
1168        box_description= wx.StaticBox(self, -1,str("Model Parameters"))
1169        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1170        sizer = wx.GridBagSizer(5,5)
1171        ## save the current model
1172        self.model = model
1173           
1174        keys = self.model.getParamList()
1175        #list of dispersion paramaters
1176        self.disp_list=self.model.getDispParamList()
1177       
1178        keys.sort()
1179   
1180        iy = 0
1181        ix = 0
1182        self.cb1 = wx.CheckBox(self, -1,"Select all", (10, 10))
1183        wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
1184        self.cb1.SetValue(False)
1185       
1186        sizer.Add(self.cb1,(iy, ix),(1,1),\
1187                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
1188        ix +=1
1189        self.text2_2 = wx.StaticText(self, -1, 'Values')
1190        sizer.Add(self.text2_2,(iy, ix),(1,1),\
1191                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1192        ix +=2 
1193        self.text2_3 = wx.StaticText(self, -1, 'Errors')
1194        sizer.Add(self.text2_3,(iy, ix),(1,1),\
1195                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
1196        self.text2_3.Hide()
1197        ix +=1 
1198        self.text2_min = wx.StaticText(self, -1, 'Min')
1199        sizer.Add(self.text2_min,(iy, ix),(1,1),\
1200                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
1201        self.text2_min.Hide()
1202        ix +=1 
1203        self.text2_max = wx.StaticText(self, -1, 'Max')
1204        sizer.Add(self.text2_max,(iy, ix),(1,1),\
1205                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
1206        self.text2_max.Hide()
1207        ix += 1
1208        self.text2_4 = wx.StaticText(self, -1, '[Units]')
1209        sizer.Add(self.text2_4,(iy, ix),(1,1),\
1210                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
1211        self.text2_4.Hide()
1212        if self.engine_type=="park":
1213            self.text2_max.Show(True)
1214            self.text2_min.Show(True)
1215
1216        for item in keys:
1217            if not item in self.disp_list and not item in self.model.orientation_params:
1218               
1219                iy += 1
1220                ix = 0
1221                ## add parameters name with checkbox for selecting to fit
1222                cb = wx.CheckBox(self, -1, item )
1223                cb.SetValue(False)
1224                wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
1225                sizer.Add( cb,( iy, ix),(1,1),
1226                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
1227               
1228                ## add parameter value
1229                ix += 1
1230                value= self.model.getParam(item)
1231                ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
1232                                    style=wx.TE_PROCESS_ENTER)
1233               
1234                ctl1.SetValue(format_number(value))
1235                sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
1236                ## text to show error sign
1237                ix += 1
1238                text2=wx.StaticText(self, -1, '+/-')
1239                sizer.Add(text2,(iy, ix),(1,1),\
1240                                wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
1241                text2.Hide() 
1242                ## txtcrtl to add error from fit
1243                ix += 1
1244                ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
1245                sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1246                ctl2.Hide()
1247               
1248                param_min, param_max= self.model.details[item][1:]
1249                ix += 1
1250                ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1251                                               kill_focus_callback = self._on_paramRangeEnter,
1252                                               set_focus_callback  = self._on_paramRangeEnter)
1253                if param_min ==None:
1254                    ctl3.SetValue("")
1255                else:
1256                    ctl3.SetValue(str(param_min))
1257                sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1258                ctl3.Hide()
1259       
1260                ix += 1
1261                ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1262                                               kill_focus_callback = self._on_paramRangeEnter,
1263                                               set_focus_callback  = self._on_paramRangeEnter)
1264                sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1265                if param_max==None:
1266                    ctl4.SetValue("")
1267                else:
1268                    ctl4.SetValue(str(param_max))
1269                ctl4.Hide()
1270               
1271                if self.engine_type=="park":
1272                    ctl3.Show(True)
1273                    ctl4.Show(True)
1274                ix +=1
1275                # Units
1276                try:
1277                    units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
1278                except:
1279                    units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
1280                sizer.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1281                   
1282                ##[cb state, name, value, "+/-", error of fit, min, max , units]
1283                self.parameters.append([cb,item, ctl1,
1284                                        text2,ctl2, ctl3, ctl4,units])
1285             
1286        iy+=1
1287        sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1288       
1289        # type can be either Guassian or Array
1290        if len(self.model.dispersion.values())>0:
1291            type= self.model.dispersion.values()[0]["type"]
1292        else:
1293            type = "Gaussian"
1294           
1295        iy += 1
1296        ix = 0
1297        #Add tile for orientational angle
1298        for item in keys:
1299            if item in self.model.orientation_params:       
1300                orient_angle = wx.StaticText(self, -1, '[For 2D only]:')
1301                sizer.Add(orient_angle,(iy, ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
1302                if not self.data.__class__.__name__ =="Data2D":
1303                    orient_angle.Hide()
1304                else:
1305                    orient_angle.Show(True)
1306                break
1307     
1308        #For Gaussian only
1309        if type.lower() != "array":
1310            for item in self.model.orientation_params:
1311                if not item in self.disp_list:
1312                    iy += 1
1313                    ix = 0
1314                    ## add parameters name with checkbox for selecting to fit
1315                    cb = wx.CheckBox(self, -1, item )
1316                    cb.SetValue(False)
1317                    wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
1318                    if self.data.__class__.__name__ =="Data2D":
1319                        cb.Show(True)
1320                        #cb.Enable()
1321                    else:
1322                        cb.Hide()
1323                        #cb.Disable()
1324                    sizer.Add( cb,( iy, ix),(1,1),
1325                                 wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
1326   
1327                    ## add parameter value
1328                    ix += 1
1329                    value= self.model.getParam(item)
1330                    ctl1 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
1331                                        style=wx.TE_PROCESS_ENTER)
1332                   
1333                    ctl1.SetValue(format_number(value))
1334                    if self.data.__class__.__name__ =="Data2D":
1335                        ctl1.Show(True)
1336                        #ctl1.Enable()
1337                    else:
1338                        ctl1.Hide()
1339                        #ctl1.Disable()
1340                    sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
1341                    ## text to show error sign
1342                    ix += 1
1343                    text2=wx.StaticText(self, -1, '+/-')
1344                    sizer.Add(text2,(iy, ix),(1,1),\
1345                                    wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
1346                    text2.Hide() 
1347                    ## txtcrtl to add error from fit
1348                    ix += 1
1349                    ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
1350                    sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1351                    ctl2.Hide()
1352                    if self.data.__class__.__name__ =="Data2D":
1353                        ctl1.Show(True)
1354                        #ctl1.Enable()
1355                    else:
1356                        ctl1.Hide()
1357                        #ctl1.Disable()
1358                    param_min, param_max= self.model.details[item][1:]
1359                    ix += 1
1360                    ctl3 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1361                                                   kill_focus_callback = self._onparamRangeEnter,
1362                                                   set_focus_callback  = self._onparamRangeEnter)
1363                    if param_min ==None:
1364                        ctl3.SetValue("")
1365                    else:
1366                        ctl3.SetValue(str(param_min))
1367                    sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1368                    ctl3.Hide()
1369                    if self.data.__class__.__name__ =="Data2D":
1370                        ctl3.Show(True)
1371                        #ctl3.Enable()
1372                    else:
1373                        ctl3.Hide()
1374                        #ctl3.Disable()
1375                    ix += 1
1376                    ctl4 = BasicPage.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
1377                                                   kill_focus_callback = self._onparamRangeEnter,
1378                                                   set_focus_callback  = self._onparamRangeEnter)
1379                    sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1380                    if param_max ==None:
1381                        ctl4.SetValue("")
1382                    else:
1383                        ctl4.SetValue(str(param_max))
1384                    ctl4.Hide()
1385                    if self.data.__class__.__name__ =="Data2D":
1386                        ctl4.Show(True)
1387                        #ctl4.Enable()
1388                    else:
1389                        ctl4.Hide()
1390                        #ctl4.Disable()
1391                    if self.engine_type !="park" or self.data.__class__.__name__ !="Data2D":                     
1392                        ctl3.Hide()
1393                        ctl4.Hide()
1394                    else:
1395                        ctl3.Show(True)
1396                        ctl4.Show(True)
1397                   
1398                    ix +=1
1399                    # Units
1400                    try:
1401                        units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
1402                    except:
1403                        units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
1404                    if self.data.__class__.__name__ =="Data2D":
1405                        units.Show(True)
1406                        #units.Enable()
1407                    else:
1408                        units.Hide()
1409                        #units.Disable()
1410                    sizer.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1411                       
1412                   
1413                    ##[cb state, name, value, "+/-", error of fit, min, max , units]
1414                    self.parameters.append([cb,item, ctl1,
1415                                            text2,ctl2, ctl3, ctl4,units])
1416                    self.orientation_params.append([cb,item, ctl1,
1417                                            text2,ctl2, ctl3, ctl4,units])
1418             
1419        iy+=1
1420        #sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1421       
1422        #Display units text on panel
1423        for item in keys:   
1424            if self.model.details[item][0]!='':
1425                self.text2_4.Show()
1426                break
1427            else:
1428                self.text2_4.Hide()
1429               
1430        self._copy_parameters_state(self.orientation_params,
1431                                     self.state.orientation_params)
1432        self._copy_parameters_state(self.orientation_params_disp,
1433                                     self.state.orientation_params_disp)
1434        self._copy_parameters_state(self.parameters, self.state.parameters)
1435        self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
1436        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
1437       
1438        boxsizer1.Add(sizer)
1439        self.sizer3.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
1440        self.sizer3.Layout()
1441        self.Layout()
1442        self.Refresh()
1443        self.SetScrollbars(20,20,25,65)
1444       
1445       
1446class HelpWindow(wx.Frame):
1447    def __init__(self, parent, id, title):
1448        wx.Frame.__init__(self, parent, id, title, size=(570, 400))
1449       
1450        from sans.models.CylinderModel import CylinderModel
1451        model = CylinderModel()
1452       
1453        from danse.common.plottools.plottables import Data1D
1454        data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
1455   
1456        from fitpanel import PageInfo
1457        myinfo = PageInfo(self,  model, data=data )
1458       
1459        ## add data
1460       
1461        from models import ModelList
1462        mylist= ModelList()
1463
1464        from sans.models.SphereModel import SphereModel
1465        from sans.models.SquareWellStructure import SquareWellStructure
1466        from sans.models.DebyeModel import DebyeModel
1467        from sans.models.LineModel import LineModel
1468        name= "shapes"
1469        list1= [SphereModel]
1470        mylist.set_list( name, list1)
1471       
1472        name= "Shape-independent"
1473        list1= [DebyeModel]
1474        mylist.set_list( name, list1)
1475       
1476        name= "Structure Factors"
1477        list1= [SquareWellStructure]
1478        mylist.set_list( name, list1)
1479       
1480        name= "Added models"
1481        list1= [LineModel]
1482        mylist.set_list( name, list1)
1483       
1484        myinfo.model_list_box = mylist.get_list()
1485       
1486        self.page = FitPage(self, myinfo) 
1487       
1488       
1489       
1490        self.Centre()
1491        self.Show(True)
1492
1493
1494   
1495if __name__=="__main__":
1496    app = wx.App()
1497    HelpWindow(None, -1, 'HelpWindow')
1498    app.MainLoop()
1499               
Note: See TracBrowser for help on using the repository browser.