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

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 0b57a57 was 479eced, checked in by Jae Cho <jhjcho@…>, 14 years ago

minor change

  • Property mode set to 100644
File size: 94.7 KB
RevLine 
[d1e0473]1
2
[c77d859]3import sys
4import wx
5import wx.lib.newevent
6import numpy
7import copy
8import math
[7975f2b]9import time
[c77d859]10from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
[7609f1a]11from DataLoader.data_info import Data1D
[c77d859]12from sans.guicomm.events import StatusEvent   
[85b3971]13from sans.guiframe.utils import format_number,check_float
[51a71a3]14#from sans.guiframe.local_perspectives.plotting.Masking import ModelPanel2D as panel2D
[7609f1a]15
[b787e68c]16## event to post model to fit to fitting plugins
[c77d859]17(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
[b787e68c]18
19## event to know the selected fit engine
[77e23a2]20(FitterTypeEvent, EVT_FITTER_TYPE)   = wx.lib.newevent.NewEvent()
[34a0c17]21(FitStopEvent, EVT_FIT_STOP)   = wx.lib.newevent.NewEvent()
[eda428b]22_BOX_WIDTH = 76
[9237df4]23_DATA_BOX_WIDTH = 300
[7609f1a]24SMEAR_SIZE_L = 0.005
25SMEAR_SIZE_H = 0.01
26
[c77d859]27import basepage
28from basepage import BasicPage
[b787e68c]29from basepage import PageInfoEvent
[997131a]30from DataLoader.qsmearing import smear_selection
[c77d859]31
32class FitPage(BasicPage):
33    """
34        FitPanel class contains fields allowing to display results when
35        fitting  a model and one data
36        @note: For Fit to be performed the user should check at least one parameter
37        on fit Panel window.
38 
39    """
[7975f2b]40   
[cfc0913]41    def __init__(self,parent, page_info):
42        BasicPage.__init__(self, parent, page_info)
[7975f2b]43
[c77d859]44        """
45            Initialization of the Panel
46        """
[51a71a3]47        ## total number of point: float
[c77d859]48        self.npts=None
[2b63df0]49        ## thread for compute Chisqr
50        self.calc_Chisqr=None
[dcf29d7]51        ## default fitengine type
[707436d]52        self.engine_type = None
[7609f1a]53        ## smear default
54        self.smearer = None
55        self.current_smearer = None
56        self.dxl = None
57        self.dxw = None
58        self.dx_min = None
59        self.dx_max = None
[edd166b]60       
[b787e68c]61        ## draw sizer
[c77d859]62        self._fill_datainfo_sizer()
[7609f1a]63       
64        # get smear info from data
65        self._get_smear_info()
66       
[c77d859]67        self._fill_model_sizer( self.sizer1)
[7609f1a]68       
69        self._get_defult_custom_smear()
70       
[cfc0913]71        self._fill_range_sizer() 
[7609f1a]72       
[ffa69b6]73        if self.data is None:
74            self.formfactorbox.Disable()
75            self.structurebox.Disable()
76        else:
[7609f1a]77            self.smearer = smear_selection( self.data )
[847091f]78            if self.smearer ==None:
79                self.enable_smearer.Disable()
[51a71a3]80            if self.data.__class__.__name__ =="Data2D":
81                if self.model != None:
82                    self.smear_description_2d.Show(True)
[2012eae]83                   
[edd166b]84        self.disp_cb_dict = {}
[cfc0913]85        ## to update the panel according to the fit engine type selected
[77e23a2]86        self.Bind(EVT_FITTER_TYPE,self._on_engine_change)
[34a0c17]87        self.Bind(EVT_FIT_STOP,self._on_fit_complete)
[ffa69b6]88       
[c69b6d5]89
[34a0c17]90    def _on_fit_complete(self, event):
91        """
92            When fit is complete ,reset the fit button label.
93        """
94        #self.btFit.SetLabel("Fit")
95        #self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
96        #self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
97        pass
98       
99       
[77e23a2]100    def _on_engine_change(self, event):
101        """
102            get an event containing the current name of the fit engine type
103            @param event: FitterTypeEvent containing  the name of the current engine
104        """
[dcf29d7]105        self.engine_type = event.type
[77e23a2]106        if len(self.parameters)==0:
[edd166b]107            self.Layout()
[77e23a2]108            return
[ad6dd4c]109        if event.type =="park":
110            self.btFit.SetLabel("Fit")
[6d91073]111
[77e23a2]112        for item in self.parameters:
[920a6e5]113            if event.type =="scipy" :
[60132ef]114                item[5].SetValue("")
[77e23a2]115                item[5].Hide()
[60132ef]116                item[6].SetValue("")
[77e23a2]117                item[6].Hide()
118                self.text2_min.Hide()
119                self.text2_max.Hide()
[7975f2b]120
[77e23a2]121            else:
122                item[5].Show(True)
123                item[6].Show(True)
124                self.text2_min.Show(True)
125                self.text2_max.Show(True)
[920a6e5]126        for item in self.fittable_param:
127            if item[5]!=None and item[6]!=None and not item in self.orientation_params_disp:
128                if event.type =="scipy" and not item in self.orientation_params:
[c5cd3b9]129                    item[5].SetValue("")
130                    item[5].Hide()
131                    item[6].SetValue("")
132                    item[6].Hide()
133                    self.text2_min.Hide()
134                    self.text2_max.Hide()
[920a6e5]135                    self.text_disp_min.Hide()
136                    self.text_disp_max.Hide()
[c5cd3b9]137                else:
138                    item[5].Show(True)
139                    item[6].Show(True)
140                    self.text2_min.Show(True)
141                    self.text2_max.Show(True)
[920a6e5]142                    self.text_disp_min.Show(True)
143                    self.text_disp_max.Show(True)
[c5cd3b9]144           
[920a6e5]145        for item in self.orientation_params:
[c5cd3b9]146            if item[5]!=None and item[6]!=None:
147                if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
148                    item[5].SetValue("")
149                    item[5].Hide()
150                    item[6].SetValue("")
151                    item[6].Hide()
152                else:
153                    item[5].Show(True)
154                    item[6].Show(True)
[920a6e5]155                   
156        for item in self.orientation_params_disp:         
157            if item[5]!=None and item[6]!=None:
158                if event.type =="scipy" or self.data.__class__.__name__ !="Data2D":
159                    item[5].SetValue("")
160                    item[5].Hide()
161                    item[6].SetValue("")
162                    item[6].Hide()
163                else:
164                    item[5].Show(True)
165                    item[6].Show(True)
[6d91073]166        self.Layout()
[7975f2b]167        self.Refresh()
168
[c77d859]169    def _fill_range_sizer(self):
170        """
171            Fill the sizer containing the plotting range
172            add  access to npts
173        """
[51a71a3]174        is_2Ddata = False
175       
176        # Check if data is 2D
[6318298]177        if self.data.__class__.__name__ == 'Data2D':
[51a71a3]178            is_2Ddata = True
179           
[ff8f99b]180        title = "Fitting"
[7609f1a]181        #smear messages & titles
182        smear_message_none  =  "No smearing is selected..."
183        smear_message_dqdata  =  "The dQ data is being used for smearing..."
184        smear_message_2d  =  "No smearing is supported for 2D model fitting..."
185        smear_message_new_ssmear  =  "Please enter only the value of interest to customize smearing..."
186        smear_message_new_psmear  =  "Please enter both; the dQ will be generated by interpolation..."
187        smear_message_pinhole_min_title = "dQ_min[1/A]:"
188        smear_message_pinhole_max_title = "dQ_max[1/A]:"
189        smear_message_slit_height_title = "Slit height[1/A]:"
190        smear_message_slit_width_title = "Slit width[1/A]:"
191       
192        self._get_smear_info()
[c77d859]193       
[51a71a3]194        #Sizers
195        box_description_range = wx.StaticBox(self, -1,str(title))
196        boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)     
[7609f1a]197        self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
[c77d859]198        sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
[7609f1a]199        self.sizer_new_smear= wx.BoxSizer(wx.HORIZONTAL)
200        self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
201        sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
202        smear_set_box= wx.StaticBox(self, -1,'Set Instrumental Smearing')
203        sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
204        sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH,85))
[2012eae]205        sizer_fit = wx.GridSizer(2, 4,2,6)
206       
207        #Fit button
208        self.btFit = wx.Button(self,wx.NewId(),'Fit', size=(88,25))
209        self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
210        self.btFit.SetToolTipString("Start fitting.")
[7609f1a]211       
212        #textcntrl for new resolution
213        self.smear_pinhole_max = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-20,20),style=wx.TE_PROCESS_ENTER,
214                                            text_enter_callback = self.onPinholeSmear)
215        self.smear_pinhole_min = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-20,20),style=wx.TE_PROCESS_ENTER,
216                                            text_enter_callback = self.onPinholeSmear)
217        self.smear_slit_height= self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-20,20),style=wx.TE_PROCESS_ENTER,
218                                            text_enter_callback = self.onSlitSmear)
219        self.smear_slit_width = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-20,20),style=wx.TE_PROCESS_ENTER,
220                                            text_enter_callback = self.onSlitSmear)
221        self.smear_data_left= BGTextCtrl(self, -1, size=(_BOX_WIDTH-20,20), style=0)
222        self.smear_data_left.SetValue(str(self.dq_l))
223        self.smear_data_right = BGTextCtrl(self, -1, size=(_BOX_WIDTH-20,20), style=0)
224        self.smear_data_right.SetValue(str(self.dq_r))
225
226        #set default values
227        self.smear_pinhole_max.SetValue(str(self.dx_max))
228        self.smear_pinhole_min.SetValue(str(self.dx_min))
229        self.smear_slit_height.SetValue(str(self.dxl))
230        self.smear_slit_width.SetValue(str(self.dxw))
231
[c77d859]232        #Filling the sizer containing instruments smearing info.
[7609f1a]233        self.disable_smearer = wx.RadioButton(self, -1, 'None', style=wx.RB_GROUP)
234        self.enable_smearer = wx.RadioButton(self, -1, 'Use dQ Data')
235        #self.enable_smearer.SetToolTipString("Click to use the loaded dQ data for smearing.")
236        self.pinhole_smearer = wx.RadioButton(self, -1, 'Custom Pinhole Smear')
237        #self.pinhole_smearer.SetToolTipString("Click to input custom resolution for pinhole smearing.")
238        self.slit_smearer = wx.RadioButton(self, -1, 'Custom Slit Smear')
239        #self.slit_smearer.SetToolTipString("Click to input custom resolution for slit smearing.")
[c77d859]240        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
241        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
[7609f1a]242        self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear, id=self.pinhole_smearer.GetId())
243        self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear, id=self.slit_smearer.GetId())
[ff8f99b]244        self.disable_smearer.SetValue(True)
[c77d859]245       
[7609f1a]246        # add 4 types of smearing to the sizer
247        sizer_smearer.Add( self.disable_smearer,0, wx.LEFT, 10)
[c77d859]248        sizer_smearer.Add((10,10))
[7609f1a]249        sizer_smearer.Add( self.enable_smearer)
250        sizer_smearer.Add((10,10))
251        sizer_smearer.Add( self.pinhole_smearer ) 
252        sizer_smearer.Add((10,10))
253        sizer_smearer.Add( self.slit_smearer ) 
254        sizer_smearer.Add((10,10))       
255       
[51a71a3]256        # StaticText for chi2, N(for fitting), Npts
257        self.tcChi    =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
258        self.tcChi.SetToolTipString("Chi2/Npts")
259        self.Npts_fit    =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
260        self.Npts_fit.SetToolTipString(" Npts : number of points selected for fitting")
261        self.Npts_total  =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
262        self.Npts_total.SetToolTipString(" Total Npts : total number of data points")
263        box_description_1= wx.StaticText(self, -1,'    Chi2/Npts')
264        box_description_2= wx.StaticText(self, -1,'  Npts')
265        box_description_3= wx.StaticText(self, -1,'  Total Npts')
266        box_description_4= wx.StaticText(self, -1,' ')
[7609f1a]267       
[2012eae]268       
[51a71a3]269        sizer_fit.Add(box_description_1,0,0)
270        sizer_fit.Add(box_description_2,0,0)
271        sizer_fit.Add(box_description_3,0,0)       
272        sizer_fit.Add(box_description_4,0,0)
273        sizer_fit.Add(self.tcChi,0,0)
274        sizer_fit.Add(self.Npts_fit ,0,0)
275        sizer_fit.Add(self.Npts_total,0,0)
276        sizer_fit.Add(self.btFit,0,0) 
277
278        # StaticText for smear
279        #self.tcChi    =  wx.StaticText(self, -1, "-", style=wx.ALIGN_LEFT)
[7609f1a]280        self.smear_description_none    =  wx.StaticText(self, -1, smear_message_none , style=wx.ALIGN_LEFT)
281        self.smear_description_dqdata    =  wx.StaticText(self, -1, smear_message_dqdata , style=wx.ALIGN_LEFT)
282        self.smear_description_type    =  wx.StaticText(self, -1, "Type:" , style=wx.ALIGN_LEFT)
283        self.smear_description_smear_type    =  BGTextCtrl(self, -1, size=(45,20), style=0)
284        self.smear_description_smear_type.SetValue(str(self.dq_l))
285        self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
286        self.smear_description_2d     =  wx.StaticText(self, -1, smear_message_2d  , style=wx.ALIGN_LEFT)
287        self.smear_message_new_s = wx.StaticText(self, -1, smear_message_new_ssmear  , style=wx.ALIGN_LEFT)
288        self.smear_message_new_p = wx.StaticText(self, -1, smear_message_new_psmear  , style=wx.ALIGN_LEFT)
289        self.smear_description_pin_min     =  wx.StaticText(self, -1, smear_message_pinhole_min_title  , style=wx.ALIGN_LEFT)
290        self.smear_description_pin_max     =  wx.StaticText(self, -1, smear_message_pinhole_max_title  , style=wx.ALIGN_LEFT)
291        self.smear_description_slit_height    =  wx.StaticText(self, -1, smear_message_slit_height_title   , style=wx.ALIGN_LEFT)
292        self.smear_description_slit_width    =  wx.StaticText(self, -1, smear_message_slit_width_title   , style=wx.ALIGN_LEFT)
293       
294        #arrange sizers
[51a71a3]295        #boxsizer1.Add( self.tcChi ) 
[7609f1a]296        self.sizer_set_smearer.Add(sizer_smearer )
297        self.sizer_set_smearer.Add((10,10))
298        self.sizer_set_smearer.Add( self.smear_description_none,0, wx.CENTER, 10 ) 
299        self.sizer_set_smearer.Add( self.smear_description_dqdata,0, wx.CENTER, 10 )
300        self.sizer_set_smearer.Add( self.smear_description_2d,0, wx.CENTER, 10 )
301        self.sizer_new_smear.Add( self.smear_description_type,0, wx.CENTER, 10 )
302        self.sizer_new_smear.Add( self.smear_description_smear_type,0, wx.CENTER, 10 )
303        self.sizer_new_smear.Add((15,-1))
304        self.sizer_new_smear.Add( self.smear_description_pin_min,0, wx.CENTER, 10 )
305        self.sizer_new_smear.Add( self.smear_description_slit_height,0, wx.CENTER, 10 )
306        self.sizer_new_smear.Add( self.smear_pinhole_min,0, wx.CENTER, 10 )
307        self.sizer_new_smear.Add( self.smear_slit_height,0, wx.CENTER, 10 )
308        self.sizer_new_smear.Add( self.smear_data_left,0, wx.CENTER, 10 )
309        self.sizer_new_smear.Add((20,-1))
310        self.sizer_new_smear.Add( self.smear_description_pin_max,0, wx.CENTER, 10 )
311        self.sizer_new_smear.Add( self.smear_description_slit_width,0, wx.CENTER, 10 )
312        self.sizer_new_smear.Add( self.smear_pinhole_max,0, wx.CENTER, 10 )
313        self.sizer_new_smear.Add( self.smear_slit_width,0, wx.CENTER, 10 )
314        self.sizer_new_smear.Add( self.smear_data_right,0, wx.CENTER, 10 )
315           
316        self.sizer_set_smearer.Add( self.smear_message_new_s,0, wx.CENTER, 10)
317        self.sizer_set_smearer.Add( self.smear_message_new_p,0, wx.CENTER, 10)
318        self.sizer_set_smearer.Add((5,2))
319        self.sizer_set_smearer.Add( self.sizer_new_smear,0, wx.CENTER, 10 )
320       
[51a71a3]321        # add all to chi2 sizer
322        sizer_smearer_box.Add(self.sizer_set_smearer)       
[7609f1a]323        sizer_chi2.Add(sizer_smearer_box)
324        sizer_chi2.Add((-1,5))
[51a71a3]325
[7609f1a]326        # hide all smear messages and textctrl
327        self._hide_all_smear_info()
328       
329        # get smear_selection
330        self.current_smearer= smear_selection( self.data )
[83ad478]331
[7609f1a]332        # Show only the relevant smear messages, etc
333        if self.current_smearer == None:
[51a71a3]334            if not is_2Ddata:
[7609f1a]335                self.smear_description_none.Show(True)
336                self.enable_smearer.Disable() 
337            else:
338                self.smear_description_2d.Show(True) 
339                self.pinhole_smearer.Disable() 
340                self.slit_smearer.Disable()   
341                self.enable_smearer.Disable() 
342        else: self._show_smear_sizer()
[51a71a3]343        boxsizer_range.Add(self.sizer_set_masking)
[7609f1a]344           
345        #Set sizer for Fitting section
346        self._set_range_sizer( title=title,box_sizer=boxsizer_range, object1=sizer_chi2, object= sizer_fit)
347       
[c77d859]348    def _fill_datainfo_sizer(self):
349        """
350            fill sizer 0 with data info
351        """
352        ## no loaded data , don't fill the sizer
[9237df4]353        if self.data is None:
354            data_min = ""
355            data_max = ""
356            data_name = ""
357        else:
358            data_name = self.data.name
359            #set maximum range for x in linear scale
360            if not hasattr(self.data,"data"): #Display only for 1D data fit
361                # Minimum value of data   
362                data_min = min(self.data.x)
363                # Maximum value of data 
364                data_max = max(self.data.x)
365            else:
366                ## Minimum value of data
367                data_min = 0
368                x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax)) 
369                y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
370                ## Maximum value of data 
371                data_max = math.sqrt(x*x + y*y)
[c77d859]372       
[fa58441]373        box_description= wx.StaticBox(self, -1, 'Data')
[c77d859]374        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
375        #----------------------------------------------------------
[9237df4]376        sizer_data = wx.BoxSizer(wx.HORIZONTAL)
[c77d859]377        #Filling the sizer containing data related fields
[9237df4]378        #self.dataSource = wx.StaticText(self, -1,str(self.data.name))
[7609f1a]379        self.dataSource = BGTextCtrl(self, -1)
[9237df4]380        self.dataSource.SetValue(str(data_name))
[7609f1a]381        #self.dataSource.SetEditable(False)
[9237df4]382        self.dataSource.SetMinSize((_DATA_BOX_WIDTH, -1))
[fa58441]383        sizer_data.Add(wx.StaticText(self, -1, 'Source Name : '))
[9237df4]384        sizer_data.Add(self.dataSource )
[2a5bba7]385        sizer_data.Add( (0,5) )
[c77d859]386       
387        #---------sizer 2 draw--------------------------------
[9237df4]388        text4_3 = wx.StaticText(self, -1, 'Total Q Range (1/A)',
389                                 style=wx.ALIGN_LEFT)
390        sizer_range = wx.BoxSizer(wx.HORIZONTAL)
391        sizer_range.Add( text4_3 ,0, wx.RIGHT, 10)
[7609f1a]392        self.minimum_q = BGTextCtrl(self, -1, size=(_BOX_WIDTH,20))
[9237df4]393        self.minimum_q.SetValue(str(data_min))
[7609f1a]394        #self.minimum_q.SetEditable(False)
395        self.maximum_q = BGTextCtrl(self, -1,size=(_BOX_WIDTH,20))
[9237df4]396        self.maximum_q.SetValue(str(data_max))
[7609f1a]397        #self.maximum_q.SetEditable(False)
[9237df4]398        sizer_range.Add(wx.StaticText(self, -1, "Min: "),0, wx.LEFT, 10)
399        sizer_range.Add(self.minimum_q,0, wx.LEFT, 10)
400        sizer_range.Add(wx.StaticText(self, -1, "Max: "),0, wx.LEFT, 10)
401        sizer_range.Add(self.maximum_q,0, wx.LEFT, 10)
402
[813334e]403        ## set q range to plot
[9237df4]404        self.qmin_x = data_min
405        self.qmax_x = data_max
[813334e]406
[9237df4]407        boxsizer1.Add(sizer_data,0, wx.ALL, 10)
408        boxsizer1.Add(sizer_range, 0 , wx.LEFT, 10)
[c77d859]409        #------------------------------------------------------------
410        self.sizer0.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
411        self.sizer0.Layout()
[813334e]412       
[c77d859]413    def _fill_model_sizer(self, sizer):
414        """
415            fill sizer containing model info
416        """
[7ad6ff5]417        ##Add model function Details button in fitpanel.
[ff8f99b]418        ##The following 3 lines are for Mac. Let JHC know before modifying...
419        title = "Model"
420        box_description= wx.StaticBox(self, -1,str(title))
421        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
422         
[7ad6ff5]423        id = wx.NewId()
[e3a76c8]424        self.model_help =wx.Button(self,id,'Details', size=(80,23))
[7ad6ff5]425        self.model_help.Bind(wx.EVT_BUTTON, self.on_model_help_clicked,id=id)
426        self.model_help.SetToolTipString("Model Function Help")
[7b0fd65]427       
[7ad6ff5]428        ## class base method  to add view 2d button   
[ff8f99b]429        self._set_model_sizer(sizer=sizer, box_sizer=boxsizer1, title="Model",object=self.model_help )   
[7ad6ff5]430
[c77d859]431   
[30d103a]432    def _set_sizer_dispersion(self, dispersity):
[c77d859]433        """
434            draw sizer with gaussian dispersity parameters
435        """
436        self.fittable_param=[]
437        self.fixed_param=[]
[60132ef]438        self.orientation_params_disp=[]
[920a6e5]439
[c77d859]440        self.sizer4_4.Clear(True)
441        if self.model==None:
442            ##no model is selected
443            return
444        if not self.enable_disp.GetValue():
445            ## the user didn't select dispersity display
446            return 
[b421b1a]447           
[376916c]448        self._reset_dispersity()
[b421b1a]449       
[376916c]450        # Create the dispersion objects
451        for item in self.model.dispersion.keys():
[30d103a]452            #disp_model =  GaussianDispersion()
453            disp_model = dispersity()
[376916c]454            self._disp_obj_dict[item] = disp_model
455            self.model.set_dispersion(item, disp_model)
[c477b31]456            self.state._disp_obj_dict[item]= disp_model
[376916c]457
[b421b1a]458
[c77d859]459        ix=0
460        iy=1
[52efcc5]461        disp = wx.StaticText(self, -1, ' ')
[c77d859]462        self.sizer4_4.Add(disp,( iy, ix),(1,1), 
463                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
464        ix += 1 
[52efcc5]465        values = wx.StaticText(self, -1, 'Sigma (STD)')
[c77d859]466        self.sizer4_4.Add(values,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
467        ix +=2 
[920a6e5]468        self.text_disp_1 = wx.StaticText(self, -1, '')
[c77d859]469        self.sizer4_4.Add( self.text_disp_1,(iy, ix),(1,1),\
470                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
471        self.text_disp_1.Hide()
[920a6e5]472       
473       
474        ix +=1 
475        self.text_disp_min = wx.StaticText(self, -1, 'Min')
476        self.sizer4_4.Add(self.text_disp_min,(iy, ix),(1,1),\
477                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
478        self.text_disp_min.Hide()
479        ix +=1 
480        self.text_disp_max = wx.StaticText(self, -1, 'Max')
481        self.sizer4_4.Add(self.text_disp_max,(iy, ix),(1,1),\
482                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
483        self.text_disp_max.Hide()
484                       
485       
[c77d859]486        ix += 1 
487        npts = wx.StaticText(self, -1, 'Npts')
488        self.sizer4_4.Add(npts,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
489        ix += 1 
490        nsigmas = wx.StaticText(self, -1, 'Nsigmas')
491        self.sizer4_4.Add(nsigmas,( iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[920a6e5]492       
493        if self.engine_type=="park":
494            self.text_disp_max.Show(True)
495            self.text_disp_min.Show(True)
[6dc9ad8]496
[c77d859]497        for item in self.model.dispersion.keys():
[780d095]498            if not item in self.model.orientation_params:
[edd166b]499                if not self.disp_cb_dict.has_key(item):
500                    self.disp_cb_dict[item]= None
[780d095]501                name1=item+".width"
502                name2=item+".npts"
503                name3=item+".nsigmas"
[c99a6c5]504                if not self.model.details.has_key(name1):
[b421b1a]505                    self.model.details [name1] = ["",None,None] 
[c99a6c5]506
[780d095]507                iy += 1
508                for p in self.model.dispersion[item].keys(): 
509       
510                    if p=="width":
511                        ix = 0
512                        cb = wx.CheckBox(self, -1, name1, (10, 10))
513                        wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
514                        self.sizer4_4.Add( cb,( iy, ix),(1,1), 
515                                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
516                        ix = 1
517                        value= self.model.getParam(name1)
[7975f2b]518                        ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[c77d859]519                                            style=wx.TE_PROCESS_ENTER)
[780d095]520                        ctl1.SetValue(str (format_number(value)))
521                        self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
522                        ## text to show error sign
523                        ix = 2
524                        text2=wx.StaticText(self, -1, '+/-')
525                        self.sizer4_4.Add(text2,(iy, ix),(1,1),
526                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
527                        text2.Hide() 
[b421b1a]528
[780d095]529                        ix = 3
[c13b8cc]530                        ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[eacf1d66]531                 
[780d095]532                        self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[eacf1d66]533                     
[b421b1a]534                        ctl2.Hide()
[eacf1d66]535
[920a6e5]536                        ix = 4
[7975f2b]537                        ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
538                                                       text_enter_callback = self._onparamRangeEnter)
[eacf1d66]539
[920a6e5]540                        self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
541                        ctl3.Hide()
542               
543                        ix = 5
[7975f2b]544                        ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
545                                                       text_enter_callback = self._onparamRangeEnter)
[920a6e5]546                        self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[eacf1d66]547
[920a6e5]548                        ctl4.Hide()
549                       
550                        if self.engine_type=="park":
551                            ctl3.Show(True)
552                            ctl4.Show(True)
553                                         
[780d095]554                        self.fittable_param.append([cb,name1,ctl1,text2,
[c99a6c5]555                                                    ctl2, ctl3, ctl4,None])                   
[920a6e5]556                   
[780d095]557                    elif p=="npts":
[920a6e5]558                            ix = 6
[780d095]559                            value= self.model.getParam(name2)
[7975f2b]560                            Tctl = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]561                                                style=wx.TE_PROCESS_ENTER)
562                           
563                            Tctl.SetValue(str (format_number(value)))
564                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
565                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
566                            self.fixed_param.append([None,name2, Tctl,None,None,
567                                                      None, None,None])
568                    elif p=="nsigmas":
[920a6e5]569                            ix = 7
[780d095]570                            value= self.model.getParam(name3)
[7975f2b]571                            Tct2 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]572                                                style=wx.TE_PROCESS_ENTER)
[c985bef]573                            Tct2.SetValue(str (format_number(value)))
574                            self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
[780d095]575                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
576                            ix +=1
577                            self.sizer4_4.Add((20,20), (iy,ix),(1,1),
578                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
579                           
[c985bef]580                            self.fixed_param.append([None,name3, Tct2
[920a6e5]581                                                     ,None,None,None, None,None])
582                           
[780d095]583        ix =0
584        iy +=1 
585        self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
586        for item in self.model.dispersion.keys():
587            if  item in self.model.orientation_params:
[edd166b]588                if not self.disp_cb_dict.has_key(item):
589                    self.disp_cb_dict[item]= None
[780d095]590                name1=item+".width"
591                name2=item+".npts"
592                name3=item+".nsigmas"
[c99a6c5]593                if not self.model.details.has_key(name1):
[b421b1a]594                    self.model.details [name1] = ["",None,None]                 
595 
596
[780d095]597                iy += 1
598                for p in self.model.dispersion[item].keys(): 
599       
600                    if p=="width":
601                        ix = 0
602                        cb = wx.CheckBox(self, -1, name1, (10, 10))
603                        wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
604                        self.sizer4_4.Add( cb,( iy, ix),(1,1), 
605                                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
606                        if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]607                            cb.Show(True)
[c99a6c5]608                        elif cb.IsShown():
[6dc9ad8]609                            cb.Hide()
[780d095]610                        ix = 1
611                        value= self.model.getParam(name1)
[7975f2b]612                        ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[c77d859]613                                            style=wx.TE_PROCESS_ENTER)
[780d095]614                        ctl1.SetValue(str (format_number(value)))
615                        if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]616                            ctl1.Show(True)
[c99a6c5]617                        elif ctl1.IsShown():
[6dc9ad8]618                            ctl1.Hide()
[780d095]619                        self.sizer4_4.Add(ctl1, (iy,ix),(1,1),wx.EXPAND)
620                        ## text to show error sign
621                        ix = 2
622                        text2=wx.StaticText(self, -1, '+/-')
623                        self.sizer4_4.Add(text2,(iy, ix),(1,1),
624                                          wx.EXPAND|wx.ADJUST_MINSIZE, 0)
625                        text2.Hide() 
[b421b1a]626
[780d095]627                        ix = 3
[c13b8cc]628                        ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[b421b1a]629                   
[780d095]630                        self.sizer4_4.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[b421b1a]631                        ctl2.Hide()
[920a6e5]632                           
633                        ix = 4
[7975f2b]634                        ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
635                                                       text_enter_callback = self._onparamRangeEnter)
[eacf1d66]636
[920a6e5]637                        self.sizer4_4.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[c99a6c5]638
[920a6e5]639                        ctl3.Hide()
640               
641                        ix = 5
[7975f2b]642                        ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
643                                                       text_enter_callback = self._onparamRangeEnter)
[920a6e5]644                        self.sizer4_4.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
645                        ctl4.Hide()
[c99a6c5]646                        #if self.data.__class__.__name__ =="Data2D":
647                            #ctl4.Enable(True)
648                        #elif ctl4.Shown():
649                            #ctl4.Hide()
[920a6e5]650                       
651                        if self.engine_type=="park" and self.data.__class__.__name__ =="Data2D":
652                            ctl3.Show(True)
[c99a6c5]653                            ctl4.Show(True) 
[920a6e5]654                           
655                           
656                           
657                           
[780d095]658                        self.fittable_param.append([cb,name1,ctl1,text2,
[920a6e5]659                                                    ctl2, ctl3, ctl4,None])
[60132ef]660                        self.orientation_params_disp.append([cb,name1,ctl1,text2,
[920a6e5]661                                                    ctl2, ctl3, ctl4,None])
[780d095]662                    elif p=="npts":
[920a6e5]663                            ix = 6
[780d095]664                            value= self.model.getParam(name2)
[7975f2b]665                            Tctl = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]666                                                style=wx.TE_PROCESS_ENTER)
667                           
668                            Tctl.SetValue(str (format_number(value)))
669                            if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]670                                Tctl.Show(True)
[780d095]671                            else:
[6dc9ad8]672                                Tctl.Hide()
[780d095]673                            self.sizer4_4.Add(Tctl, (iy,ix),(1,1),
674                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
675                            self.fixed_param.append([None,name2, Tctl,None,None,
676                                                      None, None,None])
[60132ef]677                            self.orientation_params_disp.append([None,name2, Tctl,None,None,
[780d095]678                                                      None, None,None])
679                    elif p=="nsigmas":
[920a6e5]680                            ix = 7
[780d095]681                            value= self.model.getParam(name3)
[7975f2b]682                            Tct2 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20),
[780d095]683                                                style=wx.TE_PROCESS_ENTER)
[c985bef]684                            Tct2.SetValue(str (format_number(value)))
[780d095]685                            if self.data.__class__.__name__ =="Data2D":
[c985bef]686                                Tct2.Show(True)
[780d095]687                            else:
[c985bef]688                                Tct2.Hide()
689                            self.sizer4_4.Add(Tct2, (iy,ix),(1,1),
[780d095]690                                               wx.EXPAND|wx.ADJUST_MINSIZE, 0)
691                            ix +=1
[7975f2b]692
[c985bef]693                            self.fixed_param.append([None,name3, Tct2
[780d095]694                                                     ,None,None, None, None,None])   
[920a6e5]695                                                       
[c985bef]696                            self.orientation_params_disp.append([None,name3, Tct2
[1c1436d]697                                                     ,None,None, None, None,None])
[b324aa9]698       
699
[b421b1a]700        self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict) 
701         
[71f0373]702        self.state.model = self.model.clone() 
[240b9966]703         ## save state into
[4043c96]704        self.state.cb1 = self.cb1.GetValue()
[920a6e5]705        self._copy_parameters_state(self.parameters, self.state.parameters)
[240b9966]706        self._copy_parameters_state(self.orientation_params_disp,
707                                     self.state.orientation_params_disp)
708        self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
709        self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
[d40a9cd]710
711
[c77d859]712        wx.PostEvent(self.parent, StatusEvent(status=\
713                        " Selected Distribution: Gaussian"))   
[b324aa9]714        #Fill the list of fittable parameters
715        self.select_all_param(event=None)
716       
[71f0373]717        self.Layout()
[7975f2b]718
719
[c77d859]720    def _onFit(self, event):     
721        """
722            Allow to fit
723        """
[c13b8cc]724        #make sure all parameter values are updated.
[ffa69b6]725        if self.check_invalid_panel():
726            return
[85b3971]727        flag = self._update_paramv_on_fit() 
[acd0bda3]728               
[c77d859]729        if not flag:
[7609f1a]730            msg= "Fitting range or parameters are invalid"
[c77d859]731            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
732            return 
733       
734        if len(self.param_toFit) <= 0:
735            msg= "Select at least one parameter to fit"
736            wx.PostEvent(self.parent.parent, StatusEvent(status= msg ))
737            return 
[7975f2b]738     
739
740        self.select_param(event =None)
[acd0bda3]741       
[c99a6c5]742        #Clear errors if exist from previous fitting
[edd166b]743        #self._clear_Err_on_Fit()
[d0ce2c30]744
[c13b8cc]745        # Remove or do not allow fitting on the Q=0 point, especially when y(q=0)=None at x[0].         
[1c1436d]746        self.qmin_x = float(self.qmin.GetValue())
[c13b8cc]747        self.qmax_x = float( self.qmax.GetValue())
[6f023e8]748        self.manager._reset_schedule_problem( value=0)
[ca7a626]749        self.manager.schedule_for_fit( value=1,page=self,fitproblem =None) 
750        self.manager.set_fit_range(page= self,qmin= self.qmin_x, qmax= self.qmax_x)
[ad6dd4c]751       
[c77d859]752        #single fit
[ca7a626]753        self.manager.onFit()
[ad6dd4c]754        ## allow stopping the fit
[34a0c17]755        #if self.engine_type=="scipy":
756        #    self.btFit.SetLabel("Stop")
757        #    self.btFit.Unbind(event=wx.EVT_BUTTON, id= self.btFit.GetId())
758        #    self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._StopFit, id=self.btFit.GetId())
759        #else:
760        #    self.btFit.SetLabel("Fit")
761        #    self.btFit.Bind(event= wx.EVT_BUTTON, handler=self._onFit, id=self.btFit.GetId())
[7975f2b]762           
[5f9a6b1]763
[c77d859]764       
[ad6dd4c]765    def _StopFit(self, event):
766        """
767            Stop fit
768        """
769        self.btFit.SetLabel("Fit")
770        if self.engine_type=="scipy":
771            self.manager.stop_fit()
[34a0c17]772        self.btFit.Unbind(event=wx.EVT_BUTTON, id=self.btFit.GetId())
773        self.btFit.Bind(event=wx.EVT_BUTTON, handler=self._onFit,id=self.btFit.GetId())
[7975f2b]774       
[ad6dd4c]775           
[c77d859]776    def _on_select_model(self, event): 
777        """
778             call back for model selection
[6747217]779        """ 
[59a7f2d]780        self._on_select_model_helper() 
[70c57ebf]781        self.set_model_param_sizer(self.model)                   
[b421b1a]782       
[3b605bb]783        self.enable_disp.SetValue(False)
784        self.disable_disp.SetValue(True)
[4043c96]785        try:
786            self.set_dispers_sizer()
787        except:
788            pass
[6747217]789        self.btFit.SetFocus() 
790        self.state.enable_disp = self.enable_disp.GetValue()
791        self.state.disable_disp = self.disable_disp.GetValue()
792        self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
793        self.state.slit_smearer = self.slit_smearer.GetValue()
794   
795        self.state.structurecombobox = self.structurebox.GetCurrentSelection()
796        self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
797     
[9237df4]798        if self.model != None:
[70c57ebf]799            try:
800                temp_smear= None
801                if self.enable_smearer.GetValue():
802                    temp_smear= self.smearer
803                self.compute_chisqr(temp_smear)
804            except:
805                ## error occured on chisqr computation
806                pass
[fcd8887d]807            if self.data is not None and self.data.__class__.__name__ !="Data2D":
[9237df4]808                ## set smearing value whether or not the data contain the smearing info
[6747217]809                evt = ModelEventbox(model=self.model, 
810                                        smearer=temp_smear, 
811                                        qmin= float(self.qmin_x),
[2a2af47]812                                     qmax= float(self.qmax_x)) 
[6747217]813            else:   
814                 evt = ModelEventbox(model=self.model)
[c99a6c5]815            wx.PostEvent(self.event_owner, evt) 
[4043c96]816           
[3595309d]817        if event !=None:
[3a37fe0]818            #self._undo.Enable(True)
[3595309d]819            ## post state to fit panel
820            event = PageInfoEvent(page = self)
821            wx.PostEvent(self.parent, event) 
[330573d]822     
[e2f7b92]823               
[dd5949d]824    def _onparamEnter(self,event):
825        """
826            when enter value on panel redraw model according to changed
827        """
[51a71a3]828        #default flag
[7975f2b]829        flag = False
[51a71a3]830        self.fitrange = True
831        #get event object
[69bee6d]832        tcrtl= event.GetEventObject()
[7975f2b]833
[edd166b]834        #Clear msg if previously shown.
835        msg= ""
836        wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[c99a6c5]837
[7975f2b]838        if check_float(tcrtl):
[2012eae]839            flag = self._onparamEnter_helper() 
840            self.set_npts2fit() 
[51a71a3]841            if self.fitrange:             
842                temp_smearer = None
843                if not self.disable_smearer.GetValue():
844                    temp_smearer= self.current_smearer
845                    ## set smearing value whether or not the data contain the smearing info
846                    if self.slit_smearer.GetValue():
847                        flag1 = self.update_slit_smear()
848                        flag = flag or flag1
849                    elif self.pinhole_smearer.GetValue():
850                        flag1 = self.update_pinhole_smear()
851                        flag = flag or flag1
852                elif self.data.__class__.__name__ !="Data2D":
853                    self.manager.set_smearer(smearer=temp_smearer, qmin= float(self.qmin_x),
854                                                             qmax= float(self.qmax_x)) 
855                if flag:   
856                    self.compute_chisqr(smearer= temp_smearer)
857       
858                    ## new state posted
859                    if self.state_change:
860                        #self._undo.Enable(True)
861                        event = PageInfoEvent(page = self)
862                        wx.PostEvent(self.parent, event)
863                    self.state_change= False 
864            else: 
865                return    # invalid fit range: do nothing here: msg already displayed in validate   
[69bee6d]866        else:
[7975f2b]867            self.save_current_state()
[69bee6d]868            msg= "Cannot Plot :Must enter a number!!!  "
869            wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[7609f1a]870             
871        self.save_current_state() 
872        return 
873   
[e2f7b92]874    def _onparamRangeEnter(self, event):
[920a6e5]875        """
876            Check validity of value enter in the parameters range field
877        """
[ffa69b6]878        if self.check_invalid_panel():
879            return
[920a6e5]880        tcrtl= event.GetEventObject()
[edd166b]881        #Clear msg if previously shown.
882        msg= ""
883        wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
884        # Flag to register when a parameter has changed.
885        is_modified = False
[920a6e5]886        if tcrtl.GetValue().lstrip().rstrip()!="":
887            try:
888                value = float(tcrtl.GetValue())
889                tcrtl.SetBackgroundColour(wx.WHITE)
[edd166b]890                self._check_value_enter(self.fittable_param ,is_modified)
891                self._check_value_enter(self.parameters ,is_modified) 
[920a6e5]892            except:
893                tcrtl.SetBackgroundColour("pink")
[edd166b]894                msg= "Model Error:wrong value entered : %s"% sys.exc_value
895                wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
896                return 
[920a6e5]897        else:
898           tcrtl.SetBackgroundColour(wx.WHITE)
[e2f7b92]899           
[edd166b]900        #self._undo.Enable(True)
901        self.save_current_state()
902        event = PageInfoEvent(page = self)
903        wx.PostEvent(self.parent, event)
904        self.state_change= False
[c69b6d5]905
[12eac73]906                   
907    def _onQrangeEnter(self, event):
908        """
909            Check validity of value enter in the Q range field
910        """
[ffa69b6]911        if self.check_invalid_panel():
912            return
[12eac73]913        tcrtl= event.GetEventObject()
914        #Clear msg if previously shown.
915        msg= ""
916        wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
917        # Flag to register when a parameter has changed.
918        is_modified = False
919        if tcrtl.GetValue().lstrip().rstrip()!="":
920            try:
921                value = float(tcrtl.GetValue())
922                tcrtl.SetBackgroundColour(wx.WHITE)
923
924                # If qmin and qmax have been modified, update qmin and qmax
[85b3971]925                if self._validate_qrange( self.qmin, self.qmax):
[12eac73]926                    tempmin = float(self.qmin.GetValue())
927                    if tempmin != self.qmin_x:
928                        self.qmin_x = tempmin
929                    tempmax = float(self.qmax.GetValue())
930                    if tempmax != self.qmax_x:
931                        self.qmax_x = tempmax
932                else:
933                    tcrtl.SetBackgroundColour("pink")
934                    msg= "Model Error:wrong value entered : %s"% sys.exc_value
935                    wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
936                    return 
937               
938            except:
939                tcrtl.SetBackgroundColour("pink")
940                msg= "Model Error:wrong value entered : %s"% sys.exc_value
941                wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
942                return 
943            #Check if # of points for theory model are valid(>0).
[51a71a3]944            # check for 2d
945            if self.data.__class__.__name__ =="Data2D":
946                # set mask   
947                radius= numpy.sqrt( self.data.qx_data*self.data.qx_data + self.data.qy_data*self.data.qy_data )
948                index_data = ((self.qmin <= radius)&(radius<= self.qmax))
949                index_data = (index_data)&(self.data.mask)
950                index_data = (index_data)&(numpy.isfinite(self.data.data))
951                if len(index_data[index_data]) < 10:
952                    msg= "Cannot Plot :No or too little npts in that data range!!!  "
[12eac73]953                    wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[51a71a3]954                    return
955                else:
956                    self.data.mask = index_data
957                    self.Npts_fit.Setvalue(str(len(self.data.mask)))
958            else:
959                index_data = ((self.qmin <= self.data.x)&(self.data.x <= self.qmax))
960                self.Npts_fit.SetValue(str(len(self.data.x[index_data])))
[12eac73]961           
962        else:
963           tcrtl.SetBackgroundColour("pink")
964           msg= "Model Error:wrong value entered!!!"
965           wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
966       
967        self.compute_chisqr()   
968        #self._undo.Enable(True)
969        self.save_current_state()
970        event = PageInfoEvent(page = self)
971        wx.PostEvent(self.parent, event)
972        self.state_change= False
[66c488b]973        self._draw_model()
[51a71a3]974        return
[edd166b]975    def _clear_Err_on_Fit(self):
976        """
977            hide the error text control shown
978            after fitting
979        """
980       
981        if hasattr(self,"text2_3"):
982            self.text2_3.Hide()
983
984        if len(self.parameters)>0:
985            for item in self.parameters:
986                #Skip t ifhe angle parameters if 1D data
987                if self.data.__class__.__name__ !="Data2D":
988                    if item in self.orientation_params:
989                        continue
990                if item in self.param_toFit:
991                    continue
992                ## hide statictext +/-   
993                if item[3]!=None and item[3].IsShown():
994                    item[3].Hide()
995                ## hide textcrtl  for error after fit
996                if item[4]!=None and item[4].IsShown():                   
997                    item[4].Hide()
[7975f2b]998 
[edd166b]999        if len(self.fittable_param)>0:
1000            for item in self.fittable_param:
1001                #Skip t ifhe angle parameters if 1D data
1002                if self.data.__class__.__name__ !="Data2D":
1003                    if item in self.orientation_params:
1004                        continue
1005                if item in self.param_toFit:
1006                    continue
1007                ## hide statictext +/-   
1008                if item[3]!=None and item[3].IsShown():
1009                    item[3].Hide()
1010                ## hide textcrtl  for error after fit
1011                if item[4]!=None and item[4].IsShown():
1012                    item[4].Hide()
1013        return       
[920a6e5]1014               
[7609f1a]1015    def _get_defult_custom_smear(self):
1016        """
1017           Get the defult values for custum smearing.
1018        """
1019
1020        # get the default values
1021        if self.dxl == None: self.dxl = 0.0
1022        if self.dxw == None: self.dxw = ""
1023        if self.dx_min == None: self.dx_min = SMEAR_SIZE_L
1024        if self.dx_max == None: self.dx_max = SMEAR_SIZE_H
1025       
1026    def _get_smear_info(self):
1027        """
1028           Get the smear info from data.
1029           @return: self.smear_type, self.dq_l and self.dq_r,
1030            respectively the type of the smear, dq_min and dq_max for pinhole smear data
1031            while dxl and dxw for slit smear
1032        """
1033
1034        # default
1035        self.smear_type = None
1036        self.dq_l = None
1037        self.dq_r = None
1038       
1039        if self.data is None or\
1040                self.data.__class__.__name__ == 'Data2D':           
1041            return
1042
1043        data = self.data
1044
1045        # check if it is pinhole smear and get min max if it is.
1046        if data.dx != None and all(data.dx !=0): 
1047            self.smear_type = "Pinhole" 
1048            self.dq_l = numpy.min(data.dx)
1049            self.dq_r = numpy.max(data.dx)
1050           
1051        # check if it is slit smear and get min max if it is.
1052        elif data.dxl != None or data.dxw != None: 
1053            self.smear_type = "Slit" 
1054            if data.dxl != None and all(data.dxl !=0):
1055                self.dq_l = data.dxl[0]
1056            if data.dxw != None and all(data.dxw !=0): 
1057                self.dq_r = data.dxw[0]
1058               
1059        #return self.smear_type,self.dq_l,self.dq_r       
1060   
1061    def _show_smear_sizer(self):   
1062        """
1063            Show only the sizers depending on smear selection
1064        """
1065        if self.disable_smearer.GetValue():
1066            self.smear_description_none.Show(True)
1067        elif self.enable_smearer.GetValue():
1068            self.smear_description_dqdata.Show(True)
1069            if self.smear_type != None:
1070                self.smear_description_type.Show(True)
1071                self.smear_description_smear_type.Show(True)
1072                if self.smear_type == 'Slit':
1073                    self.smear_description_slit_height.Show(True)
1074                    self.smear_description_slit_width.Show(True)
1075                elif self.smear_type == 'Pinhole':
1076                    self.smear_description_pin_min.Show(True)
1077                    self.smear_description_pin_max.Show(True)
1078                self.smear_data_left.Show(True)
1079                self.smear_data_right.Show(True)
1080        elif self.pinhole_smearer.GetValue():
1081            self.smear_message_new_p.Show(True)
1082            self.smear_description_pin_min.Show(True)
1083            self.smear_pinhole_min.Show(True)
1084            self.smear_description_pin_max.Show(True)
1085            self.smear_pinhole_max.Show(True)
1086        elif self.slit_smearer.GetValue():
1087            self.smear_message_new_s.Show(True)
1088            self.smear_description_slit_height.Show(True)
1089            self.smear_slit_height.Show(True)
1090            self.smear_description_slit_width.Show(True)
1091            self.smear_slit_width.Show(True)
1092
1093    def _hide_all_smear_info(self):
1094        """
1095           Hide all smearing messages in the set_smearer sizer
1096        """
1097        self.smear_description_none.Hide()
1098        self.smear_description_dqdata.Hide()
1099        self.smear_description_type.Hide()
1100        self.smear_description_smear_type.Hide()
1101        self.smear_description_2d.Hide()
1102        self.smear_data_left.Hide()
1103        self.smear_data_right.Hide()
1104        self.smear_description_pin_min.Hide()
1105        self.smear_pinhole_min.Hide()
1106        self.smear_description_pin_max.Hide()
1107        self.smear_pinhole_max.Hide()
1108        self.smear_description_slit_height.Hide()
1109        self.smear_slit_height.Hide()
1110        self.smear_description_slit_width.Hide()
1111        self.smear_slit_width.Hide()
1112        self.smear_message_new_p.Hide()
1113        self.smear_message_new_s.Hide()
1114   
[51a71a3]1115    def _onMask(self, event):     
1116        """
1117            Build a panel to allow to edit Mask
1118        """
1119       
1120        from sans.guiframe.local_perspectives.plotting.masking import MaskPanel as MaskDialog
1121       
1122        self.panel = MaskDialog(self, data=self.data,id =-1 )
1123        self.panel.Bind(wx.EVT_CLOSE, self._draw_masked_model)
1124        self.panel.ShowModal()
1125        #wx.PostEvent(self.parent, event)
1126       
1127    def _draw_masked_model(self,event):
1128        event.Skip()
1129
1130        is_valid_qrange = self._update_paramv_on_fit()
[247cb58]1131
[51a71a3]1132        if is_valid_qrange:
[247cb58]1133            # try re draw the model plot if it exists
[51a71a3]1134            self._draw_model()
1135            self.panel.Destroy() # frame
1136            self.set_npts2fit()
[247cb58]1137        elif self.model == None:
1138            self.panel.Destroy()
1139            self.set_npts2fit()
1140            msg= "No model is found on updating MASK in the model plot... "
1141            wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[51a71a3]1142        else:
[247cb58]1143            msg = ' Please consider your Q range, too.'
[51a71a3]1144            self.panel.ShowMessage(msg)
[882a912]1145       
[9237df4]1146    def set_data(self, data):
[882a912]1147        """
1148            reset the current data
1149        """
[9237df4]1150        self.data = data
[51a71a3]1151
1152       
[9237df4]1153        if self.data is None:
1154            data_min = ""
1155            data_max = ""
1156            data_name = ""
[ffa69b6]1157            self.formfactorbox.Disable()
1158            self.structurebox.Disable()
[9237df4]1159        else:
[00e8df8]1160            self.smearer = smear_selection( self.data )
[d493f66]1161            self.disable_smearer.SetValue(True)
[00e8df8]1162            if self.smearer == None:
1163                self.enable_smearer.Disable()
1164            else:
1165                self.enable_smearer.Enable()
[51a71a3]1166           
[7609f1a]1167            # more disables for 2D
1168            if self.data.__class__.__name__ =="Data2D":
[64017a8]1169                if self.model != None:
1170                    self.smear_description_2d.Show(True)
[7609f1a]1171                self.smear_description_none.Hide()
1172                self.pinhole_smearer.Disable()
1173                self.slit_smearer.Disable()
[51a71a3]1174                self.default_mask = copy.deepcopy(self.data.mask)
[00e8df8]1175               
[d493f66]1176               
[ffa69b6]1177            self.formfactorbox.Enable()
1178            self.structurebox.Enable()
[9237df4]1179            data_name = self.data.name
1180            #set maximum range for x in linear scale
1181            if not hasattr(self.data,"data"): #Display only for 1D data fit
1182                # Minimum value of data   
1183                data_min = min(self.data.x)
1184                # Maximum value of data 
1185                data_max = max(self.data.x)
[51a71a3]1186                #number of total data points
1187                self.Npts_total.SetValue(str(len(self.data.x)))
1188                #default:number of data points selected to fit
1189                self.Npts_fit.SetValue(str(len(self.data.x)))
[19403da]1190                self.btEditMask.Disable() 
1191                self.EditMask_title.Disable()
[9237df4]1192            else:
1193                ## Minimum value of data
1194                data_min = 0
1195                x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax)) 
1196                y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
1197                ## Maximum value of data 
1198                data_max = math.sqrt(x*x + y*y)
[51a71a3]1199                #number of total data points
1200                self.Npts_total.SetValue(str(len(self.data.data)))
1201                #default:number of data points selected to fit
1202                self.Npts_fit.SetValue(str(len(self.data.data)))
[19403da]1203                self.btEditMask.Enable() 
1204                self.EditMask_title.Enable() 
1205
[51a71a3]1206           
[9237df4]1207        self.dataSource.SetValue(data_name)
1208        self.qmin_x = data_min
1209        self.qmax_x = data_max
[ffa69b6]1210        self.minimum_q.SetValue(str(data_min))
1211        self.maximum_q.SetValue(str(data_max))
1212        self.qmin.SetValue(str(data_min))
1213        self.qmax.SetValue(str(data_max))
1214        self.qmin.SetBackgroundColour("white")
1215        self.qmax.SetBackgroundColour("white")
1216        self.state.data = data
1217        self.state.qmin = self.qmin_x
1218        self.state.qmax = self.qmax_x
1219       
[7609f1a]1220
[240b9966]1221    def reset_page(self, state,first=False):
[a074145]1222        """
1223            reset the state
1224        """
1225        self.reset_page_helper(state)
[b421b1a]1226        import sans.guiframe.gui_manager
[f20767b]1227        evt = ModelEventbox(model=state.model)
[edd166b]1228        wx.PostEvent(self.event_owner, evt) 
1229   
1230        if self.engine_type != None:
1231            self.manager._on_change_engine(engine=self.engine_type)
[7975f2b]1232
[edd166b]1233        self.select_param(event = None) 
1234        #Save state_fit
[c985bef]1235        self.save_current_state_fit()
[7975f2b]1236        self._lay_out()
[c985bef]1237        self.Refresh()
[edd166b]1238       
[2140e68]1239    def get_range(self):
1240        """
1241            return the fitting range
1242        """
[c9a4377]1243        return float(self.qmin_x) , float(self.qmax_x)
[7975f2b]1244   
1245    def get_npts2fit(self):
1246        """
1247            return numbers of data points within qrange
1248            Note: This is for Park where chi2 is not normalized by Npts of fit
1249        """
1250        npts2fit = 0
1251        qmin,qmax = self.get_range()
[51a71a3]1252        if self.data.__class__.__name__ =="Data2D": 
1253            radius= numpy.sqrt( self.data.qx_data*self.data.qx_data + self.data.qy_data*self.data.qy_data )
1254            index_data = (self.qmin_x <= radius)&(radius<= self.qmax_x)
1255            index_data= (index_data)&(self.data.mask)
1256            index_data = (index_data)&(numpy.isfinite(self.data.data))
1257            npts2fit = len(self.data.data[index_data])
[7975f2b]1258        else:
1259            for qx in self.data.x:
1260                   if qx >= qmin and qx <= qmax:
1261                       npts2fit += 1
1262        return npts2fit
1263
[51a71a3]1264    def set_npts2fit(self):
1265        """
1266            setValue Npts for fitting
1267        """
1268        self.Npts_fit.SetValue(str(self.get_npts2fit()))
1269       
1270       
[c69b6d5]1271    def get_chi2(self):
1272        """
1273            return the current chi2
1274        """
[2012eae]1275        return self.tcChi.GetValue()
[c77d859]1276       
1277    def get_param_list(self):
1278        """
1279            @return self.param_toFit: list containing  references to TextCtrl
1280            checked.Theses TextCtrl will allow reference to parameters to fit.
1281            @raise: if return an empty list of parameter fit will nnote work
1282            properly so raise ValueError,"missing parameter to fit"
1283        """
1284        if self.param_toFit !=[]:
1285            return self.param_toFit
1286        else:
1287            raise ValueError,"missing parameter to fit"   
1288     
[6d91073]1289    def onsetValues(self,chisqr,p_name, out,cov):
[c77d859]1290        """
1291            Build the panel from the fit result
1292            @param chisqr:Value of the goodness of fit metric
[6d91073]1293            @p_name: the name of parameters
[c77d859]1294            @param out:list of parameter with the best value found during fitting
1295            @param cov:Covariance matrix
1296       
1297        """
[edd166b]1298        if out == None or not numpy.isfinite(chisqr):
1299            raise ValueError,"Fit error occured..." 
1300       
1301        is_modified = False
1302        has_error = False   
[7975f2b]1303       
1304        #Hide textctrl boxes of errors.
[edd166b]1305        self._clear_Err_on_Fit()   
[7975f2b]1306       
1307        #Check if chi2 is finite
1308        if chisqr != None or numpy.isfinite(chisqr):
[e575db9]1309        #format chi2 
[7609f1a]1310            if self.engine_type == "park": 
[e575db9]1311                npt_fit = float(self.get_npts2fit()) 
[7609f1a]1312                if npt_fit > 0:
1313                    chisqr =chisqr/npt_fit   
[7975f2b]1314            chi2 = format_number(chisqr)   
[2012eae]1315            self.tcChi.SetValue(chi2)   
[7975f2b]1316            self.tcChi.Refresh()   
1317        else:
[2012eae]1318            self.tcChi.SetValue("-")
[c13b8cc]1319       
[edd166b]1320        #Hide error title
1321        if self.text2_3.IsShown():
1322            self.text2_3.Hide()
1323     
[e473e4f5]1324        try:
1325            n = self.disp_box.GetCurrentSelection()
1326            dispersity= self.disp_box.GetClientData(n)
[7975f2b]1327            if dispersity !=None and self.enable_disp.GetValue():
1328                name= dispersity.__name__
1329                if name == "GaussianDispersion":
1330                    if hasattr(self,"text_disp_1" ):
1331                        if self.text_disp_1 !=None:
1332                            self.text_disp_1.Hide()
[e473e4f5]1333        except:
[7975f2b]1334            dispersty = None
[e473e4f5]1335            pass
[c77d859]1336        #set the panel when fit result are float not list
[c99a6c5]1337        if out.__class__== numpy.float64:
[c77d859]1338            self.param_toFit[0][2].SetValue(format_number(out))
[0a518e4c]1339           
[c69b6d5]1340            if self.param_toFit[0][4].IsShown:
1341                self.param_toFit[0][4].Hide()
[c77d859]1342            if cov !=None :
1343                self.text2_3.Show(True)
[e473e4f5]1344                try:
[7975f2b]1345                    if dispersity !=None:
1346                        name= dispersity.__name__
1347                        if name == "GaussianDispersion" and self.enable_disp.GetValue():
1348                            if hasattr(self,"text_disp_1" ):
1349                                if self.text_disp_1 !=None:
1350                                    self.text_disp_1.Show(True)
[e473e4f5]1351                except:
1352                    pass
[c99a6c5]1353
[ad6dd4c]1354                if cov[0]==None or  not numpy.isfinite(cov[0]): 
[c69b6d5]1355                    if self.param_toFit[0][3].IsShown:
1356                        self.param_toFit[0][3].Hide()
[7975f2b]1357                else:                   
1358                    self.param_toFit[0][3].Show(True)               
[69bee6d]1359                    self.param_toFit[0][4].Show(True)
[c69b6d5]1360                    self.param_toFit[0][4].SetValue(format_number(cov[0]))
[ffb838f]1361                    has_error = True
[c77d859]1362        else:
[edd166b]1363
[c69b6d5]1364            i = 0
[c77d859]1365            #Set the panel when fit result are list
[7975f2b]1366            for item in self.param_toFit:     
[edd166b]1367                if len(item)>5 and item != None:     
1368                    ## reset error value to initial state
[7975f2b]1369                    item[3].Hide()
1370                    item[4].Hide()
1371                   
[edd166b]1372                    for ind in range(len(out)):
1373                       
1374                        if item[1] == p_name[ind]:
1375                            break       
[7975f2b]1376                    if len(out)<=len(self.param_toFit) and out[ind] !=None:   
1377                        val_out = format_number(out[ind])                 
1378                        item[2].SetValue(val_out)
[c99a6c5]1379
[7975f2b]1380                    if(cov !=None):
[edd166b]1381                       
1382                        try:
[7975f2b]1383                            if dispersity !=None:
1384                                name= dispersity.__name__
1385                                if name == "GaussianDispersion" and self.enable_disp.GetValue():
1386                                    if hasattr(self,"text_disp_1" ):
1387                                        if self.text_disp_1!=None:
1388                                            if not self.text_disp_1.IsShown():
1389                                                self.text_disp_1.Show(True)
[edd166b]1390                        except:
1391                            pass   
1392                   
[7975f2b]1393                        if cov[ind]!=None :
1394                            if numpy.isfinite(float(cov[ind])):
1395                                val_err = format_number(cov[ind])
1396                                item[3].Show(True)
1397                                item[4].Show(True)
1398                                item[4].SetValue(val_err)
1399
1400                                has_error = True
[e575db9]1401                    i += 1         
[c99a6c5]1402        #Show error title when any errors displayed
[b421b1a]1403        if has_error: 
[c99a6c5]1404            if not self.text2_3.IsShown():
[edd166b]1405                self.text2_3.Show(True) 
[7975f2b]1406               
1407        ## save current state 
1408        self.save_current_state()         
1409        #plot model
1410        self._draw_model()   
1411        self._lay_out()     
1412        #PostStatusEvent     
[9237df4]1413        msg = "Fit completed! "
[7975f2b]1414        wx.PostEvent(self.manager.parent, StatusEvent(status=msg))
[c69b6d5]1415
[7609f1a]1416    def onPinholeSmear(self, event):
1417        """
1418            Create a custom pinhole smear object that will change the way residuals
1419            are compute when fitting
1420        """
1421
1422        if self.check_invalid_panel():
1423            return
1424        if self.model ==None:
1425            msg="Need model and data to smear plot"
1426            wx.PostEvent(self.manager.parent, StatusEvent(status=\
1427                            "Smear: %s"%msg))
1428            return
1429       
1430        # msg default
1431        msg = None
1432        if event != None:
1433            tcrtl= event.GetEventObject()
1434            # event case of radio button
1435            if tcrtl.GetValue()== True:
1436                self.dx_min = 0.0
1437                self.dx_max = 0.0
1438                is_new_pinhole = True
1439            else:
1440                is_new_pinhole = self._is_changed_pinhole()
[d493f66]1441        else:
[1da28b8]1442            is_new_pinhole = True 
[7609f1a]1443        # if any value is changed
1444        if is_new_pinhole:
1445            msg = self._set_pinhole_smear()
1446        # hide all silt sizer   
1447        self._hide_all_smear_info()
1448           
1449        ##Calculate chi2
1450        self.compute_chisqr(smearer= self.current_smearer) 
1451        # show relevant slit sizers
1452        self._show_smear_sizer()
1453        self.sizer_set_smearer.Layout()
1454        self.Layout()   
1455       
1456        if event != None: 
1457            event.Skip()                 
1458        #self._undo.Enable(True)
1459        self.save_current_state()
1460        event = PageInfoEvent(page = self)
1461        wx.PostEvent(self.parent, event)
1462       
1463    def _is_changed_pinhole(self): 
1464        """
1465            check if any of pinhole smear is changed
1466            return: True or False
1467        """
1468        # get the values
1469        pin_min = self.smear_pinhole_min.GetValue()
1470        pin_max = self.smear_pinhole_max.GetValue()
1471                   
1472        # Check changes in slit width   
1473        try:
1474            dx_min = float(pin_min)
1475        except:
1476            return True
1477        if self.dx_min != dx_min:
1478            return True
1479       
1480        # Check changes in slit heigth
1481        try:
1482            dx_max = float(pin_max)
1483        except:
1484            return True
1485        if self.dx_max != dx_max:
1486            return True
1487        return False
1488   
1489    def _set_pinhole_smear(self):
1490        """
1491            Set custom pinhole smear
1492            return: msg
1493        """
1494        # copy data
1495        data = copy.deepcopy(self.data)
1496        data_len = len(data.x)
1497        data.dx = None
1498        data.dxl = None
1499        data.dxw = None
1500        msg = None
1501   
1502        # make sure once more if it is smearer
1503        data = copy.deepcopy(self.data)
1504        len_data = len(data.x)
1505        data.dx = numpy.zeros(len_data)
1506        get_pin_min = self.smear_pinhole_min
1507        get_pin_max = self.smear_pinhole_max       
1508
1509        if self._validate_qrange(get_pin_min, get_pin_max ):
1510            if len_data < 2: len_data = 2     
1511            self.dx_min = float(get_pin_min.GetValue())
1512            self.dx_max = float(get_pin_max.GetValue())
1513            if self.dx_min != None and self.dx_max != None:             
1514                if self.dx_min == self.dx_max:
1515                    data.dx[data.dx==0] = self.dx_min
1516                else:
1517                    step = math.fabs(self.dx_max - self.dx_min)/(len_data-1)   
1518                    data.dx = numpy.arange(self.dx_min,self.dx_max+step/1.1,step)           
1519            elif self.dx_min != None: 
1520                data.dx[data.dx==0] = self.dx_min
1521            elif self.dx_max != None:
1522                data.dx[data.dx==0] = self.dx_max         
1523            self.current_smearer = smear_selection(data)
1524
1525        else:
1526            get_pin_min.SetBackgroundColour("pink")
1527            get_pin_max.SetBackgroundColour("pink")
1528            msg= "Model Error:wrong value entered!!!"
1529            wx.PostEvent(self.manager.parent, StatusEvent(status = msg ))
1530        ## set smearing value whether or not the data contain the smearing info
1531        self.manager.set_smearer(smearer=self.current_smearer, qmin= float(self.qmin_x),qmax= float(self.qmax_x))
1532 
1533        return msg
1534       
1535    def update_pinhole_smear(self):
1536        """
1537            called by kill_focus on pinhole TextCntrl
1538            to update the changes
1539            return: msg: False when wrong value was entered
1540        """
1541        # msg default
1542        msg = None
1543        # check if any value is changed
1544        if self._is_changed_pinhole():
1545            msg = self._set_pinhole_smear()           
1546        #self._undo.Enable(True)
1547        self.save_current_state()
1548
1549        if msg != None:
1550            return False   
1551        else:
1552            return True
1553                     
1554    def onSlitSmear(self, event):
1555        """
1556            Create a custom slit smear object that will change the way residuals
1557            are compute when fitting
1558        """
1559
1560               
1561        if self.check_invalid_panel():
1562            return
1563        if self.model ==None:
1564            msg="Need model and data to smear plot"
1565            wx.PostEvent(self.manager.parent, StatusEvent(status=\
1566                            "Smear: %s"%msg))
1567            return
1568        # msg default
1569        msg = None
1570        # for event given
1571        if event != None:
1572            tcrtl= event.GetEventObject()
1573            # event case of radio button
1574            if tcrtl.GetValue():
1575                self.dxl = 0.0
1576                self.dxw = 0.0
1577                is_new_slit = True
1578            else:
1579                is_new_slit = self._is_changed_slit()
[d493f66]1580        else:
[1da28b8]1581            is_new_slit = True 
[7609f1a]1582               
1583        # if any value is changed
1584        if is_new_slit:
1585            msg = self._set_slit_smear()
1586        # hide all silt sizer
1587        self._hide_all_smear_info()       
1588        ##Calculate chi2
1589        self.compute_chisqr(smearer= self.current_smearer) 
1590        # show relevant slit sizers       
1591        self._show_smear_sizer()
1592        self.sizer_set_smearer.Layout()
1593        self.Layout()
1594       
1595        if event != None:
1596            event.Skip()     
1597        #self._undo.Enable(True)
1598        self.save_current_state()
1599        event = PageInfoEvent(page = self)
1600        wx.PostEvent(self.parent, event)
1601        if msg != None:
1602            wx.PostEvent(self.manager.parent, StatusEvent(status = msg))
1603
1604           
1605    def _is_changed_slit(self): 
1606        """
1607            check if any of slit lengths is changed
1608            return: True or False
1609        """
1610        # get the values
1611        width = self.smear_slit_width.GetValue()
1612        height = self.smear_slit_height.GetValue()
1613       
1614        # check and change the box bg color if it was pink but it should be white now
1615        # because this is the case that _set_slit_smear() will not handle
1616        if height.lstrip().rstrip()=="":
1617            self.smear_slit_height.SetBackgroundColour(wx.WHITE)
1618        if width.lstrip().rstrip()=="":
1619            self.smear_slit_width.SetBackgroundColour(wx.WHITE)
1620           
1621        # Check changes in slit width   
1622        if width == "": 
1623            dxw = 0.0
1624        else:
1625            try:
1626                dxw = float(width)
1627            except:
1628                return True
1629        if self.dxw != dxw:
1630            return True
1631       
1632        # Check changes in slit heigth
1633        if height == "": 
1634            dxl = 0.0
1635        else:
1636            try:
1637                dxl = float(height)
1638            except:
1639                return True
1640        if self.dxl != dxl:
1641            return True
1642        return False
1643   
1644    def _set_slit_smear(self):
1645        """
1646            Set custom slit smear
1647            return: msg
1648        """
1649        temp_smearer = None
1650        # make sure once more if it is smearer
1651        data = copy.deepcopy(self.data)
1652        data_len = len(data.x)
1653        data.dx = None
1654        data.dxl = None
1655        data.dxw = None
1656        msg = None
1657   
1658        try:
1659            self.dxl = float(self.smear_slit_height.GetValue())
1660            data.dxl = self.dxl* numpy.ones(data_len)
1661            self.smear_slit_height.SetBackgroundColour(wx.WHITE)
1662        except: 
1663            data.dxl = numpy.zeros(data_len)
1664            if self.smear_slit_height.GetValue().lstrip().rstrip()!="":
1665                self.smear_slit_height.SetBackgroundColour("pink")
1666                msg = "Wrong value entered... " 
1667            else:
1668                self.smear_slit_height.SetBackgroundColour(wx.WHITE)
1669        try:
1670            self.dxw = float(self.smear_slit_width.GetValue())
1671            self.smear_slit_width.SetBackgroundColour(wx.WHITE)
1672            data.dxw = self.dxw* numpy.ones(data_len)
1673        except: 
1674            data.dxw = numpy.zeros(data_len)
1675            if self.smear_slit_width.GetValue().lstrip().rstrip()!="":
1676                self.smear_slit_width.SetBackgroundColour("pink")
1677                msg = "Wrong Fit value entered... "
1678            else:
1679                self.smear_slit_width.SetBackgroundColour(wx.WHITE)
1680               
1681        self.current_smearer = smear_selection(data)
1682        #temp_smearer = self.current_smearer
1683        ## set smearing value whether or not the data contain the smearing info
1684        self.manager.set_smearer(smearer=self.current_smearer, qmin= float(self.qmin_x), qmax= float(self.qmax_x)) 
1685
1686        return msg
1687   
1688   
1689    def update_slit_smear(self):
1690        """
1691            called by kill_focus on pinhole TextCntrl
1692            to update the changes
1693            return: msg: False when wrong value was entered
1694        """             
1695        # msg default
1696        msg = None
1697        # check if any value is changed
1698        if self._is_changed_slit():
1699            msg = self._set_slit_smear()           
1700        #self._undo.Enable(True)
1701        self.save_current_state()
1702
1703        if msg != None:
1704            return False   
1705        else:
1706            return True
1707                           
1708   
[c77d859]1709    def onSmear(self, event):
1710        """
1711            Create a smear object that will change the way residuals
1712            are compute when fitting
1713        """
[ffa69b6]1714        if self.check_invalid_panel():
1715            return
[2a2af47]1716        if self.model ==None:
1717            msg="Need model and data to smear plot"
1718            wx.PostEvent(self.manager.parent, StatusEvent(status=\
1719                            "Smear: %s"%msg))
1720            return
[3370922]1721        temp_smearer = None
[7609f1a]1722        self._get_smear_info()
1723       
1724        #renew smear sizer
1725        if self.smear_type != None:
1726            self.smear_description_smear_type.SetValue(str(self.smear_type))
1727            self.smear_data_left.SetValue(str(self.dq_l))   
1728            self.smear_data_right.SetValue(str(self.dq_r))       
1729       
1730        self._hide_all_smear_info()
1731       
1732        data = copy.deepcopy(self.data)
1733        # make sure once more if it is smearer
1734        self.current_smearer = smear_selection(data)
1735
[c77d859]1736        if self.enable_smearer.GetValue():
[7609f1a]1737            temp_smearer= self.current_smearer
[c77d859]1738            if hasattr(self.data,"dxl"):
[7609f1a]1739               
[c77d859]1740                msg= ": Resolution smearing parameters"
1741            if hasattr(self.data,"dxw"):
1742                msg= ": Slit smearing parameters"
[997131a]1743            if self.smearer ==None:
[c77d859]1744                wx.PostEvent(self.manager.parent, StatusEvent(status=\
1745                            "Data contains no smearing information"))
1746            else:
1747                wx.PostEvent(self.manager.parent, StatusEvent(status=\
1748                            "Data contains smearing information %s"%msg))
[7609f1a]1749            self.smear_description_dqdata.Show(True)
1750            self.smear_data_left.Show(True)
1751            self.smear_data_right.Show(True)
1752        elif self.disable_smearer.GetValue():
1753            self.smear_description_none.Show(True)
1754           
1755        self._show_smear_sizer()
1756        self.sizer_set_smearer.Layout()
1757        self.Layout()
[fb8daaaf]1758        ## set smearing value whether or not the data contain the smearing info
[3370922]1759        self.manager.set_smearer(smearer=temp_smearer, qmin= float(self.qmin_x),
[2b63df0]1760                                     qmax= float(self.qmax_x)) 
[875f1a2]1761        ##Calculate chi2
[997131a]1762        self.compute_chisqr(smearer= temp_smearer) 
[5582a9f1]1763       
1764        self.state.enable_smearer=  self.enable_smearer.GetValue()
1765        self.state.disable_smearer=self.disable_smearer.GetValue()
[7609f1a]1766        self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
1767        self.state.slit_smearer = self.slit_smearer.GetValue()
1768       
[2b63df0]1769    def complete_chisqr(self, output, elapsed=None): 
1770        """
1771            print result chisqr
1772        """
1773        try:
[077809c]1774            if output ==None:
1775                output= "-"
[edd166b]1776
[2012eae]1777            self.tcChi.SetValue(str(format_number(output)))
[12eac73]1778
[edd166b]1779            self.state.tcChi =self.tcChi
[5582a9f1]1780         
[2b63df0]1781        except:
[0aeabc6]1782            pass
[2b63df0]1783       
1784       
1785    def compute_chisqr1D(self, smearer=None):
1786        """
1787            Compute chisqr for 1D
1788        """
1789        try:
1790            self.qmin_x = float(self.qmin.GetValue())
1791            self.qmax_x = float(self.qmax.GetValue())
1792            ##return residuals within self.qmin_x and self.qmax_x
1793            from gui_thread import CalcChisqr1D
1794            ## If a thread is already started, stop it
1795            if self.calc_Chisqr!= None and self.calc_Chisqr.isrunning():
1796                self.calc_Chisqr.stop()
1797               
[077809c]1798            self.calc_Chisqr= CalcChisqr1D( data1d= self.data,
[2b63df0]1799                                            model= self.model,
1800                                            smearer=smearer,
1801                                            qmin=self.qmin_x,
1802                                            qmax=self.qmax_x,
1803                                            completefn = self.complete_chisqr,
1804                                            updatefn   = None)
1805   
1806            self.calc_Chisqr.queue()
1807           
1808        except:
1809            raise ValueError," Could not compute Chisqr for %s Model 2D: "%self.model.name
1810           
1811           
1812   
1813       
[b787e68c]1814       
[c77d859]1815    def compute_chisqr2D(self):
1816        """
1817            compute chi square given a model and data 2D and set the value
1818            to the tcChi txtcrl
1819        """
[2b63df0]1820        try:
1821            self.qmin_x = float(self.qmin.GetValue())
1822            self.qmax_x = float(self.qmax.GetValue())
1823           
1824            ##return residuals within self.qmin_x and self.qmax_x
1825            from gui_thread import CalcChisqr2D
1826            ## If a thread is already started, stop it
1827            if self.calc_Chisqr!= None and self.calc_Chisqr.isrunning():
1828                self.calc_Chisqr.stop()
[0aeabc6]1829           
[077809c]1830            self.calc_Chisqr= CalcChisqr2D( data2d= self.data,
[2b63df0]1831                                            model= self.model,
1832                                            qmin= self.qmin_x,
1833                                            qmax = self.qmax_x,
1834                                            completefn = self.complete_chisqr,
1835                                            updatefn   = None)
[c77d859]1836   
[2b63df0]1837            self.calc_Chisqr.queue()
1838         
1839        except:
[077809c]1840           raise
[2b63df0]1841
[c77d859]1842       
[997131a]1843    def compute_chisqr(self , smearer=None):
[c77d859]1844        """
1845            compute chi square given a model and data 1D and set the value
1846            to the tcChi txtcrl
1847        """
[85b3971]1848        flag = self._validate_qrange( self.qmin, self.qmax)
[c77d859]1849        if flag== True:
1850            try:
1851                if hasattr(self.data,"data"):
1852                    self.compute_chisqr2D()
1853                    return
1854                else:
[2b63df0]1855                    self.compute_chisqr1D(smearer=smearer)
1856                    return
[c77d859]1857            except:
[077809c]1858                wx.PostEvent(self.parent.parent, StatusEvent(status=\
[2b63df0]1859                            "Chisqr Error: %s"% sys.exc_value))
[a8088d7]1860                return 
[c77d859]1861           
1862   
1863    def select_all_param(self,event): 
1864        """
1865             set to true or false all checkBox given the main checkbox value cb1
[d1e0473]1866        """           
1867
[c77d859]1868        self.param_toFit=[]
1869        if  self.parameters !=[]:
[998b6b8]1870            if  self.cb1.GetValue():
[c77d859]1871                for item in self.parameters:
[3fef0a8]1872                    ## for data2D select all to fit
1873                    if self.data.__class__.__name__=="Data2D":
[c77d859]1874                        item[0].SetValue(True)
1875                        self.param_toFit.append(item )
[3fef0a8]1876                    else:
1877                        ## for 1D all parameters except orientation
1878                        if not item in self.orientation_params:
1879                            item[0].SetValue(True)
1880                            self.param_toFit.append(item )
[b324aa9]1881                #if len(self.fittable_param)>0:
1882                for item in self.fittable_param:
1883                    if self.data.__class__.__name__=="Data2D":
1884                        item[0].SetValue(True)
1885                        self.param_toFit.append(item )
1886                    else:
1887                        ## for 1D all parameters except orientation
1888                        if not item in self.orientation_params_disp:
[3fef0a8]1889                            item[0].SetValue(True)
1890                            self.param_toFit.append(item )
[c77d859]1891            else:
1892                for item in self.parameters:
1893                    item[0].SetValue(False)
1894                for item in self.fittable_param:
1895                    item[0].SetValue(False)
1896                self.param_toFit=[]
[330573d]1897           
[52cac46]1898        self.save_current_state_fit() 
[b324aa9]1899       
[3595309d]1900        if event !=None:
[3a37fe0]1901            #self._undo.Enable(True)
[3595309d]1902            ## post state to fit panel
1903            event = PageInfoEvent(page = self)
1904            wx.PostEvent(self.parent, event) 
1905     
[c77d859]1906               
1907               
1908    def select_param(self,event):
1909        """
1910            Select TextCtrl  checked for fitting purpose and stores them
1911            in  self.param_toFit=[] list
1912        """
1913        self.param_toFit=[]
1914        for item in self.parameters:
[edd166b]1915            #Skip t ifhe angle parameters if 1D data
1916            if self.data.__class__.__name__ !="Data2D":
1917                if item in self.orientation_params:
1918                    continue
[c77d859]1919            #Select parameters to fit for list of primary parameters
[998b6b8]1920            if item[0].GetValue():
[c77d859]1921                if not (item in self.param_toFit):
1922                    self.param_toFit.append(item ) 
1923            else:
1924                #remove parameters from the fitting list
1925                if item in self.param_toFit:
1926                    self.param_toFit.remove(item)
[edd166b]1927
[c77d859]1928        #Select parameters to fit for list of fittable parameters with dispersion         
1929        for item in self.fittable_param:
[edd166b]1930            #Skip t ifhe angle parameters if 1D data
1931            if self.data.__class__.__name__ !="Data2D":
1932                if item in self.orientation_params:
1933                    continue
[998b6b8]1934            if item[0].GetValue():
[c77d859]1935                if not (item in self.param_toFit):
1936                    self.param_toFit.append(item) 
1937            else:
1938                #remove parameters from the fitting list
1939                if item in self.param_toFit:
[edd166b]1940                    self.param_toFit.remove(item)
1941
1942        #Calculate num. of angle parameters
1943        if self.data.__class__.__name__ =="Data2D": 
1944            len_orient_para = 0
1945        else:
1946            len_orient_para = len(self.orientation_params)  #assume even len
1947        #Total num. of angle parameters
1948        if len(self.fittable_param) > 0:
1949            len_orient_para *= 2
[c77d859]1950        #Set the value of checkbox that selected every checkbox or not           
[edd166b]1951        if len(self.parameters)+len(self.fittable_param)-len_orient_para ==len(self.param_toFit):
[c77d859]1952            self.cb1.SetValue(True)
1953        else:
1954            self.cb1.SetValue(False)
[52cac46]1955        self.save_current_state_fit()
[3595309d]1956        if event !=None:
[3a37fe0]1957            #self._undo.Enable(True)
[3595309d]1958            ## post state to fit panel
1959            event = PageInfoEvent(page = self)
1960            wx.PostEvent(self.parent, event) 
1961     
[c77d859]1962   
1963       
[77e23a2]1964    def set_model_param_sizer(self, model):
[c77d859]1965        """
1966            Build the panel from the model content
1967            @param model: the model selected in combo box for fitting purpose
1968        """
1969        self.sizer3.Clear(True)
1970        self.parameters = []
1971        self.param_toFit=[]
1972        self.fittable_param=[]
1973        self.fixed_param=[]
[780d095]1974        self.orientation_params=[]
[60132ef]1975        self.orientation_params_disp=[]
[c77d859]1976       
1977        if model ==None:
1978            self.sizer3.Layout()
[08ba57d]1979            self.SetScrollbars(20,20,25,65)
[c77d859]1980            return
[707436d]1981        ## the panel is drawn using the current value of the fit engine
1982        if self.engine_type==None and self.manager !=None:
1983            self.engine_type= self.manager._return_engine_type()
[c99a6c5]1984
[707436d]1985           
[c77d859]1986        box_description= wx.StaticBox(self, -1,str("Model Parameters"))
1987        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1988        sizer = wx.GridBagSizer(5,5)
1989        ## save the current model
1990        self.model = model
1991           
1992        keys = self.model.getParamList()
1993        #list of dispersion paramaters
1994        self.disp_list=self.model.getDispParamList()
[b421b1a]1995
[c77d859]1996        keys.sort()
1997   
[6dc9ad8]1998        iy = 0
[c77d859]1999        ix = 0
[2f6c260]2000        select_text = "Select All"
[b324aa9]2001        self.cb1 = wx.CheckBox(self, -1,str(select_text), (10, 10))
[c77d859]2002        wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.select_all_param)
[2f6c260]2003        self.cb1.SetToolTipString("To check/uncheck all the boxes below.")
2004        #self.cb1.SetValue(True)
[c77d859]2005       
2006        sizer.Add(self.cb1,(iy, ix),(1,1),\
[eda428b]2007                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
[c77d859]2008        ix +=1
2009        self.text2_2 = wx.StaticText(self, -1, 'Values')
2010        sizer.Add(self.text2_2,(iy, ix),(1,1),\
2011                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2012        ix +=2 
2013        self.text2_3 = wx.StaticText(self, -1, 'Errors')
2014        sizer.Add(self.text2_3,(iy, ix),(1,1),\
2015                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2016        self.text2_3.Hide()
2017        ix +=1 
2018        self.text2_min = wx.StaticText(self, -1, 'Min')
2019        sizer.Add(self.text2_min,(iy, ix),(1,1),\
2020                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2021        self.text2_min.Hide()
2022        ix +=1 
2023        self.text2_max = wx.StaticText(self, -1, 'Max')
2024        sizer.Add(self.text2_max,(iy, ix),(1,1),\
2025                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2026        self.text2_max.Hide()
[6fdfc8f]2027        ix += 1
2028        self.text2_4 = wx.StaticText(self, -1, '[Units]')
2029        sizer.Add(self.text2_4,(iy, ix),(1,1),\
2030                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2031        self.text2_4.Hide()
[dcf29d7]2032        if self.engine_type=="park":
2033            self.text2_max.Show(True)
2034            self.text2_min.Show(True)
[b421b1a]2035       
[c77d859]2036        for item in keys:
[780d095]2037            if not item in self.disp_list and not item in self.model.orientation_params:
[b421b1a]2038               
2039                ##prepare a spot to store errors
2040                if not self.model.details.has_key(item):
2041                    self.model.details [item] = ["",None,None] 
2042         
[c77d859]2043                iy += 1
2044                ix = 0
2045                ## add parameters name with checkbox for selecting to fit
[2f6c260]2046                cb = wx.CheckBox(self, -1, item )             
[479eced]2047                cb.SetToolTipString(" Check for fitting.")
[2f6c260]2048                #cb.SetValue(True)
[c77d859]2049                wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
[2f6c260]2050               
[c77d859]2051                sizer.Add( cb,( iy, ix),(1,1),
[eda428b]2052                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
[d1e0473]2053               
[c77d859]2054                ## add parameter value
2055                ix += 1
2056                value= self.model.getParam(item)
[7975f2b]2057                ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[c77d859]2058                                    style=wx.TE_PROCESS_ENTER)
2059               
[77e23a2]2060                ctl1.SetValue(format_number(value))
[c77d859]2061                sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
2062                ## text to show error sign
2063                ix += 1
2064                text2=wx.StaticText(self, -1, '+/-')
2065                sizer.Add(text2,(iy, ix),(1,1),\
2066                                wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2067                text2.Hide() 
2068                ix += 1
[c13b8cc]2069                ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[c77d859]2070                sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[b421b1a]2071                ctl2.Hide()
[c77d859]2072               
2073                ix += 1
[7975f2b]2074                ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
2075                                               text_enter_callback = self._onparamRangeEnter)
[eacf1d66]2076     
[c77d859]2077                sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2078                ctl3.Hide()
[77e23a2]2079       
[c77d859]2080                ix += 1
[7975f2b]2081                ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
2082                                               text_enter_callback = self._onparamRangeEnter)
[c77d859]2083                sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[eacf1d66]2084     
[c77d859]2085                ctl4.Hide()
[edd166b]2086
[dcf29d7]2087                if self.engine_type=="park":
2088                    ctl3.Show(True)
2089                    ctl4.Show(True)
[6fdfc8f]2090                ix +=1
2091                # Units
[7975f2b]2092                if self.model.details.has_key(item):
[6fdfc8f]2093                    units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
[7975f2b]2094                else:
[6fdfc8f]2095                    units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
2096                sizer.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[dcf29d7]2097                   
[c77d859]2098                ##[cb state, name, value, "+/-", error of fit, min, max , units]
2099                self.parameters.append([cb,item, ctl1,
[240b9966]2100                                        text2,ctl2, ctl3, ctl4,units])
[2d5f7a1]2101             
[c77d859]2102        iy+=1
[2a5bba7]2103        sizer.Add((10,10),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
[5af5183]2104       
[f20767b]2105        # type can be either Guassian or Array
[5b53a5c]2106        if len(self.model.dispersion.values())>0:
2107            type= self.model.dispersion.values()[0]["type"]
2108        else:
2109            type = "Gaussian"
[6dc9ad8]2110           
2111        iy += 1
2112        ix = 0
2113        #Add tile for orientational angle
2114        for item in keys:
2115            if item in self.model.orientation_params:       
2116                orient_angle = wx.StaticText(self, -1, '[For 2D only]:')
2117                sizer.Add(orient_angle,(iy, ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) 
2118                if not self.data.__class__.__name__ =="Data2D":
2119                    orient_angle.Hide()
2120                else:
2121                    orient_angle.Show(True)
2122                break
[5b53a5c]2123     
[f20767b]2124        #For Gaussian only
2125        if type.lower() != "array":
2126            for item in self.model.orientation_params:
2127                if not item in self.disp_list:
[c13b8cc]2128                    ##prepare a spot to store min max
2129                    if not self.model.details.has_key(item):
2130                        self.model.details [item] = ["",None,None] 
2131                         
[f20767b]2132                    iy += 1
2133                    ix = 0
2134                    ## add parameters name with checkbox for selecting to fit
2135                    cb = wx.CheckBox(self, -1, item )
2136                    cb.SetValue(False)
2137                    wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
2138                    if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]2139                        cb.Show(True)
[f20767b]2140                    else:
[6dc9ad8]2141                        cb.Hide()
[f20767b]2142                    sizer.Add( cb,( iy, ix),(1,1),
2143                                 wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 5)
2144   
2145                    ## add parameter value
2146                    ix += 1
2147                    value= self.model.getParam(item)
[7975f2b]2148                    ctl1 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH,20),
[f20767b]2149                                        style=wx.TE_PROCESS_ENTER)
[780d095]2150                   
[f20767b]2151                    ctl1.SetValue(format_number(value))
2152                    if self.data.__class__.__name__ =="Data2D":
[c5cd3b9]2153                        ctl1.Show(True)
[f20767b]2154                    else:
[c5cd3b9]2155                        ctl1.Hide()
[f20767b]2156                    sizer.Add(ctl1, (iy,ix),(1,1), wx.EXPAND)
2157                    ## text to show error sign
2158                    ix += 1
2159                    text2=wx.StaticText(self, -1, '+/-')
2160                    sizer.Add(text2,(iy, ix),(1,1),\
2161                                    wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
2162                    text2.Hide() 
2163                    ix += 1
[c13b8cc]2164                    ctl2 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=0)
[e2f7b92]2165                   
[f20767b]2166                    sizer.Add(ctl2, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2167                    ctl2.Hide()
[e2f7b92]2168                   
[c99a6c5]2169                   
[f20767b]2170                    ix += 1
[7975f2b]2171                    ctl3 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
2172                                                   text_enter_callback = self._onparamRangeEnter)
[eacf1d66]2173               
[f20767b]2174                    sizer.Add(ctl3, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
2175                    ctl3.Hide()
[b324aa9]2176                 
[f20767b]2177                    ix += 1
[7975f2b]2178                    ctl4 = self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH/2,20), style=wx.TE_PROCESS_ENTER,
2179                                                   text_enter_callback = self._onparamRangeEnter)
[f20767b]2180                    sizer.Add(ctl4, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[eacf1d66]2181                   
[f20767b]2182                    ctl4.Hide()
[edd166b]2183                   
2184                    if self.engine_type =="park" and self.data.__class__.__name__ =="Data2D":                     
[f20767b]2185                        ctl3.Show(True)
2186                        ctl4.Show(True)
[c5cd3b9]2187                   
[f20767b]2188                    ix +=1
2189                    # Units
[7975f2b]2190                    if self.model.details.has_key(item):
[f20767b]2191                        units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
[7975f2b]2192                    else:
[f20767b]2193                        units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
2194                    if self.data.__class__.__name__ =="Data2D":
[6dc9ad8]2195                        units.Show(True)
[c99a6c5]2196                   
[f20767b]2197                    else:
[6dc9ad8]2198                        units.Hide()
[c99a6c5]2199                   
[f20767b]2200                    sizer.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[7975f2b]2201                                         
[f20767b]2202                    ##[cb state, name, value, "+/-", error of fit, min, max , units]
2203                    self.parameters.append([cb,item, ctl1,
2204                                            text2,ctl2, ctl3, ctl4,units])
2205                    self.orientation_params.append([cb,item, ctl1,
2206                                            text2,ctl2, ctl3, ctl4,units])
[780d095]2207             
2208        iy+=1
[c77d859]2209       
2210        #Display units text on panel
2211        for item in keys:   
[7975f2b]2212            if self.model.details.has_key(item):
[c77d859]2213                self.text2_4.Show()
[b324aa9]2214        #Fill the list of fittable parameters
2215        self.select_all_param(event=None)
[7975f2b]2216
[52cac46]2217        self.save_current_state_fit()
[240b9966]2218        boxsizer1.Add(sizer)
[c77d859]2219        self.sizer3.Add(boxsizer1,0, wx.EXPAND | wx.ALL, 10)
2220        self.sizer3.Layout()
[330573d]2221        self.Layout()
2222        self.Refresh()
[08ba57d]2223        self.SetScrollbars(20,20,25,65)
[52cac46]2224
[7609f1a]2225class BGTextCtrl(wx.TextCtrl):
2226    """
2227        Text control used to display outputs.
2228        No editing allowed. The background is
2229        grayed out. User can't select text.
2230    """
2231    def __init__(self, *args, **kwds):
2232        wx.TextCtrl.__init__(self, *args, **kwds)
2233        self.SetEditable(False)
2234        self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
2235       
2236        # Bind to mouse event to avoid text highlighting
2237        # The event will be skipped once the call-back
2238        # is called.
2239        self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
2240       
2241    def _click(self, event):
2242        """
2243            Prevent further handling of the mouse event
2244            by not calling Skip().
2245        """ 
2246        pass
[c77d859]2247       
2248class HelpWindow(wx.Frame):
2249    def __init__(self, parent, id, title):
2250        wx.Frame.__init__(self, parent, id, title, size=(570, 400))
2251       
2252        from sans.models.CylinderModel import CylinderModel
2253        model = CylinderModel()
[2d5f7a1]2254       
[c77d859]2255        from danse.common.plottools.plottables import Data1D
2256        data= Data1D(x=[1,2], y=[3,4], dy=[0.1, 0,1])
2257   
[cfc0913]2258        from fitpanel import PageInfo
[c77d859]2259        myinfo = PageInfo(self,  model, data=data )
2260       
2261        ## add data
2262       
2263        from models import ModelList
2264        mylist= ModelList()
2265
2266        from sans.models.SphereModel import SphereModel
2267        from sans.models.SquareWellStructure import SquareWellStructure
2268        from sans.models.DebyeModel import DebyeModel
2269        from sans.models.LineModel import LineModel
2270        name= "shapes"
2271        list1= [SphereModel]
2272        mylist.set_list( name, list1)
2273       
2274        name= "Shape-independent"
2275        list1= [DebyeModel]
2276        mylist.set_list( name, list1)
2277       
2278        name= "Structure Factors"
2279        list1= [SquareWellStructure]
2280        mylist.set_list( name, list1)
2281       
2282        name= "Added models"
2283        list1= [LineModel]
2284        mylist.set_list( name, list1)
2285       
2286        myinfo.model_list_box = mylist.get_list()
2287       
2288        self.page = FitPage(self, myinfo) 
2289       
2290        self.Centre()
2291        self.Show(True)
[eacf1d66]2292 
[c77d859]2293if __name__=="__main__":
2294    app = wx.App()
2295    HelpWindow(None, -1, 'HelpWindow')
2296    app.MainLoop()
2297               
Note: See TracBrowser for help on using the repository browser.