def show(self, arguments: list = sys.argv[2:]): """Visualize an OBJ/NPZ model using Taichi THREE""" parser = argparse.ArgumentParser(prog='t3 show', description=f"{self.show.__doc__}") parser.add_argument( 'filename', help='File name of the OBJ/NPZ model to visualize, e.g. monkey.obj') parser.add_argument('-s', '--scale', default=0.75, type=float, help='Specify a scale parameter') parser.add_argument('-u', '--resx', default=512, type=int, help='Specify window width') parser.add_argument('-v', '--resy', default=512, type=int, help='Specify window height') parser.add_argument('-A', '--ambient', default=0, type=float, help='Specify ambient light strength') parser.add_argument('-o', '--ortho', action='store_true', help='Display in orthogonal mode') parser.add_argument('-l', '--lowp', action='store_true', help='Shade faces by interpolation') parser.add_argument('-x', '--flipx', action='store_true', help='Flip X axis of model when display') parser.add_argument('-y', '--flipy', action='store_true', help='Flip Y axis of model when display') parser.add_argument('-z', '--flipz', action='store_true', help='Flip Z axis of model when display') parser.add_argument('-f', '--flipface', action='store_true', help='Flip face culling direction') parser.add_argument('-F', '--flipnorm', action='store_true', help='Flip face normal direction') parser.add_argument('-b', '--bothface', action='store_true', help='Including both face, no culling') parser.add_argument('-N', '--renorm', action='store_true', help='Reset normal vectors to flat') parser.add_argument('-S', '--showhints', action='store_true', help='Show information about pixel under cursor') parser.add_argument('-T', '--taa', default=True, action='store_true', help='Enable temporal anti-aliasing') parser.add_argument('-t', '--texture', type=str, help='Path to texture to bind') parser.add_argument('-n', '--normtex', type=str, help='Path to normal map to bind') parser.add_argument('-m', '--metallic', type=str, help='Path to metallic map to bind') parser.add_argument('-r', '--roughness', type=str, help='Path to roughness map to bind') parser.add_argument('-a', '--arch', default='cpu', type=str, help='Backend to use for rendering') args = parser.parse_args(arguments) import taichi as ti import taichi_three as t3 import numpy as np ti.init(getattr(ti, args.arch)) scene = t3.Scene() obj = t3.readobj(args.filename, scale=args.scale if args.scale != 0 else 1) t3.objflipaxis(obj, args.flipx, args.flipy, args.flipz) if args.scale == 0: t3.objautoscale(obj) if args.flipface: t3.objflipface(obj) if args.flipnorm: t3.objflipnorm(obj) if args.renorm: t3.objmknorm(obj) if args.bothface: t3.objbothface(obj) model = (t3.ModelLow if args.lowp else t3.Model).from_obj(obj) if args.texture is not None: model.add_texture('color', ti.imread(args.texture)) if args.normtex is not None: model.add_texture('normal', ti.imread(args.normtex)) if args.metallic is not None: model.add_texture('metallic', ti.imread(args.metallic)) if args.roughness is not None: model.add_texture('roughness', ti.imread(args.roughness)) scene.add_model(model) camera = t3.Camera(res=(args.resx, args.resy), taa=args.taa) if args.showhints: camera.fb.add_buffer('pos', 3) camera.fb.add_buffer('texcoor', 2) camera.fb.add_buffer('normal', 3) if args.ortho: camera.type = camera.ORTHO scene.add_camera(camera) if args.ambient: light = t3.AmbientLight(args.ambient) else: light = t3.Light([0.4, -1.5, 0.8]) scene.add_light(light) gui = ti.GUI('Model', camera.res) while gui.running: gui.get_event(None) gui.running = not gui.is_pressed(ti.GUI.ESCAPE) camera.from_mouse(gui) scene.render() gui.set_image(camera.img) if args.showhints: coor = gui.get_cursor_pos() pos = camera.fb.fetchpixelinfo('pos', coor) color = camera.fb.fetchpixelinfo('img', coor) texcoor = camera.fb.fetchpixelinfo('texcoor', coor) normal = camera.fb.fetchpixelinfo('normal', coor) gui.text(f'color: [{color.x:.2f} {color.y:.2f} {color.z:.2f}]; pos: [{pos.x:+.2f} {pos.y:+.2f} {pos.z:+.2f}]', (0, 1)) gui.text(f'texcoor: [{texcoor.x:.2f} {texcoor.y:.2f}]; normal: [{normal.x:+.2f} {normal.y:+.2f} {normal.z:+.2f}]', (0, 1 - 16 / camera.res[1])) gui.show()
import taichi_three as t3 import taichi as ti N = 64 scene = t3.Scene() mesh = t3.DynamicMesh(n_faces=N, n_pos=N + 1) model = t3.Model(mesh) scene.add_model(model) camera = t3.Camera() scene.add_camera(camera) light = t3.AmbientLight(1.0) scene.add_light(light) @ti.kernel def init_mesh(): mesh.pos[N] = [0, 0, 0] for i in range(N): a = i / N * t3.tau mesh.pos[i] = [ti.sin(a), ti.cos(a), 0] mesh.faces[i] = [[i, 0, 0], [(i + 1) % N, 0, 0], [N, 0, 0]] init_mesh() gui = ti.GUI('Dynamic faces', camera.res) while gui.running: mesh.n_faces[None] = gui.frame % N scene.render()
ti.init(arch=ti.opengl) vol = np.load('assets/smoke.npy') mciso = MCISO(vol.shape[0], use_sparse=False) scene = t3.Scene() mesh = t3.DynamicMesh(n_faces=mciso.N_res, n_pos=mciso.N_res, n_nrm=mciso.N_res) model = t3.Model(mesh) scene.add_model(model) camera = t3.Camera() scene.add_camera(camera) scene.add_light(t3.Light([0.4, -1.5, -1.8], 0.8)) scene.add_light(t3.AmbientLight(0.22)) @ti.kernel def update_mesh(): mesh.n_faces[None] = mciso.Js_n[None] for i in range(mciso.Js_n[None]): for t in ti.static(range(3)): mesh.faces[i][t, 0] = mciso.Jts[i][t] mesh.faces[i][t, 2] = mciso.Jts[i][t] mesh.pos[i] = (mciso.vs[i] + 0.5) / mciso.N * 2 - 1 mesh.nrm[i] = mciso.ns[i] mciso.clear() mciso.m.from_numpy(vol * 4)
import taichi_three as t3 import numpy as np import time ti.init(ti.cpu) scene = t3.Scene() model = t3.Model(t3.Mesh.from_obj(t3.readobj('assets/torus.obj', scale=0.8))) scene.add_model(model) ball = t3.Model(t3.Mesh.from_obj(t3.readobj('assets/sphere.obj', scale=0.1))) ball.material = t3.Material( t3.BlinnPhong(emission=t3.Constant(t3.RGB(1.0, 1.0, 1.0)), )) scene.add_model(ball) camera = t3.Camera() camera.ctl = t3.CameraCtl(pos=[0, 1.8, 1.8]) scene.add_camera(camera) light = t3.PointLight(pos=[0, 1, 0]) scene.add_light(light) ambient = t3.AmbientLight(0.2) scene.add_light(ambient) gui = ti.GUI('Model', camera.res) while gui.running: gui.get_event(None) gui.running = not gui.is_pressed(ti.GUI.ESCAPE) camera.from_mouse(gui) light.pos[None].y = ti.cos(time.time()) ball.L2W[None] = t3.translate(light.pos[None].value) scene.render() gui.set_image(camera.img) gui.show()
import taichi as ti import taichi_three as t3 import numpy as np ti.init(ti.cpu) scene = t3.Scene() obj = t3.readobj('assets/torus.obj', scale=0.8) model = t3.Model(t3.Mesh.from_obj(obj)) model.material = t3.Material( t3.CookTorrance( color=t3.Texture(ti.imread('assets/cloth.jpg')), ambient=t3.Texture(ti.imread('assets/pattern.jpg')), )) scene.add_model(model) camera = t3.Camera() camera.ctl = t3.CameraCtl(pos=[0, 1, -1.8]) scene.add_camera(camera) ambient_light = t3.AmbientLight() scene.add_light(ambient_light) gui = ti.GUI('Model', camera.res) while gui.running: gui.get_event(None) gui.running = not gui.is_pressed(ti.GUI.ESCAPE) camera.from_mouse(gui) scene.render() gui.set_image(camera.img) gui.show()