Function main

scripts/map_to_sphere.py:101–116

()

    return pymesh.form_mesh(out_vertices, np.zeros((0, 3)), mesh.voxels);

def main():
    args = parse_args();
    basename, ext = os.path.splitext(args.output_mesh);

    mesh = pymesh.load_mesh(args.input_mesh);
    mesh = generate_tet_mesh(mesh);
    pymesh.save_mesh("{}_source.msh".format(basename), mesh);

    bd_vertex_indices, bd_vertex_positions = map_boundary_to_sphere(
            mesh, basename);
    out_mesh = tutte_3D(mesh, bd_vertex_indices, bd_vertex_positions);

    tet_orientations = pymesh.get_tet_orientations(out_mesh);
    out_mesh.add_attribute("orientation");
    out_mesh.set_attribute("orientation", tet_orientations);
    pymesh.save_mesh(args.output_mesh, out_mesh, "orientation");

if __name__ == "__main__":
    main();