Changeset ced5bd2 in sasmodels

Timestamp:

Oct 4, 2017 6:26:41 PM (7 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, core_shell_microgels, costrafo411, magnetic_model, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

2573fa1

Parents:

18a2bfc

Message:

allow -q=min:max on the sascomp command line

Location:

sasmodels

Files:

• compare.py (modified) (9 diffs)
• data.py (modified) (3 diffs)

sasmodels/compare.py

@@ -73 +73 @@
     -res=0 sets the resolution width dQ/Q if calculating with resolution
     -lowq*/-midq/-highq/-exq use q values up to 0.05, 0.2, 1.0, 10.0
+    -q=min:max alternative specification of qrange
     -nq=128 sets the number of Q points in the data set
     -1d*/-2d computes 1d or 2d data
@@ -768 +769 @@
     'accuracy', 'is2d' and 'view' parsed from the command line.
     """
-    qmax, nq, res = opts['qmax'], opts['nq'], opts['res']
+    qmin, qmax, nq, res = opts['qmin'], opts['qmax'], opts['nq'], opts['res']
     if opts['is2d']:
         q = np.linspace(-qmax, qmax, nq)  # type: np.ndarray
@@ -777 +778 @@
     else:
         if opts['view'] == 'log' and not opts['zero']:
-            qmax = math.log10(qmax)
-            q = np.logspace(qmax-3, qmax, nq)
+            q = np.logspace(math.log10(qmin), math.log10(qmax), nq)
         else:
-            q = np.linspace(0.001*qmax, qmax, nq)
+            q = np.linspace(qmin, qmax, nq)
         if opts['zero']:
             q = np.hstack((0, q))
@@ -955 +955 @@
         else:
             err, errstr, errview = abs(relerr), "rel err", "log"
+            if (err == 0.).all():
+                errview = 'linear'
         if 0:  # 95% cutoff
             sorted = np.sort(err.flatten())
@@ -960 +962 @@
             err[err > cutoff] = cutoff
         #err,errstr = base/comp,"ratio"
-        plot_theory(data, None, resid=err, view=errview, use_data=use_data)
-        if view == 'linear':
-            plt.xscale('linear')
+        plot_theory(data, None, resid=err, view=view, use_data=use_data)
+        plt.yscale(errview)
         plt.title("max %s = %.3g"%(errstr, abs(err).max()))
         #cbar_title = errstr if errview=="linear" else "log "+errstr
@@ -1001 +1002 @@
 
     # Data generation
-    'data=', 'noise=', 'res=',
-    'nq=', 'lowq', 'midq', 'highq', 'exq', 'zero',
+    'data=', 'noise=', 'res=', 'nq=', 'q=',
+    'lowq', 'midq', 'highq', 'exq', 'zero',
     '2d', '1d',
 
@@ -1122 +1123 @@
         'view'      : 'log',
         'is2d'      : False,
+        'qmin'      : None,
         'qmax'      : 0.05,
         'nq'        : 128,
@@ -1159 +1161 @@
         elif arg == '-zero':    opts['zero'] = True
         elif arg.startswith('-nq='):       opts['nq'] = int(arg[4:])
+        elif arg.startswith('-q='):
+            opts['qmin'], opts['qmax'] = [float(v) for v in arg[3:].split(':')]
         elif arg.startswith('-res='):      opts['res'] = float(arg[5:])
         elif arg.startswith('-noise='):    opts['noise'] = float(arg[7:])
@@ -1207 +1211 @@
 
     # Create the computational engines
+    if opts['qmin'] is None:
+        opts['qmin'] = 0.001*opts['qmax']
     if opts['datafile'] is not None:
         data = load_data(os.path.expanduser(opts['datafile']))

sasmodels/data.py

@@ -438 +438 @@
 
     scale = data.x**4 if view == 'q4' else 1.0
+    xscale = yscale = 'linear' if view == 'linear' else 'log'
 
     if use_data or use_theory:
@@ -470 +471 @@
             plt.ylim(*limits)
 
-        plt.xscale('linear' if not some_present or non_positive_x
-                   else view if view is not None
-                   else 'log')
-        plt.yscale('linear'
-                   if view == 'q4' or not some_present or not all_positive
-                   else view if view is not None
-                   else 'log')
+
+        xscale = ('linear' if not some_present or non_positive_x
+                  else view if view is not None
+                  else 'log')
+        yscale = ('linear'
+                  if view == 'q4' or not some_present or not all_positive
+                  else view if view is not None
+                  else 'log')
+        plt.xscale(xscale)
         plt.xlabel("$q$/A$^{-1}$")
+        plt.yscale(yscale)
         plt.ylabel('$I(q)$')
         title = ("data and model" if use_theory and use_data
@@ -505 +509 @@
         plt.xlabel("$q$/A$^{-1}$")
         plt.ylabel('residuals')
-        plt.xscale('linear')
         plt.title('(model - Iq)/dIq')
+        plt.xscale(xscale)
+        plt.yscale('linear')