flying_taichicken.py

from scene import Scene
import taichi as ti
from taichi.math import *

scene = Scene(voxel_edges=0, exposure=7)
scene.set_floor(-1, (1.0, 1.0, 1.0)) # height, color
scene.set_background_color((1, 1, 1))
scene.set_directional_light((.8,1,.8), 0.1, (.3, .3, .3)) # direction, noise, color

@ti.func
def sphere(pos, r, mat, color): # pos: vec3, r: i32, mat: 0/1, color: list
    for i,j,k in ti.ndrange((-64,64),(-64,64),(-64,64)):
        if (i-pos[0])**2 + (j-pos[1])**2 + (k-pos[2])**2 < r*r:
            scene.set_voxel(vec3(i,j,k), mat, color)

@ti.func
def box(pos, size, mat, color):
    for i,j,k in ti.ndrange((-64,64),(-64,64),(-64,64)):
        if ti.abs(i-pos[0]) <= size[0]/2 \
           and ti.abs(j-pos[1]) <= size[1]/2 \
           and ti.abs(k-pos[2]) <= size[2]/2:
            scene.set_voxel(vec3(i,j,k), mat, color)

@ti.func
def chicken(pos): # pos: vec3
    white = [.4,.4,.4]
    grey  = [.2,.2,.2]
    black = [.05,.05,.05]    
    red   = [.25,.0,.0]
    yellow= [.4,.3,.0]    
    # Body
    box(vec3(0,0,0)+pos, vec3(10,10,10), 1, white)
    box(vec3(0,0,10)+pos, vec3(10,10,10), 1, white)
    box(vec3(0,10,10)+pos, vec3(10,10,10), 1, white)
    # Wings
    # box(vec3(0,0,3)+pos, [15, 6, 15], 1, grey)  # Closed wings
    box(vec3(0,0,3)+pos, [25, 2, 15], 1, grey)  # Flying wings
    # Eyes
    box(vec3(0,12,12)+pos, [10,2,2], 1, black)
    # Head
    box(vec3(0,16,11)+pos, [2,2,6], 1, red)
    # Beak
    box(vec3(0,10,16)+pos, [2,6,6], 1, yellow)
    # Standing Legs
    # box(vec3(-4,-8,5)+pos, [1,6,2], 1, red)
    # box(vec3(3,-8,5)+pos, [1,6,2], 1, red)
    # box(vec3(-4,-12,6)+pos, [4,1,12], 1, red)
    # box(vec3(4,-12,6)+pos, [4,1,12], 1, red)        
    # Streching Legs
    box(vec3(-4,0,-6)+pos, [1,2,2], 1, red)
    box(vec3(3,0,-6)+pos, [1,2,2], 1, red)
    box(vec3(-4,0,-9)+pos, [4,1,4], 1, red)
    box(vec3(3,0,-9)+pos, [4,1,4], 1, red)        

@ti.func
def wave():
    for i,j,k in ti.ndrange((-64,64),(-64,64),(-64,64)):
        if k < 7*ti.sin(ti.cast(i, ti.f32)/50*3.14)*ti.sin(ti.cast(j, ti.f32)/45*3.14)-35:
            scene.set_voxel(vec3(i,k,j), 1, (0.3,.5,1.))            

@ti.func
def cloud(pos):
    sphere(vec3(0,0,0)+pos, 4, 1, [.8,.8,.8])
    sphere(vec3(0,0,-4)+pos, 4, 1, [.8,.8,.8])
    sphere(vec3(4,0,0)+pos, 4, 1, [.8,.8,.8])
    sphere(vec3(4,0,-4)+pos, 4, 1, [.8,.8,.8])
    sphere(vec3(0,0,4)+pos, 4, 1, [.8,.8,.8])
    sphere(vec3(4,0,4)+pos, 4, 1, [.8,.8,.8])    
    
            
@ti.kernel
def initialize_voxels():
    # Your code here! :-)
    # scene.set_voxel(vec3(0, 10, 0), 2, vec3(0.9, 0.1, 0.1))    # idx, mat, color
    chicken(vec3(25,-34,40))
    chicken(vec3(30,-40,-25))
    chicken(vec3(-50,-34,-30))
    chicken(vec3(-40,-34,45))

    chicken(vec3(-40,20,45))
    chicken(vec3(30,25,25))
    chicken(vec3(-28,30,-15))

    cloud(vec3(20,20,55))
    cloud(vec3(20,28,-55))
    cloud(vec3(-20,10,-55))    
    
    wave()

initialize_voxels()

scene.finish()