| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- import sys, os
- # Add the igl library to the modules search path
- sys.path.insert(0, os.getcwd() + "/../")
- import pyigl as igl
- from shared import TUTORIAL_SHARED_PATH, check_dependencies
- dependencies = ["tetgen", "viewer"]
- check_dependencies(dependencies)
- # Input polygon
- V = igl.eigen.MatrixXd()
- F = igl.eigen.MatrixXi()
- B = igl.eigen.MatrixXd()
- # Tetrahedralized interior
- TV = igl.eigen.MatrixXd()
- TT = igl.eigen.MatrixXi()
- TF = igl.eigen.MatrixXi()
- viewer = igl.viewer.Viewer()
- def key_down(viewer, key, modifier):
- if key >= ord('1') and key <= ord('9'):
- t = float((key - ord('1')) + 1) / 9.0
- v = igl.eigen.MatrixXd()
- v = B.col(2) - B.col(2).minCoeff()
- v /= v.col(0).maxCoeff()
- s = []
- for i in range(v.size()):
- if v[i, 0] < t:
- s.append(i)
- V_temp = igl.eigen.MatrixXd(len(s) * 4, 3)
- F_temp = igl.eigen.MatrixXi(len(s) * 4, 3)
- for i in range(len(s)):
- V_temp.setRow(i * 4 + 0, TV.row(TT[s[i], 0]))
- V_temp.setRow(i * 4 + 1, TV.row(TT[s[i], 1]))
- V_temp.setRow(i * 4 + 2, TV.row(TT[s[i], 2]))
- V_temp.setRow(i * 4 + 3, TV.row(TT[s[i], 3]))
- F_temp.setRow(i * 4 + 0, igl.eigen.MatrixXi([[(i*4)+0, (i*4)+1, (i*4)+3]]))
- F_temp.setRow(i * 4 + 1, igl.eigen.MatrixXi([[(i*4)+0, (i*4)+2, (i*4)+1]]))
- F_temp.setRow(i * 4 + 2, igl.eigen.MatrixXi([[(i*4)+3, (i*4)+2, (i*4)+0]]))
- F_temp.setRow(i * 4 + 3, igl.eigen.MatrixXi([[(i*4)+1, (i*4)+2, (i*4)+3]]))
- viewer.data.clear()
- viewer.data.set_mesh(V_temp, F_temp)
- viewer.data.set_face_based(True)
- else:
- return False
- return True
- # Load a surface mesh
- igl.readOFF(TUTORIAL_SHARED_PATH + "fertility.off", V, F)
- # Tetrahedralize the interior
- igl.tetgen.tetrahedralize(V, F, "pq1.414Y", TV, TT, TF)
- # Compute barycenters
- igl.barycenter(TV, TT, B)
- # Plot the generated mesh
- key_down(viewer, ord('5'), 0)
- viewer.callback_key_down = key_down
- viewer.launch()
|
