"""
Creating meshes from vtu surface files
======================================

For this example we create meshes from surface layers.
"""

# %%
from shutil import which

import ogstools as ot
from ogstools import meshlib as ml
from ogstools.examples import surface_paths

# %%
# The loaded surfaces are defined within VTU files and adhere to properties such
# as non-intersecting boundaries with consistent x and y bounds. Alternatively,
# surfaces can also be created using PyVista with the same properties.

surface1 = ml.Surface(surface_paths[0], material_id=0)
surface2 = ml.Surface(surface_paths[1], material_id=5)
surface3 = ml.Surface(surface_paths[2], material_id=2)
surface4 = ml.Surface(surface_paths[3], material_id=3)

# %%
# Create 3 layers from previously defined surfaces and add all layers to a
# layerset (ordererd from top to bottom)
layer1 = ml.Layer(top=surface1, bottom=surface2, num_subdivisions=2)
layer2 = ml.Layer(top=surface2, bottom=surface3, num_subdivisions=1)
layer3 = ml.Layer(top=surface3, bottom=surface4, num_subdivisions=0)

layer_set1 = ml.LayerSet(layers=[layer1, layer2, layer3])

# %%
# From layerset creation of simplified meshes (sm), prism meshes (pm), voxel
# meshes (vm), tetraeder mesh (tm) is possible.
sm = ml.to_region_simplified(layer_set1, xy_resolution=200, rank=3).mesh
pm = ml.to_region_prism(layer_set1, resolution=200).mesh
vm = ml.to_region_voxel(layer_set1, resolution=[200, 200, 50]).mesh
tetgen_present = (
    which("tetgen") is not None
)  # an optional requirement (needs to be installed on system or via pip)
if tetgen_present:
    tm = ml.to_region_tetraeder(layer_set1, resolution=200).mesh
else:
    tm = None

# %% [markdown]
# Simplified mesh
# ---------------

# %%
ot.plot.contourf(sm.scale([1, 1, 5]), ot.variables.material_id).show()

# %% [markdown]
# Voxel mesh
# ---------------

# %%
ot.plot.contourf(vm.scale([1, 1, 5]), ot.variables.material_id).show()

# %% [markdown]
# Prism mesh
# ---------------

# %%
ot.plot.contourf(pm.scale([1, 1, 5]), ot.variables.material_id).show()

# %% [markdown]
# Tetraeder mesh
# ---------------

# %%
if tm:
    ot.plot.contourf(tm.scale([1, 1, 5]), ot.variables.material_id).show()