nxs-analysis-tools 0.0.40__py3-none-any.whl → 0.0.41__py3-none-any.whl

_meta/__init__.py

@@ -6,5 +6,5 @@ __author__ = 'Steven J. Gomez Alvarado'
 __email__ = 'stevenjgomez@ucsb.edu'
 __copyright__ = f"2023, {__author__}"
 __license__ = 'MIT'
-__version__ = '0.0.40'
+__version__ = '0.0.41'
 __repo_url__ = 'https://github.com/stevenjgomez/nxs_analysis_tools'

nxs_analysis_tools/pairdistribution.py

@@ -1140,7 +1140,7 @@ class Interpolator:
         self.tapered = self.interpolated * self.window
 
 
-def fourier_transform_nxdata(data):
+def fourier_transform_nxdata(data, is_2d=False):
     """
     Perform a 3D Fourier Transform on the given NXdata object.
 
@@ -1156,6 +1156,10 @@ def fourier_transform_nxdata(data):
         include the `signal` field for the data and `axes` fields
         specifying the coordinate axes.
 
+    is_2d : bool
+        If true, skip FFT on out-of-plane direction and only do FFT
+        on axes 0 and 1. Default False.
+
     Returns
     -------
     NXdata
@@ -1163,16 +1167,13 @@ def fourier_transform_nxdata(data):
         result includes:
 
         - `dPDF`: The transformed data array.
-        - `x`, `y`, `z`: Arrays representing the frequency components along each axis.
+        - `x`, `y`, `z`: Arrays representing the real-space components along each axis.
 
     Notes
     -----
-    - The FFT is performed in two stages: first along the last dimension of
-    the input array and then along the first two dimensions.
-    - The function uses `pyfftw` for efficient computation of the Fourier
-    Transform.
-    - The output frequency components are computed based on the step sizes
-    of the original data axes.
+    - The FFT is performed in two stages: first along the last dimension of the input array and then along the first two dimensions.
+    - The function uses `pyfftw` for efficient computation of the Fourier Transform.
+    - The output frequency components are computed based on the step sizes of the original data axes.
 
     """
     start = time.time()
@@ -1181,7 +1182,9 @@ def fourier_transform_nxdata(data):
     padded = data[data.signal].nxdata
 
     fft_array = np.zeros(padded.shape)
+
     print("FFT on axes 1,2")
+
     for k in range(0, padded.shape[2]):
         fft_array[:, :, k] = np.real(
             np.fft.fftshift(
@@ -1190,17 +1193,18 @@ def fourier_transform_nxdata(data):
         )
         print(f'k={k}                  ', end='\r')
 
-    print("FFT on axis 3")
-    for i in range(0, padded.shape[0]):
-        for j in range(0, padded.shape[1]):
-            f_slice = fft_array[i, j, :]
-            print(f'i={i}                  ', end='\r')
-            fft_array[i, j, :] = np.real(
-                np.fft.fftshift(
-                    pyfftw.interfaces.numpy_fft.ifftn(np.fft.fftshift(f_slice),
-                                                      planner_effort='FFTW_MEASURE')
+    if not is_2d:
+        print("FFT on axis 3")
+        for i in range(0, padded.shape[0]):
+            for j in range(0, padded.shape[1]):
+                f_slice = fft_array[i, j, :]
+                print(f'i={i}                  ', end='\r')
+                fft_array[i, j, :] = np.real(
+                    np.fft.fftshift(
+                        pyfftw.interfaces.numpy_fft.ifftn(np.fft.fftshift(f_slice),
+                                                          planner_effort='FFTW_MEASURE')
+                    )
                 )
-            )
 
     end = time.time()
     print("FFT complete.")
@@ -1520,7 +1524,7 @@ class DeltaPDF:
         """
         self.padded = self.padder.pad(padding)
 
-    def perform_fft(self):
+    def perform_fft(self, is_2d=False):
         """
         Perform a 3D Fourier Transform on the padded data.
 
@@ -1548,4 +1552,4 @@ class DeltaPDF:
 
 
         """
-        self.fft = fourier_transform_nxdata(self.padded)
+        self.fft = fourier_transform_nxdata(self.padded, is_2d=is_2d)

{nxs_analysis_tools-0.0.40.dist-info → nxs_analysis_tools-0.0.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nxs-analysis-tools
-Version: 0.0.40
+Version: 0.0.41
 Summary: Reduce and transform nexus format (.nxs) scattering data.
 Author-email: "Steven J. Gomez Alvarado" <stevenjgomez@ucsb.edu>
 License: MIT License

nxs_analysis_tools-0.0.41.dist-info/RECORD

@@ -0,0 +1,11 @@
+_meta/__init__.py,sha256=QYocreoqhGDlovbW3V-ZkRa8aQqrIW5H9WvyI9XueaY,352
+nxs_analysis_tools/__init__.py,sha256=bxbTLpIcKasH3fuRZOvJ9zeu7IBBju82mOTgUV4ZqHE,530
+nxs_analysis_tools/chess.py,sha256=GUW08BcDUGD88MPZCl4yJTQp4hwd9tVNTJN_afNSUUo,24339
+nxs_analysis_tools/datareduction.py,sha256=e__mBGZIpJ-_-Dr6-T9hJURJazFnBy3PoWfwRxOra4U,41848
+nxs_analysis_tools/fitting.py,sha256=vPx75lKvm5pWOGBtRtff8k6J5dA6kRk3EJyzxCH5Tyk,8809
+nxs_analysis_tools/pairdistribution.py,sha256=bO-Ljst8eD5wmeq6oH8ylhHqaUyu2CvARznDupCygWQ,58245
+nxs_analysis_tools-0.0.41.dist-info/LICENSE,sha256=tdnoYVH1-ogW_5-gGs9bK-IkCamH1ATJqrdL37kWTHk,1102
+nxs_analysis_tools-0.0.41.dist-info/METADATA,sha256=2O7IvBuaAp_R_uXlwn-oZctEpIKGFlESR8gRAj_TP_U,3895
+nxs_analysis_tools-0.0.41.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+nxs_analysis_tools-0.0.41.dist-info/top_level.txt,sha256=8U000GNPzo6T6pOMjRdgOSO5heMzLMGjkxa1CDtyMHM,25
+nxs_analysis_tools-0.0.41.dist-info/RECORD,,

nxs_analysis_tools-0.0.40.dist-info/RECORD

@@ -1,11 +0,0 @@
-_meta/__init__.py,sha256=cgRdSYbkbQq7Gw011aNLk8NLfCZBtPjp2VfhtL5KeeE,352
-nxs_analysis_tools/__init__.py,sha256=bxbTLpIcKasH3fuRZOvJ9zeu7IBBju82mOTgUV4ZqHE,530
-nxs_analysis_tools/chess.py,sha256=GUW08BcDUGD88MPZCl4yJTQp4hwd9tVNTJN_afNSUUo,24339
-nxs_analysis_tools/datareduction.py,sha256=e__mBGZIpJ-_-Dr6-T9hJURJazFnBy3PoWfwRxOra4U,41848
-nxs_analysis_tools/fitting.py,sha256=vPx75lKvm5pWOGBtRtff8k6J5dA6kRk3EJyzxCH5Tyk,8809
-nxs_analysis_tools/pairdistribution.py,sha256=AtjllbbakFyo3MnGH18rvr6-h3ONopxXEmZK4F3VzMI,58023
-nxs_analysis_tools-0.0.40.dist-info/LICENSE,sha256=tdnoYVH1-ogW_5-gGs9bK-IkCamH1ATJqrdL37kWTHk,1102
-nxs_analysis_tools-0.0.40.dist-info/METADATA,sha256=H9FqaRexBO4zxSFcunsIieblI5jhUtosugg4sZVCMFU,3895
-nxs_analysis_tools-0.0.40.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-nxs_analysis_tools-0.0.40.dist-info/top_level.txt,sha256=8U000GNPzo6T6pOMjRdgOSO5heMzLMGjkxa1CDtyMHM,25
-nxs_analysis_tools-0.0.40.dist-info/RECORD,,