diff --git a/analysis/slices.py b/analysis/slices.py index 581927d..8b260cc 100644 --- a/analysis/slices.py +++ b/analysis/slices.py @@ -34,7 +34,7 @@ def plot_imshow_with_reference( data_list, elif isinstance(L, int) or isinstance(L, float): L = [L for data in data_list] - sep = 10 if L[0] < 100 else 20 if L[0] < 200 else 100 + sep = 10 if L[0] < 50 else 20 if L[0] < 200 else 100 ticks = [np.arange(0, l+1, sep)*len(dat)/l for l, dat in zip(L,data_list)] tick_labels = [np.arange(0, l+1, sep) for l in L] @@ -86,10 +86,11 @@ def plot_imshow_with_reference( data_list, for i, data in enumerate(data_list): im = axes[i].imshow(data, cmap=cmap, origin='lower', vmin=vmin, vmax=vmax) axes[i].set_title(titles[i], fontsize=fs_titles) - axes[0, i].set_xticks(ticks[i]) - axes[0, i].set_yticks(ticks[i]) - axes[0, i].set_xticklabels(tick_labels[i]) - axes[0, i].set_yticklabels(tick_labels[i]) + axes[i].set_xticks(ticks[i]) + axes[i].set_yticks(ticks[i]) + axes[i].set_xticklabels(tick_labels[i]) + axes[i].set_yticklabels(tick_labels[i]) + axes[i].set_xlabel('Mpc/h') fig.colorbar(im, ax=axes[:], orientation='vertical') return fig, axes @@ -123,7 +124,7 @@ if __name__ == "__main__": fields = [read_field(args.directory+f) for f in args.filenames] if args.index is None: - index = ref_field.N0//2 + index = fields[0].N0//2 else: index=args.index diff --git a/cosmo_params.py b/cosmo_params.py index 8ca1636..14e48dc 100644 --- a/cosmo_params.py +++ b/cosmo_params.py @@ -63,4 +63,32 @@ def parse_arguments_cosmo(parsed_args): return cosmo_dict def z2a(z): - return 1.0/(1.0+z) \ No newline at end of file + return 1.0/(1.0+z) + + +cosmo_defaults = { + "h": 0.6732, + "Omega_m": 0.302, + "Omega_b": 0.049, + "Omega_q": 0.6842, + "Omega_k":0.0, + "Omega_r": 0.0, + "n_s": 0.968, + "sigma8": 0.815, + "A_s": 2.148752e-09, + "Tcmb": 2.7255, + "k_p": 0.05, + "N_ur": 2.046, + "m_nu1": 0.06, + "m_nu2": 0.0, + "m_nu3": 0.0, + "w_0": -1.0, + "w_a": 0.0, + "fnl": 100.0, + "gnl": 0.0, + "k_max":10.0, + "tau_reio":0.06, + "WhichSpectrum":"EH", + "w0_fld":-1.0, + "wa_fld":0.0, + } \ No newline at end of file