Changes in / [9ed43f4:d77eca8] in sasmodels

Location:

sasmodels

Files:

• compare.py (modified) (6 diffs)
• data.py (modified) (1 diff)

sasmodels/compare.py

@@ -73 +73 @@
     -1d*/-2d computes 1d or 2d data
     -preset*/-random[=seed] preset or random parameters
-    -mono/-poly* force monodisperse/polydisperse
+    -mono*/-poly force monodisperse or allow polydisperse demo parameters
     -magnetic/-nonmagnetic* suppress magnetism
     -cutoff=1e-5* cutoff value for including a point in polydispersity
@@ -753 +753 @@
     comp = opts['engines'][1] if have_comp else None
     data = opts['data']
-    use_data = have_base ^ have_comp
+    use_data = (opts['datafile'] is not None) and (have_base ^ have_comp)
 
     # Plot if requested
     view = opts['view']
     import matplotlib.pyplot as plt
-    if limits is None:
+    if limits is None and not use_data:
         vmin, vmax = np.Inf, -np.Inf
         if have_base:
@@ -947 +947 @@
         'cutoff'    : 0.0,
         'seed'      : -1,  # default to preset
-        'mono'      : False,
+        'mono'      : True,
         # Default to magnetic a magnetic moment is set on the command line
         'magnetic'  : False,
@@ -958 +958 @@
         'html'      : False,
         'title'     : None,
-        'data'      : None,
+        'datafile'  : None,
     }
     engines = []
@@ -980 +980 @@
         elif arg.startswith('-random='):   opts['seed'] = int(arg[8:])
         elif arg.startswith('-title='):    opts['title'] = arg[7:]
-        elif arg.startswith('-data='):     opts['data'] = arg[6:]
+        elif arg.startswith('-data='):     opts['datafile'] = arg[6:]
         elif arg == '-random':  opts['seed'] = np.random.randint(1000000)
         elif arg == '-preset':  opts['seed'] = -1
@@ -1122 +1122 @@
 
     # Create the computational engines
-    if opts['data'] is not None:
-        data = load_data(os.path.expanduser(opts['data']))
+    if opts['datafile'] is not None:
+        data = load_data(os.path.expanduser(opts['datafile']))
     else:
         data, _ = make_data(opts)

sasmodels/data.py

@@ -51 +51 @@
     from sas.sascalc.dataloader.loader import Loader  # type: ignore
     loader = Loader()
-    data = loader.load(filename)
-    if data is None:
+    # Allow for one part in multipart file
+    if '[' in filename:
+        filename, indexstr = filename[:-1].split('[')
+        index = int(indexstr)
+    else:
+        index = None
+    datasets = loader.load(filename)
+    if datasets is None:
         raise IOError("Data %r could not be loaded" % filename)
+    if not isinstance(datasets, list):
+        datasets = [datasets]
+    if index is None and len(datasets) > 1:
+        raise ValueError("Need to specify filename[index] for multipart data")
+    data = datasets[index if index is not None else 0]
     if hasattr(data, 'x'):
         data.qmin, data.qmax = data.x.min(), data.x.max()
         data.mask = (np.isnan(data.y) if data.y is not None
                      else np.zeros_like(data.x, dtype='bool'))
+    elif hasattr(data, 'qx_data'):
+        data.mask = ~data.mask
     return data