source: sasview/fittingview/src/sans/perspectives/fitting/batchfitpage.py @ 1dbff4c

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 1dbff4c was 978feaa, checked in by Jae Cho <jhjcho@…>, 13 years ago

save flag
  • Property mode set to 100644
File size: 30.0 KB
Line 
1
2
3import sys
4import wx
5import wx.lib.newevent
6import numpy
7import copy
8import math
9import time
10from sans.models.dispersion_models import ArrayDispersion, GaussianDispersion
11from sans.dataloader.data_info import Data1D
12from sans.guiframe.events import StatusEvent   
13from sans.guiframe.events import NewPlotEvent 
14from sans.guiframe.utils import format_number,check_float
15
16(Chi2UpdateEvent, EVT_CHI2_UPDATE)   = wx.lib.newevent.NewEvent()
17_BOX_WIDTH = 76
18_DATA_BOX_WIDTH = 300
19SMEAR_SIZE_L = 0.00
20SMEAR_SIZE_H = 0.00
21
22import basepage
23from basepage import BasicPage
24from basepage import PageInfoEvent
25from sans.models.qsmearing import smear_selection
26from fitpage import FitPage
27
28class BatchFitPage(FitPage):
29    window_name = "BatchFit"
30    window_caption  = "BatchFit"
31
32    def __init__(self,parent, color='rand'):
33        """
34        Initialization of the Panel
35        """
36        FitPage.__init__(self, parent, color=color)
37        self.window_name = "BatchFit"
38        self.window_caption  = "BatchFit"
39        self._set_save_flag(False)
40       
41    def _fill_range_sizer(self):
42        """
43        Fill the sizer containing the plotting range
44        add  access to npts
45        """
46        is_2Ddata = False
47       
48        # Check if data is 2D
49        if self.data.__class__.__name__ ==  "Data2D" or \
50                        self.enable2D:
51            is_2Ddata = True
52           
53        title = "Fitting"
54        #smear messages & titles
55        smear_message_none  =  "No smearing is selected..."
56        smear_message_dqdata  =  "The dQ data is being used for smearing..."
57        smear_message_2d  =  "Higher accuracy is very time-expensive. Use it with care..."
58        smear_message_new_ssmear  =  "Please enter only the value of interest to customize smearing..."
59        smear_message_new_psmear  =  "Please enter both; the dQ will be generated by interpolation..."
60        smear_message_2d_x_title = "<dQp>[1/A]:"
61        smear_message_2d_y_title = "<dQs>[1/A]:"       
62        smear_message_pinhole_min_title = "dQ_low[1/A]:"
63        smear_message_pinhole_max_title = "dQ_high[1/A]:"
64        smear_message_slit_height_title = "Slit height[1/A]:"
65        smear_message_slit_width_title = "Slit width[1/A]:"
66       
67        self._get_smear_info()
68       
69        #Sizers
70        box_description_range = wx.StaticBox(self, -1,str(title))
71        boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)     
72        self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
73        sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
74        self.sizer_new_smear= wx.BoxSizer(wx.HORIZONTAL)
75        self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
76        sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
77        """
78        smear_set_box= wx.StaticBox(self, -1,'Set Instrumental Smearing')
79        sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
80        sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH,85))
81        """
82        sizer_fit = wx.GridSizer(2, 4, 2, 6)
83        """
84        # combobox for smear2d accuracy selection
85        self.smear_accuracy = wx.ComboBox(self, -1,size=(50,-1),style=wx.CB_READONLY)
86        self._set_accuracy_list()
87        self.smear_accuracy.SetValue(self.smear2d_accuracy)
88        self.smear_accuracy.SetSelection(0)
89        self.smear_accuracy.SetToolTipString("'Higher' uses more Gaussian points for smearing computation.")
90                   
91        wx.EVT_COMBOBOX(self.smear_accuracy,-1, self._on_select_accuracy)
92        """
93        #Fit button
94        self.btFit = wx.Button(self,wx.NewId(),'Fit', size=(88,25))
95        self.default_bt_colour =  self.btFit.GetDefaultAttributes()
96        self.btFit.Bind(wx.EVT_BUTTON, self._onFit,id= self.btFit.GetId())
97        self.btFit.SetToolTipString("Start fitting.")
98        """
99        #textcntrl for custom resolution
100        self.smear_pinhole_max = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
101                                            text_enter_callback = self.onPinholeSmear)
102        self.smear_pinhole_min = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
103                                            text_enter_callback = self.onPinholeSmear)
104        self.smear_slit_height= self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
105                                            text_enter_callback = self.onSlitSmear)
106        self.smear_slit_width = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH-25,20),style=wx.TE_PROCESS_ENTER,
107                                            text_enter_callback = self.onSlitSmear)
108
109        ## smear
110        self.smear_data_left= BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
111        self.smear_data_left.SetValue(str(self.dq_l))
112        self.smear_data_right = BGTextCtrl(self, -1, size=(_BOX_WIDTH-25,20), style=0)
113        self.smear_data_right.SetValue(str(self.dq_r))
114
115        #set default values for smear
116        self.smear_pinhole_max.SetValue(str(self.dx_max))
117        self.smear_pinhole_min.SetValue(str(self.dx_min))
118        self.smear_slit_height.SetValue(str(self.dxl))
119        self.smear_slit_width.SetValue(str(self.dxw))
120
121        #Filling the sizer containing instruments smearing info.
122        self.disable_smearer = wx.RadioButton(self, -1, 'None', style=wx.RB_GROUP)
123        self.enable_smearer = wx.RadioButton(self, -1, 'Use dQ Data')
124        #self.enable_smearer.SetToolTipString("Click to use the loaded dQ data for smearing.")
125        self.pinhole_smearer = wx.RadioButton(self, -1, 'Custom Pinhole Smear')
126        #self.pinhole_smearer.SetToolTipString("Click to input custom resolution for pinhole smearing.")
127        self.slit_smearer = wx.RadioButton(self, -1, 'Custom Slit Smear')
128        #self.slit_smearer.SetToolTipString("Click to input custom resolution for slit smearing.")
129        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.disable_smearer.GetId())
130        self.Bind(wx.EVT_RADIOBUTTON, self.onSmear, id=self.enable_smearer.GetId())
131        self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear, id=self.pinhole_smearer.GetId())
132        self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear, id=self.slit_smearer.GetId())
133        self.disable_smearer.SetValue(True)
134       
135        # add 4 types of smearing to the sizer
136        sizer_smearer.Add( self.disable_smearer,0, wx.LEFT, 10)
137        sizer_smearer.Add((10,10))
138        sizer_smearer.Add( self.enable_smearer)
139        sizer_smearer.Add((10,10))
140        sizer_smearer.Add( self.pinhole_smearer )
141        sizer_smearer.Add((10,10))
142        sizer_smearer.Add( self.slit_smearer )
143        sizer_smearer.Add((10,10)) 
144        """     
145        """
146        # StaticText for chi2, N(for fitting), Npts
147        self.tcChi    =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
148        self.tcChi.SetToolTipString("Chi2/Npts(Fit)")
149        self.Npts_fit    =  BGTextCtrl(self, -1, "-", size=(75,20), style=0)
150        self.Npts_fit.SetToolTipString(\
151                            " Npts : number of points selected for fitting")
152        self.Npts_total  =  self.ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
153                        style=wx.TE_PROCESS_ENTER,
154                        text_enter_callback=self._onQrangeEnter)
155        self.Npts_total.SetValue(format_number(self.npts_x))
156        self.Npts_total.SetToolTipString(\
157                                " Total Npts : total number of data points")
158        """
159        # Update and Draw button
160        self.draw_button = wx.Button(self,wx.NewId(),'Compute', size=(88,24))
161        self.draw_button.Bind(wx.EVT_BUTTON, \
162                              self._onDraw,id= self.draw_button.GetId())
163        self.draw_button.SetToolTipString("Compute and Draw.")
164        """
165        box_description_1= wx.StaticText(self, -1,'    Chi2/Npts')
166        box_description_2= wx.StaticText(self, -1,'Npts(Fit)')
167        box_description_3= wx.StaticText(self, -1,'Total Npts')
168        box_description_3.SetToolTipString( \
169                                " Total Npts : total number of data points")
170        #box_description_4= wx.StaticText(self, -1,' ')
171        """
172       
173        """
174        sizer_fit.Add(box_description_1,0,0)
175        sizer_fit.Add(box_description_2,0,0)
176        sizer_fit.Add(box_description_3,0,0) 
177        """     
178        sizer_fit.Add(self.draw_button,0,0)
179        """
180        sizer_fit.Add(self.tcChi,0,0)
181        sizer_fit.Add(self.Npts_fit ,0,0)
182        sizer_fit.Add(self.Npts_total,0,0)
183        """
184        sizer_fit.Add(self.btFit,0,0) 
185        """
186        self.smear_description_none    =  wx.StaticText(self, -1, smear_message_none , style=wx.ALIGN_LEFT)
187        self.smear_description_dqdata    =  wx.StaticText(self, -1, smear_message_dqdata , style=wx.ALIGN_LEFT)
188        self.smear_description_type    =  wx.StaticText(self, -1, "Type:" , style=wx.ALIGN_LEFT)
189        self.smear_description_accuracy_type    =  wx.StaticText(self, -1, "Accuracy:" , style=wx.ALIGN_LEFT)
190        self.smear_description_smear_type    =  BGTextCtrl(self, -1, size=(57,20), style=0)
191        self.smear_description_smear_type.SetValue(str(self.dq_l))
192        self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
193        self.smear_description_2d     =  wx.StaticText(self, -1, smear_message_2d  , style=wx.ALIGN_LEFT)
194        self.smear_message_new_s = wx.StaticText(self, -1, smear_message_new_ssmear  , style=wx.ALIGN_LEFT)
195        self.smear_message_new_p = wx.StaticText(self, -1, smear_message_new_psmear  , style=wx.ALIGN_LEFT)
196        self.smear_description_2d_x     =  wx.StaticText(self, -1, smear_message_2d_x_title  , style=wx.ALIGN_LEFT)
197        self.smear_description_2d_x.SetToolTipString("  dQp(parallel) in q_r direction.")
198        self.smear_description_2d_y     =  wx.StaticText(self, -1, smear_message_2d_y_title  , style=wx.ALIGN_LEFT)
199        self.smear_description_2d_y.SetToolTipString(" dQs(perpendicular) in q_phi direction.")
200        self.smear_description_pin_min     =  wx.StaticText(self, -1, smear_message_pinhole_min_title  , style=wx.ALIGN_LEFT)
201        self.smear_description_pin_max     =  wx.StaticText(self, -1, smear_message_pinhole_max_title  , style=wx.ALIGN_LEFT)
202        self.smear_description_slit_height    =  wx.StaticText(self, -1, smear_message_slit_height_title   , style=wx.ALIGN_LEFT)
203        self.smear_description_slit_width    =  wx.StaticText(self, -1, smear_message_slit_width_title   , style=wx.ALIGN_LEFT)
204        """
205        #arrange sizers
206        #boxsizer1.Add( self.tcChi ) 
207        """
208        self.sizer_set_smearer.Add(sizer_smearer )
209        self.sizer_set_smearer.Add((10,10))
210        self.sizer_set_smearer.Add( self.smear_description_none,0, wx.CENTER, 10 )
211        self.sizer_set_smearer.Add( self.smear_description_dqdata,0, wx.CENTER, 10 )
212        self.sizer_set_smearer.Add( self.smear_description_2d,0, wx.CENTER, 10 )
213        self.sizer_new_smear.Add( self.smear_description_type,0, wx.CENTER, 10 )
214        self.sizer_new_smear.Add( self.smear_description_accuracy_type,0, wx.CENTER, 10 )
215        self.sizer_new_smear.Add( self.smear_accuracy )
216        self.sizer_new_smear.Add( self.smear_description_smear_type,0, wx.CENTER, 10 )
217        self.sizer_new_smear.Add((15,-1))
218        self.sizer_new_smear.Add( self.smear_description_2d_x,0, wx.CENTER, 10 )
219        self.sizer_new_smear.Add( self.smear_description_pin_min,0, wx.CENTER, 10 )
220        self.sizer_new_smear.Add( self.smear_description_slit_height,0, wx.CENTER, 10 )
221
222        self.sizer_new_smear.Add( self.smear_pinhole_min,0, wx.CENTER, 10 )
223        self.sizer_new_smear.Add( self.smear_slit_height,0, wx.CENTER, 10 )
224        self.sizer_new_smear.Add( self.smear_data_left,0, wx.CENTER, 10 )
225        self.sizer_new_smear.Add((20,-1))
226        self.sizer_new_smear.Add( self.smear_description_2d_y,0, wx.CENTER, 10 )
227        self.sizer_new_smear.Add( self.smear_description_pin_max,0, wx.CENTER, 10 )
228        self.sizer_new_smear.Add( self.smear_description_slit_width,0, wx.CENTER, 10 )
229
230        self.sizer_new_smear.Add( self.smear_pinhole_max,0, wx.CENTER, 10 )
231        self.sizer_new_smear.Add( self.smear_slit_width,0, wx.CENTER, 10 )
232        self.sizer_new_smear.Add( self.smear_data_right,0, wx.CENTER, 10 )
233           
234        self.sizer_set_smearer.Add( self.smear_message_new_s,0, wx.CENTER, 10)
235        self.sizer_set_smearer.Add( self.smear_message_new_p,0, wx.CENTER, 10)
236        self.sizer_set_smearer.Add((5,2))
237        self.sizer_set_smearer.Add( self.sizer_new_smear,0, wx.CENTER, 10 )
238       
239        # add all to chi2 sizer
240        sizer_smearer_box.Add(self.sizer_set_smearer)   
241           
242        sizer_chi2.Add(sizer_smearer_box)
243        """
244        sizer_chi2.Add((-1,5))
245        """
246        # hide all smear messages and textctrl
247        self._hide_all_smear_info()
248        """
249        # get smear_selection
250        self.current_smearer= smear_selection( self.data, self.model )
251        """
252        # Show only the relevant smear messages, etc
253        if self.current_smearer == None:
254            if not is_2Ddata:
255                self.smear_description_none.Show(True)
256                self.enable_smearer.Disable() 
257            else:
258                self.smear_description_none.Show(True)
259                #self.smear_description_2d.Show(True)
260                #self.pinhole_smearer.Disable()
261                self.slit_smearer.Disable()   
262                #self.enable_smearer.Disable()
263            if self.data == None:
264                self.slit_smearer.Disable()
265                self.pinhole_smearer.Disable()
266                self.enable_smearer.Disable()
267        else: self._show_smear_sizer()
268        """
269        boxsizer_range.Add(self.sizer_set_masking)
270         #2D data? default
271        is_2Ddata = False
272       
273        #check if it is 2D data
274        if self.data.__class__.__name__ ==  "Data2D" or \
275                        self.enable2D:
276            is_2Ddata = True
277           
278        self.sizer5.Clear(True)
279     
280        self.qmin  = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
281                                          style=wx.TE_PROCESS_ENTER,
282                                    text_enter_callback = self._onQrangeEnter)
283        self.qmin.SetValue(str(self.qmin_x))
284        self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
285     
286        self.qmax  = self.ModelTextCtrl(self, -1,size=(_BOX_WIDTH,20),
287                                          style=wx.TE_PROCESS_ENTER,
288                                        text_enter_callback=self._onQrangeEnter)
289        self.qmax.SetValue(str(self.qmax_x))
290        self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
291       
292        id = wx.NewId()
293        self.reset_qrange =wx.Button(self,id,'Reset',size=(77,20))
294     
295        self.reset_qrange.Bind(wx.EVT_BUTTON, self.on_reset_clicked,id=id)
296        self.reset_qrange.SetToolTipString("Reset Q range to the default values")
297     
298        sizer_horizontal=wx.BoxSizer(wx.HORIZONTAL)
299        sizer= wx.GridSizer(2, 4, 2, 6)
300
301        self.btEditMask = wx.Button(self,wx.NewId(),'Editor', size=(88,23))
302        self.btEditMask.Bind(wx.EVT_BUTTON, self._onMask,id= self.btEditMask.GetId())
303        self.btEditMask.SetToolTipString("Edit Mask.")
304        self.EditMask_title = wx.StaticText(self, -1, ' Masking(2D)')
305
306        sizer.Add(wx.StaticText(self, -1, 'Q range'))     
307        sizer.Add(wx.StaticText(self, -1, ' Min[1/A]'))
308        sizer.Add(wx.StaticText(self, -1, ' Max[1/A]'))
309        sizer.Add(self.EditMask_title)
310        #sizer.Add(wx.StaticText(self, -1, ''))
311        sizer.Add(self.reset_qrange)   
312        sizer.Add(self.qmin)
313        sizer.Add(self.qmax)
314        #sizer.Add(self.theory_npts_tcrtl)
315        sizer.Add(self.btEditMask)
316        boxsizer_range.Add(sizer_chi2) 
317        boxsizer_range.Add((10,10))
318        boxsizer_range.Add(sizer)
319       
320        boxsizer_range.Add((10,15))
321        boxsizer_range.Add(sizer_fit)
322        if is_2Ddata:
323            self.btEditMask.Enable() 
324            self.EditMask_title.Enable() 
325        else:
326            self.btEditMask.Disable() 
327            self.EditMask_title.Disable()
328        """
329        ## save state
330        self.save_current_state()
331        """
332        self.sizer5.Add(boxsizer_range,0, wx.EXPAND | wx.ALL, 10)
333        self.sizer5.Layout()
334       
335    def _on_select_model(self, event=None): 
336        """
337        call back for model selection
338        """ 
339       
340        self.Show(False)   
341        self._on_select_model_helper() 
342        self.set_model_param_sizer(self.model)                   
343        if self.model is None:
344            self._set_bookmark_flag(False)
345            self._keep.Enable(False)
346            self._set_save_flag(False)
347        self.enable_disp.SetValue(False)
348        self.disable_disp.SetValue(True)
349        try:
350            self.set_dispers_sizer()
351        except:
352            pass
353        """
354        #self.btFit.SetFocus()
355        self.state.enable_disp = self.enable_disp.GetValue()
356        self.state.disable_disp = self.disable_disp.GetValue()
357        self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
358        self.state.slit_smearer = self.slit_smearer.GetValue()
359        """
360        self.state.structurecombobox = self.structurebox.GetCurrentSelection()
361        self.state.formfactorcombobox = self.formfactorbox.GetCurrentSelection()
362     
363        if self.model != None:
364            self._set_copy_flag(True)
365            self._set_paste_flag(True)
366            if self.data != None:
367                self._set_bookmark_flag(True)
368                self._keep.Enable(True)
369               
370            temp_smear = None
371            """
372            #self._set_save_flag(True)
373            # Reset smearer, model and data
374            self._set_smear(self.data)
375            try:
376                # update smearer sizer
377                self.onSmear(None)
378                temp_smear = None
379                if self.enable_smearer.GetValue():
380                    # Set the smearer environments
381                    temp_smear = self.smearer
382            except:
383                raise
384                ## error occured on chisqr computation
385                #pass
386            """
387            ## event to post model to fit to fitting plugins
388            (ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
389         
390            ## set smearing value whether or not
391            #    the data contain the smearing info
392            evt = ModelEventbox(model=self.model, 
393                                        smearer=temp_smear, 
394                                        qmin=float(self.qmin_x),
395                                        uid=self.uid,
396                                     qmax=float(self.qmax_x)) 
397   
398            self._manager._on_model_panel(evt=evt)
399            self.mbox_description.SetLabel("Model [%s]" % str(self.model.name))
400            self.state.model = self.model.clone()
401            self.state.model.name = self.model.name
402
403           
404        if event != None:
405            ## post state to fit panel
406            new_event = PageInfoEvent(page = self)
407            wx.PostEvent(self.parent, new_event) 
408            #update list of plugins if new plugin is available
409            if self.plugin_rbutton.GetValue():
410                temp = self.parent.update_model_list()
411                if temp:
412                    self.model_list_box = temp
413                    current_val = self.formfactorbox.GetValue()
414                    pos = self.formfactorbox.GetSelection()
415                    self._show_combox_helper()
416                    self.formfactorbox.SetSelection(pos)
417                    self.formfactorbox.SetValue(current_val)
418            self._onDraw(event=None)
419        else:
420            self._draw_model()
421        self.SetupScrolling()
422        self.Show(True)   
423       
424    def _update_paramv_on_fit(self):
425        """
426        make sure that update param values just before the fitting
427        """
428        #flag for qmin qmax check values
429        flag = True
430        self.fitrange = True
431        is_modified = False
432
433        #wx.PostEvent(self._manager.parent, StatusEvent(status=" \
434        #updating ... ",type="update"))
435
436        ##So make sure that update param values on_Fit.
437        #self._undo.Enable(True)
438        if self.model !=None:           
439            ##Check the values
440            self._check_value_enter( self.fittable_param ,is_modified)
441            self._check_value_enter( self.fixed_param ,is_modified)
442            self._check_value_enter( self.parameters ,is_modified)
443
444            # If qmin and qmax have been modified, update qmin and qmax and
445             # Here we should check whether the boundaries have been modified.
446            # If qmin and qmax have been modified, update qmin and qmax and
447            # set the is_modified flag to True
448            self.fitrange = self._validate_qrange(self.qmin, self.qmax)
449            if self.fitrange:
450                tempmin = float(self.qmin.GetValue())
451                if tempmin != self.qmin_x:
452                    self.qmin_x = tempmin
453                tempmax = float(self.qmax.GetValue())
454                if tempmax != self.qmax_x:
455                    self.qmax_x = tempmax
456                if tempmax == tempmin:
457                    flag = False   
458                temp_smearer = None
459                """
460                if not self.disable_smearer.GetValue():
461                    temp_smearer= self.current_smearer
462                    if self.slit_smearer.GetValue():
463                        flag = self.update_slit_smear()
464                    elif self.pinhole_smearer.GetValue():
465                        flag = self.update_pinhole_smear()
466                    else:
467                        self._manager.set_smearer(smearer=temp_smearer,
468                                                  uid=self.uid,
469                                                     qmin=float(self.qmin_x),
470                                                      qmax=float(self.qmax_x),
471                                                      draw=False)
472                elif not self._is_2D():
473                    self._manager.set_smearer(smearer=temp_smearer,
474                                              qmin=float(self.qmin_x),
475                                              uid=self.uid,
476                                                 qmax= float(self.qmax_x),
477                                                 draw=False)
478               
479                    if self.data != None:
480                        index_data = ((self.qmin_x <= self.data.x)&\
481                                      (self.data.x <= self.qmax_x))
482                        val = str(len(self.data.x[index_data==True]))
483                        self.Npts_fit.SetValue(val)
484                    else:
485                        # No data in the panel
486                        try:
487                            self.npts_x = float(self.Npts_total.GetValue())
488                        except:
489                            flag = False
490                            return flag
491                    flag = True
492                    """
493                if self._is_2D():
494                    # only 2D case set mask 
495                    flag = self._validate_Npts()
496                    if not flag:
497                        return flag
498            else: flag = False
499        else: 
500            flag = False
501
502        #For invalid q range, disable the mask editor and fit button, vs.   
503        if not self.fitrange:
504            #self.btFit.Disable()
505            if self._is_2D():
506                self.btEditMask.Disable()
507        else:
508            #self.btFit.Enable(True)
509            if self._is_2D() and  self.data != None:
510                self.btEditMask.Enable(True)
511
512        if not flag:
513            msg = "Cannot Plot or Fit :Must select a "
514            msg += " model or Fitting range is not valid!!!  "
515            wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
516       
517        self.save_current_state()
518   
519        return flag 
520    def save_current_state(self):
521        """
522        Currently no save option implemented for batch page
523        """
524        pass 
525    def save_current_state_fit(self):
526        """
527        Currently no save option implemented for batch page
528        """
529        pass
530    def set_data(self, data):
531        """
532        reset the current data
533        """
534        id = None
535        group_id = None
536        flag = False
537        if self.data is None and data is not None:
538            flag = True
539        if data is not None:
540            id = data.id
541            group_id = data.group_id
542            if self.data is not None:
543                flag = (data.id != self.data.id)
544        self.data = data
545        if self.data is None:
546            data_min = ""
547            data_max = ""
548            data_name = ""
549            self._set_bookmark_flag(False)
550            self._keep.Enable(False)
551            self._set_save_flag(False)
552        else:
553            if self.model != None:
554                self._set_bookmark_flag(True)
555                self._keep.Enable(True)
556            self._set_save_flag(True)
557            self._set_preview_flag(True)
558            """
559            self._set_smear(data)
560            # more disables for 2D
561            if self.data.__class__.__name__ ==  "Data2D" or \
562                        self.enable2D:
563                self.slit_smearer.Disable()
564                self.pinhole_smearer.Enable(True)
565                self.default_mask = copy.deepcopy(self.data.mask)
566            else:
567                self.slit_smearer.Enable(True)
568                self.pinhole_smearer.Enable(True)     
569             """   
570            self.formfactorbox.Enable()
571            self.structurebox.Enable()
572            data_name = self.data.name
573            #set maximum range for x in linear scale
574            if not hasattr(self.data,"data"): #Display only for 1D data fit
575                # Minimum value of data   
576                data_min = min(self.data.x)
577                # Maximum value of data 
578                data_max = max(self.data.x)
579                """
580                #number of total data points
581                self.Npts_total.SetValue(str(len(self.data.x)))
582                #default:number of data points selected to fit
583                self.Npts_fit.SetValue(str(len(self.data.x)))
584                """
585                self.btEditMask.Disable() 
586                self.EditMask_title.Disable()
587            else:
588               
589                ## Minimum value of data
590                data_min = 0
591                x = max(math.fabs(self.data.xmin), math.fabs(self.data.xmax)) 
592                y = max(math.fabs(self.data.ymin), math.fabs(self.data.ymax))
593                ## Maximum value of data 
594                data_max = math.sqrt(x*x + y*y)
595                """
596                #number of total data points
597                self.Npts_total.SetValue(str(len(self.data.data)))
598                #default:number of data points selected to fit
599                self.Npts_fit.SetValue(str(len(self.data.data)))
600                """
601                self.btEditMask.Enable() 
602                self.EditMask_title.Enable() 
603        """
604        self.Npts_total.SetEditable(False)
605        self.Npts_total.SetBackgroundColour(\
606                                    self.GetParent().GetBackgroundColour())
607       
608        self.Npts_total.Bind(wx.EVT_MOUSE_EVENTS, self._npts_click)
609        #self.Npts_total.Disable()
610        """
611        self.dataSource.SetValue(data_name)
612        self.qmin_x = data_min
613        self.qmax_x = data_max
614        #self.minimum_q.SetValue(str(data_min))
615        #self.maximum_q.SetValue(str(data_max))
616        self.qmin.SetValue(str(data_min))
617        self.qmax.SetValue(str(data_max))
618        self.qmin.SetBackgroundColour("white")
619        self.qmax.SetBackgroundColour("white")
620        self.state.data = data
621        self.state.qmin = self.qmin_x
622        self.state.qmax = self.qmax_x
623       
624        #update model plot with new data information
625        if flag:
626            #set model view button
627            if self.data.__class__.__name__ == "Data2D":
628                self.enable2D = True
629                self.model_view.SetLabel("2D Mode")
630            else:
631                self.enable2D = False
632                self.model_view.SetLabel("1D Mode")
633               
634            self.model_view.Disable()
635           
636            #replace data plot on combo box selection
637            #by removing the previous selected data
638            #wx.PostEvent(self._manager.parent, NewPlotEvent(action="remove",
639            #                                        group_id=group_id, id=id))
640            wx.PostEvent(self._manager.parent, 
641                             NewPlotEvent(group_id=group_id,
642                                               action="delete"))
643            #plot the current selected data
644            wx.PostEvent(self._manager.parent, NewPlotEvent(plot=self.data, 
645                                                           title=str(self.data.title)))
646            self._manager.store_data(uid=self.uid, data=data,
647                                     data_list=self.data_list,
648                                      caption=self.window_name)
649            self._draw_model()
650   
651
652       
653class BGTextCtrl(wx.TextCtrl):
654    """
655    Text control used to display outputs.
656    No editing allowed. The background is
657    grayed out. User can't select text.
658    """
659    def __init__(self, *args, **kwds):
660        wx.TextCtrl.__init__(self, *args, **kwds)
661        self.SetEditable(False)
662        self.SetBackgroundColour(self.GetParent().parent.GetBackgroundColour())
663       
664        # Bind to mouse event to avoid text highlighting
665        # The event will be skipped once the call-back
666        # is called.
667        self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
668       
669    def _click(self, event):
670        """
671        Prevent further handling of the mouse event
672        by not calling Skip().
673        """ 
674        pass
675 
Note: See TracBrowser for help on using the repository browser.