| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677 |
- 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 = ["png", "viewer"]
- check_dependencies(dependencies)
- def key_down(viewer, key, modifier):
- if key == ord('1'):
- # Allocate temporary buffers
- R = igl.eigen.MatrixXuc(1280, 800)
- G = igl.eigen.MatrixXuc(1280, 800)
- B = igl.eigen.MatrixXuc(1280, 800)
- A = igl.eigen.MatrixXuc(1280, 800)
- # Draw the scene in the buffers
- viewer.core.draw_buffer(viewer.data, viewer.opengl, False, R, G, B, A)
- # Save it to a PNG
- igl.png.writePNG(R, G, B, A, "out.png")
- elif key == ord('2'):
- # Allocate temporary buffers
- R = igl.eigen.MatrixXuc()
- G = igl.eigen.MatrixXuc()
- B = igl.eigen.MatrixXuc()
- A = igl.eigen.MatrixXuc()
- # Read the PNG
- igl.png.readPNG("out.png", R, G, B, A)
- # Replace the mesh with a triangulated square
- V = igl.eigen.MatrixXd([[-0.5, -0.5, 0],
- [0.5, -0.5, 0],
- [0.5, 0.5, 0],
- [-0.5, 0.5, 0]])
- F = igl.eigen.MatrixXi([[0, 1, 2], [2, 3, 0]])
- UV = igl.eigen.MatrixXd([[0, 0], [1, 0], [1, 1], [0, 1]])
- viewer.data.clear()
- viewer.data.set_mesh(V, F)
- viewer.data.set_uv(UV)
- viewer.core.align_camera_center(V)
- viewer.core.show_texture = True
- # Use the image as a texture
- viewer.data.set_texture(R, G, B)
- else:
- return False
- return True
- if __name__ == "__main__":
- V = igl.eigen.MatrixXd()
- F = igl.eigen.MatrixXi()
- # Load meshes in OFF format
- igl.readOFF(TUTORIAL_SHARED_PATH + "bunny.off", V, F)
- viewer = igl.viewer.Viewer()
- print(
- "Usage: Press 1 to render the scene and save it in a png. \nPress 2 to load the saved png and use it as a texture.")
- viewer.callback_key_down = key_down
- viewer.data.set_mesh(V, F)
- viewer.launch()
- os.remove("out.png")
|
