# Copyright (C) 2021-2025 Université Gustave Eiffel.
# This file is part of the EasyFEA project.
# EasyFEA is distributed under the terms of the GNU General Public License v3, see LICENSE.txt and CREDITS.md for more information.

"""
Hyperelas2
==========

A hyper elastic cube in compression.
"""
# sphinx_gallery_thumbnail_number = -1

from EasyFEA import Display, ElemType, Models, Simulations, PyVista, np
from EasyFEA.Geoms import Domain

if __name__ == "__main__":
    Display.Clear()

    # ----------------------------------------------
    # Configuration
    # ----------------------------------------------
    dim = 3

    # geom
    L = 1
    h = 1

    # ----------------------------------------------
    # Mesh
    # ----------------------------------------------
    meshSize = h / 10

    contour = Domain((0, 0), (L, h), h / 10)

    if dim == 2:
        mesh = contour.Mesh_2D([], ElemType.QUAD8, isOrganised=True)
    else:
        mesh = contour.Mesh_Extrude(
            [], [0, 0, h], [h / meshSize], ElemType.HEXA8, isOrganised=True
        )

    nodesX0 = mesh.Nodes_Conditions(lambda x, y, z: x == 0)
    nodesXL = mesh.Nodes_Conditions(lambda x, y, z: x == L)

    # ----------------------------------------------
    # Simulation
    # ----------------------------------------------

    isot = Models.ElasIsot(dim, E=1, v=0.3)
    lmbda = isot.get_lambda()
    mu = isot.get_mu()
    mat = Models.SaintVenantKirchhoff(dim, lmbda, mu, thickness=h)

    simu = Simulations.HyperElasticSimu(mesh, mat)

    uc = -0.3
    simu.add_dirichlet(nodesX0, [0] * dim, simu.Get_unknowns())
    values = np.zeros_like(simu.Get_unknowns(), dtype=float)
    values[0] = uc
    simu.add_dirichlet(nodesXL, values, simu.Get_unknowns())

    simu.Solve()

    # ----------------------------------------------
    # Results
    # ----------------------------------------------

    PyVista.Plot_BoundaryConditions(simu).show()
    PyVista.Plot(simu, "ux", 1, plotMesh=True).show()