| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107 |
- import igl
- bc_frac = 1.0
- bc_dir = -0.03
- deformation_field = False
- V = igl.eigen.MatrixXd()
- U = igl.eigen.MatrixXd()
- V_bc = igl.eigen.MatrixXd()
- U_bc = igl.eigen.MatrixXd()
- # Z = igl.eigen.MatrixXd()
- F = igl.eigen.MatrixXi()
- b = igl.eigen.MatrixXi()
- def pre_draw(viewer):
- global V, bc_frac, bc_dir, U, U_bc, V_bc, b, deformation_field
- # Determine boundary conditions
- if (viewer.core.is_animating):
- bc_frac += bc_dir
- bc_dir *= (-1.0 if bc_frac>=1.0 or bc_frac <= 0.0 else 1.0)
- U_bc_anim = V_bc+bc_frac*(U_bc-V_bc)
- if (deformation_field):
- D = igl.eigen.MatrixXd()
- D_bc = U_bc_anim - V_bc
- igl.harmonic(V,F,b,D_bc,2,D)
- U = V+D;
- else:
- D = igl.eigen.MatrixXd()
- igl.harmonic(V,F,b,U_bc_anim,2,D)
- U = D
- # global U
- # igl.harmonic(V,F,b,U_bc_anim,2,U)
- viewer.data.set_vertices(U)
- viewer.data.compute_normals()
- return False
- def key_down(viewer, key, mods):
- global deformation_field
- if key == ord(' '):
- viewer.core.is_animating = not viewer.core.is_animating
- return True
- if key == ord('D') or key == ord('d'):
- deformation_field = not deformation_field;
- return True
- return False
- igl.readOBJ("../tutorial/shared/decimated-max.obj",V,F)
- U=V
- # S(i) = j: j<0 (vertex i not in handle), j >= 0 (vertex i in handle j)
- S = igl.eigen.MatrixXd()
- igl.readDMAT("../tutorial/shared/decimated-max-selection.dmat",S)
- S = S.castint()
- b = igl.eigen.MatrixXi([[t[0] for t in [(i,S[i]) for i in range(0,V.rows())] if t[1] >= 0]]).transpose()
- # Boundary conditions directly on deformed positions
- U_bc.resize(b.rows(),V.cols())
- V_bc.resize(b.rows(),V.cols())
- for bi in range(0,b.rows()):
- V_bc.setRow(bi,V.row(b[bi]))
- if (S[b[bi]] == 0):
- # Don't move handle 0
- U_bc.setRow(bi,V.row(b[bi]))
- elif S[b[bi]] == 1:
- # Move handle 1 down
- U_bc.setRow(bi,V.row(b[bi]) + igl.eigen.MatrixXd([[0,-50,0]]))
- else:
- # Move other handles forward
- U_bc.setRow(bi,V.row(b[bi]) + igl.eigen.MatrixXd([[0,0,-25]]))
- # Pseudo-color based on selection
- C = igl.eigen.MatrixXd(F.rows(),3)
- purple = igl.eigen.MatrixXd([[80.0/255.0,64.0/255.0,255.0/255.0]])
- gold = igl.eigen.MatrixXd([[255.0/255.0,228.0/255.0,58.0/255.0]])
- for f in range(0,F.rows()):
- if (S[F[f,0]])>=0 and S[F[f,1]]>=0 and S[F[f,2]]>=0:
- C.setRow(f,purple)
- else:
- C.setRow(f,gold)
- # Plot the mesh with pseudocolors
- viewer = igl.viewer.Viewer()
- viewer.data.set_mesh(U, F)
- viewer.core.show_lines = False
- viewer.data.set_colors(C)
- # viewer.core.trackball_angle = igl.eigen.Quaterniond(sqrt(2.0),0,sqrt(2.0),0)
- # viewer.core.trackball_angle.normalize()
- viewer.callback_pre_draw = pre_draw
- viewer.callback_key_down = key_down
- viewer.core.animation_max_fps = 30.0
- print("Press [space] to toggle deformation.")
- print("Press 'd' to toggle between biharmonic surface or displacements.")
- viewer.launch()
|
