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

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 0fd2f27 was 3f36675, checked in by Gervaise Alina <gervyh@…>, 13 years ago

working on batch fit

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