Fully working dynamic materials example

Author: keckj

examples/dynamic_geometry.py

@@ -1,20 +1,21 @@
-import os, sys, enum, copy, logging, collections
+import os, sys, enum, logging, collections
 
-import numpy as np
 import cupy as cp
+import numpy as np
 import optix as ox
+
 import glfw, imgui
 
 from optix.sutil.gui import init_ui, display_stats
 from optix.sutil.gl_display import GLDisplay
 from optix.sutil.trackball import Trackball, TrackballViewMode
 from optix.sutil.cuda_output_buffer import CudaOutputBuffer, CudaOutputBufferType, BufferImageFormat
 
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
 logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 log = logging.getLogger()
 
-script_dir = os.path.dirname(os.path.abspath(__file__))
-
 DEBUG=False
 
 #------------------------------------------------------------------------------
@@ -83,11 +84,13 @@ def camera(self):
     def launch_dimensions(self):
         return (int(self.params.width), int(self.params.height))
 
+
 class AnimationMode(enum.Enum):
     NONE = 0
     DEFORM = 1
     EXPLODE = 2
 
+
 #------------------------------------------------------------------------------
 # Scene data
 #------------------------------------------------------------------------------

examples/dynamic_materials.py

@@ -1,19 +1,22 @@
 import os, sys, logging, collections
 
-import numpy as np
 import cupy as cp
+import numpy as np
 import optix as ox
+
 import glfw, imgui
 
 from optix.sutil.gui import init_ui, display_text
 from optix.sutil.camera import Camera
 from optix.sutil.gl_display import GLDisplay
 from optix.sutil.cuda_output_buffer import CudaOutputBuffer, CudaOutputBufferType, BufferImageFormat
 
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
 logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 log = logging.getLogger()
 
-script_dir = os.path.dirname(os.path.abspath(__file__))
+DEBUG=False
 
 
 class Params:
@@ -174,13 +177,18 @@ def build_ias(state):
     state.params.trav_handle = state.ias.handle
 
 def create_module(state):
+    if DEBUG:
+        exception_flags=ox.ExceptionFlags.DEBUG | ox.ExceptionFlags.TRACE_DEPTH | ox.ExceptionFlags.STACK_OVERFLOW,
+    else:
+        exception_flags=ox.ExceptionFlags.NONE
+
     pipeline_opts = ox.PipelineCompileOptions(
             uses_motion_blur=False,
             traversable_graph_flags=ox.TraversableGraphFlags.ALLOW_SINGLE_LEVEL_INSTANCING,
             uses_primitive_type_flags=ox.PrimitiveTypeFlags.CUSTOM,
             num_payload_values=3,
             num_attribute_values=3,
-            exception_flags=ox.ExceptionFlags.DEBUG | ox.ExceptionFlags.TRACE_DEPTH | ox.ExceptionFlags.STACK_OVERFLOW,
+            exception_flags=exception_flags,
             pipeline_launch_params_variable_name="params")
 
     compile_opts = ox.ModuleCompileOptions(
@@ -317,7 +325,7 @@ def update_sbt_header(state):
     # The right sphere will use the next compiled program group.
     material_index = state.material_index_2.nextval()
 
-    #state.hit_grps.update_program_group(3, state.hit_grps[3 + material_index])
+    state.hit_sbts.update_program_group(3, state.hit_grps[3 + material_index])
 
     state.sbt = ox.ShaderBindingTable(raygen_record=state.raygen_sbt, miss_records=state.miss_sbt,
             hitgroup_records=state.hit_sbts)

examples/hello.py

@@ -1,15 +1,20 @@
+import os, sys, logging
+
 import optix as ox
 import cupy as cp
 import numpy as np
+
 from PIL import Image, ImageOps
-import logging
-import sys
+
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
 logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 log = logging.getLogger()
 
 def create_module(ctx, pipeline_opts):
     compile_opts = ox.ModuleCompileOptions(debug_level=ox.CompileDebugLevel.LINEINFO)
-    module = ox.Module(ctx, 'cuda/hello.cu', compile_opts, pipeline_opts)
+    source = os.path.join(script_dir, 'cuda', 'hello.cu')
+    module = ox.Module(ctx, source, compile_opts, pipeline_opts)
     return module
 
 

examples/spheres.py

@@ -6,12 +6,12 @@
 
 from PIL import Image, ImageOps
 
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
 logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
 log = logging.getLogger()
 img_size = (1024, 768)
 
-script_dir = os.path.dirname(os.path.abspath(__file__))
-
 def compute_spheres_bbox(centers, radii):
     out = cp.empty((centers.shape[0], 6), dtype='f4')
     out[:, :3] = centers - radii.reshape(-1, 1)

examples/triangle.py

@@ -1,8 +1,13 @@
-import optix as ox
+import os
+
 import cupy as cp
 import numpy as np
+import optix as ox
+
 from PIL import Image, ImageOps
 
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
 img_size = (1024, 768)
 
 # use a regular function for logging
@@ -19,7 +24,8 @@ def create_acceleration_structure(ctx, vertices):
 
 def create_module(ctx, pipeline_opts):
     compile_opts = ox.ModuleCompileOptions(debug_level=ox.CompileDebugLevel.LINEINFO)
-    module = ox.Module(ctx, 'cuda/triangle.cu', compile_opts, pipeline_opts)
+    source = os.path.join(script_dir, 'cuda', 'triangle.cu')
+    module = ox.Module(ctx, source, compile_opts, pipeline_opts)
     return module
 
 
@@ -118,5 +124,3 @@ def launch_pipeline(pipeline : ox.Pipeline, sbt, gas):
     img = img.reshape(img_size[1], img_size[0], 4)
     img = ImageOps.flip(Image.fromarray(img, 'RGBA'))
     img.show()
-
-

optix/shader_binding_table.pyx

@@ -111,4 +111,3 @@ cdef class ShaderBindingTable(OptixObject):
             return array_to_device_memory(record, stream=stream), record.shape[0], record.strides[0]
         else:
             raise ValueError(f"Unsupported record type '{type(record)}'")
-