bsavery/ray-tracing-one-weekend-taichi

A fast python implementation of Ray Tracing in One Weekend using python and Taichi

ray-tracing-one-weekend-taichi

A fast python implementation of Ray Tracing in One Weekend using python and Taichi.

Taichi is a simple "Domain specific language" to use mostly pure python and run on the GPU. There are other solutions for doing similar things, but my interest in Taichi was it's wide support for platforms, and this was an expirement to see and learn it. Learn more at Taichi.

Goes up to Ray tracing "The next week" with the BVH implementation. I tried to keep the implementation fairly generic to the original text so others can follow along. There are some details that have to be "vectorized" for Taichi. For example, rather than a list of Sphere objects, we have a World class which has a list of sphere centers and radii.

To run you will have to do pip install taichi

Some notes / thoughts / TODOs.

• I have not implemented more complex memory layouts from taichi such as sparse layouts.
• More "microkernel" architectures (rather than one big render loop) seem to be slower. But maybe could be optimized. I tried to have the main kernel do 1 piece of work each pass.
• Only tested really on macOS. But performance looks good on metal gpu! Vulkan and CUDA would be interesting to compare. For reference this runs in ~9.25 sec on a macbook pro with an AMD 5500M GPU.

ray-tracing-one-weekend-taichi's Issues

A small mistake in camera.py

Thanks for your sharing these codes!

However, I think there might be a small mistake in the camera.py.

ray-tracing-one-weekend-taichi/camera.py

Line 30 in 48f887a

offset = u * rd.x + v * rd.y

It should be offset = self.u * rd.x + self.v * rd.y. Here, self.u and self.v are from u, v in __ini__()

Please let me know if you have any questions!

TypeError: build() takes 1 positional argument but 2 were given

python main.py

[Taichi] mode=release
[Taichi] version 0.7.29, llvm 10.0.0, commit 7b40a962, win, python 3.7.7
[TaiGLSL] version 0.0.11
[Taichi] Starting on arch=cuda
Traceback (most recent call last):
  File "main.py", line 78, in <module>
    world.commit()
  File "C:\DEV_PROJECT\dev-repository\ray-tracing-one-weekend-taichi\hittable.py", line 83, in commit
    self.bvh.build(self.spheres)
TypeError: build() takes 1 positional argument but 2 were given