jax_fem.utils module

jax_fem.utils.get_meshio_cell_type(ele_type)

Reference: nschloe/meshio

jax_fem.utils.json_parse(json_filepath)

jax_fem.utils.make_video(data_dir)

jax_fem.utils.modify_vtu_file(input_file_path, output_file_path)

Convert version 2.2 of vtu file to version 1.0 meshio does not accept version 2.2, raising error of meshio._exceptions.ReadError: Unknown VTU file version ‘2.2’.

jax_fem.utils.read_abaqus_and_write_vtk(abaqus_file, vtk_file)

Used for a quick inspection. Paraview can’t open .inp file so we convert it to .vtu

jax_fem.utils.save_sol(fe, sol, sol_file, cell_infos=None, point_infos=None)

jax_fem.utils.timeit(func)

jax_fem.utils.walltime(txt_dir=None, filename=None)

jax_fem.utils.wraps(wrapped, assigned=('__module__', '__name__', '__qualname__', '__doc__', '__annotations__'), updated=('__dict__',))

Decorator factory to apply update_wrapper() to a wrapper function

Returns a decorator that invokes update_wrapper() with the decorated function as the wrapper argument and the arguments to wraps() as the remaining arguments. Default arguments are as for update_wrapper(). This is a convenience function to simplify applying partial() to update_wrapper().