Changeset 9f91afe in sasmodels

Timestamp:

Apr 6, 2017 5:23:08 AM (7 years ago)

Author:

Adam Washington <adam.washington@…>

Branches:

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

Children:

668c38b

Parents:

cb0dc22

Message:

Mask out invlid q values for TOF

File:

• sasmodels/sesans.py (modified) (3 diffs)

sasmodels/sesans.py

@@ -41 +41 @@
     _H0 = None # type: np.ndarray
 
-    def __init__(self, z, SElength, zaccept, Rmax):
+    def __init__(self, z, SElength, lam, zaccept, Rmax):
         # type: (np.ndarray, float, float) -> None
         #import logging; logging.info("creating SESANS transform")
         self.q = z
-        self._set_hankel(SElength, zaccept, Rmax)
+        self._set_hankel(SElength, lam, zaccept, Rmax)
 
     def apply(self, Iq):
@@ -54 +54 @@
         return P
 
-    def _set_hankel(self, SElength, zaccept, Rmax):
+    def _set_hankel(self, SElength, lam, zaccept, Rmax):
         # type: (np.ndarray, float, float) -> None
         # Force float32 arrays, otherwise run into memory problems on some machines
@@ -71 +71 @@
         H = np.float32(dq/(2*pi)) * j0(repSE*repq) * repq
 
+        replam = np.tile(lam, (q.size, 1))
+        reptheta = np.arcsin(repq*replam/2*np.pi)
+        mask = reptheta > zaccept
+        H[mask] = 0
+
         self.q_calc = q
         self._H, self._H0 = H, H0