Taichi THREE

Taichi THREE is an extension library of the Taichi Programming Language, that helps rendering your 3D scenes into nice-looking 2D images to display in GUI.

Installation

Install Taichi THREE with pip:

# Python 3.6/3.7/3.8 (64-bit)
pip install taichi_three

This should also install its dependencies taichi and taichi_glsl as well.

How to play

First, import Taichi and Taichi THREE:

import taichi as ti
import taichi_three as t3

ti.init(ti.gpu)

---

Then, create a scene using:

scene = t3.Scene()

NOTE: scene creation should be put before any kernel invocation or host access, i.e. before materialization, so that Scene.__init__ could define its internal tensors without an error. TL;DR: Put this line as forward as possible! Ideally right below ti.init().

---

After that, define your data tensors, physics info, and pass them to scene.

pos = ti.Vector(3, ti.f32, n)
radius = ti.var(ti.f32, n)

scene.add_ball(pos, radius)      # pass the tensors directly, not pos[...]!
scene.set_light_dir([1, 2, -2])  # scene will do normalization for you :)

---

Finally, create a GUI. And here goes the main loop:

gui = ti.GUI('Ball')

while gui.running:

    ... # Update ball pos, vel, etc.

    scene.render()  # update image according to the given tensors in add_ball
    gui.set_image(scene.img)  # to display the result image
    gui.show()

Example

Running the following example code should gives you an moving ball:

import taichi as ti
import taichi_three as t3
from math import cos, sin
from time import time
ti.init(ti.opengl)

scene = t3.Scene()
pos = ti.Vector(3, ti.f32, 1)
radius = ti.var(ti.f32, 1)

scene.add_ball(pos, radius)
scene.set_light_dir([1, 2, -2])

radius[0] = 0.5

gui = ti.GUI('Ball')
while gui.running:
    pos[0] = [0.3 * sin(time()), 0.3 * cos(time()), 0]
    scene.render()
    gui.set_image(scene.img)
    gui.show()