Changeset b9578fc in sasmodels for explore

Timestamp:

Oct 27, 2017 6:54:21 AM (7 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

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

Children:

a9bc435

Parents:

f39d4a3

Message:

alternative handling of phi point density: scale phi

File:

• explore/jitter.py (modified) (2 diffs)

explore/jitter.py

@@ -15 +15 @@
 import numpy as np
 from numpy import pi, cos, sin, sqrt, exp, degrees, radians
+
+SCALED_PHI = True
 
 def draw_beam(ax, view=(0, 0)):
@@ -177 +179 @@
         raise ValueError("expected dist to be 'gaussian' or 'rectangle'")
 
+    if SCALED_PHI:
+        scale_phi = lambda dtheta, dphi: (
+            dphi/abs(cos(radians(dtheta))) if dtheta != 90
+            else 0 if dphi == 0
+            else 4*pi)
+        w = np.outer(weights, weights)
+    else:
+        scale_phi = lambda dtheta, dphe: dphi
+        w = np.outer(weights*cos(radians(dtheta*t)), weights)
+
     # mesh in theta, phi formed by rotating z
     z = np.matrix([[0], [0], [radius]])
-    points = np.hstack([Rx(phi_i)*Ry(theta_i)*z
+    points = np.hstack([Rx(scale_phi(theta_i, phi_j))*Ry(theta_i)*z
                         for theta_i in dtheta*t
-                        for phi_i in dphi*t])
+                        for phi_j in dphi*t])
+    # select just the active points (i.e., those with phi < 180
+    active = np.array([abs(scale_phi(theta_i, phi_j)) < 180
+                       for theta_i in dtheta*t
+                       for phi_j in dphi*t])
+    points = points[:, active]
+    w = w.flatten()[active]
+
     # rotate relative to beam
     points = orient_relative_to_beam(view, points)
 
-    w = np.outer(weights*cos(radians(dtheta*t)), weights)
-
     x, y, z = [np.array(v).flatten() for v in points]
-    ax.scatter(x, y, z, c=w.flatten(), marker='o', vmin=0., vmax=1.)
+    ax.scatter(x, y, z, c=w, marker='o', vmin=0., vmax=1.)
 
 def draw_labels(ax, view, jitter, text):