Changeset e89aed5 in sasview for src/sas/sasgui/perspectives/fitting/simfitpage.py

Timestamp:

Oct 6, 2016 3:47:28 PM (8 years ago)

Author:

krzywon

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

1dbdb83

Parents:

ef3e09b

Message:

#189: Can now open the simultaneous and constrained fit panel through a save state, but the constraint loading is still broken.

File:

• src/sas/sasgui/perspectives/fitting/simfitpage.py (modified) (3 diffs)

src/sas/sasgui/perspectives/fitting/simfitpage.py

@@ -768 +768 @@
         """
         Show constraint fields
+        :param dict: dictionary mapping constraint values
         """
         self.btAdd.Show(True)
@@ -871 +872 @@
     def _on_select_model(self, event):
         """
-        fill combox box with list of parameters
+        fill combo box with list of parameters
         """
         if not self.constraints_list:
@@ -1041 +1042 @@
         # List of constraints
         self.constraints_list = []
+
+    def load_from_save_state(self, fit):
+        """
+        Load in a simultaneous/constrained fit from a save state
+        :param fit: Fitpanel object
+        :return: None
+        """
+
+        model_map = {}
+        fit.fit_panel.add_sim_page()
+
+        # Process each model and associate old M# with new M#
+        i = 0
+        for model in fit.fit_panel.sim_page.model_list:
+            model_id = self._format_id(model[1].keys()[0])
+            for saved_model in self.model_list:
+                save_id = saved_model.pop('name')
+                saved_model['name'] = save_id
+                save_id = self._format_id(save_id)
+                if save_id == model_id:
+                    model_map[saved_model.pop('fit_page_source')] = \
+                        model[3].name
+                    check = bool(saved_model.pop('checked'))
+                    fit.fit_panel.sim_page.model_list[i][0].SetValue(check)
+                    continue
+            i += 1
+        fit.fit_panel.sim_page.check_model_name(None)
+
+        if len(self.constraints_list) > 0:
+            fit.fit_panel.sim_page.hide_constraint.SetValue(False)
+            fit.fit_panel.sim_page.show_constraint.SetValue(True)
+            fit.fit_panel.sim_page._display_constraint(None)
+
+        for constraint in self.constraints_list:
+            model_cbox = constraint.pop('model_cbox')
+            constraint_value = constraint.pop('constraint')
+            for key, value in model_map.iteritems():
+                model_cbox.replace(key, value)
+                constraint_value.replace(key, value)
+
+            bt_remove = wx.Button(fit.fit_panel.sim_page,
+                                  fit.fit_panel.sim_page._ids.next(), 'Remove')
+            bt_remove.Bind(wx.EVT_BUTTON, fit.fit_panel.sim_page.on_remove,
+                           id=bt_remove.GetId())
+            bt_remove.SetToolTipString("Remove constraint.")
+            param_cbox = wx.ComboBox(fit.fit_panel.sim_page, wx.ID_ANY,
+                                     style=wx.CB_READONLY, size=(100, -1))
+            egal_txt = wx.StaticText(fit.fit_panel.sim_page, wx.ID_ANY, " = ")
+            # Model list
+            model_cbox_sizer = wx.ComboBox(fit.fit_panel.sim_page,
+                                           wx.ID_ANY, style=wx.CB_READONLY)
+            model_cbox_sizer.Clear()
+            wx.EVT_COMBOBOX(model_cbox_sizer, wx.ID_ANY,
+                            fit.fit_panel.sim_page._on_select_model)
+            constraint_sizer = wx.TextCtrl(fit.fit_panel.sim_page, wx.ID_ANY)
+            sizer_constraint = wx.BoxSizer(wx.HORIZONTAL)
+            sizer_constraint.Add((5, -1))
+            sizer_constraint.Add(model_cbox_sizer, flag=wx.RIGHT | wx.EXPAND,
+                                 border=10)
+            sizer_constraint.Add(param_cbox, flag=wx.RIGHT | wx.EXPAND,
+                                 border=5)
+            sizer_constraint.Add(egal_txt, flag=wx.RIGHT | wx.EXPAND, border=5)
+            sizer_constraint.Add(constraint_sizer, flag=wx.RIGHT | wx.EXPAND,
+                                 border=10)
+            sizer_constraint.Add(bt_remove, flag=wx.RIGHT | wx.EXPAND,
+                                 border=10)
+            new_const = ConstraintLine(model_cbox=model_cbox_sizer,
+                                       param_cbox=constraint.pop('param_cbox'),
+                                       egal_txt=constraint.pop('egal_txt'),
+                                       constraint=constraint_value,
+                                       btRemove=bt_remove,
+                                       sizer=sizer_constraint
+                                       )
+            # ConstraintLine = namedtuple('ConstraintLine',
+            #      'model_cbox param_cbox egal_txt constraint btRemove sizer')
+            fit.fit_panel.sim_page.sizer_constraints.Insert(
+                before=fit.fit_panel.sim_page.nb_constraint,
+                item=sizer_constraint, flag=wx.TOP | wx.BOTTOM | wx.EXPAND,
+                border=5)
+            fit.fit_panel.sim_page.constraints_list.append(new_const)
+
+    def _format_id(self, original_id):
+        original_id = original_id.rstrip('1234567890.')
+        new_id_list = original_id.split()
+        new_id = ' '.join(new_id_list)
+        return new_id