301_Slice.py 995 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. from __future__ import print_function
  2. # Add the igl library to the modules search path
  3. import sys, os
  4. sys.path.insert(0, os.getcwd() + "/../")
  5. import pyigl as igl
  6. V = igl.eigen.MatrixXd()
  7. F = igl.eigen.MatrixXi()
  8. igl.readOFF("../../tutorial/shared/decimated-knight.off",V,F)
  9. # 100 random indicies into rows of F
  10. I = igl.eigen.MatrixXi()
  11. igl.floor((0.5*(igl.eigen.MatrixXd.Random(100,1)+1.)*F.rows()),I);
  12. # 50 random indicies into rows of I
  13. J = igl.eigen.MatrixXi()
  14. igl.floor((0.5*(igl.eigen.MatrixXd.Random(50,1)+1.)*I.rows()),J)
  15. # K = I(J);
  16. K = igl.eigen.MatrixXi()
  17. igl.slice(I,J,K)
  18. # default green for all faces
  19. #C = p2e(np.array([[0.4,0.8,0.3]])).replicate(F.rows(),1)
  20. C = igl.eigen.MatrixXd([[0.4,0.8,0.3]]).replicate(F.rows(),1)
  21. # Red for each in K
  22. R = igl.eigen.MatrixXd([[1.0,0.3,0.3]]).replicate(K.rows(),1)
  23. # C(K,:) = R
  24. igl.slice_into(R,K,1,C)
  25. # Plot the mesh with pseudocolors
  26. viewer = igl.viewer.Viewer()
  27. viewer.data.set_mesh(V, F)
  28. viewer.data.set_colors(C)
  29. viewer.launch()