201_Normals.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. import igl
  2. V = igl.eigen.MatrixXd()
  3. F = igl.eigen.MatrixXi()
  4. N_vertices = igl.eigen.MatrixXd()
  5. N_faces = igl.eigen.MatrixXd()
  6. N_corners = igl.eigen.MatrixXd()
  7. # This function is called every time a keyboard button is pressed
  8. def key_pressed(viewer, key, modifier):
  9. if key == ord('1'):
  10. viewer.data.set_normals(N_faces)
  11. return True
  12. elif key == ord('2'):
  13. viewer.data.set_normals(N_vertices)
  14. return True
  15. elif key == ord('3'):
  16. viewer.data.set_normals(N_corners)
  17. return True
  18. return False
  19. # Load a mesh in OFF format
  20. igl.readOFF("../tutorial/shared/fandisk.off", V, F);
  21. # Compute per-face normals
  22. N_faces = igl.eigen.MatrixXd()
  23. igl.per_face_normals(V,F,N_faces)
  24. # Compute per-vertex normals
  25. N_vertices = igl.eigen.MatrixXd()
  26. igl.per_vertex_normals(V,F,igl.PER_VERTEX_NORMALS_WEIGHTING_TYPE_AREA,N_vertices)
  27. # Compute per-corner normals, |dihedral angle| > 20 degrees --> crease
  28. N_corners = igl.eigen.MatrixXd()
  29. igl.per_corner_normals(V,F,20,N_corners)
  30. # Plot the mesh
  31. viewer = igl.viewer.Viewer()
  32. viewer.callback_key_pressed = key_pressed
  33. viewer.core.show_lines = False
  34. viewer.data.set_mesh(V, F)
  35. viewer.data.set_normals(N_faces)
  36. print("Press '1' for per-face normals.")
  37. print("Press '2' for per-vertex normals.")
  38. print("Press '3' for per-corner normals.")
  39. viewer.launch()