def make_plot(radius_profile, brunt_profile, mass, age): figure = pyplot.figure() semilogy(radius_profile, -brunt_profile, 'g-', label = r'convective, $N^2$ < 0') semilogy(radius_profile, brunt_profile, 'r-', label = r'radiative, $N^2$ > 0') xlabel('Radius') ylabel(r'$\|N^2\|$') pyplot.title('Brunt-Vaisala frequency squared of a {0} star at {1}'.format(mass, age)) pyplot.legend(loc=3) pyplot.show()
def internal_energy_comparison_plot(radii_SE, u_SE, radii_SPH, u_SPH, figname): if not HAS_MATPLOTLIB: return pyplot.figure(figsize = (7, 5)) semilogy(radii_SE.as_quantity_in(units.RSun), u_SE, label='stellar evolution model') semilogy(radii_SPH, u_SPH, 'go', label='SPH model') xlabel('radius') ylabel('internal energy') pyplot.legend() pyplot.savefig(figname) print "\nPlot of internal energy profiles was saved to: ", figname pyplot.close()
def internal_energy_comparison_plot(radii_SE, u_SE, radii_SPH, u_SPH, figname): if not HAS_MATPLOTLIB: return pyplot.figure(figsize=(7, 5)) semilogy(radii_SE.as_quantity_in(units.RSun), u_SE, label='stellar evolution model') semilogy(radii_SPH, u_SPH, 'go', label='SPH model') xlabel('radius') ylabel('internal energy') pyplot.legend() pyplot.savefig(figname) print("\nPlot of internal energy profiles was saved to: ", figname) pyplot.close()
def make_plot(radius_profile, brunt_profile, mass, age): figure = pyplot.figure() semilogy(radius_profile, -brunt_profile, 'g-', label=r'convective, $N^2$ < 0') semilogy(radius_profile, brunt_profile, 'r-', label=r'radiative, $N^2$ > 0') xlabel('Radius') ylabel(r'$\|N^2\|$') pyplot.title('Brunt-Vaisala frequency squared of a {0} star at {1}'.format( mass, age)) pyplot.legend(loc=3) pyplot.show()
def temperature_density_plot(data, mass, age): figure = pyplot.figure(figsize=(8, 10)) pyplot.subplot(2, 1, 1) ax = pyplot.gca() plotT = semilogy(data["radius"], data["temperature"], 'r-', label=r'$T(r)$') xlabel('Radius') ylabel('Temperature') ax.twinx() plotrho = semilogy(data["radius"], data["density"], 'g-', label=r'$\rho(r)$') plots = plotT + plotrho labels = [one_plot.get_label() for one_plot in plots] ax.legend(plots, labels, loc=3) ylabel('Density') pyplot.subplot(2, 1, 2) semilogy( data["radius"], data["composition"][0], label=data["species_names"][0]) semilogy( data["radius"], data["composition"][1], label=data["species_names"][1]) semilogy( data["radius"], data["composition"][2], label=data["species_names"][2]) semilogy( data["radius"], data["composition"][3], label=data["species_names"][3]) semilogy( data["radius"], data["composition"][4], label=data["species_names"][4]) pyplot.ylim(0.0, 1.0) xlabel('Radius') ylabel('Mass fraction') pyplot.legend() pyplot.suptitle('Structure of a {0} star at {1}'.format(mass, age)) pyplot.show()
def temperature_density_plot(data, mass, age): figure = pyplot.figure(figsize=(8, 10)) pyplot.subplot(2, 1, 1) ax = pyplot.gca() plotT = semilogy(data["radius"], data["temperature"], 'r-', label=r'$T(r)$') xlabel('Radius') ylabel('Temperature') ax.twinx() plotrho = semilogy(data["radius"], data["density"], 'g-', label=r'$\rho(r)$') plots = plotT + plotrho labels = [one_plot.get_label() for one_plot in plots] ax.legend(plots, labels, loc=3) ylabel('Density') pyplot.subplot(2, 1, 2) semilogy(data["radius"], data["composition"][0], label=data["species_names"][0]) semilogy(data["radius"], data["composition"][1], label=data["species_names"][1]) semilogy(data["radius"], data["composition"][2], label=data["species_names"][2]) semilogy(data["radius"], data["composition"][3], label=data["species_names"][3]) semilogy(data["radius"], data["composition"][4], label=data["species_names"][4]) pyplot.ylim(0.0, 1.0) xlabel('Radius') ylabel('Mass fraction') pyplot.legend() pyplot.suptitle('Structure of a {0} star at {1}'.format(mass, age)) pyplot.show()