fluidsim/run_taylor_green_proper.py

from fluidsim.solvers.ns2d.solver import Simul
import numpy as np

def init_taylor_green(sim):
    """Initialisation correcte des vortexes de Taylor-Green"""
    X, Y = sim.oper.X, sim.oper.Y

    # Vortexes de Taylor-Green classiques:
    # ux = -cos(X) * sin(Y)
    # uy = sin(X) * cos(Y)
    sim.state.init_from_uxuy_func(
        lambda x, y: -np.cos(x) * np.sin(y),
        lambda x, y: np.sin(x) * np.cos(y)
    )

# Créer les paramètres
params = Simul.create_default_params()
params.oper.type_fft = 'fft2d.with_pyfftw'
params.oper.nx = params.oper.ny = 48
params.oper.Lx = params.oper.Ly = 8
params.time_stepping.t_end = 5  # Temps plus court pour voir l'évolution
params.time_stepping.it_end = 5

# Utiliser l'initialisation personnalisée
params.init_fields.type = 'in_script'
params.init_fields._set_attrib('func', init_taylor_green)

# Créer et lancer la simulation
sim = Simul(params)
sim.time_stepping.start()

print("\n=== RÉSULTATS ===")
print("Simulation terminée avec succès !")
print(f"Énergie initiale calculée: {0.5 * np.mean(sim.state.get_var('ux')**2 + sim.state.get_var('uy')**2):.6f}")
print(f"Répertoire des résultats: {sim.output.path_run}")