Interactive raymarching in JAX. Toy project to learn about raymarching.
Try it out using colab:
Here are three examples of rendered scenes and their corresponding normals:
While in the Dash app, the scene lighting can be changed by clicking on the rendered scene.
The scene can also be modified while in the app, it is represented in a yaml format. Here is an example of a scene with a sphere, a box and a plane:
width: 200
height: 200
smoothing: 0.07
light_dir: [0.3, 1, -1]
Camera:
position: [1, 1, 0]
target: [0, 0.5, 0]
up: [0, 1, 0]
fov: 0.6
Objects:
- Sphere:
position: [0, 0.7, 0]
attribute: [0.1, 0, 0]
rotation: [6, 1, 0]
color: [1, 1, 0.5]
- Box:
position: [0, 0.3, 0]
attribute: [0.1, 0.2, 0.3]
rotation: [0, 0.5, 0]
color: [1, 0.4, 1]
rounding: 0.04
- Plane:
attribute: [0, 1, 0]
color: [1, 1, 1]Install python dependencies:
pip install -r requirements/requirements.txt
Run the app:
python app.py
The repository is structured as follows:
├── app.py # Main file to run the Dash app.
├── raymarch.py # Raymarching JAX code.
├── builder.py # Code to build the scene from a yaml file.
├── tests.py # Tests the scene building and raymarching.
├── utils # Miscellanous utility functions.
│ ├── plot.py # Plotting functions.
│ └── linalg.py # Linear algebra functions.
├── scenes # Yaml files describing the scenes.
├── assets # Assets for the app and the README.
├── requirements # Python requirements.
├── notebooks # Jupyter notebooks.
│ └── colab.ipynb # Notebook to try out the app in colab.
└── README.md # This file.