diamond_ore.py

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

scene = Scene(0, 2)
scene.set_background_color((0.2, 0.2, 0.2))
scene.set_floor(-1, (1.2, 1.2, 1.2))
scene.set_directional_light((1, 0.5, -1), 0.3, (0.85, 0.9, 0.9))

@ti.func
def set(idx, mat, color = vec3(0), noise = vec3(0)): scene.set_voxel(idx, mat, color + ti.random() * noise)
@ti.func
def get(idx): return scene.get_voxel(idx)

@ti.func
def initializeOre():
    ti.loop_config(serialize = True)
    for _ in range(256):
        d = ti.cast(4 * (ti.random() ** 2) + 2, ti.i32)
        x0, y0 = ti.cast((13 - d) * ti.random() + 2, ti.i32), ti.cast(12 * ti.random() + 2, ti.i32)
        ok = True
        for x, y in ti.ndrange((x0 - 1, x0 + d + 1), (y0 - 1, y0 + 2)):
            if get(ivec3(x, 63, y))[0] != 0: ok = False
        if ok:
            for x in range(x0, x0 + d): set(ivec3(x, 63, y0), 1)
    for x, y in ti.ndrange((2, 14), (2, 14)):
        if ti.random() < 0.1 and get(ivec3(x - 1, 63, y))[0] + get(ivec3(x + 1, 63, y))[0] == 0:
            set(ivec3(x, 63, y), 1)

@ti.func
def initializePickaxe():
    set(ivec3(0, 62, 0), 1)
    for i in range(11): set(ivec3(i + 1, 62, i), 1); set(ivec3(i, 62, i + 1), 2); set(ivec3(i + 1, 62, i + 1), 3)
    for i in range(20):
        a = i / 9 * 0.25 * pi
        x, y = ti.min(ti.cast(8 * ti.cos(a), ti.i32), 7), ti.min(ti.cast(8 * ti.sin(a), ti.i32), 7)
        set(ivec3(4 + x, 62, 4 + y), 5); set(ivec3(4 + y, 62, 4 + x), 5)
    for y, x in ti.ndrange(13, 13):
        if get(ivec3(x, 62, y))[0] == 0 and (get(ivec3(x - 1, 62, y))[0] == 5 or get(ivec3(x + 1, 62, y))[0] == 5
            or get(ivec3(x, 62, y - 1))[0] == 5 or get(ivec3(x, 62, y + 1))[0] == 5): set(ivec3(x, 62, y), 4)

@ti.kernel
def initialize():
    for _ in range(400):
        size, mat = 2 if ti.random() < 0.2 else 1, 2 if ti.random() < 0.5 else 1
        x, z = ti.cast(24 * ti.randn() - 8, ti.i32), ti.cast(24 * ti.randn() + 8, ti.i32)
        x, z = ti.min(ti.max(x, -64), 64 - size), ti.min(ti.max(z, -64), 64 - size)
        for p in ti.grouped(ti.ndrange(size, size, size)):
            set(ivec3(x, -64, z) + p, mat, vec3(0.2, 0.7, 0.6), vec3(0.1))
    initializeOre()
    c, s = ivec3(-40, -64, -24), 4
    for p in ti.grouped(ti.ndrange((c.x + s - 1, c.x + 15 * s + 1), (c.y + s - 1, c.y + 15 * s + 1),
        (c.z + s - 1, c.z + 15 * s + 1))): set(p, 2, vec3(0.2, 0.7, 0.6))
    for x, y in ti.ndrange(16, 16):
        if get(ivec3(x, 63, y))[0] == 0:
            color, noise = vec3(0.2) + ti.random() * vec3(0.1), vec3(0.05)
            for p in ti.grouped(ti.ndrange(s, s, s)):
                set(c + ivec3(s * (15 - x), s * (15 - y), 0) + p, 1, color, noise)
                set(c + ivec3(s * x, s * (15 - y), 15 * s) + p, 1, color, noise)
                set(c + ivec3(s * (15 - x), 0, s * y) + p, 1, color, noise)
                set(c + ivec3(s * x, 15 * s, s * y) + p, 1, color, noise)
                set(c + ivec3(0, s * (15 - y), s * x) + p, 1, color, noise)
                set(c + ivec3(15 * s, s * (15 - y), s * (15 - x)) + p, 1, color, noise)
    initializePickaxe()
    c, s = ivec3(0, -18, -56), 4
    for x, y in ti.ndrange(13, 13):
        d = get(ivec3(x, 62, y))[0]
        if d != 0:
            palette = (vec3(0.1, 0.08, 0.05) if d == 1 else vec3(0.15, 0.1, 0.05) if d == 2
                else vec3(0.5, 0.4, 0.2) if d == 3 else vec3(0, 0.1, 0.1) if d == 4 else vec3(0.2, 0.7, 0.6))
            mat, color, noise = 2 if d == 5 else 1, palette + ti.random() * vec3(0.05), vec3(0.05)
            for p in ti.grouped(ti.ndrange(s, s, s)):
                set(c + ivec3(0, s * y, s * x) + p, mat, color, noise)
    for p in ti.grouped(ti.ndrange((-64, 64), (62, 64), (-64, 64))): set(p, 0)

# scene.camera._camera_pos[:] = [4.25, 1.25, 2.75]; scene.camera._lookat_pos[:] = [-8 / 64, -24 / 64, 8 / 64]
# scene.renderer.set_camera_pos(*scene.camera.position); scene.renderer.set_look_at(*scene.camera.look_at)
initialize(); scene.finish()