jax_fem.basis module

jax_fem.basis.get_elements(ele_type)

Mesh node ordering is important. If the input mesh file is Gmsh .msh or Abaqus .inp, meshio would convert it to its own ordering. My experience shows that meshio ordering is the same as Abaqus. For example, for a 10-node tetrahedron element, the ordering of meshio is the following https://web.mit.edu/calculix_v2.7/CalculiX/ccx_2.7/doc/ccx/node33.html The troublesome thing is that basix has a different ordering. As shown below https://defelement.com/elements/lagrange.html The consequence is that we need to define this “re_order” variable to make sure the ordering is correct.

jax_fem.basis.get_face_shape_vals_and_grads(ele_type, gauss_order=None)

TODO: Add comments

Returns:

• face_shape_vals (ndarray) – (6, 4, 8) = (num_faces, num_face_quads, num_nodes)

• face_shape_grads_ref (ndarray) – (6, 4, 3) = (num_faces, num_face_quads, num_nodes, dim)

• face_weights (ndarray) – (6, 4) = (num_faces, num_face_quads)

• face_normals (ndarray) – (6, 3) = (num_faces, dim)

• face_inds (ndarray) – (6, 4) = (num_faces, num_face_vertices)

jax_fem.basis.get_shape_vals_and_grads(ele_type, gauss_order=None)

TODO: Add comments

Returns:

• shape_values (ndarray) – (8, 8) = (num_quads, num_nodes)

• shape_grads_ref (ndarray) – (8, 8, 3) = (num_quads, num_nodes, dim)

• weights (ndarray) – (8,) = (num_quads,)

jax_fem.basis.reorder_inds(inds, re_order)