source: sasview/sansview/perspectives/fitting/fitpage.py @ b0e8beb

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

working fitpage inheritance

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