Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion flip/__init__.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
"""Init file of the flip package."""

import os

from . import covariance, fitter, gridding, likelihood, utils, power_spectra
from . import covariance, fisher, fitter, gridding, likelihood, power_spectra, utils

__version__ = "1.0.0"
__flip_dir_path__ = os.path.dirname(__file__)
299 changes: 299 additions & 0 deletions flip/covariance/adamsblake17plane/fisher_terms.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,299 @@
import numpy as np


def get_partial_derivative_coefficients(
model_type,
parameter_values_dict,
variant=None,
):
if model_type == "density":
return get_partial_derivative_coefficients_density(
parameter_values_dict,
variant=variant,
)
elif model_type == "velocity":
return get_partial_derivative_coefficients_velocity(
parameter_values_dict,
variant=variant,
)
elif model_type == "density_velocity":
return get_partial_derivative_coefficients_density_velocity(
parameter_values_dict,
variant=variant,
)
elif model_type == "full":
return get_partial_derivative_coefficients_full(
parameter_values_dict,
variant=variant,
)


def get_partial_derivative_coefficients_velocity(
parameter_values_dict,
variant=None,
):
if variant == "growth_index":
partial_coefficients_dict = {
"Omegam": {
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["gamma"]
* parameter_values_dict["s8"] ** 2
/ parameter_values_dict["Omegam"],
],
},
"gamma": {
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["s8"] ** 2
* np.log(parameter_values_dict["Omegam"]),
],
},
"s8": {
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["s8"],
],
},
"bs8": {
"vv": [
0,
],
},
}
else:
partial_coefficients_dict = {
"fs8": {
"vv": [
2 * parameter_values_dict["fs8"],
],
},
"bs8": {
"vv": [
0,
],
},
}
return partial_coefficients_dict


def get_partial_derivative_coefficients_density(
parameter_values_dict,
variant=None,
):
if variant == "growth_index":
partial_coefficients_dict = {
"Omegam": {
"gg": [
0,
],
},
"gamma": {
"gg": [
0,
],
},
"s8": {
"gg": [
0,
],
},
"bs8": {
"gg": [
2 * parameter_values_dict["bs8"],
],
},
}
else:
partial_coefficients_dict = {
"fs8": {
"gg": [
0,
],
},
"bs8": {
"gg": [
2 * parameter_values_dict["bs8"],
],
},
}
return partial_coefficients_dict


def get_partial_derivative_coefficients_density_velocity(
parameter_values_dict,
variant=None,
):
if variant == "growth_index":
partial_coefficients_dict = {
"Omegam": {
"gg": [
0,
],
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["gamma"]
* parameter_values_dict["s8"] ** 2
/ parameter_values_dict["Omegam"],
],
},
"gamma": {
"gg": [
0,
],
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["s8"] ** 2
* np.log(parameter_values_dict["Omegam"]),
],
},
"s8": {
"gg": [
0,
],
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["s8"],
],
},
"bs8": {
"gg": [
2 * parameter_values_dict["bs8"],
],
"vv": [
0,
],
},
}
else:
partial_coefficients_dict = {
"fs8": {
"gg": [
0,
],
"vv": [
2 * parameter_values_dict["fs8"],
],
},
"bs8": {
"gg": [
2 * parameter_values_dict["bs8"],
],
"vv": [
0,
],
},
}
return partial_coefficients_dict


def get_partial_derivative_coefficients_full(
parameter_values_dict,
variant=None,
):
if variant == "growth_index":
partial_coefficients_dict = {
"Omegam": {
"gg": [
0,
],
"gv": [
parameter_values_dict["Omegam"] ** parameter_values_dict["gamma"]
* parameter_values_dict["bs8"]
* parameter_values_dict["gamma"]
* parameter_values_dict["s8"]
/ parameter_values_dict["Omegam"],
],
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["gamma"]
* parameter_values_dict["s8"] ** 2
/ parameter_values_dict["Omegam"],
],
},
"gamma": {
"gg": [
0,
],
"gv": [
parameter_values_dict["Omegam"] ** parameter_values_dict["gamma"]
* parameter_values_dict["bs8"]
* parameter_values_dict["s8"]
* np.log(parameter_values_dict["Omegam"]),
],
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["s8"] ** 2
* np.log(parameter_values_dict["Omegam"]),
],
},
"s8": {
"gg": [
0,
],
"gv": [
parameter_values_dict["Omegam"] ** parameter_values_dict["gamma"]
* parameter_values_dict["bs8"],
],
"vv": [
2
* parameter_values_dict["Omegam"]
** (2 * parameter_values_dict["gamma"])
* parameter_values_dict["s8"],
],
},
"bs8": {
"gg": [
2 * parameter_values_dict["bs8"],
],
"gv": [
parameter_values_dict["Omegam"] ** parameter_values_dict["gamma"]
* parameter_values_dict["s8"],
],
"vv": [
0,
],
},
}
else:
partial_coefficients_dict = {
"fs8": {
"gg": [
0,
],
"gv": [
parameter_values_dict["bs8"],
],
"vv": [
2 * parameter_values_dict["fs8"],
],
},
"bs8": {
"gg": [
2 * parameter_values_dict["bs8"],
],
"gv": [
parameter_values_dict["fs8"],
],
"vv": [
0,
],
},
}
return partial_coefficients_dict
Loading