| def plotcommand(cosmology='WMAP5', plotname=None): | |
| """pass""" | |
| xarray = 10 ** np.arange(1, 15, 0.2) | |
| yval = 'c' | |
| zarray = np.arange(0, 5, 0.5) | |
| xtitle = 'Halo Mass (M$_{sol}$)' | |
| ytitle = 'Concentration' | |
| linelabel = 'z=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| plt.ylim([2, 30]) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=zval, Mi=xarray) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, yarray, label=linelabel + str(zval), color=colors[zind]) | |
| ax.plot(xarray, commah.commah.cduffy(zval, xarray), color='black') | |
| ax.set_xscale('log') | |
| ax.set_yscale('log') | |
| leg = ax.legend(loc=1) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_CM_relation.png'" % plotname) | |
| fig.savefig(plotname + '_CM_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(0, 1, 0.05) - 1 | |
| yval = 'c' | |
| zarray = 10 ** np.arange(6, 14, 2) | |
| xtitle = 'Redshift' | |
| ytitle = 'NFW Concentration' | |
| linelabel = 'log$_{10}$ M$_{z}$(M$_{sol}$)=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=xarray, Mi=zval) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, yarray, label=linelabel + '{0:.1f}'.format(np.log10(zval)), color=colors[zind]) | |
| leg = ax.legend(loc=1) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_Cz_relation.png'" % plotname) | |
| fig.savefig(plotname + '_Cz_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(0, 1, 0.05) - 1 | |
| yval = 'zf' | |
| zarray = 10 ** np.arange(6, 14, 2) | |
| xtitle = 'Redshift' | |
| ytitle = 'Formation Redshift' | |
| linelabel = 'log$_{10}$ M$_{z}$(M$_{sol}$)=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=xarray, Mi=zval) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, yarray, label=linelabel + '{0:.1f}'.format(np.log10(zval)), color=colors[zind]) | |
| leg = ax.legend(loc=2) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_zfz_relation.png'" % plotname) | |
| fig.savefig(plotname + '_zfz_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(0, 1, 0.05) - 1 | |
| yval = 'dMdt' | |
| zarray = 10 ** np.arange(10, 14, 0.5) | |
| xtitle = 'log$_{10}$ (1+z)' | |
| ytitle = 'log$_{10}$ Accretion Rate M$_{sol}$ yr$^{-1}$' | |
| linelabel = 'log$_{10}$ M$_z$(M$_{sol}$)=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| cosmo = commah.getcosmo(cosmology) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=xarray, Mi=zval, com=False, mah=True) | |
| yarray = output[yval].flatten() | |
| ax.plot(np.log10(xarray + 1.0), np.log10(yarray), label=linelabel + '{0:.1f}'.format(np.log10(zval)), color=colors[zind]) | |
| semianalytic_approx = 71.6 * (zval / 1000000000000.0) * (cosmo['h'] / 0.7) * (-0.24 + 0.75 * (xarray + 1)) * np.sqrt(cosmo['omega_M_0'] * (xarray + 1) ** 3 + cosmo['omega_lambda_0']) | |
| ax.plot(np.log10(xarray + 1), np.log10(semianalytic_approx), color='black') | |
| leg = ax.legend(loc=2) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_dMdtz_relation.png'" % plotname) | |
| fig.savefig(plotname + '_dMdtz_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(10, 14, 0.5) | |
| yval = 'dMdt' | |
| zarray = np.arange(0, 5, 0.5) | |
| xtitle = 'Halo Mass M$_{sol}$' | |
| ytitle = 'Accretion Rate M$_{sol}$ yr$^{-1}$' | |
| linelabel = 'z=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=zval, Mi=xarray, com=False, mah=True) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, yarray, label=linelabel + str(zval), color=colors[zind]) | |
| ax.set_xscale('log') | |
| ax.set_yscale('log') | |
| leg = ax.legend(loc=2) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_MAH_M_relation.png'" % plotname) | |
| fig.savefig(plotname + '_MAH_M_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(10, 14, 0.5) | |
| yval = 'dMdt' | |
| zarray = np.arange(0, 5, 0.5) | |
| xtitle = 'Halo Mass M$_{sol}$' | |
| ytitle = 'Specific Accretion Rate yr$^{-1}$' | |
| linelabel = 'z=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=zval, Mi=xarray, mah=True, com=False) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, yarray / xarray, label=linelabel + str(zval), color=colors[zind]) | |
| ax.set_xscale('log') | |
| ax.set_yscale('log') | |
| leg = ax.legend(loc=1) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_specificMAH_M_relation.png'" % plotname) | |
| fig.savefig(plotname + '_specificMAH_M_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(0, 1, 0.05) - 1 | |
| yval = 'Mz' | |
| zarray = 10 ** np.arange(10, 14, 0.5) | |
| xtitle = 'Redshift' | |
| ytitle = 'M(z) (M$_{sol}$)' | |
| linelabel = 'log$_{10}$ M$_{0}$(M$_{sol}$)=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=0, Mi=zval, z=xarray) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, yarray, label=linelabel + '{0:.1f}'.format(np.log10(zval)), color=colors[zind]) | |
| ax.set_yscale('log') | |
| leg = ax.legend(loc=1) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_Mzz_relation.png'" % plotname) | |
| fig.savefig(plotname + '_Mzz_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| xarray = 10 ** np.arange(0, 1, 0.02) - 1 | |
| yval = 'Mz' | |
| zarray = 10 ** np.arange(10, 14, 0.5) | |
| xtitle = 'Redshift' | |
| ytitle = 'log$_{10}$ M(z)/M$_{0}$' | |
| linelabel = 'log$_{10}$ M$_{0}$(M$_{sol}$)=' | |
| fig = plt.figure() | |
| ax = fig.add_subplot(111) | |
| ax.set_xlabel(xtitle) | |
| ax.set_ylabel(ytitle) | |
| colors = cm.rainbow(np.linspace(0, 1, len(zarray))) | |
| for zind, zval in enumerate(zarray): | |
| output = commah.run(cosmology=cosmology, zi=0, Mi=zval, z=xarray) | |
| yarray = output[yval].flatten() | |
| ax.plot(xarray, np.log10(yarray / zval), label=linelabel + '{0:.1f}'.format(np.log10(zval)), color=colors[zind]) | |
| leg = ax.legend(loc=3) | |
| leg.get_frame().set_alpha(0) | |
| leg.get_frame().set_edgecolor('white') | |
| for label in leg.get_texts(): | |
| label.set_fontsize('small') | |
| for label in leg.get_lines(): | |
| label.set_linewidth(4) | |
| if plotname: | |
| fig.tight_layout(pad=0.2) | |
| print("Plotting to '%s_MzM0z_relation.png'" % plotname) | |
| fig.savefig(plotname + '_MzM0z_relation.png', dpi=fig.dpi * 5) | |
| else: | |
| plt.show() | |
| return 'Done' |