#=== Import ================================================
# imports-start
from mpi4py import MPI
import yaml
from pylava import LAVA
# imports-end
#===========================================================

#=== Main Script ===========================================
# loadyamls-start
# Read the YAML file name
with open("run.yaml", "r") as f:
   lava_options = yaml.safe_load(f)
# loadyamls-end

# init-start
# Setup the aero solver instance
AeroSolver = LAVA(lava_options,
                  comm=MPI.COMM_WORLD,
                  suffix="3D.NS1.INTEL.AVX2.OPENMPI-lib")
# init-end

# solve-start
# Run the LAVA solver
AeroSolver()
# solve-end

# func-start
# Evaluate some functionals
funcs = AeroSolver.evalFunction(['cl','cd'])

# Print the functionals to screen
if MPI.COMM_WORLD.rank == 0:
    print(f"C_L : {funcs['cl'][0]}")
    print(f"C_D : {funcs['cd'][0]}")
# func-end

# vol-write-start
# Write out the volume file
AeroSolver.writeVolSolution(
   filename = "vol_pylava.q",
   filetype="plot3d",
   fields=["pressure", "velocityx", "velocityy", 'velocityz', "temperature", "nut"]
)
# vol-write-end

# surf-write-start
# Write out the surface
AeroSolver.buildDVSurfaces(families=["fuse", "wing", "symm"])
AeroSolver.writeSurface(
   filename = "surface_pylava.plt",
   fields=["iblank", "pressure", "velocityx", "velocityy", 'velocityz', "temperature", "nut"]
)
# surf-write-end
#===========================================================