Changeset aa25fc7 in sasmodels for sasmodels/weights.py

Timestamp:

May 23, 2018 3:55:27 PM (6 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master

Children:

910c0f4

Parents:

33969b6

Message:

load user-defined weight functions from ~/.sasview/weights/*.py

File:

• sasmodels/weights.py (modified) (1 diff)

sasmodels/weights.py

@@ -231 +231 @@
 ))
 
+SASMODELS_WEIGHTS = "~/.sasview/weights/*.py"
+def load_weights(pattern=None):
+    # type: (str) -> None
+    """
+    Load dispersion distributions matching the given pattern
+    """
+    import logging
+    import os
+    import os.path
+    import glob
+    import traceback
+    from .custom import load_custom_kernel_module
+    if pattern is None:
+        pattern = os.environ.get("SASMODELS_WEIGHTS", SASMODELS_WEIGHTS)
+    for filename in sorted(glob.glob(os.path.expanduser(pattern))):
+        try:
+            print("loading weights from", filename)
+            module = load_custom_kernel_module(filename)
+            MODELS[module.Dispersion.type] = module.Dispersion
+        except Exception as exc:
+            logging.error(traceback.format_exc(exc))
 
 def get_weights(disperser, n, width, nsigmas, value, limits, relative):