Aquila-Consortium/libsharp2
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

support for polarization

Browse source
This commit is contained in:
Andrea Zonca 2017-11-15 07:26:20 -08:00
parent 07da503877
commit 8c33a5e699
1 changed files with 25 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

31
python/libsharp/libsharp.pyx
View file

@ -247,17 +247,36 @@ cdef class alm_info:
sharp_destroy_alm_info(self.ainfo) sharp_destroy_alm_info(self.ainfo)
self.ainfo = NULL self.ainfo = NULL
def almxfl(self, np.ndarray[double, ndim=3, mode='c'] alm, np.ndarray[double, ndim=1, mode='c'] fl, int rank): def almxfl(self, np.ndarray[double, ndim=3, mode='c'] alm, np.ndarray[double, mode='c'] fl, int rank):
"""Multiply Alm by a Ell based array """Multiply Alms by a Ell based array
For example beam smoothing"""
Parameters
----------
alm : np.ndarray
input alm, 3 dimensions = (different signal x polarizations x lm-ordering)
fl : np.ndarray
either 1 dimension, e.g. gaussian beam, or 2 dimensions e.g. a polarized beam
Returns
-------
None, it modifies alms in-place
"""
mvstart = 0 mvstart = 0
for m in self.mval(): for m in self.mval():
f = 1 if (m==0) else 2 f = 1 if (m==0) else 2
num_ells = self.ainfo.lmax + 1 - m num_ells = self.ainfo.lmax + 1 - m
for i_l in range(num_ells):
l = m + i_l has_multiple_beams = alm.shape[2] > 1 and fl.ndim > 1
alm[:,:,mvstart + f*i_l:mvstart + f*i_l +f] *= fl[l] if not has_multiple_beams:
for i_l in range(num_ells):
l = m + i_l
alm[:,:,mvstart + f*i_l:mvstart + f*i_l +f] *= fl[l]
else:
for i_l in range(num_ells):
l = m + i_l
alm[:,:,mvstart + f*i_l:mvstart + f*i_l +f] *= fl[:alm.shape[2],l]
mvstart += f * num_ells mvstart += f * num_ells
cdef class triangular_order(alm_info): cdef class triangular_order(alm_info):