SLOptixAccelStruct.cpp

Go to the documentation of this file.

/**
 * \file      SLOptixAccelStruct.cpp
 * \authors   Nic Dorner
 * \date      October 2019
 * \authors   Nic Dorner
 * \copyright http://opensource.org/licenses/GPL-3.0
 * \remarks   Please use clangformat to format the code. See more code style on
 *            https://github.com/cpvrlab/SLProject4/wiki/SLProject-Coding-Style
*/
 
#ifdef SL_HAS_OPTIX
#    include <SLOptix.h>
#    include <SLOptixAccelStruct.h>
#    include <SLOptixRaytracer.h>
 
//-----------------------------------------------------------------------------
SLOptixAccelStruct::SLOptixAccelStruct()
{
    _accelBuildOptions.buildFlags = OPTIX_BUILD_FLAG_ALLOW_UPDATE |
                                    OPTIX_BUILD_FLAG_ALLOW_RANDOM_VERTEX_ACCESS |
                                    OPTIX_BUILD_FLAG_PREFER_FAST_TRACE;
    _buffer = new SLOptixCudaBuffer<void>();
}
//-----------------------------------------------------------------------------
SLOptixAccelStruct::~SLOptixAccelStruct()
{
    delete _buffer;
}
//-----------------------------------------------------------------------------
void SLOptixAccelStruct::buildAccelerationStructure()
{
    OptixDeviceContext context = SLOptix::context;
 
    _accelBuildOptions.operation = OPTIX_BUILD_OPERATION_BUILD;
 
    OPTIX_CHECK(optixAccelComputeMemoryUsage(
      context,
      &_accelBuildOptions,
      &_buildInput,
      1, // num_build_inputs
      &_accelBufferSizes));
 
    SLOptixCudaBuffer<void> temp_buffer = SLOptixCudaBuffer<void>();
    temp_buffer.alloc(_accelBufferSizes.tempSizeInBytes);
 
    // non-compacted output
    _buffer = new SLOptixCudaBuffer<void>();
    _buffer->alloc(_accelBufferSizes.outputSizeInBytes);
 
    SLOptixCudaBuffer<OptixAabb> aabbBuffer = SLOptixCudaBuffer<OptixAabb>();
    aabbBuffer.alloc(sizeof(OptixAabb));
    SLOptixCudaBuffer<size_t> compactedSize = SLOptixCudaBuffer<size_t>();
    compactedSize.alloc(sizeof(size_t));
 
    OptixAccelEmitDesc emitProperty[1];
    emitProperty[0].type   = OPTIX_PROPERTY_TYPE_AABBS;
    emitProperty[0].result = aabbBuffer.devicePointer();
 
    OPTIX_CHECK(optixAccelBuild(
      context,
      SLOptix::stream, // CUDA stream
      &_accelBuildOptions,
      &_buildInput,
      1, // num build inputs
      temp_buffer.devicePointer(),
      _accelBufferSizes.tempSizeInBytes,
      _buffer->devicePointer(),
      _accelBufferSizes.outputSizeInBytes,
      &_handle,
      emitProperty, // emitted property list
      1             // num emitted properties
      ));
    CUDA_SYNC_CHECK(SLOptix::stream);
 
    OptixAabb aabb;
    aabbBuffer.download(&aabb);
}
//-----------------------------------------------------------------------------
void SLOptixAccelStruct::updateAccelerationStructure()
{
    OptixDeviceContext context = SLOptix::context;
 
    _accelBuildOptions.operation = OPTIX_BUILD_OPERATION_UPDATE;
 
    SLOptixCudaBuffer<void> temp_buffer = SLOptixCudaBuffer<void>();
    temp_buffer.alloc(_accelBufferSizes.tempUpdateSizeInBytes);
 
    OPTIX_CHECK(optixAccelBuild(
      context,
      SLOptix::stream, // CUDA stream
      &_accelBuildOptions,
      &_buildInput,
      1, // num build inputs
      temp_buffer.devicePointer(),
      _accelBufferSizes.tempUpdateSizeInBytes,
      _buffer->devicePointer(),
      _accelBufferSizes.outputSizeInBytes,
      &_handle,
      nullptr, // emitted property list
      0        // num emitted properties
      ));
}
//-----------------------------------------------------------------------------
#endif // SL_HAS_OPTIX