source: sasview/fittingview/src/sans/perspectives/fitting/fitpage.py @ b7c6a4a

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 b7c6a4a was 62f851f, checked in by Gervaise Alina <gervyh@…>, 13 years ago

working on plot for batch

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