source: sasview/sansview/perspectives/fitting/modelpage.py @ cfc68540

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 cfc68540 was cfc68540, checked in by Mathieu Doucet <doucetm@…>, 15 years ago

Fixed the Q range selection problem when a model is first plotted.

  • Property mode set to 100644
File size: 16.5 KB
Line 
1import sys
2import wx
3import wx.lib
4import numpy
5import copy
6
7from sans.guicomm.events import StatusEvent   
8(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
9_BOX_WIDTH = 80
10
11def format_number(value, high=False):
12    """
13        Return a float in a standardized, human-readable formatted string
14    """
15    try: 
16        value = float(value)
17    except:
18        print "returning 0"
19        return "0"
20   
21    if high:
22        return "%-6.4g" % value
23    else:
24        return "%-5.3g" % value
25
26   
27class ModelPage(wx.ScrolledWindow):
28    """
29        FitPanel class contains fields allowing to display results when
30        fitting  a model and one data
31        @note: For Fit to be performed the user should check at least one parameter
32        on fit Panel window.
33 
34    """
35    ## Internal name for the AUI manager
36    window_name = "Fit page"
37    ## Title to appear on top of the window
38    window_caption = "Fit Page"
39   
40   
41    def __init__(self, parent,model,name, *args, **kwargs):
42        wx.ScrolledWindow.__init__(self, parent, *args, **kwargs)
43        """
44            Initialization of the Panel
45        """
46        #self.scroll = wx.ScrolledWindow(self)
47        self.manager = None
48        self.parent  = parent
49        self.event_owner=None
50        #panel interface
51        self.vbox  = wx.BoxSizer(wx.VERTICAL)
52        self.sizer3 = wx.GridBagSizer(5,5)
53        self.sizer1 = wx.GridBagSizer(5,5)
54        self.sizer2 = wx.GridBagSizer(5,5)
55        self.sizer4 = wx.GridBagSizer(5,5)
56        self.sizer5 = wx.GridBagSizer(5,5)
57        self.static_line_1 = wx.StaticLine(self, -1)
58        self.modelbox = wx.ComboBox(self, -1)
59        id = wx.NewId()
60        self.vbox.Add(self.sizer3)
61        self.vbox.Add(self.sizer1)
62        self.vbox.Add(self.sizer2)
63        self.vbox.Add(self.static_line_1, 0, wx.EXPAND, 0)
64        self.vbox.Add(self.sizer5)
65        self.vbox.Add(self.sizer4)
66       
67        id = wx.NewId()
68        self.btClose =wx.Button(self,id,'Close')
69        self.btClose.Bind(wx.EVT_BUTTON, self.onClose,id=id)
70        self.btClose.SetToolTipString("Close page.")
71        ix = 1
72        iy = 1 
73        self.sizer4.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\
74                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
75        ix += 1
76        self.sizer4.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1),\
77                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
78        ix += 1
79        self.sizer4.Add(wx.StaticText(self, -1, 'Npts'),(iy, ix),(1,1),\
80                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
81        ix = 0
82        iy += 1
83        self.sizer4.Add(wx.StaticText(self, -1, 'x range'),(iy, ix),(1,1),\
84                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
85        ## Q range
86        self.qmin= 0.001
87        self.qmax= 0.1
88        self.num_points= 100
89       
90        ix += 1
91        self.xmin    = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
92        self.xmin.SetValue(format_number(self.qmin))
93        self.xmin.SetToolTipString("Minimun value of x in linear scale.")
94        self.xmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
95        self.xmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
96        self.sizer4.Add(self.xmin,(iy, ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
97       
98       
99        ix += 1
100        self.xmax    = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
101        self.xmax.SetValue(format_number(self.qmax))
102        self.xmax.SetToolTipString("Maximum value of x in linear scale.")
103        self.xmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
104        self.xmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
105        self.sizer4.Add(self.xmax,(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
106        ix += 1
107        self.npts    = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
108        self.npts.SetValue(format_number(self.num_points))
109        self.npts.SetToolTipString("Number of point to plot.")
110        self.npts.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
111        self.npts.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
112       
113        self.sizer4.Add(self.npts,(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
114        ix = 0
115        iy += 1
116        self.sizer4.Add((20,20),(iy, ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
117        ix +=3
118        self.sizer4.Add( self.btClose,(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
119        ix = 0
120        iy = 1
121        self.sizer3.Add(wx.StaticText(self,-1,'Model'),(iy,ix),(1,1)\
122                  , wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
123        ix += 1
124        self.sizer3.Add(self.modelbox,(iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
125        #ix = 0
126        #iy += 1
127        ix += 1
128        id = wx.NewId()
129        self.model_view =wx.Button(self,id,'View 2D')
130        self.model_view.Bind(wx.EVT_BUTTON, self.onModel2D,id=id)
131        self.model_view.SetToolTipString("View model in 2D")
132        self.sizer3.Add(self.model_view,(iy,ix),(1,1),\
133                   wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
134        # contains link between  model ,all its parameters, and panel organization
135        self.parameters=[]
136        #contains link between a model and selected parameters to fit
137        self.param_toFit=[]
138        # model on which the fit would be performed
139        self.model=model
140        try:
141            #print"init modelpage",model.name
142            self.set_panel(model)
143        except:
144            raise
145        # preview selected model name
146        self.prevmodel_name=name
147        #print "model view prev_model",name
148        self.modelbox.SetValue(self.prevmodel_name)
149        # flag to check if the user has selected a new model in the combox box
150        self.model_hasChanged=False
151        #dictionary of model name and model class
152        self.model_list_box={}
153       
154        #enable model 2D draw
155        self.enable2D= False
156        # Data1D to make a deep comparison between 2 Data1D for checking data
157        #change
158        self.vbox.Layout()
159        self.vbox.Fit(self) 
160       
161        self.SetSizer(self.vbox)
162        self.SetScrollbars(20,20,55,40)
163        self.Centre()
164       
165    def set_range(self, qmin, qmax, npts):
166        """
167            Set the range for the plotted models
168            @param qmin: minimum Q
169            @param qmax: maximum Q
170            @param npts: number of Q bins
171        """
172        # Set the data members
173        self.qmin = qmin
174        self.qmax = qmax
175        self.num_points = npts
176       
177        # Set the controls
178        self.xmin.SetValue(format_number(self.qmin))
179        self.xmax.SetValue(format_number(self.qmax))
180        self.npts.SetValue(format_number(self.num_points))
181       
182    def onClose(self,event):
183        """ close the page associated with this panel"""
184        self.GrandParent.onClose()
185       
186    def set_owner(self,owner):
187        """
188            set owner of fitpage
189            @param owner: the class responsible of plotting
190        """
191        self.event_owner=owner   
192   
193 
194    def set_manager(self, manager):
195        """
196             set panel manager
197             @param manager: instance of plugin fitting
198        """
199        self.manager = manager
200       
201    def onModel2D(self, event):
202        """
203         call manager to plot model in 2D
204        """
205        # If the 2D display is not currently enabled, plot the model in 2D
206        # and set the enable2D flag.
207        if self.enable2D==False:
208            self.enable2D=True
209            self._draw_model()
210           
211    def populate_box(self, dict):
212        """
213            Populate each combox box of each page
214            @param page: the page to populate
215        """
216        id=0
217        self.model_list_box=dict
218        list_name=[]
219        for item in  self.model_list_box.itervalues():
220            name = item.__name__
221            if hasattr(item, "name"):
222                name = item.name
223            list_name.append(name)
224        list_name.sort()   
225        for name in list_name:
226            self.modelbox.Insert(name,int(id))
227            id+=1
228        wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model) 
229        return 0
230   
231 
232   
233    def _on_select_model(self,event):
234        """
235            react when a model is selected from page's combo box
236            post an event to its owner to draw an appropriate theory
237        """
238        self.model_view.SetFocus()
239        for item in self.model_list_box.itervalues():
240            name = item.__name__
241            if hasattr(item, "name"):
242                name = item.name
243            if name ==event.GetString():
244                model=item()
245                self.model= model
246                self.set_panel(model)
247                print "name in model page", name,event.GetString()
248                self.name= name
249                #self.manager.draw_model(model, name)
250                self._draw_model()
251           
252           
253    def get_model_box(self): 
254        """ return reference to combox box self.model"""
255        return self.modelbox
256
257   
258    def get_param_list(self):
259        """
260            @return self.param_toFit: list containing  references to TextCtrl
261            checked.Theses TextCtrl will allow reference to parameters to fit.
262            @raise: if return an empty list of parameter fit will nnote work
263            properly so raise ValueError,"missing parameter to fit"
264        """
265        if self.param_toFit !=[]:
266            return self.param_toFit
267        else:
268            raise ValueError,"missing parameter to fit"
269       
270       
271    def set_panel(self,model):
272        """
273            Build the panel from the model content
274            @param model: the model selected in combo box for fitting purpose
275        """
276       
277        self.sizer2.Clear(True)
278        self.sizer1.Clear(True)
279        self.sizer5.Clear(True)
280        self.parameters = []
281        self.param_toFit=[]
282        self.model = model
283        keys = self.model.getParamList()
284        keys.sort()
285        description=None
286        if hasattr(self.model,'description'):
287            description =model.description
288        disp_list=self.model.getDispParamList()
289        ip=0
290        iq=1
291        if len(disp_list)>0:
292            disp = wx.StaticText(self, -1, 'Dispersion')
293            self.sizer5.Add(disp,( iq, ip),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
294            ip += 1 
295            values = wx.StaticText(self, -1, 'Values')
296            self.sizer5.Add(values,( iq, ip),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
297           
298        disp_list.sort()
299        #print "went here",self.model.name,model.description
300        iy = 1
301        ix = 0
302        self.cb0 = wx.StaticText(self, -1,'Model Description:')
303        self.sizer1.Add(self.cb0,(iy, ix),(1,1),\
304                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
305        iy += 1
306       
307        self.cb01 = wx.StaticText(self, -1,str(description),style=wx.ALIGN_LEFT)
308        self.cb01.Wrap(400) 
309        #self.cb01 = wx.StaticText(self, -1,str(description),(45, 25),style=wx.ALIGN_LEFT)
310       
311        self.sizer1.Add(self.cb01,(iy, ix),(1,1),\
312                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
313        ix = 0
314        iy = 1
315        self.cb1 = wx.StaticText(self, -1,'Parameters')
316        self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
317                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
318        ix +=1
319        self.text2_2 = wx.StaticText(self, -1, 'Values')
320        self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
321                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
322        ix +=1
323        self.text2_4 = wx.StaticText(self, -1, 'Units')
324        self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
325                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
326        self.text2_4.Hide()
327        for item in keys:
328            if not item in disp_list:
329                iy += 1
330                ix = 0
331                cb=wx.StaticText(self, -1, item)
332                self.sizer2.Add( cb,( iy, ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
333                ix += 1
334                value= self.model.getParam(item)
335                ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
336                ctl1.SetValue(str (format_number(value)))
337                ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
338                ctl1.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
339                self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
340                ix +=1
341               
342                # Units
343                try:
344                    units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
345                except:
346                    units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
347                self.sizer2.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
348            else:
349                ip = 0
350                iq += 1
351                cb = wx.CheckBox(self, -1, item, (10, 10))
352                cb.SetValue(False)
353                self.sizer5.Add( cb,( iq, ip),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
354                wx.EVT_CHECKBOX(self, cb.GetId(), self._on_select_model)
355               
356                ip += 1
357                value= self.model.getParam(item)
358                ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
359                ctl1.SetValue(str (format_number(value)))
360                ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
361                ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
362                self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
363            #save data
364            self.parameters.append([cb,ctl1])
365        iy+=1
366        self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
367       
368        #Display units text on panel
369        for item in keys:   
370            if self.model.details[item][0]!='':
371                self.text2_4.Show()
372                break
373            else:
374                self.text2_4.Hide()
375        self.vbox.Layout()
376        self.GrandParent.GetSizer().Layout()
377       
378       
379    def _onparamEnter(self,event):
380        """
381            when enter value on panel redraw model according to changed
382        """
383        self.set_model_parameter()
384       
385    def set_model_parameter(self):
386        if len(self.parameters) !=0 and self.model !=None:
387            # Flag to register when a parameter has changed.
388            is_modified = False
389            for item in self.parameters:
390                try:
391                     name=str(item[0].GetLabelText())
392                     value= float(item[1].GetValue())
393                     # If the value of the parameter has changed,
394                     # update the model and set the is_modified flag
395                     if value != self.model.getParam(name):
396                         self.model.setParam(name,value)
397                         is_modified = True
398                except:
399                     wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
400                            "Model Drawing  Error:wrong value entered : %s"% sys.exc_value))
401           
402            # Here we should check whether the boundaries have been modified.
403            # If qmin and qmax have been modified, update qmin and qmax and
404            # set the is_modified flag to True
405            if float(self.xmin.GetValue()) != self.qmin:
406                self.qmin = float(self.xmin.GetValue())
407                is_modified = True
408            if float(self.xmax.GetValue()) != self.qmax:
409                self.qmax = float(self.xmax.GetValue())
410                is_modified = True
411           
412            if float(self.npts.GetValue()) !=  self.num_points:
413                self.num_points = float(self.npts.GetValue())
414                is_modified = True
415         
416            if is_modified:
417                self._draw_model()           
418           
419    def _draw_model(self):
420        """
421            Method to draw or refresh a plotted model.
422            The method will use the data member from the model page
423            to build a call to the fitting perspective manager.
424           
425            [Note to coder: This way future changes will be done in only one place.]
426        """
427        self.manager.draw_model(self.model, self.model.name, 
428                                qmin=self.qmin, qmax=self.qmax,
429                                qstep= self.num_points,
430                                enable2D=self.enable2D)
431   
432           
433           
434             
Note: See TracBrowser for help on using the repository browser.