source: sasview/src/sas/perspectives/fitting/simfitpage.py @ f21d496

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 f21d496 was acf8e4a5, checked in by Paul Kienzle <pkienzle@…>, 10 years ago

reference BumpsFit? directly and remove fit engine selection layer

  • Property mode set to 100644
File size: 36.0 KB
Line 
1"""
2    Simultaneous fit page
3"""
4import sys, re, string, wx
5import wx.lib.newevent
6from sas.guiframe.events import StatusEvent
7from sas.guiframe.panel_base import PanelBase
8from wx.lib.scrolledpanel import ScrolledPanel
9from sas.guiframe.events import PanelOnFocusEvent
10#Control panel width
11if sys.platform.count("darwin") == 0:
12    PANEL_WID = 420
13    FONT_VARIANT = 0
14else:
15    PANEL_WID = 490
16    FONT_VARIANT = 1
17
18
19def get_fittableParam(model):
20    """
21    return list of fittable parameters name of a model
22
23    :param model: the model used
24
25    """
26    fittable_param = []
27    for item in model.getParamList():
28        if not item  in model.getDispParamList():
29            if not item in model.non_fittable:
30                fittable_param.append(item)
31
32    for item in model.fixed:
33        fittable_param.append(item)
34
35    return fittable_param
36
37
38class SimultaneousFitPage(ScrolledPanel, PanelBase):
39    """
40    Simultaneous fitting panel
41    All that needs to be defined are the
42    two data members window_name and window_caption
43    """
44    ## Internal name for the AUI manager
45    window_name = "simultaneous Fit page"
46    ## Title to appear on top of the window
47    window_caption = "Simultaneous Fit Page"
48
49    def __init__(self, parent, page_finder={}, id= -1, batch_on=False,
50                     *args, **kwargs):
51        ScrolledPanel.__init__(self, parent, id=id,
52                               style=wx.FULL_REPAINT_ON_RESIZE,
53                               *args, **kwargs)
54        PanelBase.__init__(self, parent)
55        """
56        Simultaneous page display
57        """
58        self.SetupScrolling()
59        ##Font size
60        self.SetWindowVariant(variant=FONT_VARIANT)
61        self.uid = wx.NewId()
62        self.parent = parent
63        self.batch_on = batch_on
64        ## store page_finder
65        self.page_finder = page_finder
66        ## list contaning info to set constraint
67        ## look like self.constraint_dict[page_id]= page
68        self.constraint_dict = {}
69        ## item list
70        # self.constraints_list=[combobox1, combobox2,=,textcrtl, button ]
71        self.constraints_list = []
72        ## list of current model
73        self.model_list = []
74        ## selected mdoel to fit
75        self.model_toFit = []
76        ## number of constraint
77        self.nb_constraint = 0
78        self.model_cbox_left = None
79        self.model_cbox_right = None
80        self.uid = wx.NewId()
81        ## draw page
82        self.define_page_structure()
83        self.draw_page()
84        self.set_layout()
85        self._set_save_flag(False)
86
87    def define_page_structure(self):
88        """
89        Create empty sizer for a panel
90        """
91        self.vbox = wx.BoxSizer(wx.VERTICAL)
92        self.sizer1 = wx.BoxSizer(wx.VERTICAL)
93        self.sizer2 = wx.BoxSizer(wx.VERTICAL)
94        self.sizer3 = wx.BoxSizer(wx.VERTICAL)
95
96        self.sizer1.SetMinSize((PANEL_WID, -1))
97        self.sizer2.SetMinSize((PANEL_WID, -1))
98        self.sizer3.SetMinSize((PANEL_WID, -1))
99        self.vbox.Add(self.sizer1)
100        self.vbox.Add(self.sizer2)
101        self.vbox.Add(self.sizer3)
102
103    def set_scroll(self):
104        """
105        """
106        self.Layout()
107
108    def set_layout(self):
109        """
110        layout
111        """
112        self.vbox.Layout()
113        self.vbox.Fit(self)
114        self.SetSizer(self.vbox)
115        self.set_scroll()
116        self.Centre()
117
118    def onRemove(self, event):
119        """
120        Remove constraint fields
121        """
122        if len(self.constraints_list) == 1:
123            self.hide_constraint.SetValue(True)
124            self._hide_constraint()
125            return
126        if len(self.constraints_list) == 0:
127            return
128        for item in self.constraints_list:
129            length = len(item)
130            if event.GetId() == item[length - 2].GetId():
131                sizer = item[length - 1]
132                sizer.Clear(True)
133                self.sizer_constraints.Remove(sizer)
134                #self.SetScrollbars(20,20,25,65)
135                self.constraints_list.remove(item)
136                self.nb_constraint -= 1
137                self.sizer2.Layout()
138                self.Layout()
139                break
140
141        #self._onAdd_constraint(None)
142
143    def onFit(self, event):
144        """
145        signal for fitting
146
147        """
148        flag = False
149        # check if the current page a simultaneous fit page or a batch page
150        if self == self._manager.sim_page:
151            flag = (self._manager.sim_page.uid == self.uid)
152
153        ## making sure all parameters content a constraint
154        if not self.batch_on and self.show_constraint.GetValue():
155            if not self._set_constraint():
156                return
157        ## model was actually selected from this page to be fit
158        if len(self.model_toFit) >= 1:
159            self.manager._reset_schedule_problem(value=0)
160            for item in self.model_list:
161                if item[0].GetValue():
162                    self.manager.schedule_for_fit(value=1, uid=item[2])
163            try:
164                if not self.manager.onFit(uid=self.uid):
165                    return
166            except:
167                msg = "Select at least one parameter to fit in the FitPages."
168                wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
169        else:
170            msg = "Select at least one model check box to fit "
171            wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
172
173    def set_manager(self, manager):
174        """
175        set panel manager
176
177        :param manager: instance of plugin fitting
178
179        """
180        self.manager = manager
181
182    def check_all_model_name(self, event=None):
183        """
184        check all models names
185        """
186        self.model_toFit = []
187        if self.cb1.GetValue() == True:
188            for item in self.model_list:
189                if item[0].IsEnabled():
190                    item[0].SetValue(True)
191                    self.model_toFit.append(item)
192
193            ## constraint info
194            self._store_model()
195            if not self.batch_on:
196                ## display constraint fields
197                if self.show_constraint.GetValue() and\
198                                 len(self.constraints_list) == 0:
199                    self._show_all_constraint()
200                    self._show_constraint()
201        else:
202            for item in self.model_list:
203                item[0].SetValue(False)
204
205            self.model_toFit = []
206            if not self.batch_on:
207                ##constraint info
208                self._hide_constraint()
209
210        self._update_easy_setup_cb()
211        self.Layout()
212        self.Refresh()
213
214    def check_model_name(self, event):
215        """
216        Save information related to checkbox and their states
217        """
218        self.model_toFit = []
219        cbox = event.GetEventObject()
220        for item in self.model_list:
221            if item[0].GetValue() == True:
222                self.model_toFit.append(item)
223            else:
224                if item in self.model_toFit:
225                    self.model_toFit.remove(item)
226                    self.cb1.SetValue(False)
227
228        ## display constraint fields
229        if len(self.model_toFit) >= 1:
230            self._store_model()
231            if not self.batch_on and self.show_constraint.GetValue() and\
232                             len(self.constraints_list) == 0:
233                self._show_all_constraint()
234                self._show_constraint()
235
236        elif len(self.model_toFit) < 1:
237            ##constraint info
238            self._hide_constraint()
239
240        self._update_easy_setup_cb()
241        ## set the value of the main check button
242        if len(self.model_list) == len(self.model_toFit):
243            self.cb1.SetValue(True)
244            self.Layout()
245            return
246        else:
247            self.cb1.SetValue(False)
248            self.Layout()
249
250    def _update_easy_setup_cb(self):
251        """
252        Update easy setup combobox on selecting a model
253        """
254        if self.model_cbox_left != None and self.model_cbox_right != None:
255            try:
256                # when there is something
257                self.model_cbox_left.Clear()
258                self.model_cbox_right.Clear()
259                self.model_cbox.Clear()
260            except:
261                # when there is nothing
262                pass
263            #for id, model in self.constraint_dict.iteritems():
264            for item in self.model_toFit:
265                model = item[3]
266                ## check if all parameters have been selected for constraint
267                ## then do not allow add constraint on parameters
268                if str(model.name) not in self.model_cbox_left.GetItems():
269                    self.model_cbox_left.Append(str(model.name), model)
270                if str(model.name) not in self.model_cbox_right.GetItems():
271                    self.model_cbox_right.Append(str(model.name), model)
272                if str(model.name) not in self.model_cbox.GetItems():
273                    self.model_cbox.Append(str(model.name), model)
274            self.model_cbox_left.SetSelection(0)
275            self.sizer2.Layout()
276            self.sizer3.Layout()
277
278    def draw_page(self):
279        """
280        Draw a sizer containing couples of data and model
281        """
282        self.model_list = []
283        self.model_toFit = []
284        self.constraints_list = []
285        self.constraint_dict = {}
286        self.nb_constraint = 0
287        self.model_cbox_left = None
288        self.model_cbox_right = None
289
290        if len(self.model_list) > 0:
291            for item in self.model_list:
292                item[0].SetValue(False)
293                self.manager.schedule_for_fit(value=0, uid=item[2])
294
295        self.sizer1.Clear(True)
296        box_description = wx.StaticBox(self, -1, "Fit Combinations")
297        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
298        sizer_title = wx.BoxSizer(wx.HORIZONTAL)
299        sizer_couples = wx.GridBagSizer(5, 5)
300        #------------------------------------------------------
301        if len(self.page_finder) == 0:
302            msg = " No fit combinations are found! \n\n"
303            msg += " Please load data and set up "
304            msg += "at least two fit panels first..."
305            sizer_title.Add(wx.StaticText(self, -1, msg))
306        else:
307            ## store model
308            self._store_model()
309
310            self.cb1 = wx.CheckBox(self, -1, 'Select all')
311            self.cb1.SetValue(False)
312
313            wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.check_all_model_name)
314
315            sizer_title.Add((10, 10), 0,
316                wx.TOP | wx.BOTTOM | wx.EXPAND | wx.ADJUST_MINSIZE, border=5)
317            sizer_title.Add(self.cb1, 0,
318                wx.TOP | wx.BOTTOM | wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, border=5)
319
320            ## draw list of model and data name
321            self._fill_sizer_model_list(sizer_couples)
322            ## draw the sizer containing constraint info
323            if not self.batch_on:
324                self._fill_sizer_constraint()
325            ## draw fit button
326            self._fill_sizer_fit()
327        #--------------------------------------------------------
328        boxsizer1.Add(sizer_title, flag=wx.TOP | wx.BOTTOM, border=5)
329        boxsizer1.Add(sizer_couples, 1, flag=wx.TOP | wx.BOTTOM, border=5)
330
331        self.sizer1.Add(boxsizer1, 1, wx.EXPAND | wx.ALL, 10)
332        self.sizer1.Layout()
333        #self.SetScrollbars(20,20,25,65)
334        self.AdjustScrollbars()
335        self.Layout()
336
337    def _store_model(self):
338        """
339         Store selected model
340        """
341        if len(self.model_toFit) < 1:
342            return
343        for item in self.model_toFit:
344            model = item[3]
345            page_id = item[2]
346            self.constraint_dict[page_id] = model
347
348    def _display_constraint(self, event):
349        """
350        Show fields to add constraint
351        """
352        if len(self.model_toFit) < 1:
353            msg = "Select at least 1 model to add constraint "
354            wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
355            ## hide button
356            self._hide_constraint()
357            return
358        if self.show_constraint.GetValue():
359            self._show_all_constraint()
360            self._show_constraint()
361            self.Layout()
362            return
363        else:
364            self._hide_constraint()
365            self.Layout()
366            return
367
368    def _show_all_constraint(self):
369        """
370        Show constraint fields
371        """
372        box_description = wx.StaticBox(self, -1, "Easy Setup ")
373        boxsizer = wx.StaticBoxSizer(box_description, wx.HORIZONTAL)
374        sizer_constraint = wx.BoxSizer(wx.HORIZONTAL)
375        self.model_cbox_left = wx.ComboBox(self, -1, style=wx.CB_READONLY)
376        self.model_cbox_left.Clear()
377        self.model_cbox_right = wx.ComboBox(self, -1, style=wx.CB_READONLY)
378        self.model_cbox_right.Clear()
379        wx.EVT_COMBOBOX(self.model_cbox_left, -1, self._on_select_modelcb)
380        wx.EVT_COMBOBOX(self.model_cbox_right, -1, self._on_select_modelcb)
381        egal_txt = wx.StaticText(self, -1, " = ")
382        self.set_button = wx.Button(self, wx.NewId(), 'Set All')
383        self.set_button.Bind(wx.EVT_BUTTON, self._on_set_all_equal,
384                             id=self.set_button.GetId())
385        set_tip = "Add constraints for all the adjustable parameters "
386        set_tip += "(checked in FitPages) if exist."
387        self.set_button.SetToolTipString(set_tip)
388        self.set_button.Disable()
389
390        for id, model in self.constraint_dict.iteritems():
391            ## check if all parameters have been selected for constraint
392            ## then do not allow add constraint on parameters
393            self.model_cbox_left.Append(str(model.name), model)
394        self.model_cbox_left.Select(0)
395        for id, model in self.constraint_dict.iteritems():
396            ## check if all parameters have been selected for constraint
397            ## then do not allow add constraint on parameters
398            self.model_cbox_right.Append(str(model.name), model)
399        boxsizer.Add(self.model_cbox_left,
400                             flag=wx.RIGHT | wx.EXPAND, border=10)
401        boxsizer.Add(wx.StaticText(self, -1, ".parameters"),
402                             flag=wx.RIGHT | wx.EXPAND, border=5)
403        boxsizer.Add(egal_txt, flag=wx.RIGHT | wx.EXPAND, border=5)
404        boxsizer.Add(self.model_cbox_right,
405                             flag=wx.RIGHT | wx.EXPAND, border=10)
406        boxsizer.Add(wx.StaticText(self, -1, ".parameters"),
407                             flag=wx.RIGHT | wx.EXPAND, border=5)
408        boxsizer.Add((20, -1))
409        boxsizer.Add(self.set_button, flag=wx.RIGHT | wx.EXPAND, border=5)
410        sizer_constraint.Add(boxsizer, flag=wx.RIGHT | wx.EXPAND, border=5)
411        self.sizer_all_constraints.Insert(before=0,
412                             item=sizer_constraint,
413                             flag=wx.TOP | wx.BOTTOM | wx.EXPAND, border=5)
414
415        self.sizer_all_constraints.Layout()
416        self.sizer2.Layout()
417        #self.SetScrollbars(20,20,25,65)
418
419    def _on_select_modelcb(self, event):
420        """
421        On select model left or right combobox
422        """
423        event.Skip()
424        flag = True
425        if self.model_cbox_left.GetValue().strip() == '':
426            flag = False
427        if self.model_cbox_right.GetValue().strip() == '':
428            flag = False
429        if self.model_cbox_left.GetValue() == \
430                self.model_cbox_right.GetValue():
431            flag = False
432        self.set_button.Enable(flag)
433
434    def _on_set_all_equal(self, event):
435        """
436        On set button
437        """
438        event.Skip()
439        length = len(self.constraints_list)
440        if length < 1:
441            return
442        param_list = []
443        param_listB = []
444        selection = self.model_cbox_left.GetCurrentSelection()
445        model_left = self.model_cbox_left.GetValue()
446        model = self.model_cbox_left.GetClientData(selection)
447        selectionB = self.model_cbox_right.GetCurrentSelection()
448        model_right = self.model_cbox_right.GetValue()
449        modelB = self.model_cbox_right.GetClientData(selectionB)
450        for id, dic_model in self.constraint_dict.iteritems():
451            if model == dic_model:
452                param_list = self.page_finder[id].get_param2fit()
453            if modelB == dic_model:
454                param_listB = self.page_finder[id].get_param2fit()
455            if len(param_list) > 0 and len(param_listB) > 0:
456                break
457        num_cbox = 0
458        has_param = False
459        for param in param_list:
460            num_cbox += 1
461            if param in param_listB:
462                self.model_cbox.SetStringSelection(model_left)
463                self._on_select_model(None)
464                self.param_cbox.Clear()
465                self.param_cbox.Append(str(param), model)
466                self.param_cbox.SetStringSelection(str(param))
467                self.ctl2.SetValue(str(model_right + "." + str(param)))
468                has_param = True
469                if num_cbox == (len(param_list) + 1):
470                    break
471                self._show_constraint()
472
473        self.sizer_constraints.Layout()
474        self.sizer2.Layout()
475        self.SetScrollbars(20, 20, 25, 65)
476        self.Layout()
477        if not has_param:
478            msg = " There is no adjustable parameter (checked to fit)"
479            msg += " either one of the models."
480            wx.PostEvent(self.parent.parent, StatusEvent(info="warning",
481                                                         status=msg))
482        else:
483            msg = " The constraints are added."
484            wx.PostEvent(self.parent.parent, StatusEvent(info="info",
485                                                         status=msg))
486
487    def _show_constraint(self):
488        """
489        Show constraint fields
490        """
491        self.btAdd.Show(True)
492        if len(self.constraints_list) != 0:
493            nb_fit_param = 0
494            for id, model in self.constraint_dict.iteritems():
495                nb_fit_param += len(self.page_finder[id].get_param2fit())
496            ##Don't add anymore
497            if len(self.constraints_list) == nb_fit_param:
498                msg = "Cannot add another constraint .Maximum of number "
499                msg += "Parameters name reached %s" % str(nb_fit_param)
500                wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
501                self.sizer_constraints.Layout()
502                self.sizer2.Layout()
503                return
504        if len(self.model_toFit) < 1:
505            msg = "Select at least 1 model to add constraint "
506            wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
507            self.sizer_constraints.Layout()
508            self.sizer2.Layout()
509            return
510
511        sizer_constraint = wx.BoxSizer(wx.HORIZONTAL)
512        model_cbox = wx.ComboBox(self, -1, style=wx.CB_READONLY)
513        model_cbox.Clear()
514        param_cbox = wx.ComboBox(self, -1, style=wx.CB_READONLY, size=(100, -1),)
515        param_cbox.Hide()
516
517        #This is for GetCLientData() _on_select_param: Was None return on MAC.
518        self.param_cbox = param_cbox
519
520        wx.EVT_COMBOBOX(param_cbox, -1, self._on_select_param)
521        self.ctl2 = wx.TextCtrl(self, -1)
522        egal_txt = wx.StaticText(self, -1, " = ")
523        self.btRemove = wx.Button(self, wx.NewId(), 'Remove')
524        self.btRemove.Bind(wx.EVT_BUTTON, self.onRemove,
525                           id=self.btRemove.GetId())
526        self.btRemove.SetToolTipString("Remove constraint.")
527        self.btRemove.Hide()
528        if hasattr(self, "btAdd"):
529            self.btAdd.Hide()
530        for id, model in self.constraint_dict.iteritems():
531            ## check if all parameters have been selected for constraint
532            ## then do not allow add constraint on parameters
533            model_cbox.Append(str(model.name), model)
534
535        #This is for GetCLientData() passing to self._on_select_param: Was None return on MAC.
536        self.model_cbox = model_cbox
537
538        wx.EVT_COMBOBOX(model_cbox, -1, self._on_select_model)
539        sizer_constraint.Add((5, -1))
540        sizer_constraint.Add(model_cbox, flag=wx.RIGHT | wx.EXPAND, border=10)
541        sizer_constraint.Add(param_cbox, flag=wx.RIGHT | wx.EXPAND, border=5)
542        sizer_constraint.Add(egal_txt, flag=wx.RIGHT | wx.EXPAND, border=5)
543        sizer_constraint.Add(self.ctl2, flag=wx.RIGHT | wx.EXPAND, border=10)
544        sizer_constraint.Add(self.btRemove, flag=wx.RIGHT | wx.EXPAND, border=10)
545
546        self.sizer_constraints.Insert(before=self.nb_constraint,
547                        item=sizer_constraint, flag=wx.TOP | wx.BOTTOM | wx.EXPAND,
548                        border=5)
549        self.constraints_list.append([model_cbox, param_cbox, egal_txt,
550                                    self.ctl2, self.btRemove, sizer_constraint])
551
552        self.nb_constraint += 1
553        self.sizer_constraints.Layout()
554        self.sizer2.Layout()
555
556    def _hide_constraint(self):
557        """
558        hide buttons related constraint
559        """
560        for id in  self.page_finder.iterkeys():
561            self.page_finder[id].clear_model_param()
562
563        self.nb_constraint = 0
564        self.constraint_dict = {}
565        if hasattr(self, "btAdd"):
566            self.btAdd.Hide()
567        self._store_model()
568        if self.model_cbox_left != None:
569            try:
570                self.model_cbox_left.Clear()
571            except:
572                pass
573            self.model_cbox_left = None
574        if self.model_cbox_right != None:
575            try:
576                self.model_cbox_right.Clear()
577            except:
578                pass
579            self.model_cbox_right = None
580        self.constraints_list = []
581        self.sizer_all_constraints.Clear(True)
582        self.sizer_all_constraints.Layout()
583        self.sizer_constraints.Clear(True)
584        self.sizer_constraints.Layout()
585        self.sizer2.Layout()
586
587    def _on_select_model(self, event):
588        """
589        fill combox box with list of parameters
590        """
591        param_list = []
592        ##This way PC/MAC both work, instead of using event.GetClientData().
593        n = self.model_cbox.GetCurrentSelection()
594        model = self.model_cbox.GetClientData(n)
595        for id, dic_model in self.constraint_dict.iteritems():
596            if model == dic_model:
597                param_list = self.page_finder[id].get_param2fit()
598                #break
599        length = len(self.constraints_list)
600        if length < 1:
601            return
602        param_cbox = self.constraints_list[length - 1][1]
603        param_cbox.Clear()
604        ## insert only fittable paramaters
605        for param in param_list:
606            param_cbox.Append(str(param), model)
607
608        param_cbox.Show(True)
609        self.btRemove.Show(True)
610        self.btAdd.Show(True)
611        self.sizer2.Layout()
612
613    def _on_select_param(self, event):
614        """
615        Store the appropriate constraint in the page_finder
616        """
617        ##This way PC/MAC both work, instead of using event.GetClientData().
618        #n = self.param_cbox.GetCurrentSelection()
619        #model = self.param_cbox.GetClientData(n)
620        #param = event.GetString()
621
622        length = len(self.constraints_list)
623        if length < 1:
624            return
625        egal_txt = self.constraints_list[length - 1][2]
626        egal_txt.Show(True)
627
628        ctl2 = self.constraints_list[length - 1][3]
629        ctl2.Show(True)
630
631    def _onAdd_constraint(self, event):
632        """
633        Add another line for constraint
634        """
635        if not self.show_constraint.GetValue():
636            msg = " Select Yes to add Constraint "
637            wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
638            return
639        ## check that a constraint is added
640        # before allow to add another constraint
641        for item in self.constraints_list:
642            model_cbox = item[0]
643            if model_cbox.GetString(0) == "":
644                msg = " Select a model Name! "
645                wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
646                return
647            param_cbox = item[1]
648            if param_cbox.GetString(0) == "":
649                msg = " Select a parameter Name! "
650                wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
651                return
652            ctl2 = item[3]
653            if ctl2.GetValue().lstrip().rstrip() == "":
654                model = param_cbox.GetClientData(\
655                                            param_cbox.GetCurrentSelection())
656                if model != None:
657                    msg = " Enter a constraint for %s.%s! " % (model.name,
658                                                        param_cbox.GetString(0))
659                else:
660                     msg = " Enter a constraint"
661                wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
662                return
663        ## some model or parameters can be constrained
664        self._show_constraint()
665        self.sizer3.Layout()
666        self.Layout()
667        self.Refresh()
668
669    def _fill_sizer_fit(self):
670        """
671        Draw fit button
672        """
673        self.sizer3.Clear(True)
674        box_description = wx.StaticBox(self, -1, "Fit ")
675        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
676        sizer_button = wx.BoxSizer(wx.HORIZONTAL)
677
678        self.btFit = wx.Button(self, wx.NewId(), 'Fit', size=wx.DefaultSize)
679        self.btFit.Bind(wx.EVT_BUTTON, self.onFit, id=self.btFit.GetId())
680        self.btFit.SetToolTipString("Perform fit.")
681        if self.batch_on:
682            text = " Fit in Parallel all Data set and model selected.\n"
683        else:
684            text = " This page requires at least one FitPage with a data\n"
685            text = " and a model for fitting."
686        text_hint = wx.StaticText(self, -1, text)
687
688        sizer_button.Add(text_hint, wx.RIGHT | wx.EXPAND, 10)
689        sizer_button.Add(self.btFit, 0, wx.LEFT | wx.ADJUST_MINSIZE, 10)
690
691        boxsizer1.Add(sizer_button, flag=wx.TOP | wx.BOTTOM, border=10)
692        self.sizer3.Add(boxsizer1, 0, wx.EXPAND | wx.ALL, 10)
693        self.sizer3.Layout()
694
695    def _fill_sizer_constraint(self):
696        """
697        Fill sizer containing constraint info
698        """
699        msg = "Select at least 2 model to add constraint "
700        wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
701
702        self.sizer2.Clear(True)
703        if self.batch_on:
704            if self.sizer2.IsShown():
705                self.sizer2.Show(False)
706            return
707        box_description = wx.StaticBox(self, -1, "Fit Constraints")
708        boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
709        sizer_title = wx.BoxSizer(wx.HORIZONTAL)
710        self.sizer_all_constraints = wx.BoxSizer(wx.HORIZONTAL)
711        self.sizer_constraints = wx.BoxSizer(wx.VERTICAL)
712        sizer_button = wx.BoxSizer(wx.HORIZONTAL)
713
714        self.hide_constraint = wx.RadioButton(self, -1, 'No', (10, 10),
715                                              style=wx.RB_GROUP)
716        self.show_constraint = wx.RadioButton(self, -1, 'Yes', (10, 30))
717        self.Bind(wx.EVT_RADIOBUTTON, self._display_constraint,
718                  id=self.hide_constraint.GetId())
719        self.Bind(wx.EVT_RADIOBUTTON, self._display_constraint,
720                  id=self.show_constraint.GetId())
721        if self.batch_on:
722            self.hide_constraint.Enable(False)
723            self.show_constraint.Enable(False)
724        self.hide_constraint.SetValue(True)
725        self.show_constraint.SetValue(False)
726
727        sizer_title.Add(wx.StaticText(self, -1, " Model"))
728        sizer_title.Add((10, 10))
729        sizer_title.Add(wx.StaticText(self, -1, " Parameter"))
730        sizer_title.Add((10, 10))
731        sizer_title.Add(wx.StaticText(self, -1, " Add Constraint?"))
732        sizer_title.Add((10, 10))
733        sizer_title.Add(self.show_constraint)
734        sizer_title.Add(self.hide_constraint)
735        sizer_title.Add((10, 10))
736
737        self.btAdd = wx.Button(self, wx.NewId(), 'Add')
738        self.btAdd.Bind(wx.EVT_BUTTON, self._onAdd_constraint,
739                        id=self.btAdd.GetId())
740        self.btAdd.SetToolTipString("Add another constraint?")
741        self.btAdd.Hide()
742
743        text_hint = wx.StaticText(self, -1,
744                                  "Example: [M0][paramter] = M1.parameter")
745        sizer_button.Add(text_hint, 0 , wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 10)
746        sizer_button.Add(self.btAdd, 0, wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 10)
747
748        boxsizer1.Add(sizer_title, flag=wx.TOP | wx.BOTTOM, border=10)
749        boxsizer1.Add(self.sizer_all_constraints, flag=wx.TOP | wx.BOTTOM,
750                      border=10)
751        boxsizer1.Add(self.sizer_constraints, flag=wx.TOP | wx.BOTTOM,
752                      border=10)
753        boxsizer1.Add(sizer_button, flag=wx.TOP | wx.BOTTOM, border=10)
754
755        self.sizer2.Add(boxsizer1, 0, wx.EXPAND | wx.ALL, 10)
756        self.sizer2.Layout()
757
758        #self.SetScrollbars(20,20,25,65)
759
760    def _set_constraint(self):
761        """
762        get values from the constrainst textcrtl ,parses them into model name
763        parameter name and parameters values.
764        store them in a list self.params .when when params is not empty
765        set_model uses it to reset the appropriate model
766        and its appropriates parameters
767        """
768        for item in self.constraints_list:
769            select0 = item[0].GetSelection()
770            if select0 == wx.NOT_FOUND:
771                continue
772            model = item[0].GetClientData(select0)
773            select1 = item[1].GetSelection()
774            if select1 == wx.NOT_FOUND:
775                continue
776            param = item[1].GetString(select1)
777            constraint = item[3].GetValue().lstrip().rstrip()
778            if param.lstrip().rstrip() == "":
779                param = None
780                msg = " Constraint will be ignored!. missing parameters"
781                msg += " in combobox to set constraint! "
782                wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
783            for id, value in self.constraint_dict.iteritems():
784                if model == value:
785                    if constraint == "":
786                        msg = " Constraint will be ignored!. missing value"
787                        msg += " in textcrtl to set constraint! "
788                        wx.PostEvent(self.parent.parent,
789                                     StatusEvent(status=msg))
790                        constraint = None
791                    if str(param) in self.page_finder[id].get_param2fit():
792                        msg = " Checking constraint for parameter: %s ", param
793                        wx.PostEvent(self.parent.parent,
794                                     StatusEvent(info="info", status=msg))
795                    else:
796                        model_name = item[0].GetLabel()
797                        fitpage = self.page_finder[id].get_fit_tab_caption()
798                        msg = "All constrainted parameters must be set "
799                        msg += " adjustable: '%s.%s' " % (model_name, param)
800                        msg += "is NOT checked in '%s'. " % fitpage
801                        msg += " Please check it to fit or"
802                        msg += " remove the line of the constraint."
803                        wx.PostEvent(self.parent.parent,
804                                StatusEvent(info="error", status=msg))
805                        return False
806
807                    for fid in self.page_finder[id].iterkeys():
808                        # wrap in param/constraint in str() to remove unicode
809                        self.page_finder[id].set_model_param(str(param),
810                                                        str(constraint), fid=fid)
811                    break
812        return True
813
814    def _fill_sizer_model_list(self, sizer):
815        """
816        Receive a dictionary containing information to display model name
817        """
818        ix = 0
819        iy = 0
820        list = []
821        sizer.Clear(True)
822
823        new_name = wx.StaticText(self, -1, '  Model Title ',
824                                 style=wx.ALIGN_CENTER)
825        new_name.SetBackgroundColour('orange')
826        new_name.SetForegroundColour(wx.WHITE)
827        sizer.Add(new_name, (iy, ix), (1, 1),
828                            wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
829        ix += 2
830        model_type = wx.StaticText(self, -1, '  Model ')
831        model_type.SetBackgroundColour('grey')
832        model_type.SetForegroundColour(wx.WHITE)
833        sizer.Add(model_type, (iy, ix), (1, 1),
834                            wx.EXPAND | wx.ADJUST_MINSIZE, 0)
835        ix += 1
836        data_used = wx.StaticText(self, -1, '  Data ')
837        data_used.SetBackgroundColour('grey')
838        data_used.SetForegroundColour(wx.WHITE)
839        sizer.Add(data_used, (iy, ix), (1, 1),
840                            wx.EXPAND | wx.ADJUST_MINSIZE, 0)
841        ix += 1
842        tab_used = wx.StaticText(self, -1, '  FitPage ')
843        tab_used.SetBackgroundColour('grey')
844        tab_used.SetForegroundColour(wx.WHITE)
845        sizer.Add(tab_used, (iy, ix), (1, 1),
846                            wx.EXPAND | wx.ADJUST_MINSIZE, 0)
847        for id, value in self.page_finder.iteritems():
848            if id not in self.parent.opened_pages:
849                continue
850
851            if self.batch_on != self.parent.get_page_by_id(id).batch_on:
852                continue
853
854            data_list = []
855            model_list = []
856            # get data name and model objetta
857            for fitproblem in value.get_fit_problem():
858
859                data = fitproblem.get_fit_data()
860                if not data.is_data:
861                    continue
862                name = '-'
863                if data is not None and data.is_data:
864                    name = str(data.name)
865                data_list.append(name)
866
867                model = fitproblem.get_model()
868                if model is None:
869                    continue
870                model_list.append(model)
871
872            if len(model_list) == 0:
873                continue
874            # Draw sizer
875            ix = 0
876            iy += 1
877            model = model_list[0]
878            name = '_'
879            if model is not None:
880                name = str(model.name)
881            cb = wx.CheckBox(self, -1, name)
882            cb.SetValue(False)
883            cb.Enable(model is not None and data.is_data)
884            sizer.Add(cb, (iy, ix), (1, 1),
885                       wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
886            wx.EVT_CHECKBOX(self, cb.GetId(), self.check_model_name)
887            ix += 2
888            type = model.__class__.__name__
889            model_type = wx.StaticText(self, -1, str(type))
890            sizer.Add(model_type, (iy, ix), (1, 1),
891                      wx.EXPAND | wx.ADJUST_MINSIZE, 0)
892            if self.batch_on:
893                data_used = wx.ComboBox(self, -1, style=wx.CB_READONLY)
894                data_used.AppendItems(data_list)
895                data_used.SetSelection(0)
896            else:
897                data_used = wx.StaticText(self, -1, data_list[0])
898
899            ix += 1
900            sizer.Add(data_used, (iy, ix), (1, 1),
901                      wx.EXPAND | wx.ADJUST_MINSIZE, 0)
902            ix += 1
903            caption = value.get_fit_tab_caption()
904            tab_caption_used = wx.StaticText(self, -1, str(caption))
905            sizer.Add(tab_caption_used, (iy, ix), (1, 1),
906                      wx.EXPAND | wx.ADJUST_MINSIZE, 0)
907
908            self.model_list.append([cb, value, id, model])
909
910        iy += 1
911        sizer.Add((20, 20), (iy, ix), (1, 1),
912                  wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
913        sizer.Layout()
914
915    def on_set_focus(self, event=None):
916        """
917        The  derivative class is on focus if implemented
918        """
919        if self.parent is not None:
920            if self.parent.parent is not None:
921                wx.PostEvent(self.parent.parent, PanelOnFocusEvent(panel=self))
922            self.page_finder = self.parent._manager.get_page_finder()
Note: See TracBrowser for help on using the repository browser.