def setUpClass(cls): cls.cf = cambfast.CAMBfast() cls.cf.add_parameter('tau', 0.0522, 0.02, 0.1, 10, fixed=False) cls.cf.add_parameter('r', 0.01, 0.0, 0.1, 10, fixed=False) cls.cf.add_parameter('As', 2e-9, 1e-9, 3e-9, 10, fixed=False) cls.lmax = 60 cls.cf.generate_interp(cls.lmax, CMB_unit='muK') cls.cf.write_funcs('tau_r_As_lmax60.npz')
import time import numpy as np import cambfast from cambfast import get_spectrum_camb lmax = 50 npts = 200 cf = cambfast.CAMBfast() cf.inifile = '../ini/planck_2018.ini' cf.add_parameter('tau', 0.05, 0.003, 0.12, npts, fixed=False) cf.add_parameter('r', 0.05, -0.3, 0.3, npts, fixed=True) cf.generate_interp(lmax, CMB_unit='muK') #cf.write_funcs(f'As_tau_r_lmax{lmax}_npts{npts}.npz') cf.write_funcs(f'tau_lmax{lmax}_npts{npts}.npz')
import numpy as np import pylab as plt import time import cambfast cf = cambfast.CAMBfast() cf.load_funcs('./tau_r_lmax50_npts100.npz') cfmc = cambfast.CAMBfast() cfmc.load_MCdata('./tau_r_lmax50_npts5000_MC.npz', method='cubic') t0 = time.time() cl = cf.get_spectrum(lmax=50, tau=0.024827, r=0.02987) print(f'interpolation : {time.time()-t0} s') t0 = time.time() clmc = cfmc.get_spectrum(lmax=50, tau=0.024827, r=0.02987) #*(2.7255*1e6)**2 print(f'interpolation MC: {time.time()-t0} s') t0 = time.time() cl1 = cambfast.get_spectrum_camb(lmax=50, tau=0.024827, r=0.02987, CMB_unit='muK') print(f'CAMB : {time.time()-t0} s') par0 = cf.pars[0] par1 = cf.pars[1] ## grid vs random points par0, par1 = np.mgrid[par0.min:par0.max:1j * par0.nstep, par1.min:par1.max:1j * par1.nstep] rpts = cfmc.MCpoints plt.plot(par0, par1, 'b.', markersize=1)
def setUpClass(cls): cls.cf = cambfast.CAMBfast() cls.cf.load_funcs('tau_r_As_lmax60.npz') cls.lmax = 60