diff --git a/README.md b/README.md index 0133d8f69ae5c4c66b0c4b4cbd7ecc427936ae76..373780304cb52013b1198cb8a9e549ea405b6f98 100644 --- a/README.md +++ b/README.md @@ -5,6 +5,8 @@ pyTurbulence is a Python package for generating synthetic compressible turbulent TODO: - Add dilatational velocity fluctuations +  + ## Features - Generate solenoidal velocity fields using the method of Rogallo. diff --git a/pyTurbulence/solver.py b/pyTurbulence/solver.py index 53beea3c1d3f8889a98d808185f27476dbb57b17..a5f3201ec3ec6e8c38d1d4c3dca1d09ca8b1db9c 100644 --- a/pyTurbulence/solver.py +++ b/pyTurbulence/solver.py @@ -1,6 +1,7 @@ import numpy as np from pyTurbulence.syntheticTurbulence import compute_solenoidal_velocities, compute_solenoidal_pressure, compute_thermodynamic_fields from pyTurbulence.spectrum import compute_tke_spectrum, plot_spectrum, energy_spectrum +from pyTurbulence.plot3D import plot3D import time import os @@ -44,11 +45,15 @@ def solve(params, results_dir): start = time.time() knyquist, wave_numbers, tke_spectrum = compute_tke_spectrum(u, v, w, domain_size[0], domain_size[1], domain_size[2]) real_spectrum = energy_spectrum(spectrum, wave_numbers, urms, k0) - spectrum_path = os.path.join(results_dir, "spectrum.png") + spectrum_path = os.path.join(results_dir, "spectrum.pdf") plot_spectrum(wmax, wave_numbers, tke_spectrum, real_spectrum=real_spectrum, save_path=spectrum_path) end = time.time() spectrum_time = end - start + # Plot the 3D fields + # plot3D(u, domain_size, domain_resolution, name="u") + + # Save the data start = time.time() u.tofile(os.path.join(results_dir, "u.dat"))