Source code for firecrown.modeling_tools
"""Basic module for Cosmology and cosmological tools definitions.
This module contains the ModelingTools class, which is built around the
:class:`pyccl.Cosmology` class. This is used by likelihoods that need to access
reusable objects, such as perturbation theory or halo model calculators.
"""
from __future__ import annotations
from abc import ABC, abstractmethod
from typing import Collection
import pyccl.nl_pt
from firecrown.models.cluster.abundance import ClusterAbundance
from firecrown.updatable import Updatable, UpdatableCollection
[docs]class ModelingTools(Updatable):
"""Modeling tools for likelihoods.
A class that bundles together a :class:`pyccl.Cosmology` object and associated
objects, such as perturbation theory or halo model calculator workspaces.
"""
def __init__(
self,
*,
pt_calculator: None | pyccl.nl_pt.EulerianPTCalculator = None,
pk_modifiers: None | Collection[PowerspectrumModifier] = None,
cluster_abundance: None | ClusterAbundance = None,
):
super().__init__()
self.ccl_cosmo: None | pyccl.Cosmology = None
self.pt_calculator: None | pyccl.nl_pt.EulerianPTCalculator = pt_calculator
pk_modifiers = pk_modifiers if pk_modifiers is not None else []
self.pk_modifiers: UpdatableCollection = UpdatableCollection(pk_modifiers)
self.powerspectra: dict[str, pyccl.Pk2D] = {}
self._prepared: bool = False
self.cluster_abundance = cluster_abundance
[docs] def add_pk(self, name: str, powerspectrum: pyccl.Pk2D) -> None:
"""Add a :python:`pyccl.Pk2D` to the table of power spectra."""
if name in self.powerspectra:
raise KeyError(f"Power spectrum {name} already exists")
self.powerspectra[name] = powerspectrum
[docs] def get_pk(self, name: str) -> pyccl.Pk2D:
"""Access a power spectrum from the table of power spectra.
Either retrive a pyccl.Pk2D from the table of power spectra, or fall back
to what the pyccl.Cosmology object can provide.
"""
if self.ccl_cosmo is None:
raise RuntimeError("Cosmology has not been set")
if name in self.powerspectra:
return self.powerspectra[name]
return self.ccl_cosmo.get_nonlin_power(name)
[docs] def has_pk(self, name: str) -> bool:
"""Check if a power spectrum with name `name` is available."""
# There should probably a pyccl.Cosmology method to check if a specific
# power spectrum exists
try:
self.get_pk(name)
except KeyError:
return False
return True
[docs] def prepare(self, ccl_cosmo: pyccl.Cosmology) -> None:
"""Prepare the Cosmology for use in likelihoods.
This method will prepare the ModelingTools for use in likelihoods. This
includes building the perturbation theory and halo model calculators
if they are needed.
:param ccl_cosmo: the current CCL cosmology object
"""
if not self.is_updated():
raise RuntimeError("ModelingTools has not been updated.")
if self._prepared:
raise RuntimeError("ModelingTools has already been prepared")
if self.ccl_cosmo is not None:
raise RuntimeError("Cosmology has already been set")
self.ccl_cosmo = ccl_cosmo
if self.pt_calculator is not None:
self.pt_calculator.update_ingredients(ccl_cosmo)
for pkm in self.pk_modifiers:
self.add_pk(name=pkm.name, powerspectrum=pkm.compute_p_of_k_z(tools=self))
if self.cluster_abundance is not None:
self.cluster_abundance.update_ingredients(ccl_cosmo)
self._prepared = True
[docs] def _reset(self) -> None:
"""Resets all CCL objects in ModelingTools.
This method is called by the Updatable base class when the object is
destroyed. It also resets the power spectra, the cosmology and the
_prepared state variable.
"""
self.ccl_cosmo = None
# Also reset the power spectra
# TODO: is that always needed?
self.powerspectra = {}
self._prepared = False
[docs] def get_ccl_cosmology(self) -> pyccl.Cosmology:
"""Return the CCL cosmology object."""
if self.ccl_cosmo is None:
raise RuntimeError("Cosmology has not been set")
return self.ccl_cosmo
[docs] def get_pt_calculator(self) -> pyccl.nl_pt.EulerianPTCalculator:
"""Return the perturbation theory calculator object."""
if self.pt_calculator is None:
raise RuntimeError("A PT calculator has not been set")
return self.pt_calculator