###################################################################
############# BAO Lyman-alpha DR11 ################################
from McMc import model_lyaDR11 as lya
reload(lya)
B=pymc.MCMC(lya)
B.use_step_method(pymc.AdaptiveMetropolis,B.stochastics,delay=1000)
B.sample(iter=50000,burn=10000,thin=10)
clf()
xlim(0,1)
ylim(0,1.5)
mcmc.cont_gkde(B.trace('om')[:],B.trace('ol')[:],nsig=5)
xx=np.linspace(0,1,1000)
plot(xx,1-xx,'k:')
reload(lya)
hvals=B.trace('h')[:]
omvals=B.trace('om')[:]
olvals=B.trace('ol')[:]
ll=np.zeros(hvals.size)
invhrs=np.zeros(hvals.size)
da_rs=np.zeros(hvals.size)
for i in np.arange(hvals.size):
print(i,hvals.size)
ll[i],invhrs[i],da_rs[i]=lya.theproba_ext(h=hvals[i], om=omvals[i],ol=olvals[i],ob=mycosmo['omega_b_0'])
mcmc.cont_gkde(invhrs,da_rs,fill=False,color='red',alpha=1)
obh2=Planck.trace('omega_b_0')[:]*Planck.trace('h')[:]**2
och2=(Planck.trace('omega_M_0')[:]-Planck.trace('omega_b_0')[:])*Planck.trace('h')[:]**2
ol=Planck.trace('omega_lambda_0')[:]
h=Planck.trace('h')[:]
om=Planck.trace('omega_M_0')[:]
ok=Planck.trace('omega_k_0')
clf()
subplot(2,2,1)
xlim(0.02,0.025)
ylim(0.60,0.82)
scatter(obh2,ol,c=h,linewidth=0)
colorbar()
mcmc.cont_gkde(obh2,ol,color='red',fill=False,alpha=1)
xlabel('$\Omega_b h^2$')
ylabel('$\Omega_\Lambda$')
subplot(2,2,2)
xlim(0.098,0.132)
ylim(0.60,0.82)
scatter(och2,ol,c=h,linewidth=0)
colorbar()
mcmc.cont_gkde(och2,ol,color='red',fill=False)
xlabel('$\Omega_c h^2$')
ylabel('$\Omega_\Lambda$')
subplot(2,2,3)
xlim(0.02,0.025)
ylim(0.096,0.132)
Andersonh.use_step_method(pymc.AdaptiveMetropolis,Andersonh.stochastics,delay=1000)
Andersonh.sample(iter=niter,burn=nburn,thin=nthin)
Andersonh.db.close()
### Anderson+Beutler+DR7+LyaDR11+h
AllBAOh=pymc.MCMC(mcmc.generic_model([data_Anderson,data_Beutler, data_DR7, data_lyaDR11, data_hPlanck],variables=variables,library=library),db='pickle',dbname='BAOh_'+model+'_'+library+'.db')
AllBAOh.use_step_method(pymc.AdaptiveMetropolis,AllBAOh.stochastics,delay=1000)
AllBAOh.sample(iter=niter,burn=nburn,thin=nthin)
AllBAOh.db.close()
reload(mcmc)
clf()
xlim(0,1)
ylim(0,1.5)
mcmc.cont_gkde(Beutlerh.trace('omega_M_0')[:],Beutlerh.trace('omega_lambda_0')[:],color='green')
mcmc.cont_gkde(DR7h.trace('omega_M_0')[:],DR7h.trace('omega_lambda_0')[:],color='orange')
mcmc.cont_gkde(Andersonh.trace('omega_M_0')[:],Andersonh.trace('omega_lambda_0')[:],color='pink')
mcmc.cont_gkde(BAOh.trace('omega_M_0')[:],BAOh.trace('omega_lambda_0')[:],color='blue')
mcmc.cont_gkde(AllBAOh.trace('omega_M_0')[:],AllBAOh.trace('omega_lambda_0')[:],color='red')
xx=np.linspace(0,1,1000)
plot(xx,1-xx,'k:')
range=[[-0.5,1.5],[-0.5,2.]]
reload(mcmc)
clf()
xlim(0,1)
ylim(0,1.5)
sm=2
mcmc.cont(Beutlerh.trace('omega_M_0')[:],Beutlerh.trace('omega_lambda_0')[:],color='green',nsmooth=sm)
mcmc.cont(DR7h.trace('omega_M_0')[:],DR7h.trace('omega_lambda_0')[:],color='orange',nsmooth=sm)
