From 3ac0f054882e23fc72560dfc4154f6e02925831b Mon Sep 17 00:00:00 2001
From: Amaury Bilocq <amaurybilocq@mac.home>
Date: Sat, 22 Feb 2025 00:12:37 +0100
Subject: [PATCH] Init doc folder - add pictures to readme

---
 README.md              | 2 ++
 pyTurbulence/solver.py | 7 ++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 0133d8f..3737803 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
 
+![alt text](image.png) ![alt text](image-1.png)
+
 ## Features
 
 - Generate solenoidal velocity fields using the method of Rogallo.
diff --git a/pyTurbulence/solver.py b/pyTurbulence/solver.py
index 53beea3..a5f3201 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"))
-- 
GitLab