Getting Started with CosmoWAP

This guide provides examples of how to use CosmoWAP for various cosmological calculations.

Basic Setup

First, let’s set up a cosmology and survey parameters and initialize the main class (ClassWAP):

import numpy as np
import matplotlib.pyplot as plt
import cosmo_wap as cw
from cosmo_wap.lib import utils

# Initialize CLASS with Planck 2018 cosmology (default)
# Can pass: h, Omega_m, Omega_b, A_s, n_s, sigma8, k_max, z_max
cosmo = utils.get_cosmo()

# Get survey parameters for Euclid Hα survey (0.9 < z < 1.8)
survey_params = cw.SurveyParams.Euclid(cosmo)

# Initialize ClassWAP (the main interface)
cosmo_funcs = cw.ClassWAP(cosmo, survey_params)

Computing Power Spectra

Once you have a ClassWAP instance, you can very simply compute the power spectrum multipoles:

import cosmo_wap.pk as pk

# Define k values and redshift
k = np.linspace(0.01, 0.2, 50)
z = 1.0

# Compute the Newtonian plane-parallel monopole (l=0)
P_0 = pk.NPP.l0(cosmo_funcs, k, zz=z)

# Compute 2nd order wide-angle corrections
P_0_WA = pk.WA2.l0(cosmo_funcs, k, zz=z)

Computing Bispectra

Similarly for the bispectrum:

import cosmo_wap.bk as bk

# Define triangle configuration
k1, k2, k3 = 0.1, 0.1, 0.1  # equilateral
z = 1.0

# Compute the Newtonian monopole
B0 = bk.NPP.l0(cosmo_funcs, k1, k2, k3=k3, zz=z)

# Second-order relativistic corrections O((H/k)^2)
B0_GR = bk.GR2.l0(cosmo_funcs, k1, k2, k3=k3, zz=z)

Forecasting with Fisher Matrices

CosmoWAP includes a full forecasting pipeline for SNRs, Fisher matrices and MCMCs

Lets run through a basic example!

from cosmo_wap.forecast import FullForecast

# so we are using a Euclid-like survey with Euclid range (0.9,1.8)

# Create a forecast with 4 redshift bins, kmax = 0.15 h/Mpc
forecast = FullForecast(cosmo_funcs, kmax_func=0.15, N_bins=4)

# Compute Fisher matrix for cosmological parameters
# using Pk monopole (l=0) and quadrupole (l=2)
fisher = forecast.get_fish(
    ["A_s", "n_s", "h"],
    terms="NPP",          # Newtonian plane-parallel
    pkln=[0, 2],          # Pk multipoles
    bkln=None,            # No bispectrum multipoles
    verbose=True
)

# Access results
print("1-sigma errors:", fisher.errors)
print("Correlation matrix:\n", fisher.correlation)

# Get error on a specific parameter
sigma_ns = fisher.get_error("n_s")

See the Forecasting page for more details.

Example Notebooks

See example notebooks in the examples/ directory for usage of CosmoWAP:

example_pk.ipynb - Power spectrum calculations
example_bk.ipynb - Bispectrum calculations