import k3d
from math import sqrt, sin, cos
width = 100
height = 100
length = 100
def r(x, y, z):
r = sqrt((x - width / 2) * (x - width / 2) + (y - height / 2) * (y - height / 2) + (z - length / 2) * (z - length / 2))
r += sin(x / 2) * 3
r += cos(y / 10) * 5
return r
def f(x, y, z):
return 0 if r(x, y, z) > width / 2 else (1 if y + sin(x / 20) * 10 > height / 2 else 2)
color_map = (0xffff00, 0xff0000)
voxels = [[[f(x, y, z) for x in range(width)] for y in range(height)] for z in range(length)]
# add red voxels in the corners, just for
for x in [0, -1]:
for y in [0, -1]:
for z in [0, -1]:
voxels[z][y][x] = 2
plot = k3d.plot()
obj = k3d.voxels(voxels, color_map, compression_level=1)
plot += obj
plot.display()