guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-05-14 03:51:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

adding cic compensation tools

Browse source
This commit is contained in:
EiffL 2022-03-26 00:06:34 +01:00
parent 079cebbdea
commit 3e1b3d8a3b
2 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
jaxpm/kernels.py
View file

@ -83,3 +83,18 @@ def longrange_kernel(kvec, r_split):
return np.exp(-kk * r_split**2)
else:
return 1.
def cic_compensation(kvec):
"""
Computes cic compensation kernel.
Adapted from https://github.com/bccp/nbodykit/blob/a387cf429d8cb4a07bb19e3b4325ffdf279a131e/nbodykit/source/mesh/catalog.py#L499
Itself based on equation 18 (with p=2) of
`Jing et al 2005 <https://arxiv.org/abs/astro-ph/0409240>`_
Args:
kvec: array of k values in Fourier space
Returns:
v: array of kernel
"""
kwts = [np.sinc(kvec[i] / (2 * np.pi)) for i in range(3)]
wts = (kwts[0] * kwts[1] * kwts[2])**(-2)
return wts