source: sasview/sansview/perspectives/fitting/fitpage.py @ 904193cd

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 904193cd was c365765, checked in by Jae Cho <jhjcho@…>, 13 years ago

another test

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