Changeset 9a66e65 in sasmodels for sasmodels/convert.py

Timestamp:

Dec 9, 2015 11:12:17 AM (8 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, core_shell_microgels, costrafo411, magnetic_model, release_v0.94, release_v0.95, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

716abe0

Parents:

dcef2ee

Message:

refactor sasmodels to sasview parameter conversion

File:

• sasmodels/convert.py (modified) (3 diffs)

sasmodels/convert.py

@@ -2 +2 @@
 Convert models to and from sasview.
 """
+import warnings
 
-def _rename_pars(pars, mapping):
+PD_DOT = [
+    ("", ""),
+    ("_pd", ".width"),
+    ("_pd_n", ".npts"),
+    ("_pd_nsigma", ".nsigmas"),
+    ("_pd_type", ".type"),
+    ]
+def _convert_pars(pars, mapping):
     """
     Rename the parameters and any associated polydispersity attributes.
     """
     newpars = pars.copy()
-    for old,new in mapping.items():
+    for new, old in mapping.items():
         if old == new: continue
-        # Bumps style parameter names
-        for variant in ("", "_pd", "_pd_n", "_pd_nsigma", "_pd_type"):
-            if old+variant in newpars:
+        for pd, dot in PD_DOT:
+            if old+dot in newpars:
                 if new is not None:
-                    newpars[new+variant] = pars[old+variant]
-                del newpars[old+variant]
-        # Sasview style parameter names
-        for variant in (".width", ".nsigmas", ".type", ".npts"):
-            if old+variant in newpars:
-                if new is not None:
-                    newpars[new+variant] = pars[old+variant]
-                del newpars[old+variant]
+                    newpars[new+pd] = pars[old+dot]
+                del newpars[old+dot]
     return newpars
 
@@ -51 +52 @@
                 for p,v in pars.items())
 
+def _remove_pd(pars, key, name):
+    """
+    Remove polydispersity from the parameter list.
+
+    Note: operates in place
+    """
+    # Bumps style parameter names
+    pd = pars.pop(key+".width", 0.0)
+    pd_n = pars.pop(key+".npts", 0)
+    if pd != 0.0 and pd_n != 0:
+        warnings.warn("parameter %s not polydisperse in sasview %s"%(key, name))
+    pars.pop(key+".nsigmas", None)
+    pars.pop(key+".type", None)
+    return pars
+
+def _revert_pars(pars, mapping):
+    """
+    Rename the parameters and any associated polydispersity attributes.
+    """
+    newpars = pars.copy()
+    for new, old in mapping.items():
+        for pd, dot in PD_DOT:
+            if old+pd == new+dot:
+                continue
+            if new+pd in newpars:
+                if old is not None:
+                    newpars[old+dot] = pars[new+pd]
+                del newpars[new+pd]
+    return newpars
+
 def revert_model(model_definition, pars):
     """
@@ -57 +88 @@
     mapping = model_definition.oldpars
     oldname = model_definition.oldname
-    oldpars = _rename_pars(_unscale_sld(pars), mapping)
+    oldpars = _revert_pars(_unscale_sld(pars), mapping)
+
+    # Note: update compare.constrain_pars to match
+    name = model_definition.name
+    if name in ('teubner_strey', 'broad_peak'):
+        if oldpars.pop('scale', 1.0) != 1.0:
+            warnings.warn("parameter scale not used in sasview %s"%name)
+    elif name in ('guinier',):
+        if oldpars.pop('background', 0.0) != 0.0:
+            warnings.warn("parameter background not used in sasview %s"%name)
+    elif getattr(model_definition, 'category', None) == 'structure-factor':
+        if oldpars.pop('scale', 1.0) != 1.0:
+            warnings.warn("parameter scale not used in sasview %s"%name)
+        if oldpars.pop('background', 0.0) != 0.0:
+            warnings.warn("parameter background not used in sasview %s"%name)
+    elif name == 'pearl_necklace':
+        _remove_pd(oldpars, 'num_pearls', name)
+        _remove_pd(oldpars, 'thick_string', name)
+
     return oldname, oldpars
 
+def constrain_new_to_old(model_definition, pars):
+    """
+    Restrict parameter values to those that will match sasview.
+    """
+    # Note: update convert.revert_model to match
+    name = model_definition.name
+    if name in ('teubner_strey','broad_peak'):
+        pars['scale'] = 1
+    elif name in ('guinier',):
+        pars['background'] = 0
+    elif name == 'pearl_necklace':
+        pars['string_thickness_pd_n'] = 0
+        pars['number_of_pearls_pd_n'] = 0
+    elif getattr(model_definition, 'category', None) == 'structure-factor':
+        pars['scale'], pars['background'] = 1, 0
+