"""
===========================================================
Plot Young modulus as a 3D surface
===========================================================

This example shows how to plot the directional dependence of the Young modulus as a 3D surface.
"""
###############################################################################
# Define the stiffness tensor for NiTi
# -------------------------------------
#
# We define the stiffness tensor for a monoclinic NiTi material using its elastic constants. They are taken from the
# `Materials Project (mp-1048) <https://next-gen.materialsproject.org/materials/mp-1048>`_.
from elasticipy.tensors.elasticity import StiffnessTensor
C = StiffnessTensor.monoclinic(
    phase_name='TiNi',
    C11=231, C12=127, C13=104,
    C22=240, C23=131, C33=175,
    C44=81, C55=11, C66=85,
    C15=-18, C25=1, C35=-3, C46=3
)
print(C)

###############################################################################
# Get the Young modulus from the stiffness tensor
# -------------------------------------------------
E = C.Young_modulus
print("Young's modulus:")
print(E)

###############################################################################
# Plot it as a 3D surface
# --------------------------
E.plot3D()

###############################################################################
# Use Matplotlib instead of Plotly
# --------------------------------
# By default, elasticipy uses `Plotly <https://plotly.com/python/>`_ to render the surface, allowing interactive plot
# (see above). Matplolib can be used instead:
E.plot3D(backend='matplotlib')