-
Notifications
You must be signed in to change notification settings - Fork 0
/
IDM_Fisher_python.py
118 lines (92 loc) · 4.18 KB
/
IDM_Fisher_python.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Aug 2 11:39:44 2018
@author: aizhana
"""
import numpy as np
import matplotlib.pyplot as plt
import fishchips.experiments_change as experiments
from fishchips.cosmo import Observables
import fishchips.util
# Get all data points for all parameters
# from output .dat files
# after "=" write path to the .dat file with Pk data (..._pk) or Cl data (..._cl)
# Output files (from .ini files above) I used are in PT-check/class/test_output/
# for example, lcdm_pk = "/Users/aizhan.akh/Documents/Projects/PT-check/class/test_output/lcdm_pk.dat"
# for LCDM
fiducial_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_linear_cl_lensed.dat")[:,:]
# for dmeff
dmeff_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_dmeff_linear_cl_lensed.dat")[:,:]
#for omega_b
omega_b_l_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_l_omega_b_linear_cl_lensed.dat")[:,:]
omega_b_r_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_r_omega_b_linear_cl_lensed.dat")[:,:]
#for omega_cdm
omega_cdm_l_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_l_omega_cdm_linear_cl_lensed.dat")[:,:]
omega_cdm_r_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_r_omega_cdm_linear_cl_lensed.dat")[:,:]
#for tau
tau_l_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_l_tau_linear_cl_lensed.dat")[:,:]
tau_r_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_r_tau_linear_cl_lensed.dat")[:,:]
#for h
h_l_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_l_h_linear_cl_lensed.dat")[:,:]
h_r_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_r_h_linear_cl_lensed.dat")[:,:]
#for As
As_l_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_l_As_linear_cl_lensed.dat")[:,:]
As_r_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_r_As_linear_cl_lensed.dat")[:,:]
#for ns
ns_l_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_l_ns_linear_cl_lensed.dat")[:,:]
ns_r_data = np.genfromtxt("/u/aizhana/Projects/CodeCombined/Fisher/output/linear/lcdm_pt_r_ns_linear_cl_lensed.dat")[:,:]
pars = np.array( ['omega_b', 'omega_cdm', 'h', 'A_s', 'n_s', 'tau_reio'])
centers = np.array([0.02230, 0.1188, 0.6774, 2.142e-9, 0.9619, 0.66])
steps = np.array([0.0002230, 0.001188, 0.006774, 2.142e-11, 0.009619, 0.0066])
fiducial = {}
dmeff = {}
omega_b_l = {}
omega_b_r = {}
omega_cdm_l = {}
omega_cdm_r = {}
tau_l = {}
tau_r = {}
h_l = {}
h_r = {}
As_l = {}
As_r = {}
ns_l = {}
ns_r = {}
channels = ['ell', 'tt', 'ee', 'te', 'bb', 'pp', 'tp', 'ep']
for i in channels:
j = channels.index(i)
fiducial[i] = fiducial_data[:, j]
dmeff[i] = dmeff_data[:, j]
omega_b_l[i] = omega_b_l_data[:, j]
omega_b_r[i] = omega_b_r_data[:, j]
omega_cdm_l[i] = omega_cdm_l_data[:, j]
omega_cdm_r[i] = omega_cdm_r_data[:, j]
tau_l[i] = tau_l_data[:, j]
tau_r[i] = tau_r_data[:, j]
h_l[i] = h_l_data[:, j]
h_r[i] = h_r_data[:, j]
As_l[i] = As_l_data[:, j]
As_r[i] = As_r_data[:, j]
ns_l[i] = ns_l_data[:, j]
ns_r[i] = ns_r_data[:, j]
cl_left = []
cl_right = []
#pars = np.array( ['omega_b', 'omega_cdm', 'h', 'A_s', 'n_s', 'z_reio', 'sigma'])
cl_left.append(omega_b_l)
cl_left.append(omega_cdm_l)
cl_left.append(h_l)
cl_left.append(As_l)
cl_left.append(ns_l)
cl_left.append(tau_l)
#cl_left.append(fiducial)
cl_right.append(omega_b_r)
cl_right.append(omega_cdm_r)
cl_right.append(h_r)
cl_right.append(As_r)
cl_right.append(ns_r)
cl_right.append(tau_r)
#cl_right.append(dmeff)
#import fishchips.experiments_change as experiments_change
#example = experiments_change.CMB_Primary()
#fisher = example.get_fisher_different(obs)