import firedrake
import viskex
import common_firedrake as common # isort: skip
Generate meshes of the unit cube by dividing each edge of the cube in 6 segments, using either a tetrahedral or hexahedral mesh.
cube_tetra = firedrake.UnitCubeMesh(6, 6, 6, hexahedral=False)
cube_hexa = firedrake.UnitCubeMesh(6, 6, 6, hexahedral=True)
Interpolate the vector field $(x^3 + y^2 + z^4, x^6 + y^5 + z^7, x^9 + y^8 + z^{10})$, and store it either in a dolfinx function or a UFL expression.
cube_tetra_vector_field, cube_tetra_vector_field_ufl = common.prepare_vector_field_cases(
cube_tetra, lambda x: (x[0]**3 + x[1]**2 + x[2]**4, x[0]**6 + x[1]**5 + x[2]**7, x[0]**9 + x[1]**8 + x[2]**10))
cube_hexa_vector_field, cube_hexa_vector_field_ufl = common.prepare_vector_field_cases(
cube_hexa, lambda x: (x[0]**3 + x[1]**2 + x[2]**4, x[0]**6 + x[1]**5 + x[2]**7, x[0]**9 + x[1]**8 + x[2]**10))
Plot the vector field with a contour plot.
viskex.firedrake.plot_vector_field(cube_tetra_vector_field, "vector")
error: XDG_RUNTIME_DIR is invalid or not set in the environment. MESA: error: ZINK: failed to choose pdev glx: failed to create drisw screen
viskex.firedrake.plot_vector_field(cube_tetra_vector_field_ufl, "vector")
error: XDG_RUNTIME_DIR is invalid or not set in the environment. MESA: error: ZINK: failed to choose pdev glx: failed to create drisw screen
# PYTEST_XFAIL: Temporarily broken due to firedrake#3315
"""Expect this cell to fail.
viskex.firedrake.plot_vector_field(cube_hexa_vector_field, "vector")
"""
'Expect this cell to fail.\n\nviskex.firedrake.plot_vector_field(cube_hexa_vector_field, "vector")\n'
# PYTEST_XFAIL: Temporarily broken due to firedrake#3315
"""Expect this cell to fail.
viskex.firedrake.plot_vector_field(cube_hexa_vector_field_ufl, "vector")
"""
'Expect this cell to fail.\n\nviskex.firedrake.plot_vector_field(cube_hexa_vector_field_ufl, "vector")\n'
Plot the vector field with a quiver plot.
viskex.firedrake.plot_vector_field(cube_tetra_vector_field, "vector", glyph_factor=0.1)
error: XDG_RUNTIME_DIR is invalid or not set in the environment. MESA: error: ZINK: failed to choose pdev glx: failed to create drisw screen
# PYTEST_XFAIL: Temporarily broken due to firedrake#3315
"""Expect this cell to fail.
viskex.firedrake.plot_vector_field(cube_hexa_vector_field, "vector", glyph_factor=0.1)
"""
'Expect this cell to fail.\n\nviskex.firedrake.plot_vector_field(cube_hexa_vector_field, "vector", glyph_factor=0.1)\n'
Apply the vector field as a deformation to the mesh.
viskex.firedrake.plot_vector_field(cube_tetra_vector_field, "vector", warp_factor=1.0)
error: XDG_RUNTIME_DIR is invalid or not set in the environment. MESA: error: ZINK: failed to choose pdev glx: failed to create drisw screen
# PYTEST_XFAIL: Temporarily broken due to firedrake#3315
"""Expect this cell to fail.
viskex.firedrake.plot_vector_field(cube_hexa_vector_field, "vector", warp_factor=1.0)
"""
'Expect this cell to fail.\n\nviskex.firedrake.plot_vector_field(cube_hexa_vector_field, "vector", warp_factor=1.0)\n'