def Pg_Delta_m(model, lcdm_model, f_NL, **cosmo):
Pg_Delta_m = var.galaxy_power(model, lcdm_model, f_NL, **cosmo)
loglog(cc.k, Pg_Delta_m, next(linecycler), linewidth = 2, label = model + "_%s" %f_NL)
legend(loc = 'best')
axvline(x = 0.33 * 10**-3, color='r', ls = '--') # vertical line at k = H_0
axvline(x = 0.14 * 10**-1, color='b', ls = '-.') # vertical line at k_eq
xlim((cc.k_min, cc.k_max))
xlabel('k')
ylabel('$Pg_m$(k, a = 1)')
def update(val):
f_NL = sf_NL.val
gamma = Decimal(sgamma.val).quantize(Decimal('0.1'),rounding=ROUND_DOWN)
z = Decimal(int(sz.val)).quantize(Decimal('0.0'),rounding=ROUND_DOWN)
w_x = Decimal(sw_x.val).quantize(Decimal('0.1'),rounding=ROUND_DOWN)
ref_l_model1 = "lcdm_%s" %z
ref_w_model1 = "wcdm_%s_%s_%s" %(z, w_x, cc.c2_s[0])
ref_g_model1 = "gwcdm_%s_%s_%s_%s_%s" %(z, w_x, cc.c2_s[0], gamma, cc.Inrct)
Pg_Delta_m = var.galaxy_power(ref_w_model1, ref_l_model1, f_NL, **cosmo)
Pg_Delta_m11 = var.galaxy_power(ref_g_model1, ref_l_model1, f_NL0, **cosmo)
Pg_Delta_mt = var.galaxy_power(ref_g_model1, ref_l_model1, f_NL, **cosmo)
l.set_ydata(Pg_Delta_m0)
l1.set_ydata(Pg_Delta_m)
l2.set_ydata(Pg_Delta_m11)
l3.set_ydata(Pg_Delta_mt)
draw()
def IN_Pg_var_plot():
#---------- Cosmo parameters -----------------
cosmo = {'omega_M_0': 0.315, 'omega_b_0': 0.045, 'omega_lambda_0': 1 - 0.315}
ax = subplot(111)
subplots_adjust(left=0.25, bottom=0.25)
f_NL0 = cc.f_NL[0]
f_NL1 = cc.f_NL[1]
#galaxy_power(model, lcdm_model, f_NL, **cosmo)
ref_l_model = "lcdm_%s" %cc.z[0]
ref_w_model = "wcdm_%s_%s_%s" %(cc.z[0], cc.w_x[0], cc.c2_s[0])
ref_g_model = "gwcdm_%s_%s_%s_%s_%s" %(cc.z[0], cc.w_x[0], cc.c2_s[0], cc.gamma[2], cc.Inrct)
Pg_Delta_m0 = var.galaxy_power(ref_w_model, ref_l_model, f_NL0, **cosmo)
Pg_Delta_m1 = var.galaxy_power(ref_w_model, ref_l_model, f_NL1, **cosmo)
Pg_Delta_m01 = var.galaxy_power(ref_g_model, ref_l_model, f_NL0, **cosmo)
PPg_Delta_m01 = var.galaxy_power(ref_g_model, ref_l_model, f_NL0, **cosmo)[0]
f_NL = linspace(f_NL0, f_NL1, 1000)
for i in range(len(f_NL)):
PPg_Delta_m1 = var.galaxy_power(ref_w_model, ref_l_model, f_NL[i], **cosmo)[0]
if PPg_Delta_m01 >= PPg_Delta_m1:
f_NL01 = f_NL[i]
else:
exit
print "Effective f_NL: ", f_NL01
text(0.05, 0.9,'$f^{eff}_{NL}$ = %s' %round(f_NL01, 3) , ha ='left', va ='top', transform = ax.transAxes)
l, = loglog(cc.k, Pg_Delta_m0, '-', linewidth = 2)
l1, = loglog(cc.k, Pg_Delta_m1, '--', linewidth = 2, label = "wcdm - $f_{NL}$")
l2, = loglog(cc.k, Pg_Delta_m01, '-.', linewidth = 2, label = "$\Gamma$-wcdm")
l3, = loglog(cc.k, Pg_Delta_m01, ':', linewidth = 2, label = "$\Gamma$-wcdm + $f_{NL}$" )
axvline(x = 0.17 * 10**-2, color='r', ls = '--') # vertical line at k_eq
axvline(x = 0.14 * 10**-1, color='b', ls = '-.') # vertical line at k_eq
xlim((cc.k_min, cc.k_max))
#xlim((10**-3, cc.k_max))
axvline(x = 10**-3, color='g', ls = '--') # vertical line at k_eq
legend(loc = 'best')
xlabel('k')
ylabel('$Pg_m$(k, a = 1)')
axcolor = 'lightgoldenrodyellow'
axf_NL = axes([0.25, 0.05, 0.65, 0.03], axisbg=axcolor)
axgamma = axes([0.25, 0.1, 0.65, 0.03], axisbg=axcolor)
axw_x = axes([0.25, 0.15, 0.65, 0.03], axisbg=axcolor)
axz = axes([0.025, 0.2, 0.1, 0.04], axisbg=axcolor)
sf_NL = Slider(axf_NL, '$f_{NL}$', f_NL0, f_NL1, valfmt='%1.3f', valinit = f_NL01)
sgamma = DiscreteSlider(axgamma, '$\Gamma$', 0.0, 1.0, allowed_vals = cc.gamma, valfmt='%1.1f', valinit = cc.gamma[2])
sz = DiscreteSlider(axz,'z', cc.z[0], cc.z[-1], valfmt='%1.1f' , allowed_vals = cc.z, valinit = cc.z[0])
sw_x = DiscreteSlider(axw_x, '$w_x$', cc.w_x[0], cc.w_x[-1], allowed_vals = cc.w_x , valfmt='%1.1f', valinit = cc.w_x[0])
def update(val):
f_NL = sf_NL.val
gamma = Decimal(sgamma.val).quantize(Decimal('0.1'),rounding=ROUND_DOWN)
z = Decimal(int(sz.val)).quantize(Decimal('0.0'),rounding=ROUND_DOWN)
w_x = Decimal(sw_x.val).quantize(Decimal('0.1'),rounding=ROUND_DOWN)
ref_l_model1 = "lcdm_%s" %z
ref_w_model1 = "wcdm_%s_%s_%s" %(z, w_x, cc.c2_s[0])
ref_g_model1 = "gwcdm_%s_%s_%s_%s_%s" %(z, w_x, cc.c2_s[0], gamma, cc.Inrct)
Pg_Delta_m = var.galaxy_power(ref_w_model1, ref_l_model1, f_NL, **cosmo)
Pg_Delta_m11 = var.galaxy_power(ref_g_model1, ref_l_model1, f_NL0, **cosmo)
Pg_Delta_mt = var.galaxy_power(ref_g_model1, ref_l_model1, f_NL, **cosmo)
l.set_ydata(Pg_Delta_m0)
l1.set_ydata(Pg_Delta_m)
l2.set_ydata(Pg_Delta_m11)
l3.set_ydata(Pg_Delta_mt)
draw()
sf_NL.on_changed(update)
sgamma.on_changed(update)
sz.on_changed(update)
sw_x.on_changed(update)
resetax = axes([0.8, 0.0, 0.1, 0.04])
button = Button(resetax, 'Reset', color=axcolor, hovercolor='0.975')
def reset(event):
sf_NL.reset()
sgamma.reset()
sz.reset()
sw_x.reset()
button.on_clicked(reset)
show()
return
