PhantomGaze: A GPU-Accelerated Rendering Engine for Scientific Visualization

Introduction

PhantomGaze is a Python package that provides a basic rendering engine optimized for scientific computing visualizations. Utilizing the power of Numba, it enables efficient operations directly on the GPU, enhancing the visualization of complex scientific data. This package is particularly useful for rendering volumetric data, geometric shapes, and contour visualizations with ease.

Features

• GPU-accelerated rendering engine
• Volumetric rendering
• Geometric rendering
• Contour rendering
• Customizable color maps
• Integration with GPU libraries such as CuPy, JAX, PyTorch, and Warp.

Installation

PhantomGaze can be installed using pip:

pip install .

(TODO: Make package available on PyPI)

Usage

There are several examples in the examples directory. A minimal example of a contour plot is shown below:

import cupy as cp
import matplotlib.pyplot as plt

import phantomgaze as pg

# Create SDF feild of a sphere using cupy
lin = cp.linspace(-1, 1, 256)
X, Y, Z = cp.meshgrid(lin, lin, lin, indexing="ij")
sphere = -(cp.sqrt(X**2 + Y**2 + Z**2) - 1.0)
sphere_volume = pg.objects.Volume(
    sphere, spacing=(2 / 256, 2 / 256, 2 / 256), origin=(-1.0, -1.0, -1.0)
)
color_volume = pg.objects.Volume(
    cp.sin(X * 2 * cp.pi) * cp.sin(Y * 2 * cp.pi) * cp.sin(Z * 2 * cp.pi),
    spacing=(2 / 256, 2 / 256, 2 / 256),
    origin=(-1.0, -1.0, -1.0),
)

# Create camera object
camera = pg.Camera(position=(2.0, 1.0, -4.0), focal_point=(0.0, 0.0, 0.0), view_up=(0.0, 1.0, 0.0))

# Render the contour of the inner sphere
screen_buffer = pg.render.contour(sphere_volume, camera, threshold=0.0, color=color_volume)

# Show the rendered image
plt.imshow(screen_buffer.image.get())
plt.show()

This produces the following image:

Gallery

The following images were generated using the examples in the examples directory.