PyCosmoMMF 0.0.2__tar.gz → 0.0.4__tar.gz

{PyCosmoMMF-0.0.2 → PyCosmoMMF-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyCosmoMMF
-Version: 0.0.2
+Version: 0.0.4
 Summary: A package for detecting structures in the Cosmic Web.
 Home-page: https://github.com/James11222/PyCosmoMMF/
 Author: James Sunseri

{PyCosmoMMF-0.0.2 → PyCosmoMMF-0.0.4}/PyCosmoMMF/filter.py

@@ -50,6 +50,62 @@ def kspace_gaussian_filter(R_S, kv):
                     -(kx[ix]**2 + ky[iy]**2 + kz[iz]**2) * R_S**2 / 2)
     return filter_k
 
+@nb.njit(parallel=True, fastmath=True)
+def kspace_top_hat_filter(R_S, kv):
+    """
+    create a top-hat filter in k-space.
+
+    Parameters
+    ----------
+    R_S : float
+        The smoothing scale.
+    kv : tuple
+        The wavevectors in each dimension.
+    
+    Returns
+    -------
+    filter_k : array
+        The filter in k-space.
+    """
+
+    kx, ky, kz = kv
+    nx, ny, nz = len(kx), len(ky), len(kz)
+    filter_k = np.zeros((nx, ny, nz))
+
+    for ix in nb.prange(nx):
+        for iy in range(ny):
+            for iz in range(nz):
+                k = np.sqrt(kx[ix]**2 + ky[iy]**2 + kz[iz]**2)
+                if k == 0.0:
+                    filter_k[ix, iy, iz] = 1.0
+                else:
+                    filter_k[ix, iy, iz] = 3 * (np.sin(k * R_S) - (k * R_S) * np.cos(k * R_S)) / (k * R_S)**3
+
+    return filter_k
+
+def smooth_top_hat(f, R_S, kv):
+    """
+    apply a top-hat filter to a field f.
+
+    Parameters
+    ----------
+    f : array
+        The field to be smoothed.
+    R_S : float
+        The smoothing scale.
+    kv : tuple
+        The wavevectors in each dimension.
+
+    Returns
+    -------
+    f_Rn : array
+        The smoothed field.
+    """
+    GF = kspace_top_hat_filter(R_S, kv)
+    f_Rn = np.real(np.fft.ifftn(GF * np.fft.fftn(f)))
+    f_Rn = f_Rn * (np.sum(f) / np.sum(f_Rn))
+    return f_Rn
+
 def smooth_gauss(f, R_S, kv):
     """
     apply a Gaussian filter to a field f.

{PyCosmoMMF-0.0.2 → PyCosmoMMF-0.0.4}/PyCosmoMMF.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyCosmoMMF
-Version: 0.0.2
+Version: 0.0.4
 Summary: A package for detecting structures in the Cosmic Web.
 Home-page: https://github.com/James11222/PyCosmoMMF/
 Author: James Sunseri

{PyCosmoMMF-0.0.2 → PyCosmoMMF-0.0.4}/setup.py

@@ -12,7 +12,7 @@ here = pathlib.Path(__file__).parent.resolve()
 
 setup(
     name='PyCosmoMMF',  # Required
-    version='0.0.2',  # Required
+    version='0.0.4',  # Required
     description='A package for detecting structures in the Cosmic Web.',  # Optional
     long_description=(here / 'README.md').read_text(encoding='utf-8'),  # Optional
     long_description_content_type='text/markdown',  # Optional (see note above)