fluidsim/plot_results.py

import h5py
import matplotlib.pyplot as plt
import numpy as np

# Charger les données NetCDF directement
print("Chargement des données NetCDF...")

# État initial
with h5py.File("/root/Sim_data/NS2D_48x48_S8x8_2026-01-08_19-25-12/state_phys_t0000.000.nc", 'r') as f:
    rot_init = f['state_phys']['rot'][:]
    ux_init = f['state_phys']['ux'][:]
    uy_init = f['state_phys']['uy'][:]

# État final
with h5py.File("/root/Sim_data/NS2D_48x48_S8x8_2026-01-08_19-25-12/state_phys_t0010.200.nc", 'r') as f:
    rot_final = f['state_phys']['rot'][:]
    ux_final = f['state_phys']['ux'][:]
    uy_final = f['state_phys']['uy'][:]

print("Données chargées. Dimensions:", rot_init.shape)

# Tracer les champs physiques initiaux
print("Tracé des champs initiaux...")
fig, axes = plt.subplots(1, 3, figsize=(15, 5))

# Vorticité initiale
im1 = axes[0].imshow(rot_init.T, origin='lower', extent=[0, 8, 0, 8])
axes[0].set_title('Vorticité initiale (t=0)')
plt.colorbar(im1, ax=axes[0])

# Vitesse Ux initiale
im2 = axes[1].imshow(ux_init.T, origin='lower', extent=[0, 8, 0, 8])
axes[1].set_title('Vitesse Ux initiale (t=0)')
plt.colorbar(im2, ax=axes[1])

# Vitesse Uy initiale
im3 = axes[2].imshow(uy_init.T, origin='lower', extent=[0, 8, 0, 8])
axes[2].set_title('Vitesse Uy initiale (t=0)')
plt.colorbar(im3, ax=axes[2])

plt.tight_layout()
plt.savefig('simulation_initial.png', dpi=150, bbox_inches='tight')
print("Image sauvegardée: simulation_initial.png")

# Tracer les champs physiques finaux
print("Tracé des champs finaux...")
fig, axes = plt.subplots(1, 3, figsize=(15, 5))

# Vorticité finale
im1 = axes[0].imshow(rot_final.T, origin='lower', extent=[0, 8, 0, 8])
axes[0].set_title('Vorticité finale (t=10.2)')
plt.colorbar(im1, ax=axes[0])

# Vitesse Ux finale
im2 = axes[1].imshow(ux_final.T, origin='lower', extent=[0, 8, 0, 8])
axes[1].set_title('Vitesse Ux finale (t=10.2)')
plt.colorbar(im2, ax=axes[1])

# Vitesse Uy finale
im3 = axes[2].imshow(uy_final.T, origin='lower', extent=[0, 8, 0, 8])
axes[2].set_title('Vitesse Uy finale (t=10.2)')
plt.colorbar(im3, ax=axes[2])

plt.tight_layout()
plt.savefig('simulation_final.png', dpi=150, bbox_inches='tight')
print("Image sauvegardée: simulation_final.png")

print("Tracé terminé. Images sauvegardées: simulation_initial.png et simulation_final.png")