source: sasview/sansview/perspectives/fitting/fitpage.py @ 02775ab8

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 02775ab8 was f95301b, checked in by Gervaise Alina <gervyh@…>, 14 years ago

work on saving state

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