Function tutte_3D

scripts/map_to_sphere.py:84–99

(mesh, bd_vertex_indices, bd_vertex_positions)

    return x;

def tutte_3D(mesh, bd_vertex_indices, bd_vertex_positions):
    assembler = pymesh.Assembler(mesh);
    L = assembler.assemble("laplacian");

    is_constraint = np.zeros(mesh.num_vertices, dtype=bool);
    is_constraint[bd_vertex_indices] = True;
    is_variable = np.logical_not(is_constraint);

    M, rhs = init_harmonic_system(L, bd_vertex_indices, bd_vertex_positions, is_variable);
    x = solve(M, rhs, "SparseLU", 1e-6);

    out_vertices = np.zeros((mesh.num_vertices, 3));
    out_vertices[is_constraint,:] = bd_vertex_positions;
    out_vertices[is_variable,:] = x;

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

def main():
    args = parse_args();