Changeset 0b03001 in sasmodels

Timestamp:

Jul 27, 2016 6:17:31 PM (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:

7f1ee79

Parents:

6351bfa

Message:

list models by kind of model

File:

• sasmodels/core.py (modified) (2 diffs)

sasmodels/core.py

@@ -57 +57 @@
 #    build_model
 
-def list_models():
+def list_models(kind=None):
     # type: () -> List[str]
     """
     Return the list of available models on the model path.
     """
+    KINDS = ("all", "py", "c", "double", "oriented", "magnetic")
+    if kind and kind not in KINDS:
+        raise ValueError("kind not in "+", ".join(KINDS))
     root = dirname(__file__)
     files = sorted(glob(joinpath(root, 'models', "[a-zA-Z]*.py")))
     available_models = [basename(f)[:-3] for f in files]
-    return available_models
+    selected = [name for name in available_models if _matches(name, kind)]
+
+    return selected
+
+def _matches(name, kind):
+    if kind is None or kind=="all":
+        return True
+    info = load_model_info(name)
+    pars = info.parameters.kernel_parameters
+    if kind == "py" and callable(info.Iq):
+        return True
+    elif kind == "c" and not callable(info.Iq):
+        return True
+    elif kind == "double" and not info.single:
+        return True
+    elif kind == "oriented" and any(p.type=='orientation' for p in pars):
+        return True
+    elif kind == "magnetic" and any(p.type=='sld' for p in pars):
+        return True
+    return False
 
 def load_model(model_name, dtype=None, platform='ocl'):
@@ -223 +245 @@
 
 if __name__ == "__main__":
-    print("\n".join(list_models()))
+    import sys
+    kind = sys.argv[1] if len(sys.argv) > 1 else "all"
+    print("\n".join(list_models(kind)))