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

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

Take care of white spaces (pylint)

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