Changeset 4e9f227 in sasview for src

Timestamp:

May 15, 2014 5:07:43 PM (10 years ago)

Author:

pkienzle

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, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

786685e

Parents:

e3efa6b3

Message:

support simultaneous fitting in bumps

Location:

src/sans

Files:

• fit/BumpsFitting.py (modified) (7 diffs)
• fit/expression.py (added)
• perspectives/fitting/fitpage.py (modified) (4 diffs)
• perspectives/fitting/fitting.py (modified) (2 diffs)
• perspectives/fitting/simfitpage.py (modified) (1 diff)

src/sans/fit/BumpsFitting.py

@@ -2 +2 @@
 BumpsFitting module runs the bumps optimizer.
 """
-import time
-
 import numpy
 
@@ -11 +9 @@
 from bumps.fitproblem import FitProblem
 
-from sans.fit.AbstractFitEngine import FitEngine
-from sans.fit.AbstractFitEngine import FResult
+from .AbstractFitEngine import FitEngine
+from .AbstractFitEngine import FResult
+from .expression import compile_constraints
 
 class BumpsMonitor(object):
@@ -52 +51 @@
 
 
+# Note: currently using bumps parameters for each parameter object so that
+# a SasFitness can be used directly in bumps with the usual semantics.
+# The disadvantage of this technique is that we need to copy every parameter
+# back into the model each time the function is evaluated.  We could instead
+# define reference parameters for each sans parameter, but then we would not
+# be able to express constraints using python expressions in the usual way
+# from bumps, and would instead need to use string expressions.
 class SasFitness(object):
     """
     Wrap SAS model as a bumps fitness object
     """
-    def __init__(self, name, model, data, fitted=[], **kw):
-        self.name = name
-        self.model = model
+    def __init__(self, model, data, fitted=[], constraints={}, **kw):
+        self.name = model.name
+        self.model = model.model
         self.data = data
         self._define_pars()
         self._init_pars(kw)
+        self.constraints = dict(constraints)
         self.set_fitted(fitted)
         self._dirty = True
@@ -73 +80 @@
             self._pars[k] = parameter.Parameter(value=value, bounds=bounds,
                                                 fixed=True, name=name)
+        #print parameter.summarize(self._pars.values())
 
     def _init_pars(self, kw):
@@ -102 +110 @@
         """
         for k,p in self._pars.items():
-            p.fixed = (k not in param_list)
+            p.fixed = (k not in param_list or k in self.constraints)
         self.fitted_pars = [self._pars[k] for k in param_list]
         self.fitted_par_names = param_list
@@ -112 +120 @@
     def update(self):
         for k,v in self._pars.items():
+            #print "updating",k,v,v.value
             self.model.setParam(k,v.value)
         self._dirty = True
@@ -154 +163 @@
             ftol=1.49012e-8, reset_flag=False):
         # Build collection of bumps fitness calculators
-        models = [ SasFitness(name="M%d"%(i+1),
-                              model=M.get_model().model,
+        models = [ SasFitness(model=M.get_model(),
                               data=M.get_data(),
+                              constraints=M.constraints,
                               fitted=M.pars)
                    for i,M in enumerate(self.fit_arrange_dict.values())
                    if M.get_to_fit() == 1 ]
         problem = FitProblem(models)
+
+
+        # Build constraint expressions
+        exprs = {}
+        for M in models:
+            exprs.update((".".join((M.name,k)),v) for k,v in M.constraints.items())
+        if exprs:
+            symtab = dict((".".join((M.name,k)),p)
+                          for M in models
+                          for k,p in M.parameters().items())
+            constraints = compile_constraints(symtab,exprs)
+        else:
+            constraints = lambda: 0
+
+        # Override model update so that parameter constraints are applied
+        problem._model_update = problem.model_update
+        def model_update():
+            constraints()
+            problem._model_update()
+        problem.model_update = model_update
 
         # Run the fit

src/sans/perspectives/fitting/fitpage.py

@@ -661 +661 @@
                           wx.EXPAND | wx.ADJUST_MINSIZE, 0)
         
-        if self.engine_type == "park":
+        if self.engine_type in ("park","bumps"):
             self.text_disp_max.Show(True)
             self.text_disp_min.Show(True)
@@ -738 +738 @@
                                           wx.EXPAND | wx.ADJUST_MINSIZE, 0)
 
-                        if self.engine_type == "park":
+                        if self.engine_type in ("park","bumps"):
                             ctl3.Show(True)
                             ctl4.Show(True)
@@ -1005 +1005 @@
             return
 
-        if len(self._manager.fit_thread_list) > 0 and\
-                    self._manager._fit_engine != "park" and\
-                    self._manager.sim_page != None and \
-                    self._manager.sim_page.uid == self.uid:
+        if (len(self._manager.fit_thread_list) > 0
+                and self._manager._fit_engine not in ("park","bumps")
+                and self._manager.sim_page != None
+                and self._manager.sim_page.uid == self.uid):
             msg = "The FitEnging will be set to 'ParkMC'\n"
             msg += " to fit with more than one data set..."
@@ -2108 +2108 @@
         if chisqr != None and numpy.isfinite(chisqr):
             #format chi2
-            if self.engine_type == "park":
+            if self.engine_type in ("park","bumps"):
                 npt_fit = float(self.get_npts2fit())
             chi2 = format_number(chisqr, True)

src/sans/perspectives/fitting/fitting.py

@@ -999 +999 @@
             #simulatanous fit only one engine need to be created
             ## if simultaneous fit change automatically the engine to park
-            self._on_change_engine(engine='park')
+            if self._fit_engine not in ("park","bumps"):
+                self._on_change_engine(engine='park')
             sim_fitter = Fit(self._fit_engine)
             sim_fitter.fitter_id = self.sim_page.uid
@@ -1007 +1008 @@
             
         self.fitproblem_count = fitproblem_count
-        if self._fit_engine == "park":
+        if self._fit_engine in ("park","bumps"):
             engineType = "Simultaneous Fit"
         else:

src/sans/perspectives/fitting/simfitpage.py

@@ -152 +152 @@
         ## making sure all parameters content a constraint
         ## validity of the constraint expression is own by fit engine
-        if self.parent._manager._fit_engine != "park" and flag:
+        if self.parent._manager._fit_engine not in ("park","bumps") and flag:
             msg = "The FitEnging will be set to 'Park' fit engine\n"
             msg += " for the simultaneous fit..."