Index

C

compute_all_slices() (in module s2fft.recursions.price_mcewen) compute_all_slices_jax() (in module s2fft.recursions.price_mcewen) compute_eighth() (in module s2fft.recursions.trapani) compute_full() (in module s2fft.recursions.risbo) (in module s2fft.recursions.risbo_jax) (in module s2fft.recursions.trapani) (in module s2fft.recursions.turok) compute_full_jax() (in module s2fft.recursions.trapani) compute_full_loop() (in module s2fft.recursions.trapani)

compute_full_vectorized() (in module s2fft.recursions.trapani) compute_quarter() (in module s2fft.recursions.trapani) (in module s2fft.recursions.turok) compute_quarter_jax() (in module s2fft.recursions.trapani) compute_quarter_slice() (in module s2fft.recursions.turok) compute_quarter_vectorized() (in module s2fft.recursions.trapani) compute_slice() (in module s2fft.recursions.turok) (in module s2fft.recursions.turok_jax) critical_log() (in module s2fft.logs)

D

debug_log() (in module s2fft.logs)

downsample_by_two_mwss() (in module s2fft.utils.resampling)

E

elm2ind() (in module s2fft.sampling.s2_samples)

elmn2ind() (in module s2fft.sampling.so3_samples)

F

f_shape() (in module s2fft.sampling.s2_samples) (in module s2fft.sampling.so3_samples) fill() (in module s2fft.recursions.turok) fill_eighth2quarter() (in module s2fft.recursions.trapani) fill_half2full() (in module s2fft.recursions.trapani) fill_half2full_jax() (in module s2fft.recursions.trapani) fill_half2full_vectorized() (in module s2fft.recursions.trapani) fill_quarter2half() (in module s2fft.recursions.trapani) fill_quarter2half_jax() (in module s2fft.recursions.trapani) fill_quarter2half_vectorized() (in module s2fft.recursions.trapani) flm_1d_to_2d() (in module s2fft.sampling.s2_samples) flm_2d_to_1d() (in module s2fft.sampling.s2_samples) flm_2d_to_hp() (in module s2fft.sampling.s2_samples) flm_hp_to_2d() (in module s2fft.sampling.s2_samples) flm_shape() (in module s2fft.sampling.s2_samples) flmn_1d_to_3d() (in module s2fft.sampling.so3_samples) flmn_3d_to_1d() (in module s2fft.sampling.so3_samples) flmn_shape() (in module s2fft.sampling.so3_samples) flmn_shape_1d() (in module s2fft.sampling.so3_samples) fnab_shape() (in module s2fft.sampling.so3_samples)

forward() (in module s2fft.base_transforms.spherical) (in module s2fft.base_transforms.wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) forward_jax() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) forward_jax_ssht() (in module s2fft.transforms.wigner) forward_latitudinal_step() (in module s2fft.transforms.otf_recursions) forward_latitudinal_step_jax() (in module s2fft.transforms.otf_recursions) forward_numpy() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) forward_transform() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) forward_transform_jax() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) forward_transform_torch() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) ftm_shape() (in module s2fft.sampling.s2_samples)

G

generate_flm() (in module s2fft.utils.signal_generator) generate_flmn() (in module s2fft.utils.signal_generator) generate_precomputes() (in module s2fft.recursions.price_mcewen)

generate_precomputes_jax() (in module s2fft.recursions.price_mcewen) generate_precomputes_wigner() (in module s2fft.recursions.price_mcewen) generate_precomputes_wigner_jax() (in module s2fft.recursions.price_mcewen) generate_rotate_dls() (in module s2fft.utils.rotation)

H

healpix_fft() (in module s2fft.utils.healpix_ffts) healpix_fft_jax() (in module s2fft.utils.healpix_ffts) healpix_fft_numpy() (in module s2fft.utils.healpix_ffts) healpix_fft_torch() (in module s2fft.utils.healpix_ffts) healpix_ifft() (in module s2fft.utils.healpix_ffts)

healpix_ifft_jax() (in module s2fft.utils.healpix_ffts) healpix_ifft_numpy() (in module s2fft.utils.healpix_ffts) healpix_ifft_torch() (in module s2fft.utils.healpix_ffts) healpix_phase_shifts() (in module s2fft.precompute_transforms.construct) hp_ang2pix() (in module s2fft.sampling.s2_samples) hp_getidx() (in module s2fft.sampling.s2_samples)

I

ind2elm() (in module s2fft.sampling.s2_samples) info_log() (in module s2fft.logs) init() (in module s2fft.recursions.trapani) init_jax() (in module s2fft.recursions.trapani) init_nonjax() (in module s2fft.recursions.trapani) inverse() (in module s2fft.base_transforms.spherical) (in module s2fft.base_transforms.wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) inverse_jax() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner)

inverse_jax_ssht() (in module s2fft.transforms.wigner) inverse_latitudinal_step() (in module s2fft.transforms.otf_recursions) inverse_latitudinal_step_jax() (in module s2fft.transforms.otf_recursions) inverse_numpy() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) inverse_transform() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) inverse_transform_jax() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) inverse_transform_torch() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner)

L

lm2lm_hp() (in module s2fft.sampling.s2_samples)

M

module s2fft.base_transforms.spherical s2fft.base_transforms.wigner s2fft.logs s2fft.precompute_transforms.construct s2fft.precompute_transforms.spherical s2fft.precompute_transforms.wigner s2fft.recursions.price_mcewen s2fft.recursions.risbo s2fft.recursions.risbo_jax s2fft.recursions.trapani s2fft.recursions.turok s2fft.recursions.turok_jax s2fft.sampling.s2_samples s2fft.sampling.so3_samples s2fft.transforms.c_backend_spherical s2fft.transforms.otf_recursions s2fft.transforms.spherical s2fft.transforms.wigner s2fft.utils s2fft.utils.healpix_ffts s2fft.utils.quadrature s2fft.utils.quadrature_jax s2fft.utils.quadrature_torch s2fft.utils.resampling s2fft.utils.resampling_jax s2fft.utils.resampling_torch s2fft.utils.rotation s2fft.utils.signal_generator

mw_to_mwss() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) mw_to_mwss_phi() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) mw_to_mwss_theta() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) mw_weights() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch)

N

ncoeff() (in module s2fft.sampling.s2_samples) nphi_equiang() (in module s2fft.sampling.s2_samples) nphi_equitorial_band() (in module s2fft.sampling.s2_samples)

nphi_ring() (in module s2fft.sampling.s2_samples) ntheta() (in module s2fft.sampling.s2_samples) ntheta_extension() (in module s2fft.sampling.s2_samples)

P

p2phi_equiang() (in module s2fft.sampling.s2_samples) p2phi_ring() (in module s2fft.sampling.s2_samples) p2phi_rings() (in module s2fft.utils.healpix_ffts) p2phi_rings_jax() (in module s2fft.utils.healpix_ffts) periodic_extension() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch)

periodic_extension_spatial_mwss() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) phis_equiang() (in module s2fft.sampling.s2_samples) phis_ring() (in module s2fft.sampling.s2_samples)

Q

quad_weight_dh_theta_only() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_dh() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_gl() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_hp() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch)

quad_weights_mw() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mw_theta_only() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mwss() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mwss_theta_only() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_transform() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch)

R

reindex() (in module s2fft.recursions.turok_jax) ring_phase_shift_hp() (in module s2fft.sampling.s2_samples)

ring_phase_shifts_hp() (in module s2fft.utils.healpix_ffts) ring_phase_shifts_hp_jax() (in module s2fft.utils.healpix_ffts) rotate_flms() (in module s2fft.utils.rotation)

S

s2fft.base_transforms.spherical module s2fft.base_transforms.wigner module s2fft.logs module s2fft.precompute_transforms.construct module s2fft.precompute_transforms.spherical module s2fft.precompute_transforms.wigner module s2fft.recursions.price_mcewen module s2fft.recursions.risbo module s2fft.recursions.risbo_jax module s2fft.recursions.trapani module s2fft.recursions.turok module s2fft.recursions.turok_jax module s2fft.sampling.s2_samples module s2fft.sampling.so3_samples module s2fft.transforms.c_backend_spherical module s2fft.transforms.otf_recursions module

s2fft.transforms.spherical module s2fft.transforms.wigner module s2fft.utils module s2fft.utils.healpix_ffts module s2fft.utils.quadrature module s2fft.utils.quadrature_jax module s2fft.utils.quadrature_torch module s2fft.utils.resampling module s2fft.utils.resampling_jax module s2fft.utils.resampling_torch module s2fft.utils.rotation module s2fft.utils.signal_generator module setup_logging() (in module s2fft.logs) spectral_folding() (in module s2fft.utils.healpix_ffts) spectral_folding_jax() (in module s2fft.utils.healpix_ffts) spectral_folding_torch() (in module s2fft.utils.healpix_ffts) spectral_periodic_extension() (in module s2fft.utils.healpix_ffts) spectral_periodic_extension_jax() (in module s2fft.utils.healpix_ffts) spectral_periodic_extension_torch() (in module s2fft.utils.healpix_ffts) spin_spherical_kernel() (in module s2fft.precompute_transforms.construct) spin_spherical_kernel_jax() (in module s2fft.precompute_transforms.construct)

T

t2theta() (in module s2fft.sampling.s2_samples)

thetas() (in module s2fft.sampling.s2_samples)

U

unextend() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) upsample_by_two_mwss() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch)

upsample_by_two_mwss_ext() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch)

W

warning_log() (in module s2fft.logs)

wigner_kernel() (in module s2fft.precompute_transforms.construct) wigner_kernel_jax() (in module s2fft.precompute_transforms.construct)