# -*- coding: utf-8 -*-
# TorsoCAE Validation Journal — hyperelastic case 2: Treloar-style equibiaxial tension (3-term Ogden)
# See README.md in this directory for theory and results.
# disp=6.0 m drives this 4m sheet to lambda=2.5 (both in-plane axes), displacement-controlled,
# matching run_validation.py's approach -- see that file for the fitted material.

import sys
sys.path.insert(0, "./server/")
from pytorsocae import TorsoCAESession

session = TorsoCAESession()
session.csg_make_shape("box", {"dx": 4.0, "dy": 4.0, "dz": 0.5, "cx": 0, "cy": 0, "cz": 0},
                        name="EquibiaxialSheet", body_id="body_0")
session.csg_select("body_0")
session.mesh(mesh_id="mesh_0", algo_id="hex_transfinite", size_factor=1.2, size_min=0, size_max=0, order=1, dim=3)
session.solid("EquibiaxialSheet").material(E=210e9, nu=0.49,
                                            ogden_mu=[230881.77, 100.0, 100.0],
                                            ogden_alpha=[0.3192, 5.8917, -3.0290])
session.surface(1, scope_id="body_0").bc("fixed", frame="global", components=["x"])
session.surface(3, scope_id="body_0").bc("fixed", frame="global", components=["y"])
session.surface(5, scope_id="body_0").bc("fixed", frame="global", components=["z"])
session.surface(2, scope_id="body_0").bc("displacement", values=[6.0, 0, 0], components=["x"])   # -> lambda1 = 2.5
session.surface(4, scope_id="body_0").bc("displacement", values=[0, 6.0, 0], components=["y"])   # -> lambda2 = 2.5
session.set_physics("structural", submodel="hyperelastic-ogden", backend="dolfinx")
session.set_solver_options(algo="direct", precond="none", tol=1e-10, max_iter=1000,
                            num_steps=20, inner_tol=1e-9, max_inner_iter=60, n_cores=2, device="cpu")
session.compute(mesh_ids=["mesh_0"])