source: sasview/sansview/perspectives/fitting/fitpage.py @ 852354c8

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 852354c8 was 9a22655, checked in by Gervaise Alina <gervyh@…>, 14 years ago

working on plugin update

  • Property mode set to 100644
File size: 114.3 KB
Line 
1
2
3import sys
4import wx
5import wx.lib.newevent
6import numpy
7import copy
8import math
9import time
10from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
11from DataLoader.data_info import Data1D
12from sans.guiframe.events import StatusEvent   
13from sans.guiframe.utils import format_number,check_float
14
15(Chi2UpdateEvent, EVT_CHI2_UPDATE)   = wx.lib.newevent.NewEvent()
16_BOX_WIDTH = 76
17_DATA_BOX_WIDTH = 300
18SMEAR_SIZE_L = 0.005
19SMEAR_SIZE_H = 0.006
20
21import basepage
22from basepage import BasicPage
23from basepage import PageInfoEvent
24from sans.models.qsmearing import smear_selection
25
26class FitPage(BasicPage):
27    """
28    FitPanel class contains fields allowing to display results when
29    fitting  a model and one data
30   
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, color='rand'):
36        """
37        Initialization of the Panel
38        """
39        BasicPage.__init__(self, parent, color=color)
40       
41        ## draw sizer
42        self._fill_datainfo_sizer()
43        # get smear info from data
44        self._get_smear_info()
45        self._fill_model_sizer( self.sizer1)
46        self._get_defult_custom_smear()
47        self._fill_range_sizer() 
48        self._set_smear(self.data)
49        self.Bind(EVT_CHI2_UPDATE, self.on_complete_chisqr)
50        self._set_bookmark_flag(False)
51        self._set_save_flag(False)
52
53    def _on_fit_complete(self):
54        """
55        When fit is complete ,reset the fit button label.
56        """
57        self.btFit.SetLabel("Fit")
58        self.bind_fit_button()
59       
60    def _is_2D(self):
61        """
62        Check if data_name is Data2D
63       
64        :return: True or False
65       
66        """
67        if self.data.__class__.__name__ =="Data2D":
68            return True
69        return False
70           
71    def _on_engine_change(self, name):
72        """
73        get the current name of the fit engine type
74         and update the panel accordingly
75        """
76       
77        self.engine_type = str(name)
78        self.state.engine_type = self.engine_type
79        if len(self.parameters) == 0:
80            self.Layout()
81            return
82        for item in self.parameters:
83            if self.engine_type =="scipy" :
84                item[5].SetValue("")
85                item[5].Hide()
86                item[6].SetValue("")
87                item[6].Hide()
88                self.text2_min.Hide()
89                self.text2_max.Hide()
90
91            else:
92                item[5].Show(True)
93                item[6].Show(True)
94                self.text2_min.Show(True)
95                self.text2_max.Show(True)
96        for item in self.fittable_param:
97            if item[5]!=None and item[6]!=None and not \
98                item in self.orientation_params_disp:
99                if self.engine_type =="scipy" and \
100                    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 self.engine_type =="scipy" or \
120                self.data.__class__.__name__ !="Data2D":
121                    item[5].SetValue("")
122                    item[5].Hide()
123                    item[6].SetValue("")
124                    item[6].Hide()
125                else:
126                    item[5].Show(True)
127                    item[6].Show(True)
128                   
129        for item in self.orientation_params_disp:         
130            if item[5]!=None and item[6]!=None:
131                if self.engine_type =="scipy" or \
132                    self.data.__class__.__name__ !="Data2D":
133                    item[5].SetValue("")
134                    item[5].Hide()
135                    item[6].SetValue("")
136                    item[6].Hide()
137                else:
138                    item[5].Show(True)
139                    item[6].Show(True)
140        self.Layout()
141        self.Refresh()
142       
143 
144       
145    def _fill_range_sizer(self):
146        """
147        Fill the sizer containing the plotting range
148        add  access to npts
149        """
150        is_2Ddata = False
151       
152        # Check if data is 2D
153        if self.data.__class__.__name__ == 'Data2D':
154            is_2Ddata = True
155           
156        title = "Fitting"
157        #smear messages & titles
158        smear_message_none  =  "No smearing is selected..."
159        smear_message_dqdata  =  "The dQ data is being used for smearing..."
160        smear_message_2d  =  "Higher accuracy is very time-expensive. Use it with care..."
161        smear_message_new_ssmear  =  "Please enter only the value of interest to customize smearing..."
162        smear_message_new_psmear  =  "Please enter both; the dQ will be generated by interpolation..."
163        smear_message_2d_x_title = "<dQx>[1/A]:"
164        smear_message_2d_y_title = "<dQy>[1/A]:"
165        smear_message_pinhole_min_title = "dQ_low[1/A]:"
166        smear_message_pinhole_max_title = "dQ_high[1/A]:"
167        smear_message_slit_height_title = "Slit height[1/A]:"
168        smear_message_slit_width_title = "Slit width[1/A]:"
169       
170        self._get_smear_info()
171       
172        #Sizers
173        box_description_range = wx.StaticBox(self, -1,str(title))
174        boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)     
175        self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
176        sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
177        self.sizer_new_smear= wx.BoxSizer(wx.HORIZONTAL)
178        self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
179        sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
180        smear_set_box= wx.StaticBox(self, -1,'Set Instrumental Smearing')
181        sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
182        sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH,85))
183        sizer_fit = wx.GridSizer(2, 4,2,6)
184       
185        # combobox for smear2d accuracy selection
186        self.smear_accuracy = wx.ComboBox(self, -1,size=(50,-1),style=wx.CB_READONLY)
187        self._set_accuracy_list()
188        self.smear_accuracy.SetValue(self.smear2d_accuracy)
189        self.smear_accuracy.SetSelection(0)
190        self.smear_accuracy.SetToolTipString("'Higher' uses more Gaussian points for smearing computation.")
191                   
192        wx.EVT_COMBOBOX(self.smear_accuracy,-1, self._on_select_accuracy)
193
194        #Fit button
195        self.btFit = wx.Button(self,wx.NewId(),'Fit', size=(88,25))
196        self.default_bt_colour = self.btFit.GetBackgroundColour()
197        self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
198        self.btFit.SetToolTipString("Start fitting.")
199       
200        #textcntrl for custom resolution
201        self.smear_pinhole_max = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
202                                            text_enter_callback = self.onPinholeSmear)
203        self.smear_pinhole_min = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
204                                            text_enter_callback = self.onPinholeSmear)
205        self.smear_slit_height= self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
206                                            text_enter_callback = self.onSlitSmear)
207        self.smear_slit_width = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
208                                            text_enter_callback = self.onSlitSmear)
209
210        ## smear
211        self.smear_data_left= BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
212        self.smear_data_left.SetValue(str(self.dq_l))
213        self.smear_data_right = BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
214        self.smear_data_right.SetValue(str(self.dq_r))
215
216        #set default values for smear
217        self.smear_pinhole_max.SetValue(str(self.dx_max))
218        self.smear_pinhole_min.SetValue(str(self.dx_min))
219        self.smear_slit_height.SetValue(str(self.dxl))
220        self.smear_slit_width.SetValue(str(self.dxw))
221
222        #Filling the sizer containing instruments smearing info.
223        self.disable_smearer = wx.RadioButton(self, -1, 'None', style=wx.RB_GROUP)
224        self.enable_smearer = wx.RadioButton(self, -1, 'Use dQ Data')
225        #self.enable_smearer.SetToolTipString("Click to use the loaded dQ data for smearing.")
226        self.pinhole_smearer = wx.RadioButton(self, -1, 'Custom Pinhole Smear')
227        #self.pinhole_smearer.SetToolTipString("Click to input custom resolution for pinhole smearing.")
228        self.slit_smearer = wx.RadioButton(self, -1, 'Custom Slit Smear')
229        #self.slit_smearer.SetToolTipString("Click to input custom resolution for slit smearing.")
230        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
231        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
232        self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear, id=self.pinhole_smearer.GetId())
233        self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear, id=self.slit_smearer.GetId())
234        self.disable_smearer.SetValue(True)
235       
236        # add 4 types of smearing to the sizer
237        sizer_smearer.Add( self.disable_smearer,0, wx.LEFT, 10)
238        sizer_smearer.Add((10,10))
239        sizer_smearer.Add( self.enable_smearer)
240        sizer_smearer.Add((10,10))
241        sizer_smearer.Add( self.pinhole_smearer ) 
242        sizer_smearer.Add((10,10))
243        sizer_smearer.Add( self.slit_smearer ) 
244        sizer_smearer.Add((10,10))       
245       
246        # StaticText for chi2, N(for fitting), Npts
247        self.tcChi    =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
248        self.tcChi.SetToolTipString("Chi2/Npts")
249        self.Npts_fit    =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
250        self.Npts_fit.SetToolTipString(" Npts : number of points selected for fitting")
251        self.Npts_total  =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
252        self.Npts_total.SetToolTipString(" Total Npts : total number of data points")
253        box_description_1= wx.StaticText(self, -1,'    Chi2/Npts')
254        box_description_2= wx.StaticText(self, -1,'Fitted Npts')
255        box_description_3= wx.StaticText(self, -1,'Data Npts')
256        box_description_4= wx.StaticText(self, -1,' ')
257       
258       
259        sizer_fit.Add(box_description_1,0,0)
260        sizer_fit.Add(box_description_2,0,0)
261        sizer_fit.Add(box_description_3,0,0)       
262        sizer_fit.Add(box_description_4,0,0)
263        sizer_fit.Add(self.tcChi,0,0)
264        sizer_fit.Add(self.Npts_fit ,0,0)
265        sizer_fit.Add(self.Npts_total,0,0)
266        sizer_fit.Add(self.btFit,0,0) 
267
268        # StaticText for smear
269        #self.tcChi    =  wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
270        self.smear_description_none    =  wx.StaticText(self, -1, smear_message_none , style=wx.ALIGN_LEFT)
271        self.smear_description_dqdata    =  wx.StaticText(self, -1, smear_message_dqdata , style=wx.ALIGN_LEFT)
272        self.smear_description_type    =  wx.StaticText(self, -1, "Type:" , style=wx.ALIGN_LEFT)
273        self.smear_description_accuracy_type    =  wx.StaticText(self, -1, "Accuracy:" , style=wx.ALIGN_LEFT)
274        self.smear_description_smear_type    =  BGTextCtrl(self, -1, size=(57,20), style=0)
275        self.smear_description_smear_type.SetValue(str(self.dq_l))
276        self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
277        self.smear_description_2d     =  wx.StaticText(self, -1, smear_message_2d  , style=wx.ALIGN_LEFT)
278        self.smear_message_new_s = wx.StaticText(self, -1, smear_message_new_ssmear  , style=wx.ALIGN_LEFT)
279        self.smear_message_new_p = wx.StaticText(self, -1, smear_message_new_psmear  , style=wx.ALIGN_LEFT)
280        self.smear_description_2d_x     =  wx.StaticText(self, -1, smear_message_2d_x_title  , style=wx.ALIGN_LEFT)
281        self.smear_description_2d_y     =  wx.StaticText(self, -1, smear_message_2d_y_title  , style=wx.ALIGN_LEFT)
282        self.smear_description_pin_min     =  wx.StaticText(self, -1, smear_message_pinhole_min_title  , style=wx.ALIGN_LEFT)
283        self.smear_description_pin_max     =  wx.StaticText(self, -1, smear_message_pinhole_max_title  , style=wx.ALIGN_LEFT)
284        self.smear_description_slit_height    =  wx.StaticText(self, -1, smear_message_slit_height_title   , style=wx.ALIGN_LEFT)
285        self.smear_description_slit_width    =  wx.StaticText(self, -1, smear_message_slit_width_title   , style=wx.ALIGN_LEFT)
286       
287        #arrange sizers
288        #boxsizer1.Add( self.tcChi ) 
289        self.sizer_set_smearer.Add(sizer_smearer )
290        self.sizer_set_smearer.Add((10,10))
291        self.sizer_set_smearer.Add( self.smear_description_none,0, wx.CENTER, 10 ) 
292        self.sizer_set_smearer.Add( self.smear_description_dqdata,0, wx.CENTER, 10 )
293        self.sizer_set_smearer.Add( self.smear_description_2d,0, wx.CENTER, 10 )
294        self.sizer_new_smear.Add( self.smear_description_type,0, wx.CENTER, 10 )
295        self.sizer_new_smear.Add( self.smear_description_accuracy_type,0, wx.CENTER, 10 )
296        self.sizer_new_smear.Add( self.smear_accuracy )
297        self.sizer_new_smear.Add( self.smear_description_smear_type,0, wx.CENTER, 10 )
298        self.sizer_new_smear.Add((15,-1))
299        self.sizer_new_smear.Add( self.smear_description_2d_x,0, wx.CENTER, 10 )
300        self.sizer_new_smear.Add( self.smear_description_pin_min,0, wx.CENTER, 10 )
301        self.sizer_new_smear.Add( self.smear_description_slit_height,0, wx.CENTER, 10 )
302
303        self.sizer_new_smear.Add( self.smear_pinhole_min,0, wx.CENTER, 10 )
304        self.sizer_new_smear.Add( self.smear_slit_height,0, wx.CENTER, 10 )
305        self.sizer_new_smear.Add( self.smear_data_left,0, wx.CENTER, 10 )
306        self.sizer_new_smear.Add((20,-1))
307        self.sizer_new_smear.Add( self.smear_description_2d_y,0, wx.CENTER, 10 )
308        self.sizer_new_smear.Add( self.smear_description_pin_max,0, wx.CENTER, 10 )
309        self.sizer_new_smear.Add( self.smear_description_slit_width,0, wx.CENTER, 10 )
310
311        self.sizer_new_smear.Add( self.smear_pinhole_max,0, wx.CENTER, 10 )
312        self.sizer_new_smear.Add( self.smear_slit_width,0, wx.CENTER, 10 )
313        self.sizer_new_smear.Add( self.smear_data_right,0, wx.CENTER, 10 )
314           
315        self.sizer_set_smearer.Add( self.smear_message_new_s,0, wx.CENTER, 10)
316        self.sizer_set_smearer.Add( self.smear_message_new_p,0, wx.CENTER, 10)
317        self.sizer_set_smearer.Add((5,2))
318        self.sizer_set_smearer.Add( self.sizer_new_smear,0, wx.CENTER, 10 )
319       
320        # add all to chi2 sizer
321        sizer_smearer_box.Add(self.sizer_set_smearer)       
322        sizer_chi2.Add(sizer_smearer_box)
323        sizer_chi2.Add((-1,5))
324
325        # hide all smear messages and textctrl
326        self._hide_all_smear_info()
327       
328        # get smear_selection
329        self.current_smearer= smear_selection( self.data, self.model )
330
331        # Show only the relevant smear messages, etc
332        if self.current_smearer == None:
333            if not is_2Ddata:
334                self.smear_description_none.Show(True)
335                self.enable_smearer.Disable() 
336            else:
337                self.smear_description_none.Show(True)
338                #self.smear_description_2d.Show(True)
339                #self.pinhole_smearer.Disable()
340                self.slit_smearer.Disable()   
341                #self.enable_smearer.Disable()
342        else: self._show_smear_sizer()
343        boxsizer_range.Add(self.sizer_set_masking)
344         #2D data? default
345        is_2Ddata = False
346       
347        #check if it is 2D data
348        if self.data.__class__.__name__ == 'Data2D':
349            is_2Ddata = True
350           
351        self.sizer5.Clear(True)
352     
353        self.qmin  = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
354                                          style=wx.TE_PROCESS_ENTER,
355                                    text_enter_callback = self._onQrangeEnter)
356        self.qmin.SetValue(str(self.qmin_x))
357        self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
358     
359        self.qmax  = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
360                                          style=wx.TE_PROCESS_ENTER,
361                                        text_enter_callback=self._onQrangeEnter)
362        self.qmax.SetValue(str(self.qmax_x))
363        self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
364       
365        self.theory_npts_tcrtl  = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20), 
366                        style=wx.TE_PROCESS_ENTER, 
367                        text_enter_callback=self._onQrangeEnter)
368        self.theory_npts_tcrtl.SetValue(format_number(self.npts_x))
369        self.theory_npts_tcrtl.SetToolTipString("Number of point to plot.")
370        id = wx.NewId()
371        self.reset_qrange =wx.Button(self,id,'Reset',size=(77,20))
372     
373        self.reset_qrange.Bind(wx.EVT_BUTTON, self.on_reset_clicked,id=id)
374        self.reset_qrange.SetToolTipString("Reset Q range to the default values")
375     
376        sizer_horizontal=wx.BoxSizer(wx.HORIZONTAL)
377        sizer= wx.GridSizer(2, 5,0, 0)
378
379        self.btEditMask = wx.Button(self,wx.NewId(),'Editor', size=(88,23))
380        self.btEditMask.Bind(wx.EVT_BUTTON, self._onMask,id= self.btEditMask.GetId())
381        self.btEditMask.SetToolTipString("Edit Mask.")
382        self.EditMask_title = wx.StaticText(self, -1, ' Masking(2D)')
383
384        sizer.Add(wx.StaticText(self, -1, 'Q range'))     
385        sizer.Add(wx.StaticText(self, -1, ' Min[1/A]'))
386        sizer.Add(wx.StaticText(self, -1, ' Max[1/A]'))
387        sizer.Add(wx.StaticText(self, -1, ' Theory Npts'))
388        sizer.Add(self.EditMask_title)
389        sizer.Add(self.reset_qrange)   
390        sizer.Add(self.qmin)
391        sizer.Add(self.qmax)
392        sizer.Add(self.theory_npts_tcrtl)
393        sizer.Add(self.btEditMask)
394        boxsizer_range.Add(sizer_chi2) 
395        boxsizer_range.Add((10,10))
396        boxsizer_range.Add(sizer)
397       
398        boxsizer_range.Add((10,15))
399        boxsizer_range.Add(sizer_fit)
400        if is_2Ddata:
401            self.btEditMask.Enable() 
402            self.EditMask_title.Enable() 
403        else:
404            self.btEditMask.Disable() 
405            self.EditMask_title.Disable()
406        ## save state
407        self.save_current_state()
408        self.sizer5.Add(boxsizer_range,0, wx.EXPAND | wx.ALL, 10)
409        self.sizer5.Layout()
410
411    def _fill_datainfo_sizer(self):
412        """
413        fill sizer 0 with data info
414        """
415        ## no loaded data , don't fill the sizer
416        if self.data is None:
417            data_min = ""
418            data_max = ""
419            data_name = ""
420        else:
421            data_name = self.data.name
422            #set maximum range for x in linear scale
423            if not hasattr(self.data,"data"): #Display only for 1D data fit
424                # Minimum value of data   
425                data_min = min(self.data.x)
426                # Maximum value of data 
427                data_max = max(self.data.x)
428            else:
429                ## Minimum value of data
430                data_min = 0
431                x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax)) 
432                y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
433                ## Maximum value of data 
434                data_max = math.sqrt(x*x + y*y)
435            ## set q range to plot
436            self.qmin_x = data_min
437            self.qmax_x = data_max
438        box_description= wx.StaticBox(self, -1, 'Data')
439        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
440        #----------------------------------------------------------
441        sizer_data = wx.BoxSizer(wx.HORIZONTAL)
442        #Filling the sizer containing data related fields
443        #self.dataSource = wx.StaticText(self, -1,str(self.data.name))
444        self.dataSource = BGTextCtrl(self, -1)
445        self.dataSource.SetValue(str(data_name))
446        #self.dataSource.SetEditable(False)
447        self.dataSource.SetMinSize((_DATA_BOX_WIDTH, -1))
448        sizer_data.Add(wx.StaticText(self, -1, 'Source Name : '))
449        sizer_data.Add(self.dataSource )
450        sizer_data.Add( (0,5) )
451       
452        #---------sizer 2 draw--------------------------------
453        text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
454                                 style=wx.ALIGN_LEFT)
455        sizer_range = wx.BoxSizer(wx.HORIZONTAL)
456        sizer_range.Add( text4_3 ,0, wx.RIGHT, 10)
457        self.minimum_q = BGTextCtrl(self, -1, size=(_BOX_WIDTH,20))
458        self.minimum_q.SetValue(str(data_min))
459        self.maximum_q = BGTextCtrl(self, -1,size=(_BOX_WIDTH,20))
460        self.maximum_q.SetValue(str(data_max))
461        sizer_range.Add(wx.StaticText(self, -1, "Min: "),0, wx.LEFT, 10)
462        sizer_range.Add(self.minimum_q,0, wx.LEFT, 10)
463        sizer_range.Add(wx.StaticText(self, -1, "Max: "),0, wx.LEFT, 10)
464        sizer_range.Add(self.maximum_q,0, wx.LEFT, 10)
465        boxsizer1.Add(sizer_data,0, wx.ALL, 10)
466        boxsizer1.Add(sizer_range, 0 , wx.LEFT, 10)
467        #------------------------------------------------------------
468        self.sizer0.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
469        self.sizer0.Layout()
470       
471    def _fill_model_sizer(self, sizer):
472        """
473        fill sizer containing model info
474        """
475        ##Add model function Details button in fitpanel.
476        ##The following 3 lines are for Mac. Let JHC know before modifying...
477        title = "Model"
478        box_description= wx.StaticBox(self, -1,str(title))
479        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
480         
481        id = wx.NewId()
482        self.model_help =wx.Button(self,id,'Details', size=(80,23))
483        self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,id=id)
484        self.model_help.SetToolTipString("Model Function Help")
485        id = wx.NewId()
486        self.model_view = wx.Button(self, id,"Switch to 2D", size=(80, 23))
487        self.model_view.Bind(wx.EVT_BUTTON, self._onModel2D, id=id)
488        hint = "toggle view of model from 1D to 2D  or 2D from 1D"
489        self.model_view.SetToolTipString(hint)
490     
491        self.shape_rbutton = wx.RadioButton(self, -1, 'Shapes', style=wx.RB_GROUP)
492        self.shape_indep_rbutton = wx.RadioButton(self, -1, "Shape-Independent")
493        self.struct_rbutton = wx.RadioButton(self, -1, "Structure Factor ")
494        self.plugin_rbutton = wx.RadioButton(self, -1, "Customized Models")
495               
496        self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
497                            id= self.shape_rbutton.GetId()) 
498        self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
499                            id= self.shape_indep_rbutton.GetId()) 
500        self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
501                            id= self.struct_rbutton.GetId()) 
502        self.Bind(wx.EVT_RADIOBUTTON, self._show_combox,
503                            id= self.plugin_rbutton.GetId()) 
504        #MAC needs SetValue
505        self.shape_rbutton.SetValue(True)
506     
507        sizer_radiobutton = wx.GridSizer(2, 3, 5, 5)
508        sizer_radiobutton.Add(self.shape_rbutton)
509        sizer_radiobutton.Add(self.shape_indep_rbutton)
510        #sizer_radiobutton.Add((5, 5))
511        sizer_radiobutton.Add(self.model_view,1, wx.LEFT, 20)
512        sizer_radiobutton.Add(self.plugin_rbutton)
513        sizer_radiobutton.Add(self.struct_rbutton)
514        #sizer_radiobutton.Add((5, 5))
515        sizer_radiobutton.Add(self.model_help,1, wx.LEFT, 20)
516       
517        sizer_selection = wx.BoxSizer(wx.HORIZONTAL)
518        mutifactor_selection = wx.BoxSizer(wx.HORIZONTAL)
519       
520        self.text1 = wx.StaticText(self,-1,"" )
521        self.text2 = wx.StaticText(self,-1,"P(Q)*S(Q)" )
522        self.mutifactor_text = wx.StaticText( self,-1,"No. of Shells: ")
523        self.mutifactor_text1 = wx.StaticText( self,-1,"" )
524        self.show_sld_button = wx.Button( self,-1,"Show SLD Profile" )
525        self.show_sld_button.Bind(wx.EVT_BUTTON,self._on_show_sld)
526
527        self.formfactorbox = wx.ComboBox(self, -1,style=wx.CB_READONLY)
528        if self.model!= None:
529            self.formfactorbox.SetValue(self.model.name)
530        self.structurebox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
531        self.multifactorbox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
532        self.initialize_combox()
533        wx.EVT_COMBOBOX(self.formfactorbox, -1, self._on_select_model)
534        wx.EVT_COMBOBOX(self.structurebox, -1, self._on_select_model)
535        wx.EVT_COMBOBOX(self.multifactorbox, -1, self._on_select_model)
536        ## check model type to show sizer
537        if self.model !=None:
538            self._set_model_sizer_selection(self.model)
539       
540        sizer_selection.Add(self.text1)
541        sizer_selection.Add((5, 5))
542        sizer_selection.Add(self.formfactorbox)
543        sizer_selection.Add((5, 5))
544        sizer_selection.Add(self.text2)
545        sizer_selection.Add((5, 5))
546        sizer_selection.Add(self.structurebox)
547       
548        mutifactor_selection.Add((10,5))
549        mutifactor_selection.Add(self.mutifactor_text)
550        mutifactor_selection.Add(self.multifactorbox)
551        mutifactor_selection.Add((5, 5))
552        mutifactor_selection.Add(self.mutifactor_text1)
553        mutifactor_selection.Add((10, 5))
554        mutifactor_selection.Add(self.show_sld_button)
555
556       
557        boxsizer1.Add(sizer_radiobutton)
558        boxsizer1.Add((10, 10))
559        boxsizer1.Add(sizer_selection)
560        boxsizer1.Add((10, 10))
561        boxsizer1.Add(mutifactor_selection)
562       
563        self._set_multfactor_combobox()
564        self.multifactorbox.SetSelection(1)
565        self.show_sld_button.Hide()
566        sizer.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
567        sizer.Layout()
568       
569    def _set_sizer_dispersion(self, dispersity):
570        """
571        draw sizer with gaussian dispersity parameters
572        """
573        self.fittable_param=[]
574        self.fixed_param=[]
575        self.orientation_params_disp=[]
576
577        self.sizer4_4.Clear(True)
578        if self.model==None:
579            ##no model is selected
580            return
581        if not self.enable_disp.GetValue():
582            ## the user didn't select dispersity display
583            return 
584           
585        self._reset_dispersity()
586       
587        # Create the dispersion objects
588        for item in self.model.dispersion.keys():
589            #disp_model =  GaussianDispersion()
590            disp_model = dispersity()
591            self._disp_obj_dict[item] = disp_model
592            self.model.set_dispersion(item, disp_model)
593            self.state._disp_obj_dict[item]= disp_model
594
595
596        ix=0
597        iy=1
598        disp = wx.StaticText(self, -1, ' ')
599        self.sizer4_4.Add(disp,( iy, ix),(1,1), 
600                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
601        ix += 1 
602        values = wx.StaticText(self, -1, 'PD[ratio] ')
603        polytext = "Polydispersity (= STD/mean); "
604        polytext +=  "the standard deviation over the mean value."
605        values.SetToolTipString(polytext)
606
607        self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 
608                          0)
609        ix +=2 
610        self.text_disp_1 = wx.StaticText(self, -1, '')
611        self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
612                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
613        self.text_disp_1.Hide()
614       
615       
616        ix +=1 
617        self.text_disp_min = wx.StaticText(self, -1, 'Min')
618        self.sizer4_4.Add(self.text_disp_min,(iy, ix),(1,1),\
619                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
620        self.text_disp_min.Hide()
621        ix +=1 
622        self.text_disp_max = wx.StaticText(self, -1, 'Max')
623        self.sizer4_4.Add(self.text_disp_max,(iy, ix),(1,1),\
624                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
625        self.text_disp_max.Hide()
626                       
627       
628        ix += 1 
629        npts = wx.StaticText(self, -1, 'Npts')
630        npts.SetToolTipString("Number of sampling points for the numerical\n\
631        integration over the distribution function.")
632        self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
633        ix += 1 
634        nsigmas = wx.StaticText(self, -1, 'Nsigmas')
635        nsigmas.SetToolTipString("   Number of sigmas between which the range\n\
636         of the distribution function will be used for weighting. \n\
637        The value '3' covers 99.5% for Gaussian distribution \n\
638        function. Note: Not recommended to change this value.")
639        self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE,
640                           0)
641       
642        if self.engine_type=="park":
643            self.text_disp_max.Show(True)
644            self.text_disp_min.Show(True)
645
646        for item in self.model.dispersion.keys():
647            if not item in self.model.orientation_params:
648                if not self.disp_cb_dict.has_key(item):
649                    self.disp_cb_dict[item]= None
650                name0="Distribution of " + item
651                name1=item+".width"
652                name2=item+".npts"
653                name3=item+".nsigmas"
654                if not self.model.details.has_key(name1):
655                    self.model.details [name1] = ["",None,None] 
656
657                iy += 1
658                for p in self.model.dispersion[item].keys(): 
659       
660                    if p=="width":
661                        ix = 0
662                        cb = wx.CheckBox(self, -1, name0, (10, 10))
663                        cb.SetToolTipString("Check for fitting")
664                        wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
665                        self.sizer4_4.Add( cb,( iy, ix),(1,1), 
666                                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 
667                                           15)
668                        ix = 1
669                        value= self.model.getParam(name1)
670                        ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20)
671                                                  ,style=wx.TE_PROCESS_ENTER)
672                        ctl1.SetLabel('PD[ratio]')
673                        poly_text = "Polydispersity (STD/mean) of %s\n" % item
674                        poly_text += "STD: the standard deviation"
675                        poly_text += " from the mean value."
676                        ctl1.SetToolTipString(poly_text)
677                        ctl1.SetValue(str (format_number(value)))
678                        self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
679                        ## text to show error sign
680                        ix = 2
681                        text2=wx.StaticText(self, -1, '+/-')
682                        self.sizer4_4.Add(text2,(iy, ix),(1,1),
683                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
684                        text2.Hide() 
685
686                        ix = 3
687                        ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), 
688                                           style=0)
689                 
690                        self.sizer4_4.Add(ctl2, (iy,ix),(1,1), 
691                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
692                     
693                        ctl2.Hide()
694
695                        ix = 4
696                        ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,
697                                            20), style=wx.TE_PROCESS_ENTER,
698                                text_enter_callback = self._onparamRangeEnter)
699                       
700                        self.sizer4_4.Add(ctl3, (iy,ix),(1,1), 
701                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
702                        ctl3.Hide()
703               
704                        ix = 5
705                        ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,
706                                            20), style=wx.TE_PROCESS_ENTER,
707                            text_enter_callback = self._onparamRangeEnter)
708                       
709                        self.sizer4_4.Add(ctl4, (iy,ix),(1,1), 
710                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
711
712                        ctl4.Hide()
713                       
714                        if self.engine_type=="park":
715                            ctl3.Show(True)
716                            ctl4.Show(True)
717                                         
718                        self.fittable_param.append([cb,name1,ctl1,text2,
719                                                    ctl2, ctl3, ctl4,None])                   
720                   
721                    elif p=="npts":
722                            ix = 6
723                            value= self.model.getParam(name2)
724                            Tctl = self.ModelTextCtrl(self, -1, 
725                                                      size=(_BOX_WIDTH/2,20),
726                                                style=wx.TE_PROCESS_ENTER)
727                           
728                            Tctl.SetValue(str (format_number(value)))
729                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
730                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
731                            self.fixed_param.append([None,name2, Tctl,None,None,
732                                                      None, None,None])
733                    elif p=="nsigmas":
734                            ix = 7
735                            value= self.model.getParam(name3)
736                            Tct2 = self.ModelTextCtrl(self, -1, 
737                                                      size=(_BOX_WIDTH/2,20),
738                                                style=wx.TE_PROCESS_ENTER)
739                           
740                            Tct2.SetValue(str (format_number(value)))
741                            self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
742                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
743                            ix +=1
744                            self.sizer4_4.Add((20,20), (iy,ix),(1,1),
745                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
746                           
747                            self.fixed_param.append([None,name3, Tct2
748                                                     ,None,None,None, 
749                                                     None,None])
750                           
751        ix =0
752        iy +=1 
753        self.sizer4_4.Add((20,20),(iy,ix),(1,1), 
754                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
755        first_orient  = True
756        for item in self.model.dispersion.keys():
757            if  item in self.model.orientation_params:
758                if not self.disp_cb_dict.has_key(item):
759                    self.disp_cb_dict[item]= None
760                name0="Distribution of " + item
761                name1=item+".width"
762                name2=item+".npts"
763                name3=item+".nsigmas"
764               
765                if not self.model.details.has_key(name1):
766                    self.model.details [name1] = ["",None,None]                 
767 
768
769                iy += 1
770                for p in self.model.dispersion[item].keys(): 
771       
772                    if p=="width":
773                        ix = 0
774                        cb = wx.CheckBox(self, -1, name0, (10, 10))
775                        cb.SetToolTipString("Check for fitting")
776                        wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
777                        self.sizer4_4.Add( cb,( iy, ix),(1,1), 
778                                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 
779                                           15)
780                        if self.data.__class__.__name__ =="Data2D":
781                            cb.Show(True)
782                        elif cb.IsShown():
783                            cb.Hide()
784                        ix = 1
785                        value= self.model.getParam(name1)
786                        ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,
787                                        20),style=wx.TE_PROCESS_ENTER)
788                        poly_tip = "Absolute Sigma for %s." % item
789                        ctl1.SetToolTipString(poly_tip)
790                        ctl1.SetValue(str (format_number(value)))
791                        if self.data.__class__.__name__ =="Data2D":
792                            if first_orient:
793                                values.SetLabel('PD[ratio], Sig[deg]')
794                                poly_text = "PD(polydispersity for lengths):\n"
795                                poly_text +=  "It should be a value between"
796                                poly_text +=  "0 and 1\n"
797                                poly_text += "Sigma for angles: \n"
798                                poly_text += "It is the STD (ratio*mean)"
799                                poly_text += " of the distribution.\n "
800                           
801                                values.SetToolTipString(poly_text)
802                                first_orient = False
803                            ctl1.Show(True)
804                        elif ctl1.IsShown():
805                            ctl1.Hide()
806                       
807                        self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
808                        ## text to show error sign
809                        ix = 2
810                        text2=wx.StaticText(self, -1, '+/-')
811                        self.sizer4_4.Add(text2,(iy, ix),(1,1),
812                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
813                        text2.Hide() 
814
815                        ix = 3
816                        ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), 
817                                           style=0)
818                   
819                        self.sizer4_4.Add(ctl2, (iy,ix),(1,1), 
820                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
821                        ctl2.Hide()
822                           
823                        ix = 4
824                        ctl3 = self.ModelTextCtrl(self, -1, 
825                                                  size=(_BOX_WIDTH/2,20), 
826                                                  style=wx.TE_PROCESS_ENTER,
827                                text_enter_callback = self._onparamRangeEnter)
828
829                        self.sizer4_4.Add(ctl3, (iy,ix),(1,1), 
830                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
831
832                        ctl3.Hide()
833               
834                        ix = 5
835                        ctl4 = self.ModelTextCtrl(self, -1, 
836                            size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
837                                text_enter_callback = self._onparamRangeEnter)
838                        self.sizer4_4.Add(ctl4, (iy,ix),(1,1), 
839                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
840                        ctl4.Hide()
841                        #if self.data.__class__.__name__ =="Data2D":
842                            #ctl4.Enable(True)
843                        #elif ctl4.Shown():
844                            #ctl4.Hide()
845                       
846                        if self.engine_type=="park" and \
847                                    self.data.__class__.__name__ =="Data2D":
848                            ctl3.Show(True)
849                            ctl4.Show(True) 
850                           
851                           
852                           
853                           
854                        self.fittable_param.append([cb,name1,ctl1,text2,
855                                                    ctl2, ctl3, ctl4,None])
856                        self.orientation_params_disp.append([cb,name1,ctl1,
857                                                  text2, ctl2, ctl3, ctl4,None])
858                    elif p=="npts":
859                            ix = 6
860                            value= self.model.getParam(name2)
861                            Tctl = self.ModelTextCtrl(self, -1, 
862                                                      size=(_BOX_WIDTH/2,20),
863                                                style=wx.TE_PROCESS_ENTER)
864                           
865                            Tctl.SetValue(str (format_number(value)))
866                            if self.data.__class__.__name__ =="Data2D":
867                                Tctl.Show(True)
868                            else:
869                                Tctl.Hide()
870                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
871                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
872                            self.fixed_param.append([None,name2, Tctl,None,None,
873                                                      None, None,None])
874                            self.orientation_params_disp.append([None,name2, 
875                                                    Tctl,None,None,
876                                                      None, None,None])
877                    elif p=="nsigmas":
878                            ix = 7
879                            value= self.model.getParam(name3)
880                            Tct2 = self.ModelTextCtrl(self, -1, 
881                                                      size=(_BOX_WIDTH/2,20),
882                                                style=wx.TE_PROCESS_ENTER)
883                           
884                            Tct2.SetValue(str (format_number(value)))
885                            if self.data.__class__.__name__ =="Data2D":
886                                Tct2.Show(True)
887                            else:
888                                Tct2.Hide()
889                            self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
890                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
891                            ix +=1
892
893                            self.fixed_param.append([None,name3, Tct2
894                                                 ,None,None, None, None,None])   
895                                                       
896                            self.orientation_params_disp.append([None,name3, 
897                                            Tct2 ,None,None, None, None,None])
898       
899        self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict) 
900         
901        self.state.model = self.model.clone() 
902         ## save state into
903        self.state.cb1 = self.cb1.GetValue()
904        self._copy_parameters_state(self.parameters, self.state.parameters)
905        self._copy_parameters_state(self.orientation_params_disp,
906                                     self.state.orientation_params_disp)
907        self._copy_parameters_state(self.fittable_param, 
908                                    self.state.fittable_param)
909        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
910
911
912        wx.PostEvent(self.parent, StatusEvent(status=\
913                        " Selected Distribution: Gaussian"))   
914        #Fill the list of fittable parameters
915        self.select_all_param(event=None)
916       
917        self.Layout()
918
919    def _onFit(self, event):     
920        """
921        Allow to fit
922        """
923        if self.data is None:
924            msg = "Please select a Data first..."
925            wx.MessageBox(msg, 'Info')
926            wx.PostEvent(self._manager.parent, StatusEvent(status=\
927                            "Fit: %s" % msg))
928            return
929        if self.model is None:
930            msg = "Please select a Model first..."
931            wx.MessageBox(msg, 'Info')
932            wx.PostEvent(self._manager.parent, StatusEvent(status=\
933                            "Fit: %s" % msg))
934            return
935
936        flag = self._update_paramv_on_fit() 
937               
938        if not flag:
939            msg= "Fitting range or parameters are invalid"
940            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
941            return 
942       
943        if len(self.param_toFit) <= 0:
944            msg= "Select at least one parameter to fit"
945            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
946            return 
947     
948        self.select_param(event =None)
949       
950        #Clear errors if exist from previous fitting
951        #self._clear_Err_on_Fit()
952
953        # Remove or do not allow fitting on the Q=0 point, especially
954        # when y(q=0)=None at x[0].         
955        self.qmin_x = float(self.qmin.GetValue())
956        self.qmax_x = float( self.qmax.GetValue())
957        self._manager._reset_schedule_problem(value=0)
958        self._manager.schedule_for_fit(uid=self.uid,value=1, fitproblem =None) 
959        self._manager.set_fit_range(uid=self.uid,qmin= self.qmin_x, 
960                                   qmax= self.qmax_x)
961       
962        #single fit
963        self._manager.onFit()
964        self.btFit.SetLabel("Stop")
965        self.bind_fit_button()
966           
967    def bind_fit_button(self):
968        """
969        bind the fit button to either fit handler or stop fit handler
970        """
971        self.btFit.Unbind(event=wx.EVT_BUTTON, id= self.btFit.GetId())
972        if self.btFit.GetLabel().lower() == "stop":
973            self.btFit.SetBackgroundColour('red')
974            self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._StopFit,
975                             id=self.btFit.GetId())
976        elif self.btFit.GetLabel().lower() == "fit":
977            self.btFit.SetBackgroundColour(self.default_bt_colour)
978            self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit, 
979                            id=self.btFit.GetId())
980        else:
981            msg = "FitPage: fit button has unknown label"
982            raise ValuerError, msg
983        self._manager._reset_schedule_problem(value=0)
984         
985
986    def _StopFit(self, event):
987        """
988        Stop fit
989        """
990        #if self.engine_type=="scipy":
991        self._manager.stop_fit(self.uid)
992        self._manager._reset_schedule_problem(value=0)
993        self._on_fit_complete()
994         
995    def _on_select_model(self, event=None): 
996        """
997        call back for model selection
998        """ 
999        if self.plugin_rbutton.GetValue():
1000            self._show_combox_helper()
1001            print "went here"
1002           
1003        self._on_select_model_helper() 
1004        self.set_model_param_sizer(self.model)                   
1005       
1006        if self.model is None:
1007            self._set_bookmark_flag(False)
1008            self._set_save_flag(False)
1009        self.enable_disp.SetValue(False)
1010        self.disable_disp.SetValue(True)
1011        try:
1012            self.set_dispers_sizer()
1013        except:
1014            pass
1015        self.btFit.SetFocus() 
1016        self.state.enable_disp = self.enable_disp.GetValue()
1017        self.state.disable_disp = self.disable_disp.GetValue()
1018        self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
1019        self.state.slit_smearer = self.slit_smearer.GetValue()
1020   
1021        self.state.structurecombobox = self.structurebox.GetCurrentSelection()
1022        self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
1023     
1024        if self.model != None:
1025            self._set_bookmark_flag(True)
1026            self._set_save_flag(True)
1027            # Reset smearer, model and data
1028            self._set_smear(self.data)
1029            try:
1030                # update smearer sizer
1031                self.onSmear(None)
1032                temp_smear = None
1033                if self.enable_smearer.GetValue():
1034                    # Set the smearer environments
1035                    temp_smear = self.smearer
1036            except:
1037                raise
1038                ## error occured on chisqr computation
1039                #pass
1040            ## event to post model to fit to fitting plugins
1041            (ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
1042           
1043            ## set smearing value whether or not
1044            #    the data contain the smearing info
1045            evt = ModelEventbox(model=self.model, 
1046                                        smearer=temp_smear, 
1047                                        qmin=float(self.qmin_x),
1048                                        uid=self.uid,
1049                                     qmax=float(self.qmax_x)) 
1050   
1051            self._manager._on_model_panel(evt=evt)
1052            self.state.model = self.model.clone()
1053            self.state.model.name = self.model.name
1054            if event is not None:
1055                self._draw_model()
1056        if event != None:
1057           
1058            ## post state to fit panel
1059            event = PageInfoEvent(page = self)
1060            wx.PostEvent(self.parent, event) 
1061        self.SetupScrolling()
1062     
1063    def _onparamEnter(self,event):
1064        """
1065        when enter value on panel redraw model according to changed
1066        """
1067        if self.model ==None:
1068            msg="Please select a Model first..."
1069            wx.MessageBox(msg, 'Info')
1070            #wx.PostEvent(self._manager.parent, StatusEvent(status=\
1071            #                "Parameters: %s"%msg))
1072            return
1073
1074        #default flag
1075        flag = False
1076        self.fitrange = True
1077        #get event object
1078        tcrtl = event.GetEventObject()
1079       
1080        #wx.PostEvent(self._manager.parent, StatusEvent(status=" \
1081        #                        updating ... ",type="update"))
1082        #Clear msg if previously shown.
1083        msg= ""
1084        wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1085
1086        if check_float(tcrtl):
1087            flag = self._onparamEnter_helper() 
1088            self.set_npts2fit() 
1089            if self.fitrange:             
1090                temp_smearer = None
1091                if not self.disable_smearer.GetValue():
1092                    temp_smearer= self.current_smearer
1093                    ## set smearing value whether or not
1094                    #        the data contain the smearing info
1095                    if self.slit_smearer.GetValue():
1096                        flag1 = self.update_slit_smear()
1097                        flag = flag or flag1
1098                    elif self.pinhole_smearer.GetValue():
1099                        flag1 = self.update_pinhole_smear()
1100                        flag = flag or flag1
1101                elif self.data.__class__.__name__ !="Data2D":
1102                    self._manager.set_smearer(smearer=temp_smearer, 
1103                                              uid=self.uid,
1104                                             qmin= float(self.qmin_x),
1105                                            qmax= float(self.qmax_x),
1106                                            draw=True) 
1107                if flag:   
1108                    #self.compute_chisqr(smearer= temp_smearer)
1109       
1110                    ## new state posted
1111                    if self.state_change:
1112                        #self._undo.Enable(True)
1113                        event = PageInfoEvent(page = self)
1114                        wx.PostEvent(self.parent, event)
1115                    self.state_change= False 
1116            else: 
1117                # invalid fit range: do nothing here:
1118                # msg already displayed in validate
1119                return     
1120        else:
1121            self.save_current_state()
1122            msg= "Cannot Plot :Must enter a number!!!  "
1123            wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1124             
1125        self.save_current_state() 
1126        return 
1127   
1128    def _onparamRangeEnter(self, event):
1129        """
1130        Check validity of value enter in the parameters range field
1131        """
1132        #if self.check_invalid_panel():
1133        #    return
1134        tcrtl= event.GetEventObject()
1135        #Clear msg if previously shown.
1136        msg= ""
1137        wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1138        # Flag to register when a parameter has changed.
1139        is_modified = False
1140        if tcrtl.GetValue().lstrip().rstrip()!="":
1141            try:
1142                value = float(tcrtl.GetValue())
1143                tcrtl.SetBackgroundColour(wx.WHITE)
1144                self._check_value_enter(self.fittable_param ,is_modified)
1145                self._check_value_enter(self.parameters ,is_modified) 
1146            except:
1147                tcrtl.SetBackgroundColour("pink")
1148                msg= "Model Error:wrong value entered : %s"% sys.exc_value
1149                wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1150                return 
1151        else:
1152           tcrtl.SetBackgroundColour(wx.WHITE)
1153           
1154        #self._undo.Enable(True)
1155        self.save_current_state()
1156        event = PageInfoEvent(page = self)
1157        wx.PostEvent(self.parent, event)
1158        self.state_change= False
1159
1160                   
1161    def _onQrangeEnter(self, event):
1162        """
1163        Check validity of value enter in the Q range field
1164        """
1165        #if self.check_invalid_panel():
1166        #    return
1167        tcrtl= event.GetEventObject()
1168        #Clear msg if previously shown.
1169        msg= ""
1170        wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1171        # Flag to register when a parameter has changed.
1172        is_modified = False
1173        if tcrtl.GetValue().lstrip().rstrip()!="":
1174            try:
1175                value = float(tcrtl.GetValue())
1176                tcrtl.SetBackgroundColour(wx.WHITE)
1177
1178                # If qmin and qmax have been modified, update qmin and qmax
1179                if self._validate_qrange( self.qmin, self.qmax):
1180                    tempmin = float(self.qmin.GetValue())
1181                    if tempmin != self.qmin_x:
1182                        self.qmin_x = tempmin
1183                    tempmax = float(self.qmax.GetValue())
1184                    if tempmax != self.qmax_x:
1185                        self.qmax_x = tempmax
1186                else:
1187                    tcrtl.SetBackgroundColour("pink")
1188                    msg= "Model Error:wrong value entered : %s"% sys.exc_value
1189                    wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1190                    return 
1191               
1192            except:
1193                tcrtl.SetBackgroundColour("pink")
1194                msg= "Model Error:wrong value entered : %s"% sys.exc_value
1195                wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1196                return 
1197            #Check if # of points for theory model are valid(>0).
1198            # check for 2d
1199            if self.data.__class__.__name__ =="Data2D":
1200                # set mask   
1201                radius= numpy.sqrt( self.data.qx_data*self.data.qx_data + 
1202                                    self.data.qy_data*self.data.qy_data )
1203                index_data = ((self.qmin_x <= radius)& \
1204                                (radius<= self.qmax_x))
1205                index_data = (index_data)&(self.data.mask)
1206                index_data = (index_data)&(numpy.isfinite(self.data.data))
1207                if len(index_data[index_data]) < 10:
1208                    msg = "Cannot Plot :No or too little npts in"
1209                    msg += " that data range!!!  "
1210                    wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1211                    return
1212                else:
1213                    self.data.mask = index_data
1214                    #self.Npts_fit.SetValue(str(len(self.data.mask)))
1215                    self.set_npts2fit() 
1216            else:
1217                if self.data is not None:
1218                   
1219                    index_data = ((self.qmin_x <= self.data.x)& \
1220                                  (self.data.x <= self.qmax_x))
1221                    self.Npts_fit.SetValue(str(len(self.data.x[index_data])))
1222
1223           
1224        else:
1225           tcrtl.SetBackgroundColour("pink")
1226           msg= "Model Error:wrong value entered!!!"
1227           wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1228       
1229        self._draw_model()
1230        ##update chi2
1231        #self.compute_chisqr(smearer= temp_smearer)
1232        #self._undo.Enable(True)
1233        self.save_current_state()
1234        event = PageInfoEvent(page = self)
1235        wx.PostEvent(self.parent, event)
1236        self.state_change= False
1237        return
1238   
1239    def _clear_Err_on_Fit(self):
1240        """
1241        hide the error text control shown
1242        after fitting
1243        """
1244       
1245        if hasattr(self,"text2_3"):
1246            self.text2_3.Hide()
1247
1248        if len(self.parameters)>0:
1249            for item in self.parameters:
1250                #Skip t ifhe angle parameters if 1D data
1251                if self.data.__class__.__name__ !="Data2D":
1252                    if item in self.orientation_params:
1253                        continue
1254                if item in self.param_toFit:
1255                    continue
1256                ## hide statictext +/-   
1257                if item[3]!=None and item[3].IsShown():
1258                    item[3].Hide()
1259                ## hide textcrtl  for error after fit
1260                if item[4]!=None and item[4].IsShown():                   
1261                    item[4].Hide()
1262 
1263        if len(self.fittable_param)>0:
1264            for item in self.fittable_param:
1265                #Skip t ifhe angle parameters if 1D data
1266                if self.data.__class__.__name__ !="Data2D":
1267                    if item in self.orientation_params:
1268                        continue
1269                if item in self.param_toFit:
1270                    continue
1271                ## hide statictext +/-   
1272                if item[3]!=None and item[3].IsShown():
1273                    item[3].Hide()
1274                ## hide textcrtl  for error after fit
1275                if item[4]!=None and item[4].IsShown():
1276                    item[4].Hide()
1277        return       
1278               
1279    def _get_defult_custom_smear(self):
1280        """
1281        Get the defult values for custum smearing.
1282        """
1283        # get the default values
1284        if self.dxl == None: self.dxl = 0.0
1285        if self.dxw == None: self.dxw = ""
1286        if self.dx_min == None: self.dx_min = SMEAR_SIZE_L
1287        if self.dx_max == None: self.dx_max = SMEAR_SIZE_H
1288       
1289    def _get_smear_info(self):
1290        """
1291        Get the smear info from data.
1292       
1293        :return: self.smear_type, self.dq_l and self.dq_r,
1294            respectively the type of the smear, dq_min and
1295            dq_max for pinhole smear data
1296            while dxl and dxw for slit smear
1297           
1298        """
1299
1300        # default
1301        self.smear_type = None
1302        self.dq_l = None
1303        self.dq_r = None
1304        data = self.data
1305        if self.data is None:
1306            return
1307        elif self.data.__class__.__name__ == 'Data2D': 
1308            if data.dqx_data == None or  data.dqy_data ==None: 
1309                return
1310            elif self.smearer != None and (data.dqx_data.any()!=0) and \
1311                            (data.dqx_data.any()!=0): 
1312                self.smear_type = "Pinhole2d"
1313                self.dq_l = format_number(numpy.average(data.dqx_data)) 
1314                self.dq_r = format_number(numpy.average(data.dqy_data)) 
1315                return 
1316            else: 
1317                return
1318        # check if it is pinhole smear and get min max if it is.
1319        if data.dx != None and all(data.dx !=0): 
1320            self.smear_type = "Pinhole" 
1321            self.dq_l = data.dx[0]
1322            self.dq_r = data.dx[-1]
1323           
1324        # check if it is slit smear and get min max if it is.
1325        elif data.dxl != None or data.dxw != None: 
1326            self.smear_type = "Slit" 
1327            if data.dxl != None and all(data.dxl !=0):
1328                self.dq_l = data.dxl[0]
1329            if data.dxw != None and all(data.dxw !=0): 
1330                self.dq_r = data.dxw[0]   
1331        #return self.smear_type,self.dq_l,self.dq_r       
1332   
1333    def _show_smear_sizer(self):   
1334        """
1335        Show only the sizers depending on smear selection
1336        """
1337        # smear disabled
1338        if self.disable_smearer.GetValue():
1339            self.smear_description_none.Show(True)
1340        # 2Dsmear
1341        elif self._is_2D():
1342            self.smear_description_accuracy_type.Show(True)
1343            self.smear_accuracy.Show(True)
1344            self.smear_description_accuracy_type.Show(True)
1345            self.smear_description_2d.Show(True)
1346            self.smear_description_2d_x.Show(True)
1347            self.smear_description_2d_y.Show(True)
1348            if self.pinhole_smearer.GetValue():
1349                self.smear_pinhole_min.Show(True)
1350                self.smear_pinhole_max.Show(True)
1351        # smear from data
1352        elif self.enable_smearer.GetValue():
1353
1354            self.smear_description_dqdata.Show(True)
1355            if self.smear_type != None:
1356                self.smear_description_smear_type.Show(True)
1357                if self.smear_type == 'Slit':
1358                    self.smear_description_slit_height.Show(True)
1359                    self.smear_description_slit_width.Show(True)               
1360                elif self.smear_type == 'Pinhole':
1361                    self.smear_description_pin_min.Show(True)
1362                    self.smear_description_pin_max.Show(True)
1363                self.smear_description_smear_type.Show(True)
1364                self.smear_description_type.Show(True)
1365                self.smear_data_left.Show(True)
1366                self.smear_data_right.Show(True)
1367        # custom pinhole smear
1368        elif self.pinhole_smearer.GetValue():
1369            if self.smear_type == 'Pinhole':
1370                self.smear_message_new_p.Show(True)
1371                self.smear_description_pin_min.Show(True)
1372                self.smear_description_pin_max.Show(True)
1373
1374            self.smear_pinhole_min.Show(True)
1375            self.smear_pinhole_max.Show(True)
1376        # custom slit smear
1377        elif self.slit_smearer.GetValue():
1378            self.smear_message_new_s.Show(True)
1379            self.smear_description_slit_height.Show(True)
1380            self.smear_slit_height.Show(True)
1381            self.smear_description_slit_width.Show(True)
1382            self.smear_slit_width.Show(True)
1383
1384    def _hide_all_smear_info(self):
1385        """
1386        Hide all smearing messages in the set_smearer sizer
1387        """
1388        self.smear_description_none.Hide()
1389        self.smear_description_dqdata.Hide()
1390        self.smear_description_type.Hide()
1391        self.smear_description_smear_type.Hide()
1392        self.smear_description_accuracy_type.Hide()
1393        self.smear_description_2d_x.Hide()
1394        self.smear_description_2d_y.Hide()
1395        self.smear_description_2d.Hide()
1396       
1397        self.smear_accuracy.Hide()
1398        self.smear_data_left.Hide()
1399        self.smear_data_right.Hide()
1400        self.smear_description_pin_min.Hide()
1401        self.smear_pinhole_min.Hide()
1402        self.smear_description_pin_max.Hide()
1403        self.smear_pinhole_max.Hide()
1404        self.smear_description_slit_height.Hide()
1405        self.smear_slit_height.Hide()
1406        self.smear_description_slit_width.Hide()
1407        self.smear_slit_width.Hide()
1408        self.smear_message_new_p.Hide()
1409        self.smear_message_new_s.Hide()
1410   
1411    def _set_accuracy_list(self):
1412        """
1413        Set the list of an accuracy in 2D custum smear: Xhigh, High, Med, or Low
1414        """
1415        # list of accuracy choices
1416        list = ['Low','Med','High','Xhigh']
1417        for idx in range(len(list)):
1418            self.smear_accuracy.Append(list[idx],idx)
1419           
1420    def _set_fun_box_list(self,fun_box):
1421        """
1422        Set the list of func for multifunctional models
1423        """
1424        # Check if it is multi_functional model
1425        if self.model.__class__ not in self.model_list_box["Multi-Functions"]:
1426            return None
1427        # Get the func name list
1428        list = self.model.fun_list
1429        if len(list) == 0:
1430            return None
1431        # build function (combo)box
1432        ind = 0
1433        while(ind < len(list)):
1434            for key, val in list.iteritems():
1435                if (val == ind):
1436                    fun_box.Append(key,val)
1437                    break
1438            ind += 1
1439       
1440    def _on_select_accuracy(self,event):
1441        """
1442        Select an accuracy in 2D custom smear: Xhigh, High, Med, or Low
1443        """
1444        #event.Skip()
1445        # Check if the accuracy is same as before
1446        #self.smear2d_accuracy = event.GetEventObject().GetValue()
1447        self.smear2d_accuracy = self.smear_accuracy.GetValue()
1448        if self.pinhole_smearer.GetValue():
1449            self.onPinholeSmear(event=None)
1450        else:   
1451            self.onSmear(event=None)
1452            if self.current_smearer != None:
1453                self.current_smearer.set_accuracy(accuracy = 
1454                                                  self.smear2d_accuracy) 
1455        event.Skip()
1456
1457    def _on_fun_box(self,event):
1458        """
1459        Select an func: Erf,Rparabola,LParabola
1460        """
1461        fun_val = None
1462        fun_box = event.GetEventObject()
1463        name = fun_box.Name
1464        value = fun_box.GetValue()
1465        if self.model.fun_list.has_key(value):
1466            fun_val = self.model.fun_list[value]
1467
1468        self.model.setParam(name,fun_val)
1469        # save state
1470        self._copy_parameters_state(self.str_parameters, 
1471                                    self.state.str_parameters)
1472        # update params
1473        self._update_paramv_on_fit() 
1474        # draw
1475        self._draw_model()
1476        self.Refresh()
1477        # get ready for new event
1478        event.Skip()
1479       
1480    def _onMask(self, event):     
1481        """
1482        Build a panel to allow to edit Mask
1483        """
1484       
1485        from sans.guiframe.local_perspectives.plotting.masking \
1486        import MaskPanel as MaskDialog
1487       
1488        self.panel = MaskDialog(self, data=self.data,id =-1 )
1489        self.panel.Bind(wx.EVT_CLOSE, self._draw_masked_model)
1490        self.panel.ShowModal()
1491        #wx.PostEvent(self.parent, event)
1492       
1493    def _draw_masked_model(self,event):
1494        """
1495        Draw model image w/mask
1496        """
1497        event.Skip()
1498
1499        is_valid_qrange = self._update_paramv_on_fit()
1500
1501        if is_valid_qrange:
1502            # try re draw the model plot if it exists
1503            self._draw_model()
1504            self.panel.Destroy() # frame
1505            self.set_npts2fit()
1506        elif self.model == None:
1507            self.panel.Destroy()
1508            self.set_npts2fit()
1509            msg= "No model is found on updating MASK in the model plot... "
1510            wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
1511        else:
1512            msg = ' Please consider your Q range, too.'
1513            self.panel.ShowMessage(msg)
1514       
1515    def _set_smear(self, data):
1516        """
1517        """
1518        if data is None:
1519            return
1520        self.smearer = smear_selection(self.data, self.model)
1521        self.disable_smearer.SetValue(True)
1522        if self.smearer == None:
1523            self.enable_smearer.Disable()
1524        else:
1525            self.enable_smearer.Enable()
1526           
1527    def get_view_mode(self):
1528        """
1529        return True if the panel allow 2D or False if 1D
1530        """
1531        return self.enable2D
1532   
1533    def set_data(self, data):
1534        """
1535        reset the current data
1536        """
1537        flag = False
1538        if self.data is None and data is not None:
1539            self.window_name = str(data.name)
1540            ## Title to appear on top of the window
1541            self.window_caption = str(data.name)
1542            flag = True
1543        self.data = data
1544        if self.data is None:
1545            data_min = ""
1546            data_max = ""
1547            data_name = ""
1548            self._set_bookmark_flag(False)
1549            self._set_save_flag(False)
1550        else:
1551            self._set_bookmark_flag(True)
1552            self._set_save_flag(True)
1553            self._set_smear(data)
1554            # more disables for 2D
1555            if self.data.__class__.__name__ =="Data2D":
1556                self.slit_smearer.Disable()
1557                self.default_mask = copy.deepcopy(self.data.mask)
1558               
1559               
1560            self.formfactorbox.Enable()
1561            self.structurebox.Enable()
1562            data_name = self.data.name
1563            #set maximum range for x in linear scale
1564            if not hasattr(self.data,"data"): #Display only for 1D data fit
1565                # Minimum value of data   
1566                data_min = min(self.data.x)
1567                # Maximum value of data 
1568                data_max = max(self.data.x)
1569                #number of total data points
1570                self.Npts_total.SetValue(str(len(self.data.x)))
1571                #default:number of data points selected to fit
1572                self.Npts_fit.SetValue(str(len(self.data.x)))
1573                self.btEditMask.Disable() 
1574                self.EditMask_title.Disable()
1575            else:
1576               
1577                ## Minimum value of data
1578                data_min = 0
1579                x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax)) 
1580                y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
1581                ## Maximum value of data 
1582                data_max = math.sqrt(x*x + y*y)
1583                #number of total data points
1584                self.Npts_total.SetValue(str(len(self.data.data)))
1585                #default:number of data points selected to fit
1586                self.Npts_fit.SetValue(str(len(self.data.data)))
1587                self.btEditMask.Enable() 
1588                self.EditMask_title.Enable() 
1589        self.dataSource.SetValue(data_name)
1590        self.qmin_x = data_min
1591        self.qmax_x = data_max
1592        self.minimum_q.SetValue(str(data_min))
1593        self.maximum_q.SetValue(str(data_max))
1594        self.qmin.SetValue(str(data_min))
1595        self.qmax.SetValue(str(data_max))
1596        self.qmin.SetBackgroundColour("white")
1597        self.qmax.SetBackgroundColour("white")
1598        self.state.data = data
1599        self.state.qmin = self.qmin_x
1600        self.state.qmax = self.qmax_x
1601       
1602        #update model plot with new data information
1603        if flag:
1604            #set model view button
1605            if self.data.__class__.__name__ == "Data2D":
1606                self.enable2D = True
1607                self.model_view.SetLabel("2D Mode")
1608            else:
1609                self.enable2D = False
1610                self.model_view.SetLabel("1D Mode")
1611               
1612            self.model_view.Disable()
1613            self._draw_model()
1614       
1615    def reset_page(self, state,first=False):
1616        """
1617        reset the state
1618        """
1619        self.reset_page_helper(state)
1620        #import sans.guiframe.gui_manager
1621        #evt = ModelEventbox(model=state.model)
1622        #wx.PostEvent(self.event_owner, evt) 
1623   
1624        if self.engine_type != None:
1625            self._manager._on_change_engine(engine=self.engine_type)
1626
1627        self.select_param(event = None) 
1628        #Save state_fit
1629        self.save_current_state_fit()
1630        self._lay_out()
1631        self.Refresh()
1632       
1633    def get_range(self):
1634        """
1635        return the fitting range
1636        """
1637        return float(self.qmin_x) , float(self.qmax_x)
1638   
1639    def get_npts2fit(self):
1640        """
1641        return numbers of data points within qrange
1642       
1643        :Note: This is for Park where chi2 is not normalized by Npts of fit
1644       
1645        """
1646        if self.data is None:
1647            return
1648        npts2fit = 0
1649        qmin,qmax = self.get_range()
1650        if self.data.__class__.__name__ =="Data2D": 
1651            radius= numpy.sqrt( self.data.qx_data*self.data.qx_data + 
1652                                self.data.qy_data*self.data.qy_data )
1653            index_data = (self.qmin_x <= radius)&(radius<= self.qmax_x)
1654            index_data= (index_data)&(self.data.mask)
1655            index_data = (index_data)&(numpy.isfinite(self.data.data))
1656            npts2fit = len(self.data.data[index_data])
1657        else:
1658            for qx in self.data.x:
1659                   if qx >= qmin and qx <= qmax:
1660                       npts2fit += 1
1661        return npts2fit
1662
1663    def set_npts2fit(self):
1664        """
1665        setValue Npts for fitting
1666        """
1667        self.Npts_fit.SetValue(str(self.get_npts2fit()))
1668       
1669    def get_chi2(self):
1670        """
1671        return the current chi2
1672        """
1673        return self.tcChi.GetValue()
1674       
1675    def get_param_list(self):
1676        """
1677        :return self.param_toFit: list containing  references to TextCtrl
1678            checked.Theses TextCtrl will allow reference to parameters to fit.
1679       
1680        :raise: if return an empty list of parameter fit will nnote work
1681            properly so raise ValueError,"missing parameter to fit"
1682        """
1683        if self.param_toFit !=[]:
1684            return self.param_toFit
1685        else:
1686            raise ValueError,"missing parameter to fit"   
1687     
1688    def onsetValues(self,chisqr,p_name, out,cov):
1689        """
1690        Build the panel from the fit result
1691       
1692        :param chisqr: Value of the goodness of fit metric
1693        :param p_name: the name of parameters
1694        :param out: list of parameter with the best value found during fitting
1695        :param cov: Covariance matrix
1696   
1697        """
1698        if out == None or not numpy.isfinite(chisqr):
1699            raise ValueError,"Fit error occured..." 
1700       
1701        is_modified = False
1702        has_error = False   
1703       
1704        #Hide textctrl boxes of errors.
1705        self._clear_Err_on_Fit()   
1706       
1707        #Check if chi2 is finite
1708        if chisqr != None or numpy.isfinite(chisqr):
1709        #format chi2 
1710            if self.engine_type == "park": 
1711                npt_fit = float(self.get_npts2fit()) 
1712                if npt_fit > 0:
1713                    chisqr =chisqr/npt_fit   
1714            chi2 = format_number(chisqr)   
1715            self.tcChi.SetValue(chi2)   
1716            self.tcChi.Refresh()   
1717        else:
1718            self.tcChi.SetValue("-")
1719       
1720        #Hide error title
1721        if self.text2_3.IsShown():
1722            self.text2_3.Hide()
1723     
1724        try:
1725            n = self.disp_box.GetCurrentSelection()
1726            dispersity= self.disp_box.GetClientData(n)
1727            if dispersity !=None and self.enable_disp.GetValue():
1728                name= dispersity.__name__
1729                if name == "GaussianDispersion":
1730                    if hasattr(self,"text_disp_1" ):
1731                        if self.text_disp_1 !=None:
1732                            self.text_disp_1.Hide()
1733        except:
1734            dispersty = None
1735            pass
1736        #set the panel when fit result are float not list
1737        if out.__class__== numpy.float64:
1738            self.param_toFit[0][2].SetValue(format_number(out))
1739           
1740            if self.param_toFit[0][4].IsShown:
1741                self.param_toFit[0][4].Hide()
1742            if cov !=None :
1743                self.text2_3.Show(True)
1744                try:
1745                    if dispersity !=None:
1746                        name= dispersity.__name__
1747                        if name == "GaussianDispersion" and \
1748                        self.enable_disp.GetValue():
1749                            if hasattr(self,"text_disp_1" ):
1750                                if self.text_disp_1 !=None:
1751                                    self.text_disp_1.Show(True)
1752                except:
1753                    pass
1754
1755                if cov[0]==None or  not numpy.isfinite(cov[0]): 
1756                    if self.param_toFit[0][3].IsShown:
1757                        self.param_toFit[0][3].Hide()
1758                else:                   
1759                    self.param_toFit[0][3].Show(True)               
1760                    self.param_toFit[0][4].Show(True)
1761                    self.param_toFit[0][4].SetValue(format_number(cov[0]))
1762                    has_error = True
1763        else:
1764
1765            i = 0
1766            #Set the panel when fit result are list
1767            for item in self.param_toFit:     
1768                if len(item)>5 and item != None:     
1769                    ## reset error value to initial state
1770                    item[3].Hide()
1771                    item[4].Hide()
1772                   
1773                    for ind in range(len(out)):
1774                       
1775                        if item[1] == p_name[ind]:
1776                            break       
1777                    if len(out)<=len(self.param_toFit) and out[ind] !=None:   
1778                        val_out = format_number(out[ind])                 
1779                        item[2].SetValue(val_out)
1780
1781                    if(cov !=None):
1782                       
1783                        try:
1784                            if dispersity !=None:
1785                                name= dispersity.__name__
1786                                if name == "GaussianDispersion" and \
1787                                                self.enable_disp.GetValue():
1788                                    if hasattr(self,"text_disp_1" ):
1789                                        if self.text_disp_1!=None:
1790                                            if not self.text_disp_1.IsShown():
1791                                                self.text_disp_1.Show(True)
1792                        except:
1793                            pass   
1794                   
1795                        if cov[ind]!=None :
1796                            if numpy.isfinite(float(cov[ind])):
1797                                val_err = format_number(cov[ind])
1798                                item[3].Show(True)
1799                                item[4].Show(True)
1800                                item[4].SetValue(val_err)
1801
1802                                has_error = True
1803                    i += 1         
1804        #Show error title when any errors displayed
1805        if has_error: 
1806            if not self.text2_3.IsShown():
1807                self.text2_3.Show(True) 
1808               
1809        ## save current state 
1810        self.save_current_state()         
1811        #plot model
1812        self._draw_model()   
1813        self._lay_out()     
1814        #PostStatusEvent     
1815        msg = "Fit completed! "
1816        wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
1817
1818    def onPinholeSmear(self, event):
1819        """
1820        Create a custom pinhole smear object that will change the way residuals
1821        are compute when fitting
1822       
1823        :Note: accuracy is given by strings'High','Med', 'Low' FOR 2d,
1824                     None for 1D
1825                     
1826        """
1827
1828        #if self.check_invalid_panel():
1829        #    return
1830        if self.model ==None:
1831            self.disable_smearer.SetValue(True)
1832            msg="Please select a Model first..."
1833            wx.MessageBox(msg, 'Info')
1834            wx.PostEvent(self._manager.parent, StatusEvent(status=\
1835                            "Smear: %s"%msg))
1836            return
1837
1838        # Need update param values
1839        self._update_paramv_on_fit()     
1840
1841        # msg default
1842        msg = None
1843        if event != None:
1844            tcrtl= event.GetEventObject()
1845            # event case of radio button
1846            if tcrtl.GetValue()== True:
1847                self.dx_min = 0.0
1848                self.dx_max = 0.0
1849                is_new_pinhole = True
1850            else:
1851                is_new_pinhole = self._is_changed_pinhole()
1852        else:
1853            is_new_pinhole = True 
1854        # if any value is changed
1855        if is_new_pinhole:
1856            msg = self._set_pinhole_smear()
1857        # hide all silt sizer   
1858        self._hide_all_smear_info()
1859       
1860        ##Calculate chi2
1861        temp_smearer = self.current_smearer
1862        #self.compute_chisqr(smearer= temp_smearer)
1863           
1864        # show relevant slit sizers
1865        self._show_smear_sizer()
1866
1867        self.sizer_set_smearer.Layout()
1868        self.Layout()   
1869       
1870        if event != None: 
1871            event.Skip()                 
1872        #self._undo.Enable(True)
1873        self.save_current_state()
1874        event = PageInfoEvent(page = self)
1875        wx.PostEvent(self.parent, event)
1876       
1877    def _is_changed_pinhole(self): 
1878        """
1879        check if any of pinhole smear is changed
1880       
1881        :return: True or False
1882       
1883        """
1884        # get the values
1885        pin_min = self.smear_pinhole_min.GetValue()
1886        pin_max = self.smear_pinhole_max.GetValue()
1887                   
1888        # Check changes in slit width   
1889        try:
1890            dx_min = float(pin_min)
1891        except:
1892            return True
1893        if self.dx_min != dx_min:
1894            return True
1895       
1896        # Check changes in slit heigth
1897        try:
1898            dx_max = float(pin_max)
1899        except:
1900            return True
1901        if self.dx_max != dx_max:
1902            return True
1903        return False
1904   
1905    def _set_pinhole_smear(self):
1906        """
1907        Set custom pinhole smear
1908       
1909        :return: msg
1910       
1911        """
1912        # copy data
1913        data = copy.deepcopy(self.data)
1914        if self._is_2D():
1915            self.smear_type = 'Pinhole2d'
1916            len_data = len(data.data)
1917            data.dqx_data = numpy.zeros(len_data)
1918            data.dqy_data = numpy.zeros(len_data)
1919        else:
1920            self.smear_type = 'Pinhole'
1921            len_data = len(data.x)
1922            data.dx = numpy.zeros(len_data)
1923            data.dxl = None
1924            data.dxw = None
1925        msg = None
1926   
1927        get_pin_min = self.smear_pinhole_min
1928        get_pin_max = self.smear_pinhole_max       
1929
1930        if not check_float(get_pin_min):
1931            get_pin_min.SetBackgroundColour("pink")
1932            msg= "Model Error:wrong value entered!!!"
1933        elif not check_float(get_pin_max ):
1934            get_pin_max.SetBackgroundColour("pink")
1935            msg= "Model Error:wrong value entered!!!"
1936        else:
1937            if len_data < 2: len_data = 2     
1938            self.dx_min = float(get_pin_min.GetValue())
1939            self.dx_max = float(get_pin_max.GetValue())
1940            if self.dx_min < 0:
1941                get_pin_min.SetBackgroundColour("pink")
1942                msg= "Model Error:This value can not be negative!!!"
1943            elif self.dx_max <0:
1944                get_pin_max.SetBackgroundColour("pink")
1945                msg= "Model Error:This value can not be negative!!!"
1946            elif self.dx_min != None and self.dx_max != None:   
1947                if self._is_2D():
1948                    data.dqx_data[data.dqx_data==0] = self.dx_min
1949                    data.dqy_data[data.dqy_data==0] = self.dx_max         
1950                elif self.dx_min == self.dx_max:
1951                    data.dx[data.dx==0] = self.dx_min
1952                else:
1953                    step = (self.dx_max - self.dx_min)/(len_data-1)   
1954                    data.dx = numpy.arange(self.dx_min,self.dx_max+step/1.1,
1955                                           step)           
1956            elif self.dx_min != None: 
1957                if self._is_2D(): data.dqx_data[data.dqx_data==0] = self.dx_min
1958                else: data.dx[data.dx==0] = self.dx_min
1959            elif self.dx_max != None:
1960                if self._is_2D(): data.dqy_data[data.dqy_data==0] = self.dx_max
1961                else: data.dx[data.dx==0] = self.dx_max         
1962            self.current_smearer = smear_selection(data, self.model)
1963            # 2D need to set accuracy
1964            if self._is_2D(): 
1965                self.current_smearer.set_accuracy(accuracy = \
1966                                                  self.smear2d_accuracy)
1967
1968        if msg != None:
1969            wx.PostEvent(self._manager.parent, StatusEvent(status = msg ))
1970        else:
1971            get_pin_min.SetBackgroundColour("white")
1972            get_pin_max.SetBackgroundColour("white")
1973        ## set smearing value whether or not the data contain the smearing info
1974        self._manager.set_smearer(smearer=self.current_smearer,
1975                                 qmin=float(self.qmin_x),
1976                                 qmax= float(self.qmax_x),
1977                                 uid=self.uid)
1978        return msg
1979       
1980    def update_pinhole_smear(self):
1981        """
1982        called by kill_focus on pinhole TextCntrl
1983        to update the changes
1984       
1985        :return: False when wrong value was entered
1986       
1987        """
1988        # msg default
1989        msg = None
1990        # check if any value is changed
1991        if self._is_changed_pinhole():
1992            msg = self._set_pinhole_smear()           
1993        #self._undo.Enable(True)
1994        self.save_current_state()
1995
1996        if msg != None:
1997            return False   
1998        else:
1999            return True
2000                     
2001    def onSlitSmear(self, event):
2002        """
2003        Create a custom slit smear object that will change the way residuals
2004        are compute when fitting
2005        """
2006 
2007        #if self.check_invalid_panel():
2008        #    return
2009
2010        if self.model ==None:
2011            self.disable_smearer.SetValue(True)
2012           
2013            msg="Please select a Model first..."
2014            wx.MessageBox(msg, 'Info')
2015            wx.PostEvent(self._manager.parent, StatusEvent(status=\
2016                            "Smear: %s"%msg))
2017            return
2018
2019        # Need update param values
2020        self._update_paramv_on_fit()
2021
2022        # msg default
2023        msg = None
2024        # for event given
2025        if event != None:
2026            tcrtl= event.GetEventObject()
2027            # event case of radio button
2028            if tcrtl.GetValue():
2029                self.dxl = 0.0
2030                self.dxw = 0.0
2031                is_new_slit = True
2032            else:
2033                is_new_slit = self._is_changed_slit()
2034        else:
2035            is_new_slit = True 
2036       
2037        # if any value is changed
2038        if is_new_slit:
2039            msg = self._set_slit_smear()
2040           
2041        # hide all silt sizer
2042        self._hide_all_smear_info()       
2043        ##Calculate chi2
2044        #self.compute_chisqr(smearer= self.current_smearer) 
2045        # show relevant slit sizers       
2046        self._show_smear_sizer()
2047        self.sizer_set_smearer.Layout()
2048        self.Layout()
2049
2050        if event != None:
2051            event.Skip()     
2052        #self._undo.Enable(True)
2053        self.save_current_state()
2054        event = PageInfoEvent(page = self)
2055        wx.PostEvent(self.parent, event)
2056        if msg != None:
2057            wx.PostEvent(self._manager.parent, StatusEvent(status = msg))
2058
2059    def _is_changed_slit(self): 
2060        """
2061        check if any of slit lengths is changed
2062       
2063        :return: True or False
2064       
2065        """
2066        # get the values
2067        width = self.smear_slit_width.GetValue()
2068        height = self.smear_slit_height.GetValue()
2069       
2070        # check and change the box bg color if it was pink
2071        #    but it should be white now
2072        # because this is the case that _set_slit_smear() will not handle
2073        if height.lstrip().rstrip()=="":
2074            self.smear_slit_height.SetBackgroundColour(wx.WHITE)
2075        if width.lstrip().rstrip()=="":
2076            self.smear_slit_width.SetBackgroundColour(wx.WHITE)
2077           
2078        # Check changes in slit width   
2079        if width == "": 
2080            dxw = 0.0
2081        else:
2082            try:
2083                dxw = float(width)
2084            except:
2085                return True
2086        if self.dxw != dxw:
2087            return True
2088       
2089        # Check changes in slit heigth
2090        if height == "": 
2091            dxl = 0.0
2092        else:
2093            try:
2094                dxl = float(height)
2095            except:
2096                return True
2097        if self.dxl != dxl:
2098            return True
2099
2100        return False
2101   
2102    def _set_slit_smear(self):
2103        """
2104        Set custom slit smear
2105       
2106        :return: message to inform the user about the validity
2107            of the values entered for slit smear
2108        """
2109        if self.data.__class__.__name__ == "Data2D":
2110            return
2111        temp_smearer = None
2112        # make sure once more if it is smearer
2113        data = copy.deepcopy(self.data)
2114        data_len = len(data.x)
2115        data.dx = None
2116        data.dxl = None
2117        data.dxw = None
2118        msg = None
2119   
2120        try:
2121            self.dxl = float(self.smear_slit_height.GetValue())
2122            data.dxl = self.dxl* numpy.ones(data_len)
2123            self.smear_slit_height.SetBackgroundColour(wx.WHITE)
2124        except: 
2125            data.dxl = numpy.zeros(data_len)
2126            if self.smear_slit_height.GetValue().lstrip().rstrip()!="":
2127                self.smear_slit_height.SetBackgroundColour("pink")
2128                msg = "Wrong value entered... " 
2129            else:
2130                self.smear_slit_height.SetBackgroundColour(wx.WHITE)
2131        try:
2132            self.dxw = float(self.smear_slit_width.GetValue())
2133            self.smear_slit_width.SetBackgroundColour(wx.WHITE)
2134            data.dxw = self.dxw* numpy.ones(data_len)
2135        except: 
2136            data.dxw = numpy.zeros(data_len)
2137            if self.smear_slit_width.GetValue().lstrip().rstrip()!="":
2138                self.smear_slit_width.SetBackgroundColour("pink")
2139                msg = "Wrong Fit value entered... "
2140            else:
2141                self.smear_slit_width.SetBackgroundColour(wx.WHITE)
2142             
2143        self.current_smearer = smear_selection(data, self.model)
2144        #temp_smearer = self.current_smearer
2145        ## set smearing value whether or not the data contain the smearing info
2146        self._manager.set_smearer(smearer=self.current_smearer, 
2147                                 qmin=float(self.qmin_x), 
2148                                 qmax= float(self.qmax_x),
2149                                 id=self.id) 
2150        return msg
2151   
2152    def update_slit_smear(self):
2153        """
2154        called by kill_focus on pinhole TextCntrl
2155        to update the changes
2156       
2157        :return: False when wrong value was entered
2158       
2159        """             
2160        # msg default
2161        msg = None
2162        # check if any value is changed
2163        if self._is_changed_slit():
2164            msg = self._set_slit_smear()           
2165        #self._undo.Enable(True)
2166        self.save_current_state()
2167
2168        if msg != None:
2169            return False   
2170        else:
2171            return True
2172                           
2173    def onSmear(self, event):
2174        """
2175        Create a smear object that will change the way residuals
2176        are compute when fitting
2177        """
2178        if event != None: 
2179            event.Skip()   
2180        if self.data is None:
2181            return
2182       
2183        if self.model == None:
2184            self.disable_smearer.SetValue(True)
2185            msg="Please select a Model first..."
2186            wx.MessageBox(msg, 'Info')
2187            wx.PostEvent(self._manager.parent, StatusEvent(status=\
2188                            "Smear: %s"%msg))
2189            return
2190       
2191        # Need update param values
2192        self._update_paramv_on_fit()
2193       
2194        temp_smearer = None
2195        self._get_smear_info()
2196       
2197        #renew smear sizer
2198        if self.smear_type != None:
2199            self.smear_description_smear_type.SetValue(str(self.smear_type))
2200            self.smear_data_left.SetValue(str(self.dq_l))   
2201            self.smear_data_right.SetValue(str(self.dq_r))       
2202
2203        self._hide_all_smear_info()
2204       
2205        data = copy.deepcopy(self.data)
2206        # make sure once more if it is smearer
2207        self.current_smearer = smear_selection(data, self.model)
2208       
2209        if self.enable_smearer.GetValue():
2210            if hasattr(self.data,"dxl"):
2211               
2212                msg= ": Resolution smearing parameters"
2213            if hasattr(self.data,"dxw"):
2214                msg= ": Slit smearing parameters"
2215            if self.smearer ==None:
2216                wx.PostEvent(self._manager.parent, StatusEvent(status=\
2217                            "Data contains no smearing information"))
2218            else:
2219                wx.PostEvent(self._manager.parent, StatusEvent(status=\
2220                            "Data contains smearing information"))
2221
2222            #self.smear_description_dqdata.Show(True)
2223            self.smear_data_left.Show(True)
2224            self.smear_data_right.Show(True)
2225            temp_smearer= self.current_smearer
2226        elif self.disable_smearer.GetValue():
2227            self.smear_description_none.Show(True)
2228           
2229        self._show_smear_sizer()
2230       
2231        self.sizer_set_smearer.Layout()
2232        self.Layout()
2233        ## set smearing value whether or not the data contain the smearing info
2234        self._manager.set_smearer(uid=self.uid, smearer=temp_smearer, qmin= float(self.qmin_x),
2235                        qmax= float(self.qmax_x), draw=True) 
2236       
2237        ##Calculate chi2
2238        #self.compute_chisqr(smearer= temp_smearer)
2239       
2240        self.state.enable_smearer=  self.enable_smearer.GetValue()
2241        self.state.disable_smearer=self.disable_smearer.GetValue()
2242        self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
2243        self.state.slit_smearer = self.slit_smearer.GetValue()
2244     
2245    def on_complete_chisqr(self, event): 
2246        """
2247        print result chisqr
2248       
2249        :event: activated by fitting/ complete after draw
2250       
2251        """
2252        try:
2253            if event ==None:
2254                output= "-"
2255            else:
2256                output = event.output
2257            self.tcChi.SetValue(str(format_number(output)))
2258
2259            self.state.tcChi = self.tcChi.GetValue()
2260        except:
2261            pass 
2262           
2263
2264    def select_all_param(self,event): 
2265        """
2266        set to true or false all checkBox given the main checkbox value cb1
2267        """           
2268        self.param_toFit=[]
2269        if  self.parameters !=[]:
2270            if  self.cb1.GetValue():
2271                for item in self.parameters:
2272                    ## for data2D select all to fit
2273                    if self.data.__class__.__name__=="Data2D":
2274                        item[0].SetValue(True)
2275                        self.param_toFit.append(item )
2276                    else:
2277                        ## for 1D all parameters except orientation
2278                        if not item in self.orientation_params:
2279                            item[0].SetValue(True)
2280                            self.param_toFit.append(item )
2281                #if len(self.fittable_param)>0:
2282                for item in self.fittable_param:
2283                    if self.data.__class__.__name__=="Data2D":
2284                        item[0].SetValue(True)
2285                        self.param_toFit.append(item )
2286                    else:
2287                        ## for 1D all parameters except orientation
2288                        if not item in self.orientation_params_disp:
2289                            item[0].SetValue(True)
2290                            self.param_toFit.append(item )
2291            else:
2292                for item in self.parameters:
2293                    item[0].SetValue(False)
2294                for item in self.fittable_param:
2295                    item[0].SetValue(False)
2296                self.param_toFit=[]
2297           
2298        self.save_current_state_fit() 
2299       
2300        if event !=None:
2301            #self._undo.Enable(True)
2302            ## post state to fit panel
2303            event = PageInfoEvent(page = self)
2304            wx.PostEvent(self.parent, event) 
2305     
2306    def select_param(self,event):
2307        """
2308        Select TextCtrl  checked for fitting purpose and stores them
2309        in  self.param_toFit=[] list
2310        """
2311        self.param_toFit=[]
2312        for item in self.parameters:
2313            #Skip t ifhe angle parameters if 1D data
2314            if self.data.__class__.__name__ !="Data2D":
2315                if item in self.orientation_params:
2316                    continue
2317            #Select parameters to fit for list of primary parameters
2318            if item[0].GetValue():
2319                if not (item in self.param_toFit):
2320                    self.param_toFit.append(item ) 
2321            else:
2322                #remove parameters from the fitting list
2323                if item in self.param_toFit:
2324                    self.param_toFit.remove(item)
2325
2326        #Select parameters to fit for list of fittable parameters
2327        #        with dispersion         
2328        for item in self.fittable_param:
2329            #Skip t ifhe angle parameters if 1D data
2330            if self.data.__class__.__name__ !="Data2D":
2331                if item in self.orientation_params:
2332                    continue
2333            if item[0].GetValue():
2334                if not (item in self.param_toFit):
2335                    self.param_toFit.append(item) 
2336            else:
2337                #remove parameters from the fitting list
2338                if item in self.param_toFit:
2339                    self.param_toFit.remove(item)
2340
2341        #Calculate num. of angle parameters
2342        if self.data.__class__.__name__ =="Data2D": 
2343            len_orient_para = 0
2344        else:
2345            len_orient_para = len(self.orientation_params)  #assume even len
2346        #Total num. of angle parameters
2347        if len(self.fittable_param) > 0:
2348            len_orient_para *= 2
2349        #Set the value of checkbox that selected every checkbox or not           
2350        if len(self.parameters)+len(self.fittable_param)-len_orient_para ==\
2351                len(self.param_toFit):
2352            self.cb1.SetValue(True)
2353        else:
2354            self.cb1.SetValue(False)
2355        self.save_current_state_fit()
2356        if event !=None:
2357            #self._undo.Enable(True)
2358            ## post state to fit panel
2359            event = PageInfoEvent(page = self)
2360            wx.PostEvent(self.parent, event) 
2361   
2362    def set_model_param_sizer(self, model):
2363        """
2364        Build the panel from the model content
2365       
2366        :param model: the model selected in combo box for fitting purpose
2367       
2368        """
2369        self.sizer3.Clear(True)
2370        self.parameters = []
2371        self.str_parameters = []
2372        self.param_toFit=[]
2373        self.fittable_param=[]
2374        self.fixed_param=[]
2375        self.orientation_params=[]
2376        self.orientation_params_disp=[]
2377       
2378        if model ==None:
2379            self.sizer3.Layout()
2380            self.SetupScrolling()
2381            return
2382        ## the panel is drawn using the current value of the fit engine
2383        if self.engine_type==None and self._manager !=None:
2384            self.engine_type= self._manager._return_engine_type()
2385
2386        box_description= wx.StaticBox(self, -1,str("Model Parameters"))
2387        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
2388        sizer = wx.GridBagSizer(5,5)
2389        ## save the current model
2390        self.model = model
2391           
2392        keys = self.model.getParamList()
2393        #list of dispersion paramaters
2394        self.disp_list=self.model.getDispParamList()
2395
2396        def custom_compare(a,b):
2397            """
2398            Custom compare to order, first by alphabets then second by number.
2399            """ 
2400            # number at the last digit
2401            a_last = a[len(a)-1]
2402            b_last = b[len(b)-1]
2403            # default
2404            num_a = None
2405            num_b = None
2406            # split the names
2407            a2 = a.lower().split('_')
2408            b2 = b.lower().split('_')
2409            # check length of a2, b2
2410            len_a2 = len(a2)
2411            len_b2 = len(b2)
2412            # check if it contains a int number(<10)
2413            try: 
2414                num_a = int(a_last)
2415            except: pass
2416            try:
2417                num_b = int(b_last)
2418            except: pass
2419            # Put 'scale' near the top; happens
2420            # when numbered param name exists
2421            if a == 'scale':
2422                return -1
2423            # both have a number   
2424            if num_a != None and num_b != None:
2425                if num_a > num_b: return -1
2426                # same number
2427                elif num_a == num_b: 
2428                    # different last names
2429                    if a2[len_a2-1] != b2[len_b2-1] and num_a != 0:
2430                        return -cmp(a2[len_a2-1], b2[len_b2-1])
2431                    else: 
2432                        return cmp(a, b) 
2433                else: return 1
2434            # one of them has a number
2435            elif num_a != None: return 1
2436            elif num_b != None: return -1
2437            # no numbers
2438            else: return cmp(a.lower(), b.lower())
2439
2440                       
2441        keys.sort(custom_compare)
2442   
2443        iy = 0
2444        ix = 0
2445        select_text = "Select All"
2446        self.cb1 = wx.CheckBox(self, -1,str(select_text), (10, 10))
2447        wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
2448        self.cb1.SetToolTipString("To check/uncheck all the boxes below.")
2449        #self.cb1.SetValue(True)
2450       
2451        sizer.Add(self.cb1,(iy, ix),(1,1),\
2452                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
2453        ix +=1
2454        self.text2_2 = wx.StaticText(self, -1, 'Values')
2455        sizer.Add(self.text2_2,(iy, ix),(1,1),\
2456                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2457        ix +=2 
2458        self.text2_3 = wx.StaticText(self, -1, 'Errors')
2459        sizer.Add(self.text2_3,(iy, ix),(1,1),\
2460                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2461        self.text2_3.Hide()
2462        ix +=1 
2463        self.text2_min = wx.StaticText(self, -1, 'Min')
2464        sizer.Add(self.text2_min,(iy, ix),(1,1),\
2465                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2466        self.text2_min.Hide()
2467        ix +=1 
2468        self.text2_max = wx.StaticText(self, -1, 'Max')
2469        sizer.Add(self.text2_max,(iy, ix),(1,1),\
2470                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2471        self.text2_max.Hide()
2472        ix += 1
2473        self.text2_4 = wx.StaticText(self, -1, '[Units]')
2474        sizer.Add(self.text2_4,(iy, ix),(1,1),\
2475                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2476        self.text2_4.Hide()
2477        if self.engine_type=="park":
2478            self.text2_max.Show(True)
2479            self.text2_min.Show(True)
2480       
2481        for item in keys:
2482            if not item in self.disp_list and not item in \
2483                    self.model.orientation_params:
2484               
2485                ##prepare a spot to store errors
2486                if not self.model.details.has_key(item):
2487                    self.model.details [item] = ["",None,None] 
2488         
2489                iy += 1
2490                ix = 0
2491                if self.model.__class__ in \
2492                    self.model_list_box["Multi-Functions"] \
2493                    and item in self.model.non_fittable:
2494                    non_fittable_name = wx.StaticText(self, -1, item )
2495                    sizer.Add(non_fittable_name,(iy, ix),(1,1),\
2496                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 21)
2497                    ## add parameter value
2498                    ix += 1
2499                    value= self.model.getParam(item)
2500                    if len(self.model.fun_list) > 0:
2501                        num = item.split('_')[1][5:7]
2502                        fun_box = wx.ComboBox(self, -1,size=(100,-1),
2503                                    style=wx.CB_READONLY, name = '%s'% item)
2504                        self._set_fun_box_list(fun_box)
2505                        fun_box.SetSelection(0)
2506                        #self.fun_box.SetToolTipString("A function
2507                        #    describing the interface")
2508                        wx.EVT_COMBOBOX(fun_box,-1, self._on_fun_box)
2509                    else:
2510                        fun_box = self.ModelTextCtrl(self, -1, 
2511                                                     size=(_BOX_WIDTH,20),
2512                                style=wx.TE_PROCESS_ENTER, name ='%s'% item)
2513                        fun_box.SetToolTipString("Hit 'Enter' after typing.")
2514                        fun_box.SetValue(format_number(value))
2515                    sizer.Add(fun_box, (iy,ix),(1,1), wx.EXPAND)
2516                    self.str_parameters.append([None,item, fun_box,
2517                                                None,None,None,None,None])
2518
2519                   
2520                else:
2521                    ## add parameters name with checkbox for selecting to fit
2522                    cb = wx.CheckBox(self, -1, item )             
2523                    cb.SetToolTipString(" Check for fitting.")
2524                    #cb.SetValue(True)
2525                    wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
2526                   
2527                    sizer.Add( cb,( iy, ix),(1,1),
2528                                 wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
2529
2530                    ## add parameter value
2531                    ix += 1
2532                    value= self.model.getParam(item)
2533                    ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
2534                                        style=wx.TE_PROCESS_ENTER)
2535                    ctl1.SetToolTipString("Hit 'Enter' after typing.")
2536                    ctl1.SetValue(format_number(value))
2537                    sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
2538                    ## text to show error sign
2539                    ix += 1
2540                    text2=wx.StaticText(self, -1, '+/-')
2541                    sizer.Add(text2,(iy, ix),(1,1),\
2542                                    wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2543                    text2.Hide() 
2544                    ix += 1
2545                    ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
2546                    sizer.Add(ctl2, (iy,ix),(1,1), 
2547                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2548                    ctl2.Hide()
2549                   
2550                    ix += 1
2551                    ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
2552                                               style=wx.TE_PROCESS_ENTER,
2553                                text_enter_callback = self._onparamRangeEnter)
2554         
2555                    sizer.Add(ctl3, (iy,ix),(1,1), 
2556                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2557                    ctl3.Hide()
2558           
2559                    ix += 1
2560                    ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
2561                                               style=wx.TE_PROCESS_ENTER,
2562                                text_enter_callback = self._onparamRangeEnter)
2563                    sizer.Add(ctl4, (iy,ix),(1,1), 
2564                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2565         
2566                    ctl4.Hide()
2567   
2568                    if self.engine_type=="park":
2569                        ctl3.Show(True)
2570                        ctl4.Show(True)
2571                    ix +=1
2572                    # Units
2573                    if self.model.details.has_key(item):
2574                        units = wx.StaticText(self, -1, 
2575                            self.model.details[item][0], style=wx.ALIGN_LEFT)
2576                    else:
2577                        units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
2578                    sizer.Add(units, (iy,ix),(1,1), 
2579                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2580                       
2581                    ##[cb state, name, value, "+/-", error of fit, min, max , units]
2582                    self.parameters.append([cb,item, ctl1,
2583                                            text2,ctl2, ctl3, ctl4,units])
2584                                 
2585        iy+=1
2586        sizer.Add((10,10),(iy,ix),(1,1), 
2587                  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
2588       
2589        # type can be either Guassian or Array
2590        if len(self.model.dispersion.values())>0:
2591            type= self.model.dispersion.values()[0]["type"]
2592        else:
2593            type = "Gaussian"
2594           
2595        iy += 1
2596        ix = 0
2597        #Add tile for orientational angle
2598        for item in keys:
2599            if item in self.model.orientation_params:       
2600                orient_angle = wx.StaticText(self, -1, '[For 2D only]:')
2601                sizer.Add(orient_angle,(iy, ix),(1,1), 
2602                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
2603                if not self.data.__class__.__name__ =="Data2D":
2604                    orient_angle.Hide()
2605                else:
2606                    orient_angle.Show(True)
2607                break
2608     
2609        #For Gaussian only
2610        if type.lower() != "array":
2611            for item in self.model.orientation_params:
2612                if not item in self.disp_list:
2613                    ##prepare a spot to store min max
2614                    if not self.model.details.has_key(item):
2615                        self.model.details [item] = ["",None,None] 
2616                         
2617                    iy += 1
2618                    ix = 0
2619                    ## add parameters name with checkbox for selecting to fit
2620                    cb = wx.CheckBox(self, -1, item )
2621                    cb.SetValue(False)
2622                    cb.SetToolTipString("Check for fitting")
2623                    wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
2624                    if self.data.__class__.__name__ =="Data2D":
2625                        cb.Show(True)
2626                    else:
2627                        cb.Hide()
2628                    sizer.Add( cb,( iy, ix),(1,1),
2629                                 wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
2630   
2631                    ## add parameter value
2632                    ix += 1
2633                    value= self.model.getParam(item)
2634                    ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
2635                                        style=wx.TE_PROCESS_ENTER)
2636                    ctl1.SetToolTipString("Hit 'Enter' after typing.")
2637                    ctl1.SetValue(format_number(value))
2638                    if self.data.__class__.__name__ =="Data2D":
2639                        ctl1.Show(True)
2640                    else:
2641                        ctl1.Hide()
2642                    sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
2643                    ## text to show error sign
2644                    ix += 1
2645                    text2=wx.StaticText(self, -1, '+/-')
2646                    sizer.Add(text2,(iy, ix),(1,1),\
2647                                    wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2648                    text2.Hide() 
2649                    ix += 1
2650                    ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
2651                    sizer.Add(ctl2, (iy,ix),(1,1), 
2652                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2653                    ctl2.Hide()
2654                   
2655                   
2656                    ix += 1
2657                    ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), 
2658                                              style=wx.TE_PROCESS_ENTER,
2659                                text_enter_callback = self._onparamRangeEnter)
2660                   
2661                    sizer.Add(ctl3, (iy,ix),(1,1), 
2662                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2663                    ctl3.Hide()
2664                 
2665                    ix += 1
2666                    ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
2667                                               style=wx.TE_PROCESS_ENTER,
2668                            text_enter_callback = self._onparamRangeEnter)
2669                    sizer.Add(ctl4, (iy,ix),(1,1), 
2670                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2671                   
2672                    ctl4.Hide()
2673                   
2674                    if self.engine_type =="park" and \
2675                            self.data.__class__.__name__ =="Data2D":                     
2676                        ctl3.Show(True)
2677                        ctl4.Show(True)
2678                   
2679                    ix +=1
2680                    # Units
2681                    if self.model.details.has_key(item):
2682                        units = wx.StaticText(self, -1, 
2683                                              self.model.details[item][0], 
2684                                              style=wx.ALIGN_LEFT)
2685                    else:
2686                        units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
2687                    if self.data.__class__.__name__ =="Data2D":
2688                        units.Show(True)
2689                   
2690                    else:
2691                        units.Hide()
2692                   
2693                    sizer.Add(units, (iy,ix),(1,1), 
2694                              wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2695                                         
2696                    ##[cb state, name, value, "+/-", error of fit,min,max,units]
2697                    self.parameters.append([cb,item, ctl1,
2698                                            text2,ctl2, ctl3, ctl4,units])
2699                    self.orientation_params.append([cb,item, ctl1,
2700                                            text2,ctl2, ctl3, ctl4,units])
2701             
2702        iy+=1
2703       
2704        #Display units text on panel
2705        for item in keys:   
2706            if self.model.details.has_key(item):
2707                self.text2_4.Show()
2708        #Fill the list of fittable parameters
2709        self.select_all_param(event=None)
2710
2711        self.save_current_state_fit()
2712        boxsizer1.Add(sizer)
2713        self.sizer3.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
2714        self.sizer3.Layout()
2715        self.Layout()
2716        self.Refresh()
2717     
2718    def _onModel2D(self, event):
2719        """
2720        toggle view of model from 1D to 2D  or 2D from 1D
2721        """
2722        if self.model_view.GetLabelText() == "Switch to 2D":
2723            self.model_view.SetLabel("Switch to 1D")
2724            self.enable2D = True
2725        else:
2726            self.model_view.SetLabel("Switch to 2D")
2727            self.enable2D = False
2728        self._draw_model()
2729
2730class BGTextCtrl(wx.TextCtrl):
2731    """
2732    Text control used to display outputs.
2733    No editing allowed. The background is
2734    grayed out. User can't select text.
2735    """
2736    def __init__(self, *args, **kwds):
2737        wx.TextCtrl.__init__(self, *args, **kwds)
2738        self.SetEditable(False)
2739        self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
2740       
2741        # Bind to mouse event to avoid text highlighting
2742        # The event will be skipped once the call-back
2743        # is called.
2744        self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
2745       
2746    def _click(self, event):
2747        """
2748        Prevent further handling of the mouse event
2749        by not calling Skip().
2750        """ 
2751        pass
2752 
Note: See TracBrowser for help on using the repository browser.