607_Picking.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. import sys, os
  2. # Add the igl library to the modules search path
  3. sys.path.insert(0, os.getcwd() + "/../")
  4. import pyigl as igl
  5. from shared import TUTORIAL_SHARED_PATH, check_dependencies
  6. dependencies = ["viewer"]
  7. check_dependencies(dependencies)
  8. # Mesh with per-face color
  9. V = igl.eigen.MatrixXd()
  10. F = igl.eigen.MatrixXi()
  11. C = igl.eigen.MatrixXd()
  12. viewer = igl.viewer.Viewer()
  13. def mouse_down(viewer, a, b):
  14. bc = igl.eigen.MatrixXd()
  15. # Cast a ray in the view direction starting from the mouse position
  16. fid = igl.eigen.MatrixXi([-1])
  17. coord = igl.eigen.MatrixXd([viewer.current_mouse_x, viewer.core.viewport[3] - viewer.current_mouse_y])
  18. hit = igl.unproject_onto_mesh(coord, viewer.core.view * viewer.core.model,
  19. viewer.core.proj, viewer.core.viewport, V, F, fid, bc)
  20. if hit:
  21. C.setRow(fid[0, 0], igl.eigen.MatrixXd([[1, 0, 0]]))
  22. viewer.data.set_colors(C)
  23. return True
  24. return False
  25. print("Usage: [LeftMouseClick] to select a face")
  26. # Load a mesh in OFF format
  27. igl.readOFF(TUTORIAL_SHARED_PATH + "fertility.off", V, F)
  28. # Initialize white
  29. C.setConstant(F.rows(), 3, 1.0)
  30. viewer.data.set_mesh(V, F)
  31. viewer.data.set_colors(C)
  32. viewer.core.show_lines = False
  33. viewer.callback_mouse_down = mouse_down
  34. viewer.launch()