diff options
| -rwxr-xr-x | calc_mass_potential.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/calc_mass_potential.py b/calc_mass_potential.py index e61645b..3254648 100755 --- a/calc_mass_potential.py +++ b/calc_mass_potential.py @@ -6,6 +6,7 @@ # # Change logs: # 2016-07-10: +# * Allow disable the calculation of potential profile # * Use class 'SmoothSpline' from module 'spline.py' # 2016-07-04: # * Remove unnecessary configuration options @@ -142,6 +143,7 @@ rho_total_profile = rho_total_profile.txt rho_total_profile_image = rho_total_profile.png # output gravitational potential profile +# NOTE: to disable potential calculation, do not specified the output files potential_profile = potential_profile.txt potential_profile_image = potential_profile.png ------------------------------------------------------------ @@ -646,14 +648,16 @@ def main(): density_profile.plot(profile="rho_total", ax=ax, fig=fig) fig.savefig(config["rho_total_profile_image"], dpi=150) - density_profile.calc_potential(verbose=True) - density_profile.save(profile="potential", - outfile=config["potential_profile"]) - fig = Figure(figsize=(10, 8)) - FigureCanvas(fig) - ax = fig.add_subplot(1, 1, 1) - density_profile.plot(profile="potential", ax=ax, fig=fig) - fig.savefig(config["potential_profile_image"], dpi=150) + outfile_potential = config["potential_profile"] + if outfile_potential != "": + density_profile.calc_potential(verbose=True) + density_profile.save(profile="potential", + outfile=outfile_potential) + fig = Figure(figsize=(10, 8)) + FigureCanvas(fig) + ax = fig.add_subplot(1, 1, 1) + density_profile.plot(profile="potential", ax=ax, fig=fig) + fig.savefig(config["potential_profile_image"], dpi=150) if __name__ == "__main__": |