def log_likelihood(N_nu=mycosmo['N_nu'], Y_He=mycosmo['Y_He'], h=mycosmo['h'], n=mycosmo['n'], omega_M_0=mycosmo['omega_M_0'], omega_b_0=mycosmo['omega_b_0'], omega_lambda_0=mycosmo['omega_lambda_0'], omega_n_0=mycosmo['omega_n_0'], sigma_8=mycosmo['sigma_8'], t_0=mycosmo['t_0'], tau=mycosmo['tau'], w=mycosmo['w'], z_reion=mycosmo['z_reion'],nmassless=mycosmo['Num_Nu_massless'],nmassive=mycosmo['Num_Nu_massive'],library='astropy'):
cosmo=mcmc.get_cosmology(N_nu,Y_He,h,n,omega_M_0,omega_b_0,omega_lambda_0,omega_n_0,sigma_8,t_0,tau,w,z_reion,nmassless,nmassive,library=library)
if library == 'astropy': cosast=astropy.cosmology.wCDM(cosmo['h']*100,cosmo['omega_M_0'],cosmo['omega_lambda_0'],w0=cosmo['w'])
try:
if library == 'cosmolopy':
#print('log_likelihood Beutler Da: cosmolopy')
daval=cosmolopy.distance.angular_diameter_distance(z,**cosmo)
elif library == 'astropy':
#print('log_likelihood Beutler Da: astropy')
daval=cosast.angular_diameter_distance(z)
elif library == 'jc':
#print('log_likelihood Beutler Da: jc')
daval=cosmo_utils.angdist(z,**cosmo)
except:
daval=-1.
try:
if library == 'cosmolopy':
#print('log_likelihood Beutler Hz: cosmolopy')
hval=cosmolopy.distance.e_z(z,**cosmo)*cosmo['h']*100
elif library == 'astropy':
#print('log_likelihood Beutler Hz: astropy')
hval=100*cosmo['h']/cosast.inv_efunc(z)
elif library == 'jc':
#print('log_likelihood Beutler Hz: jc')
hval=cosmo_utils.e_z(z,**cosmo)*cosmo['h']*100
except:
hval=-1.
dvval=((1+z)**2*daval**2*cc*z/hval)**(1./3)
rsval=cosmo_utils.rs_zdrag_fast_camb(**cosmo)*corr
rs_dv=rsval/dvval
chi2=(rs_dv-rs_dv_mes)**2/drs_dv_mes**2
return(-0.5*chi2)
def my_da_rs(h=h,om=om,w=w,ob=ob):
cosmo=cosmolopy.fidcosmo.copy()
cosmo['h']=h
cosmo['omega_M_0']=om
cosmo['omega_lambda_0']=1.-om
cosmo['omega_k_0']=0.
cosmo['omega_b_0']=ob
cosmo['w']=w
#cosast=astropy.cosmology.wCDM(cosmo['h']*100,cosmo['omega_M_0'],cosmo['omega_lambda_0'],w0=cosmo['w'])
try:
daval=cosmo_utils.angdist(zlya,**cosmo)
#daval=cosmolopy.distance.angular_diameter_distance(zlya,**cosmo)
#daval=cosast.angular_diameter_distance(zlya)
except ValueError:
daval=-1.
rsval=cosmo_utils.rs(**cosmo)
theda_rs=daval/rsval
return(theda_rs)
def thelogproba_ext(h=h,om=om,w=w,ob=ob):
cosmo=cosmolopy.fidcosmo.copy()
cosmo['h']=h
cosmo['omega_M_0']=om
cosmo['omega_lambda_0']=1.-om
cosmo['omega_k_0']=0.
cosmo['omega_b_0']=ob
cosmo['w']=w
try:
daval=cosmo_utils.angdist(zlya,**cosmo)
except ValueError:
daval=-1.
try:
hval=cosmo_utils.e_z(zlya,**cosmo)*cosmo['h']
except ValueError:
hval=-1.
rsval=cosmo_utils.rs(**cosmo)
invhrs=1./(hval*rsval)
da_rs=daval/rsval
vallogprob=logproba_interp(invhrs,da_rs)
return(vallogprob,invhrs,da_rs)
def log_likelihood(N_nu=np.array(mycosmo['N_nu']), Y_He=np.array(mycosmo['Y_He']), h=np.array(mycosmo['h']), n=np.array(mycosmo['n']), omega_M_0=np.array(mycosmo['omega_M_0']), omega_b_0=np.array(mycosmo['omega_b_0']), omega_lambda_0=np.array(mycosmo['omega_lambda_0']), omega_n_0=np.array(mycosmo['omega_n_0']), sigma_8=np.array(mycosmo['sigma_8']), t_0=np.array(mycosmo['t_0']), tau=np.array(mycosmo['tau']), w=np.array(mycosmo['w']), z_reion=np.array(mycosmo['z_reion']),nmassless=mycosmo['Num_Nu_massless'],nmassive=mycosmo['Num_Nu_massive'],library='astropy'):
cosmo=mcmc.get_cosmology(N_nu,Y_He,h,n,omega_M_0,omega_b_0,omega_lambda_0,omega_n_0,sigma_8,t_0,tau,w,z_reion,nmassless,nmassive,library=library)
#print(cosmo['h'],cosmo['omega_M_0'],cosmo['omega_lambda_0'],cosmo['omega_M_0']+cosmo['omega_lambda_0'],cosmo['omega_k_0'],cosmo['w'])
if library == 'astropy': cosast=astropy.cosmology.wCDM(cosmo['h']*100,cosmo['omega_M_0'],cosmo['omega_lambda_0'],w0=cosmo['w'])
try:
if library == 'cosmolopy':
#print('log_likelihood lyaDR11 Da: cosmolopy')
daval=cosmolopy.distance.angular_diameter_distance(zlya,**cosmo)
elif library == 'astropy':
#print('log_likelihood lyaDR11 Da: astropy')
daval=cosast.angular_diameter_distance(zlya)
elif library == 'jc':
#print('log_likelihood lyaDR11 Da: jc')
daval=cosmo_utils.angdist(zlya,**cosmo)
except:
daval=-1.
try:
if library == 'cosmolopy':
#print('log_likelihood lyaDR11 Hz: cosmolopy')
hval=cosmolopy.distance.e_z(zlya,**cosmo)*cosmo['h']
elif library == 'astropy':
#print('log_likelihood lyaDR11 Hz: astropy')
hval=cosmo['h']/cosast.inv_efunc(zlya)
elif library == 'jc':
#print('log_likelihood lyaDR11 Hz: jc')
hval=cosmo_utils.e_z(zlya,**cosmo)*cosmo['h']
except:
hval=-1.
rsval=cosmo_utils.rs_zdrag_fast_camb(**cosmo)
invhrs=1./(hval*rsval)
da_rs=daval/rsval
valprob=likelihood_interp(invhrs,da_rs)
return(np.log(valprob))
names=np.loadtxt(rep+'base_planck_lowl_lowLike.paramnames',dtype='str',usecols=[0])
planck=dict([names[i],planck_chains[:,i+2]] for i in range(np.size(names)))
sz=len(planck['H0*'])
#### Calculate Da and H for Planck
davals=np.zeros((sz,nbz))
hvals=np.zeros((sz,nbz))
for i in np.arange(sz):
print(i)
cosmo=mycosmo.copy()
cosmo['h']=planck['H0*'][i]/100
cosmo['omega_M_0']=planck['omegam*'][i]
cosmo['omega_lambda_0']=planck['omegal*'][i]
cosmo['omega_k_0']=1.-cosmo['omega_M_0']-cosmo['omega_lambda_0']
cosmo['w']=-1.
davals[i,:]=cosmo_utils.angdist(zvals,**cosmo)
hvals[i,:]=cosmo_utils.e_z(zvals,**cosmo)*cosmo['h']
### modele autre
cosmonew=cosmo.copy()
cosmonew['h']=planck['H0*'][i]/100
cosmonew['omega_M_0']=0.26
cosmonew['omega_k_0']=-0.007
cosmonew['omega_lambda_0']=1-cosmonew['omega_M_0']-cosmonew['omega_k_0']
cosmonew['w']=-1.
dazarb=cosmo_utils.angdist(zvals,**cosmonew)
hzarb=cosmo_utils.e_z(zvals,**cosmonew)*cosmonew['h']
mda=np.zeros(nbz)
import astropy
cosast=astropy.cosmology.wCDM(H0=cosm[4]*100,Om0=cosm[0],Ode0=cosm[1],w0=cosm[2])
#### my own code
from McMc import cosmo_utils
reload(cosmo_utils)
cosmojc=cosmolopy.fidcosmo.copy()
cosmojc['h']=cosm[4]
cosmojc['omega_M_0']=cosm[0]
cosmojc['omega_lambda_0']=cosm[1]
cosmojc['omega_k_0']=1.-(cosm[0]+cosm[1])
cosmojc['w']=cosm[2]
cosmolopy.distance.angular_diameter_distance(zlya,**cosmol)
cosast.angular_diameter_distance(zlya)
cosmo_utils.angdist(zlya,accurate=True,**cosmojc)
### Systematic comparison of astropy and cosmolopy and IDL
wvals=linspace(-3,0,100)
da_astropy=np.zeros(100)
da_cosmolopy=np.zeros(100)
for i in np.arange(100):
cosmol=cosmolopy.fidcosmo.copy()
cosmol['h']=cosm[4]
cosmol['omega_M_0']=cosm[0]
cosmol['omega_lambda_0']=cosm[1]
cosmol['omega_k_0']=1.-(cosm[0]+cosm[1])
##### there is a bug in the cosmolopy.e_z code, a factor 3 missing so one needs to correct w the following way
cosmol['w']=(wvals[i]+1)*3-1
da_cosmolopy[i]=cosmolopy.distance.angular_diameter_distance(zlya,**cosmol)
cosast=astropy.cosmology.wCDM(H0=cosm[4]*100,Om0=cosm[0],Ode0=cosm[1],w0=wvals[i])
