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

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

model page change on interface

  • Property mode set to 100644
File size: 16.2 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       
166    def onClose(self,event):
167        """ close the page associated with this panel"""
168        self.GrandParent.onClose()
169       
170    def set_owner(self,owner):
171        """
172            set owner of fitpage
173            @param owner: the class responsible of plotting
174        """
175        self.event_owner=owner   
176   
177 
178    def set_manager(self, manager):
179        """
180             set panel manager
181             @param manager: instance of plugin fitting
182        """
183        self.manager = manager
184       
185    def onModel2D(self, event):
186        """
187         call manager to plot model in 2D
188        """
189        # If the 2D display is not currently enabled, plot the model in 2D
190        # and set the enable2D flag.
191        if self.enable2D==False:
192            self.enable2D=True
193            self.manager.draw_model(model=self.model,
194                                    name=self.model.name,
195                                    description=None,
196                                     enable2D=self.enable2D,
197                                     qmin=float(self.qmin),
198                                    qmax=float(self.qmax))
199           
200    def populate_box(self, dict):
201        """
202            Populate each combox box of each page
203            @param page: the page to populate
204        """
205        id=0
206        self.model_list_box=dict
207        list_name=[]
208        for item in  self.model_list_box.itervalues():
209            name = item.__name__
210            if hasattr(item, "name"):
211                name = item.name
212            list_name.append(name)
213        list_name.sort()   
214        for name in list_name:
215            self.modelbox.Insert(name,int(id))
216            id+=1
217        wx.EVT_COMBOBOX(self.modelbox,-1, self._on_select_model) 
218        return 0
219   
220 
221   
222    def _on_select_model(self,event):
223        """
224            react when a model is selected from page's combo box
225            post an event to its owner to draw an appropriate theory
226        """
227        #print "modelpage: self.model_list_box ",self.model_list_box
228        for item in self.model_list_box.itervalues():
229            name = item.__name__
230            if hasattr(item, "name"):
231                name = item.name
232            if name ==event.GetString():
233                model=item()
234                #print "fitpage: _on_select_model model name",name ,event.GetString()
235                self.model= model
236                self.set_panel(model)
237                print "name in model page", name,event.GetString()
238                self.name= name
239                self.manager.draw_model(model, name)
240               
241            self.model_view.SetFocus()
242           
243    def get_model_box(self): 
244        """ return reference to combox box self.model"""
245        return self.modelbox
246
247   
248    def get_param_list(self):
249        """
250            @return self.param_toFit: list containing  references to TextCtrl
251            checked.Theses TextCtrl will allow reference to parameters to fit.
252            @raise: if return an empty list of parameter fit will nnote work
253            properly so raise ValueError,"missing parameter to fit"
254        """
255        if self.param_toFit !=[]:
256            return self.param_toFit
257        else:
258            raise ValueError,"missing parameter to fit"
259       
260       
261    def set_panel(self,model):
262        """
263            Build the panel from the model content
264            @param model: the model selected in combo box for fitting purpose
265        """
266       
267        self.sizer2.Clear(True)
268        self.sizer1.Clear(True)
269        self.sizer5.Clear(True)
270        self.parameters = []
271        self.param_toFit=[]
272        self.model = model
273        keys = self.model.getParamList()
274        keys.sort()
275        description=None
276        if hasattr(self.model,'description'):
277            description =model.description
278        disp_list=self.model.getDispParamList()
279        ip=0
280        iq=1
281        if len(disp_list)>0:
282            disp = wx.StaticText(self, -1, 'Dispersion')
283            self.sizer5.Add(disp,( iq, ip),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
284            ip += 1 
285            values = wx.StaticText(self, -1, 'Values')
286            self.sizer5.Add(values,( iq, ip),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
287           
288        disp_list.sort()
289        #print "went here",self.model.name,model.description
290        iy = 1
291        ix = 0
292        self.cb0 = wx.StaticText(self, -1,'Model Description:')
293        self.sizer1.Add(self.cb0,(iy, ix),(1,1),\
294                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
295        iy += 1
296       
297        self.cb01 = wx.StaticText(self, -1,str(description),style=wx.ALIGN_LEFT)
298        self.cb01.Wrap(400) 
299        #self.cb01 = wx.StaticText(self, -1,str(description),(45, 25),style=wx.ALIGN_LEFT)
300       
301        self.sizer1.Add(self.cb01,(iy, ix),(1,1),\
302                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
303        ix = 0
304        iy = 1
305        self.cb1 = wx.StaticText(self, -1,'Parameters')
306        self.sizer2.Add(self.cb1,(iy, ix),(1,1),\
307                          wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
308        ix +=1
309        self.text2_2 = wx.StaticText(self, -1, 'Values')
310        self.sizer2.Add(self.text2_2,(iy, ix),(1,1),\
311                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
312        ix +=1
313        self.text2_4 = wx.StaticText(self, -1, 'Units')
314        self.sizer2.Add(self.text2_4,(iy, ix),(1,1),\
315                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
316        self.text2_4.Hide()
317        for item in keys:
318            if not item in disp_list:
319                iy += 1
320                ix = 0
321                cb=wx.StaticText(self, -1, item)
322                self.sizer2.Add( cb,( iy, ix),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
323                ix += 1
324                value= self.model.getParam(item)
325                ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
326                ctl1.SetValue(str (format_number(value)))
327                ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
328                ctl1.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
329                self.sizer2.Add(ctl1, (iy,ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
330                ix +=1
331               
332                # Units
333                try:
334                    units = wx.StaticText(self, -1, self.model.details[item][0], style=wx.ALIGN_LEFT)
335                except:
336                    units = wx.StaticText(self, -1, "", style=wx.ALIGN_LEFT)
337                self.sizer2.Add(units, (iy,ix),(1,1),  wx.EXPAND|wx.ADJUST_MINSIZE, 0)
338            else:
339                ip = 0
340                iq += 1
341                cb = wx.CheckBox(self, -1, item, (10, 10))
342                cb.SetValue(False)
343                self.sizer5.Add( cb,( iq, ip),(1,1),  wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
344                wx.EVT_CHECKBOX(self, cb.GetId(), self._on_select_model)
345               
346                ip += 1
347                value= self.model.getParam(item)
348                ctl1 = wx.TextCtrl(self, -1, size=(_BOX_WIDTH,20), style=wx.TE_PROCESS_ENTER)
349                ctl1.SetValue(str (format_number(value)))
350                ctl1.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
351                ctl1.Bind(wx.EVT_TEXT_ENTER,self._onparamEnter)
352                self.sizer5.Add(ctl1, (iq,ip),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0)
353            #save data
354            self.parameters.append([cb,ctl1])
355        iy+=1
356        self.sizer2.Add((20,20),(iy,ix),(1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
357       
358        #Display units text on panel
359        for item in keys:   
360            if self.model.details[item][0]!='':
361                self.text2_4.Show()
362                break
363            else:
364                self.text2_4.Hide()
365        self.vbox.Layout()
366        self.GrandParent.GetSizer().Layout()
367       
368       
369    def _onparamEnter(self,event):
370        """
371            when enter value on panel redraw model according to changed
372        """
373        self.set_model_parameter()
374       
375    def set_model_parameter(self):
376        if len(self.parameters) !=0 and self.model !=None:
377            # Flag to register when a parameter has changed.
378            is_modified = False
379            for item in self.parameters:
380                try:
381                     name=str(item[0].GetLabelText())
382                     value= float(item[1].GetValue())
383                     # If the value of the parameter has changed,
384                     # update the model and set the is_modified flag
385                     if value != self.model.getParam(name):
386                         self.model.setParam(name,value)
387                         is_modified = True
388                except:
389                     wx.PostEvent(self.parent.GrandParent, StatusEvent(status=\
390                            "Model Drawing  Error:wrong value entered : %s"% sys.exc_value))
391           
392            # Here we should check whether the boundaries have been modified.
393            # If qmin and qmax have been modified, update qmin and qmax and
394            # set the is_modified flag to True
395            if float(self.xmin.GetValue()) != self.qmin:
396                self.qmin = float(self.xmin.GetValue())
397                is_modified = True
398            if float(self.xmax.GetValue()) != self.qmax:
399                self.qmax = float(self.xmax.GetValue())
400                is_modified = True
401            if float(self.npts.GetValue()) !=  self.num_points:
402                self.qmax = float(self.npts.GetValue())
403                is_modified = True
404           
405            if is_modified:
406                self.manager.draw_model(self.model, self.model.name, 
407                                        qmin=self.qmin, qmax=self.qmax,
408                                        qstep= self.num_points,
409                                        enable2D=self.enable2D)
410            #self.manager.draw_model(self,model,description=None,
411            # enable1D=True,qmin=None,qmax=None, qstep=None)
412           
413           
414           
415           
416             
Note: See TracBrowser for help on using the repository browser.