ehpk_bm Namespace Reference

Functions
def	do_all (z_arr, k_arr, cpar, prefix)

Variables
float	TCMB = 2.725
int	PLOT_STUFF = 0
int	WRITE_STUFF = 1
int	FS = 16
int	LKMAX = 7

Function Documentation

def ehpk_bm.do_all	(		z_arr,
			k_arr,
			cpar,
			prefix
	)

Definition at line 11 of file ehpk_bm.py.

def do_all(z_arr,k_arr,cpar,prefix) :
    pcs=csm.PcsPar()
    pcs.background_set(cpar['om'],cpar['ol'],cpar['ob'],cpar['w0'],cpar['wa'],cpar['hh'],TCMB)
    pcs.set_linear_pk('EH',-3,LKMAX,0.01,cpar['ns'],cpar['s8'])

    gf0=pcs.growth_factor(1)
    a_arr=1./(z_arr+1)
    gf_arr=np.array([pcs.growth_factor(a) for a in a_arr])
    pk_arr=np.array([[pcs.Pk_linear_0(k)*(gf/gf0)**2 for k in k_arr] for gf in gf_arr])

    if PLOT_STUFF==1 :
        for i in np.arange(len(pk_arr)) :
            plt.plot(k_arr,pk_arr[i],label='$z=%.2lf$'%(z_arr[i]))
        plt.xlabel('$k\\,[h\\,{\\rm Mpc}^{-1}]$',fontsize=FS)
        plt.ylabel('$P(k)\\,[{\\rm Mpc}\\,h^{-1}]^3$',fontsize=FS)
        plt.gca().set_xscale('log');
        plt.gca().set_yscale('log');
        plt.legend(loc='lower left')
        plt.show()

    if WRITE_STUFF==1 :
        header_pk="[0] k (Mpc/h)^-1"
        for i in np.arange(len(z_arr)) :
            header_pk+=", [%d]"%(i+2)+" P(k,z=%.1lf) (Mpc/h)^3"%(z_arr[i])
        np.savetxt(prefix+"_pk_eh.txt",np.transpose(np.vstack((k_arr,pk_arr))),header=header_pk)

z_arr=np.array([0.,1.,2.,3.,4.,5.])
lk_arr=0.1*np.arange(41)-3.
k_arr=10**lk_arr

cpar_model1={'om': 0.3,'ol': 0.7,'ob':0.05,'hh': 0.7,'s8': 0.8,'ns': 0.96,'w0': -1.0, 'wa': 0.0}

do_all(z_arr,k_arr,cpar_model1,"model1")

Variable Documentation

int ehpk_bm.FS = 16

Definition at line 8 of file ehpk_bm.py.

int ehpk_bm.LKMAX = 7

Definition at line 9 of file ehpk_bm.py.

int ehpk_bm.PLOT_STUFF = 0

Definition at line 6 of file ehpk_bm.py.

float ehpk_bm.TCMB = 2.725

Definition at line 5 of file ehpk_bm.py.

int ehpk_bm.WRITE_STUFF = 1

Definition at line 7 of file ehpk_bm.py.